flytalk/models/visitors.go

package models

type Visitor struct {
	Model
	Name      string `gorm:"size:50 not null default:''" json:"name"`
	Avator    string `gorm:"size:255 not null default:''" json:"avator"`
	SourceIp  string `gorm:"size:32 not null default:''" json:"source_ip"`
	ToId      string `gorm:"size:50 not null default:''" json:"to_id"`
	VisitorId string `gorm:"size:100 not null default:'' uniqueIndex:uidx_visitor_id" json:"visitor_id"`
	Status    uint   `gorm:"size:4 not null default:0" json:"status"`
	Refer     string `gorm:"size:500 not null default:''" json:"refer"`
	City      string `gorm:"size:100 not null default:''" json:"city"`
	ClientIp  string `gorm:"size:32 not null default:''" json:"client_ip"`
}

func CreateVisitor(name string, avator string, sourceIp string, toId string, visitorId string, refer string, city string, clientIp string) {
	old := FindVisitorByVistorId(visitorId)
	if old.Name != "" {
		//更新状态上线
		UpdateVisitor(visitorId, 1, clientIp, sourceIp, refer)
		return
	}
	v := &Visitor{
		Name:      name,
		Avator:    avator,
		SourceIp:  sourceIp,
		ToId:      toId,
		VisitorId: visitorId,
		Status:    1,
		Refer:     refer,
		City:      city,
		ClientIp:  clientIp,
	}
	DB.Create(v)
}
func FindVisitorByVistorId(visitorId string) Visitor {
	var v Visitor
	DB.Where("visitor_id = ?", visitorId).First(&v)
	return v
}
func FindVisitors(page int, pagesize int) []Visitor {
	offset := (page - 1) * pagesize
	if offset < 0 {
		offset = 0
	}
	var visitors []Visitor
	DB.Offset(offset).Limit(pagesize).Order("status desc, updated_at desc").Find(&visitors)
	return visitors
}
func FindVisitorsByKefuId(page int, pagesize int, kefuId string) []Visitor {
	offset := (page - 1) * pagesize
	if offset < 0 {
		offset = 0
	}
	var visitors []Visitor
	DB.Where("to_id=?", kefuId).Offset(offset).Limit(pagesize).Order("status desc, updated_at desc").Find(&visitors)
	return visitors
}
func FindVisitorsOnline() []Visitor {
	var visitors []Visitor
	DB.Where("status = ?", 1).Find(&visitors)
	return visitors
}
func UpdateVisitorStatus(visitorId string, status uint) {
	visitor := Visitor{}
	DB.Model(&visitor).Where("visitor_id = ?", visitorId).Update("status", status)
}
func UpdateVisitor(visitorId string, status uint, clientIp string, sourceIp string, refer string) {
	visitor := &Visitor{
		Status:   status,
		ClientIp: clientIp,
		SourceIp: sourceIp,
		Refer:    refer,
	}
	DB.Model(visitor).Where("visitor_id = ?", visitorId).Updates(visitor)
}

//查询条数
func CountVisitors() int64 {
	var count int64
	DB.Model(&Visitor{}).Count(&count)
	return count
}

//查询条数
func CountVisitorsByKefuId(kefuId string) int64 {
	var count int64
	DB.Model(&Visitor{}).Where("to_id=?", kefuId).Count(&count)
	return count
}