hongze_cygx/models/wx_user_rai_label.go

package models

import (
	"fmt"
	"github.com/beego/beego/v2/client/orm"
	"strings"
	"time"
)

type WxUserRaiLabelRedis struct {
	UserId           int       `description:"用户ID"`
	SourceId         int       `description:"资源ID"`
	Label            string    `description:"标签内容"`
	SourceType       int       `description:"来源1:搜索关键字标签、2:产业/个股标签(线下活动)、3:产业/个股标签(线下路演)、4:产业/个股标签(线上活动)、5:产业/个股标签(线上路演)、6:销售输入标签、7:产业/个股标签(报告)、8:报告类型标签"`
	CreateTime       time.Time `description:"创建时间"`
	RegisterPlatform int       `description:"来源 1小程序，2：网页"`
	TableName        string    `description:"数据来源的表名"`
}

type WxUserRaiLabel struct {
	RaiLabelId       int       `orm:"column(rai_label_id);pk"`
	UserId           int       `description:"用户ID"`
	RealName         string    `description:"用户实际名称"`
	Mobile           string    `description:"手机号"`
	Email            string    `description:"邮箱"`
	CompanyId        int       `description:"公司id"`
	CompanyName      string    `description:"公司名称"`
	Label            string    `description:"标签内容"`
	SourceId         int       `description:"来源ID"`
	SourceType       int       `description:"来源1:搜索关键字标签、2:产业/个股标签(线下活动)、3:产业/个股标签(线下路演)、4:产业/个股标签(线上活动)、5:产业/个股标签(线上路演)、6:销售输入标签、7:产业/个股标签(报告)、8:报告类型标签"`
	SysUserId        int       `description:"创建人id"`
	SysUserRealName  string    `description:"创建人名称"`
	CreateTime       time.Time `description:"创建时间"`
	ModifyTime       time.Time `description:"更新时间"`
	RegisterPlatform int       `description:"来源 1小程序，2：网页"`
	TableName        string    `description:"数据来源的表名"`
}

// 添加
func AddWxUserRaiLabel(item *WxUserRaiLabel) (err error) {
	o, err := orm.NewOrm().Begin()
	if err != nil {
		return
	}
	defer func() {
		fmt.Println(err)
		if err == nil {
			o.Commit()
		} else {
			o.Rollback()
		}
	}()

	//删除原有数据
	sql := ` DELETE FROM wx_user_rai_label WHERE user_id = ? AND label = ?  `
	_, err = o.Raw(sql, item.UserId, item.Label).Exec()
	if err != nil {
		return
	}

	_, err = o.Insert(item)
	return
}

// 批量插入
func AddWxUserRaiLabelMulti(items []*WxUserRaiLabel, label []string, userId int) (err error) {
	o, err := orm.NewOrm().Begin()
	if err != nil {
		return
	}
	defer func() {
		if err == nil {
			o.Commit()
		} else {
			o.Rollback()
		}
	}()

	if len(label) > 0 {
		//删除老的数据信息
		sql := ` DELETE FROM wx_user_rai_label WHERE user_id = ? AND  label IN ('` + strings.Join(label, "','") + `')  `
		_, err = o.Raw(sql, userId).Exec()
		if err != nil {
			return
		}
	}

	//批量插入新的关联数据
	if len(items) > 0 {
		//批量添加流水信息
		_, err = o.InsertMulti(len(items), items)
		if err != nil {
			return
		}
	}

	return
}