[#263] v2: Support new rpc library
Implement `message.Message` interface on all structures and use new methods for conversion instead of functions. make `Unmarshal` and JSON methods to use encoding functions from `message` library. Remove all per-service clients and implement `rpc` library of the functions which execute NeoFS API RPC through new RPC client. Remove no longer used gRPC per-service clients. Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
This commit is contained in:
parent
30c6ca0714
commit
1031f3122e
102 changed files with 7554 additions and 12298 deletions
190
v2/rpc/object.go
Normal file
190
v2/rpc/object.go
Normal file
|
@ -0,0 +1,190 @@
|
|||
package rpc
|
||||
|
||||
import (
|
||||
"github.com/nspcc-dev/neofs-api-go/rpc/client"
|
||||
"github.com/nspcc-dev/neofs-api-go/rpc/common"
|
||||
"github.com/nspcc-dev/neofs-api-go/rpc/message"
|
||||
"github.com/nspcc-dev/neofs-api-go/v2/object"
|
||||
)
|
||||
|
||||
const serviceObject = serviceNamePrefix + "object.ObjectService"
|
||||
|
||||
const (
|
||||
rpcObjectPut = "Put"
|
||||
rpcObjectGet = "Get"
|
||||
rpcObjectSearch = "Search"
|
||||
rpcObjectRange = "GetRange"
|
||||
rpcObjectHash = "GetRangeHash"
|
||||
rpcObjectHead = "Head"
|
||||
rpcObjectDelete = "Delete"
|
||||
)
|
||||
|
||||
// PutRequestWriter is an object.PutRequest
|
||||
// message streaming component.
|
||||
type PutRequestWriter struct {
|
||||
wc client.MessageWriterCloser
|
||||
|
||||
resp message.Message
|
||||
}
|
||||
|
||||
// Write writes req to the stream.
|
||||
func (w *PutRequestWriter) Write(req *object.PutRequest) error {
|
||||
return w.wc.WriteMessage(req)
|
||||
}
|
||||
|
||||
// Close closes the stream.
|
||||
func (w *PutRequestWriter) Close() error {
|
||||
return w.wc.Close()
|
||||
}
|
||||
|
||||
// PutObject executes ObjectService.Put RPC.
|
||||
func PutObject(
|
||||
cli *client.Client,
|
||||
resp *object.PutResponse,
|
||||
opts ...client.CallOption,
|
||||
) (*PutRequestWriter, error) {
|
||||
wc, err := client.OpenClientStream(cli, common.CallMethodInfoClientStream(serviceObject, rpcObjectPut), resp, opts...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &PutRequestWriter{
|
||||
wc: wc,
|
||||
resp: resp,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// GetResponseReader is an object.GetResponse
|
||||
// stream reader.
|
||||
type GetResponseReader struct {
|
||||
r client.MessageReader
|
||||
}
|
||||
|
||||
// Read reads response from the stream.
|
||||
//
|
||||
// Returns io.EOF of streaming is finished.
|
||||
func (r *GetResponseReader) Read(resp *object.GetResponse) error {
|
||||
return r.r.ReadMessage(resp)
|
||||
}
|
||||
|
||||
// GetObject executes ObjectService.Get RPC.
|
||||
func GetObject(
|
||||
cli *client.Client,
|
||||
req *object.GetRequest,
|
||||
opts ...client.CallOption,
|
||||
) (*GetResponseReader, error) {
|
||||
wc, err := client.OpenServerStream(cli, common.CallMethodInfoServerStream(serviceObject, rpcObjectGet), req, opts...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &GetResponseReader{
|
||||
r: wc,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// GetResponseReader is an object.SearchResponse
|
||||
// stream reader.
|
||||
type SearchResponseReader struct {
|
||||
r client.MessageReader
|
||||
}
|
||||
|
||||
// Read reads response from the stream.
|
||||
//
|
||||
// Returns io.EOF of streaming is finished.
|
||||
func (r *SearchResponseReader) Read(resp *object.SearchResponse) error {
|
||||
return r.r.ReadMessage(resp)
|
||||
}
|
||||
|
||||
// SearchObjects executes ObjectService.Search RPC.
|
||||
func SearchObjects(
|
||||
cli *client.Client,
|
||||
req *object.SearchRequest,
|
||||
opts ...client.CallOption,
|
||||
) (*SearchResponseReader, error) {
|
||||
wc, err := client.OpenServerStream(cli, common.CallMethodInfoServerStream(serviceObject, rpcObjectSearch), req, opts...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &SearchResponseReader{
|
||||
r: wc,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// GetResponseReader is an object.GetRangeResponse
|
||||
// stream reader.
|
||||
type ObjectRangeResponseReader struct {
|
||||
r client.MessageReader
|
||||
}
|
||||
|
||||
// Read reads response from the stream.
|
||||
//
|
||||
// Returns io.EOF of streaming is finished.
|
||||
func (r *ObjectRangeResponseReader) Read(resp *object.GetRangeResponse) error {
|
||||
return r.r.ReadMessage(resp)
|
||||
}
|
||||
|
||||
// GetObjectRange executes ObjectService.GetRange RPC.
|
||||
func GetObjectRange(
|
||||
cli *client.Client,
|
||||
req *object.GetRangeRequest,
|
||||
opts ...client.CallOption,
|
||||
) (*ObjectRangeResponseReader, error) {
|
||||
wc, err := client.OpenServerStream(cli, common.CallMethodInfoServerStream(serviceObject, rpcObjectRange), req, opts...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &ObjectRangeResponseReader{
|
||||
r: wc,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// HeadObject executes ObjectService.Head RPC.
|
||||
func HeadObject(
|
||||
cli *client.Client,
|
||||
req *object.HeadRequest,
|
||||
opts ...client.CallOption,
|
||||
) (*object.HeadResponse, error) {
|
||||
resp := new(object.HeadResponse)
|
||||
|
||||
err := client.SendUnary(cli, common.CallMethodInfoUnary(serviceObject, rpcObjectHead), req, resp, opts...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return resp, nil
|
||||
}
|
||||
|
||||
// DeleteObject executes ObjectService.Delete RPC.
|
||||
func DeleteObject(
|
||||
cli *client.Client,
|
||||
req *object.DeleteRequest,
|
||||
opts ...client.CallOption,
|
||||
) (*object.DeleteResponse, error) {
|
||||
resp := new(object.DeleteResponse)
|
||||
|
||||
err := client.SendUnary(cli, common.CallMethodInfoUnary(serviceObject, rpcObjectDelete), req, resp, opts...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return resp, nil
|
||||
}
|
||||
|
||||
// HashObjectRange executes ObjectService.GetRangeHash RPC.
|
||||
func HashObjectRange(
|
||||
cli *client.Client,
|
||||
req *object.GetRangeHashRequest,
|
||||
opts ...client.CallOption,
|
||||
) (*object.GetRangeHashResponse, error) {
|
||||
resp := new(object.GetRangeHashResponse)
|
||||
|
||||
err := client.SendUnary(cli, common.CallMethodInfoUnary(serviceObject, rpcObjectHash), req, resp, opts...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return resp, nil
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue