重新调整w,r的入口

2023-07-24 13:17:02 +08:00
parent 4c2f823178
commit 7ec78d1c35
82 changed files with 89 additions and 88 deletions
--- a/server/assistant/internal/handler/redirecthandler.go
+++ b/server/assistant/internal/handler/redirecthandler.go
@@ -28,7 +28,7 @@ func RedirectHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {

 		resp := l.Redirect(&req, userinfo)

-		if !basic.AfterLogic(w, r, rl) {
+		if !basic.AfterLogic(w, r, rl, resp) {
 			basic.NormalAfterLogic(w, r, resp)
 		}
 	}
--- a/server/assistant/internal/logic/redirectlogic.go
+++ b/server/assistant/internal/logic/redirectlogic.go
@@ -36,7 +36,7 @@ func NewRedirectLogic(ctx context.Context, svcCtx *svc.ServiceContext) *Redirect
 // }

 // 处理逻辑后 w,r 如:重定向
-func (l *RedirectLogic) AfterLogic(w http.ResponseWriter, r *http.Request) {
+func (l *RedirectLogic) AfterLogic(w http.ResponseWriter, r *http.Request, resp *basic.Response) {
 	if l.RedirectUrl != "" {
 		html := fmt.Sprintf(`
 	<!DOCTYPE html>