XT_New/controllers/base_api_controller.go

package controllers

import (
	"XT/enums"
	"XT/service"
)

type BaseAPIController struct {
	BaseController
}

// func (this *BaseAPIController) Prepare() {
// 	this.BaseController.Prepare()
// 	beego.Trace("============================================================")
// 	beego.Trace("session ID: %v", this.Ctx.Input.Cookie("beegosessionID"))
// 	beego.Trace("session : %v", this.GetSession("info"))
// 	this.SetSession("info", time.Now().Format("2006/01/02 15:04:05"))
// 	beego.Trace("============================================================")
// }

// 输出数据格式化
/*
	success json:
	{
		"state": 1,
		"code": 0,
		"data": json,
	}

	fail json:
	{
		"state": 0,
		"code": int,
		"msg": string,
	}
*/
func (this *BaseAPIController) ServeSuccessJSON(data map[string]interface{}) {
	this.Data["json"] = enums.MakeSuccessResponseJSON(data)
	this.ServeJSON()
}

func (this *BaseAPIController) ServeFailJSONWithSGJErrorCode(code int) {
	this.Data["json"] = enums.MakeFailResponseJSONWithSGJErrorCode(code)
	this.ServeJSON()
}

func (this *BaseAPIController) ServeFailJSONWithSGJError(err *enums.SGJError) {
	this.Data["json"] = enums.MakeFailResponseJSONWithSGJError(err)
	this.ServeJSON()
}

func (this *BaseAPIController) ServeFailJsonSend(code int, msg string) {
	this.Data["json"] = enums.MakeFailResponseJSON(msg, code)
	this.ServeJSON()
}

type BaseAuthAPIController struct {
	BaseAPIController
}

func (this *BaseAuthAPIController) Prepare() {
	this.BaseAPIController.Prepare()
	if this.GetAdminUserInfo() == nil {
		//var userAdmin models.AdminUser
		//userAdmin.Id = 380
		//userAdmin.Mobile = "13532250447"
		//userAdmin.IsSuperAdmin = true
		//userAdmin.Status = 1
		//userAdmin.CreateTime = 1530786071
		//userAdmin.ModifyTime = 1530786071
		//var subscibe models.ServeSubscibe
		//subscibe.ID = 1
		//subscibe.OrgId = 4
		//subscibe.PeriodStart = 1538035409
		//subscibe.PeriodEnd = 1569571409
		//subscibe.State = 1
		//subscibe.Status = 1
		//subscibe.CreatedTime = 1538035409
		//subscibe.UpdatedTime = 1538035409
		//subscibes := make(map[int64]*models.ServeSubscibe, 0)
		//subscibes[4] = &subscibe
		//
		//var template models.GobalTemplate
		//template.TemplateId = 2
		//
		//
		//var adminUserInfo service.AdminUserInfo
		//adminUserInfo.CurrentOrgId = 4
		//adminUserInfo.CurrentAppId = 5
		//adminUserInfo.AdminUser = &userAdmin
		//adminUserInfo.Subscibes = subscibes
		//this.SetSession("admin_user_info", &adminUserInfo)

		this.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeNotLogin)
		this.StopRun()
	}

	if this.Ctx.Request.Method != "GET" {
		adminUserInfo := this.GetAdminUserInfo()

		err := service.GetOrgSubscibeState(adminUserInfo.Subscibes[adminUserInfo.CurrentOrgId])
		if err != nil || adminUserInfo.Subscibes[adminUserInfo.CurrentOrgId].State == 3 {
			this.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeNotSubscibe)
			this.StopRun()
		}
	}

	//if this.Ctx.Request.Header.Get("Permission") == "1" {
	//
	//	isPermission := false
	//	roles := []int64{1,2}		//模拟该用户有多少角色
	//	var targetRole models.RolePurview
	//	var userRolePurview []string
	//
	//	//用户角色1对应的权限
	//	role1 := models.RolePurview{PurviewIds:"11,12,13"}
	//
	//	//用户角色2对应的权限
	//	role2 := models.RolePurview{PurviewIds:"11,15,16"}
	//
	//
	//	//模拟角色2
	//	//判断该用户有多少个角色，合并最大角色权限
	//	if len(roles) == 1{ //单个
	//		targetRole = role1
	//		userRolePurview = strings.Split(targetRole.PurviewIds, ",")
	//
	//	}else{ //多个
	//		targetRole = role1
	//		targetRole.PurviewIds = targetRole.PurviewIds + "," + role2.PurviewIds
	//		userRolePurview = strings.Split(targetRole.PurviewIds, ",")
	//	}
	//
	//	userRolePurview = RemoveRepeatedElement2(userRolePurview)
	//
	//	fmt.Println(userRolePurview)
	//	//所有权限列表
	//	allPermission := []models.Purview{{Id:10,Urlfor:"/m/api/dialysis/dialysisPrescription-/api/dialysis/prescription"},
	//		{Id:11,Urlfor:"/m/api/dialysis/dialysisPrescription1-/api/dialysis/prescription1"},
	//		{Id:12,Urlfor:"/m/api/dialysis/dialysisPrescription2-/api/dialysis/prescription2"},
	//		{Id:13,Urlfor:"/m/api/dialysis/dialysisPrescription3-/api/dialysis/prescription3"},
	//		{Id:14,Urlfor:"/m/api/dialysis/dialysisPrescription4-/api/dialysis/prescription4"},
	//		{Id:15,Urlfor:"/m/api/dialysis/dialysisPrescription5-/api/dialysis/prescription5"},
	//		{Id:16,Urlfor:"/m/api/dialysis/dialysisPrescription6-/api/dialysis/prescription6"}}
	//	for _, item := range allPermission {
	//		//判断当前路由是否在权限路由列表里面
	//		if strings.Split(item.Urlfor, "-")[1] == strings.Split(this.Ctx.Request.RequestURI , "?")[0]{
	//			fmt.Println(strings.Split(this.Ctx.Request.RequestURI , "?")[0])
	//
	//			//获取该角色的所有权限
	//			for _, items := range userRolePurview{
	//				id, _ := strconv.ParseInt(items, 10, 64)
	//				if id == item.Id{
	//					isPermission = true
	//				}
	//			}
	//			if !isPermission{
	//				this.ServeFailJSONWithSGJErrorCode(enums.ErrorCodePreExist)
	//				this.StopRun()
	//			}
	//		}
	//
	//	}
	//}
}

