token 中间件切换4

@@ -18,8 +18,11 @@ func (controller *RoleController) CreateCycle() {
 	if err := controller.Unmarshal(in); err != nil {
 		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
 	} else {
-		in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
-		in.CreatorId = middlewares.GetUserId(controller.Ctx)
+		ua := middlewares.GetUser(controller.Ctx)
+		in.CompanyId = ua.CompanyId
+		in.CreatorId = ua.UserId
+		//in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
+		//in.CreatorId = middlewares.GetUserId(controller.Ctx)
 		controller.Response(ruService.Create(in))
 	}
 }
@@ -30,7 +33,9 @@ func (controller *CycleController) UpdateCycle() {
 	if err := controller.Unmarshal(in); err != nil {
 		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
 	} else {
-		in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
+		ua := middlewares.GetUser(controller.Ctx)
+		in.CompanyId = ua.CompanyId
+		//in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
 		controller.Response(ruService.Update(in))
 	}
 }
@@ -65,7 +70,9 @@ func (controller *CycleController) ListCycle() {
 		if len(in.Name) > 0 {
 			in.Name = "%" + in.Name + "%"
 		}
-		in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
+		ua := middlewares.GetUser(controller.Ctx)
+		in.CompanyId = ua.CompanyId
+		//in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
 		controller.Response(ruService.List(in))
 	}
 }

@@ -18,8 +18,11 @@ func (controller *RoleController) CreateProject() {
 	if err := controller.Unmarshal(in); err != nil {
 		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
 	} else {
-		in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
-		in.CreatorId = middlewares.GetUserId(controller.Ctx)
+		ua := middlewares.GetUser(controller.Ctx)
+		in.CompanyId = ua.CompanyId
+		in.CreatorId = ua.UserId
+		//in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
+		//in.CreatorId = middlewares.GetUserId(controller.Ctx)
 		controller.Response(ruService.Create(in))
 	}
 }
@@ -30,7 +33,9 @@ func (controller *ProjectController) UpdateProject() {
 	if err := controller.Unmarshal(in); err != nil {
 		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
 	} else {
-		in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
+		ua := middlewares.GetUser(controller.Ctx)
+		in.CompanyId = ua.CompanyId
+		//in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
 		controller.Response(ruService.Update(in))
 	}
 }
@@ -41,7 +46,9 @@ func (controller *ProjectController) UpdateProjectForTemplate() {
 	if err := controller.Unmarshal(in); err != nil {
 		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
 	} else {
-		in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
+		ua := middlewares.GetUser(controller.Ctx)
+		in.CompanyId = ua.CompanyId
+		//in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
 		controller.Response(ruService.UpdateTemplate(in))
 	}
 }
@@ -52,7 +59,9 @@ func (controller *ProjectController) UpdateProjectForTemplateNode() {
 	if err := controller.Unmarshal(in); err != nil {
 		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
 	} else {
-		in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
+		ua := middlewares.GetUser(controller.Ctx)
+		in.CompanyId = ua.CompanyId
+		//in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
 		controller.Response(ruService.UpdateTemplateNode(in))
 	}
 }
@@ -88,7 +97,9 @@ func (controller *ProjectController) ListProject() {
 		if len(in.Name) > 0 {
 			in.Name = "%" + in.Name + "%"
 		}
-		in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
+		ua := middlewares.GetUser(controller.Ctx)
+		in.CompanyId = ua.CompanyId
+		//in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
 		controller.Response(ruService.List(in))
 	}
 }
@@ -119,7 +130,9 @@ func (controller *ProjectController) StatisticCycleUser() {
 	if err := controller.Unmarshal(in); err != nil {
 		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
 	} else {
-		in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
+		ua := middlewares.GetUser(controller.Ctx)
+		in.CompanyId = ua.CompanyId
+		//in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
 		controller.Response(ruService.StatisticCycleUser(in))
 	}
 }

@@ -18,8 +18,11 @@ func (controller *RoleController) CreateRule() {
 	if err := controller.Unmarshal(in); err != nil {
 		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
 	} else {
-		in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
-		in.CreatorId = middlewares.GetUserId(controller.Ctx)
+		ua := middlewares.GetUser(controller.Ctx)
+		in.CompanyId = ua.CompanyId
+		in.CreatorId = ua.UserId
+		//in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
+		//in.CreatorId = middlewares.GetUserId(controller.Ctx)
 		controller.Response(ruService.Create(in))
 	}
 }
