B
|
0óJf% ã @ s¾ d Z ddlZddlZddlZddlZddlmZ ddlmZm Z ddl
|
m Z m Z e j Zdd edD ZG d d
|
d
|
ZG d d d Zed krºx$edD ]Ze dde ¡ d¡ qW e ¡ dS )u
|
redis管çå¨
|
é N)Úasync_log_util)Úlogger_redis_debugÚ logger_system)ÚconstantÚmiddle_api_protocolc
|
C s. g | ]&}t jtd td td |dddqS )ÚhostÚportÚpwdTé2 )r r ÚpasswordÚdbÚdecode_responsesZmax_connections)ÚredisÚConnectionPoolÚconfig)Ú.0r © r ú2D:\workspace\trade_cb\db\redis_manager_delegate.pyú
|
<listcomp> s r é c @ s, e Zd Zed fddZdd Zdd ZdS ) Ú RedisManagerr c C s t | | _|| _d S )N)Ú pool_cachesÚpoolr )Úselfr r r r Ú__init__ s
|
zRedisManager.__init__c C s t j| jdS )N)Úconnection_pool)r ÚRedisr )r r r r ÚgetRedis s zRedisManager.getRedisc C s" t jtd td td | jddS )Nr r r T)r r r r r )r r r r )r r r r ÚgetRedisNoPool s zRedisManager.getRedisNoPoolN)Ú__name__Ú
|
__module__Ú __qualname__r r r r r r r r r s r c @ s¾ e Zd Ze ¡ Zedd ZedCddZedd Z edd Z
|
edDd d Z edEd dZ edFddZ edGddZedHddZedIddZedJddZedKddZedLd d ZedMd d ZedNd d!ZedOd"d#ZedPd$d%ZedQd&d'ZedRd(d)ZedSd*d+ZedTd,d-ZedUd.d/Z edVd0d1Z edWd2d3Z edXd4d5Z edYd6d7Z edZd8d9Z!ed:d; Z"ee#d<d=d>Z$ed?d@ Z%edAdB Z&dS )[Ú
|
RedisUtilsc C s t ¡ }z| S X d S )N)Útime)ÚclsÚ method_nameÚkeyZ lamada_methodZ_RedisUtils__start_timer r r Úexec% s zRedisUtils.execNc C s4 |||d}|d k r ||d< t |¡}t |¡}|S )N)r Úcmdr& Úargs)r Zload_redis_cmdÚrequest)r$ r r( r& r) ÚdataÚfdataÚresultr r r Ú __request. s
|
|
zRedisUtils.__requestc C sd g }xF|D ]>}|d |d |d d}|d d k r>|d |d< | |¡ q
|
W t |¡}t |¡}|S )Nr é é )r r( r& é r) )Úappendr Zload_redis_cmdsr* )r$ ZodatasZ_datasÚdr+ r, Úresultsr r r Z__batch__request< s
|
|
|
zRedisUtils.__batch__requestc C s |j jd S )Nr )r Zconnection_kwargs)r$ Úredis_r r r Z__get_dbL s zRedisUtils.__get_dbTc s d fdd¡S )NÚgetc s ¡d¡S )Nr6 )Ú_RedisUtils__requestÚ_RedisUtils__get_dbr )r$ r& r5 r r Ú<lambda>R ó z RedisUtils.get.<locals>.<lambda>)r' )r$ r5 r& Ú auto_freer )r$ r& r5 r r6 P s zRedisUtils.getc s d fdd¡S )NÚscardc s ¡d¡S )Nr< )r7 r8 r )r$ r& r5 r r r9 V r: z"RedisUtils.scard.<locals>.<lambda>)r' )r$ r5 r& r; r )r$ r& r5 r r< T s zRedisUtils.scardFc s d fdd¡S )NÚdeletec s ¡d¡S )Nr= )r7 r8 r )r$ r& r5 r r r9 Z r: z#RedisUtils.delete.<locals>.<lambda>)r' )r$ r5 r& r; Ú_asyncr )r$ r& r5 r r= X s zRedisUtils.deletec C s | |d|¡ d S )Nr= )Úadd_async_task)r$ r r& r; r r r Ú delete_async\ s zRedisUtils.delete_asyncc s, t td ¡ d fdd¡S )Nzkeys:Úkeysc s ¡d¡S )NrA )r7 r8 r )r$ r& r5 r r r9 d r: z!RedisUtils.keys.<locals>.<lambda>)r Úinfor r' )r$ r5 r& r; r )r$ r& r5 r rA a s zRedisUtils.keysc s d fdd¡S )NÚsetc s ¡d¡S )NrC )r7 r8 r )r$ r& r5 Úvalr r r9 h r: z RedisUtils.set.<locals>.<lambda>)r' )r$ r5 r& rD r; r )r$ r& r5 rD r rC f s zRedisUtils.setc C s | |d||f¡ d S )NrC )r? )r$ r r& rD r; r r r Ú set_asyncj s zRedisUtils.set_asyncc s d fdd¡S )NÚsetexc s ¡dg¡S )NrF )r7 r8 r )r$ Úexpirer& r5 rD r r r9 q r: z"RedisUtils.setex.<locals>.<lambda>)r' )r$ r5 r& rG rD r; r> r )r$ rG r& r5 rD r rF o s zRedisUtils.setexc C s | |d|||f¡ d S )NrF )r? )r$ r r& rG rD r; r r r Ú setex_asyncs s zRedisUtils.setex_asyncc s d fdd¡S )NÚsetnxc s ¡d¡S )NrI )r7 r8 r )r$ r& r5 rD r r r9 z r: z"RedisUtils.setnx.<locals>.<lambda>)r' )r$ r5 r& rD r; r )r$ r& r5 rD r rI x s zRedisUtils.setnxc s d fdd¡S )NrG c s ¡d¡S )NrG )r7 r8 r )r$ rG r& r5 r r r9 ~ r: z#RedisUtils.expire.<locals>.<lambda>)r' )r$ r5 r& rG r; r )r$ rG r& r5 r rG | s zRedisUtils.expirec C s | |d||f¡ d S )NrG )r? )r$ r r& rG r; r r r Ú expire_async s zRedisUtils.expire_asyncc s d fdd¡S )NÚsaddc s ¡d¡S )NrK )r7 r8 r )r$ r& r5 rD r r r9 r: z!RedisUtils.sadd.<locals>.<lambda>)r' )r$ r5 r& rD r; r )r$ r& r5 rD r rK
s zRedisUtils.saddc C s | |d||f¡ d S )NrK )r? )r$ r r& rD r; r r r Ú
|
sadd_async s zRedisUtils.sadd_asyncc s d fdd¡S )NÚ sismemberc s ¡d¡S )NrM )r7 r8 r )r$ r& r5 rD r r r9 r: z&RedisUtils.sismember.<locals>.<lambda>)r' )r$ r5 r& rD r; r )r$ r& r5 rD r rM s zRedisUtils.sismemberc s2 d fdd¡}t|tkr.t|}|S )NÚsmembersc s ¡d¡S )NrN )r7 r8 r )r$ r& r5 r r r9 r: z%RedisUtils.smembers.<locals>.<lambda>)r' ÚtypeÚlistrC )r$ r5 r& r; r- r )r$ r& r5 r rN s zRedisUtils.smembersc s d fdd¡S )NÚsremc s ¡d¡S )NrQ )r7 r8 r )r$ r& r5 rD r r r9 r: z!RedisUtils.srem.<locals>.<lambda>)r' )r$ r5 r& rD r; r )r$ r& r5 rD r rQ s zRedisUtils.sremc C s | |d||f¡ d S )NrQ )r? )r$ r r& rD r; r r r Ú
|
srem_async s zRedisUtils.srem_asyncc s d fdd¡S )NÚincrbyc s ¡d¡S )NrS )r7 r8 r )r$ r& Únumr5 r r r9 ¤ r: z#RedisUtils.incrby.<locals>.<lambda>)r' )r$ r5 r& rT r; r> r )r$ r& rT r5 r rS ¢ s zRedisUtils.incrbyc C s | |d||f¡ d S )NrS )r? )r$ r r& rT r; r r r Ú incrby_async¦ s zRedisUtils.incrby_asyncc s d fdd¡S )NÚlpushc s ¡d¡S )NrV )r7 r8 r )r$ r& r5 rD r r r9 r: z"RedisUtils.lpush.<locals>.<lambda>)r' )r$ r5 r& rD r; r )r$ r& r5 rD r rV « s zRedisUtils.lpushc s d fdd¡S )NÚlpopc s ¡d¡S )NrW )r7 r8 r )r$ r& r5 r r r9 ± r: z!RedisUtils.lpop.<locals>.<lambda>)r' )r$ r5 r& r; r )r$ r& r5 r rW ¯ s zRedisUtils.lpopc s d fdd¡S )NÚrpushc s ¡d¡S )NrX )r7 r8 r )r$ r& r5 rD r r r9 µ r: z"RedisUtils.rpush.<locals>.<lambda>)r' )r$ r5 r& rD r; r )r$ r& r5 rD r rX ³ s zRedisUtils.rpushc C s d S )Nr )r$ r5 r r r Úrealse· s zRedisUtils.realse)r c C s | j |||f¡ d S )N)Ú _RedisUtils__async_task_queueÚ
|
put_nowait)r$ r Úmethodr) r r r r? » s zRedisUtils.add_async_taskc C s
|
| j ¡ S )N)rZ Úqsize)r$ r r r Úget_async_task_count¿ s z RedisUtils.get_async_task_countc
|
C sz t d¡ t dt ¡ ¡ g }t ¡ }xHy| j ¡ }|r|d |d g}t|d tksrt|d t kr¸|
|
|d d ¡ t |d dkr¬|
|
|d dd
¡ qÐ|
|
d ¡ n|
|
|d ¡ |
|
d ¡ |
|
t|¡ t |dkrz| |¡}t ¡ }W d | ¡ X |r@t ¡ | dkr@| |¡}t ¡ }| ¡ W q. tk
|
rp } zt |¡ W d d }~X Y q.X q.W d S )Nu å¯å¨Redisæ°æ®åæ¥æå¡u redis 线ç¨ID:r r/ r0 é é )r rB ÚtoolÚ get_thread_idr# rZ r6 rO ÚtuplerP r2 ÚlenÚ_RedisUtils__batch__requestÚclearÚ ExceptionÚloggingÚ exception)r$ ZdataListZlast_upload_timer+ Z temp_datar4 Úe1r r r Úrun_loopÄ s:
|
|
|
|
|
|
|
zRedisUtils.run_loop)N)T)T)TF)T)T)T)T)TF)T)T)T)T)T)T)T)T)T)T)TF)T)T)T)T)'r r r! ÚqueueÚQueuerZ Ú classmethodr' r7 re r8 r6 r< r= r@ rA rC rE rF rH rI rG rJ rK rL rM rN rQ rR rS rU rV rW rX rY Úintr? r^ rk r r r r r" " sr r" Ú__main__é zbuy_position_info-002547ió )Ú__doc__rh rl r# r Z
|
log_moduler Úlog_module.logr r Úutilsr r Ú REDIS_CONFIGr Úranger r r" r ÚirH ra Z
|
get_expirerk r r r r Ú<module> s" X
|
