package service

import (
	"ems.agt/features/udm_user/repo"
	"ems.agt/lib/core/datasource"
	"ems.agt/lib/log"
)

// 实例化服务层 ServiceUdmSubUser 结构体
var NewServiceUdmSubUser = &ServiceUdmSubUser{
	repoAuthUser: *repo.NewRepoUdmAuthUser,
	repoSunUser:  *repo.NewRepoUdmSubUser,
}

// ServiceUdmSubUser UDM签约用户 服务层处理
type ServiceUdmSubUser struct {
	repoAuthUser repo.RepoUdmAuthUser
	repoSunUser  repo.RepoUdmSubUser
}

// SubUserSave UDM签约用户-获取全部保存数据库
func (r *ServiceUdmSubUser) SubUserSave() int64 {
	var num int64 = 0
	subArr := redisUdmSubUserList()
	// 有数据才清空
	if len(subArr) == 0 {
		return num
	}

	_, err := datasource.ExecDB("", "TRUNCATE TABLE u_sub_user", nil)
	if err != nil {
		log.Errorf("TRUNCATE err => %v", err)
	}

	for _, u := range subArr {
		results, err := datasource.DefaultDB().Table("u_sub_user").Insert(u)
		if err != nil {
			log.Errorf("Insert err => %v", err)
			return 0
		}
		num += results
	}
	return num
}

// SubUserSave UDM签约用户-分页查询数据库
func (r *ServiceUdmSubUser) SubUserList(query map[string]any) map[string]any {
	return r.repoSunUser.SelectPage(query)
}