@@ -30,8 +33,11 @@ func (controller *RuleController) UpdateRule() {
 	if err := controller.Unmarshal(in); err != nil {
 		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
 	} else {
-		in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
-		in.CreatorId = middlewares.GetUserId(controller.Ctx)
+		ua := middlewares.GetUser(controller.Ctx)
+		in.CompanyId = ua.CompanyId
+		in.CreatorId = ua.UserId
+		//in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
+		//in.CreatorId = middlewares.GetUserId(controller.Ctx)
 		controller.Response(ruService.Update(in))
 	}
 }
@@ -82,7 +88,9 @@ func (controller *RuleController) ListRuleRelCreator() {
 		if len(in.NameOrRemark) > 0 {
 			in.NameOrRemark = "%" + in.NameOrRemark + "%"
 		}
-		in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
+		ua := middlewares.GetUser(controller.Ctx)
+		in.CompanyId = ua.CompanyId
+		//in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
 		controller.Response(ruService.ListRelCreator(in))
 	}
 }

@@ -19,8 +19,11 @@ func (controller *RoleController) CreateTemplate() {
 	if err := controller.Unmarshal(in); err != nil {
 		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
 	} else {
-		in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
-		in.CreatorId = middlewares.GetUserId(controller.Ctx)
+		ua := middlewares.GetUser(controller.Ctx)
+		in.CompanyId = ua.CompanyId
+		in.CreatorId = ua.UserId
+		//in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
+		//in.CreatorId = middlewares.GetUserId(controller.Ctx)
 		controller.Response(ruService.Create(in))
 	}
 }
@@ -31,7 +34,9 @@ func (controller *TemplateController) UpdateTemplate() {
 	if err := controller.Unmarshal(in); err != nil {
 		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
 	} else {
-		in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
+		ua := middlewares.GetUser(controller.Ctx)
+		in.CompanyId = ua.CompanyId
+		//in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
 		controller.Response(ruService.Update(in))
 	}
 }
@@ -67,7 +72,9 @@ func (controller *TemplateController) ListTemplate() {
 		if len(in.Name) > 0 {
 			in.Name = "%" + in.Name + "%"
 		}
-		in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
+		ua := middlewares.GetUser(controller.Ctx)
+		in.CompanyId = ua.CompanyId
+		//in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
 		controller.Response(ruService.List(in))
 	}
 }
@@ -81,7 +88,9 @@ func (controller *TemplateController) ListEnableTemplate() {
 	if err := controller.Unmarshal(in); err != nil {
 		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
 	} else {
-		in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
+		ua := middlewares.GetUser(controller.Ctx)
+		in.CompanyId = ua.CompanyId
+		//in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
 		controller.Response(ruService.List(in))
 	}
 }
@@ -92,7 +101,9 @@ func (controller *TemplateController) StateTemplate() {
 	if err := controller.Unmarshal(in); err != nil {
 		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
 	} else {
-		in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
+		ua := middlewares.GetUser(controller.Ctx)
+		in.CompanyId = ua.CompanyId
+		//in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
 		controller.Response(ruService.State(in))
 	}
 }
@@ -103,8 +114,11 @@ func (controller *TemplateController) CopyTemplate() {
 	if err := controller.Unmarshal(in); err != nil {
 		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
 	} else {
-		in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
-		in.CreatorId = middlewares.GetUserId(controller.Ctx)
+		ua := middlewares.GetUser(controller.Ctx)
+		in.CompanyId = ua.CompanyId
+		in.CreatorId = ua.UserId
+		//in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
+		//in.CreatorId = middlewares.GetUserId(controller.Ctx)
 		controller.Response(ruService.Copy(in))
 	}
 }

@@ -51,7 +51,9 @@ func (controller *RoleController) ListForUserRole() {
 	if err := controller.Unmarshal(in); err != nil {
 		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
 	} else {
-		in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
+		ua := middlewares.GetUser(controller.Ctx)
+		in.CompanyId = ua.CompanyId
+		//in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
 		controller.Response(ruService.ListForUser(in))
 	}
 }

