def post(self, request, node_id): """ 这个接口操作比较重,所以为了避免发信号 所有写操作都需要用BULK的方式 1 更新节点流量 2 更新用户流量 3 记录节点在线IP 4 关闭超出流量的节点 """ ss_node = SSNode.get_or_none_by_node_id(node_id) if not ss_node: return HttpResponseNotFound() data = request.json["data"] node_total_traffic = 0 log_time = get_current_datetime() active_tcp_connections = 0 need_clear_cache = False user_model_list = [] trafficlog_model_list = [] online_ip_log_model_list = [] for user_data in data: user_id = user_data["user_id"] u = int(user_data["upload_traffic"] * ss_node.enlarge_scale) d = int(user_data["download_traffic"] * ss_node.enlarge_scale) # 个人流量增量 user = User.get_by_pk(user_id) user.download_traffic += d user.upload_traffic += u user.last_use_time = log_time user_model_list.append(user) if user.overflow: need_clear_cache = True # 个人流量记录 trafficlog_model_list.append( UserTrafficLog( node_type=UserTrafficLog.NODE_TYPE_SS, node_id=node_id, user_id=user_id, download_traffic=u, upload_traffic=d, ) ) # 节点流量增量 node_total_traffic += u + d # active_tcp_connections active_tcp_connections += user_data["tcp_conn_num"] # online ip log for ip in user_data.get("ip_list", []): online_ip_log_model_list.append( UserOnLineIpLog(user_id=user_id, node_id=node_id, ip=ip) ) # 用户流量 User.objects.bulk_update( user_model_list, ["download_traffic", "upload_traffic", "last_use_time"], ) # 节点流量记录 SSNode.increase_used_traffic(node_id, node_total_traffic) # 流量记录 UserTrafficLog.objects.bulk_create(trafficlog_model_list) # 在线IP UserOnLineIpLog.objects.bulk_create(online_ip_log_model_list) # 节点在线人数 NodeOnlineLog.add_log( NodeOnlineLog.NODE_TYPE_SS, node_id, len(data), active_tcp_connections ) # check node && user traffic if ss_node.overflow: ss_node.enable = False if need_clear_cache or ss_node.overflow: ss_node.save() return JsonResponse(data={})
def post(self, request, node_id): """ 这个接口操作比较重,所以为了避免发信号 所有写操作都需要用BULK的方式 1 更新节点流量 2 更新用户流量 3 记录节点在线IP 4 关闭超出流量的用户 5 关闭超出流量的节点 """ data = request.json["data"] log_time = pendulum.now() node_total_traffic = 0 trafficlog_model_list = [] user_traffic_model_list = [] online_ip_log_model_list = [] user_ss_config_model_list = [] for user_data in data: user_id = user_data["user_id"] u = user_data["upload_traffic"] d = user_data["download_traffic"] # 个人流量增量 user_traffic = UserTraffic.get_by_user_id(user_id) user_traffic.download_traffic += d user_traffic.upload_traffic += u user_traffic.last_use_time = log_time user_traffic_model_list.append(user_traffic) if user_traffic.overflow: user_ss_config = UserSSConfig.get_by_user_id(user_id) user_ss_config.enable = False user_ss_config_model_list.append(user_ss_config) # 个人流量记录 trafficlog_model_list.append( UserTrafficLog( node_id=node_id, user_id=user_id, download_traffic=u, upload_traffic=d, ) ) # 节点流量增量 node_total_traffic += u + d # online ip log for ip in user_data.get("ip_list", []): online_ip_log_model_list.append( UserOnLineIpLog(user_id=user_id, node_id=node_id, ip=ip) ) # 节点流量记录 SSNode.increase_used_traffic(node_id, node_total_traffic) # 流量记录 UserTrafficLog.objects.bulk_create(trafficlog_model_list) # 在线IP UserOnLineIpLog.objects.bulk_create(online_ip_log_model_list) # 个人流量记录 UserTraffic.objects.bulk_update( user_traffic_model_list, ["download_traffic", "upload_traffic", "last_use_time"], ) # 用户开关 UserSSConfig.objects.bulk_update(user_ss_config_model_list, ["enable"]) # 节点在线人数 SSNodeOnlineLog.add_log(node_id, len(data)) # check node && user traffic ss_node = SSNode.get_or_none_by_node_id(node_id) if ss_node.overflow: ss_node.enable = False if user_ss_config_model_list or ss_node.overflow: # NOTE save for clear cache ss_node.save() return JsonResponse({"ret": 1, "data": []})