aiforge/models/user_invitation.go

package models

import (
	"fmt"

	"code.gitea.io/gitea/modules/log"
	"code.gitea.io/gitea/modules/timeutil"
)

// Follow represents relations of user and his/her followers.
type Invitation struct {
	ID                int64              `xorm:"pk autoincr"`
	SrcUserID         int64              `xorm:"NOT NULL DEFAULT 0"`
	UserID            int64              `xorm:"NOT NULL DEFAULT 0"`
	Phone             string             `xorm:"INDEX"`
	Avatar            string             `xorm:"-"`
	Name              string             `xorm:"-"`
	InvitationUserNum int                `xorm:"-"`
	IsActive          bool               `xorm:"-"`
	CreatedUnix       timeutil.TimeStamp `xorm:"created"`
}

func QueryInvitaionByPhone(phone string) []*Invitation {
	statictisSess := xStatistic.NewSession()
	defer statictisSess.Close()
	cond := "phone ='" + phone + "'"
	invitationList := make([]*Invitation, 0)
	if err := statictisSess.Table(new(Invitation)).Where(cond).
		Find(&invitationList); err != nil {
		return nil
	} else {
		return invitationList
	}
}

func GetAllUserName() map[int64]string {
	sess := x.NewSession()
	defer sess.Close()
	sess.Select("id,name").Table("user")
	userList := make([]*User, 0)
	reMap := make(map[int64]string)
	sess.Find(&userList)
	for _, user := range userList {
		reMap[user.ID] = user.Name
	}
	return reMap
}

func QueryInvitaionPage(start int, pageSize int) ([]*Invitation, int64) {
	statictisSess := xStatistic.NewSession()
	defer statictisSess.Close()
	//cond := "created_unix >=" + fmt.Sprint(startTime) + " and created_unix <=" + fmt.Sprint(endTime)

	allCount, err := statictisSess.Count(new(Invitation))
	if err != nil {
		log.Info("query error." + err.Error())
		return nil, 0
	}
	invitationList := make([]*Invitation, 0)
	if err := statictisSess.Table(new(Invitation)).OrderBy("created_unix desc").Limit(pageSize, start).
		Find(&invitationList); err != nil {
		return nil, 0
	}
	return invitationList, allCount
}

func QueryInvitaionByTime(startTime int64, endTime int64) []*Invitation {
	statictisSess := xStatistic.NewSession()
	defer statictisSess.Close()
	cond := "created_unix >=" + fmt.Sprint(startTime) + " and created_unix <=" + fmt.Sprint(endTime)
	invitationList := make([]*Invitation, 0)
	if err := statictisSess.Table(new(Invitation)).Where(cond).OrderBy("created_unix desc").
		Find(&invitationList); err != nil {
		return nil
	}
	return invitationList
}

func InsertInvitaion(invitationUser *Invitation) error {
	statictisSess := xStatistic.NewSession()
	defer statictisSess.Close()
	_, err := statictisSess.Insert(invitationUser)
	return err
}

func QueryInvitaionBySrcUserId(srcUserId int64, start int, pageSize int) ([]*Invitation, int64) {
	statictisSess := xStatistic.NewSession()
	defer statictisSess.Close()
	cond := "src_user_id =" + fmt.Sprint(srcUserId)
	allCount, err := statictisSess.Where(cond).Count(new(Invitation))
	if err != nil {
		log.Info("query error." + err.Error())
		return nil, 0
	}
	invitationList := make([]*Invitation, 0)

	if err := statictisSess.Table(new(Invitation)).Where(cond).OrderBy("created_unix desc").Limit(pageSize, start).
		Find(&invitationList); err != nil {
		return nil, 0
	}
	return invitationList, allCount
}