JCS-pub/client/internal/metacache/hubmeta.go

package metacache

import (
	"context"
	"time"

	"gitlink.org.cn/cloudream/common/pkgs/logger"
	stgglb "gitlink.org.cn/cloudream/jcs-pub/common/globals"
	corrpc "gitlink.org.cn/cloudream/jcs-pub/common/pkgs/rpc/coordinator"
	cortypes "gitlink.org.cn/cloudream/jcs-pub/coordinator/types"
)

func (m *MetaCacheHost) AddHubMeta() *HubMeta {
	meta := &HubMeta{}
	meta.cache = NewSimpleMetaCache(SimpleMetaCacheConfig[cortypes.HubID, cortypes.Hub]{
		Getter: meta.load,
		Expire: time.Minute * 5,
	})

	m.caches = append(m.caches, meta)
	return meta
}

type HubMeta struct {
	cache *SimpleMetaCache[cortypes.HubID, cortypes.Hub]
}

func (h *HubMeta) Get(hubID cortypes.HubID) *cortypes.Hub {
	v, ok := h.cache.Get(hubID)
	if ok {
		return &v
	}
	return nil
}

func (h *HubMeta) GetMany(hubIDs []cortypes.HubID) []*cortypes.Hub {
	vs, oks := h.cache.GetMany(hubIDs)
	ret := make([]*cortypes.Hub, len(vs))
	for i := range vs {
		if oks[i] {
			ret[i] = &vs[i]
		}
	}
	return ret
}

func (h *HubMeta) ClearOutdated() {
	h.cache.ClearOutdated()
}

func (h *HubMeta) load(keys []cortypes.HubID) ([]cortypes.Hub, []bool) {
	vs := make([]cortypes.Hub, len(keys))
	oks := make([]bool, len(keys))

	coorCli := stgglb.CoordinatorRPCPool.Get()
	defer coorCli.Release()

	get, cerr := coorCli.GetHubs(context.Background(), corrpc.NewGetHubs(keys))
	if cerr != nil {
		logger.Warnf("get hubs: %v", cerr)
		return vs, oks
	}

	for i := range keys {
		if get.Hubs[i] != nil {
			vs[i] = *get.Hubs[i]
			oks[i] = true
		}
	}

	return vs, oks
}