|
|
@@ -28,7 +28,43 @@ formats = {
|
|
|
'net': 'l',
|
|
|
'disk': 'l',
|
|
|
}
|
|
|
-def datum(split, environ):
|
|
|
+
|
|
|
+def get_data(split, environ):
|
|
|
+ group = int(split[1])
|
|
|
+ client_id = int(split[3])
|
|
|
+ data_path = path.join(DATA_DIR, str(group), str(client_id))
|
|
|
+ with open(data_path, 'r') as f:
|
|
|
+ buf = f.read()
|
|
|
+ rval = {}
|
|
|
+ index = 0
|
|
|
+ while index < len(buf):
|
|
|
+ # read the key
|
|
|
+ key_size = ord(buf[index])
|
|
|
+ fmt = '%dp' % (key_size + 1)
|
|
|
+ data = struct.unpack(fmt, buf[index:index+key_size+1])
|
|
|
+ key = data[0]
|
|
|
+
|
|
|
+ # get the format_code
|
|
|
+ split = key.split('.')
|
|
|
+ stat_group = split[0]
|
|
|
+ format_code = formats[stat_group]
|
|
|
+
|
|
|
+ # read the data
|
|
|
+ fmt = '%dp 1440%s' % (key_size + 1, format_code)
|
|
|
+ size = struct.calcsize(fmt)
|
|
|
+ data = struct.unpack(fmt, buf[index:index+size])
|
|
|
+ dict_insert(rval, split, data[1:])
|
|
|
+ index += size
|
|
|
+ return json.dumps(rval)
|
|
|
+
|
|
|
+def dict_insert(d, split, value):
|
|
|
+ if len(split) > 1:
|
|
|
+ d.setdefault(split[0], {})
|
|
|
+ dict_insert(d[split[0]], split[1:], value)
|
|
|
+ else:
|
|
|
+ d[split[0]] = value
|
|
|
+
|
|
|
+def post_datum(split, environ):
|
|
|
group = int(split[1])
|
|
|
body = json.load(environ['wsgi.input'])
|
|
|
client_id = body['client_id']
|
|
|
@@ -50,12 +86,13 @@ def datum(split, environ):
|
|
|
|
|
|
def write_stat(f, key, format_code, data):
|
|
|
fmt = '%dp 1440%s' % (len(key) + 1, format_code)
|
|
|
- array = [0] * 1440
|
|
|
+ array = [-1] * 1440
|
|
|
array[0] = data
|
|
|
f.write(struct.pack(fmt, key, *array))
|
|
|
|
|
|
handlers = {
|
|
|
- 'datum': datum,
|
|
|
+ 'data': get_data,
|
|
|
+ 'datum': post_datum,
|
|
|
}
|
|
|
|
|
|
server = gevent.pywsgi.WSGIServer(('0.0.0.0', 8892), application)
|
