eta_hub/models/smart_resource.go

package models

import (
	sql2 "database/sql"
	"eta/eta_hub/global"
	"eta/eta_hub/utils"
	"fmt"
	"github.com/rdlucklib/rdluck_tools/paging"
	"strings"
	"time"
)

type SmartReportResource struct {
	ResourceId int       `orm:"column(resource_id);pk" description:"智能研报资源ID"`
	ImgUrl     string    // 图片链接
	Style      string    // 版图样式
	ImgName    string    // 图片名称
	Type       int       // 类型 1-版头 2-版尾
	CreateTime time.Time // 创建时间
}

func (m *SmartReportResource) TableName() string {
	return "smart_report_resource"
}

func (m *SmartReportResource) PrimaryId() string {
	return "resource_id"
}

func (m *SmartReportResource) Create() (err error) {
	//o := orm.NewOrmUsingDB("rddp")
	err = global.DbMap[utils.MYSQL_URL_RDDP].Create(m).Error
	if err != nil {
		return
	}
	return
}

func (m *SmartReportResource) Update(cols []string) (err error) {
	//o := orm.NewOrmUsingDB("rddp")
	err = global.DbMap[utils.MYSQL_URL_RDDP].Select(cols).Updates(m).Error
	return
}

func (m *SmartReportResource) Del() (err error) {
	//o := orm.NewOrmUsingDB("rddp")
	sql := fmt.Sprintf(`DELETE FROM %s WHERE %s = ? LIMIT 1`, m.TableName(), m.PrimaryId())
	err = global.DbMap[utils.MYSQL_URL_RDDP].Exec(sql, m.ResourceId).Error
	return
}

type SmartReportResourceItem struct {
	ResourceId int    `orm:"column(resource_id);pk" description:"智能研报资源ID"`
	ImgUrl     string // 图片链接
	ImgName    string // 图片名称
	Style      string // 版图样式
	Type       int    // 类型 1-版头 2-版尾
	CreateTime string // 创建时间
}

// SmartReportResourceListResp 智能研报资源库
type SmartReportResourceListResp struct {
	List   []*SmartReportResourceItem
	Paging *paging.PagingItem `description:"分页数据"`
}

func (m *SmartReportResource) GetCountByCondition(condition string, pars []interface{}) (count int, err error) {
	//o := orm.NewOrmUsingDB("rddp")
	var totalNull sql2.NullInt64
	sql := fmt.Sprintf(`SELECT COUNT(1) FROM %s WHERE 1=1 %s`, m.TableName(), condition)
	err = global.DbMap[utils.MYSQL_URL_RDDP].Raw(sql, pars...).Scan(&totalNull).Error
	if err != nil {
		return
	}
	count = int(totalNull.Int64)

	return
}

func (m *SmartReportResource) GetPageItemsByCondition(condition string, pars []interface{}, fieldArr []string, startSize, pageSize int) (items []*SmartReportResourceItem, err error) {
	//o := orm.NewOrmUsingDB("rddp")
	fields := strings.Join(fieldArr, ",")
	if len(fieldArr) == 0 {
		fields = `*`
	}
	order := ` ORDER BY create_time DESC`

	sql := fmt.Sprintf(`SELECT %s FROM %s WHERE 1=1 %s %s LIMIT ?,?`, fields, m.TableName(), condition, order)
	pars = append(pars, startSize, pageSize)
	err = global.DbMap[utils.MYSQL_URL_RDDP].Raw(sql, pars...).Find(&items).Error

	return
}

// SmartReportResourceEditReq 智能研报资源编辑请求体
type SmartReportResourceEditReq struct {
	ResourceId int    `description:"资源ID"`
	ImgName    string `description:"图片名称"`
	Style      string `description:"版图样式"`
}

func (m *SmartReportResource) GetItemById(id int) (item *SmartReportResource, err error) {
	//o := orm.NewOrmUsingDB("rddp")
	sql := fmt.Sprintf(`SELECT * FROM %s WHERE %s = ? LIMIT 1`, m.TableName(), m.PrimaryId())
	err = global.DbMap[utils.MYSQL_URL_RDDP].Raw(sql, id).First(&item).Error
	return
}

// SmartReportResourceRemoveReq 删除智能研报资源请求体
type SmartReportResourceRemoveReq struct {
	ResourceIds string `description:"资源IDs"`
}

// SmartReportResourceAddReq 新增智能研报资源请求体
type SmartReportResourceAddReq struct {
	Type    int    `description:"类型 1-版头 2-版尾"`
	ImgUrl  string `description:"图片链接"`
	ImgName string `description:"图片名称"`
	Style   string `description:"版图样式"`
}

func GetResourceItemById(id int) (item *SmartReportResource, err error) {
	//o := orm.NewOrmUsingDB("rddp")
	sql := fmt.Sprintf(`SELECT * FROM smart_report_resource WHERE resource_id = ? LIMIT 1`)
	err = global.DbMap[utils.MYSQL_URL_RDDP].Raw(sql, id).First(&item).Error
	return
}