cas_tt_cloud_backend/app/enterprise/controller/user/tenant.go

package user

import (
	"SciencesServer/app/basic/config"
	model2 "SciencesServer/app/common/model"
	"SciencesServer/app/enterprise/model"
	"SciencesServer/app/service"
	config2 "SciencesServer/config"
	"SciencesServer/serve/orm"
	"SciencesServer/utils"
	"errors"
	"gorm.io/gorm"
	"time"
)

type Tenant struct{ *service.SessionEnterprise }

type TenantHandle func(session *service.SessionEnterprise) *Tenant

type TenantBasicParams struct {
	Avatar  string `json:"avatar"`  // 头像
	Name    string `json:"name"`    // 名称
	Email   string `json:"email"`   // 邮箱
	Job     string `json:"job"`     // 职务
	Address string `json:"address"` // 详细地址
}

type TenantIndustryParams struct {
}

type (
	// TenantPerfectParams 完善信息参数
	TenantPerfectParams struct {
		*TenantBasicParams      // 基本信息
		*TenantParamsForCompany // 公司身份信息
	}
	// TenantParamsForCompany 公司参数信息
	TenantParamsForCompany struct {
		Name string `json:"name"`
		Code string `json:"code"`
		config.Area
		model2.Position
		Industry  string `json:"industry"`
		Introduce string `json:"introduce"`
	}
)

var tenantHandlePerfect = map[int]func(params *TenantPerfectParams) (string, error){
	config.TenantUserIdentityForCompany: perfectForCompany,
}

func perfectForCompany(params *TenantPerfectParams) (string, error) {
	if params.TenantParamsForCompany == nil {
		return "", errors.New("企业信息异常")
	}
	return utils.AnyToJSON(params.TenantParamsForCompany), nil
}

func perfectForExpert() error {
	return nil
}

func perfectForResearch() error {
	return nil
}

// Perfect 完善信息
func (c *Tenant) Perfect(params *TenantPerfectParams) error {
	// 事件处理
	_handle, has := tenantHandlePerfect[c.SelectIdentity]

	if !has {
		return errors.New("未知的身份处理方式")
	}
	// 查询用户身份信息
	mUserTenant := model.NewUserTenant()

	isExist, err := model2.FirstField(mUserTenant.UserTenant, []string{"id", "name", "identity", "status"},
		model2.NewWhere("uid", c.UID), model2.NewWhere("identity", c.Identity))

	if err != nil {
		return err
	} else if isExist {
		if mUserTenant.Status != model2.UserTenantStatusForExamining {
			return errors.New("资料审核中，不可修改")
		} else if mUserTenant.Status != model2.UserTenantStatusForExaminePass {
			return errors.New("资料审核已通过，不可修改")
		}
	}
	mUserTenant.Avatar = params.TenantBasicParams.Avatar
	mUserTenant.Name = params.TenantBasicParams.Name
	mUserTenant.Email = params.TenantBasicParams.Email
	mUserTenant.Identity = c.SelectIdentity
	mUserTenant.Address = params.TenantBasicParams.Address

	if mUserTenant.Other, err = _handle(params); err != nil {
		return err
	}
	mUserTenant.Status = model2.UserTenantStatusForExamining

	if isExist {
		if err = model2.Updates(mUserTenant.UserTenant, mUserTenant.UserTenant); err != nil {
			return err
		}
		return nil
	}
	if err = orm.GetDB().Transaction(func(tx *gorm.DB) error {
		if err = model2.Updates(mUserTenant.UserTenant, map[string]interface{}{
			"selected": model2.UserTenantSelectedForNo, "updated_at": time.Now(),
		}); err != nil {
			return err
		}
		mUserTenant.Selected = model2.UserTenantSelectedForYes

		return model2.Create(mUserTenant.UserTenant, tx)
	}); err != nil {
		return err
	}
	return nil
}

// Auth 认证
func (c *Tenant) Auth() {

}

// Switch 切换身份
func (c *Tenant) Switch(identity int) error {
	if _, has := config.TenantUserIdentityData[identity]; !has {
		return errors.New("未知的身份信息")
	}
	if c.SelectIdentity == identity {
		return nil
	}
	// 已存在相应的身份，更新最后
	if c.Identity&identity > 0 {
		mUserTenant := model.NewUserTenant()
		// 查询用户身份信息
		now := time.Now()

		var err error

		if err = orm.GetDB().Transaction(func(tx *gorm.DB) error {
			if err = model2.UpdatesWhere(mUserTenant.UserTenant, map[string]interface{}{
				"selected": model2.UserTenantSelectedForNo, "updated_at": now,
			}, []*model2.ModelWhere{model2.NewWhere("uid", c.UID)}, tx); err != nil {
				return err
			}
			return model2.UpdatesWhere(mUserTenant.UserTenant, map[string]interface{}{
				"selected": model2.UserTenantSelectedForYes, "updated_at": now,
			}, []*model2.ModelWhere{model2.NewWhere("uid", c.UID),
				model2.NewWhere("identity", identity)}, tx)
		}); err != nil {
			return err
		}
	}
	c.SelectIdentity = identity
	service.Publish(config2.EventForRedisHashProduce, config2.RedisKeyForAccount, c.UIDToString(), c.SessionEnterprise)
	return nil
}

func NewTenant() TenantHandle {
	return func(session *service.SessionEnterprise) *Tenant {
		return &Tenant{session}
	}
}