wangshan 3 days ago
parent
commit
3c41de7f28
2 changed files with 10 additions and 11 deletions
  1. 5 6
      internal/cmd/cmd.go
  2. 5 5
      internal/middleware/cookie.go

+ 5 - 6
internal/cmd/cmd.go

@@ -2,12 +2,11 @@ package cmd
 
 import (
 	"context"
-	"jybxseo/internal/controller"
-	"jybxseo/internal/middleware"
-
 	"github.com/gogf/gf/v2/frame/g"
 	"github.com/gogf/gf/v2/net/ghttp"
 	"github.com/gogf/gf/v2/os/gcmd"
+	"jybxseo/internal/controller"
+	"jybxseo/internal/middleware"
 )
 
 var (
@@ -17,10 +16,10 @@ var (
 		Brief: "start http jybxseo server",
 		Func: func(ctx context.Context, parser *gcmd.Parser) (err error) {
 			s := g.Server()
+			s.BindMiddlewareDefault(
+				middleware.SetSeoRefer, // SetSeoRefer seo来源标记
+			)
 			s.Group("/", func(group *ghttp.RouterGroup) {
-				group.Middleware(
-					middleware.SetSeoRefer, // SetSeoRefer seo来源标记
-				)
 				group.GET("/", controller.IndexTransfer)           //地区、行业首页
 				group.GET("/index.html", controller.IndexTransfer) //地区、行业首页
 

+ 5 - 5
internal/middleware/cookie.go

@@ -1,17 +1,17 @@
 package middleware
 
 import (
-	"github.com/gogf/gf/v2/frame/g"
 	"github.com/gogf/gf/v2/net/ghttp"
-	"github.com/gogf/gf/v2/os/gctx"
+	"strings"
 )
 
 // SetSeoRefer seo来源标记
 func SetSeoRefer(r *ghttp.Request) {
 	val := r.Cookie.Get("SESSIONID")
-	g.Log().Info(gctx.New(), "-------------------------------", val.String())
-	if !val.IsNil() && val.String() != "" {
-		if val = r.Cookie.Get("SeoRefer"); val.IsNil() {
+	sessionId := val.String()
+	if sessionId != "" && !strings.Contains(r.Referer(), "jianyu360.c") {
+		val = r.Cookie.Get("SeoRefer")
+		if val.IsNil() {
 			r.Cookie.Set("SeoRefer", r.Referer())
 		}
 	}