openbilibili/app/job/main/block/service/limit.go

package service

import (
	"context"
	"time"

	"go-common/app/job/main/block/conf"
	"go-common/app/job/main/block/model"
	xsql "go-common/library/database/sql"
	"go-common/library/log"

	"github.com/pkg/errors"
)

func (s *Service) limitExpireHandler(c context.Context) {
	if conf.Conf.Property.LimitExpireCheckLimit <= 0 {
		log.Error("conf.Conf.Property.LimitExpireCheckLimit [%d] <= 0", conf.Conf.Property.LimitExpireCheckLimit)
		return
	}
	var (
		mids    = make([]int64, conf.Conf.Property.LimitExpireCheckLimit)
		startID int64
		err     error
	)
	for len(mids) >= conf.Conf.Property.LimitExpireCheckLimit {
		log.Info("limit expire handle startID (%d)", startID)
		if startID, mids, err = s.dao.UserStatusList(c, model.BlockStatusLimit, startID, conf.Conf.Property.LimitExpireCheckLimit); err != nil {
			log.Error("%+v", err)
			return
		}
		for _, mid := range mids {
			var ok bool
			if ok, err = s.limitExpireCheck(c, mid); err != nil {
				log.Error("%+v", err)
				continue
			}
			if ok {
				s.limitExpireRemove(c, mid)
			}
			time.Sleep(time.Millisecond * 100)
		}
	}
}

func (s *Service) limitExpireCheck(c context.Context, mid int64) (ok bool, err error) {
	var (
		his *model.DBHistory
	)
	if his, err = s.dao.UserLastHistory(c, mid); err != nil {
		return
	}
	if his == nil {
		return
	}
	if his.Action != model.BlockActionLimit {
		return
	}
	if his.StartTime.Add(time.Duration(his.Duration) * time.Second).After(time.Now()) {
		return
	}
	ok = true
	return
}

func (s *Service) limitExpireRemove(c context.Context, mid int64) (err error) {
	var (
		_reason, _comment = "系统自动解封", "系统自动解封"
		stime             = time.Now()
		db                = &model.DBHistory{
			MID:       mid,
			AdminID:   model.BlockJOBManagerID,
			AdminName: model.BlockJOBManagerName,
			Source:    model.BlockSourceRemove,
			Area:      model.BlockAreaNone,
			Reason:    _reason,
			Comment:   _comment,
			Action:    model.BlockActionSelfRemove,
			StartTime: stime,
			Duration:  0,
			Notify:    false,
		}
		tx *xsql.Tx
	)
	if tx, err = s.dao.BeginTX(c); err != nil {
		return
	}
	if err = s.dao.TxInsertHistory(c, tx, db); err != nil {
		tx.Rollback()
		return
	}
	if err = s.dao.TxUpsertUser(c, tx, mid, model.BlockStatusFalse); err != nil {
		tx.Rollback()
		return
	}
	if err = tx.Commit(); err != nil {
		err = errors.WithStack(err)
	}
	s.mission(func() {
		if err := s.notifyRemoveMSG(context.TODO(), []int64{mid}); err != nil {
			log.Error("%+v", err)
		}
	})
	s.mission(func() {
		s.AddAuditLog(context.TODO(), model.BlockActionSelfRemove, model.BlockJOBManagerID, model.BlockJOBManagerName, []int64{mid}, 0, model.BlockSourceRemove, model.BlockAreaNone, _reason, _comment, false, stime)
	})
	s.cache.Save(func() {
		if err := s.dao.DeleteUserCache(context.TODO(), mid); err != nil {
			log.Error("%+v", err)
		}
		if databusErr := s.accountNotify(context.TODO(), mid); databusErr != nil {
			log.Error("%+v", databusErr)
		}
	})
	return
}