Cold_Api/models/Account/Company.go

package Account

import (
	"Cold_Api/conf"
	"Cold_Api/controllers/lib"
	"Cold_Api/logs"
	"encoding/json"
	"fmt"
	"github.com/astaxie/beego/cache"
	_ "github.com/astaxie/beego/cache/redis"
	"github.com/beego/beego/v2/adapter/orm"
	orm2 "github.com/beego/beego/v2/client/orm"
	_ "github.com/go-sql-driver/mysql"
	"strconv"
	"time"
)

type Company struct {
	Id     int    `orm:"column(ID);size(11);auto;pk"`
	T_mid  int    `orm:"size(200);null"`       //  上一级 ID
	T_name string `orm:"size(256);null"`       // 公司名称
	T_key  string `orm:"size(256);index;null"` // 公司密钥

	T_plan string `orm:"type(text);null"` //  平面图
	T_data string `orm:"type(text);null"` //  大数据
	T_v3d  string `orm:"type(text);null"` //  3D 视图

	T_path string `orm:"size(256);null"` // 公司路径 /0/1/5/

	T_State    int       `orm:"size(200);default(1)"`                                  //  0删除  1 正常
	CreateTime time.Time `orm:"column(create_time);type(timestamp);null;auto_now_add"` //auto_now 每次 model 保存时都会对时间自动更新
	UpdateTime time.Time `orm:"column(update_time);type(timestamp);null;auto_now"`     //auto_now_add 第一次保存时才设置时间

	Children []Company `orm:"-"`
}

type Company_R struct {
	Id       int
	T_mid    int    // 上一级 ID
	T_name   string // 公司名称
	T_plan   string // 平面图
	T_data   string // 大数据
	T_v3d    string // 3D 视图
	Children []Company_R
}

func CompanyToCompany_R(r Company) (v Company_R) {
	v.Id = r.Id
	v.T_mid = r.T_mid
	v.T_name = r.T_name
	return v
}

func (t *Company) TableName() string {
	return "company" // 数据库名称   // ************** 替换 FormulaList **************
}

var redisCache_Company cache.Cache

func init() {
	//注册模型
	orm.RegisterModel(new(Company))

	config := fmt.Sprintf(`{"key":"%s","conn":"%s","dbNum":"%s","password":"%s"}`,
		"redis_User_Company", conf.Redis_address, conf.Redis_dbNum, conf.Redis_password)
	fmt.Println(config)
	var err error
	redisCache_Company, err = cache.NewCache("redis", config)
	if err != nil || redisCache_Company == nil {
		errMsg := "failed to init redis"
		logs.Error(errMsg, err)
		panic(errMsg)
	}

}

// ---------------- Redis -------------------
func Redis_Company_Set(r Company) (err error) {
	//json序列化
	str, err := json.Marshal(r)
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return
	}

	err = redisCache_Company.Put(strconv.Itoa(r.Id), str, 24*time.Hour)
	if err != nil {
		logs.Error("set key:", strconv.Itoa(r.Id), ",value:", str, err)
	}
	return
}

func Redis_Company_Get(key string) (r Company, is bool) {
	if redisCache_Company.IsExist(key) {
		//println("找到key:",key)
		v := redisCache_Company.Get(key)

		err := json.Unmarshal(v.([]byte), &r)
		if err != nil {
			logs.Error(lib.FuncName(), err)
			return Company{}, false
		}
		return r, true
	}
	//println("没有 找到key:",key)
	return Company{}, false
}

func Redis_Company_DelK(key string) (err error) {
	err = redisCache_Company.Delete(key)
	if err != nil {
		logs.Error(lib.FuncName(), err)
	}
	return
}

// ---------------- 特殊方法 -------------------

// 添加
func Add_Company(var_ Company) (id int64, err error) {
	o := orm.NewOrm()
	o.Begin()

	var rand_x int64
	for true {
		var_.T_key = lib.GetRandstring(16, "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789", rand_x)
		err = o.Read(&var_, "T_key") // o.Read(&r,"Tokey") 如果不是 主键 就得指定字段名
		if err != nil {
			break
		}
		rand_x++
	}

	id, err = o.Insert(&var_)
	if err != nil {
		o.Rollback()
		logs.Error(lib.FuncName(), err)
		return id, err
	}

	T_Path := "/0/" + strconv.Itoa(int(id)) + "/"
	if var_.T_mid != 0 {
		Company_r, err := Read_Company_ById(var_.T_mid)
		if err != nil {
			o.Rollback()
			logs.Error(lib.FuncName(), err)
			return id, err
		}
		T_Path = Company_r.T_path + strconv.Itoa(int(id)) + "/"
	}

	var_.Id = int(id)
	var_.T_path = T_Path
	_, err = o.Update(&var_, "T_path")
	if err != nil {
		o.Rollback()
		logs.Error(lib.FuncName(), err)
		return id, err
	}
	o.Commit()

	Redis_Company_Set(var_)
	return id, err

}