@@ -18,7 +18,9 @@ func (controller *RoleUserController) CreateRoleUser() {
 	if err := controller.Unmarshal(in); err != nil {
 		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
 	} else {
-		in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
+		ua := middlewares.GetUser(controller.Ctx)
+		in.CompanyId = ua.CompanyId
+		//in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
 		controller.Response(ruService.Create(in))
 	}
 }
@@ -29,7 +31,9 @@ func (controller *RoleUserController) RemoveRoleUser() {
 	if err := controller.Unmarshal(in); err != nil {
 		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
 	} else {
-		in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
+		ua := middlewares.GetUser(controller.Ctx)
+		in.CompanyId = ua.CompanyId
+		//in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
 		controller.Response(ruService.Remove(in))
 	}
 }
@@ -40,7 +44,9 @@ func (controller *RoleUserController) ListRoleUser() {
 	if err := controller.Unmarshal(in); err != nil {
 		controller.Response(nil, application.ThrowError(application.ARG_ERROR, err.Error()))
 	} else {
-		in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
+		ua := middlewares.GetUser(controller.Ctx)
+		in.CompanyId = ua.CompanyId
+		//in.CompanyId = middlewares.GetCompanyId(controller.Ctx)
 		controller.Response(ruService.ListRole(in))
 	}
 }

@@ -6,6 +6,14 @@ import (
 	"gitlab.fjmaimaimai.com/allied-creation/performance/pkg/domain"
 )
 