type BaseServeAPIController struct {
	BaseAPIController
}

func (this *BaseServeAPIController) Prepare() {
	this.BaseAPIController.Prepare()
	if this.GetAdminUserInfo() == nil {
		//var userAdmin models.AdminUser
		//userAdmin.Id = 380
		//userAdmin.Mobile = "13532250447"
		//userAdmin.IsSuperAdmin = true
		//userAdmin.Status = 1
		//userAdmin.CreateTime = 1530786071
		//userAdmin.ModifyTime = 1530786071
		//var subscibe models.ServeSubscibe
		//subscibe.ID = 1
		//subscibe.OrgId = 4
		//subscibe.PeriodStart = 1538035409
		//subscibe.PeriodEnd = 1569571409
		//subscibe.State = 1
		//subscibe.Status = 1
		//subscibe.CreatedTime = 1538035409
		//subscibe.UpdatedTime = 1538035409
		//subscibes := make(map[int64]*models.ServeSubscibe, 0)
		//subscibes[4] = &subscibe
		//var adminUserInfo service.AdminUserInfo
		//adminUserInfo.CurrentOrgId = 4
		//adminUserInfo.CurrentAppId = 5
		//adminUserInfo.AdminUser = &userAdmin
		//adminUserInfo.Subscibes = subscibes
		//this.SetSession("admin_user_info", &adminUserInfo)

		this.ServeFailJSONWithSGJErrorCode(enums.ErrorCodeNotLogin)
		this.StopRun()
	}
}

func RemoveRepeatedElement2(arr []string) (newArr []string) {
	newArr = make([]string, 0)
	for i := 0; i < len(arr); i++ {
		repeat := false
		for j := i + 1; j < len(arr); j++ {
			if arr[i] == arr[j] {
				repeat = true
				break
			}
		}
		if !repeat {
			newArr = append(newArr, arr[i])
		}
	}
	return
}