|
@ -17,7 +17,6 @@ type Publisher struct { |
|
|
publisherId string |
|
|
publisherId string |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
/* |
|
|
|
|
|
func (mc *MessagingClient) NewPublisher(publisherId, namespace, topic string) (*Publisher, error) { |
|
|
func (mc *MessagingClient) NewPublisher(publisherId, namespace, topic string) (*Publisher, error) { |
|
|
// read topic configuration
|
|
|
// read topic configuration
|
|
|
topicConfiguration := &messaging_pb.TopicConfiguration{ |
|
|
topicConfiguration := &messaging_pb.TopicConfiguration{ |
|
@ -25,11 +24,16 @@ func (mc *MessagingClient) NewPublisher(publisherId, namespace, topic string) (* |
|
|
} |
|
|
} |
|
|
publishClients := make([]messaging_pb.SeaweedMessaging_PublishClient, topicConfiguration.PartitionCount) |
|
|
publishClients := make([]messaging_pb.SeaweedMessaging_PublishClient, topicConfiguration.PartitionCount) |
|
|
for i := 0; i < int(topicConfiguration.PartitionCount); i++ { |
|
|
for i := 0; i < int(topicConfiguration.PartitionCount); i++ { |
|
|
client, err := setupPublisherClient(broker.TopicPartition{ |
|
|
|
|
|
|
|
|
tp := broker.TopicPartition{ |
|
|
Namespace: namespace, |
|
|
Namespace: namespace, |
|
|
Topic: topic, |
|
|
Topic: topic, |
|
|
Partition: int32(i), |
|
|
Partition: int32(i), |
|
|
}) |
|
|
|
|
|
|
|
|
} |
|
|
|
|
|
grpcClientConn, err := mc.findBroker(tp) |
|
|
|
|
|
if err != nil { |
|
|
|
|
|
return nil, err |
|
|
|
|
|
} |
|
|
|
|
|
client, err := setupPublisherClient(grpcClientConn, tp) |
|
|
if err != nil { |
|
|
if err != nil { |
|
|
return nil, err |
|
|
return nil, err |
|
|
} |
|
|
} |
|
@ -40,7 +44,6 @@ func (mc *MessagingClient) NewPublisher(publisherId, namespace, topic string) (* |
|
|
topicConfiguration: topicConfiguration, |
|
|
topicConfiguration: topicConfiguration, |
|
|
}, nil |
|
|
}, nil |
|
|
} |
|
|
} |
|
|
*/ |
|
|
|
|
|
|
|
|
|
|
|
func setupPublisherClient(grpcConnection *grpc.ClientConn, tp broker.TopicPartition) (messaging_pb.SeaweedMessaging_PublishClient, error) { |
|
|
func setupPublisherClient(grpcConnection *grpc.ClientConn, tp broker.TopicPartition) (messaging_pb.SeaweedMessaging_PublishClient, error) { |
|
|
|
|
|
|
|
@ -113,9 +116,3 @@ func (p *Publisher) Publish(m *messaging_pb.Message) error { |
|
|
Data: m, |
|
|
Data: m, |
|
|
}) |
|
|
}) |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
func (p *Publisher) Shutdown() { |
|
|
|
|
|
for _, client := range p.publishClients { |
|
|
|
|
|
client.CloseSend() |
|
|
|
|
|
} |
|
|
|
|
|
} |
|
|
|