B
|
de7% ã @ sÎ d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl m
|
Z
|
m Z ddl m Z mZ ejZdd edD ZG d d
|
d
|
ZG d d d Zed krÊx$edD ]Ze dde ¡ d¡ q¦W e ¡ dS )u
|
redis管çå¨
|
é N)Úasync_log_util)Úlogger_redis_debugÚ logger_system)ÚtoolÚmiddle_api_protocolc
|
C s. g | ]&}t jtd td td |dddqS )ÚhostÚportÚpwdTé2 )r r ÚpasswordÚdbÚdecode_responsesZmax_connections)ÚredisZConnectionPoolÚconfig)Ú.0r © r ú4D:\workspace\trade_sell\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 Z __request0 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_dbN s zRedisUtils.__get_dbTc s d fdd¡S )NÚgetc s ¡d¡S )Nr4 )Ú_RedisUtils__requestÚ_RedisUtils__get_dbr )r# r% r3 r r Ú<lambda>T ó z RedisUtils.get.<locals>.<lambda>)r& )r# r3 r% Ú auto_freer )r# r% r3 r r4 R s zRedisUtils.getc s d fdd¡S )NÚscardc s ¡d¡S )Nr: )r5 r6 r )r# r% r3 r r r7 X r8 z"RedisUtils.scard.<locals>.<lambda>)r& )r# r3 r% r9 r )r# r% r3 r r: V s zRedisUtils.scardFc s d fdd¡S )NÚdeletec s ¡d¡S )Nr; )r5 r6 r )r# r% r3 r r r7 \ r8 z#RedisUtils.delete.<locals>.<lambda>)r& )r# r3 r% r9 Ú_asyncr )r# r% r3 r r; Z s zRedisUtils.deletec C s | |d|¡ d S )Nr; )Úadd_async_task)r# r r% r9 r r r Ú delete_async^ s zRedisUtils.delete_asyncc s, t td ¡ d fdd¡S )Nzkeys:Úkeysc s ¡d¡S )Nr? )r5 r6 r )r# r% r3 r r r7 f r8 z!RedisUtils.keys.<locals>.<lambda>)r Úinfor r& )r# r3 r% r9 r )r# r% r3 r r? c s zRedisUtils.keysc s d fdd¡S )NÚsetc s ¡d¡S )NrA )r5 r6 r )r# r% r3 Úvalr r r7 j r8 z RedisUtils.set.<locals>.<lambda>)r& )r# r3 r% rB r9 r )r# r% r3 rB r rA h s zRedisUtils.setc C s | |d||f¡ d S )NrA )r= )r# r r% rB r9 r r r Ú set_asyncl s zRedisUtils.set_asyncc s d fdd¡S )NÚsetexc s ¡dg¡S )NrD )r5 r6 r )r# Úexpirer% r3 rB r r r7 s r8 z"RedisUtils.setex.<locals>.<lambda>)r& )r# r3 r% rE rB r9 r< r )r# rE r% r3 rB r rD q s zRedisUtils.setexc C s | |d|||f¡ d S )NrD )r= )r# r r% rE rB r9 r r r Ú setex_asyncu s zRedisUtils.setex_asyncc s d fdd¡S )NÚsetnxc s ¡d¡S )NrG )r5 r6 r )r# r% r3 rB r r r7 | r8 z"RedisUtils.setnx.<locals>.<lambda>)r& )r# r3 r% rB r9 r )r# r% r3 rB r rG z s zRedisUtils.setnxc s d fdd¡S )NrE c s ¡d¡S )NrE )r5 r6 r )r# rE r% r3 r r r7 r8 z#RedisUtils.expire.<locals>.<lambda>)r& )r# r3 r% rE r9 r )r# rE r% r3 r rE ~ s zRedisUtils.expirec C s | |d||f¡ d S )NrE )r= )r# r r% rE r9 r r r Ú expire_async s zRedisUtils.expire_asyncc s d fdd¡S )NÚsaddc s ¡d¡S )NrI )r5 r6 r )r# r% r3 rB r r r7 r8 z!RedisUtils.sadd.<locals>.<lambda>)r& )r# r3 r% rB r9 r )r# r% r3 rB r rI s zRedisUtils.saddc C s | |d||f¡ d S )NrI )r= )r# r r% rB r9 r r r Ú
|
sadd_async s zRedisUtils.sadd_asyncc s d fdd¡S )NÚ sismemberc s ¡d¡S )NrK )r5 r6 r )r# r% r3 rB r r r7 r8 z&RedisUtils.sismember.<locals>.<lambda>)r& )r# r3 r% rB r9 r )r# r% r3 rB r rK s zRedisUtils.sismemberc s2 d fdd¡}t|tkr.t|}|S )NÚsmembersc s ¡d¡S )NrL )r5 r6 r )r# r% r3 r r r7 r8 z%RedisUtils.smembers.<locals>.<lambda>)r& ÚtypeÚlistrA )r# r3 r% r9 r, r )r# r% r3 r rL s zRedisUtils.smembersc s d fdd¡S )NÚsremc s ¡d¡S )NrO )r5 r6 r )r# r% r3 rB r r r7 r8 z!RedisUtils.srem.<locals>.<lambda>)r& )r# r3 r% rB r9 r )r# r% r3 rB r rO s zRedisUtils.sremc C s | |d||f¡ d S )NrO )r= )r# r r% rB r9 r r r Ú
|
srem_async s zRedisUtils.srem_asyncc s d fdd¡S )NÚincrbyc s ¡d¡S )NrQ )r5 r6 r )r# r% Únumr3 r r r7 ¦ r8 z#RedisUtils.incrby.<locals>.<lambda>)r& )r# r3 r% rR r9 r< r )r# r% rR r3 r rQ ¤ s zRedisUtils.incrbyc C s | |d||f¡ d S )NrQ )r= )r# r r% rR r9 r r r Ú incrby_async¨ s zRedisUtils.incrby_asyncc s d fdd¡S )NÚlpushc s ¡d¡S )NrT )r5 r6 r )r# r% r3 rB r r r7 ¯ r8 z"RedisUtils.lpush.<locals>.<lambda>)r& )r# r3 r% rB r9 r )r# r% r3 rB r rT s zRedisUtils.lpushc s d fdd¡S )NÚlpopc s ¡d¡S )NrU )r5 r6 r )r# r% r3 r r r7 ³ r8 z!RedisUtils.lpop.<locals>.<lambda>)r& )r# r3 r% r9 r )r# r% r3 r rU ± s zRedisUtils.lpopc s d fdd¡S )NÚrpushc s ¡d¡S )NrV )r5 r6 r )r# r% r3 rB r r r7 · r8 z"RedisUtils.rpush.<locals>.<lambda>)r& )r# r3 r% rB r9 r )r# r% r3 rB r rV µ s zRedisUtils.rpushc C s d S )Nr )r# r3 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)rX Ú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- r. é é )r r@ r Z get_thread_idr" rX r4 rM ÚtuplerN r0 ÚlenÚ_RedisUtils__batch__requestÚclearÚ ExceptionÚloggingÚ exception)r# ZdataListZlast_upload_timer* Z temp_datar2 Ú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ÚQueuerX Ú classmethodr& r5 ra r6 r4 r: r; r> r? rA rC rD rF rG rE rH rI rJ rK rL rO rP rQ rS rT rU rV rW Úintr= r\ rg r r r r r! $ sr r! Ú__main__é zbuy_position_info-002547ió )Ú__doc__rd rh Ú threadingr" r ÚconstantÚ
|
log_moduler Úlog_module.logr r Úutilsr r Z REDIS_CONFIGr Úranger r r! r ÚirF Ú
|
get_expirerg r r r r Ú<module> s& X
|
