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.
48 lines
1.8 KiB
48 lines
1.8 KiB
package server
|
|
|
|
import (
|
|
"context"
|
|
"github.com/go-kratos/kratos/v2/middleware"
|
|
"github.com/go-kratos/kratos/v2/transport"
|
|
"github.com/go-kratos/kratos/v2/transport/http"
|
|
)
|
|
|
|
// 跨域中间件
|
|
|
|
func MWCors() middleware.Middleware {
|
|
return func(handler middleware.Handler) middleware.Handler {
|
|
return func(ctx context.Context, req interface{}) (reply interface{}, err error) {
|
|
if tr, ok := transport.FromServerContext(ctx); ok {
|
|
// Do something on entering
|
|
//在响应头中添加特定的头信息
|
|
ht := tr.(http.Transporter)
|
|
ht.ReplyHeader().Set("Access-Control-Allow-Origin", "*")
|
|
ht.ReplyHeader().Set("Access-Control-Allow-Methods", "POST, GET, PUT, DELETE, PATCH,OPTIONS")
|
|
ht.ReplyHeader().Set("Access-Control-Allow-Credentials", "true")
|
|
ht.ReplyHeader().Set("Access-Control-Allow-Headers", "Access-Control-Allow-Credentials,Content-Type,X-Requested-With ,Content-Length, User-Agent,Authorization")
|
|
defer func() {
|
|
// Do something on exiting
|
|
//method := c.Request.Method
|
|
//origin := c.Request.Header.Get("Origin")
|
|
//if origin != "" {
|
|
// c.Header("Access-Control-Allow-Origin", origin)
|
|
// c.Header("Access-Control-Allow-Methods", "POST, GET, PUT, DELETE, OPTIONS")
|
|
// c.Header("Access-Control-Allow-Headers", "Content-Type, Content-Length, Token")
|
|
// c.Header("Access-Control-Expose-Headers", "Access-Control-Allow-Headers, Token")
|
|
// c.Header("Access-Control-Max-Age", "172800")
|
|
// c.Header("Access-Control-Allow-Credentials", "true")
|
|
// c.Set("content-type", "application/json")
|
|
//}
|
|
//if method == "OPTIONS" {
|
|
// //c.JSON(200, Controller.R(200, nil, "Options Request"))
|
|
// c.AbortWithStatus(http.StatusNoContent)
|
|
//}
|
|
//c.Next()
|
|
|
|
}()
|
|
}
|
|
return handler(ctx, req)
|
|
}
|
|
}
|
|
}
|