mirror of https://github.com/D4-project/d4-core
fix: [Worker 8] fix buffer type
parent
e7c83391b4
commit
fe22c69512
|
@ -21,7 +21,7 @@
|
|||
|
||||
<body>
|
||||
|
||||
<nav class="navbar navbar-expand-sm navbar-dark bg-dark">
|
||||
<nav class="navbar navbar-expand-lg navbar-dark bg-dark">
|
||||
<a class="navbar-brand" href="{{ url_for('index') }}">
|
||||
<img src="{{ url_for('static', filename='img/d4-logo.png')}}" alt="D4 Project" style="width:80px;">
|
||||
</a>
|
||||
|
|
|
@ -37,6 +37,8 @@ rotation_save_cycle = 300 #seconds
|
|||
|
||||
analyzer_list_max_default_size = 10000
|
||||
|
||||
max_buffer_length = 10000
|
||||
|
||||
save_to_file = True
|
||||
|
||||
def get_save_dir(dir_data_uuid, year, month, day):
|
||||
|
@ -54,7 +56,7 @@ if __name__ == "__main__":
|
|||
session_uuid = sys.argv[1]
|
||||
stream_name = 'stream:{}:{}'.format(type, session_uuid)
|
||||
id = '0'
|
||||
buffer = None
|
||||
buffer = b''
|
||||
|
||||
# track launched worker
|
||||
redis_server_stream.sadd('working_session_uuid:{}'.format(type), session_uuid)
|
||||
|
@ -89,9 +91,9 @@ if __name__ == "__main__":
|
|||
|
||||
if id and data:
|
||||
# reconstruct data
|
||||
if buffer:
|
||||
if buffer != b'':
|
||||
data[b'message'] = '{}{}'.format(buffer, data[b'message'])
|
||||
buffer = None
|
||||
buffer = b''
|
||||
|
||||
# send data to redis
|
||||
# new line in received data
|
||||
|
@ -110,7 +112,12 @@ if __name__ == "__main__":
|
|||
if all_line[-1] != b'':
|
||||
buffer += data[b'message']
|
||||
else:
|
||||
if len(buffer) < max_buffer_length:
|
||||
buffer += data[b'message']
|
||||
else:
|
||||
print('Error, infinite loop, buffer may length reached')
|
||||
# force new line
|
||||
buffer += b'{}\n'.format(data[b'message'])
|
||||
|
||||
|
||||
# save data on disk
|
||||
|
|
Loading…
Reference in New Issue