eta_index_lib/models/data_stat/edb_source_stat.go

package data_stat

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

// EdbSourceStat 数据源统计表
type EdbSourceStat struct {
	Id               int       `orm:"column(id);pk"`
	SourceName       string    `description:"来源名称"`
	Source           int       `description:"来源id"`
	TerminalCode     string    `description:"终端编码，用于配置在机器上"`
	EdbNum           int       `description:"指标总数"`
	EdbNewNum        int       `description:"今日新增指标数"`
	EdbDelNum        int       `description:"今日删除指标数"`
	NeedRefreshNum   int       `description:"今日需刷新指标"`
	HasRefreshNum    int       `description:"今日发起刷新任务指标数"`
	UpdateSuccessNum int       `description:"今日已刷新成功指标数"`
	UpdateFailedNum  int       `description:"今日已刷新失败指标数"`
	CreateTime       time.Time `description:"创建时间"`
	ModifyTime       time.Time `description:"修改时间"`
}

func GetEdbSourceStatByCondition(condition string, pars []interface{}) (item []*EdbSourceStat, err error) {
	o := orm.NewOrm()
	sql := ` SELECT * FROM edb_source_stat WHERE 1=1 `
	if condition != "" {
		sql += condition
	}
	_, err = o.Raw(sql, pars).QueryRows(&item)
	return
}

// Add 新增
func (r *EdbSourceStat) Add(list []*EdbSourceStat) (err error) {
	o := orm.NewOrm()
	_, err = o.InsertMulti(len(list), list)
	return
}

// UpdateEdbSourceStatMulti 批量更新
func UpdateEdbSourceStatMulti(list []*EdbSourceStat) (err error) {
	o := orm.NewOrm()
	sql := `UPDATE edb_source_stat 
SET 
edb_num=?,                              
edb_new_num=?,                          
edb_del_num=?,                          
need_refresh_num=?,                      
has_refresh_num=?,                       
update_success_num=?,                   
update_failed_num=?,                                                         
modify_time=?                           
WHERE
	id = ?`
	p, err := o.Raw(sql).Prepare()
	if err != nil {
		return
	}
	defer func() {
		_ = p.Close() // 别忘记关闭 statement
	}()
	for _, v := range list {
		_, err = p.Exec(v.EdbNum, v.EdbNewNum, v.EdbDelNum, v.NeedRefreshNum, v.HasRefreshNum, v.UpdateFailedNum, v.ModifyTime, v.Id)
		if err != nil {
			return
		}
	}
	return
}