Cold_Api/models/AfterSales/AfterSales.go

package AfterSales

import (
	"Cold_Api/conf"
	"Cold_Api/controllers/lib"
	"encoding/json"
	"fmt"
	"strconv"
	"time"

	"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/beego/beego/v2/core/logs"
	_ "github.com/go-sql-driver/mysql"
)

// 附件结构体
type Attachment struct {
	Name string `json:"name"` // 文件名称
	Url  string `json:"url"`  // 文件路径/URL
}

// 售后服务主表
type AfterSales struct {
	Id            int    `orm:"column(ID);size(11);auto;pk"`
	T_name        string `orm:"size(256);null"`      // 标题
	T_category    int    `orm:"index;size(11);null"` // 分类ID，关联AfterSalesCategory表
	T_content     string `orm:"type(text);null"`     // 内容
	T_attachments string `orm:"size(1000);null"`     // 附件列表，用JSON存储文件路径数组
	T_sort        int    `orm:"size(11);default(0)"` // 排序
	T_display     int    `orm:"size(2);default(1)"`  // 是否显示: 2隐藏 1显示
	T_State       int    `orm:"size(2);default(1)"`  // 0 删除   1 正常

	CreateTime time.Time `orm:"column(create_time);type(timestamp);null;auto_now_add"` //auto_now_add 第一次保存时才设置时间
	UpdateTime time.Time `orm:"column(update_time);type(timestamp);null;auto_now"`     //auto_now 每次 model 保存时都会对时间自动更新
}

// 返回结构体
type AfterSales_R struct {
	Id              int          `json:"id"`
	T_name          string       `json:"t_name"`          // 服务名称
	T_category      int          `json:"t_category"`      // 分类
	T_category_name string       `json:"t_category_name"` // 分类名称
	T_content       string       `json:"t_content"`       // 内容
	T_attachments   []Attachment `json:"t_attachments"`   // 附件列表
	T_sort          int          `json:"t_sort"`          // 排序
	T_display       int          `json:"t_display"`       // 是否显示
	T_State         int          `json:"t_state"`         // 状态
	CreateTime      time.Time    `json:"create_time"`     // 创建时间
	UpdateTime      time.Time    `json:"update_time"`     // 更新时间
}

// 简化返回结构体
type AfterSales_Simple struct {
	Id              int       `json:"id"`
	T_name          string    `json:"t_name"`          // 服务名称
	T_category      int       `json:"t_category"`      // 分类
	T_category_name string    `json:"t_category_name"` // 分类名称
	T_sort          int       `json:"t_sort"`          // 排序
	CreateTime      time.Time `json:"create_time"`     // 创建时间
}

func (t *AfterSales) TableName() string {
	return "after_sales"
}

var redisCache_AfterSales cache.Cache

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

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

// ---------------- Redis 缓存方法 -------------------
func Redis_AfterSales_Set(r AfterSales) (err error) {
	key := strconv.Itoa(r.Id)

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

	err = redisCache_AfterSales.Put(key, str, 2*time.Hour)
	if err != nil {
		logs.Error("set key:", key, ",value:", str, err)
	}
	return
}

func Redis_AfterSales_Get(key string) (AfterSales, bool) {
	if redisCache_AfterSales.IsExist(key) {
		v := redisCache_AfterSales.Get(key)
		var r AfterSales
		err := json.Unmarshal(v.([]byte), &r)
		if err != nil {
			logs.Error(lib.FuncName(), err)
			return AfterSales{}, false
		}
		return r, true
	}
	return AfterSales{}, false
}

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

// ---------------- 转换方法 -------------------
func AfterSalesToAfterSales_R(t AfterSales) (r AfterSales_R) {
	r.Id = t.Id
	r.T_name = t.T_name
	r.T_category = t.T_category

	// 从分类表获取分类名称
	if t.T_category > 0 {
		if category, err := Read_AfterSalesCategory_ById(t.T_category); err == nil {
			r.T_category_name = category.T_name
		} else {
			r.T_category_name = "未知分类"
		}
	} else {
		r.T_category_name = "无分类"
	}

	r.T_content = t.T_content
	r.T_sort = t.T_sort
	r.T_display = t.T_display
	r.T_State = t.T_State
	r.CreateTime = t.CreateTime
	r.UpdateTime = t.UpdateTime

	// 解析附件JSON
	if len(t.T_attachments) > 0 {
		err := json.Unmarshal([]byte(t.T_attachments), &r.T_attachments)
		if err != nil {
			logs.Error(lib.FuncName(), err)
			r.T_attachments = []Attachment{}
		}
	} else {
		r.T_attachments = []Attachment{}
	}

	return r
}

