50 lines
1.3 KiB
Go
50 lines
1.3 KiB
Go
|
package grpc
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/go-kit/kit/endpoint"
|
||
|
|
kitgrpc "github.com/go-kit/kit/transport/grpc"
|
||
|
|
"github.com/mainflux/mainflux"
|
||
|
|
"golang.org/x/net/context"
|
||
|
|
"google.golang.org/grpc"
|
||
|
|
)
|
||
|
|
|
||
|
|
var _ mainflux.UsersServiceClient = (*grpcClient)(nil)
|
||
|
|
|
||
|
|
type grpcClient struct {
|
||
|
|
identify endpoint.Endpoint
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewClient returns new gRPC client instance.
|
||
|
|
func NewClient(conn *grpc.ClientConn) mainflux.UsersServiceClient {
|
||
|
|
endpoint := kitgrpc.NewClient(
|
||
|
|
conn,
|
||
|
|
"mainflux.UsersService",
|
||
|
|
"Identify",
|
||
|
|
encodeIdentifyRequest,
|
||
|
|
decodeIdentifyResponse,
|
||
|
|
mainflux.Identity{},
|
||
|
|
).Endpoint()
|
||
|
|
|
||
|
|
return &grpcClient{endpoint}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (client grpcClient) Identify(ctx context.Context, token *mainflux.Token, _ ...grpc.CallOption) (*mainflux.Identity, error) {
|
||
|
|
res, err := client.identify(ctx, identityReq{token.GetValue()})
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
ir := res.(identityRes)
|
||
|
|
return &mainflux.Identity{Value: ir.id}, ir.err
|
||
|
|
}
|
||
|
|
|
||
|
|
func encodeIdentifyRequest(_ context.Context, grpcReq interface{}) (interface{}, error) {
|
||
|
|
req := grpcReq.(identityReq)
|
||
|
|
return &mainflux.Token{Value: req.token}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func decodeIdentifyResponse(_ context.Context, grpcRes interface{}) (interface{}, error) {
|
||
|
|
res := grpcRes.(*mainflux.Identity)
|
||
|
|
return identityRes{res.GetValue(), nil}, nil
|
||
|
|
}
|