From 8b69d78c3b88269540434e75a09e7968a667cb29 Mon Sep 17 00:00:00 2001 From: wanglei <34475144@qq.com> Date: Fri, 24 Oct 2025 11:11:57 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E9=AA=8C=E8=AF=81=E5=87=BD?= =?UTF-8?q?=E6=95=B0=E6=96=87=E4=BB=B6=EF=BC=8C=E5=AF=B9=E4=BD=BF=E7=94=A8?= =?UTF-8?q?=E7=9A=84=E5=85=AC=E5=85=B1=E5=87=BD=E6=95=B0=E8=BF=9B=E8=A1=8C?= =?UTF-8?q?=E5=B0=81=E8=A3=85?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/api/system/account.lua | 4 +--- src/api/system/user.lua | 16 ++++------------ src/service/system/account.lua | 15 ++------------- src/util/validator.lua | 32 ++++++++++++++++++++++++++++++++ 4 files changed, 39 insertions(+), 28 deletions(-) create mode 100644 src/util/validator.lua diff --git a/src/api/system/account.lua b/src/api/system/account.lua index 3d6c03a..a0e2c37 100644 --- a/src/api/system/account.lua +++ b/src/api/system/account.lua @@ -38,9 +38,7 @@ end --根据账号id获取账号信息 function _M.add_account() --获取请求头并进行校验 - if checkReqHeader() == false then - return - end + if checkReqHeader() == false then return end --读取请求体的数据 ngx.req.read_body() --获取请求数据 diff --git a/src/api/system/user.lua b/src/api/system/user.lua index c44e1b8..5696cee 100644 --- a/src/api/system/user.lua +++ b/src/api/system/user.lua @@ -8,17 +8,7 @@ local _M = {} local dao = require("service.system.user") local resp = require("util.response") - ---验证请求头是否正确 -local function checkReqHeader() - local headers = ngx.req.get_headers() - if headers["content-type"] ~= "application/json" then - local result = resp:json(0x000001) - resp:send(result) - return false - end - return true -end +local validator = require("util.validator") --获取所有用户信息 function _M.get_alluser() @@ -38,7 +28,9 @@ end --根据用户id获取用户信息 function _M.add_user() --获取请求头并进行校验 - if checkReqHeader() == false then + if validator.checkReqHeader() == false then + local result = resp:json(0x000001) + resp:send(result) return end --读取请求体的数据 diff --git a/src/service/system/account.lua b/src/service/system/account.lua index 320a352..bb8f2bd 100644 --- a/src/service/system/account.lua +++ b/src/service/system/account.lua @@ -8,6 +8,7 @@ local pgmoon = require('pgmoon') local dbconf = require("config.database") local status = require("config.status") local snowflake = require("util.snowflake") +local validator = require("util.validator") local _M = {} @@ -57,18 +58,6 @@ local function execSQL(sql) return code,res end ---校验json数据的正确性,并返回json解析后的数据 -local function checkJson(jsonData) - local success, result = pcall(function() - return cjson.decode(jsonData) - end) - if success == true then - return true, result - end - local res = nil - return false,res -end - --通过查询条件判断数据库中的数据记录 local function checkAccountExist(where) --组装sql语句 @@ -95,7 +84,7 @@ end --增加账号信息到数据表 function _M.addAccount(jsonData) --验证数据的正确性,错误时返回 - local success, result = checkJson(jsonData) + local success, result = validator.checkJson(jsonData) if success == false then return 0x000001,result end diff --git a/src/util/validator.lua b/src/util/validator.lua new file mode 100644 index 0000000..1e46380 --- /dev/null +++ b/src/util/validator.lua @@ -0,0 +1,32 @@ +--- +--- Generated by EmmyLua(https://github.com/EmmyLua) +--- Created by admin. +--- DateTime: 2025/10/24 11:01 +--- 提供公共需要的验证接口等功能 + +local cjson = require('cjson') + +local _M = {} + +--验证请求头是否正确 +function _M:checkReqHeader() + local headers = ngx.req.get_headers() + if headers["content-type"] ~= "application/json" then + return false + end + return true +end + +--校验json数据的正确性,并返回json解析后的数据 +function _M:checkJson(jsonData) + local success, result = pcall(function() + return cjson.decode(jsonData) + end) + if success == true then + return true, result + end + local res = nil + return false,res +end + +return _M \ No newline at end of file