Cold_Api/models/Admin/Admin.go

package Admin

import (
	"Cold_Api/conf"
	"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"
	"time"
)

type Admin struct {
	Id           int    `orm:"column(ID);size(11);auto;pk"`
	Admin_uuid   string `orm:"size(256);null"` //
	Admin_master int    `orm:"size(200);null"` //  上一级
	Admin_power  int    `orm:"size(2);0"`      // 权限
	Admin_name   string `orm:"size(256);null"` //
	Admin_user   string `orm:"size(256);null"` //
	Admin_pass   string `orm:"size(256);null"` //

	Admin_telephone string `orm:"size(256);null"`  // 电话
	Admin_mail      string `orm:"size(200);null"`  //  邮箱
	Admin_gender    int    `orm:"size(2);null"`    // 性别  0未知   1男   2女
	Admin_text      string `orm:"type(text);null"` // 详情

	Admin_panel_plan string `orm:"size(200);null"`  //  平面图
	Admin_panel_data string `orm:"size(200);null"`  //  大数据
	Admin_panel_v3d  string `orm:"type(text);null"` //  3D 视图
	Admin_panel_v3d1 string `orm:"type(text);null"` //  3D 视图 1
	Admin_panel_v3d2 string `orm:"type(text);null"` //  3D 视图 2
	Admin_panel_v3d3 string `orm:"type(text);null"` //  3D 视图 3
	Admin_panel_v3d4 string `orm:"type(text);null"` //  3D 视图 4
	Admin_panel_v3d5 string `orm:"type(text);null"` //  3D 视图 5

	Admin_rh int `orm:"type(text);1"` // 显示湿度  0 不显示   1  显示

	T_State    int       `orm:"size(200);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 第一次保存时才设置时间
}
type Admin_R struct {
	Admin_uuid       string //
	Admin_master     int    //  上一级
	Admin_power      int    // 权限
	Admin_name       string //
	Admin_user       string //
	Admin_telephone  string // 电话
	Admin_mail       string //  邮箱
	Admin_gender     int    // 性别  0未知   1男   2女
	Admin_text       string // 详情
	Admin_rh         int    // 显示湿度
	Admin_panel_plan string //  平面图
	Admin_panel_data string //  大数据
	Admin_panel_v3d  string //  大数据
	Admin_panel_v3d1 string //  大数据
	Admin_panel_v3d2 string //  大数据
	Admin_panel_v3d3 string //  大数据
	Admin_panel_v3d4 string //  大数据
	Admin_panel_v3d5 string //  大数据
}

func Admin_To_Admin_R(r Admin) (v Admin_R) {
	v.Admin_uuid = r.Admin_uuid
	v.Admin_master = r.Admin_master
	v.Admin_power = r.Admin_power
	v.Admin_name = r.Admin_name
	v.Admin_user = r.Admin_user
	v.Admin_telephone = r.Admin_telephone
	v.Admin_mail = r.Admin_mail
	v.Admin_gender = r.Admin_gender
	v.Admin_text = r.Admin_text
	v.Admin_rh = r.Admin_rh
	v.Admin_panel_plan = r.Admin_panel_plan
	v.Admin_panel_data = r.Admin_panel_data
	v.Admin_panel_v3d = r.Admin_panel_v3d
	v.Admin_panel_v3d1 = r.Admin_panel_v3d1
	v.Admin_panel_v3d2 = r.Admin_panel_v3d2
	v.Admin_panel_v3d3 = r.Admin_panel_v3d3
	v.Admin_panel_v3d4 = r.Admin_panel_v3d4
	v.Admin_panel_v3d5 = r.Admin_panel_v3d5

	return v
}

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

var redisCache_Admin cache.Cache

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

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

}

// ---------------- Redis -------------------
//Redis_Set(m.T_sn,m) // Redis 更新缓存
func Redis_Admin_Set(r Admin) (err error) {
	//json序列化
	str, err := json.Marshal(r)
	if err != nil {
		fmt.Print(err)
		return
	}

	err = redisCache_Admin.Put(r.Admin_uuid, str, 24*time.Hour)
	if err != nil {
		fmt.Println("set key:", r.Admin_uuid, ",value:", str, err)
	}
	return
}

//if r,is :=Redis_Get(T_sn);is{
//return r,nil
//}
func Redis_Admin_Get(key string) (r Admin, is bool) {
	if redisCache_Admin.IsExist(key) {
		//println("找到key:",key)
		v := redisCache_Admin.Get(key)

		json.Unmarshal(v.([]byte), &r)
		return r, true
	}
	//println("没有 找到key:",key)
	return Admin{}, false
}

func Redis_Admin_DelK(key string) (err error) {
	err = redisCache_Admin.Delete(key)
	return
}

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

// 添加
func Add_Admin(m Admin) (id int64, err error) {
	o := orm.NewOrm()
	id, err = o.Insert(&m)
	if err != nil {
		fmt.Println(err)
	}
	m.Id = int(id)
	Redis_Admin_Set(m)
	return id, err
}

//// 修改
//func Update_Delete_Admin_ByAdmin_uuid(Admin_uuid string) (err error) {
//	o := orm.NewOrm()
//	v := Admin{Admin_uuid: Admin_uuid}
//	// ascertain id exists in the database
//	if err = o.Read(&v, "Admin_uuid"); err == nil {
//		var num int64
//		v.T_State = 0
//
//		if num, err = o.Update(&v, "T_State"); err == nil {
//			fmt.Println("Number of records updated in database:", num)
//		}
//
//		Redis_Admin_Set(v)
//	}
//	return err
//}

func Read_Admin_Admin_user(Admin_user string) (error, Admin) {

	o := orm.NewOrm()
	r := Admin{Admin_user: Admin_user, T_State: 1}
	err := o.Read(&r, "Admin_user", "T_State") // o.Read(&r,"Tokey") 如果不是 主键 就得指定字段名
	if err != nil {
		fmt.Println(err)
	}
	return err, r
}
func ReadLogin_verification(Admin_user string, Admin_pass string) (error, Admin) {

	o := orm.NewOrm()
	r := Admin{Admin_user: Admin_user, Admin_pass: Admin_pass, T_State: 1}
	err := o.Read(&r, "Admin_user", "Admin_pass", "T_State") // o.Read(&r,"Tokey") 如果不是 主键 就得指定字段名
	if err != nil {
		fmt.Println(err)
	}
	Redis_Admin_Set(r) // Redis 更新缓存
	return err, r
}

// 修改
func Update_Admin_ById_Basics(m Admin) (err error) {
	o := orm.NewOrm()
	v := Admin{Admin_uuid: m.Admin_uuid, T_State: 1}
	// ascertain id exists in the database
	if err = o.Read(&v, "Admin_uuid", "T_State"); err == nil {
		var num int64
		if len(m.Admin_name) > 0 {
			v.Admin_name = m.Admin_name
		}
		if len(m.Admin_mail) > 0 {
			v.Admin_mail = m.Admin_mail
		}
		if len(m.Admin_telephone) > 0 {
			v.Admin_telephone = m.Admin_telephone
		}
		if len(m.Admin_text) > 0 {
			v.Admin_text = m.Admin_text
		}
		v.Admin_gender = m.Admin_gender
		if m.Admin_power > 0 {
			v.Admin_power = m.Admin_power
		}
		if len(m.Admin_pass) > 0 {
			v.Admin_pass = m.Admin_pass
		}
		println("m.Admin_pass:", m.Admin_pass)
		if num, err = o.Update(&v, "Admin_name", "Admin_telephone", "Admin_mail", "Admin_text", "Admin_gender", "Admin_power", "Admin_pass"); err == nil {
			fmt.Println("Number of records updated in database:", num)
		}
		Redis_Admin_Set(m) // Redis 更新缓存
	}
	return err
}

// 修改
func Update_Admin(m Admin, cols ...string) bool {
	o := orm.NewOrm()
	if num, err := o.Update(&m, cols...); err == nil {
		fmt.Println("Number of records updated in database:", num)
		Redis_Admin_Set(m) // Redis 更新缓存
		return true
	}
	return false
}

// 获取 ById
func Read_Admin_ByAdmin_uuid(Admin_uuid string) (e error, r Admin) {
	if r, is := Redis_Admin_Get(Admin_uuid); is {
		//println("Redis_Get  OK")
		return nil, r
	}

	o := orm.NewOrm()
	qs := o.QueryTable(new(Admin))
	e = qs.Filter("Admin_uuid", Admin_uuid).Filter("T_State", 1).One(&r)
	return e, r
}

// 获取 ById
func Read_Admin_ByUuid(Admin_uuid string) (r Admin, err error) {
	if r, is := Redis_Admin_Get(Admin_uuid); is {
		//println("Redis_Get  OK")
		return r, nil
	}

	o := orm.NewOrm()
	r = Admin{Admin_uuid: Admin_uuid, T_State: 1}
	err = o.Read(&r, "Admin_uuid", "T_State") // o.Read(&r,"Tokey") 如果不是 主键 就得指定字段名
	if err != nil {
		fmt.Println(err)
	}
	return r, err
}

// 获取 ById
func Read_Admin_ById(id int) (r Admin, err error) {
	o := orm.NewOrm()
	r = Admin{Id: id}
	err = o.Read(&r) // o.Read(&r,"Tokey") 如果不是 主键 就得指定字段名
	if err != nil {
		fmt.Println(err)
	}
	return r, err
}

// 获取 ById
func Read_Admin_ById_x(Id int) (r Admin) {
	o := orm.NewOrm()
	qs := o.QueryTable(new(Admin))
	qs.Filter("Id", Id).One(&r, "Id", "Admin_uuid", "Admin_master", "Admin_name", "Admin_user", "CreateTime", "UpdateTime")
	return r
}

// 获取列表
func Read_Admin_List_ALL_User(page int, Admin_name string, Admin_master int) (maps []Admin, cnt int64) {

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

	qs := o.QueryTable(new(Admin))
	var offset int64
	if page <= 1 {
		offset = 0
	} else {
		offset = int64((page - 1) * conf.Page_size)
	}

	qs.Limit(conf.Page_size, offset).Filter("Admin_name__icontains", Admin_name).Exclude("Admin_master", 0).Filter("Admin_master", Admin_master).Filter("T_State", 1).OrderBy("-Id").All(&maps)
	cnt, _ = qs.Filter("Admin_name__icontains", Admin_name).Filter("Admin_master", Admin_master).Filter("T_State", 1).Count()

	return maps, cnt
}

// 获取列表
func Read_v2_Admin_List(Admin_master int, Admin_name string, page int, page_z int) (maps []Admin, cnt int64) {

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

	qs := o.QueryTable(new(Admin))
	var offset int64
	if page_z == 0 {
		page_z = conf.Page_size
	}
	if page <= 1 {
		offset = 0
	} else {
		offset = int64((page - 1) * page_z)
	}
	cond := orm.NewCondition()
	cond1 := cond.And("T_State", 1) // .AndNot("status__in", 1).Or("profile__age__gt", 2000)

	if Admin_master > 0 {
		cond1 = cond1.AndCond(cond.And("Admin_master", Admin_master))
	}

	if len(Admin_name) > 0 {
		cond1 = cond1.AndCond(cond.Or("Admin_name__icontains", Admin_name).Or("Admin_user__icontains", Admin_name))
	}

	qs.Limit(page_z, offset).SetCond((*orm2.Condition)(cond1)).OrderBy("-Id").All(&maps)
	cnt, _ = qs.SetCond((*orm2.Condition)(cond1)).Count()

	return maps, cnt
}

// 获取列表
func Read_Admin_List_ALL_Admin_master(page int, Admin_name string, Admin_master int) (maps []Admin, cnt int64) {

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

	qs := o.QueryTable(new(Admin))
	var offset int64
	if page <= 1 {
		offset = 0
	} else {
		offset = int64((page - 1) * conf.Page_size)
	}

	qs.Limit(conf.Page_size, offset).Filter("Admin_name__icontains", Admin_name).Filter("Admin_master", Admin_master).Filter("T_State", 1).OrderBy("-Id").All(&maps)
	cnt, _ = qs.Filter("Admin_name__icontains", Admin_name).Filter("Admin_master", Admin_master).Filter("T_State", 1).Count()

	return maps, cnt
}

// 获取列表 - 权限相同
func Read_Admin_List_ALL_AdminPower(AdminPower int) (maps []Admin) {

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

	qs := o.QueryTable(new(Admin))
	qs.Filter("Admin_power", AdminPower).Filter("T_State", 1).OrderBy("Id").All(&maps)
	//cnt, _ := qs.Filter("Id", AdminPower).Count()

	return maps
}

// 获取列表 - 权限相同
func Read_Admin_List_Admin_master_ALL(Admin_master int) (maps []Admin) {

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

	qs := o.QueryTable(new(Admin))
	qs.Filter("Admin_master", Admin_master).Filter("T_State", 1).Filter("Admin_power", 6).OrderBy("Id").All(&maps)
	//cnt, _ := qs.Filter("Id", AdminPower).Count()

	return maps
}

// 获取列表 - 权限相同
func Read_Admin_master_next_ALL(Admin_master int) (maps []Admin) {

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

	qs := o.QueryTable(new(Admin))
	qs.Filter("Admin_master", Admin_master).Filter("T_State", 1).OrderBy("Id").All(&maps)
	//cnt, _ := qs.Filter("Id", AdminPower).Count()

	return maps
}