gp-server.git - Gitblit

B  
Öd+0ã@sddlZddlZddlZddlZddlZddlZddlZddlZddlZddl    Z    ddl
Z
ddlZddlm Z ddlmZmZddlmZmZmZddlmZe ¡ZGdddejZGdd    d    ejeZGd
ddejZdd ZddZ ddZ!e"dkrüdS)éN)Ú
mysql_data)Ú
RedisUtilsÚRedisManager)Úsocket_utilÚkpl_api_utilÚhosting_api_util)Ú JueJinHttpApic@seZdZddZdS)ÚMyTCPServercCstjj|||dddS)NT)Úbind_and_activate)ÚsocketserverÚ    TCPServerÚ__init__)ÚselfÚserver_addressÚRequestHandlerClass©rú1D:\workspace\GP\gp_middle_server\middle_server.pyr szMyTCPServer.__init__N)Ú__name__Ú
__module__Ú__qualname__r rrrrr    sr    c@seZdZdS)ÚMyThreadingTCPServerN)rrrrrrrrsrcsTeZdZdZddZeddZddZedd    Zfd
dZ    fdd Z
ZS)ÚMyBaseRequestHandleFcCs| ¡dS)N)Ú_MyBaseRequestHandle__init)rrrrÚsetup"szMyBaseRequestHandle.setupcCs|jr
dSd|_i|_dS)NT)Ú_MyBaseRequestHandle__initedÚ%_MyBaseRequestHandle__req_socket_dict)ÚclsrrrÚ__init%szMyBaseRequestHandle.__initcCs|g}|d}| d¡x$|D]}| |d||¡qW| ¡d |¡d}t |jdd¡ ¡}||krxtddS)NÚsignú=ú&z
JiaBei@!*.zutf-8)Úencodinguç¾ååºé)    ÚpopÚappendÚsortÚjoinÚhashlibÚmd5ÚencodeÚ    hexdigestÚ    Exception)rÚ    data_jsonÚlist_strrÚkÚ_MyBaseRequestHandle__strr'rrrÚ__is_sign_right,s
 
z#MyBaseRequestHandle.__is_sign_rightc
Csd}d}| |¡}|}|rt ¡}| d¡}| d¡r~t|dd}d}xJ||ksz| d¡}    |t|    7}||     d¡7}qLWn| d¡}|| d¡}||fS)    NÚé
zutf-8z##éri(i)ÚrecvÚtimeÚdecodeÚ
startswithÚintÚlen)
rZskkÚdataÚheader_sizeÚbufÚ
header_strÚ
start_timeÚcontent_lengthÚ received_sizeÚr_datarrrÚgetRecvData8s"
 
 
 
 
 
zMyBaseRequestHandle.getRecvDatacs$|jd}t ¡|j}xyÊ| |¡\}}|ræ|}d}yt |¡}WnLtjjk
r}z*|     t
 t ddd¡j dd¡¡wWdd}~XYnXt|d|ddkrè|d    d
}|d}    tj ||    |¡|     t ddi¡j dd¡yx| |¡\}
}y,t |
¡}|ddkr0tj |d ¡WnBtjjk
rt}z|
sX| ¡td|
|Wdd}~XYnXt d¡qøWWn\tk
r¶}ztj |    ¡Wdd}~XYn.tk
râ}zt |¡Wdd}~XYnXqè|ddkrBz"|d } t | |d|d    ¡Wd|     t
 t ddi¡j dd¡¡Xqè|ddkrz|d    }|d    }td|Wd|     t
 t ddi¡j dd¡¡Xqè|ddkry|d    }|d}|d    }d}|dkràt ddi¡}nº|dkr|d}|d}|d}| d¡}t| ¡}t t!|}||g}|rzt"|t#ksLt"|t$krpt$|}x$|D]}| %|¡qZWn
| %|¡t#|}||}
t d|
d¡}|     t
 |j dd¡¡WnVtk
