def get_image(req: HttpRequest, code):
b_ret = False
op_type = Const.OpType.查询
if code:
assets_impl = AssetsImpl(_db=req.my_db, _img_db=req.my_img_db)
tmp_code, header, image = assets_impl.get_image(code=code)
if image:
content_type = "application/octet-stream"
if header:
if type(header) == bytes or type(header) == bytearray:
header = header.decode("utf-8")
tmp = re.findall('Content-Type:.*', header, re.I)
if tmp:
content_type = re.sub('Content-Type:', '', tmp[0], re.I)
req.res_head['Content-Type'] = content_type
req.res_body = image
b_ret = True
else:
req.res_command = ResponseCode.NOT_FOUND
package_body(req=req,
status=Const.OpStatus.成功,
op_type=op_type,
message='资产代码:{0}, 没有图像记录或者图像记录格式不对'.format(code, ))
else:
req.res_command = ResponseCode.BAD_REQUEST
package_body(req=req,
status=Const.OpStatus.失败,
op_type=op_type,
message='没有上送资产编码')
return b_ret
def do_get(req: HttpRequest):
paras = req.parameters
action = paras.get('action', None)
limit = paras.get('limit', 10)
offset = paras.get('offset', 0)
code = paras.get('code', None)
login_name = paras.get('userInfo.login_name', None)
if not login_name:
user_info = paras.get('userInfo')
if user_info:
login_name = user_info.get('login_name', None)
if isinstance(req, HttpRequest):
req.res_head['Content-Type'] = 'application/json; charset=UTF-8'
if action == 'get_image':
b_ret = get_image(req=req, code=code)
elif action == 'get_assets':
b_ret = get_assets(req=req,
code=code,
login_name=login_name,
limit=limit,
offset=offset)
else:
req.res_command = ResponseCode.NOT_IMPLEMENTED
package_body(req=req,
status=Const.OpStatus.失败,
op_type=Const.OpType.查询,
message="不支持的动作")
b_ret = False
else:
raise Exception('para req is not HttpRequest')
return b_ret
def tcp_link(sock, addr, alive_sock_queue, _db, _img_db):
t_start = time.time()
my_print('Accept new connection from {0}...'.format(addr, ))
exception_trace_id = generate_trace_id()
exception_message = None
has_exception = True
req = HttpRequest(sock,
addr,
_db=_db,
_img_db=_img_db,
trace_id=exception_trace_id)
need_send = False
try:
success = req.do()
if success: # 成功的请求，以后可以缓存
pass
has_exception = False
need_send = True
except Exception as e:
tb = traceback.format_exc()
if str(e).find('client socket closed.') != -1:
return
exception_message = str(e)
except BaseException as be:
tb = traceback.format_exc()
exception_message = str(be)
if has_exception:
my_print('tcp_link TraceId:{0} {1} {2}'.format(exception_trace_id,
exception_message, tb))
req.res_command = ResponseCode.INNER_ERROR
body = dict()
body['status'] = 9999
body['message'] = exception_message
body['traceId'] = exception_trace_id
ymp = json.dumps(body, ensure_ascii=False)
req.res_body = ymp.encode('UTF-8')
req.res_head['Content-Type'] = 'text/html; charset=UTF-8'
need_send = True
if need_send:
if req.res_body is None or 0 == len(req.res_body):
my_print("trace_id:{} header:{} body:{}".format(
req.trace_id, req._raw_head, get_print_string(req._raw_body)))
req.res_body = 'UNKNOWN ERROR,trace_id:{} req.trace_id:{}'.format(
req.trace_id, req.parameters.get('trace_id',
None)).encode('utf-8')
req.res_command = b"HTTP/1.1 500 OK\r\n"
req.res_head['Content-Length'] = str(len(req.res_body))
response_header = dict2header(req.res_head).encode('UTF-8')
if Conf.get('is_save_response'):
write_to_file(Conf.get('save_response_file_path'),
req.get_res_data())
sock.sendall(req.res_command)
sock.sendall(response_header)
sock.sendall(req.res_body)
if req.res_head.get('Connection', 'Close').lower() == 'keep-alive':
data = types.SimpleNamespace(sock=sock, addr=addr)
alive_sock_queue.put(data)
else:
sock.close()
else:
sock.close()
t_stop = time.time()
print(
"[%d] %s,%s %s %.5f" %
(os.getpid(), addr[0], addr[1], req.command.command, t_stop - t_start))