+func GetUser(ctx *context.Context) *domain.UserAuth {
+	user := ctx.Input.GetData(domain.UserAuth{})
+	if user != nil {
+		return user.(*domain.UserAuth)
+	}
+	return nil
+}
+
 func CheckAdminToken() func(ctx *context.Context) {
 	return func(ctx *context.Context) {
 		adminToken := ctx.Input.Header("x-admin-token")

 package middlewares
 
-import (
-	"github.com/beego/beego/v2/server/web/context"
-)
-
-func setUserId(userId int64, ctx *context.Context) {
-	ctx.Input.SetData("_UserId", userId)
-}
-
-func GetUserId(ctx *context.Context) int64 {
-	userId := ctx.Input.GetData("_UserId")
-	return userId.(int64)
-}
-
-func setCompanyId(companyId int64, ctx *context.Context) {
-	ctx.Input.SetData("_CompanyId", companyId)
-}
-
-func GetCompanyId(ctx *context.Context) int64 {
-	companyId := ctx.Input.GetData("_CompanyId")
-	return companyId.(int64)
-}
-
-func setCompanyType(companyId int, ctx *context.Context) {
-	ctx.Input.SetData("_CompanyType", companyId)
-}
-
-func GetCompanyType(ctx *context.Context) int {
-	companyId := ctx.Input.GetData("_CompanyType")
-	return companyId.(int)
-}
-
-func invalidOrExpired(ctx *context.Context) {
-	resp := map[string]interface{}{
-		"code": 902,
-		"msg":  "Authorization过期或无效，需要进行重新获取令牌",
-	}
-	_ = ctx.Output.JSON(resp, false, false)
-}
-
-func CheckToken() func(ctx *context.Context) {
-	return func(ctx *context.Context) {
-		tokenStr := ctx.Input.Header("x-mmm-accesstoken")
-		if tokenStr == "" { //没有带token
-			invalidOrExpired(ctx)
-			return
-		}
-
-		//userServe := service.UserService{}
-		//userTk, err := userServe.ValidLoginToken(tokenStr)
-		//if err != nil {
-		//	invalidOrExpired(ctx)
-		//	return
-		//}
-		//setUserId(userTk.UserId, ctx)
-		//setCompanyId(userTk.CompanyId, ctx)
-		//setCompanyType(userTk.CompanyType, ctx)
-	}
-}
+//
+//import (
+//	"github.com/beego/beego/v2/server/web/context"
+//)
+//
+//func setUserId(userId int64, ctx *context.Context) {
+//	ctx.Input.SetData("_UserId", userId)
+//}
+//
+//func GetUserId(ctx *context.Context) int64 {
+//	userId := ctx.Input.GetData("_UserId")
+//	return userId.(int64)
+//}
+//
+//func setCompanyId(companyId int64, ctx *context.Context) {
+//	ctx.Input.SetData("_CompanyId", companyId)
+//}
+//
+//func GetCompanyId(ctx *context.Context) int64 {
+//	companyId := ctx.Input.GetData("_CompanyId")
+//	return companyId.(int64)
+//}
+//
+//func setCompanyType(companyId int, ctx *context.Context) {
+//	ctx.Input.SetData("_CompanyType", companyId)
+//}
+//
+//func GetCompanyType(ctx *context.Context) int {
+//	companyId := ctx.Input.GetData("_CompanyType")
+//	return companyId.(int)
+//}
+//
+//func invalidOrExpired(ctx *context.Context) {
+//	resp := map[string]interface{}{
+//		"code": 902,
+//		"msg":  "Authorization过期或无效，需要进行重新获取令牌",
+//	}
+//	_ = ctx.Output.JSON(resp, false, false)
+//}
+//
+//func CheckToken() func(ctx *context.Context) {
+//	return func(ctx *context.Context) {
+//		tokenStr := ctx.Input.Header("x-mmm-accesstoken")
+//		if tokenStr == "" { //没有带token
+//			invalidOrExpired(ctx)
+//			return
+//		}
+//
+//		//userServe := service.UserService{}
+//		//userTk, err := userServe.ValidLoginToken(tokenStr)
+//		//if err != nil {
+//		//	invalidOrExpired(ctx)
+//		//	return
+//		//}
+//		//setUserId(userTk.UserId, ctx)
+//		//setCompanyId(userTk.CompanyId, ctx)
+//		//setCompanyType(userTk.CompanyType, ctx)
+//	}
+//}

@@ -9,7 +9,7 @@ import (
 
 func init() {
 	ns := web.NewNamespace("/v1/evaluation-rule",
-		web.NSBefore(filters.AllowCors(), middlewares.CheckToken()),
+		web.NSBefore(filters.AllowCors(), middlewares.CheckAdminToken()),
 		web.NSRouter("/", &controllers.RuleController{}, "Post:CreateRule"),
 		web.NSRouter("/", &controllers.RuleController{}, "Put:UpdateRule"),
 		web.NSRouter("/", &controllers.RuleController{}, "Delete:RemoveRule"),

@@ -9,7 +9,7 @@ import (
 
 func init() {
 	ns := web.NewNamespace("/v1/evaluation-template",
-		web.NSBefore(filters.AllowCors(), middlewares.CheckToken()),
+		web.NSBefore(filters.AllowCors(), middlewares.CheckAdminToken()),
 		web.NSRouter("/", &controllers.TemplateController{}, "Post:CreateTemplate"),
 		web.NSRouter("/", &controllers.TemplateController{}, "Put:UpdateTemplate"),
 		web.NSRouter("/", &controllers.TemplateController{}, "Delete:RemoveTemplate"),

@@ -9,7 +9,7 @@ import (
 
 func init() {
 	ns := web.NewNamespace("/v1/role",
-		web.NSBefore(filters.AllowCors(), middlewares.CheckToken()),
+		web.NSBefore(filters.AllowCors(), middlewares.CheckAdminToken()),
 		//web.NSRouter("/", &controllers.RoleController{}, "Post:CreateRole"),
 		//web.NSRouter("/", &controllers.RoleController{}, "Put:UpdateRole"),
 		//web.NSRouter("/", &controllers.RoleController{}, "Delete:RemoveRole"),

@@ -9,7 +9,7 @@ import (
 
 func init() {
 	ns := web.NewNamespace("/v1/role-user",
-		web.NSBefore(filters.AllowCors(), middlewares.CheckToken()),
+		web.NSBefore(filters.AllowCors(), middlewares.CheckAdminToken()),
 		web.NSRouter("/", &controllers.RoleUserController{}, "Post:CreateRole"),
 		web.NSRouter("/", &controllers.RoleUserController{}, "Delete:RemoveRole"),
 		web.NSRouter("/all", &controllers.RoleUserController{}, "Post:ListRoleUser"),