JCS-pub/common/pkgs/rpc/coordinator/hub.go

package corrpc

import (
	context "context"

	"gitlink.org.cn/cloudream/jcs-pub/common/pkgs/rpc"
	jcstypes "gitlink.org.cn/cloudream/jcs-pub/common/types"
)

type HubService interface {
	GetHubConfig(ctx context.Context, msg *GetHubConfig) (*GetHubConfigResp, *rpc.CodeError)

	GetHubs(ctx context.Context, msg *GetHubs) (*GetHubsResp, *rpc.CodeError)

	GetHubConnectivities(ctx context.Context, msg *GetHubConnectivities) (*GetHubConnectivitiesResp, *rpc.CodeError)

	ReportHubConnectivity(ctx context.Context, msg *ReportHubConnectivity) (*ReportHubConnectivityResp, *rpc.CodeError)
}

type GetHubConfig struct {
	HubID jcstypes.HubID `json:"hubID"`
}
type GetHubConfigResp struct {
	Hub jcstypes.Hub `json:"hub"`
}

func ReqGetHubConfig(hubID jcstypes.HubID) *GetHubConfig {
	return &GetHubConfig{
		HubID: hubID,
	}
}
func RespGetHubConfig(hub jcstypes.Hub) *GetHubConfigResp {
	return &GetHubConfigResp{
		Hub: hub,
	}
}
func (c *Client) GetHubConfig(ctx context.Context, msg *GetHubConfig) (*GetHubConfigResp, *rpc.CodeError) {
	if c.fusedErr != nil {
		return nil, c.fusedErr
	}
	return rpc.UnaryClient[*GetHubConfigResp](c.cli.GetHubConfig, ctx, msg)
}
func (s *Server) GetHubConfig(ctx context.Context, req *rpc.Request) (*rpc.Response, error) {
	return rpc.UnaryServer(s.svrImpl.GetHubConfig, ctx, req)
}

// 获取指定节点的信息。如果HubIDs为nil，则返回所有Hub
type GetHubs struct {
	HubIDs []jcstypes.HubID `json:"hubIDs"`
}
type GetHubsResp struct {
	Hubs []*jcstypes.Hub `json:"hubs"`
}

var _ = TokenAuth(Coordinator_GetHubs_FullMethodName)

func NewGetHubs(hubIDs []jcstypes.HubID) *GetHubs {
	return &GetHubs{
		HubIDs: hubIDs,
	}
}
func NewGetHubsResp(hubs []*jcstypes.Hub) *GetHubsResp {
	return &GetHubsResp{
		Hubs: hubs,
	}
}
func (r *GetHubsResp) GetHub(id jcstypes.HubID) *jcstypes.Hub {
	for _, n := range r.Hubs {
		if n.HubID == id {
			return n
		}
	}

	return nil
}
func (c *Client) GetHubs(ctx context.Context, msg *GetHubs) (*GetHubsResp, *rpc.CodeError) {
	if c.fusedErr != nil {
		return nil, c.fusedErr
	}
	return rpc.UnaryClient[*GetHubsResp](c.cli.GetHubs, ctx, msg)
}
func (s *Server) GetHubs(ctx context.Context, req *rpc.Request) (*rpc.Response, error) {
	return rpc.UnaryServer(s.svrImpl.GetHubs, ctx, req)
}

// 获取节点连通性信息

type GetHubConnectivities struct {
	HubIDs []jcstypes.HubID `json:"hubIDs"`
}
type GetHubConnectivitiesResp struct {
	Connectivities []jcstypes.HubConnectivity `json:"hubs"`
}

func ReqGetHubConnectivities(hubIDs []jcstypes.HubID) *GetHubConnectivities {
	return &GetHubConnectivities{
		HubIDs: hubIDs,
	}
}
func RespGetHubConnectivities(cons []jcstypes.HubConnectivity) *GetHubConnectivitiesResp {
	return &GetHubConnectivitiesResp{
		Connectivities: cons,
	}
}
func (c *Client) GetHubConnectivities(ctx context.Context, msg *GetHubConnectivities) (*GetHubConnectivitiesResp, *rpc.CodeError) {
	if c.fusedErr != nil {
		return nil, c.fusedErr
	}
	return rpc.UnaryClient[*GetHubConnectivitiesResp](c.cli.GetHubConnectivities, ctx, msg)
}
func (s *Server) GetHubConnectivities(ctx context.Context, req *rpc.Request) (*rpc.Response, error) {
	return rpc.UnaryServer(s.svrImpl.GetHubConnectivities, ctx, req)
}

// 上报节点连通性信息
type ReportHubConnectivity struct {
	Connecttivities []jcstypes.HubConnectivity
}
type ReportHubConnectivityResp struct {
}

func (c *Client) ReportHubConnectivity(ctx context.Context, msg *ReportHubConnectivity) (*ReportHubConnectivityResp, *rpc.CodeError) {
	if c.fusedErr != nil {
		return nil, c.fusedErr
	}
	return rpc.UnaryClient[*ReportHubConnectivityResp](c.cli.ReportHubConnectivity, ctx, msg)
}
func (s *Server) ReportHubConnectivity(ctx context.Context, req *rpc.Request) (*rpc.Response, error) {
	return rpc.UnaryServer(s.svrImpl.ReportHubConnectivity, ctx, req)
}