func AfterSalesToAfterSales_Simple(t AfterSales) (r AfterSales_Simple) {
	r.Id = t.Id
	r.T_name = t.T_name
	r.T_category = t.T_category

	// 从分类表获取分类名称
	if t.T_category > 0 {
		if category, err := Read_AfterSalesCategory_ById(t.T_category); err == nil {
			r.T_category_name = category.T_name
		} else {
			r.T_category_name = "未知分类"
		}
	} else {
		r.T_category_name = "无分类"
	}

	r.T_sort = t.T_sort
	r.CreateTime = t.CreateTime
	return r
}

// ---------------- CRUD 方法 -------------------

// 添加售后服务
func Add_AfterSales(m AfterSales) (id int64, err error) {
	o := orm.NewOrm()
	id, err = o.Insert(&m)
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return
	}
	m.Id = int(id)
	Redis_AfterSales_Set(m)
	return
}

// 根据ID获取售后服务
func Read_AfterSales_ById(id int) (r AfterSales, err error) {
	key := strconv.Itoa(id)
	if r, is := Redis_AfterSales_Get(key); is {
		return r, nil
	}
	o := orm.NewOrm()
	r = AfterSales{Id: id, T_State: 1}
	err = o.Read(&r, "Id", "T_State")
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return r, err
	}
	Redis_AfterSales_Set(r)
	return r, err
}

// 修改售后服务
func Update_AfterSales(r AfterSales, cols ...string) bool {
	o := orm.NewOrm()
	num, err := o.Update(&r, cols...)
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return false
	}
	logs.Info("Number of records updated in database:", num)
	Redis_AfterSales_Set(r)
	return true
}

// 删除售后服务（软删除）
func Delete_AfterSales_ById(id int) bool {
	o := orm.NewOrm()
	v := AfterSales{Id: id}
	if err := o.Read(&v); err == nil {
		var num int64
		v.T_State = 0
		num, err = o.Update(&v, "T_State")
		if err != nil {
			logs.Error(lib.FuncName(), err)
			return false
		}
		logs.Info("Number of records updated in database:", num)
		key := strconv.Itoa(v.Id)
		Redis_AfterSales_DelK(key)
		return true
	}
	return false
}

// 获取售后服务列表
func Read_AfterSales_List(T_name string, T_category int, T_display int, page int, page_z int) (r []AfterSales_R, cnt int64) {
	o := orm.NewOrm()
	var map_r []AfterSales
	qs := o.QueryTable(new(AfterSales))
	var offset int64
	if page <= 1 {
		offset = 0
	} else {
		offset = int64((page - 1) * page_z)
	}

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

	if len(T_name) > 0 {
		cond1 = cond1.And("T_name__icontains", T_name)
	}
	if T_category > 0 {
		cond1 = cond1.And("T_category", T_category)
	}
	if T_display > 0 { // T_display可以为0（隐藏）或1（显示），-1或其他值表示不过滤
		cond1 = cond1.And("T_display", T_display)
	}

	_, err := qs.Limit(page_z, offset).SetCond((*orm2.Condition)(cond1)).OrderBy("T_sort", "-Id").All(&map_r)
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return
	}
	cnt, err = qs.SetCond((*orm2.Condition)(cond1)).Count()
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return
	}

	for _, v := range map_r {
		r = append(r, AfterSalesToAfterSales_R(v))
	}

	return r, cnt
}

// 获取所有售后服务（简化版本，用于下拉选择等）
func Read_AfterSales_All(T_category int) (r []AfterSales_Simple) {
	o := orm.NewOrm()
	var map_r []AfterSales
	qs := o.QueryTable(new(AfterSales))

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

	if T_category > 0 {
		cond1 = cond1.And("T_category", T_category)
	}

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

	for _, v := range map_r {
		r = append(r, AfterSalesToAfterSales_Simple(v))
	}

	return r
}

// 根据分类获取售后服务数量
func Read_AfterSales_Count_ByCategory(T_pid int, T_category int) int64 {
	o := orm.NewOrm()
	qs := o.QueryTable(new(AfterSales))

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

	cnt, err := qs.SetCond((*orm2.Condition)(cond1)).Count()
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return 0
	}
	return cnt
}