// 修改
func Update_Company(m Company, cols ...string) bool {
	o := orm.NewOrm()
	num, err := o.Update(&m, cols...)
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return false
	}
	fmt.Println("Number of records updated in database:", num)
	Redis_Company_Set(m) // Redis 更新缓存
	return true
}

// 删除
func Delete_Company(orm orm.Ormer, id int) bool {
	var m = Company{
		Id:      id,
		T_State: 0,
	}
	num, err := orm.Update(&m, "T_State")
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return false
	}
	fmt.Println("Number of records updated in database:", num)
	Redis_Company_DelK(strconv.Itoa(m.Id))
	return true
}

// 获取 ById
func Read_Company_ById(Id int) (r Company, e error) {
	if r, is := Redis_Company_Get(strconv.Itoa(Id)); is {
		//println("Redis_Get  OK")
		return r, nil
	}

	o := orm.NewOrm()
	qs := o.QueryTable(new(Company))
	e = qs.Filter("Id", Id).Filter("T_State", 1).One(&r)
	if e != nil {
		logs.Error("Read_Company_ById", e)
	}
	return r, e
}

// 获取列表
func Read_Company_Tree(admin_r Admin, T_name string) (CompanyList []Company_R) {

	// 内部用户未绑定公司
	if admin_r.T_pid > 0 || len(admin_r.T_pids) == 0 {
		return CompanyList
	}

	o := orm.NewOrm()
	// 也可以直接使用 Model 结构体作为表名

	qs := o.QueryTable(new(Company))
	var maps []Company

	cond := orm.NewCondition()
	cond1 := cond.And("T_State", 1)

	if admin_r.T_pid > 0 {
		cond1 = cond1.And("Id", admin_r.T_pid)
	}

	// 内部用户已绑定公司，* 绑定所有公司
	if admin_r.T_pids != "*" {
		T_pids := lib.SplitStringToIntIds(admin_r.T_pids, "P")
		cond1 = cond1.And("Id__in", ReadCompanyIds_T_pids(T_pids))
	}

	if len(T_name) > 0 {
		cond1 = cond1.And("T_name__icontains", T_name)
	}

	_, err := qs.SetCond((*orm2.Condition)(cond1)).OrderBy("-Id").All(&maps)
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return CompanyList
	}

	parentMap, flag := getCompanyParent(maps)
	if flag {
		for i := 0; i < len(maps); i++ {
			if parentMap[maps[i].T_mid] {
				continue
			}
			r := Company_R{
				Id:       maps[i].Id,
				T_mid:    maps[i].T_mid,
				T_name:   maps[i].T_name,
				Children: nil,
			}
			info := CompanyCall(maps, r)
			CompanyList = append(CompanyList, info)
		}
	} else {
		for i := 0; i < len(maps); i++ {
			r := Company_R{
				Id:       maps[i].Id,
				T_mid:    maps[i].T_mid,
				T_name:   maps[i].T_name,
				Children: nil,
			}
			info := CompanyCall(maps, r)
			CompanyList = append(CompanyList, info)
		}
	}

	return CompanyList
}

func ReadCompanyIds_T_pids(T_pids []int) []int {

	var companyList []Company
	var companyIds []int
	o := orm.NewOrm()
	qs := o.QueryTable(new(Company))
	_, err := qs.Filter("Id__in", T_pids).Filter("T_State", 1).All(&companyList)
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return companyIds
	}
	for _, company := range companyList {
		subIds := ReadCompanyIds_T_path(company.T_path)
		companyIds = append(companyIds, subIds...)
	}

	return lib.IntIdsDistinct(companyIds)
}

// 通过T_pid查询所有子id
func ReadCompanyIds_T_path(T_path string) (companyIds []int) {
	o := orm.NewOrm()
	qs := o.QueryTable(new(Company))
	var CompanyList []Company
	_, err := qs.Filter("T_path__startswith", T_path).Filter("T_State", 1).All(&CompanyList)
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return companyIds
	}
	for _, v := range CompanyList {
		companyIds = append(companyIds, v.Id)
	}
	return companyIds
}

