eta_mini_crm/models/report_push_status.go

package models

import (
	"eta/eta_mini_crm/utils"
	"time"

	"github.com/beego/beego/v2/client/orm"
)

type ReportPushStatus struct {
	ReportPushStatusId int       `orm:"pk"`
	ReportId           int       `description:"报告id"`
	State              int       `description:"报告状态:0-未推送，1-已推送"`
	Title              string    `description:"报告标题"`
	Abstract           string    `description:"报告摘要"`
	Stage              int       `description:"期数"`
	ClassifyIdFirst    int       `description:"一级分类id"`
	ClassifyNameFirst  string    `description:"一级分类名称"`
	ClassifyIdSecond   int       `description:"二级分类id"`
	ClassifyNameSecond string    `description:"二级分类名称"`
	ClassifyIdThird    int       `description:"三级分类id"`
	ClassifyNameThird  string    `description:"三级分类名称"`
	Author             string    `description:"报告作者"`
	ReportType         int       `description:"报告类型:1-eta报告 2-pdf报告"`
	PublishTime        time.Time `description:"报告发布时间"`
	CreateTime         time.Time `description:"创建时间"`
	ModifyTime         time.Time `description:"修改时间"`
	PushTime           time.Time `description:"推送时间"`
}

type Report struct {
	Id                 int       `description:"报告id"`   // id
	ClassifyIdFirst    int       `description:"一级分类id"` // 一级分类id
	ClassifyNameFirst  string    `description:"一级分类名称"` // 一级分类名称
	ClassifyIdSecond   int       `description:"二级分类id"` // 二级分类id
	ClassifyNameSecond string    `description:"二级分类名称"` // 二级分类名称
	ClassifyIdThird    int       `description:"三级分类id"` // 三级分类id
	ClassifyNameThird  string    `description:"三级分类名称"` // 三级分类名称
	Title              string    `description:"报告标题"`   // 标题
	Abstract           string    `description:"摘要"`     // 摘要
	Author             string    `description:"作者"`     // 作者
	CreateTime         time.Time `description:"创建时间"`   // 创建时间
	ModifyTime         time.Time `description:"修改时间"`   // 修改时间
	PublishTime        time.Time `description:"发布时间"`   // 发布时间
	Stage              int       `description:"期数"`     // 期数
	ContentModifyTime  time.Time `description:"内容更新时间"` // 内容更新时间
	ReportCreateTime   time.Time `description:"创建时间"`   // 报告时间创建时间
}

type ReportPushView struct {
	ReportPushStatusId int    `orm:"pk"`
	ReportId           int    `description:"报告id"`
	Title              string `description:"报告标题"`
	Abstract           string `description:"报告摘要"`
	ClassifyIdFirst    int    `description:"一级分类id"`
	ClassifyNameFirst  string `description:"一级分类名称"`
	ClassifyIdSecond   int    `description:"二级分类id"`
	ClassifyNameSecond string `description:"二级分类名称"`
	ClassifyIdThird    int    `description:"二级分类id"`
	ClassifyNameThird  string `description:"二级分类名称"`
	Author             string `description:"报告作者"`
	State              int    `description:"报告状态:0-未推送，1-已推送"`
	PushTime           string `description:"推送时间"`
	PublishTime        string `description:"报告发布时间"`
	ReportType         int    `description:"报告类型:1-eta报告 2-pdf报告"`
	CreateTime         string `description:"创建时间"`
	ModifyTime         string `description:"修改时间"`
}

func (r *ReportPushStatus) Insert() (insertId int64, err error) {
	o := orm.NewOrm()
	insertId, err = o.Insert(r)
	return
}

func (r *ReportPushStatus) MultiInsert(items []*ReportPushStatus) (err error) {
	o := orm.NewOrm()
	_, err = o.InsertMulti(500, items)
	return
}

func (r *ReportPushStatus) Update(cols []string) (err error) {
	o := orm.NewOrm()
	_, err = o.Update(r, cols...)
	return
}

func GetReportPushStatusByReportId(reportId, state int) (item *ReportPushStatus, err error) {
	o := orm.NewOrm()
	sql := `SELECT * FROM report_push_status WHERE report_id=? AND state=?`
	err = o.Raw(sql, reportId, state).QueryRow(&item)
	return
}

