You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
55 lines
1.3 KiB
55 lines
1.3 KiB
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
kgin "github.com/go-kratos/gin"
|
|
"github.com/go-kratos/kratos/v2"
|
|
"github.com/go-kratos/kratos/v2/errors"
|
|
"github.com/go-kratos/kratos/v2/middleware"
|
|
"github.com/go-kratos/kratos/middleware/recovery/v2"
|
|
"github.com/go-kratos/kratos/v2/transport"
|
|
"github.com/go-kratos/kratos/v2/transport/http"
|
|
)
|
|
|
|
func customMiddleware(handler middleware.Handler) middleware.Handler {
|
|
return func(ctx context.Context, req interface{}) (reply interface{}, err error) {
|
|
if tr, ok := transport.FromServerContext(ctx); ok {
|
|
fmt.Println("operation:", tr.Operation())
|
|
}
|
|
reply, err = handler(ctx, req)
|
|
return
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
router := gin.Default()
|
|
// 使用kratos中间件
|
|
router.Use(kgin.Middlewares(recovery.Recovery(), customMiddleware))
|
|
|
|
router.GET("/helloworld/:name", func(ctx *gin.Context) {
|
|
name := ctx.Param("name")
|
|
if name == "error" {
|
|
// 返回kratos error
|
|
kgin.Error(ctx, errors.Unauthorized("auth_error", "no authentication"))
|
|
} else {
|
|
ctx.JSON(200, map[string]string{"welcome": name})
|
|
}
|
|
})
|
|
|
|
httpSrv := http.NewServer(http.Address(":8000"))
|
|
httpSrv.HandlePrefix("/", router)
|
|
|
|
app := kratos.New(
|
|
kratos.Name("gin"),
|
|
kratos.Server(
|
|
httpSrv,
|
|
),
|
|
)
|
|
if err := app.Run(); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|