r
}z6t |¡|     t
 t dt&|d¡j dd¡¡Wdd}~XYnXqè|ddkr,y²|d    }|d    }|d}|d}| d¡}t' (¡}t ||}g}|rt"|t#kst"|t$krt$|}n
| %|¡t#|}||}
t d|
d¡}|     t
 |j dd¡¡WnVtk
r&}z6t |¡|     t
 t dt&|d¡j dd¡¡Wdd}~XYnXqè|ddkrîyZ|d    }|d    }|d}| d ¡}t) ||¡}
t d|
d¡}|     t
 |j dd¡¡WnVtk
rê}z6t |¡|     t
 t dt&|d¡j dd¡¡Wdd}~XYnXnö|dd!kr°yZ|d    }|d    }|d"}| d    ¡}t* ||¡}
t d|
d¡}|     t
 |j dd¡¡WnVtk
r¬}z6t |¡|     t
 t dt&|d¡j dd¡¡Wdd}~XYnXn4|dd#krè|d    }|d}|d$krÚn
|d%krènPWqtk
r}zt |¡PWdd}~XYqXqWdS)&NréduJSONè§£æå¤±è´¥)ÚcodeÚmsgzutf-8)r!ÚtypeÚregisterr9Úclient_typeÚridÚheartÚ    client_iduJSONè§£æåºééÚresponseÚ
request_idrCZl2_subscript_codesÚredisÚctyper0Z
queue_sizeÚcmdÚdbÚkeyÚargs)rCr9ÚmysqlZjuejinÚpathÚparamsÚkplÚurlÚtradeÚorderÚcancel_order)+Úclient_addressÚsuperÚhandleÚrequestrAÚjsonÚloadsÚdecoderÚJSONDecodeErrorÚsendallrÚload_headerÚdumpsr(ÚprintÚsocket_managerÚClientSocketManagerÚ
add_clientrIÚcloser4ÚsleepÚConnectionResetErrorÚ
del_clientr*ÚloggingÚ    exceptionrÚset_responseÚgetrÚgetRedisÚgetattrrrEÚtupleÚlistr#ÚstrrÚMysqldbrr)rÚhostÚskr9ÚheaderÚdata_strr+ÚerGrHÚresultÚ
resultJSONÚeerJÚdatasrOZ
result_strrQrPrRrSrNÚmethodZargs_ÚarTZpath_rVrXZdata_)Ú    __class__rrr^Ms
 
 
 ((
 
 
 
 
 
 
6
 
 
 
 
6
 
4
 
4
 
 
zMyBaseRequestHandle.handlecst ¡dS)N)r]Úfinish)r)rrrròszMyBaseRequestHandle.finish)rrrrrÚclassmethodrÚ#_MyBaseRequestHandle__is_sign_rightrAr^rÚ __classcell__rr)rrrs&rc    Cs8x2z ytj ¡WnYnXWdt d¡XqWdS)Nr2)rhriÚdel_invalid_clientsr4rlrrrrÚclear_invalid_clientös
rcCsL|dk    rH|dk    rHx6y$| ¡}|r4t |¡}td|WqYqXqWdS)Nuæ¶å°æ¥èªL1çæ°æ®ï¼)r3r`rarg)Z
pipe_tradeZpipe_l1ÚvalrrrÚ__recv_pipe_l1s
rcCs<tdtjdddd}| ¡d}t|t}| ¡dS)Nzcreate MiddleServercSstS)N)rrrrrÚ<lambda>ózrun.<locals>.<lambda>T)ÚtargetÚdaemon)z0.0.0.0i')rgÚ    threadingÚThreadÚstartrrÚ serve_forever)Út1ÚladdrÚ    tcpserverrrrÚrun s
rÚ__main__)#Údatetimer&Úior`roÚqueueÚrandomÚsocketrrr4rhrQrZdb.redis_managerrrÚutilsrrrZutils.juejin_utilrÚQueueZtrade_data_request_queuerr    ÚThreadingMixInrÚBaseRequestHandlerrrrrrrrrrÚ<module>s2X