hd/models/live_viewer.go

package models

import (
	"time"

	"git.ouxuan.net/tommy/osmanthuswine/src/core"
)

const LiveViewerTN = TableNamePrefix + "live_viewer"

type LiveViewer struct {
	Id           int       `json:"id" xorm:"not null pk autoincr INT(11)"`
	IsDelete     bool      `json:"-" xorm:"not null default 0 comment('是否删除') TINYINT(1)"`
	CreatedAt    time.Time `json:"created_at" xorm:"not null created comment('创建时间') DATETIME"`
	UpdatedAt    time.Time `json:"updated_at" xorm:"not null updated default CURRENT_TIMESTAMP  comment('更新时间') TIMESTAMP"`
	UserId       int       `json:"user_id" xorm:"not null default 0 comment('用户id') INT(11)"`
	ArchId       int       `json:"arch_id" xorm:"not null default 0 comment('归档id') INT(11)"`
	ActivityId   int       `json:"activity_id" xorm:"not null default 0 comment('活动id') INT(11)"`
	LiveConfigId int       `json:"live_config_id" xorm:"not null default 0 comment('直播id') INT(11)"`
}

func (t *LiveViewer) TableName() string {
	return LiveViewerTN
}

func (t *LiveViewer) Record(uid, aid, lid int) error {
	session := core.GetXormAuto().NewSession()
	defer session.Close()
	session.Begin()
	exist, err := session.Where("is_delete=0 and user_id=? and activity_id=?", uid, aid).Exist(t)
	if err != nil {
		session.Rollback()
		return err
	}
	if exist {
		session.Commit()
		return nil
	}
	t.UserId = uid
	t.ActivityId = aid
	t.LiveConfigId = lid
	_, err = session.InsertOne(t)
	if err != nil {
		session.Rollback()
		return err
	}
	_, err = session.Where("is_delete=0 and activity_id=?", aid).Incr("watch_num").Update(&LiveConfig{})
	if err != nil {
		session.Rollback()
		return err
	}
	session.Commit()
	return nil
}

func GetLiveViewerByLiveConfigIds(ids interface{}) ([]*LiveViewer, error) {
	viewers := make([]*LiveViewer, 0)
	err := core.GetXormAuto().Table(&LiveViewer{}).Alias("l").
		Join("left", Alias(&Activity{}, "a"),
			"a.id=l.activity_id and l.arch_id=a.arch_id").
		Where("l.is_delete=0").In("l.live_config_id", ids).
		Find(&viewers)
	return viewers, err
}

func CountLiveViewerByLiveConfigId(lid interface{}) (int64, error) {
	return core.GetXormAuto().Where("is_delete=0 and live_config_id=?", lid).Count(&LiveViewer{})
}

func (t *LiveViewer) ExistByUserId(userId, liveId interface{}) (bool, error) {
	return core.GetXormAuto().Where("is_delete=0 and user_id=? and live_config_id=?", userId, liveId).Exist(t)
}