visit.go 817 Bytes
Newer Older
hujiebin's avatar
hujiebin committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37
package visit_m

import (
	"git.hilo.cn/hilo-common/domain"
	"git.hilo.cn/hilo-common/resource/mysql"
	"hilo-group/myerr"
)

type UserVisit struct {
	mysql.Entity
	*domain.Model `gorm:"-"`
	UserId        mysql.ID
	VisitUserId   mysql.ID
	N             mysql.Num
}

func GetVisitInstanceOrInit(model *domain.Model, userId mysql.ID, visitUserId mysql.ID) (*UserVisit, error) {
	var userVisit UserVisit
	if err := model.Db.Where(&UserVisit{
		UserId:      userId,
		VisitUserId: visitUserId,
	}).FirstOrInit(&userVisit, UserVisit{
		UserId:      userId,
		VisitUserId: visitUserId,
		N:           0,
	}).Error; err != nil {
		return nil, myerr.WrapErr(err)
	}
	userVisit.Model = model
	return &userVisit, nil
}

func (userVisit *UserVisit) UserVisitAdd() *UserVisit {
	userVisit.N = userVisit.N + 1
	return userVisit
}