forked from TrueCloudLab/frostfs-sdk-go
107 lines
2.7 KiB
Go
107 lines
2.7 KiB
Go
package rpc
|
|
|
|
import (
|
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/container"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/rpc/client"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/rpc/common"
|
|
)
|
|
|
|
const serviceContainer = serviceNamePrefix + "container.ContainerService"
|
|
|
|
const (
|
|
rpcContainerPut = "Put"
|
|
rpcContainerGet = "Get"
|
|
rpcContainerDel = "Delete"
|
|
rpcContainerList = "List"
|
|
rpcContainerStream = "ListStream"
|
|
rpcContainerGetEACL = "GetExtendedACL"
|
|
rpcContainerUsedSpace = "AnnounceUsedSpace"
|
|
)
|
|
|
|
// PutContainer executes ContainerService.Put RPC.
|
|
func PutContainer(
|
|
cli *client.Client,
|
|
req *container.PutRequest,
|
|
opts ...client.CallOption,
|
|
) (*container.PutResponse, error) {
|
|
resp := new(container.PutResponse)
|
|
|
|
err := client.SendUnary(cli, common.CallMethodInfoUnary(serviceContainer, rpcContainerPut), req, resp, opts...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return resp, nil
|
|
}
|
|
|
|
// GetContainer executes ContainerService.Get RPC.
|
|
func GetContainer(
|
|
cli *client.Client,
|
|
req *container.GetRequest,
|
|
opts ...client.CallOption,
|
|
) (*container.GetResponse, error) {
|
|
resp := new(container.GetResponse)
|
|
|
|
err := client.SendUnary(cli, common.CallMethodInfoUnary(serviceContainer, rpcContainerGet), req, resp, opts...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return resp, nil
|
|
}
|
|
|
|
// DeleteContainer executes ContainerService.Delete RPC.
|
|
func DeleteContainer(
|
|
cli *client.Client,
|
|
req *container.DeleteRequest,
|
|
opts ...client.CallOption,
|
|
) (*container.PutResponse, error) {
|
|
resp := new(container.PutResponse)
|
|
|
|
err := client.SendUnary(cli, common.CallMethodInfoUnary(serviceContainer, rpcContainerDel), req, resp, opts...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return resp, nil
|
|
}
|
|
|
|
// ListContainers executes ContainerService.List RPC.
|
|
func ListContainers(
|
|
cli *client.Client,
|
|
req *container.ListRequest,
|
|
opts ...client.CallOption,
|
|
) (*container.ListResponse, error) {
|
|
resp := new(container.ListResponse)
|
|
|
|
err := client.SendUnary(cli, common.CallMethodInfoUnary(serviceContainer, rpcContainerList), req, resp, opts...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return resp, nil
|
|
}
|
|
|
|
type ListStreamResponseReader struct {
|
|
r client.MessageReader
|
|
}
|
|
|
|
func (r *ListStreamResponseReader) Read(resp *container.ListStreamResponse) error {
|
|
return r.r.ReadMessage(resp)
|
|
}
|
|
|
|
// ListContainersStream executes ContainerService.ListStream RPC.
|
|
func ListContainersStream(
|
|
cli *client.Client,
|
|
req *container.ListStreamRequest,
|
|
opts ...client.CallOption,
|
|
) (*ListStreamResponseReader, error) {
|
|
wc, err := client.OpenServerStream(cli, common.CallMethodInfoServerStream(serviceContainer, rpcContainerStream), req, opts...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &ListStreamResponseReader{
|
|
r: wc,
|
|
}, nil
|
|
}
|