61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
# -*- coding: gbk -*-
|
||
import grpc
|
||
import qmt_service_pb2
|
||
import qmt_service_pb2_grpc
|
||
import time
|
||
|
||
class QmtServiceServicer(qmt_service_pb2_grpc.QmtServiceServicer):
|
||
"""实现QMT服务（单线程版本）"""
|
||
|
||
def GetAsset(self, request, context):
|
||
"""实现GetAsset方法"""
|
||
print(f"收到查询请求，账户: {request.account_id}")
|
||
|
||
# 这里是你调用大QMT API获取数据的逻辑
|
||
# 实际使用时，请替换为真实的xt_trader查询代码
|
||
# 参考: asset = xt_trader.query_stock_asset(acc)
|
||
|
||
# 模拟数据
|
||
total = 1000000.0
|
||
cash = 500000.0
|
||
market_val = 500000.0
|
||
|
||
# 模拟一些耗时操作（如查询数据库）
|
||
# time.sleep(0.1) # 如果需要可以取消注释
|
||
|
||
# 返回响应
|
||
return qmt_service_pb2.AssetResponse(
|
||
total_asset=total,
|
||
cash=cash,
|
||
market_value=market_val
|
||
)
|
||
|
||
def serve():
|
||
"""启动gRPC服务（单线程）"""
|
||
# 使用单线程服务器，通过设置maximum_concurrent_rpcs参数限制并发
|
||
# 或者使用同步服务器，直接处理请求
|
||
server = grpc.server()
|
||
|
||
# 注册服务
|
||
qmt_service_pb2_grpc.add_QmtServiceServicer_to_server(
|
||
QmtServiceServicer(),
|
||
server
|
||
)
|
||
|
||
# 监听端口
|
||
server.add_insecure_port('[::]:58051')
|
||
|
||
# 启动服务器
|
||
server.start()
|
||
print("QMT gRPC 服务已启动（单线程模式），监听端口 58051...")
|
||
print("所有请求将串行处理，不会并发执行")
|
||
|
||
# 保持服务运行
|
||
try:
|
||
server.wait_for_termination()
|
||
except KeyboardInterrupt:
|
||
print("\n服务已停止")
|
||
server.stop(0)
|
||
|
||
if __name__ == '__main__':
|
||
serve() |