mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-01-07 03:13:22 +08:00
86eff6e589
issue: #33285 - add streaming node grpc client wrapper - add unittest for streaming node grpc client side - fix binary unsafe bug for message --------- Signed-off-by: chyezh <chyezh@outlook.com>
31 lines
881 B
Go
31 lines
881 B
Go
package producer
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/milvus-io/milvus/pkg/streaming/util/message"
|
|
"github.com/milvus-io/milvus/pkg/streaming/util/types"
|
|
)
|
|
|
|
var _ Producer = (*producerImpl)(nil)
|
|
|
|
// Producer is the interface that wraps the basic produce method on grpc stream.
|
|
// Producer is work on a single stream on grpc,
|
|
// so Producer cannot recover from failure because of the stream is broken.
|
|
type Producer interface {
|
|
// Assignment returns the assignment of the producer.
|
|
Assignment() types.PChannelInfoAssigned
|
|
|
|
// Produce sends the produce message to server.
|
|
Produce(ctx context.Context, msg message.MutableMessage) (message.MessageID, error)
|
|
|
|
// Check if a producer is available.
|
|
IsAvailable() bool
|
|
|
|
// Available returns a channel that will be closed when the producer is unavailable.
|
|
Available() <-chan struct{}
|
|
|
|
// Close close the producer client.
|
|
Close()
|
|
}
|