package rag
import (
"database/sql"
"eta/eta_api/global"
"eta/eta_api/utils"
"fmt"
"time"
)
type WechatPlatform struct {
WechatPlatformId int `gorm:"column:wechat_platform_id;type:int(10) UNSIGNED;primaryKey;not null;" description:"wechat_platform_id"`
FakeId string `gorm:"column:fake_id;type:varchar(255);comment:公众号唯一id;" description:"fake_id"` // 公众号唯一id
Nickname string `gorm:"column:nickname;type:varchar(255);comment:公众号名称;" description:"nickname"` // 公众号名称
Alias string `gorm:"column:alias;type:varchar(255);comment:别名;" description:"alias"` // 别名
RoundHeadImg string `gorm:"column:round_head_img;type:varchar(255);comment:头像;" description:"round_head_img"` // 头像
ServiceType int `gorm:"column:service_type;type:int(11);comment:类型;default:0;" description:"service_type"` // 类型
Signature string `gorm:"column:signature;type:varchar(255);comment:签名;" description:"signature"` // 签名
Verified int `gorm:"column:verified;type:int(11);comment:是否认证，0：未认证，1：已认证；这个我不确定，再核实下;default:0;" description:"verified"` // 是否认证，0：未认证，1：已认证；这个我不确定，再核实下
ArticleLink string `gorm:"column:article_link;type:varchar(255);comment:添加公众时的文章链接;" description:"article_link"` // 添加公众时的文章链接
Enabled int `gorm:"column:enabled;type:tinyint(9);comment:是否启用，0：禁用，1：启用;default:1;" description:"enabled"` // 是否启用，0：禁用，1：启用
SysUserId int `gorm:"column:sys_user_id;type:int(9) UNSIGNED;comment:用户id;default:0;" description:"sys_user_id"` // 用户id
ModifyTime time.Time `gorm:"column:modify_time;type:datetime;comment:最后一次修改时间;default:NULL;" description:"modify_time"` // 最后一次修改时间
CreateTime time.Time `gorm:"column:create_time;type:datetime;comment:添加时间;default:NULL;" description:"create_time"` // 添加时间
}
// TableName get sql table name.获取数据库表名
func (m *WechatPlatform) TableName() string {
return "wechat_platform"
}
// WechatPlatformColumns get sql column name.获取数据库列名
var WechatPlatformColumns = struct {
WechatPlatformID string
FakeID string
Nickname string
Alias string
RoundHeadImg string
ServiceType string
Signature string
Verified string
ArticleLink string
Enabled string
SysUserID string
ModifyTime string
CreateTime string
}{
WechatPlatformID: "wechat_platform_id",
FakeID: "fake_id",
Nickname: "nickname",
Alias: "alias",
RoundHeadImg: "round_head_img",
ServiceType: "service_type",
Signature: "signature",
Verified: "verified",
ArticleLink: "article_link",
Enabled: "enabled",
SysUserID: "sys_user_id",
ModifyTime: "modify_time",
CreateTime: "create_time",
}
func (m *WechatPlatform) Create() (err error) {
err = global.DbMap[utils.DbNameAI].Create(&m).Error
return
}
func (m *WechatPlatform) Update(updateCols []string) (err error) {
err = global.DbMap[utils.DbNameAI].Select(updateCols).Updates(&m).Error
return
}
func (m *WechatPlatform) Del() (err error) {
err = global.DbMap[utils.DbNameAI].Delete(&m).Error
return
}
func (m *WechatPlatform) GetById(wechatPlatformId int) (item *WechatPlatform, err error) {
err = global.DbMap[utils.DbNameAI].Where(fmt.Sprintf("%s = ?", WechatPlatformColumns.WechatPlatformID), wechatPlatformId).First(&item).Error
return
}
func (m *WechatPlatform) GetByCondition(condition string, pars []interface{}) (item *WechatPlatform, err error) {
sqlStr := fmt.Sprintf(`SELECT * FROM %s WHERE 1=1 %s`, m.TableName(), condition)
err = global.DbMap[utils.DbNameAI].Raw(sqlStr, pars...).First(&item).Error
return
}
func (m *WechatPlatform) GetListByCondition(condition string, pars []interface{}, startSize, pageSize int) (items []*WechatPlatform, err error) {
sqlStr := fmt.Sprintf(`SELECT * FROM %s WHERE 1=1 %s LIMIT ?,?`, m.TableName(), condition)
pars = append(pars, startSize, pageSize)
err = global.DbMap[utils.DbNameAI].Raw(sqlStr, pars...).Find(&items).Error
return
}
func (m *WechatPlatform) GetCountByCondition(condition string, pars []interface{}) (total int, err error) {
var intNull sql.NullInt64
sqlStr := fmt.Sprintf(`SELECT COUNT(1) total FROM %s WHERE 1=1 %s`, m.TableName(), condition)
err = global.DbMap[utils.DbNameAI].Raw(sqlStr, pars...).Scan(&intNull).Error
if err == nil && intNull.Valid {
total = int(intNull.Int64)
}
return
}
func (m *WechatPlatform) GetPageListByCondition(condition string, pars []interface{}, startSize, pageSize int) (total int, items []*WechatPlatform, err error) {
total, err = m.GetCountByCondition(condition, pars)
if err != nil {
return
}
if total > 0 {
items, err = m.GetListByCondition(condition, pars, startSize, pageSize)
}
return
}
func (m *WechatPlatform) GetByFakeID(fakeId string) (item *WechatPlatform, err error) {
err = global.DbMap[utils.DbNameAI].Where(fmt.Sprintf("%s = ?", WechatPlatformColumns.FakeID), fakeId).First(&item).Error
return
}
// Add
// @Description: 添加一个新的公众号
// @author: Roc
// @receiver m
// @datetime 2025-03-04 17:48:30
// @param tagIdList []int
// @return err error
func (m *WechatPlatform) Add(tagIdList []int) (err error) {
tx := global.DbMap[utils.DbNameAI].Begin()
defer func() {
if err != nil {
_ = tx.Rollback()
} else {
_ = tx.Commit()
}
}()
err = tx.Create(&m).Error
if err != nil {
return
}
// 标签与公众号关系
if len(tagIdList) > 0 {
addTagMappingList := make([]*WechatPlatformTagMapping, 0)
for _, tagId := range tagIdList {
addTagMappingList = append(addTagMappingList, &WechatPlatformTagMapping{
WechatPlatformTagMappingId: 0,
WechatPlatformId: m.WechatPlatformId,
TagId: tagId,
ModifyTime: time.Now(),
CreateTime: time.Now(),
})
}
err = tx.CreateInBatches(addTagMappingList, utils.MultiAddNum).Error
if err != nil {
return
}
}
// 用户与公众号关系
userMapping := &WechatPlatformUserMapping{
WechatPlatformUserMappingId: 0,
WechatPlatformId: m.WechatPlatformId,
SysUserId: m.SysUserId,
ModifyTime: time.Now(),
CreateTime: time.Now(),
}
err = tx.Create(userMapping).Error
return
}