mirror of
https://github.com/hamster1963/HomeDash-Backend.git
synced 2025-06-22 06:51:32 +08:00
update: 初始化代码提交
This commit is contained in:
hamster1963
42
internal/global/g_functions/functions.go
Normal file
42
internal/global/g_functions/functions.go
Normal file
@ -0,0 +1,42 @@
|
||||
package g_functions
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/gogf/gf/v2/errors/gcode"
|
||||
"github.com/gogf/gf/v2/errors/gerror"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/os/glog"
|
||||
"github.com/gogf/gf/v2/util/gconv"
|
||||
)
|
||||
|
||||
func ResErr(code int, errMsg ...interface{}) error {
|
||||
resMsg := ""
|
||||
errDetail := ""
|
||||
switch len(errMsg) {
|
||||
case 0:
|
||||
resMsg = "Unknown error reason"
|
||||
case 1:
|
||||
resMsg = gconv.String(errMsg[0])
|
||||
default:
|
||||
for _, v := range gconv.SliceStr(errMsg[:len(errMsg)-1]) {
|
||||
resMsg += v + ","
|
||||
}
|
||||
resMsg = resMsg[:len(resMsg)-1]
|
||||
errDetail = gconv.String(errMsg[len(errMsg)-1])
|
||||
}
|
||||
return gerror.NewCode(gcode.New(code, resMsg, nil), errDetail)
|
||||
}
|
||||
|
||||
// SetDefaultHandler 替代默认的日志handler
|
||||
func SetDefaultHandler() {
|
||||
glog.SetDefaultHandler(func(ctx context.Context, in *glog.HandlerInput) {
|
||||
m := map[string]interface{}{
|
||||
"stdout": true,
|
||||
"writerColorEnable": false,
|
||||
"file": "inside-{Y-m-d}.log",
|
||||
"path": g.Config().MustGet(ctx, "server.logger.path", "log/").String(), // 此处必须重新设置，才可以实现db的log写入文件
|
||||
}
|
||||
_ = in.Logger.SetConfigWithMap(m)
|
||||
in.Next(ctx)
|
||||
})
|
||||
}
|
||||
33
internal/global/g_index/index.go
Normal file
33
internal/global/g_index/index.go
Normal file
@ -0,0 +1,33 @@
|
||||
package g_index
|
||||
|
||||
const IndexHTML = `
|
||||
<!DOCTYPE html>
|
||||
<html lang="zh">
|
||||
<head>
|
||||
<title>仓鼠咖啡馆</title>
|
||||
<style>
|
||||
body {
|
||||
margin: 0;
|
||||
height: 100vh;
|
||||
background-image: url('https://s2.loli.net/2023/05/16/Nyaxz3QrsSKmvHE.jpg');
|
||||
background-size: cover;
|
||||
background-position: center;
|
||||
display: flex;
|
||||
justify-content: center;
|
||||
align-items: center;
|
||||
text-align: center;
|
||||
}
|
||||
.text {
|
||||
font-family: Arial, sans-serif;
|
||||
font-size: 24px;
|
||||
color: white;
|
||||
background-color: rgba(0, 0, 0, 0.7);
|
||||
padding: 10px 20px;
|
||||
}
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<div class="text">仓鼠咖啡馆</div>
|
||||
</body>
|
||||
</html>
|
||||
`
|
||||
191
internal/global/g_middleware/middleware.go
Normal file
191
internal/global/g_middleware/middleware.go
Normal file
@ -0,0 +1,191 @@
|
||||
package g_middleware
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/gogf/gf/v2/errors/gerror"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/net/ghttp"
|
||||
"github.com/gogf/gf/v2/os/glog"
|
||||
"github.com/golang-jwt/jwt/v5"
|
||||
"home-network-watcher/manifest"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
type sMiddleware struct{} // 创建结构体
|
||||
var (
|
||||
once = &sync.Once{} // 创建互锁
|
||||
s *sMiddleware // 创建指针
|
||||
SMiddlewares = newMiddleware() // 对外暴露
|
||||
)
|
||||
|
||||
// defaultHandlerResponse 返回结构体
|
||||
type defaultHandlerResponse struct {
|
||||
Status int `json:"status" dc:"Error code"` // 可业务需要可以更改json字段
|
||||
Message string `json:"msg" dc:"Error message"`
|
||||
Data interface{} `json:"data" dc:"Result data for certain request according API definition"`
|
||||
}
|
||||
|
||||
// newMiddleware 单例中间件
|
||||
func newMiddleware() *sMiddleware {
|
||||
once.Do(func() {
|
||||
s = &sMiddleware{}
|
||||
})
|
||||
return s
|
||||
}
|
||||
|
||||
func (s *sMiddleware) MiddlewareCORS(r *ghttp.Request) {
|
||||
r.Response.CORSDefault()
|
||||
r.Middleware.Next()
|
||||
}
|
||||
|
||||
// resWriteJson 返回json输出
|
||||
func (s *sMiddleware) resWriteJson(r *ghttp.Request, in defaultHandlerResponse) {
|
||||
r.Response.ClearBuffer()
|
||||
r.Response.WriteJson(defaultHandlerResponse{
|
||||
Status: in.Status,
|
||||
Message: in.Message,
|
||||
Data: in.Data,
|
||||
})
|
||||
}
|
||||
|
||||
// ErrorsStatus 服务器错误码处理
|
||||
func (s *sMiddleware) ErrorsStatus(server *ghttp.Server) {
|
||||
server.BindStatusHandlerByMap(map[int]ghttp.HandlerFunc{
|
||||
500: func(r *ghttp.Request) {
|
||||
s.resWriteJson(r, defaultHandlerResponse{
|
||||
Status: 500,
|
||||
Message: "Error 500,Internal Server Error",
|
||||
Data: "",
|
||||
})
|
||||
},
|
||||
404: func(r *ghttp.Request) {
|
||||
s.resWriteJson(r, defaultHandlerResponse{
|
||||
Status: 404,
|
||||
Message: "Error 404,Not Found",
|
||||
Data: "",
|
||||
})
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
// JWTAuth 鉴权中间件
|
||||
func (s *sMiddleware) JWTAuth(r *ghttp.Request) {
|
||||
JWTString := r.GetHeader("Authorization")
|
||||
if JWTString == "" {
|
||||
s.resWriteJson(r, defaultHandlerResponse{
|
||||
Status: 401,
|
||||
Message: "JWT为空",
|
||||
Data: nil,
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// 验证JWT
|
||||
token, err := jwt.Parse(JWTString, func(token *jwt.Token) (interface{}, error) {
|
||||
return manifest.JWTKey, nil
|
||||
})
|
||||
if err != nil {
|
||||
s.resWriteJson(r, defaultHandlerResponse{
|
||||
Status: 401,
|
||||
Message: "JWT验证失败",
|
||||
Data: nil,
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
if !token.Valid {
|
||||
s.resWriteJson(r, defaultHandlerResponse{
|
||||
Status: 401,
|
||||
Message: "JWT验证失败",
|
||||
Data: nil,
|
||||
})
|
||||
return
|
||||
}
|
||||
// 验证是否过期
|
||||
expirationTime, err := token.Claims.GetExpirationTime()
|
||||
if err != nil {
|
||||
s.resWriteJson(r, defaultHandlerResponse{
|
||||
Status: 401,
|
||||
Message: "JWT验证失败",
|
||||
Data: nil,
|
||||
})
|
||||
return
|
||||
}
|
||||
if expirationTime.Before(time.Now()) {
|
||||
s.resWriteJson(r, defaultHandlerResponse{
|
||||
Status: 401,
|
||||
Message: "JWT已过期",
|
||||
Data: nil,
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
glog.Info(context.TODO(), "JWT验证通过")
|
||||
// 验证通过，设置用户信息
|
||||
audience, err := token.Claims.GetAudience()
|
||||
if err != nil {
|
||||
s.resWriteJson(r, defaultHandlerResponse{
|
||||
Status: 401,
|
||||
Message: "JWT验证失败",
|
||||
Data: nil,
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
r.SetCtxVar("user_id", audience[0])
|
||||
r.Middleware.Next()
|
||||
}
|
||||
|
||||
// ResponseHandler is the middleware handling handler response object and its error. 中间件处理处理程序响应对象及其错误
|
||||
func (s *sMiddleware) ResponseHandler(r *ghttp.Request) {
|
||||
//defer action_log.NewActionLog().LogAdd(r) // 日志钩子
|
||||
r.Middleware.Next()
|
||||
|
||||
// There's custom buffer content, it then exits current handler.
|
||||
if r.Response.BufferLength() > 0 {
|
||||
return
|
||||
}
|
||||
|
||||
// 定义参数和类型.方便调用
|
||||
var (
|
||||
code int
|
||||
msg string
|
||||
resData = r.GetHandlerResponse() // 检索并返回处理程序响应对象及其错误
|
||||
err = r.GetError() // 返回请求过程中发生的错误。如果没有错误，则返回 nil
|
||||
errCode = gerror.Code(err) // 将错误信息通过接口解析处理
|
||||
)
|
||||
code = 200 // 默认返回状态码信息码
|
||||
msg = "success" // 默认成功返回响应体文本信息
|
||||
if err != nil {
|
||||
code = 400 // 默认返回状态码信息码
|
||||
switch errCode.Code() {
|
||||
case -1:
|
||||
msg = "Nonstandard error return"
|
||||
resData = "I won't expose it to you. Go back and change it!"
|
||||
case 500:
|
||||
r.Response.Writer.Status = 500
|
||||
return
|
||||
default:
|
||||
msg = errCode.Message()
|
||||
resData = err.Error()
|
||||
r.Response.Writer.Status = 400
|
||||
if msg == "" || msg == "Unknown error reason" {
|
||||
msg = "Nonstandard error return"
|
||||
resData = "I won't expose it to you. Go back and change it!"
|
||||
}
|
||||
if resData == msg {
|
||||
resData = ""
|
||||
}
|
||||
}
|
||||
}
|
||||
// 如果是空res返回，则resData返回空字符串
|
||||
if g.IsNil(resData) {
|
||||
resData = ""
|
||||
}
|
||||
s.resWriteJson(r, defaultHandlerResponse{
|
||||
Status: code,
|
||||
Message: msg,
|
||||
Data: resData,
|
||||
})
|
||||
}
|
||||
Reference in New Issue
Block a user