server/game2004api
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1

Browse Source
This commit is contained in:
aozhiwei 2020-11-10 19:05:50 +08:00
commit b7da6b18a9
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tools/migratedb/import_redis.py
View File

@ -12,11 +12,12 @@ import pprint
argv = sys.argv argv = sys.argv
host = argv[1] host = argv[1]
port = argv[2] port = argv[2]
script_name = argv[3] passwd = argv[3]
script_name = argv[4]
def writeToRedis(conn, cmdline): def writeToRedis(conn, cmdline):
assert len(cmdline) > 1 assert len(cmdline) > 1
# print(cmdline) print(cmdline)
if cmdline[0] == 'hset': if cmdline[0] == 'hset':
conn.hset(cmdline[1], cmdline[2], cmdline[3]) conn.hset(cmdline[1], cmdline[2], cmdline[3])
elif cmdline[0] == 'zadd': elif cmdline[0] == 'zadd':
@ -24,14 +25,19 @@ def writeToRedis(conn, cmdline):
elif cmdline[0] == 'sadd': elif cmdline[0] == 'sadd':
conn.sadd(cmdline[1], cmdline[2]) conn.sadd(cmdline[1], cmdline[2])
elif cmdline[0] == 'set': elif cmdline[0] == 'set':
conn.set(cmdline[1], cmdline[2]) if cmdline[2] != None:
conn.set(cmdline[1], cmdline[2])
else:
print('[warning]', cmdline)
elif cmdline[0] == 'expire': elif cmdline[0] == 'expire':
conn.expire(cmdline[1], cmdline[2]) if int(cmdline[2]) >= 0:
conn.expire(cmdline[1], cmdline[2])
else: else:
assert False assert False
conn = redis.Redis(host = host, conn = redis.Redis(host = host,
port = port, port = port,
password = passwd,
db = 0, db = 0,
decode_responses = True decode_responses = True
) )