func Read_Company_List(T_name string) (CompanyList []Company_R) {

	o := orm.NewOrm()
	// 也可以直接使用 Model 结构体作为表名

	qs := o.QueryTable(new(Company))
	var maps []Company

	cond := orm.NewCondition()
	cond1 := cond.And("T_State", 1)

	if len(T_name) > 0 {
		cond1 = cond1.And("T_name__icontains", T_name)
	}

	_, err := qs.SetCond((*orm2.Condition)(cond1)).OrderBy("-Id").All(&maps)
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return CompanyList
	}

	parentMap, flag := getCompanyParent(maps)
	if flag {
		for i := 0; i < len(maps); i++ {
			if parentMap[maps[i].T_mid] {
				continue
			}
			r := Company_R{
				Id:       maps[i].Id,
				T_mid:    maps[i].T_mid,
				T_name:   maps[i].T_name,
				Children: nil,
			}
			info := CompanyCall(maps, r)
			CompanyList = append(CompanyList, info)
		}
	} else {
		for i := 0; i < len(maps); i++ {
			r := Company_R{
				Id:       maps[i].Id,
				T_mid:    maps[i].T_mid,
				T_name:   maps[i].T_name,
				Children: nil,
			}
			info := CompanyCall(maps, r)
			CompanyList = append(CompanyList, info)
		}
	}

	return CompanyList
}

func CompanyCall(CompanyList []Company, company Company_R) Company_R {
	list := CompanyList
	min := make([]Company_R, 0)
	for j := 0; j < len(list); j++ {
		if company.Id != list[j].T_mid {
			continue
		}
		mi := Company_R{}
		mi.Id = list[j].Id
		mi.T_mid = list[j].T_mid
		mi.T_name = list[j].T_name
		mi.Children = []Company_R{}
		ms := CompanyCall(CompanyList, mi)
		min = append(min, ms)
	}
	company.Children = min
	return company
}

func getCompanyParent(CompanyList []Company) (map[int]bool, bool) {
	list := CompanyList
	var flag = false
	var parentMap = map[int]bool{}
	for j := 0; j < len(list); j++ {
		parentMap[list[j].T_mid] = false
	}

	for j := 0; j < len(list); j++ {
		if _, ok := parentMap[list[j].Id]; !ok {
			continue
		}
		parentMap[list[j].Id] = true
		flag = true
	}
	return parentMap, flag
}

func Read_Company_List_All_ByT_name(T_name string) (maps []Company) {
	o := orm.NewOrm()
	// 也可以直接使用 Model 结构体作为表名

	qs := o.QueryTable(new(Company))
	cond := orm.NewCondition()

	if len(T_name) > 0 {
		cond = cond.And("T_name__icontains", T_name)
	}
	_, err := qs.SetCond((*orm2.Condition)(cond)).All(&maps)
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return maps
	}
	return maps
}

// 获取内部用户绑定的公司列表
func Read_Company_List_ByT_pids(T_pids string) (CompanyList []Company_R) {

	o := orm.NewOrm()
	// 也可以直接使用 Model 结构体作为表名

	qs := o.QueryTable(new(Company))
	var maps []Company
	cond := orm.NewCondition()
	cond1 := cond.And("T_State", 1)
	if len(T_pids) >= 0 {
		list := lib.SplitStringIds(T_pids, "P")
		cond1 = cond1.And("Id__in", list)
	}

	_, err := qs.SetCond((*orm2.Condition)(cond1)).OrderBy("Id").All(&maps)
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return CompanyList
	}

	parentMap, flag := getCompanyParent(maps)
	if flag {
		for i := 0; i < len(maps); i++ {
			if parentMap[maps[i].T_mid] {
				continue
			}
			r := Company_R{
				Id:       maps[i].Id,
				T_mid:    maps[i].T_mid,
				T_name:   maps[i].T_name,
				Children: nil,
			}
			info := CompanyCall(maps, r)
			CompanyList = append(CompanyList, info)
		}
	} else {
		for i := 0; i < len(maps); i++ {
			r := Company_R{
				Id:       maps[i].Id,
				T_mid:    maps[i].T_mid,
				T_name:   maps[i].T_name,
				Children: nil,
			}
			info := CompanyCall(maps, r)
			CompanyList = append(CompanyList, info)
		}
	}

	return CompanyList
}