bilibili-go-common/app/admin/main/aegis/service/service.go

package service

import (
	"context"
	"fmt"
	"strings"
	"time"

	"go-common/app/admin/main/aegis/conf"
	daoOrm "go-common/app/admin/main/aegis/dao/gorm"
	daoHttp "go-common/app/admin/main/aegis/dao/http"
	daoMc "go-common/app/admin/main/aegis/dao/mc"
	daoMysql "go-common/app/admin/main/aegis/dao/mysql"
	daoRedis "go-common/app/admin/main/aegis/dao/redis"
	daoRpc "go-common/app/admin/main/aegis/dao/rpc"
	"go-common/app/admin/main/aegis/model/common"
	"go-common/app/admin/main/aegis/model/middleware"
	"go-common/app/admin/main/aegis/model/net"
	"go-common/library/queue/databus"
	"go-common/library/sync/pipeline/fanout"
)

// Service struct
type Service struct {
	c *conf.Config

	gorm     *daoOrm.Dao
	http     *daoHttp.Dao
	mc       *daoMc.Dao
	mysql    *daoMysql.Dao
	redis    *daoRedis.Dao
	rpc      *daoRpc.Dao
	aegisPub *databus.Databus

	// cache
	groupCache         map[int64]*common.Group
	bizCfgCache        map[string]string
	bizRoleCache       map[int64]map[string]int64  //business:cfg_name:role_id/manager_id映射
	taskRoleCache      map[int64]map[int64][]int64 //business:manager_id:flow_id映射
	netCacheCh         chan map[string]string
	bizMiddlewareCache map[int64][]*middleware.Aggregate

	netCache        map[int64]*net.Net
	tokenCache      map[int64]*net.Token
	flowCache       map[int64]*net.Flow
	transitionCache map[int64]*net.Transition
	bindCache       map[int64]*net.TokenBind

	//async
	async *fanout.Fanout

	//gray
	gray map[int64][][]common.GrayField
}

// Cache .
func (s *Service) Cache() map[string]interface{} {
	return map[string]interface{}{
		"groupCache":         s.groupCache,
		"bizCfgCache":        s.bizCfgCache,
		"bizRoleCache":       s.bizRoleCache,
		"taskRoleCache":      s.taskRoleCache,
		"bizMiddlewareCache": s.bizMiddlewareCache,
	}
}

// New init
func New(c *conf.Config) (s *Service) {
	s = &Service{
		c: c,
		// dao
		gorm:     daoOrm.New(c),
		http:     daoHttp.New(c),
		mc:       daoMc.New(c),
		mysql:    daoMysql.New(c),
		redis:    daoRedis.New(c),
		rpc:      daoRpc.New(c),
		aegisPub: databus.New(c.AegisPub),

		// cache
		netCache:        map[int64]*net.Net{},
		tokenCache:      map[int64]*net.Token{},
		flowCache:       map[int64]*net.Flow{},
		transitionCache: map[int64]*net.Transition{},
		bindCache:       map[int64]*net.TokenBind{},

		//gray
		gray:       loadGray(c),
		netCacheCh: make(chan map[string]string, 10240),

		//async
		async: fanout.New("async", fanout.Worker(10), fanout.Buffer(10240)),
	}

	go s.cacheProc()
	go s.setNetCache()
	return s
}

// Debug .
func (s *Service) Debug() string {
	return s.c.Debug
}

// Ping Service
func (s *Service) Ping(c context.Context) (err error) {
	return s.mysql.Ping(c)
}

// Close Service
func (s *Service) Close() {
	s.async.Close()
	s.gorm.Close()
	s.mysql.Close()
	s.mc.Close()
	s.redis.Close()
	s.aegisPub.Close()
}

func (s *Service) cacheProc() {
	for {
		s.syncUpCache(context.Background())
		s.syncBizCache(context.Background())

		time.Sleep(10 * time.Minute)
	}
}

//IsAdmin uid是否为管理员
func (s *Service) IsAdmin(uid int64) bool {
	return len(s.c.Admin) > 0 && strings.Contains(","+s.c.Admin+",", fmt.Sprintf(",%d,", uid))
}

func loadGray(c *conf.Config) (gray map[int64][][]common.GrayField) {
	if c.Gray == nil {
		gray = make(map[int64][][]common.GrayField)
		return
	}

	gray = make(map[int64][][]common.GrayField)
	for _, biz := range c.Gray.Biz {
		ones := [][]common.GrayField{}
		for _, opts := range biz.Options {
			options := []common.GrayField{}
			for _, field := range opts.Fields {
				field.Name = strings.TrimSpace(field.Name)
				field.Value = strings.TrimSpace(field.Value)
				if field.Value == "" || field.Name == "" {
					continue
				}
				options = append(options, common.GrayField{
					Name:  field.Name,
					Value: fmt.Sprintf(",%s,", field.Value),
				})
			}
			ones = append(ones, options)
		}
		gray[biz.BusinessID] = ones
	}
	return
}

//GetMiddlewareCache get cache by bizid
func (s *Service) GetMiddlewareCache(bizid int64) []*middleware.Aggregate {
	return s.bizMiddlewareCache[bizid]
}