adulau/hashlookup-server
1
0
Fork 0
mirror of https://github.com/adulau/hashlookup-server.git synced 2024-11-21 17:47:06 +00:00
Code Issues Projects Releases Packages Wiki Activity

new: [api:bulk] add support for pub-sub channel of existing and

Browse source 
non-existing hashes
This commit is contained in:
Alexandre Dulaunoy 2021-08-29 11:52:07 +02:00
parent ecc2baf2f9
commit a14e5aedf1
Signed by: adulau
GPG key ID: 09E2CD4944E6CBCD
1 changed files with 21 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
bin/server.py
View file

@ -8,7 +8,7 @@ import json
config = configparser.ConfigParser() config = configparser.ConfigParser()
config.read('../etc/server.conf') config.read('../etc/server.conf')
stats = config['global'].getboolean('stats') stats = config['global'].getboolean('stats')
stats_pubsub = config['global'].getboolean('stats') stats_pubsub = config['global'].getboolean('stats_pubsub')
score = 1 score = 1
session = config['session'].getboolean('enable') session = config['session'].getboolean('enable')
session_ttl = config['session'].get('ttl') session_ttl = config['session'].get('ttl')
@ -181,10 +181,17 @@ class bulkmd5(Resource):
return {'message': 'JSON format incorrect. An array of hashes in the key \'hashes\' is expected.'}, 404 return {'message': 'JSON format incorrect. An array of hashes in the key \'hashes\' is expected.'}, 404
ret = [] ret = []
for val in json_data['hashes']: for val in json_data['hashes']:
if not rdb.exists("l:{}".format(val.upper())): k = val.upper()
if not rdb.exists("l:{}".format(k)):
if stats_pubsub:
pub_lookup(channel='nx', k=k)
continue continue
sha1 = rdb.get("l:{}".format(val.upper())) sha1 = rdb.get("l:{}".format(k))
ret.append(rdb.hgetall("h:{}".format(sha1))) ret.append(rdb.hgetall("h:{}".format(sha1)))
if stats:
rdb.zincrby("s:exist:sha1", score, k)
if stats_pubsub:
pub_lookup(channel='exist', k=k)
return ret return ret
@api.route('/bulk/sha1') @api.route('/bulk/sha1')
@ -196,7 +203,17 @@ class bulksha1(Resource):
return {'message': 'JSON format incorrect. An array of hashes in the key \'hashes\' is expected.'}, 404 return {'message': 'JSON format incorrect. An array of hashes in the key \'hashes\' is expected.'}, 404
ret = [] ret = []
for val in json_data['hashes']: for val in json_data['hashes']:
ret.append(rdb.hgetall("h:{}".format(val.upper()))) k = val.upper()
if not rdb.exists("h:{}".format(k)):
if stats_pubsub:
pub_lookup(channel='nx', k=k)
continue
k = val.upper()
ret.append(rdb.hgetall("h:{}".format(k)))
if stats:
rdb.zincrby("s:exist:sha1", score, k)
if stats_pubsub:
pub_lookup(channel='exist', k=k)
return ret return ret
@api.route('/session/create/<string:name>') @api.route('/session/create/<string:name>')