BaseController.php 5.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185
  1. <?php
  2. namespace App\Http\Controllers\Api\WebApi;
  3. use App\Components\Helpers;
  4. use App\Models\Node;
  5. use App\Models\NodeHeartBeat;
  6. use App\Models\NodeOnlineLog;
  7. use App\Models\NodeOnlineUserIp;
  8. use App\Models\Rule;
  9. use App\Models\RuleGroup;
  10. use App\Models\RuleGroupNode;
  11. use App\Models\RuleLog;
  12. use App\Models\User;
  13. use App\Models\UserDataFlowLog;
  14. use Arr;
  15. use Illuminate\Http\JsonResponse;
  16. use Illuminate\Http\Request;
  17. use Response;
  18. class BaseController
  19. {
  20. // 上报节点心跳信息
  21. public function setNodeStatus(Request $request, $id): JsonResponse
  22. {
  23. $cpu = (int) $request->input('cpu') / 100;
  24. $mem = (int) $request->input('mem') / 100;
  25. $disk = (int) $request->input('disk') / 100;
  26. if (is_null($request->input('uptime'))) {
  27. return $this->returnData('上报节点心跳信息失败,请检查字段');
  28. }
  29. $obj = new NodeHeartBeat();
  30. $obj->node_id = $id;
  31. $obj->uptime = (int) $request->input('uptime');
  32. //$obj->load = $request->input('load');
  33. $obj->load = implode(' ', [$cpu, $mem, $disk]);
  34. $obj->log_time = time();
  35. $obj->save();
  36. if (! $obj->id) {
  37. return $this->returnData('生成节点心跳信息失败');
  38. }
  39. return $this->returnData('上报节点心跳信息成功', 'success', 200);
  40. }
  41. // 返回数据
  42. public function returnData($message, $status = 'fail', $code = 400, $data = [], $addition = []): JsonResponse
  43. {
  44. $etag = Helpers::abortIfNotModified($data);
  45. $data = [
  46. 'status' => $status,
  47. 'code' => $code,
  48. 'data' => $data,
  49. 'message' => $message,
  50. ];
  51. if ($addition) {
  52. $data = array_merge($data, $addition);
  53. }
  54. return Response::json($data)->header('ETAG', $etag);
  55. }
  56. // 上报节点在线人数
  57. public function setNodeOnline(Request $request, $id): JsonResponse
  58. {
  59. $inputArray = $request->all();
  60. $onlineCount = 0;
  61. foreach ($inputArray as $input) {
  62. if (! Arr::has($input, ['ip', 'uid'])) {
  63. return $this->returnData('上报节点在线用户IP信息失败,请检查字段');
  64. }
  65. $obj = new NodeOnlineUserIp();
  66. $obj->node_id = $id;
  67. $obj->user_id = $input['uid'];
  68. $obj->ip = $input['ip'];
  69. $obj->port = User::find($input['uid'])->port;
  70. $obj->created_at = time();
  71. $obj->save();
  72. if (! $obj->id) {
  73. return $this->returnData('生成节点在线用户IP信息失败');
  74. }
  75. $onlineCount++;
  76. }
  77. $obj = new NodeOnlineLog();
  78. $obj->node_id = $id;
  79. $obj->online_user = $onlineCount;
  80. $obj->log_time = time();
  81. $obj->save();
  82. if ($obj->id) {
  83. return $this->returnData('上报节点在线情况成功', 'success', 200);
  84. }
  85. return $this->returnData('生成节点在线情况失败');
  86. }
  87. // 上报用户流量日志
  88. public function setUserTraffic(Request $request, $id): JsonResponse
  89. {
  90. foreach ($request->all() as $input) {
  91. if (! Arr::exists($input, 'uid')) {
  92. return $this->returnData('上报用户流量日志失败,请检查字段');
  93. }
  94. $rate = Node::find($id)->traffic_rate;
  95. $log = new UserDataFlowLog();
  96. $log->user_id = (int) $input['uid'];
  97. $log->u = (int) $input['upload'] * $rate;
  98. $log->d = (int) $input['download'] * $rate;
  99. $log->node_id = $id;
  100. $log->rate = $rate;
  101. $log->traffic = flowAutoShow($log->u + $log->d);
  102. $log->log_time = time();
  103. $log->save();
  104. if (! $log->id) {
  105. return $this->returnData('生成用户流量日志失败');
  106. }
  107. $user = User::find($log->user_id);
  108. if ($user) {
  109. $user->u += $log->u;
  110. $user->d += $log->d;
  111. $user->t = time();
  112. $user->save();
  113. }
  114. }
  115. return $this->returnData('上报用户流量日志成功', 'success', 200);
  116. }
  117. // 获取节点的审计规则
  118. public function getNodeRule($id): JsonResponse
  119. {
  120. $nodeRule = RuleGroupNode::whereNodeId($id)->first();
  121. $data = [];
  122. //节点未设置任何审计规则
  123. if ($nodeRule) {
  124. $ruleGroup = RuleGroup::find($nodeRule->rule_group_id);
  125. if ($ruleGroup && $ruleGroup->rules) {
  126. foreach ($ruleGroup->rules as $ruleId) {
  127. $rule = Rule::find($ruleId);
  128. if ($rule) {
  129. $data[] = [
  130. 'id' => $rule->id,
  131. 'type' => $rule->type_api_label,
  132. 'pattern' => $rule->pattern,
  133. ];
  134. }
  135. }
  136. return $this->returnData('获取节点审计规则成功', 'success', 200, ['mode' => $ruleGroup->type ? 'reject' : 'allow', 'rules' => $data]);
  137. }
  138. }
  139. //放行
  140. return $this->returnData('获取节点审计规则成功', 'success', 200, ['mode' => 'all', 'rules' => $data]);
  141. }
  142. // 上报用户触发的审计规则记录
  143. public function addRuleLog(Request $request, $id): JsonResponse
  144. {
  145. if ($request->has(['uid', 'rule_id', 'reason'])) {
  146. $obj = new RuleLog();
  147. $obj->user_id = $request->input('uid');
  148. $obj->node_id = $id;
  149. $obj->rule_id = $request->input('rule_id');
  150. $obj->reason = $request->input('reason');
  151. $obj->save();
  152. if ($obj->id) {
  153. return $this->returnData('上报用户触发审计规则日志成功', 'success', 200);
  154. }
  155. }
  156. return $this->returnData('上报用户触发审计规则日志失败');
  157. }
  158. }