frostfs-api-go/rpc/netmap.go

package rpc

import (
	"git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/netmap"
	"git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/rpc/client"
	"git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/rpc/common"
)

const serviceNetmap = serviceNamePrefix + "netmap.NetmapService"

const (
	rpcNetmapNodeInfo = "LocalNodeInfo"
	rpcNetmapNetInfo  = "NetworkInfo"
	rpcNetmapSnapshot = "NetmapSnapshot"
)

// LocalNodeInfo executes NetmapService.LocalNodeInfo RPC.
func LocalNodeInfo(
	cli *client.Client,
	req *netmap.LocalNodeInfoRequest,
	opts ...client.CallOption,
) (*netmap.LocalNodeInfoResponse, error) {
	resp := new(netmap.LocalNodeInfoResponse)

	err := client.SendUnary(cli, common.CallMethodInfoUnary(serviceNetmap, rpcNetmapNodeInfo), req, resp, opts...)
	if err != nil {
		return nil, err
	}

	return resp, nil
}

// NetworkInfo executes NetmapService.NetworkInfo RPC.
func NetworkInfo(
	cli *client.Client,
	req *netmap.NetworkInfoRequest,
	opts ...client.CallOption,
) (*netmap.NetworkInfoResponse, error) {
	resp := new(netmap.NetworkInfoResponse)
	err := client.SendUnary(cli, common.CallMethodInfoUnary(serviceNetmap, rpcNetmapNetInfo), req, resp, opts...)
	if err != nil {
		return nil, err
	}

	return resp, nil
}

// NetMapSnapshot executes NetmapService.NetmapSnapshot RPC.
func NetMapSnapshot(
	cli *client.Client,
	req *netmap.SnapshotRequest,
	opts ...client.CallOption,
) (*netmap.SnapshotResponse, error) {
	resp := new(netmap.SnapshotResponse)

	err := client.SendUnary(cli, common.CallMethodInfoUnary(serviceNetmap, rpcNetmapSnapshot), req, resp, opts...)
	if err != nil {
		return nil, err
	}

	return resp, nil
}