openbilibili/app/admin/main/videoup-task/service/consumer.go

package service

import (
	"context"
	"fmt"
	"go-common/library/sync/errgroup"
	"strings"
	"time"

	"go-common/app/admin/main/videoup-task/model"
	"go-common/library/log"
)

// HandsUp 签入
func (s *Service) HandsUp(c context.Context, uid int64, uname string) (err error) {
	if s.CheckOnline(c, uid) {
		log.Info("已经登入(%d)", uid)
		return
	}

	_, err = s.dao.TaskUserCheckIn(c, uid)
	if err != nil {
		log.Error("s.dao.TaskUserCheckIn(%d) error(%v)", uid, err)
		return
	}
	s.sendConsumerLog(c, &model.ConsumerLog{
		UID:    uid,
		Uname:  uname,
		Action: model.ActionHandsUP,
		Ctime:  time.Now().Format(model.TimeFormatSec),
		Desc:   "checkin",
	})

	mapParas := map[string]interface{}{
		"action": model.ActionHandsUP,
		"uid":    uid,
	}

	if _, err = s.dao.AddTaskHis(c, 0, model.ActionHandsUP, 0, 0, uid, 0, 0, "checkin"); err != nil {
		log.Error("s.dao.AddTaskLog(%v) error(%v)", mapParas, uid)
		return
	}
	log.Info("用户签入(%d)", uid)
	return
}

// HandsOff 签出
func (s *Service) HandsOff(c context.Context, uid int64, fuid int64) (err error) {
	if fuid != 0 { //管理员强制踢出组员
		if !s.isLeader(c, uid) {
			return fmt.Errorf("只有组长能强制踢出")
		}
		log.Info("管理员%d踢出组员%d", uid, fuid)
		uid = fuid
	}

	err = s.checkOut(c, uid)
	if err != nil {
		log.Error("s.checkOut(%d) error(%v)", uid, err)
		return
	}
	s.Free(c, uid)
	return
}

// Online 用户列表
func (s *Service) Online(c context.Context) (cms []*model.Consumers, err error) {
	cms, err = s.dao.Consumers(c)
	if err != nil {
		log.Error("s.dao.Consumers error(%v)", err)
		return
	}

	if len(cms) > 0 {
		var wg errgroup.Group
		wg.Go(func() error {
			if err := s.mulIDtoName(c, cms, s.dao.GetNameByUID, "UID", "UserName"); err != nil {
				log.Error("mulIDtoName s.dao.GetNameByUID error(%v)", err)
			}
			return nil
		})
		wg.Go(func() error {
			if err := s.mulIDtoName(c, cms, s.dao.OutTime, "UID", "LastOut"); err != nil {
				log.Error("mulIDtoName s.dao.OutTime error(%v)", err)
			}
			return nil
		})
		wg.Wait()
	}

	return
}

// InOutList 用户登入登出历史
func (s *Service) InOutList(c context.Context, unames string, bt, et string) (l []*model.InQuit, err error) {
	uids := []int64{}
	if len(unames) > 0 {
		if res, err := s.dao.Uids(c, strings.Split(unames, ",")); err == nil {
			for _, uid := range res {
				uids = append(uids, uid)
			}
		}
	}
	// 前端参数是日期，搜索参数必须到秒
	if len(bt) > 0 && len(et) > 0 {
		bt = bt + " 00:00:00"
		et = et + " 23:59:59"
	}

	return s.dao.InQuitList(c, uids, bt, et)
}

// CheckOnline 检查在线状态
func (s *Service) CheckOnline(c context.Context, uid int64) (on bool) {
	if s.dao.IsConsumerOn(c, uid) == 1 {
		on = true
	}
	return
}

// CheckGroup 检查用户组权限
func (s *Service) CheckGroup(c context.Context, uid int64) (role int8, err error) {
	role, err = s.dao.GetUserRole(c, uid)
	if err != nil || role == 0 {
		log.Error("非法用户(%d) error(%v)", uid, err)
		return
	}
	return
}

func (s *Service) checkOut(c context.Context, uid int64) (err error) {
	if s.dao.IsConsumerOn(c, uid) == 0 {
		log.Info("已经签出(%d)", uid)
		return
	}

	_, err = s.dao.TaskUserCheckOff(c, uid)
	if err != nil {
		log.Error("s.dao.TaskUserCheckOff(%d) error(%v)", uid, err)
		return
	}

	s.sendConsumerLog(c, &model.ConsumerLog{
		UID:    uid,
		Uname:  "",
		Action: model.ActionHandsOFF,
		Ctime:  time.Now().Format(model.TimeFormatSec),
		Desc:   "checkout",
	})

	mapParas := map[string]interface{}{
		"action": model.ActionHandsOFF,
		"uid":    uid,
	}

	if _, err = s.dao.AddTaskHis(c, 0, model.ActionHandsOFF, 0, 0, uid, 0, 0, "checkOut"); err != nil {
		log.Error("s.dao.AddTaskLog(%v) error(%v)", mapParas, uid)
	}
	return
}

func (s *Service) isLeader(c context.Context, uid int64) bool {
	role, e := s.dao.GetUserRole(c, uid)
	if e != nil {
		log.Error("s.dao.GetUserRole(%d) error(%v)", uid, e)
		return false
	}
	if role == model.TaskLeader {
		return true
	}
	return false
}