B
|
ÚéJfõ= ã @ s d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl m
|
Z
|
m Z m Z ddl mZmZmZ dZdZdZd Zd
|
Zd Zd
|
Zd Zd
|
Zd Zd Zd Z d Z dZ dZ dZ dZ!G dd de"Z#G dd dZ$G dd dZ%e&dkrþe$ej'ej(dZ)e) *¡ e+ dS )u
|
å¤é¨æ¥å£ç®¡ç
|
é N)ÚSendResponseSkManager)Ú logger_systemÚlogger_request_apiÚprintlog)Úmiddle_api_protocolÚtoolÚ socket_utilÚheartÚcmdZ
|
trade_sellé é é é ÚtradeZ sell_ruleZrefresh_trade_dataÚget_code_position_infoZcommon_requestc @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) ÚActionCallbackc C s d S )N© )ÚselfÚ client_idÚ
|
request_idÚdatar r ú:D:\workspace\trade_cb\utils\outside_api_command_manager.pyÚOnTrade/ s zActionCallback.OnTradec C s d S )Nr )r r r r r r r Ú
|
OnSellRule3 s zActionCallback.OnSellRulec C s d S )Nr )r r r r r r r ÚOnRefreshTradeData6 s z!ActionCallback.OnRefreshTradeDatac C s d S )Nr )r r r r r r r ÚOnGetCodePositionInfo9 s z$ActionCallback.OnGetCodePositionInfoc C s d S )Nr )r r r r r r r ÚOnCommonRequest< s z ActionCallback.OnCommonRequestN)Ú__name__Ú
|
__module__Ú __qualname__r r r r r r r r r r - s
|
r c s¢ e Zd Zi Zi ZdZdZdZdZ fddZ e
|
dd Z e
|
dddZ e
|
dd
|
d Z e
|
d d Ze
|
dd Ze
|
dd Ze
|
dd Ze
|
dd ZdddZ ZS ) ÚApiCommandManagerr Nc s" | j s t j| f||| _ | j S )N)Ú _instanceÚsuperÚ__new__)ÚclsÚargsÚkwargs)Ú __class__r r r# K s zApiCommandManager.__new__c
|
C sR t t jt j¡}| | j¡ | t t dd|i|d¡
|
d¡¡¡ | d¡ |S )NÚregisterÚ client_type)Útyper Úridzutf-8i ) ÚsocketÚAF_INETÚ SOCK_STREAMÚconnectÚip_portÚsendr Úformat_responseÚjsonÚdumpsÚencodeÚrecv)r$ r) r+ Úclientr r r Ú__create_clientP s
|
z!ApiCommandManager.__create_clientc C sz | dt t ¡ d dt dd¡ }|d k r>|d| 7 }| ||¡}| |||¡ | |||¡ td|| ||fS )NÚ_iè r z create_and_run_client success)ÚroundÚtimeÚrandomÚrandintÚ!_ApiCommandManager__create_clientÚ%_ApiCommandManager__heartbeats_threadÚ)_ApiCommandManager__listen_command_threadr )r$ r* ÚindexÚkeyÚskr r r Ú__create_and_run_client[ s ( z)ApiCommandManager.__create_and_run_clienté c C sT | j ¡ || _|| _||f| _x.t|D ]"}| t|¡}|d | j |d < q*W d S )Nr r )Útrade_client_dictÚclearÚtrade_client_countÚaction_callbackr0 ÚrangeÚ)_ApiCommandManager__create_and_run_clientÚCLIENT_TYPE_TRADE_SELL)r$ ÚaddrÚportÚtrade_action_callbackrH ÚiÚresultr r r Úinitg s
|
|
zApiCommandManager.initc C sP xHyÔt |¡d }|rÐt ¡ }zNytd|| t |¡}|d tkrl| t d|d¡ d¡¡ w|d }|d }td| |
|
d ¡} t |¡sºtd
|
t || |d d
|
d ¡ w|tkrÔ| j || |¡ nl|tkrî| j || |¡ nR|tkr
|
| j || |¡ n6|tkr&| j || |¡ n|tkr@| j || |¡ W n. tk
|
rp }
|
zt |
|
¡ W d d }
|
~
|
X Y nX W d t t ¡ | } | d kr²t |¡}t d|d d ¡ | t ddi¡ d¡¡ X ntdW q tk
|
rF }
|
zLt |
|
¡ |t kr|| j kr| j !|¡ y | "¡ W n Y nX P W d d }
|
~
|
X Y qX qW d S )Nr u æ¥æ¶æ°æ®r* r )r* r zutf-8r u æ¥æ¶å
容r u ç¾åé误éÿÿÿÿ)ÚcodeÚmsgé u è¶
æ¶5s以ä¸ï¼Ú cmd_recieveu æ¥æ¶çå
容为空)#r Ú recv_datar; r r3 ÚloadsÚMSG_TYPE_HEARTr1 r4 r5 ÚgetÚis_client_params_sign_rightr Úsend_error_responseÚAPI_TYPE_TRADErI r ÚAPI_TYPE_SELL_RULEr ÚAPI_TYPE_REFRESH_TRADE_DATAr Ú API_TYPE_GET_CODE_POSITION_INFOr ÚAPI_TYPE_COMMON_REQUESTr Ú ExceptionÚloggingÚ exceptionÚintr ÚinforL rF ÚpopÚclose) r$ Ú_typer rC rQ Ú
|
start_timeÚ result_jsonr Ú content_typer ÚeÚuse_timer r r Ú__listen_commandr s`
|
|
|
|
|
|
|
|
|
|
|
|
z"ApiCommandManager.__listen_commandc
|
C s xy&| t t d|d¡ d¡¡¡ W n\ tk
|
r } z>|tkrX|| jkrX| j |¡ y |
|
¡ W n Y nX P W d d }~X Y nX t t ¡ qW d S )Nr )r* r zutf-8)r1 r Ú load_headerr3 r4 r5 rc rL rF rh ri r; ÚsleepÚHEART_SPACE_TIME)r$ rj r rC rn r r r Ú __heart_beats® s &
|
z ApiCommandManager.__heart_beatsc s0 t j fddd}| d¡ | ¡ d S )Nc s ¡S )N)Ú"_ApiCommandManager__listen_commandr )rj r$ r+ rC r r Ú<lambda>Â ó z;ApiCommandManager.__listen_command_thread.<locals>.<lambda>)ÚtargetT)Ú threadingÚThreadÚ setDaemonÚstart)r$ rj r+ rC Út1r )rj r$ r+ rC r Ú__listen_command_threadÀ s
|
z)ApiCommandManager.__listen_command_threadc s0 t j fddd}| d¡ | ¡ d S )Nc s ¡S )N)Ú _ApiCommandManager__heart_beatsr )rj r$ r+ rC r r rv È rw z7ApiCommandManager.__heartbeats_thread.<locals>.<lambda>)rx T)ry rz r{ r| )r$ rj r+ rC r} r )rj r$ r+ rC r Ú__heartbeats_threadÆ s
|
z%ApiCommandManager.__heartbeats_threadc C s t dt ¡ ¡ xjyNt| j| jk rbx8t| jt| j D ] }| t ¡}|d | j|d < q>W W n Y nX t
|
d¡ qW d S )Nu' outside_api __maintain_client 线ç¨ID:r r ) r rg r Ú get_thread_idÚlenrF rH rJ rK rL r; rr )r$ rP rQ r r r Ú__maintain_clientÌ s
|
z#ApiCommandManager.__maintain_clientTc s8 |r ¡ n&tj fddd}| d¡ | ¡ d S )Nc s ¡ S )N)Ú#_ApiCommandManager__maintain_clientr )r r r rv ß rw z'ApiCommandManager.run.<locals>.<lambda>)rx T)r ry rz r{ r| )r Úblockingr} r )r r ÚrunÚ s
|
|
|
zApiCommandManager.run)N)rE )T)r r r Zcommon_client_dictrF rH Zcommon_client_countrI r! r# Ú classmethodr> rK rR ru r r@ r? r r Ú __classcell__r r )r' r r B s$
|
< r c s e Zd ZdZi Zi ZdZdZ fddZe dd Z
|
e dddZ e d d
|
Z e d d Z e d d Ze dd Ze dd Ze dd ZdddZ ZS )ÚNewApiCommandManageru"
|
æ°ç交ææ令管ç
|
Nc s" | j s t j| f||| _ | j S )N)r! r" r# )r$ r% r& )r' r r r# î s z NewApiCommandManager.__new__c
|
C sR t t jt j¡}| | j¡ | t t dd|i|d¡
|
d¡¡¡ | d¡ |S )Nr( r) )r* r r+ zutf-8i ) r, r- r. r/ r0 r1 r r2 r3 r4 r5 r6 )r$ r) r+ r7 r r r r8 ó s
|
z$NewApiCommandManager.__create_clientc C sz | dt t ¡ d dt dd¡ }|d k r>|d| 7 }| ||¡}| |||¡ | |||¡ td|| ||fS )Nr9 iè r z create_and_run_client success)r: r; r< r= Ú$_NewApiCommandManager__create_clientÚ(_NewApiCommandManager__heartbeats_threadÚ,_NewApiCommandManager__listen_command_threadr )r$ r* rA rB rC r r r rD þ s ( z,NewApiCommandManager.__create_and_run_clientc C s | j ¡ | j ¡ || _||f| _x,|D ]$}i | j |d < |d | j|d < q*W xH| jD ]>}x8t| j| D ]&}| ||¡}|d | j | |d < qnW qZW dS )uä
|
åå§å
|
:param addr: æå¡å¨å°å
|
:param port: æå¡å¨ç«¯å£
|
:param trade_action_callback: åè°
|
:param clients_info: 客æ·ç«¯ä¿¡æ¯ï¼[(ç±»å,æ°é)]
|
:return:
|
r r N)Ú client_dictrG Úclient_count_dictrI r0 rJ Ú,_NewApiCommandManager__create_and_run_client) r$ rM rN rI Z clients_infoÚ client_infor) rP rQ r r r rR
|
s
|
|
|
|
|
zNewApiCommandManager.initc C sÜ xÔyVt |¡d }|rRt ¡ }zÒy td|| t |¡}|d tkrh| t d|d¡ d¡¡ w|d }|d }td| |
|
d ¡} t |¡s¶td
|
t || |d d
|
d ¡ w| || |¡ W n, tk
|
rò }
|
zt |
|
¡ W d d }
|
~
|
X Y nX W d tt ¡ | } | d kr4t |¡}t d|d d ¡ | t ddi¡ d¡¡ X ntdW q tk
|
rÒ }
|
zVt |
|
¡ || jkr¦|| j| kr¦| j| |¡ y | ¡ W n Y nX P W d d }
|
~
|
X Y qX qW d S )Nr u æ¥æ¶æ°æ®r* r )r* r zutf-8r u æ¥æ¶å
容r u ç¾åé误rS )rT rU rV u è¶
æ¶5s以ä¸ï¼rW u æ¥æ¶çå
容为空)r rX r; r r3 rY rZ r1 r4 r5 r[ r\ r r] rI rc rd re rf r rg r rh ri ) r$ rj r rC rQ rk rl r rm r rn ro r r r rp # sN
|
|
|
|
|
|
|
|
z%NewApiCommandManager.__listen_commandc
|
C s¢ xy&| t t d|d¡ d¡¡¡ W nf tk
|
r } zH|| jkrb|| j| krb| j| |¡ y | ¡ W n Y nX P W d d }~X Y nX t
|
t ¡ qW d S )Nr )r* r zutf-8) r1 r rq r3 r4 r5 rc r rh ri r; rr rs )r$ rj r rC rn r r r rt T s &
|
z"NewApiCommandManager.__heart_beatsc s0 t j fddd}| d¡ | ¡ d S )Nc s ¡S )N)Ú%_NewApiCommandManager__listen_commandr )rj r$ r+ rC r r rv h rw z>NewApiCommandManager.__listen_command_thread.<locals>.<lambda>)rx T)ry rz r{ r| )r$ rj r+ rC r} r )rj r$ r+ rC r r~ f s
|
z,NewApiCommandManager.__listen_command_threadc s0 t j fddd}| d¡ | ¡ d S )Nc s ¡S )N)Ú"_NewApiCommandManager__heart_beatsr )rj r$ r+ rC r r rv n rw z:NewApiCommandManager.__heartbeats_thread.<locals>.<lambda>)rx T)ry rz r{ r| )r$ rj r+ rC r} r )rj r$ r+ rC r r l s
|
z(NewApiCommandManager.__heartbeats_threadc C s¨ t dt ¡ ¡ xyrxl| jD ]b}t| j| | j| k r xDt| j| t| j| D ]$}| |¡}|d | j| |d < qZW q W W n Y nX t
|
d¡ qW d S )Nu' outside_api __maintain_client 线ç¨ID:r r ) r rg r r r r r rJ r r; rr )r$ r) rP rQ r r r r r s "
|
"z&NewApiCommandManager.__maintain_clientTc s8 |r ¡ n&tj fddd}| d¡ | ¡ d S )Nc s ¡ S )N)Ú&_NewApiCommandManager__maintain_clientr )r r r rv rw z*NewApiCommandManager.run.<locals>.<lambda>)rx T)r ry rz r{ r| )r r
r} r )r r r s
|
|
|
zNewApiCommandManager.run)N)T)r r r Ú__doc__r r rI r! r# r r r rR r r r r r r r r r )r' r r ä s 1 r Ú__main__),r r3 rd r< r, ry r; Z huaxin_client.client_networkr Úlog_module.logr r r Úutilsr r r rZ Ú MSG_TYPE_CMDrL rs ZTRADE_DIRECTION_BUYZTRADE_DIRECTION_SELLZTRADE_TYPE_ORDERZTRADE_TYPE_CANCEL_ORDERZ OPERRATE_SETZOPERRATE_DELETEZ OPERRATE_GETZ OPERRATE_ADDr^ r_ r` ra rb Úobjectr r r r Ú SERVER_HOSTÚ SERVER_PORTÚmanagerr Úinputr r r r Ú<module> sF # (
|
