gitee.com/liuxuezhan/go-micro-v1.18.0@v1.0.0/transport/grpc/proto/transport.micro.go (about)

     1  // Code generated by protoc-gen-micro. DO NOT EDIT.
     2  // source: go-micro/transport/grpc/proto/transport.proto
     3  
     4  package go_micro_grpc_transport
     5  
     6  import (
     7  	fmt "fmt"
     8  	proto "github.com/golang/protobuf/proto"
     9  	math "math"
    10  )
    11  
    12  import (
    13  	context "context"
    14  	client "gitee.com/liuxuezhan/go-micro-v1.18.0/client"
    15  	server "gitee.com/liuxuezhan/go-micro-v1.18.0/server"
    16  )
    17  
    18  // Reference imports to suppress errors if they are not otherwise used.
    19  var _ = proto.Marshal
    20  var _ = fmt.Errorf
    21  var _ = math.Inf
    22  
    23  // This is a compile-time assertion to ensure that this generated file
    24  // is compatible with the proto package it is being compiled against.
    25  // A compilation error at this line likely means your copy of the
    26  // proto package needs to be updated.
    27  const _ = proto.ProtoPackageIsVersion3 // please upgrade the proto package
    28  
    29  // Reference imports to suppress errors if they are not otherwise used.
    30  var _ context.Context
    31  var _ client.Option
    32  var _ server.Option
    33  
    34  // Client API for Transport service
    35  
    36  type TransportService interface {
    37  	Stream(ctx context.Context, opts ...client.CallOption) (Transport_StreamService, error)
    38  }
    39  
    40  type transportService struct {
    41  	c    client.Client
    42  	name string
    43  }
    44  
    45  func NewTransportService(name string, c client.Client) TransportService {
    46  	if c == nil {
    47  		c = client.NewClient()
    48  	}
    49  	if len(name) == 0 {
    50  		name = "go.micro.grpc.transport"
    51  	}
    52  	return &transportService{
    53  		c:    c,
    54  		name: name,
    55  	}
    56  }
    57  
    58  func (c *transportService) Stream(ctx context.Context, opts ...client.CallOption) (Transport_StreamService, error) {
    59  	req := c.c.NewRequest(c.name, "Transport.Stream", &Message{})
    60  	stream, err := c.c.Stream(ctx, req, opts...)
    61  	if err != nil {
    62  		return nil, err
    63  	}
    64  	return &transportServiceStream{stream}, nil
    65  }
    66  
    67  type Transport_StreamService interface {
    68  	SendMsg(interface{}) error
    69  	RecvMsg(interface{}) error
    70  	Close() error
    71  	Send(*Message) error
    72  	Recv() (*Message, error)
    73  }
    74  
    75  type transportServiceStream struct {
    76  	stream client.Stream
    77  }
    78  
    79  func (x *transportServiceStream) Close() error {
    80  	return x.stream.Close()
    81  }
    82  
    83  func (x *transportServiceStream) SendMsg(m interface{}) error {
    84  	return x.stream.Send(m)
    85  }
    86  
    87  func (x *transportServiceStream) RecvMsg(m interface{}) error {
    88  	return x.stream.Recv(m)
    89  }
    90  
    91  func (x *transportServiceStream) Send(m *Message) error {
    92  	return x.stream.Send(m)
    93  }
    94  
    95  func (x *transportServiceStream) Recv() (*Message, error) {
    96  	m := new(Message)
    97  	err := x.stream.Recv(m)
    98  	if err != nil {
    99  		return nil, err
   100  	}
   101  	return m, nil
   102  }
   103  
   104  // Server API for Transport service
   105  
   106  type TransportHandler interface {
   107  	Stream(context.Context, Transport_StreamStream) error
   108  }
   109  
   110  func RegisterTransportHandler(s server.Server, hdlr TransportHandler, opts ...server.HandlerOption) error {
   111  	type transport interface {
   112  		Stream(ctx context.Context, stream server.Stream) error
   113  	}
   114  	type Transport struct {
   115  		transport
   116  	}
   117  	h := &transportHandler{hdlr}
   118  	return s.Handle(s.NewHandler(&Transport{h}, opts...))
   119  }
   120  
   121  type transportHandler struct {
   122  	TransportHandler
   123  }
   124  
   125  func (h *transportHandler) Stream(ctx context.Context, stream server.Stream) error {
   126  	return h.TransportHandler.Stream(ctx, &transportStreamStream{stream})
   127  }
   128  
   129  type Transport_StreamStream interface {
   130  	SendMsg(interface{}) error
   131  	RecvMsg(interface{}) error
   132  	Close() error
   133  	Send(*Message) error
   134  	Recv() (*Message, error)
   135  }
   136  
   137  type transportStreamStream struct {
   138  	stream server.Stream
   139  }
   140  
   141  func (x *transportStreamStream) Close() error {
   142  	return x.stream.Close()
   143  }
   144  
   145  func (x *transportStreamStream) SendMsg(m interface{}) error {
   146  	return x.stream.Send(m)
   147  }
   148  
   149  func (x *transportStreamStream) RecvMsg(m interface{}) error {
   150  	return x.stream.Recv(m)
   151  }
   152  
   153  func (x *transportStreamStream) Send(m *Message) error {
   154  	return x.stream.Send(m)
   155  }
   156  
   157  func (x *transportStreamStream) Recv() (*Message, error) {
   158  	m := new(Message)
   159  	if err := x.stream.Recv(m); err != nil {
   160  		return nil, err
   161  	}
   162  	return m, nil
   163  }