rank_item.go 2.4 KB
package models

import (
	"oppmg/common/log"
	"time"

	"github.com/astaxie/beego/orm"
)

type RankItem struct {
	Id         int64     `orm:"column(id);auto"`
	CompanyId  int64     `orm:"column(company_id);null" description:"公司编号 company.id"`
	RankTypeId int64     `orm:"column(rank_type_id)" description:"表rank_type.id 榜单类型编号"`
	CreateAt   time.Time `orm:"column(create_at);type(timestamp);null" description:"创建时间"`
	UpdateAt   time.Time `orm:"column(update_at);type(timestamp);null" description:"更新时间"`
	SortNum    int       `orm:"column(sort_num);null" description:"序号"`
	ItemName   string    `orm:"column(item_name);size(50);null" description:"评比项名称"`
	ItemKey    string    `orm:"column(item_key);size(50);null" description:"评比项键值(排行榜排序使用)"`
}

func (t *RankItem) TableName() string {
	return "rank_item"
}

func init() {
	orm.RegisterModel(new(RankItem))
}

// AddRankItem insert a new NewRankItem into database and returns
// last inserted Id on success.
func AddRankItem(m []RankItem, om orm.Ormer) (successSum int64, err error) {
	nowTime := time.Now()
	for i := range m {
		m[i].CreateAt = nowTime
		m[i].UpdateAt = nowTime
	}
	successSum, err = om.InsertMulti(10, &m)
	return
}

// GetRankItemById retrieves NewRankItem by Id. Returns error if
// Id doesn't exist
func GetRankItemById(id int64) (v *RankItem, err error) {
	o := orm.NewOrm()
	v = &RankItem{Id: id}
	if err = o.Read(v); err == nil {
		return v, nil
	}
	return nil, err
}

func GetRankItemByCompanyid(companyid int64, rankTypeId int64) ([]RankItem, error) {
	var (
		data []RankItem
		err  error
	)
	o := orm.NewOrm()
	_, err = o.QueryTable(&RankItem{}).
		Filter("company_id", companyid).
		Filter("rank_type_id", rankTypeId).
		All(&data)
	if err == orm.ErrNoRows {
		return data, nil
	}
	return data, err
}

func DeleteRanKItemByIds(ids []int64, om orm.Ormer) error {
	_, err := om.QueryTable(&RankItem{}).Filter("id__in", ids).Delete()
	return err
}

// UpdateRankItem updates RankItem by Id and returns error if
// the record to be updated doesn't exist
func UpdateRankItemById(m *RankItem, cols []string, om orm.Ormer) (err error) {
	if len(cols) > 0 {
		cols = append(cols, "UpdateAt")
	}
	m.UpdateAt = time.Now()
	var num int64
	if num, err = om.Update(m, cols...); err == nil {
		log.Info("Number of records updated in database:%d", num)
	}
	return
}