B
|
ËÕdÄ ã @ s¤ d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z ej
|
Z dd e dD Z G dd d ZG d
|
d d Zed kr ed ¡ Zejjd Zee dS )u
|
redis管çå¨
|
é N)Úlogger_redis_debug)Útoolc
|
C s. g | ]&}t jtd td td |dddqS )ÚhostÚportÚpwdTé2 )r r ÚpasswordÚdbÚdecode_responsesZmax_connections)ÚredisÚConnectionPoolÚconfig)Ú.0r © r õ2 D:\éåä½å¸\low_suction_hx\db\redis_manager.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 Zed:ddZed;ddZ ed<d d
|
Z
|
ed=d d Z ed>d dZ ed?ddZ ed@ddZedAddZedBddZedCddZedDddZedEdd ZedFd d ZedGd d ZedHd!d"ZedId#d$ZedJd%d&ZedKd'd(ZedLd)d*ZedMd+d,ZedNd-d.Z edOd/d0Z ed1d2 Z ed3d4 Z ed5d6 Z ed7d8 Z!d9S )PÚ
|
RedisUtilsc C s8 t ¡ }z| S t d|tt ¡ | d d|¡ X d S )Nz {}({}):{}iè é )Útimer ÚinfoÚround)ÚclsÚ method_nameÚkeyZ lamada_methodZ_RedisUtils__start_timer r r Úexec% s zRedisUtils.execTc s | d fdd¡S )NÚgetc s
|
¡S )N)r( r )r& Úredis_r r Ú<lambda>/ ó z RedisUtils.get.<locals>.<lambda>)r' )r$ r) r& Ú auto_freer )r& r) r r( - s zRedisUtils.getc s | d fdd¡S )NÚscardc s
|
¡S )N)r- r )r& r) r r r* 3 r+ z"RedisUtils.scard.<locals>.<lambda>)r' )r$ r) r& r, r )r& r) r r- 1 s zRedisUtils.scardFc s | d fdd¡S )NÚdeletec s
|
¡S )N)r. r )r& r) r r r* 7 r+ z#RedisUtils.delete.<locals>.<lambda>)r' )r$ r) r& r, Ú_asyncr )r& r) r r. 5 s zRedisUtils.deletec C s | |d|¡ t dd|¡ d S )Nr. zdelete_async({}):{}r )Úadd_async_taskr r" )r$ r r& r, r r r Ú delete_async9 s zRedisUtils.delete_asyncc s | d fdd¡S )NÚkeysc s
|
¡S )N)r2 r )r& r) r r r* @ r+ z!RedisUtils.keys.<locals>.<lambda>)r' )r$ r) r& r, r )r& r) r r2 > s zRedisUtils.keysc s | d fdd¡S )NÚsetc s ¡S )N)r3 r )r& r) Úvalr r r* D r+ z RedisUtils.set.<locals>.<lambda>)r' )r$ r) r& r4 r, r )r& r) r4 r r3 B s zRedisUtils.setc s | d fdd¡S )NÚsetexc s ¡S )N)r5 r )Úexpirer& r) r4 r r r* H r+ z"RedisUtils.setex.<locals>.<lambda>)r' )r$ r) r& r6 r4 r, r/ r )r6 r& r) r4 r r5 F s zRedisUtils.setexc C s& | |d|||f¡ t dd|¡ d S )Nr5 zsetex_async({}):{}r )r0 r r" )r$ r r& r6 r4 r, r r r Ú setex_asyncJ s zRedisUtils.setex_asyncc s | d fdd¡S )NÚsetnxc s ¡S )N)r8 r )r& r) r4 r r r* Q r+ z"RedisUtils.setnx.<locals>.<lambda>)r' )r$ r) r& r4 r, r )r& r) r4 r r8 O s zRedisUtils.setnxc s | d fdd¡S )Nr6 c s ¡S )N)r6 r )r6 r& r) r r r* U r+ z#RedisUtils.expire.<locals>.<lambda>)r' )r$ r) r& r6 r, r )r6 r& r) r r6 S s zRedisUtils.expirec C s$ | |d||f¡ t dd|¡ d S )Nr6 zexpire_async({}):{}r )r0 r r" )r$ r r& r6 r, r r r Ú expire_asyncW s zRedisUtils.expire_asyncc s | d fdd¡S )NÚsaddc s ¡S )N)r: r )r& r) r4 r r r* ^ r+ z!RedisUtils.sadd.<locals>.<lambda>)r' )r$ r) r& r4 r, r )r& r) r4 r r: \ s zRedisUtils.saddc C s$ | |d||f¡ t dd|¡ d S )Nr: zsadd_async({}):{}r )r0 r r" )r$ r r& r4 r, r r r Ú
|
sadd_async` s zRedisUtils.sadd_asyncc s | d fdd¡S )NÚ sismemberc s ¡S )N)r< r )r& r) r4 r r r* g r+ z&RedisUtils.sismember.<locals>.<lambda>)r' )r$ r) r& r4 r, r )r& r) r4 r r< e s zRedisUtils.sismemberc s | d fdd¡S )NÚsmembersc s
|
¡S )N)r= r )r& r) r r r* k r+ z%RedisUtils.smembers.<locals>.<lambda>)r' )r$ r) r& r, r )r& r) r r= i s zRedisUtils.smembersc s | d fdd¡S )NÚsremc s ¡S )N)r> r )r& r) r4 r r r* o r+ z!RedisUtils.srem.<locals>.<lambda>)r' )r$ r) r& r4 r, r )r& r) r4 r r> m s zRedisUtils.sremc C s$ | |d||f¡ t dd|¡ d S )Nr> zsrem_async({}):{}r )r0 r r" )r$ r r& r4 r, r r r Ú
|
srem_asyncq s zRedisUtils.srem_asyncc s | d fdd¡S )NÚincrbyc s ¡S )N)r@ r )r& Únumr) r r r* x r+ z#RedisUtils.incrby.<locals>.<lambda>)r' )r$ r) r& rA r, r/ r )r& rA r) r r@ v s zRedisUtils.incrbyc C s$ | |d||f¡ t dd|¡ d S )Nr@ zincrby_async({}):{}r )r0 r r" )r$ r r& rA r, r r r Ú incrby_asyncz s zRedisUtils.incrby_asyncc s | d fdd¡S )NÚlpushc s ¡S )N)rC r )r& r) r4 r r r* r+ z"RedisUtils.lpush.<locals>.<lambda>)r' )r$ r) r& r4 r, r )r& r) r4 r rC s zRedisUtils.lpushc s | d fdd¡S )NÚlpopc s
|
¡S )N)rD r )r& r) r r r*
r+ z!RedisUtils.lpop.<locals>.<lambda>)r' )r$ r) r& r, r )r& r) r rD s zRedisUtils.lpopc s | d fdd¡S )NÚrpushc s ¡S )N)rE r )r& r) r4 r r r* r+ z"RedisUtils.rpush.<locals>.<lambda>)r' )r$ r) r& r4 r, r )r& r) r4 r rE s zRedisUtils.rpushc C s d S )Nr )r$ r) r r r Úrealse s zRedisUtils.realsec C s | j |||f¡ d S )N)Ú _RedisUtils__async_task_queueÚ
|
put_nowait)r$ r ÚmethodÚargsr r r r0 s zRedisUtils.add_async_taskc C s
|
| j ¡ S )N)rG Úqsize)r$ r r r Úget_async_task_count s z RedisUtils.get_async_task_countc
|
C s xy^| j ¡ }|r^|d }|d }|d }t| ¡ }t||}t|tkrV|| }n||}W q tk
|
r } zt |¡ W d d }~X Y qX qW d S )Nr é é )
|
rG r( r r ÚgetattrÚtypeÚtupleÚ ExceptionÚloggingÚ exception) r$ Údatar r% rJ Z_redisrI ÚresultÚe1r r r Úrun_loop s
|
|
|
|
zRedisUtils.run_loopN)T)T)TF)T)T)T)TF)T)T)T)T)T)T)T)T)T)T)TF)T)T)T)T)"r r r ÚqueueÚQueuerG Ú classmethodr' r( r- r. r1 r2 r3 r5 r7 r8 r6 r9 r: r; r< r= r> r? r@ rB rC rD rE rF r0 rL rX r r r r r " sd r Ú__main__rM r )Ú__doc__rS rY r! r ÚconstantÚlog_module.logr Úutilsr Ú REDIS_CONFIGr Úranger r r r r r Zconnection_kwargsr Úprintr r r r Ú<module> s"
|
