You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

603 lines
24 KiB

3 years ago
3 years ago
3 years ago
3 years ago
1 year ago
3 years ago
3 years ago
3 years ago
1 year ago
1 year ago
1 year ago
1 year ago
3 years ago
3 years ago
3 years ago
1 year ago
3 years ago
3 years ago
1 year ago
1 year ago
3 years ago
3 years ago
3 years ago
3 years ago
1 year ago
1 year ago
1 year ago
3 years ago
3 years ago
3 years ago
3 years ago
1 year ago
3 years ago
3 years ago
  1. // Code generated by protoc-gen-go-grpc. DO NOT EDIT.
  2. // versions:
  3. // - protoc-gen-go-grpc v1.2.0
  4. // - protoc v4.23.3
  5. // source: mq.proto
  6. package mq_pb
  7. import (
  8. context "context"
  9. grpc "google.golang.org/grpc"
  10. codes "google.golang.org/grpc/codes"
  11. status "google.golang.org/grpc/status"
  12. )
  13. // This is a compile-time assertion to ensure that this generated file
  14. // is compatible with the grpc package it is being compiled against.
  15. // Requires gRPC-Go v1.32.0 or later.
  16. const _ = grpc.SupportPackageIsVersion7
  17. // SeaweedMessagingClient is the client API for SeaweedMessaging service.
  18. //
  19. // For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream.
  20. type SeaweedMessagingClient interface {
  21. // control plane
  22. FindBrokerLeader(ctx context.Context, in *FindBrokerLeaderRequest, opts ...grpc.CallOption) (*FindBrokerLeaderResponse, error)
  23. AssignSegmentBrokers(ctx context.Context, in *AssignSegmentBrokersRequest, opts ...grpc.CallOption) (*AssignSegmentBrokersResponse, error)
  24. CheckSegmentStatus(ctx context.Context, in *CheckSegmentStatusRequest, opts ...grpc.CallOption) (*CheckSegmentStatusResponse, error)
  25. CheckBrokerLoad(ctx context.Context, in *CheckBrokerLoadRequest, opts ...grpc.CallOption) (*CheckBrokerLoadResponse, error)
  26. // control plane for balancer
  27. ConnectToBalancer(ctx context.Context, opts ...grpc.CallOption) (SeaweedMessaging_ConnectToBalancerClient, error)
  28. // control plane for topic partitions
  29. LookupTopicBrokers(ctx context.Context, in *LookupTopicBrokersRequest, opts ...grpc.CallOption) (*LookupTopicBrokersResponse, error)
  30. CreateTopic(ctx context.Context, in *CreateTopicRequest, opts ...grpc.CallOption) (*CreateTopicResponse, error)
  31. // a pub client will call this to get the topic partitions assignment
  32. RequestTopicPartitions(ctx context.Context, in *RequestTopicPartitionsRequest, opts ...grpc.CallOption) (*RequestTopicPartitionsResponse, error)
  33. AssignTopicPartitions(ctx context.Context, in *AssignTopicPartitionsRequest, opts ...grpc.CallOption) (*AssignTopicPartitionsResponse, error)
  34. CheckTopicPartitionsStatus(ctx context.Context, in *CheckTopicPartitionsStatusRequest, opts ...grpc.CallOption) (*CheckTopicPartitionsStatusResponse, error)
  35. // data plane
  36. Publish(ctx context.Context, opts ...grpc.CallOption) (SeaweedMessaging_PublishClient, error)
  37. Subscribe(ctx context.Context, in *SubscribeRequest, opts ...grpc.CallOption) (SeaweedMessaging_SubscribeClient, error)
  38. }
  39. type seaweedMessagingClient struct {
  40. cc grpc.ClientConnInterface
  41. }
  42. func NewSeaweedMessagingClient(cc grpc.ClientConnInterface) SeaweedMessagingClient {
  43. return &seaweedMessagingClient{cc}
  44. }
  45. func (c *seaweedMessagingClient) FindBrokerLeader(ctx context.Context, in *FindBrokerLeaderRequest, opts ...grpc.CallOption) (*FindBrokerLeaderResponse, error) {
  46. out := new(FindBrokerLeaderResponse)
  47. err := c.cc.Invoke(ctx, "/messaging_pb.SeaweedMessaging/FindBrokerLeader", in, out, opts...)
  48. if err != nil {
  49. return nil, err
  50. }
  51. return out, nil
  52. }
  53. func (c *seaweedMessagingClient) AssignSegmentBrokers(ctx context.Context, in *AssignSegmentBrokersRequest, opts ...grpc.CallOption) (*AssignSegmentBrokersResponse, error) {
  54. out := new(AssignSegmentBrokersResponse)
  55. err := c.cc.Invoke(ctx, "/messaging_pb.SeaweedMessaging/AssignSegmentBrokers", in, out, opts...)
  56. if err != nil {
  57. return nil, err
  58. }
  59. return out, nil
  60. }
  61. func (c *seaweedMessagingClient) CheckSegmentStatus(ctx context.Context, in *CheckSegmentStatusRequest, opts ...grpc.CallOption) (*CheckSegmentStatusResponse, error) {
  62. out := new(CheckSegmentStatusResponse)
  63. err := c.cc.Invoke(ctx, "/messaging_pb.SeaweedMessaging/CheckSegmentStatus", in, out, opts...)
  64. if err != nil {
  65. return nil, err
  66. }
  67. return out, nil
  68. }
  69. func (c *seaweedMessagingClient) CheckBrokerLoad(ctx context.Context, in *CheckBrokerLoadRequest, opts ...grpc.CallOption) (*CheckBrokerLoadResponse, error) {
  70. out := new(CheckBrokerLoadResponse)
  71. err := c.cc.Invoke(ctx, "/messaging_pb.SeaweedMessaging/CheckBrokerLoad", in, out, opts...)
  72. if err != nil {
  73. return nil, err
  74. }
  75. return out, nil
  76. }
  77. func (c *seaweedMessagingClient) ConnectToBalancer(ctx context.Context, opts ...grpc.CallOption) (SeaweedMessaging_ConnectToBalancerClient, error) {
  78. stream, err := c.cc.NewStream(ctx, &SeaweedMessaging_ServiceDesc.Streams[0], "/messaging_pb.SeaweedMessaging/ConnectToBalancer", opts...)
  79. if err != nil {
  80. return nil, err
  81. }
  82. x := &seaweedMessagingConnectToBalancerClient{stream}
  83. return x, nil
  84. }
  85. type SeaweedMessaging_ConnectToBalancerClient interface {
  86. Send(*ConnectToBalancerRequest) error
  87. Recv() (*ConnectToBalancerResponse, error)
  88. grpc.ClientStream
  89. }
  90. type seaweedMessagingConnectToBalancerClient struct {
  91. grpc.ClientStream
  92. }
  93. func (x *seaweedMessagingConnectToBalancerClient) Send(m *ConnectToBalancerRequest) error {
  94. return x.ClientStream.SendMsg(m)
  95. }
  96. func (x *seaweedMessagingConnectToBalancerClient) Recv() (*ConnectToBalancerResponse, error) {
  97. m := new(ConnectToBalancerResponse)
  98. if err := x.ClientStream.RecvMsg(m); err != nil {
  99. return nil, err
  100. }
  101. return m, nil
  102. }
  103. func (c *seaweedMessagingClient) LookupTopicBrokers(ctx context.Context, in *LookupTopicBrokersRequest, opts ...grpc.CallOption) (*LookupTopicBrokersResponse, error) {
  104. out := new(LookupTopicBrokersResponse)
  105. err := c.cc.Invoke(ctx, "/messaging_pb.SeaweedMessaging/LookupTopicBrokers", in, out, opts...)
  106. if err != nil {
  107. return nil, err
  108. }
  109. return out, nil
  110. }
  111. func (c *seaweedMessagingClient) CreateTopic(ctx context.Context, in *CreateTopicRequest, opts ...grpc.CallOption) (*CreateTopicResponse, error) {
  112. out := new(CreateTopicResponse)
  113. err := c.cc.Invoke(ctx, "/messaging_pb.SeaweedMessaging/CreateTopic", in, out, opts...)
  114. if err != nil {
  115. return nil, err
  116. }
  117. return out, nil
  118. }
  119. func (c *seaweedMessagingClient) RequestTopicPartitions(ctx context.Context, in *RequestTopicPartitionsRequest, opts ...grpc.CallOption) (*RequestTopicPartitionsResponse, error) {
  120. out := new(RequestTopicPartitionsResponse)
  121. err := c.cc.Invoke(ctx, "/messaging_pb.SeaweedMessaging/RequestTopicPartitions", in, out, opts...)
  122. if err != nil {
  123. return nil, err
  124. }
  125. return out, nil
  126. }
  127. func (c *seaweedMessagingClient) AssignTopicPartitions(ctx context.Context, in *AssignTopicPartitionsRequest, opts ...grpc.CallOption) (*AssignTopicPartitionsResponse, error) {
  128. out := new(AssignTopicPartitionsResponse)
  129. err := c.cc.Invoke(ctx, "/messaging_pb.SeaweedMessaging/AssignTopicPartitions", in, out, opts...)
  130. if err != nil {
  131. return nil, err
  132. }
  133. return out, nil
  134. }
  135. func (c *seaweedMessagingClient) CheckTopicPartitionsStatus(ctx context.Context, in *CheckTopicPartitionsStatusRequest, opts ...grpc.CallOption) (*CheckTopicPartitionsStatusResponse, error) {
  136. out := new(CheckTopicPartitionsStatusResponse)
  137. err := c.cc.Invoke(ctx, "/messaging_pb.SeaweedMessaging/CheckTopicPartitionsStatus", in, out, opts...)
  138. if err != nil {
  139. return nil, err
  140. }
  141. return out, nil
  142. }
  143. func (c *seaweedMessagingClient) Publish(ctx context.Context, opts ...grpc.CallOption) (SeaweedMessaging_PublishClient, error) {
  144. stream, err := c.cc.NewStream(ctx, &SeaweedMessaging_ServiceDesc.Streams[1], "/messaging_pb.SeaweedMessaging/Publish", opts...)
  145. if err != nil {
  146. return nil, err
  147. }
  148. x := &seaweedMessagingPublishClient{stream}
  149. return x, nil
  150. }
  151. type SeaweedMessaging_PublishClient interface {
  152. Send(*PublishRequest) error
  153. Recv() (*PublishResponse, error)
  154. grpc.ClientStream
  155. }
  156. type seaweedMessagingPublishClient struct {
  157. grpc.ClientStream
  158. }
  159. func (x *seaweedMessagingPublishClient) Send(m *PublishRequest) error {
  160. return x.ClientStream.SendMsg(m)
  161. }
  162. func (x *seaweedMessagingPublishClient) Recv() (*PublishResponse, error) {
  163. m := new(PublishResponse)
  164. if err := x.ClientStream.RecvMsg(m); err != nil {
  165. return nil, err
  166. }
  167. return m, nil
  168. }
  169. func (c *seaweedMessagingClient) Subscribe(ctx context.Context, in *SubscribeRequest, opts ...grpc.CallOption) (SeaweedMessaging_SubscribeClient, error) {
  170. stream, err := c.cc.NewStream(ctx, &SeaweedMessaging_ServiceDesc.Streams[2], "/messaging_pb.SeaweedMessaging/Subscribe", opts...)
  171. if err != nil {
  172. return nil, err
  173. }
  174. x := &seaweedMessagingSubscribeClient{stream}
  175. if err := x.ClientStream.SendMsg(in); err != nil {
  176. return nil, err
  177. }
  178. if err := x.ClientStream.CloseSend(); err != nil {
  179. return nil, err
  180. }
  181. return x, nil
  182. }
  183. type SeaweedMessaging_SubscribeClient interface {
  184. Recv() (*SubscribeResponse, error)
  185. grpc.ClientStream
  186. }
  187. type seaweedMessagingSubscribeClient struct {
  188. grpc.ClientStream
  189. }
  190. func (x *seaweedMessagingSubscribeClient) Recv() (*SubscribeResponse, error) {
  191. m := new(SubscribeResponse)
  192. if err := x.ClientStream.RecvMsg(m); err != nil {
  193. return nil, err
  194. }
  195. return m, nil
  196. }
  197. // SeaweedMessagingServer is the server API for SeaweedMessaging service.
  198. // All implementations must embed UnimplementedSeaweedMessagingServer
  199. // for forward compatibility
  200. type SeaweedMessagingServer interface {
  201. // control plane
  202. FindBrokerLeader(context.Context, *FindBrokerLeaderRequest) (*FindBrokerLeaderResponse, error)
  203. AssignSegmentBrokers(context.Context, *AssignSegmentBrokersRequest) (*AssignSegmentBrokersResponse, error)
  204. CheckSegmentStatus(context.Context, *CheckSegmentStatusRequest) (*CheckSegmentStatusResponse, error)
  205. CheckBrokerLoad(context.Context, *CheckBrokerLoadRequest) (*CheckBrokerLoadResponse, error)
  206. // control plane for balancer
  207. ConnectToBalancer(SeaweedMessaging_ConnectToBalancerServer) error
  208. // control plane for topic partitions
  209. LookupTopicBrokers(context.Context, *LookupTopicBrokersRequest) (*LookupTopicBrokersResponse, error)
  210. CreateTopic(context.Context, *CreateTopicRequest) (*CreateTopicResponse, error)
  211. // a pub client will call this to get the topic partitions assignment
  212. RequestTopicPartitions(context.Context, *RequestTopicPartitionsRequest) (*RequestTopicPartitionsResponse, error)
  213. AssignTopicPartitions(context.Context, *AssignTopicPartitionsRequest) (*AssignTopicPartitionsResponse, error)
  214. CheckTopicPartitionsStatus(context.Context, *CheckTopicPartitionsStatusRequest) (*CheckTopicPartitionsStatusResponse, error)
  215. // data plane
  216. Publish(SeaweedMessaging_PublishServer) error
  217. Subscribe(*SubscribeRequest, SeaweedMessaging_SubscribeServer) error
  218. mustEmbedUnimplementedSeaweedMessagingServer()
  219. }
  220. // UnimplementedSeaweedMessagingServer must be embedded to have forward compatible implementations.
  221. type UnimplementedSeaweedMessagingServer struct {
  222. }
  223. func (UnimplementedSeaweedMessagingServer) FindBrokerLeader(context.Context, *FindBrokerLeaderRequest) (*FindBrokerLeaderResponse, error) {
  224. return nil, status.Errorf(codes.Unimplemented, "method FindBrokerLeader not implemented")
  225. }
  226. func (UnimplementedSeaweedMessagingServer) AssignSegmentBrokers(context.Context, *AssignSegmentBrokersRequest) (*AssignSegmentBrokersResponse, error) {
  227. return nil, status.Errorf(codes.Unimplemented, "method AssignSegmentBrokers not implemented")
  228. }
  229. func (UnimplementedSeaweedMessagingServer) CheckSegmentStatus(context.Context, *CheckSegmentStatusRequest) (*CheckSegmentStatusResponse, error) {
  230. return nil, status.Errorf(codes.Unimplemented, "method CheckSegmentStatus not implemented")
  231. }
  232. func (UnimplementedSeaweedMessagingServer) CheckBrokerLoad(context.Context, *CheckBrokerLoadRequest) (*CheckBrokerLoadResponse, error) {
  233. return nil, status.Errorf(codes.Unimplemented, "method CheckBrokerLoad not implemented")
  234. }
  235. func (UnimplementedSeaweedMessagingServer) ConnectToBalancer(SeaweedMessaging_ConnectToBalancerServer) error {
  236. return status.Errorf(codes.Unimplemented, "method ConnectToBalancer not implemented")
  237. }
  238. func (UnimplementedSeaweedMessagingServer) LookupTopicBrokers(context.Context, *LookupTopicBrokersRequest) (*LookupTopicBrokersResponse, error) {
  239. return nil, status.Errorf(codes.Unimplemented, "method LookupTopicBrokers not implemented")
  240. }
  241. func (UnimplementedSeaweedMessagingServer) CreateTopic(context.Context, *CreateTopicRequest) (*CreateTopicResponse, error) {
  242. return nil, status.Errorf(codes.Unimplemented, "method CreateTopic not implemented")
  243. }
  244. func (UnimplementedSeaweedMessagingServer) RequestTopicPartitions(context.Context, *RequestTopicPartitionsRequest) (*RequestTopicPartitionsResponse, error) {
  245. return nil, status.Errorf(codes.Unimplemented, "method RequestTopicPartitions not implemented")
  246. }
  247. func (UnimplementedSeaweedMessagingServer) AssignTopicPartitions(context.Context, *AssignTopicPartitionsRequest) (*AssignTopicPartitionsResponse, error) {
  248. return nil, status.Errorf(codes.Unimplemented, "method AssignTopicPartitions not implemented")
  249. }
  250. func (UnimplementedSeaweedMessagingServer) CheckTopicPartitionsStatus(context.Context, *CheckTopicPartitionsStatusRequest) (*CheckTopicPartitionsStatusResponse, error) {
  251. return nil, status.Errorf(codes.Unimplemented, "method CheckTopicPartitionsStatus not implemented")
  252. }
  253. func (UnimplementedSeaweedMessagingServer) Publish(SeaweedMessaging_PublishServer) error {
  254. return status.Errorf(codes.Unimplemented, "method Publish not implemented")
  255. }
  256. func (UnimplementedSeaweedMessagingServer) Subscribe(*SubscribeRequest, SeaweedMessaging_SubscribeServer) error {
  257. return status.Errorf(codes.Unimplemented, "method Subscribe not implemented")
  258. }
  259. func (UnimplementedSeaweedMessagingServer) mustEmbedUnimplementedSeaweedMessagingServer() {}
  260. // UnsafeSeaweedMessagingServer may be embedded to opt out of forward compatibility for this service.
  261. // Use of this interface is not recommended, as added methods to SeaweedMessagingServer will
  262. // result in compilation errors.
  263. type UnsafeSeaweedMessagingServer interface {
  264. mustEmbedUnimplementedSeaweedMessagingServer()
  265. }
  266. func RegisterSeaweedMessagingServer(s grpc.ServiceRegistrar, srv SeaweedMessagingServer) {
  267. s.RegisterService(&SeaweedMessaging_ServiceDesc, srv)
  268. }
  269. func _SeaweedMessaging_FindBrokerLeader_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
  270. in := new(FindBrokerLeaderRequest)
  271. if err := dec(in); err != nil {
  272. return nil, err
  273. }
  274. if interceptor == nil {
  275. return srv.(SeaweedMessagingServer).FindBrokerLeader(ctx, in)
  276. }
  277. info := &grpc.UnaryServerInfo{
  278. Server: srv,
  279. FullMethod: "/messaging_pb.SeaweedMessaging/FindBrokerLeader",
  280. }
  281. handler := func(ctx context.Context, req interface{}) (interface{}, error) {
  282. return srv.(SeaweedMessagingServer).FindBrokerLeader(ctx, req.(*FindBrokerLeaderRequest))
  283. }
  284. return interceptor(ctx, in, info, handler)
  285. }
  286. func _SeaweedMessaging_AssignSegmentBrokers_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
  287. in := new(AssignSegmentBrokersRequest)
  288. if err := dec(in); err != nil {
  289. return nil, err
  290. }
  291. if interceptor == nil {
  292. return srv.(SeaweedMessagingServer).AssignSegmentBrokers(ctx, in)
  293. }
  294. info := &grpc.UnaryServerInfo{
  295. Server: srv,
  296. FullMethod: "/messaging_pb.SeaweedMessaging/AssignSegmentBrokers",
  297. }
  298. handler := func(ctx context.Context, req interface{}) (interface{}, error) {
  299. return srv.(SeaweedMessagingServer).AssignSegmentBrokers(ctx, req.(*AssignSegmentBrokersRequest))
  300. }
  301. return interceptor(ctx, in, info, handler)
  302. }
  303. func _SeaweedMessaging_CheckSegmentStatus_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
  304. in := new(CheckSegmentStatusRequest)
  305. if err := dec(in); err != nil {
  306. return nil, err
  307. }
  308. if interceptor == nil {
  309. return srv.(SeaweedMessagingServer).CheckSegmentStatus(ctx, in)
  310. }
  311. info := &grpc.UnaryServerInfo{
  312. Server: srv,
  313. FullMethod: "/messaging_pb.SeaweedMessaging/CheckSegmentStatus",
  314. }
  315. handler := func(ctx context.Context, req interface{}) (interface{}, error) {
  316. return srv.(SeaweedMessagingServer).CheckSegmentStatus(ctx, req.(*CheckSegmentStatusRequest))
  317. }
  318. return interceptor(ctx, in, info, handler)
  319. }
  320. func _SeaweedMessaging_CheckBrokerLoad_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
  321. in := new(CheckBrokerLoadRequest)
  322. if err := dec(in); err != nil {
  323. return nil, err
  324. }
  325. if interceptor == nil {
  326. return srv.(SeaweedMessagingServer).CheckBrokerLoad(ctx, in)
  327. }
  328. info := &grpc.UnaryServerInfo{
  329. Server: srv,
  330. FullMethod: "/messaging_pb.SeaweedMessaging/CheckBrokerLoad",
  331. }
  332. handler := func(ctx context.Context, req interface{}) (interface{}, error) {
  333. return srv.(SeaweedMessagingServer).CheckBrokerLoad(ctx, req.(*CheckBrokerLoadRequest))
  334. }
  335. return interceptor(ctx, in, info, handler)
  336. }
  337. func _SeaweedMessaging_ConnectToBalancer_Handler(srv interface{}, stream grpc.ServerStream) error {
  338. return srv.(SeaweedMessagingServer).ConnectToBalancer(&seaweedMessagingConnectToBalancerServer{stream})
  339. }
  340. type SeaweedMessaging_ConnectToBalancerServer interface {
  341. Send(*ConnectToBalancerResponse) error
  342. Recv() (*ConnectToBalancerRequest, error)
  343. grpc.ServerStream
  344. }
  345. type seaweedMessagingConnectToBalancerServer struct {
  346. grpc.ServerStream
  347. }
  348. func (x *seaweedMessagingConnectToBalancerServer) Send(m *ConnectToBalancerResponse) error {
  349. return x.ServerStream.SendMsg(m)
  350. }
  351. func (x *seaweedMessagingConnectToBalancerServer) Recv() (*ConnectToBalancerRequest, error) {
  352. m := new(ConnectToBalancerRequest)
  353. if err := x.ServerStream.RecvMsg(m); err != nil {
  354. return nil, err
  355. }
  356. return m, nil
  357. }
  358. func _SeaweedMessaging_LookupTopicBrokers_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
  359. in := new(LookupTopicBrokersRequest)
  360. if err := dec(in); err != nil {
  361. return nil, err
  362. }
  363. if interceptor == nil {
  364. return srv.(SeaweedMessagingServer).LookupTopicBrokers(ctx, in)
  365. }
  366. info := &grpc.UnaryServerInfo{
  367. Server: srv,
  368. FullMethod: "/messaging_pb.SeaweedMessaging/LookupTopicBrokers",
  369. }
  370. handler := func(ctx context.Context, req interface{}) (interface{}, error) {
  371. return srv.(SeaweedMessagingServer).LookupTopicBrokers(ctx, req.(*LookupTopicBrokersRequest))
  372. }
  373. return interceptor(ctx, in, info, handler)
  374. }
  375. func _SeaweedMessaging_CreateTopic_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
  376. in := new(CreateTopicRequest)
  377. if err := dec(in); err != nil {
  378. return nil, err
  379. }
  380. if interceptor == nil {
  381. return srv.(SeaweedMessagingServer).CreateTopic(ctx, in)
  382. }
  383. info := &grpc.UnaryServerInfo{
  384. Server: srv,
  385. FullMethod: "/messaging_pb.SeaweedMessaging/CreateTopic",
  386. }
  387. handler := func(ctx context.Context, req interface{}) (interface{}, error) {
  388. return srv.(SeaweedMessagingServer).CreateTopic(ctx, req.(*CreateTopicRequest))
  389. }
  390. return interceptor(ctx, in, info, handler)
  391. }
  392. func _SeaweedMessaging_RequestTopicPartitions_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
  393. in := new(RequestTopicPartitionsRequest)
  394. if err := dec(in); err != nil {
  395. return nil, err
  396. }
  397. if interceptor == nil {
  398. return srv.(SeaweedMessagingServer).RequestTopicPartitions(ctx, in)
  399. }
  400. info := &grpc.UnaryServerInfo{
  401. Server: srv,
  402. FullMethod: "/messaging_pb.SeaweedMessaging/RequestTopicPartitions",
  403. }
  404. handler := func(ctx context.Context, req interface{}) (interface{}, error) {
  405. return srv.(SeaweedMessagingServer).RequestTopicPartitions(ctx, req.(*RequestTopicPartitionsRequest))
  406. }
  407. return interceptor(ctx, in, info, handler)
  408. }
  409. func _SeaweedMessaging_AssignTopicPartitions_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
  410. in := new(AssignTopicPartitionsRequest)
  411. if err := dec(in); err != nil {
  412. return nil, err
  413. }
  414. if interceptor == nil {
  415. return srv.(SeaweedMessagingServer).AssignTopicPartitions(ctx, in)
  416. }
  417. info := &grpc.UnaryServerInfo{
  418. Server: srv,
  419. FullMethod: "/messaging_pb.SeaweedMessaging/AssignTopicPartitions",
  420. }
  421. handler := func(ctx context.Context, req interface{}) (interface{}, error) {
  422. return srv.(SeaweedMessagingServer).AssignTopicPartitions(ctx, req.(*AssignTopicPartitionsRequest))
  423. }
  424. return interceptor(ctx, in, info, handler)
  425. }
  426. func _SeaweedMessaging_CheckTopicPartitionsStatus_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
  427. in := new(CheckTopicPartitionsStatusRequest)
  428. if err := dec(in); err != nil {
  429. return nil, err
  430. }
  431. if interceptor == nil {
  432. return srv.(SeaweedMessagingServer).CheckTopicPartitionsStatus(ctx, in)
  433. }
  434. info := &grpc.UnaryServerInfo{
  435. Server: srv,
  436. FullMethod: "/messaging_pb.SeaweedMessaging/CheckTopicPartitionsStatus",
  437. }
  438. handler := func(ctx context.Context, req interface{}) (interface{}, error) {
  439. return srv.(SeaweedMessagingServer).CheckTopicPartitionsStatus(ctx, req.(*CheckTopicPartitionsStatusRequest))
  440. }
  441. return interceptor(ctx, in, info, handler)
  442. }
  443. func _SeaweedMessaging_Publish_Handler(srv interface{}, stream grpc.ServerStream) error {
  444. return srv.(SeaweedMessagingServer).Publish(&seaweedMessagingPublishServer{stream})
  445. }
  446. type SeaweedMessaging_PublishServer interface {
  447. Send(*PublishResponse) error
  448. Recv() (*PublishRequest, error)
  449. grpc.ServerStream
  450. }
  451. type seaweedMessagingPublishServer struct {
  452. grpc.ServerStream
  453. }
  454. func (x *seaweedMessagingPublishServer) Send(m *PublishResponse) error {
  455. return x.ServerStream.SendMsg(m)
  456. }
  457. func (x *seaweedMessagingPublishServer) Recv() (*PublishRequest, error) {
  458. m := new(PublishRequest)
  459. if err := x.ServerStream.RecvMsg(m); err != nil {
  460. return nil, err
  461. }
  462. return m, nil
  463. }
  464. func _SeaweedMessaging_Subscribe_Handler(srv interface{}, stream grpc.ServerStream) error {
  465. m := new(SubscribeRequest)
  466. if err := stream.RecvMsg(m); err != nil {
  467. return err
  468. }
  469. return srv.(SeaweedMessagingServer).Subscribe(m, &seaweedMessagingSubscribeServer{stream})
  470. }
  471. type SeaweedMessaging_SubscribeServer interface {
  472. Send(*SubscribeResponse) error
  473. grpc.ServerStream
  474. }
  475. type seaweedMessagingSubscribeServer struct {
  476. grpc.ServerStream
  477. }
  478. func (x *seaweedMessagingSubscribeServer) Send(m *SubscribeResponse) error {
  479. return x.ServerStream.SendMsg(m)
  480. }
  481. // SeaweedMessaging_ServiceDesc is the grpc.ServiceDesc for SeaweedMessaging service.
  482. // It's only intended for direct use with grpc.RegisterService,
  483. // and not to be introspected or modified (even as a copy)
  484. var SeaweedMessaging_ServiceDesc = grpc.ServiceDesc{
  485. ServiceName: "messaging_pb.SeaweedMessaging",
  486. HandlerType: (*SeaweedMessagingServer)(nil),
  487. Methods: []grpc.MethodDesc{
  488. {
  489. MethodName: "FindBrokerLeader",
  490. Handler: _SeaweedMessaging_FindBrokerLeader_Handler,
  491. },
  492. {
  493. MethodName: "AssignSegmentBrokers",
  494. Handler: _SeaweedMessaging_AssignSegmentBrokers_Handler,
  495. },
  496. {
  497. MethodName: "CheckSegmentStatus",
  498. Handler: _SeaweedMessaging_CheckSegmentStatus_Handler,
  499. },
  500. {
  501. MethodName: "CheckBrokerLoad",
  502. Handler: _SeaweedMessaging_CheckBrokerLoad_Handler,
  503. },
  504. {
  505. MethodName: "LookupTopicBrokers",
  506. Handler: _SeaweedMessaging_LookupTopicBrokers_Handler,
  507. },
  508. {
  509. MethodName: "CreateTopic",
  510. Handler: _SeaweedMessaging_CreateTopic_Handler,
  511. },
  512. {
  513. MethodName: "RequestTopicPartitions",
  514. Handler: _SeaweedMessaging_RequestTopicPartitions_Handler,
  515. },
  516. {
  517. MethodName: "AssignTopicPartitions",
  518. Handler: _SeaweedMessaging_AssignTopicPartitions_Handler,
  519. },
  520. {
  521. MethodName: "CheckTopicPartitionsStatus",
  522. Handler: _SeaweedMessaging_CheckTopicPartitionsStatus_Handler,
  523. },
  524. },
  525. Streams: []grpc.StreamDesc{
  526. {
  527. StreamName: "ConnectToBalancer",
  528. Handler: _SeaweedMessaging_ConnectToBalancer_Handler,
  529. ServerStreams: true,
  530. ClientStreams: true,
  531. },
  532. {
  533. StreamName: "Publish",
  534. Handler: _SeaweedMessaging_Publish_Handler,
  535. ServerStreams: true,
  536. ClientStreams: true,
  537. },
  538. {
  539. StreamName: "Subscribe",
  540. Handler: _SeaweedMessaging_Subscribe_Handler,
  541. ServerStreams: true,
  542. },
  543. },
  544. Metadata: "mq.proto",
  545. }