diff --git a/qieziGo/controller/application.go b/qieziGo/controller/application.go index 8bf878a..04911a5 100644 --- a/qieziGo/controller/application.go +++ b/qieziGo/controller/application.go @@ -1,11 +1,21 @@ package controller -import "github.com/gin-gonic/gin" +import ( + "encoding/json" + "github.com/gin-gonic/gin" + "qieziGo/util" +) var base = `/application` +type test struct { + aa string + bb string +} + func sign(c *gin.Context) { - c.String(200, `ok`) + data, _ := json.Marshal(test{"asdf", "asdf"}) + util.Success(c, data) } func visit(c *gin.Context) { @@ -20,7 +30,7 @@ func download(c *gin.Context) { } -func LoadApplication(e *gin.Engine) { +func LoadApplication(e *gin.RouterGroup) { e.POST(base+"/sign", sign). GET(base+"/visit", visit). POST(base+"/check", check). diff --git a/qieziGo/go.mod b/qieziGo/go.mod index 683be64..1d0400f 100644 --- a/qieziGo/go.mod +++ b/qieziGo/go.mod @@ -11,6 +11,7 @@ require ( github.com/go-playground/universal-translator v0.18.1 // indirect github.com/go-playground/validator/v10 v10.11.2 // indirect github.com/goccy/go-json v0.10.0 // indirect + github.com/golang-jwt/jwt/v4 v4.5.0 // indirect github.com/json-iterator/go v1.1.12 // indirect github.com/klauspost/cpuid/v2 v2.2.4 // indirect github.com/leodido/go-urn v1.2.2 // indirect diff --git a/qieziGo/go.sum b/qieziGo/go.sum index 38c3449..cb121ab 100644 --- a/qieziGo/go.sum +++ b/qieziGo/go.sum @@ -18,6 +18,8 @@ github.com/go-playground/validator/v10 v10.11.2 h1:q3SHpufmypg+erIExEKUmsgmhDTyh github.com/go-playground/validator/v10 v10.11.2/go.mod h1:NieE624vt4SCTJtD87arVLvdmjPAeV8BQlHtMnw9D7s= github.com/goccy/go-json v0.10.0 h1:mXKd9Qw4NuzShiRlOXKews24ufknHO7gx30lsDyokKA= github.com/goccy/go-json v0.10.0/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I= +github.com/golang-jwt/jwt/v4 v4.5.0 h1:7cYmW1XlMY7h7ii7UhUyChSgS5wUJEnm9uZVTGqOWzg= +github.com/golang-jwt/jwt/v4 v4.5.0/go.mod h1:m21LjoU+eqJr34lmDMbreY2eSTRJ1cv77w39/MY0Ch0= github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk= github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg= diff --git a/qieziGo/main.go b/qieziGo/main.go index 465377f..dd9089f 100644 --- a/qieziGo/main.go +++ b/qieziGo/main.go @@ -11,7 +11,8 @@ import ( func main() { r := gin.Default() r.Use(middleware.Auth()) - controller.LoadApplication(r) + v1 := r.Group("/qiezi/api") + controller.LoadApplication(v1) if err := r.Run(":8080"); err != nil { fmt.Printf("start service error:%v\n", err) } diff --git a/qieziGo/middleware/auth.go b/qieziGo/middleware/auth.go index a3988bf..90900c4 100644 --- a/qieziGo/middleware/auth.go +++ b/qieziGo/middleware/auth.go @@ -2,7 +2,6 @@ package middleware import ( "github.com/gin-gonic/gin" - "net/http" "time" ) @@ -15,8 +14,7 @@ func Auth() gin.HandlerFunc { if ok { context.Next() } else { - context.Status(http.StatusUnauthorized) - context.Writer.WriteString(`无访问权限`) + context.String(401, "无访问权限") context.Abort() } ok = !ok diff --git a/qieziGo/util/resultUtil.go b/qieziGo/util/resultUtil.go new file mode 100644 index 0000000..50739b7 --- /dev/null +++ b/qieziGo/util/resultUtil.go @@ -0,0 +1,17 @@ +package util + +import "github.com/gin-gonic/gin" + +type result struct { + Code int + Message string + Data any +} + +func Success(ctx *gin.Context, data any) { + ctx.JSON(200, result{200, "success", data}) +} + +func Fail(ctx *gin.Context) { + ctx.JSON(200, result{0, "fail", nil}) +}