func GetReportPushStatusByReportIdAndState(reportId []int, state int) (items []*ReportPushStatus, err error) {
	if len(reportId) == 0 {
		return
	}
	o := orm.NewOrm()
	sql := `SELECT * FROM report_push_status WHERE report_id IN (` + utils.GetOrmReplaceHolder(len(reportId)) + `) AND state=?`
	_, err = o.Raw(sql, reportId, state).QueryRows(&items)
	return
}

func GetReportPushStatusByReportIds(reportType int, reportId []int) (items []*ReportPushStatus, err error) {
	if len(reportId) == 0 {
		return
	}
	o := orm.NewOrm()
	sql := `SELECT * FROM report_push_status WHERE report_type=? AND report_id IN (` + utils.GetOrmReplaceHolder(len(reportId)) + `) `
	_, err = o.Raw(sql, reportType, reportId).QueryRows(&items)
	return
}

func GetReportPushStatusListByCondition(condition, sortCondition string, pars []interface{}, startSize, pageSize int) (items []*ReportPushView, err error) {
	o := orm.NewOrm()
	sql := `SELECT * FROM report_push_status WHERE 1=1  `
	if condition != "" {
		sql += condition
	}
	if sortCondition != "" {
		sql += sortCondition
	}
	sql += ` LIMIT ?,? `
	_, err = o.Raw(sql, pars, startSize, pageSize).QueryRows(&items)
	return
}

func GetReportByCondition(condition string, pars []interface{}) (items []*Report, err error) {
	o := orm.NewOrmUsingDB("rddp")
	sql := `SELECT * FROM report WHERE 1=1 AND (state=2 OR state=6) `
	if condition != "" {
		sql += condition
	}
	_, err = o.Raw(sql, pars...).QueryRows(&items)
	return
}

func GetReportIdListByCondition(condition string, pars []interface{}) (reportId []int, err error) {
	o := orm.NewOrm()
	sql := `SELECT report_id FROM report_push_status WHERE 1=1 `
	if condition != "" {
		sql += condition
	}
	_, err = o.Raw(sql, pars).QueryRows(&reportId)
	return
}

func GetReportCountById(id int) (count int, err error) {
	o := orm.NewOrmUsingDB("rddp")
	sql := `SELECT COUNT(*) AS count FROM report WHERE (state=2 OR state=6) AND id=?`
	err = o.Raw(sql, id).QueryRow(&count)
	return
}

func GetReportCountByCondition(condition string, pars []interface{}) (count int, err error) {
	o := orm.NewOrm()
	sql := `SELECT COUNT(*) AS count FROM report_push_status WHERE 1=1 `
	if condition != "" {
		sql += condition
	}
	err = o.Raw(sql, pars).QueryRow(&count)
	return
}

// BatchPushReport 批量推送报告
func BatchPushReport(reportId []int) (err error) {
	if len(reportId) == 0 {
		return
	}
	o := orm.NewOrm()
	sql := `UPDATE report_push_status SET state=1, push_time=NOW() WHERE report_id IN (` + utils.GetOrmReplaceHolder(len(reportId)) + `)`
	_, err = o.Raw(sql, reportId).Exec()
	return
}

// BatchPushCancelReport 批量撤销推送报告
func BatchPushCancelReport(reportId []int) (err error) {
	if len(reportId) == 0 {
		return
	}
	o := orm.NewOrm()
	sql := `UPDATE report_push_status SET state=0, modify_time=NOW() WHERE report_id IN (` + utils.GetOrmReplaceHolder(len(reportId)) + `)`
	_, err = o.Raw(sql, reportId).Exec()
	return
}

// BatchAddReportPushStatus 批量添加报告
func BatchAddReportPushStatus(items []*ReportPushStatus) (err error) {
	if len(items) == 0 {
		return
	}
	o := orm.NewOrm()
	_, err = o.InsertMulti(100, items)
	return
}

func GetMaxSyncIdReportPush(report_type int) (count int, err error) {
	o := orm.NewOrm()
	sql := `SELECT MAX(report_id) AS max_id FROM report_push_status WHERE report_type=?`
	err = o.Raw(sql, report_type).QueryRow(&count)
	return
}

func GetBatchReport(maxId, batchSize int) (items []*Report, err error) {
	o := orm.NewOrmUsingDB("rddp")
	sql := `SELECT * FROM report WHERE id>? AND (state=2 OR state=6) LIMIT ?`
	o.Raw(sql, maxId, batchSize).QueryRows(&items)
	return
}