mirror of
https://github.com/crazybber/go-pattern-examples.git
synced 2024-11-22 11:56:03 +03:00
finish fanout complex example error
This commit is contained in:
parent
fe3617814c
commit
e1aeee0d6e
@ -75,7 +75,9 @@ func (p *Pipeline) Dispatch(msg interface{}) {
|
|||||||
|
|
||||||
//NewPipeline create a Workflow with a dispacher builder and some workers
|
//NewPipeline create a Workflow with a dispacher builder and some workers
|
||||||
func NewPipeline(d DispatcherBuilder, idle uint32, debug bool) *Pipeline {
|
func NewPipeline(d DispatcherBuilder, idle uint32, debug bool) *Pipeline {
|
||||||
|
|
||||||
ch := make(chan interface{}, MasterQueueSize)
|
ch := make(chan interface{}, MasterQueueSize)
|
||||||
|
|
||||||
wk := make(map[int]*worker)
|
wk := make(map[int]*worker)
|
||||||
for i := 0; i < MaxWorkers; i++ {
|
for i := 0; i < MaxWorkers; i++ {
|
||||||
wk[i] = &worker{
|
wk[i] = &worker{
|
||||||
|
65
gomore/05_fan_out/fan_out_complex_test.go
Normal file
65
gomore/05_fan_out/fan_out_complex_test.go
Normal file
@ -0,0 +1,65 @@
|
|||||||
|
package fanout
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"google.golang.org/grpc"
|
||||||
|
)
|
||||||
|
|
||||||
|
//taggingDispatcher implement our dispatcher interface
|
||||||
|
type taggingDispatcher struct {
|
||||||
|
Address string
|
||||||
|
// stream proto.StreamClient
|
||||||
|
conn *grpc.ClientConn
|
||||||
|
}
|
||||||
|
type messageContent struct{}
|
||||||
|
|
||||||
|
func TestComplexStreamingFanOut(t *testing.T) {
|
||||||
|
|
||||||
|
builder := func() IDispatcher {
|
||||||
|
return &taggingDispatcher{Address: "SH"}
|
||||||
|
}
|
||||||
|
tagging := &Tagging{
|
||||||
|
topic: "new topic",
|
||||||
|
pipeline: NewPipeline(builder, 2, true),
|
||||||
|
}
|
||||||
|
tagging.pipeline.Dispatch(messageContent{})
|
||||||
|
|
||||||
|
tagging.pipeline.Start(context.Background())
|
||||||
|
}
|
||||||
|
|
||||||
|
type Tagging struct {
|
||||||
|
topic string
|
||||||
|
pipeline *Pipeline
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *taggingDispatcher) Before(ctx context.Context) error {
|
||||||
|
conn, err := grpc.Dial(d.Address, grpc.WithInsecure())
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
d.conn = conn
|
||||||
|
// // // client := proto.NewClient(conn)
|
||||||
|
// // stream, err := client.StreamMetric(ctx)
|
||||||
|
// // if err != nil {
|
||||||
|
// // return err
|
||||||
|
// // }
|
||||||
|
// // d.stream = stream
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *taggingDispatcher) After() error {
|
||||||
|
// _, err := d.stream.CloseAndRecv()
|
||||||
|
// e := d.conn.Close()
|
||||||
|
// if e != nil {
|
||||||
|
// log.Error("close connection error", field.Error(e))
|
||||||
|
// }
|
||||||
|
//return err
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *taggingDispatcher) Process(msg interface{}) error {
|
||||||
|
//return d.stream.Send(msg.(*proto.Tagging))
|
||||||
|
return nil
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user