Compare commits
	
		
			2 Commits
		
	
	
		
			v4.1.7
			...
			vtolstov-p
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
|  | 629087ec7e | ||
| c576abf26a | 
| @@ -26,24 +26,24 @@ jobs: | |||||||
|  |  | ||||||
|     - name: test coverage |     - name: test coverage | ||||||
|       run: | |       run: | | ||||||
|         go test -v -cover ./... -covermode=count -coverprofile coverage.out -coverpkg ./... |         go test -v -cover ./... -coverprofile coverage.out -coverpkg ./... | ||||||
|         go tool cover -func coverage.out -o coverage.out |         go tool cover -func coverage.out -o coverage.out | ||||||
|  |  | ||||||
|     - name: coverage badge |     - name: coverage badge | ||||||
|       uses: tj-actions/coverage-badge-go@v2 |       uses: tj-actions/coverage-badge-go@v1 | ||||||
|       with: |       with: | ||||||
|         green: 80 |         green: 80 | ||||||
|         filename: coverage.out |         filename: coverage.out | ||||||
|  |  | ||||||
|     - uses: stefanzweifel/git-auto-commit-action@v4 |     - uses: stefanzweifel/git-auto-commit-action@v4 | ||||||
|       name: autocommit |       id: auto-commit-action | ||||||
|       with: |       with: | ||||||
|         commit_message: Apply Code Coverage Badge |         commit_message: Apply Code Coverage Badge | ||||||
|         skip_fetch: true |         skip_fetch: true | ||||||
|         skip_checkout: true |         skip_checkout: true | ||||||
|         file_pattern: ./README.md |         file_pattern: ./README.md | ||||||
|  |  | ||||||
|     - name: push |     - name: Push Changes | ||||||
|       if: steps.auto-commit-action.outputs.changes_detected == 'true' |       if: steps.auto-commit-action.outputs.changes_detected == 'true' | ||||||
|       uses: ad-m/github-push-action@master |       uses: ad-m/github-push-action@master | ||||||
|       with: |       with: | ||||||
|   | |||||||
							
								
								
									
										26
									
								
								README.md
									
									
									
									
									
								
							
							
						
						
									
										26
									
								
								README.md
									
									
									
									
									
								
							| @@ -1,9 +1,5 @@ | |||||||
| # Micro  | # Micro  | ||||||
|  |  | ||||||
| [](https://opensource.org/licenses/Apache-2.0) |  | ||||||
| [](https://pkg.go.dev/go.unistack.org/micro/v4?tab=overview) |  | ||||||
| [](https://git.unistack.org/unistack-org/micro/actions?query=workflow%3Abuild+branch%3Av4+event%3Apush) |  | ||||||
| [](https://goreportcard.com/report/go.unistack.org/micro/v4) |  | ||||||
|  |  | ||||||
| Micro is a standard library for microservices. | Micro is a standard library for microservices. | ||||||
|  |  | ||||||
| @@ -15,20 +11,30 @@ Micro provides the core requirements for distributed systems development includi | |||||||
|  |  | ||||||
| Micro abstracts away the details of distributed systems. Here are the main features. | Micro abstracts away the details of distributed systems. Here are the main features. | ||||||
|  |  | ||||||
|  | - **Authentication** - Auth is built in as a first class citizen. Authentication and authorization enable secure  | ||||||
|  | zero trust networking by providing every service an identity and certificates. This additionally includes rule  | ||||||
|  | based access control. | ||||||
|  |  | ||||||
| - **Dynamic Config** - Load and hot reload dynamic config from anywhere. The config interface provides a way to load application  | - **Dynamic Config** - Load and hot reload dynamic config from anywhere. The config interface provides a way to load application  | ||||||
| level config from any source such as env vars, cmdline, file, consul, vault... You can merge the sources and even define fallbacks. | level config from any source such as env vars, file, etcd. You can merge the sources and even define fallbacks. | ||||||
|  |  | ||||||
| - **Data Storage** - A simple data store interface to read, write and delete records. It includes support for memory, file and  | - **Data Storage** - A simple data store interface to read, write and delete records. It includes support for memory, file and  | ||||||
| s3. State and persistence becomes a core requirement beyond prototyping and Micro looks to build that into the framework. | CockroachDB by default. State and persistence becomes a core requirement beyond prototyping and Micro looks to build that into the framework. | ||||||
|  |  | ||||||
| - **Service Discovery** - Automatic service registration and name resolution. Service discovery is at the core of micro service  | - **Service Discovery** - Automatic service registration and name resolution. Service discovery is at the core of micro service  | ||||||
| development. When service A needs to speak to service B it needs the location of that service. | development. When service A needs to speak to service B it needs the location of that service. | ||||||
|  |  | ||||||
|  | - **Load Balancing** - Client side load balancing built on service discovery. Once we have the addresses of any number of instances  | ||||||
|  | of a service we now need a way to decide which node to route to. We use random hashed load balancing to provide even distribution  | ||||||
|  | across the services and retry a different node if there's a problem.  | ||||||
|  |  | ||||||
| - **Message Encoding** - Dynamic message encoding based on content-type. The client and server will use codecs along with content-type  | - **Message Encoding** - Dynamic message encoding based on content-type. The client and server will use codecs along with content-type  | ||||||
| to seamlessly encode and decode Go types for you. Any variety of messages could be encoded and sent from different clients. The client  | to seamlessly encode and decode Go types for you. Any variety of messages could be encoded and sent from different clients. The client  | ||||||
| and server handle this by default. | and server handle this by default. | ||||||
|  |  | ||||||
| - **Async Messaging** - Pub/Sub is built in as a first class citizen for asynchronous communication and event driven architectures. | - **Transport** - gRPC or http based request/response with support for bidirectional streaming. We provide an abstraction for synchronous communication. A request made to a service will be automatically resolved, load balanced, dialled and streamed. | ||||||
|  |  | ||||||
|  | - **Async Messaging** - PubSub is built in as a first class citizen for asynchronous communication and event driven architectures. | ||||||
| Event notifications are a core pattern in micro service development. | Event notifications are a core pattern in micro service development. | ||||||
|  |  | ||||||
| - **Synchronization** - Distributed systems are often built in an eventually consistent manner. Support for distributed locking and  | - **Synchronization** - Distributed systems are often built in an eventually consistent manner. Support for distributed locking and  | ||||||
| @@ -37,6 +43,10 @@ leadership are built in as a Sync interface. When using an eventually consistent | |||||||
| - **Pluggable Interfaces** - Micro makes use of Go interfaces for each system abstraction. Because of this these interfaces  | - **Pluggable Interfaces** - Micro makes use of Go interfaces for each system abstraction. Because of this these interfaces  | ||||||
| are pluggable and allows Micro to be runtime agnostic. | are pluggable and allows Micro to be runtime agnostic. | ||||||
|  |  | ||||||
|  | ## Getting Started | ||||||
|  |  | ||||||
|  | To be created. | ||||||
|  |  | ||||||
| ## License | ## License | ||||||
|  |  | ||||||
| Micro is Apache 2.0 licensed. | Micro is Apache 2.0 licensed. | ||||||
|   | |||||||
							
								
								
									
										15
									
								
								SECURITY.md
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										15
									
								
								SECURITY.md
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,15 @@ | |||||||
|  | # Security Policy | ||||||
|  |  | ||||||
|  | ## Supported Versions | ||||||
|  |  | ||||||
|  | Use this section to tell people about which versions of your project are | ||||||
|  | currently being supported with security updates. | ||||||
|  |  | ||||||
|  | | Version | Supported          | | ||||||
|  | | ------- | ------------------ | | ||||||
|  | | 3.7.x   | :white_check_mark: | | ||||||
|  | | < 3.7.0 | :x:                | | ||||||
|  |  | ||||||
|  | ## Reporting a Vulnerability | ||||||
|  |  | ||||||
|  | If you find any issue, please create github issue in this repo | ||||||
| @@ -6,8 +6,8 @@ import ( | |||||||
| 	"errors" | 	"errors" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/codec" | 	"go.unistack.org/micro/v3/codec" | ||||||
| 	"go.unistack.org/micro/v4/metadata" | 	"go.unistack.org/micro/v3/metadata" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // DefaultBroker default memory broker | // DefaultBroker default memory broker | ||||||
| @@ -18,10 +18,6 @@ var ( | |||||||
| 	ErrNotConnected = errors.New("broker not connected") | 	ErrNotConnected = errors.New("broker not connected") | ||||||
| 	// ErrDisconnected returns when broker disconnected | 	// ErrDisconnected returns when broker disconnected | ||||||
| 	ErrDisconnected = errors.New("broker disconnected") | 	ErrDisconnected = errors.New("broker disconnected") | ||||||
| 	// ErrInvalidMessage returns when invalid Message passed |  | ||||||
| 	ErrInvalidMessage = errors.New("invalid message") |  | ||||||
| 	// ErrInvalidHandler returns when subscriber passed to Subscribe |  | ||||||
| 	ErrInvalidHandler = errors.New("invalid handler, ony func(Message) error and func([]Message) error supported") |  | ||||||
| 	// DefaultGracefulTimeout | 	// DefaultGracefulTimeout | ||||||
| 	DefaultGracefulTimeout = 5 * time.Second | 	DefaultGracefulTimeout = 5 * time.Second | ||||||
| ) | ) | ||||||
| @@ -40,12 +36,14 @@ type Broker interface { | |||||||
| 	Connect(ctx context.Context) error | 	Connect(ctx context.Context) error | ||||||
| 	// Disconnect disconnect from broker | 	// Disconnect disconnect from broker | ||||||
| 	Disconnect(ctx context.Context) error | 	Disconnect(ctx context.Context) error | ||||||
| 	// NewMessage create new broker message to publish. |  | ||||||
| 	NewMessage(ctx context.Context, hdr metadata.Metadata, body interface{}, opts ...PublishOption) (Message, error) |  | ||||||
| 	// Publish message to broker topic | 	// Publish message to broker topic | ||||||
| 	Publish(ctx context.Context, topic string, messages ...Message) error | 	Publish(ctx context.Context, topic string, msg *Message, opts ...PublishOption) error | ||||||
| 	// Subscribe subscribes to topic message via handler | 	// Subscribe subscribes to topic message via handler | ||||||
| 	Subscribe(ctx context.Context, topic string, handler interface{}, opts ...SubscribeOption) (Subscriber, error) | 	Subscribe(ctx context.Context, topic string, h Handler, opts ...SubscribeOption) (Subscriber, error) | ||||||
|  | 	// BatchPublish messages to broker with multiple topics | ||||||
|  | 	BatchPublish(ctx context.Context, msgs []*Message, opts ...PublishOption) error | ||||||
|  | 	// BatchSubscribe subscribes to topic messages via handler | ||||||
|  | 	BatchSubscribe(ctx context.Context, topic string, h BatchHandler, opts ...SubscribeOption) (Subscriber, error) | ||||||
| 	// String type of broker | 	// String type of broker | ||||||
| 	String() string | 	String() string | ||||||
| 	// Live returns broker liveness | 	// Live returns broker liveness | ||||||
| @@ -57,27 +55,72 @@ type Broker interface { | |||||||
| } | } | ||||||
|  |  | ||||||
| type ( | type ( | ||||||
| 	FuncPublish   func(ctx context.Context, topic string, messages ...Message) error | 	FuncPublish        func(ctx context.Context, topic string, msg *Message, opts ...PublishOption) error | ||||||
| 	HookPublish        func(next FuncPublish) FuncPublish | 	HookPublish        func(next FuncPublish) FuncPublish | ||||||
| 	FuncSubscribe func(ctx context.Context, topic string, handler interface{}, opts ...SubscribeOption) (Subscriber, error) | 	FuncBatchPublish   func(ctx context.Context, msgs []*Message, opts ...PublishOption) error | ||||||
|  | 	HookBatchPublish   func(next FuncBatchPublish) FuncBatchPublish | ||||||
|  | 	FuncSubscribe      func(ctx context.Context, topic string, h Handler, opts ...SubscribeOption) (Subscriber, error) | ||||||
| 	HookSubscribe      func(next FuncSubscribe) FuncSubscribe | 	HookSubscribe      func(next FuncSubscribe) FuncSubscribe | ||||||
|  | 	FuncBatchSubscribe func(ctx context.Context, topic string, h BatchHandler, opts ...SubscribeOption) (Subscriber, error) | ||||||
|  | 	HookBatchSubscribe func(next FuncBatchSubscribe) FuncBatchSubscribe | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Message is given to a subscription handler for processing | // Handler is used to process messages via a subscription of a topic. | ||||||
| type Message interface { | type Handler func(Event) error | ||||||
| 	// Context for the message. |  | ||||||
|  | // Events contains multiple events | ||||||
|  | type Events []Event | ||||||
|  |  | ||||||
|  | // Ack try to ack all events and return | ||||||
|  | func (evs Events) Ack() error { | ||||||
|  | 	var err error | ||||||
|  | 	for _, ev := range evs { | ||||||
|  | 		if err = ev.Ack(); err != nil { | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SetError sets error on event | ||||||
|  | func (evs Events) SetError(err error) { | ||||||
|  | 	for _, ev := range evs { | ||||||
|  | 		ev.SetError(err) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // BatchHandler is used to process messages in batches via a subscription of a topic. | ||||||
|  | type BatchHandler func(Events) error | ||||||
|  |  | ||||||
|  | // Event is given to a subscription handler for processing | ||||||
|  | type Event interface { | ||||||
|  | 	// Context return context.Context for event | ||||||
| 	Context() context.Context | 	Context() context.Context | ||||||
| 	// Topic returns message destination topic. | 	// Topic returns event topic | ||||||
| 	Topic() string | 	Topic() string | ||||||
| 	// Header returns message headers. | 	// Message returns broker message | ||||||
| 	Header() metadata.Metadata | 	Message() *Message | ||||||
| 	// Body returns broker message []byte slice | 	// Ack acknowledge message | ||||||
| 	Body() []byte |  | ||||||
| 	// Unmarshal try to decode message body to dst. |  | ||||||
| 	// This is helper method that uses codec.Unmarshal. |  | ||||||
| 	Unmarshal(dst interface{}, opts ...codec.Option) error |  | ||||||
| 	// Ack acknowledge message if supported. |  | ||||||
| 	Ack() error | 	Ack() error | ||||||
|  | 	// Error returns message error (like decoding errors or some other) | ||||||
|  | 	Error() error | ||||||
|  | 	// SetError set event processing error | ||||||
|  | 	SetError(err error) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Message is used to transfer data | ||||||
|  | type Message struct { | ||||||
|  | 	// Header contains message metadata | ||||||
|  | 	Header metadata.Metadata | ||||||
|  | 	// Body contains message body | ||||||
|  | 	Body codec.RawMessage | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewMessage create broker message with topic filled | ||||||
|  | func NewMessage(topic string) *Message { | ||||||
|  | 	m := &Message{Header: metadata.New(2)} | ||||||
|  | 	m.Header.Set(metadata.HeaderTopic, topic) | ||||||
|  | 	return m | ||||||
| } | } | ||||||
|  |  | ||||||
| // Subscriber is a convenience return type for the Subscribe method | // Subscriber is a convenience return type for the Subscribe method | ||||||
|   | |||||||
| @@ -42,16 +42,6 @@ func SetSubscribeOption(k, v interface{}) SubscribeOption { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // SetPublishOption returns a function to setup a context with given value |  | ||||||
| func SetPublishOption(k, v interface{}) PublishOption { |  | ||||||
| 	return func(o *PublishOptions) { |  | ||||||
| 		if o.Context == nil { |  | ||||||
| 			o.Context = context.Background() |  | ||||||
| 		} |  | ||||||
| 		o.Context = context.WithValue(o.Context, k, v) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // SetOption returns a function to setup a context with given value | // SetOption returns a function to setup a context with given value | ||||||
| func SetOption(k, v interface{}) Option { | func SetOption(k, v interface{}) Option { | ||||||
| 	return func(o *Options) { | 	return func(o *Options) { | ||||||
| @@ -61,3 +51,13 @@ func SetOption(k, v interface{}) Option { | |||||||
| 		o.Context = context.WithValue(o.Context, k, v) | 		o.Context = context.WithValue(o.Context, k, v) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // SetPublishOption returns a function to setup a context with given value | ||||||
|  | func SetPublishOption(k, v interface{}) PublishOption { | ||||||
|  | 	return func(o *PublishOptions) { | ||||||
|  | 		if o.Context == nil { | ||||||
|  | 			o.Context = context.Background() | ||||||
|  | 		} | ||||||
|  | 		o.Context = context.WithValue(o.Context, k, v) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -49,6 +49,17 @@ func TestSetSubscribeOption(t *testing.T) { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func TestSetPublishOption(t *testing.T) { | ||||||
|  | 	type key struct{} | ||||||
|  | 	o := SetPublishOption(key{}, "test") | ||||||
|  | 	opts := &PublishOptions{} | ||||||
|  | 	o(opts) | ||||||
|  |  | ||||||
|  | 	if v, ok := opts.Context.Value(key{}).(string); !ok || v == "" { | ||||||
|  | 		t.Fatal("SetPublishOption not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| func TestSetOption(t *testing.T) { | func TestSetOption(t *testing.T) { | ||||||
| 	type key struct{} | 	type key struct{} | ||||||
| 	o := SetOption(key{}, "test") | 	o := SetOption(key{}, "test") | ||||||
|   | |||||||
| @@ -2,104 +2,66 @@ package broker | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"strings" |  | ||||||
| 	"sync" | 	"sync" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/broker" | 	"go.unistack.org/micro/v3/broker" | ||||||
| 	"go.unistack.org/micro/v4/codec" | 	"go.unistack.org/micro/v3/logger" | ||||||
| 	"go.unistack.org/micro/v4/logger" | 	"go.unistack.org/micro/v3/metadata" | ||||||
| 	"go.unistack.org/micro/v4/metadata" | 	"go.unistack.org/micro/v3/options" | ||||||
| 	"go.unistack.org/micro/v4/options" | 	maddr "go.unistack.org/micro/v3/util/addr" | ||||||
| 	maddr "go.unistack.org/micro/v4/util/addr" | 	"go.unistack.org/micro/v3/util/id" | ||||||
| 	"go.unistack.org/micro/v4/util/id" | 	mnet "go.unistack.org/micro/v3/util/net" | ||||||
| 	mnet "go.unistack.org/micro/v4/util/net" | 	"go.unistack.org/micro/v3/util/rand" | ||||||
| 	"go.unistack.org/micro/v4/util/rand" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type Broker struct { | type memoryBroker struct { | ||||||
| 	funcPublish        broker.FuncPublish | 	funcPublish        broker.FuncPublish | ||||||
|  | 	funcBatchPublish   broker.FuncBatchPublish | ||||||
| 	funcSubscribe      broker.FuncSubscribe | 	funcSubscribe      broker.FuncSubscribe | ||||||
| 	subscribers   map[string][]*Subscriber | 	funcBatchSubscribe broker.FuncBatchSubscribe | ||||||
|  | 	subscribers        map[string][]*memorySubscriber | ||||||
| 	addr               string | 	addr               string | ||||||
| 	opts               broker.Options | 	opts               broker.Options | ||||||
| 	sync.RWMutex | 	sync.RWMutex | ||||||
| 	connected bool | 	connected bool | ||||||
| } | } | ||||||
|  |  | ||||||
| type memoryMessage struct { | type memoryEvent struct { | ||||||
| 	c     codec.Codec | 	err     error | ||||||
|  | 	message interface{} | ||||||
| 	topic   string | 	topic   string | ||||||
| 	ctx   context.Context | 	opts    broker.Options | ||||||
| 	body  []byte |  | ||||||
| 	hdr   metadata.Metadata |  | ||||||
| 	opts  broker.PublishOptions |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *memoryMessage) Ack() error { | type memorySubscriber struct { | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryMessage) Body() []byte { |  | ||||||
| 	return m.body |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryMessage) Header() metadata.Metadata { |  | ||||||
| 	return m.hdr |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryMessage) Context() context.Context { |  | ||||||
| 	return m.ctx |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryMessage) Topic() string { |  | ||||||
| 	return "" |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryMessage) Unmarshal(dst interface{}, opts ...codec.Option) error { |  | ||||||
| 	return m.c.Unmarshal(m.body, dst) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type Subscriber struct { |  | ||||||
| 	ctx          context.Context | 	ctx          context.Context | ||||||
| 	exit         chan bool | 	exit         chan bool | ||||||
| 	handler interface{} | 	handler      broker.Handler | ||||||
|  | 	batchhandler broker.BatchHandler | ||||||
| 	id           string | 	id           string | ||||||
| 	topic        string | 	topic        string | ||||||
| 	opts         broker.SubscribeOptions | 	opts         broker.SubscribeOptions | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Broker) newCodec(ct string) (codec.Codec, error) { | func (m *memoryBroker) Options() broker.Options { | ||||||
| 	if idx := strings.IndexRune(ct, ';'); idx >= 0 { | 	return m.opts | ||||||
| 		ct = ct[:idx] |  | ||||||
| 	} |  | ||||||
| 	b.RLock() |  | ||||||
| 	c, ok := b.opts.Codecs[ct] |  | ||||||
| 	b.RUnlock() |  | ||||||
| 	if ok { |  | ||||||
| 		return c, nil |  | ||||||
| 	} |  | ||||||
| 	return nil, codec.ErrUnknownContentType |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Broker) Options() broker.Options { | func (m *memoryBroker) Address() string { | ||||||
| 	return b.opts | 	return m.addr | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Broker) Address() string { | func (m *memoryBroker) Connect(ctx context.Context) error { | ||||||
| 	return b.addr |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (b *Broker) Connect(ctx context.Context) error { |  | ||||||
| 	select { | 	select { | ||||||
| 	case <-ctx.Done(): | 	case <-ctx.Done(): | ||||||
| 		return ctx.Err() | 		return ctx.Err() | ||||||
| 	default: | 	default: | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	b.Lock() | 	m.Lock() | ||||||
| 	defer b.Unlock() | 	defer m.Unlock() | ||||||
|  |  | ||||||
| 	if b.connected { | 	if m.connected { | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -113,128 +75,156 @@ func (b *Broker) Connect(ctx context.Context) error { | |||||||
| 	// set addr with port | 	// set addr with port | ||||||
| 	addr = mnet.HostPort(addr, 10000+i) | 	addr = mnet.HostPort(addr, 10000+i) | ||||||
|  |  | ||||||
| 	b.addr = addr | 	m.addr = addr | ||||||
| 	b.connected = true | 	m.connected = true | ||||||
|  |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Broker) Disconnect(ctx context.Context) error { | func (m *memoryBroker) Disconnect(ctx context.Context) error { | ||||||
| 	select { | 	select { | ||||||
| 	case <-ctx.Done(): | 	case <-ctx.Done(): | ||||||
| 		return ctx.Err() | 		return ctx.Err() | ||||||
| 	default: | 	default: | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	b.Lock() | 	m.Lock() | ||||||
| 	defer b.Unlock() | 	defer m.Unlock() | ||||||
|  |  | ||||||
| 	if !b.connected { | 	if !m.connected { | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	b.connected = false | 	m.connected = false | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Broker) Init(opts ...broker.Option) error { | func (m *memoryBroker) Init(opts ...broker.Option) error { | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&b.opts) | 		o(&m.opts) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	b.funcPublish = b.fnPublish | 	m.funcPublish = m.fnPublish | ||||||
| 	b.funcSubscribe = b.fnSubscribe | 	m.funcBatchPublish = m.fnBatchPublish | ||||||
|  | 	m.funcSubscribe = m.fnSubscribe | ||||||
|  | 	m.funcBatchSubscribe = m.fnBatchSubscribe | ||||||
|  |  | ||||||
| 	b.opts.Hooks.EachPrev(func(hook options.Hook) { | 	m.opts.Hooks.EachPrev(func(hook options.Hook) { | ||||||
| 		switch h := hook.(type) { | 		switch h := hook.(type) { | ||||||
| 		case broker.HookPublish: | 		case broker.HookPublish: | ||||||
| 			b.funcPublish = h(b.funcPublish) | 			m.funcPublish = h(m.funcPublish) | ||||||
|  | 		case broker.HookBatchPublish: | ||||||
|  | 			m.funcBatchPublish = h(m.funcBatchPublish) | ||||||
| 		case broker.HookSubscribe: | 		case broker.HookSubscribe: | ||||||
| 			b.funcSubscribe = h(b.funcSubscribe) | 			m.funcSubscribe = h(m.funcSubscribe) | ||||||
|  | 		case broker.HookBatchSubscribe: | ||||||
|  | 			m.funcBatchSubscribe = h(m.funcBatchSubscribe) | ||||||
| 		} | 		} | ||||||
| 	}) | 	}) | ||||||
|  |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Broker) NewMessage(ctx context.Context, hdr metadata.Metadata, body interface{}, opts ...broker.PublishOption) (broker.Message, error) { | func (m *memoryBroker) Publish(ctx context.Context, topic string, msg *broker.Message, opts ...broker.PublishOption) error { | ||||||
| 	options := broker.NewPublishOptions(opts...) | 	return m.funcPublish(ctx, topic, msg, opts...) | ||||||
| 	m := &memoryMessage{ctx: ctx, hdr: hdr, opts: options} |  | ||||||
| 	c, err := b.newCodec(m.opts.ContentType) |  | ||||||
| 	if err == nil { |  | ||||||
| 		m.body, err = c.Marshal(body) |  | ||||||
| 	} |  | ||||||
| 	if err != nil { |  | ||||||
| 		return nil, err |  | ||||||
| } | } | ||||||
|  |  | ||||||
| 	return m, nil | func (m *memoryBroker) fnPublish(ctx context.Context, topic string, msg *broker.Message, opts ...broker.PublishOption) error { | ||||||
|  | 	msg.Header.Set(metadata.HeaderTopic, topic) | ||||||
|  | 	return m.publish(ctx, []*broker.Message{msg}, opts...) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Broker) Publish(ctx context.Context, topic string, messages ...broker.Message) error { | func (m *memoryBroker) BatchPublish(ctx context.Context, msgs []*broker.Message, opts ...broker.PublishOption) error { | ||||||
| 	return b.funcPublish(ctx, topic, messages...) | 	return m.funcBatchPublish(ctx, msgs, opts...) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Broker) fnPublish(ctx context.Context, topic string, messages ...broker.Message) error { | func (m *memoryBroker) fnBatchPublish(ctx context.Context, msgs []*broker.Message, opts ...broker.PublishOption) error { | ||||||
| 	return b.publish(ctx, topic, messages...) | 	return m.publish(ctx, msgs, opts...) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Broker) publish(ctx context.Context, topic string, messages ...broker.Message) error { | func (m *memoryBroker) publish(ctx context.Context, msgs []*broker.Message, opts ...broker.PublishOption) error { | ||||||
| 	b.RLock() | 	m.RLock() | ||||||
| 	if !b.connected { | 	if !m.connected { | ||||||
| 		b.RUnlock() | 		m.RUnlock() | ||||||
| 		return broker.ErrNotConnected | 		return broker.ErrNotConnected | ||||||
| 	} | 	} | ||||||
| 	b.RUnlock() | 	m.RUnlock() | ||||||
|  |  | ||||||
|  | 	var err error | ||||||
|  |  | ||||||
| 	select { | 	select { | ||||||
| 	case <-ctx.Done(): | 	case <-ctx.Done(): | ||||||
| 		return ctx.Err() | 		return ctx.Err() | ||||||
| 	default: | 	default: | ||||||
|  | 		options := broker.NewPublishOptions(opts...) | ||||||
|  |  | ||||||
|  | 		msgTopicMap := make(map[string]broker.Events) | ||||||
|  | 		for _, v := range msgs { | ||||||
|  | 			p := &memoryEvent{opts: m.opts} | ||||||
|  |  | ||||||
|  | 			if m.opts.Codec == nil || options.BodyOnly { | ||||||
|  | 				p.topic, _ = v.Header.Get(metadata.HeaderTopic) | ||||||
|  | 				p.message = v.Body | ||||||
|  | 			} else { | ||||||
|  | 				p.topic, _ = v.Header.Get(metadata.HeaderTopic) | ||||||
|  | 				p.message, err = m.opts.Codec.Marshal(v) | ||||||
|  | 				if err != nil { | ||||||
|  | 					return err | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			msgTopicMap[p.topic] = append(msgTopicMap[p.topic], p) | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 	b.RLock() | 		beh := m.opts.BatchErrorHandler | ||||||
| 	subs, ok := b.subscribers[topic] | 		eh := m.opts.ErrorHandler | ||||||
| 	b.RUnlock() |  | ||||||
|  | 		for t, ms := range msgTopicMap { | ||||||
|  | 			m.RLock() | ||||||
|  | 			subs, ok := m.subscribers[t] | ||||||
|  | 			m.RUnlock() | ||||||
| 			if !ok { | 			if !ok { | ||||||
| 		return nil | 				continue | ||||||
| 			} | 			} | ||||||
|  |  | ||||||
| 	var err error |  | ||||||
|  |  | ||||||
| 			for _, sub := range subs { | 			for _, sub := range subs { | ||||||
| 		switch s := sub.handler.(type) { | 				if sub.opts.BatchErrorHandler != nil { | ||||||
| 		default: | 					beh = sub.opts.BatchErrorHandler | ||||||
| 			if b.opts.Logger.V(logger.ErrorLevel) { |  | ||||||
| 				b.opts.Logger.Error(ctx, "broker  handler error", broker.ErrInvalidHandler) |  | ||||||
| 				} | 				} | ||||||
| 		case func(broker.Message) error: | 				if sub.opts.ErrorHandler != nil { | ||||||
| 			for _, message := range messages { | 					eh = sub.opts.ErrorHandler | ||||||
| 				msg, ok := message.(*memoryMessage) | 				} | ||||||
| 				if !ok { |  | ||||||
| 					if b.opts.Logger.V(logger.ErrorLevel) { | 				switch { | ||||||
| 						b.opts.Logger.Error(ctx, "broker handler error", broker.ErrInvalidMessage) | 				// batch processing | ||||||
|  | 				case sub.batchhandler != nil: | ||||||
|  | 					if err = sub.batchhandler(ms); err != nil { | ||||||
|  | 						ms.SetError(err) | ||||||
|  | 						if beh != nil { | ||||||
|  | 							_ = beh(ms) | ||||||
|  | 						} else if m.opts.Logger.V(logger.ErrorLevel) { | ||||||
|  | 							m.opts.Logger.Error(m.opts.Context, err.Error()) | ||||||
|  | 						} | ||||||
|  | 					} else if sub.opts.AutoAck { | ||||||
|  | 						if err = ms.Ack(); err != nil { | ||||||
|  | 							m.opts.Logger.Error(m.opts.Context, "broker ack error", err) | ||||||
| 						} | 						} | ||||||
| 					} | 					} | ||||||
| 				msg.topic = topic | 					// single processing | ||||||
| 				if err = s(msg); err == nil && sub.opts.AutoAck { | 				case sub.handler != nil: | ||||||
| 					err = msg.Ack() | 					for _, p := range ms { | ||||||
|  | 						if err = sub.handler(p); err != nil { | ||||||
|  | 							p.SetError(err) | ||||||
|  | 							if eh != nil { | ||||||
|  | 								_ = eh(p) | ||||||
|  | 							} else if m.opts.Logger.V(logger.ErrorLevel) { | ||||||
|  | 								m.opts.Logger.Error(m.opts.Context, "broker handler error", err) | ||||||
| 							} | 							} | ||||||
| 				if err != nil { | 						} else if sub.opts.AutoAck { | ||||||
| 					if b.opts.Logger.V(logger.ErrorLevel) { | 							if err = p.Ack(); err != nil { | ||||||
| 						b.opts.Logger.Error(ctx, "broker handler error", err) | 								m.opts.Logger.Error(m.opts.Context, "broker ack error", err) | ||||||
| 							} | 							} | ||||||
| 						} | 						} | ||||||
| 					} | 					} | ||||||
| 		case func([]broker.Message) error: |  | ||||||
| 			if err = s(messages); err == nil && sub.opts.AutoAck { |  | ||||||
| 				for _, message := range messages { |  | ||||||
| 					err = message.Ack() |  | ||||||
| 					if err != nil { |  | ||||||
| 						if b.opts.Logger.V(logger.ErrorLevel) { |  | ||||||
| 							b.opts.Logger.Error(ctx, "broker handler error", err) |  | ||||||
| 						} |  | ||||||
| 					} |  | ||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| @@ -243,21 +233,17 @@ func (b *Broker) publish(ctx context.Context, topic string, messages ...broker.M | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Broker) Subscribe(ctx context.Context, topic string, handler interface{}, opts ...broker.SubscribeOption) (broker.Subscriber, error) { | func (m *memoryBroker) BatchSubscribe(ctx context.Context, topic string, handler broker.BatchHandler, opts ...broker.SubscribeOption) (broker.Subscriber, error) { | ||||||
| 	return b.funcSubscribe(ctx, topic, handler, opts...) | 	return m.funcBatchSubscribe(ctx, topic, handler, opts...) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Broker) fnSubscribe(ctx context.Context, topic string, handler interface{}, opts ...broker.SubscribeOption) (broker.Subscriber, error) { | func (m *memoryBroker) fnBatchSubscribe(ctx context.Context, topic string, handler broker.BatchHandler, opts ...broker.SubscribeOption) (broker.Subscriber, error) { | ||||||
| 	if err := broker.IsValidHandler(handler); err != nil { | 	m.RLock() | ||||||
| 		return nil, err | 	if !m.connected { | ||||||
| 	} | 		m.RUnlock() | ||||||
|  |  | ||||||
| 	b.RLock() |  | ||||||
| 	if !b.connected { |  | ||||||
| 		b.RUnlock() |  | ||||||
| 		return nil, broker.ErrNotConnected | 		return nil, broker.ErrNotConnected | ||||||
| 	} | 	} | ||||||
| 	b.RUnlock() | 	m.RUnlock() | ||||||
|  |  | ||||||
| 	sid, err := id.New() | 	sid, err := id.New() | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| @@ -266,7 +252,56 @@ func (b *Broker) fnSubscribe(ctx context.Context, topic string, handler interfac | |||||||
|  |  | ||||||
| 	options := broker.NewSubscribeOptions(opts...) | 	options := broker.NewSubscribeOptions(opts...) | ||||||
|  |  | ||||||
| 	sub := &Subscriber{ | 	sub := &memorySubscriber{ | ||||||
|  | 		exit:         make(chan bool, 1), | ||||||
|  | 		id:           sid, | ||||||
|  | 		topic:        topic, | ||||||
|  | 		batchhandler: handler, | ||||||
|  | 		opts:         options, | ||||||
|  | 		ctx:          ctx, | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	m.Lock() | ||||||
|  | 	m.subscribers[topic] = append(m.subscribers[topic], sub) | ||||||
|  | 	m.Unlock() | ||||||
|  |  | ||||||
|  | 	go func() { | ||||||
|  | 		<-sub.exit | ||||||
|  | 		m.Lock() | ||||||
|  | 		newSubscribers := make([]*memorySubscriber, 0, len(m.subscribers)-1) | ||||||
|  | 		for _, sb := range m.subscribers[topic] { | ||||||
|  | 			if sb.id == sub.id { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			newSubscribers = append(newSubscribers, sb) | ||||||
|  | 		} | ||||||
|  | 		m.subscribers[topic] = newSubscribers | ||||||
|  | 		m.Unlock() | ||||||
|  | 	}() | ||||||
|  |  | ||||||
|  | 	return sub, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryBroker) Subscribe(ctx context.Context, topic string, handler broker.Handler, opts ...broker.SubscribeOption) (broker.Subscriber, error) { | ||||||
|  | 	return m.funcSubscribe(ctx, topic, handler, opts...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryBroker) fnSubscribe(ctx context.Context, topic string, handler broker.Handler, opts ...broker.SubscribeOption) (broker.Subscriber, error) { | ||||||
|  | 	m.RLock() | ||||||
|  | 	if !m.connected { | ||||||
|  | 		m.RUnlock() | ||||||
|  | 		return nil, broker.ErrNotConnected | ||||||
|  | 	} | ||||||
|  | 	m.RUnlock() | ||||||
|  |  | ||||||
|  | 	sid, err := id.New() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	options := broker.NewSubscribeOptions(opts...) | ||||||
|  |  | ||||||
|  | 	sub := &memorySubscriber{ | ||||||
| 		exit:    make(chan bool, 1), | 		exit:    make(chan bool, 1), | ||||||
| 		id:      sid, | 		id:      sid, | ||||||
| 		topic:   topic, | 		topic:   topic, | ||||||
| @@ -275,64 +310,102 @@ func (b *Broker) fnSubscribe(ctx context.Context, topic string, handler interfac | |||||||
| 		ctx:     ctx, | 		ctx:     ctx, | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	b.Lock() | 	m.Lock() | ||||||
| 	b.subscribers[topic] = append(b.subscribers[topic], sub) | 	m.subscribers[topic] = append(m.subscribers[topic], sub) | ||||||
| 	b.Unlock() | 	m.Unlock() | ||||||
|  |  | ||||||
| 	go func() { | 	go func() { | ||||||
| 		<-sub.exit | 		<-sub.exit | ||||||
| 		b.Lock() | 		m.Lock() | ||||||
| 		newSubscribers := make([]*Subscriber, 0, len(b.subscribers)-1) | 		newSubscribers := make([]*memorySubscriber, 0, len(m.subscribers)-1) | ||||||
| 		for _, sb := range b.subscribers[topic] { | 		for _, sb := range m.subscribers[topic] { | ||||||
| 			if sb.id == sub.id { | 			if sb.id == sub.id { | ||||||
| 				continue | 				continue | ||||||
| 			} | 			} | ||||||
| 			newSubscribers = append(newSubscribers, sb) | 			newSubscribers = append(newSubscribers, sb) | ||||||
| 		} | 		} | ||||||
| 		b.subscribers[topic] = newSubscribers | 		m.subscribers[topic] = newSubscribers | ||||||
| 		b.Unlock() | 		m.Unlock() | ||||||
| 	}() | 	}() | ||||||
|  |  | ||||||
| 	return sub, nil | 	return sub, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Broker) String() string { | func (m *memoryBroker) String() string { | ||||||
| 	return "memory" | 	return "memory" | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Broker) Name() string { | func (m *memoryBroker) Name() string { | ||||||
| 	return b.opts.Name | 	return m.opts.Name | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Broker) Live() bool { | func (m *memoryBroker) Live() bool { | ||||||
| 	return true | 	return true | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Broker) Ready() bool { | func (m *memoryBroker) Ready() bool { | ||||||
| 	return true | 	return true | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *Broker) Health() bool { | func (m *memoryBroker) Health() bool { | ||||||
| 	return true | 	return true | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *Subscriber) Options() broker.SubscribeOptions { | func (m *memoryEvent) Topic() string { | ||||||
| 	return m.opts |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *Subscriber) Topic() string { |  | ||||||
| 	return m.topic | 	return m.topic | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *Subscriber) Unsubscribe(ctx context.Context) error { | func (m *memoryEvent) Message() *broker.Message { | ||||||
|  | 	switch v := m.message.(type) { | ||||||
|  | 	case *broker.Message: | ||||||
|  | 		return v | ||||||
|  | 	case []byte: | ||||||
|  | 		msg := &broker.Message{} | ||||||
|  | 		if err := m.opts.Codec.Unmarshal(v, msg); err != nil { | ||||||
|  | 			if m.opts.Logger.V(logger.ErrorLevel) { | ||||||
|  | 				m.opts.Logger.Error(m.opts.Context, "[memory]: failed to unmarshal: %v", err) | ||||||
|  | 			} | ||||||
|  | 			return nil | ||||||
|  | 		} | ||||||
|  | 		return msg | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryEvent) Ack() error { | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryEvent) Error() error { | ||||||
|  | 	return m.err | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryEvent) SetError(err error) { | ||||||
|  | 	m.err = err | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryEvent) Context() context.Context { | ||||||
|  | 	return m.opts.Context | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memorySubscriber) Options() broker.SubscribeOptions { | ||||||
|  | 	return m.opts | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memorySubscriber) Topic() string { | ||||||
|  | 	return m.topic | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memorySubscriber) Unsubscribe(ctx context.Context) error { | ||||||
| 	m.exit <- true | 	m.exit <- true | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewBroker return new memory broker | // NewBroker return new memory broker | ||||||
| func NewBroker(opts ...broker.Option) broker.Broker { | func NewBroker(opts ...broker.Option) broker.Broker { | ||||||
| 	return &Broker{ | 	return &memoryBroker{ | ||||||
| 		opts:        broker.NewOptions(opts...), | 		opts:        broker.NewOptions(opts...), | ||||||
| 		subscribers: make(map[string][]*Subscriber), | 		subscribers: make(map[string][]*memorySubscriber), | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -5,23 +5,12 @@ import ( | |||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"testing" | 	"testing" | ||||||
|  |  | ||||||
| 	"go.uber.org/atomic" | 	"go.unistack.org/micro/v3/broker" | ||||||
| 	"go.unistack.org/micro/v4/broker" | 	"go.unistack.org/micro/v3/metadata" | ||||||
| 	"go.unistack.org/micro/v4/codec" |  | ||||||
| 	"go.unistack.org/micro/v4/metadata" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type hldr struct { | func TestMemoryBatchBroker(t *testing.T) { | ||||||
| 	c atomic.Int64 | 	b := NewBroker() | ||||||
| } |  | ||||||
|  |  | ||||||
| func (h *hldr) Handler(m broker.Message) error { |  | ||||||
| 	h.c.Add(1) |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestMemoryBroker(t *testing.T) { |  | ||||||
| 	b := NewBroker(broker.Codec("application/octet-stream", codec.NewCodec())) |  | ||||||
| 	ctx := context.Background() | 	ctx := context.Background() | ||||||
|  |  | ||||||
| 	if err := b.Init(); err != nil { | 	if err := b.Init(); err != nil { | ||||||
| @@ -33,31 +22,32 @@ func TestMemoryBroker(t *testing.T) { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	topic := "test" | 	topic := "test" | ||||||
| 	count := int64(10) | 	count := 10 | ||||||
|  |  | ||||||
| 	h := &hldr{} | 	fn := func(evts broker.Events) error { | ||||||
|  | 		return evts.Ack() | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	sub, err := b.Subscribe(ctx, topic, h.Handler) | 	sub, err := b.BatchSubscribe(ctx, topic, fn) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		t.Fatalf("Unexpected error subscribing %v", err) | 		t.Fatalf("Unexpected error subscribing %v", err) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	for i := int64(0); i < count; i++ { | 	msgs := make([]*broker.Message, 0, count) | ||||||
| 		message, err := b.NewMessage(ctx, | 	for i := 0; i < count; i++ { | ||||||
| 			metadata.Pairs( | 		message := &broker.Message{ | ||||||
| 				"foo", "bar", | 			Header: map[string]string{ | ||||||
| 				"id", fmt.Sprintf("%d", i), | 				metadata.HeaderTopic: topic, | ||||||
| 			), | 				"foo":                "bar", | ||||||
| 			[]byte(`"hello world"`), | 				"id":                 fmt.Sprintf("%d", i), | ||||||
| 			broker.PublishContentType("application/octet-stream"), | 			}, | ||||||
| 		) | 			Body: []byte(`"hello world"`), | ||||||
| 		if err != nil { | 		} | ||||||
| 			t.Fatal(err) | 		msgs = append(msgs, message) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 		if err := b.Publish(ctx, topic, message); err != nil { | 	if err := b.BatchPublish(ctx, msgs); err != nil { | ||||||
| 			t.Fatalf("Unexpected error publishing %d err: %v", i, err) | 		t.Fatalf("Unexpected error publishing %v", err) | ||||||
| 		} |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if err := sub.Unsubscribe(ctx); err != nil { | 	if err := sub.Unsubscribe(ctx); err != nil { | ||||||
| @@ -67,8 +57,58 @@ func TestMemoryBroker(t *testing.T) { | |||||||
| 	if err := b.Disconnect(ctx); err != nil { | 	if err := b.Disconnect(ctx); err != nil { | ||||||
| 		t.Fatalf("Unexpected connect error %v", err) | 		t.Fatalf("Unexpected connect error %v", err) | ||||||
| 	} | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| 	if h.c.Load() != count { | func TestMemoryBroker(t *testing.T) { | ||||||
| 		t.Fatal("invalid messages count received") | 	b := NewBroker() | ||||||
|  | 	ctx := context.Background() | ||||||
|  |  | ||||||
|  | 	if err := b.Init(); err != nil { | ||||||
|  | 		t.Fatalf("Unexpected init error %v", err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if err := b.Connect(ctx); err != nil { | ||||||
|  | 		t.Fatalf("Unexpected connect error %v", err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	topic := "test" | ||||||
|  | 	count := 10 | ||||||
|  |  | ||||||
|  | 	fn := func(_ broker.Event) error { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	sub, err := b.Subscribe(ctx, topic, fn) | ||||||
|  | 	if err != nil { | ||||||
|  | 		t.Fatalf("Unexpected error subscribing %v", err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	msgs := make([]*broker.Message, 0, count) | ||||||
|  | 	for i := 0; i < count; i++ { | ||||||
|  | 		message := &broker.Message{ | ||||||
|  | 			Header: map[string]string{ | ||||||
|  | 				metadata.HeaderTopic: topic, | ||||||
|  | 				"foo":                "bar", | ||||||
|  | 				"id":                 fmt.Sprintf("%d", i), | ||||||
|  | 			}, | ||||||
|  | 			Body: []byte(`"hello world"`), | ||||||
|  | 		} | ||||||
|  | 		msgs = append(msgs, message) | ||||||
|  |  | ||||||
|  | 		if err := b.Publish(ctx, topic, message); err != nil { | ||||||
|  | 			t.Fatalf("Unexpected error publishing %d err: %v", i, err) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if err := b.BatchPublish(ctx, msgs); err != nil { | ||||||
|  | 		t.Fatalf("Unexpected error publishing %v", err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if err := sub.Unsubscribe(ctx); err != nil { | ||||||
|  | 		t.Fatalf("Unexpected error unsubscribing from %s: %v", topic, err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if err := b.Disconnect(ctx); err != nil { | ||||||
|  | 		t.Fatalf("Unexpected connect error %v", err) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -3,37 +3,24 @@ package broker | |||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"strings" | 	"strings" | ||||||
| 	"sync" |  | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/codec" | 	"go.unistack.org/micro/v3/options" | ||||||
| 	"go.unistack.org/micro/v4/metadata" |  | ||||||
| 	"go.unistack.org/micro/v4/options" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type NoopBroker struct { | type NoopBroker struct { | ||||||
| 	funcPublish        FuncPublish | 	funcPublish        FuncPublish | ||||||
|  | 	funcBatchPublish   FuncBatchPublish | ||||||
| 	funcSubscribe      FuncSubscribe | 	funcSubscribe      FuncSubscribe | ||||||
|  | 	funcBatchSubscribe FuncBatchSubscribe | ||||||
| 	opts               Options | 	opts               Options | ||||||
| 	sync.RWMutex |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (b *NoopBroker) newCodec(ct string) (codec.Codec, error) { |  | ||||||
| 	if idx := strings.IndexRune(ct, ';'); idx >= 0 { |  | ||||||
| 		ct = ct[:idx] |  | ||||||
| 	} |  | ||||||
| 	b.RLock() |  | ||||||
| 	c, ok := b.opts.Codecs[ct] |  | ||||||
| 	b.RUnlock() |  | ||||||
| 	if ok { |  | ||||||
| 		return c, nil |  | ||||||
| 	} |  | ||||||
| 	return nil, codec.ErrUnknownContentType |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func NewBroker(opts ...Option) *NoopBroker { | func NewBroker(opts ...Option) *NoopBroker { | ||||||
| 	b := &NoopBroker{opts: NewOptions(opts...)} | 	b := &NoopBroker{opts: NewOptions(opts...)} | ||||||
| 	b.funcPublish = b.fnPublish | 	b.funcPublish = b.fnPublish | ||||||
|  | 	b.funcBatchPublish = b.fnBatchPublish | ||||||
| 	b.funcSubscribe = b.fnSubscribe | 	b.funcSubscribe = b.fnSubscribe | ||||||
|  | 	b.funcBatchSubscribe = b.fnBatchSubscribe | ||||||
|  |  | ||||||
| 	return b | 	return b | ||||||
| } | } | ||||||
| @@ -68,14 +55,20 @@ func (b *NoopBroker) Init(opts ...Option) error { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	b.funcPublish = b.fnPublish | 	b.funcPublish = b.fnPublish | ||||||
|  | 	b.funcBatchPublish = b.fnBatchPublish | ||||||
| 	b.funcSubscribe = b.fnSubscribe | 	b.funcSubscribe = b.fnSubscribe | ||||||
|  | 	b.funcBatchSubscribe = b.fnBatchSubscribe | ||||||
|  |  | ||||||
| 	b.opts.Hooks.EachPrev(func(hook options.Hook) { | 	b.opts.Hooks.EachPrev(func(hook options.Hook) { | ||||||
| 		switch h := hook.(type) { | 		switch h := hook.(type) { | ||||||
| 		case HookPublish: | 		case HookPublish: | ||||||
| 			b.funcPublish = h(b.funcPublish) | 			b.funcPublish = h(b.funcPublish) | ||||||
|  | 		case HookBatchPublish: | ||||||
|  | 			b.funcBatchPublish = h(b.funcBatchPublish) | ||||||
| 		case HookSubscribe: | 		case HookSubscribe: | ||||||
| 			b.funcSubscribe = h(b.funcSubscribe) | 			b.funcSubscribe = h(b.funcSubscribe) | ||||||
|  | 		case HookBatchSubscribe: | ||||||
|  | 			b.funcBatchSubscribe = h(b.funcBatchSubscribe) | ||||||
| 		} | 		} | ||||||
| 	}) | 	}) | ||||||
|  |  | ||||||
| @@ -94,72 +87,43 @@ func (b *NoopBroker) Address() string { | |||||||
| 	return strings.Join(b.opts.Addrs, ",") | 	return strings.Join(b.opts.Addrs, ",") | ||||||
| } | } | ||||||
|  |  | ||||||
| type noopMessage struct { | func (b *NoopBroker) fnBatchPublish(_ context.Context, _ []*Message, _ ...PublishOption) error { | ||||||
| 	c    codec.Codec |  | ||||||
| 	ctx  context.Context |  | ||||||
| 	body []byte |  | ||||||
| 	hdr  metadata.Metadata |  | ||||||
| 	opts PublishOptions |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *noopMessage) Ack() error { |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *noopMessage) Body() []byte { | func (b *NoopBroker) BatchPublish(ctx context.Context, msgs []*Message, opts ...PublishOption) error { | ||||||
| 	return m.body | 	return b.funcBatchPublish(ctx, msgs, opts...) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *noopMessage) Header() metadata.Metadata { | func (b *NoopBroker) fnPublish(_ context.Context, _ string, _ *Message, _ ...PublishOption) error { | ||||||
| 	return m.hdr |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *noopMessage) Context() context.Context { |  | ||||||
| 	return m.ctx |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *noopMessage) Topic() string { |  | ||||||
| 	return "" |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *noopMessage) Unmarshal(dst interface{}, opts ...codec.Option) error { |  | ||||||
| 	return m.c.Unmarshal(m.body, dst) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (b *NoopBroker) NewMessage(ctx context.Context, hdr metadata.Metadata, body interface{}, opts ...PublishOption) (Message, error) { |  | ||||||
| 	options := NewPublishOptions(opts...) |  | ||||||
| 	m := &noopMessage{ctx: ctx, hdr: hdr, opts: options} |  | ||||||
| 	c, err := b.newCodec(m.opts.ContentType) |  | ||||||
| 	if err == nil { |  | ||||||
| 		m.body, err = c.Marshal(body) |  | ||||||
| 	} |  | ||||||
| 	if err != nil { |  | ||||||
| 		return nil, err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return m, nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (b *NoopBroker) fnPublish(_ context.Context, _ string, _ ...Message) error { |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *NoopBroker) Publish(ctx context.Context, topic string, msg ...Message) error { | func (b *NoopBroker) Publish(ctx context.Context, topic string, msg *Message, opts ...PublishOption) error { | ||||||
| 	return b.funcPublish(ctx, topic, msg...) | 	return b.funcPublish(ctx, topic, msg, opts...) | ||||||
| } | } | ||||||
|  |  | ||||||
| type NoopSubscriber struct { | type NoopSubscriber struct { | ||||||
| 	ctx          context.Context | 	ctx          context.Context | ||||||
| 	topic        string | 	topic        string | ||||||
| 	handler interface{} | 	handler      Handler | ||||||
|  | 	batchHandler BatchHandler | ||||||
| 	opts         SubscribeOptions | 	opts         SubscribeOptions | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *NoopBroker) fnSubscribe(ctx context.Context, topic string, handler interface{}, opts ...SubscribeOption) (Subscriber, error) { | func (b *NoopBroker) fnBatchSubscribe(ctx context.Context, topic string, handler BatchHandler, opts ...SubscribeOption) (Subscriber, error) { | ||||||
|  | 	return &NoopSubscriber{ctx: ctx, topic: topic, opts: NewSubscribeOptions(opts...), batchHandler: handler}, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) BatchSubscribe(ctx context.Context, topic string, handler BatchHandler, opts ...SubscribeOption) (Subscriber, error) { | ||||||
|  | 	return b.funcBatchSubscribe(ctx, topic, handler, opts...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) fnSubscribe(ctx context.Context, topic string, handler Handler, opts ...SubscribeOption) (Subscriber, error) { | ||||||
| 	return &NoopSubscriber{ctx: ctx, topic: topic, opts: NewSubscribeOptions(opts...), handler: handler}, nil | 	return &NoopSubscriber{ctx: ctx, topic: topic, opts: NewSubscribeOptions(opts...), handler: handler}, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *NoopBroker) Subscribe(ctx context.Context, topic string, handler interface{}, opts ...SubscribeOption) (Subscriber, error) { | func (b *NoopBroker) Subscribe(ctx context.Context, topic string, handler Handler, opts ...SubscribeOption) (Subscriber, error) { | ||||||
| 	return b.funcSubscribe(ctx, topic, handler, opts...) | 	return b.funcSubscribe(ctx, topic, handler, opts...) | ||||||
| } | } | ||||||
|  |  | ||||||
|   | |||||||
| @@ -10,9 +10,9 @@ type testHook struct { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (t *testHook) Publish1(fn FuncPublish) FuncPublish { | func (t *testHook) Publish1(fn FuncPublish) FuncPublish { | ||||||
| 	return func(ctx context.Context, topic string, messages ...Message) error { | 	return func(ctx context.Context, topic string, msg *Message, opts ...PublishOption) error { | ||||||
| 		t.f = true | 		t.f = true | ||||||
| 		return fn(ctx, topic, messages...) | 		return fn(ctx, topic, msg, opts...) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|   | |||||||
| @@ -5,13 +5,13 @@ import ( | |||||||
| 	"crypto/tls" | 	"crypto/tls" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/codec" | 	"go.unistack.org/micro/v3/codec" | ||||||
| 	"go.unistack.org/micro/v4/logger" | 	"go.unistack.org/micro/v3/logger" | ||||||
| 	"go.unistack.org/micro/v4/meter" | 	"go.unistack.org/micro/v3/meter" | ||||||
| 	"go.unistack.org/micro/v4/options" | 	"go.unistack.org/micro/v3/options" | ||||||
| 	"go.unistack.org/micro/v4/register" | 	"go.unistack.org/micro/v3/register" | ||||||
| 	"go.unistack.org/micro/v4/sync" | 	"go.unistack.org/micro/v3/sync" | ||||||
| 	"go.unistack.org/micro/v4/tracer" | 	"go.unistack.org/micro/v3/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Options struct | // Options struct | ||||||
| @@ -23,8 +23,8 @@ type Options struct { | |||||||
| 	Tracer tracer.Tracer | 	Tracer tracer.Tracer | ||||||
| 	// Register can be used for clustering | 	// Register can be used for clustering | ||||||
| 	Register register.Register | 	Register register.Register | ||||||
| 	// Codecs holds the codecs for marshal/unmarshal based on content-type | 	// Codec holds the codec for marshal/unmarshal | ||||||
| 	Codecs map[string]codec.Codec | 	Codec codec.Codec | ||||||
| 	// Logger used for logging | 	// Logger used for logging | ||||||
| 	Logger logger.Logger | 	Logger logger.Logger | ||||||
| 	// Meter used for metrics | 	// Meter used for metrics | ||||||
| @@ -37,6 +37,11 @@ type Options struct { | |||||||
| 	// TLSConfig holds tls.TLSConfig options | 	// TLSConfig holds tls.TLSConfig options | ||||||
| 	TLSConfig *tls.Config | 	TLSConfig *tls.Config | ||||||
|  |  | ||||||
|  | 	// ErrorHandler used when broker can't unmarshal incoming message | ||||||
|  | 	ErrorHandler Handler | ||||||
|  | 	// BatchErrorHandler used when broker can't unmashal incoming messages | ||||||
|  | 	BatchErrorHandler BatchHandler | ||||||
|  |  | ||||||
| 	// Addrs holds the broker address | 	// Addrs holds the broker address | ||||||
| 	Addrs []string | 	Addrs []string | ||||||
| 	// Hooks can be run before broker Publish/BatchPublish and | 	// Hooks can be run before broker Publish/BatchPublish and | ||||||
| @@ -54,11 +59,10 @@ func NewOptions(opts ...Option) Options { | |||||||
| 		Logger:          logger.DefaultLogger, | 		Logger:          logger.DefaultLogger, | ||||||
| 		Context:         context.Background(), | 		Context:         context.Background(), | ||||||
| 		Meter:           meter.DefaultMeter, | 		Meter:           meter.DefaultMeter, | ||||||
| 		Codecs:          make(map[string]codec.Codec), | 		Codec:           codec.DefaultCodec, | ||||||
| 		Tracer:          tracer.DefaultTracer, | 		Tracer:          tracer.DefaultTracer, | ||||||
| 		GracefulTimeout: DefaultGracefulTimeout, | 		GracefulTimeout: DefaultGracefulTimeout, | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&options) | 		o(&options) | ||||||
| 	} | 	} | ||||||
| @@ -74,13 +78,10 @@ func Context(ctx context.Context) Option { | |||||||
|  |  | ||||||
| // PublishOptions struct | // PublishOptions struct | ||||||
| type PublishOptions struct { | type PublishOptions struct { | ||||||
| 	// ContentType for message body | 	// Context holds external options | ||||||
| 	ContentType string |  | ||||||
| 	// BodyOnly flag says the message contains raw body bytes and don't need |  | ||||||
| 	// codec Marshal method |  | ||||||
| 	BodyOnly bool |  | ||||||
| 	// Context holds custom options |  | ||||||
| 	Context context.Context | 	Context context.Context | ||||||
|  | 	// BodyOnly flag says the message contains raw body bytes | ||||||
|  | 	BodyOnly bool | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewPublishOptions creates PublishOptions struct | // NewPublishOptions creates PublishOptions struct | ||||||
| @@ -98,6 +99,10 @@ func NewPublishOptions(opts ...PublishOption) PublishOptions { | |||||||
| type SubscribeOptions struct { | type SubscribeOptions struct { | ||||||
| 	// Context holds external options | 	// Context holds external options | ||||||
| 	Context context.Context | 	Context context.Context | ||||||
|  | 	// ErrorHandler used when broker can't unmarshal incoming message | ||||||
|  | 	ErrorHandler Handler | ||||||
|  | 	// BatchErrorHandler used when broker can't unmashal incoming messages | ||||||
|  | 	BatchErrorHandler BatchHandler | ||||||
| 	// Group holds consumer group | 	// Group holds consumer group | ||||||
| 	Group string | 	Group string | ||||||
| 	// AutoAck flag specifies auto ack of incoming message when no error happens | 	// AutoAck flag specifies auto ack of incoming message when no error happens | ||||||
| @@ -116,13 +121,6 @@ type Option func(*Options) | |||||||
| // PublishOption func | // PublishOption func | ||||||
| type PublishOption func(*PublishOptions) | type PublishOption func(*PublishOptions) | ||||||
|  |  | ||||||
| // PublishContentType sets message content-type that used to Marshal |  | ||||||
| func PublishContentType(ct string) PublishOption { |  | ||||||
| 	return func(o *PublishOptions) { |  | ||||||
| 		o.ContentType = ct |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // PublishBodyOnly publish only body of the message | // PublishBodyOnly publish only body of the message | ||||||
| func PublishBodyOnly(b bool) PublishOption { | func PublishBodyOnly(b bool) PublishOption { | ||||||
| 	return func(o *PublishOptions) { | 	return func(o *PublishOptions) { | ||||||
| @@ -130,6 +128,13 @@ func PublishBodyOnly(b bool) PublishOption { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // PublishContext sets the context | ||||||
|  | func PublishContext(ctx context.Context) PublishOption { | ||||||
|  | 	return func(o *PublishOptions) { | ||||||
|  | 		o.Context = ctx | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // Addrs sets the host addresses to be used by the broker | // Addrs sets the host addresses to be used by the broker | ||||||
| func Addrs(addrs ...string) Option { | func Addrs(addrs ...string) Option { | ||||||
| 	return func(o *Options) { | 	return func(o *Options) { | ||||||
| @@ -137,10 +142,51 @@ func Addrs(addrs ...string) Option { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // Codec sets the codec used for encoding/decoding messages | // Codec sets the codec used for encoding/decoding used where | ||||||
| func Codec(ct string, c codec.Codec) Option { | // a broker does not support headers | ||||||
|  | func Codec(c codec.Codec) Option { | ||||||
| 	return func(o *Options) { | 	return func(o *Options) { | ||||||
| 		o.Codecs[ct] = c | 		o.Codec = c | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ErrorHandler will catch all broker errors that cant be handled | ||||||
|  | // in normal way, for example Codec errors | ||||||
|  | func ErrorHandler(h Handler) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.ErrorHandler = h | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // BatchErrorHandler will catch all broker errors that cant be handled | ||||||
|  | // in normal way, for example Codec errors | ||||||
|  | func BatchErrorHandler(h BatchHandler) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.BatchErrorHandler = h | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SubscribeErrorHandler will catch all broker errors that cant be handled | ||||||
|  | // in normal way, for example Codec errors | ||||||
|  | func SubscribeErrorHandler(h Handler) SubscribeOption { | ||||||
|  | 	return func(o *SubscribeOptions) { | ||||||
|  | 		o.ErrorHandler = h | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SubscribeBatchErrorHandler will catch all broker errors that cant be handled | ||||||
|  | // in normal way, for example Codec errors | ||||||
|  | func SubscribeBatchErrorHandler(h BatchHandler) SubscribeOption { | ||||||
|  | 	return func(o *SubscribeOptions) { | ||||||
|  | 		o.BatchErrorHandler = h | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Queue sets the subscribers queue | ||||||
|  | // Deprecated | ||||||
|  | func Queue(name string) SubscribeOption { | ||||||
|  | 	return func(o *SubscribeOptions) { | ||||||
|  | 		o.Group = name | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -207,6 +253,14 @@ func SubscribeContext(ctx context.Context) SubscribeOption { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // DisableAutoAck disables auto ack | ||||||
|  | // Deprecated | ||||||
|  | func DisableAutoAck() SubscribeOption { | ||||||
|  | 	return func(o *SubscribeOptions) { | ||||||
|  | 		o.AutoAck = false | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // SubscribeAutoAck contol auto acking of messages | // SubscribeAutoAck contol auto acking of messages | ||||||
| // after they have been handled. | // after they have been handled. | ||||||
| func SubscribeAutoAck(b bool) SubscribeOption { | func SubscribeAutoAck(b bool) SubscribeOption { | ||||||
|   | |||||||
| @@ -1,14 +0,0 @@ | |||||||
| package broker |  | ||||||
|  |  | ||||||
| // IsValidHandler func signature |  | ||||||
| func IsValidHandler(sub interface{}) error { |  | ||||||
| 	switch sub.(type) { |  | ||||||
| 	default: |  | ||||||
| 		return ErrInvalidHandler |  | ||||||
| 	case func(Message) error: |  | ||||||
| 		break |  | ||||||
| 	case func([]Message) error: |  | ||||||
| 		break |  | ||||||
| 	} |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
| @@ -5,7 +5,7 @@ import ( | |||||||
| 	"math" | 	"math" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/util/backoff" | 	"go.unistack.org/micro/v3/util/backoff" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // BackoffFunc is the backoff call func | // BackoffFunc is the backoff call func | ||||||
|   | |||||||
| @@ -5,8 +5,8 @@ import ( | |||||||
| 	"context" | 	"context" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/codec" | 	"go.unistack.org/micro/v3/codec" | ||||||
| 	"go.unistack.org/micro/v4/metadata" | 	"go.unistack.org/micro/v3/metadata" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| @@ -29,14 +29,18 @@ var ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
| // Client is the interface used to make requests to services. | // Client is the interface used to make requests to services. | ||||||
|  | // It supports Request/Response via Transport and Publishing via the Broker. | ||||||
| // It also supports bidirectional streaming of requests. | // It also supports bidirectional streaming of requests. | ||||||
| type Client interface { | type Client interface { | ||||||
| 	Name() string | 	Name() string | ||||||
| 	Init(opts ...Option) error | 	Init(opts ...Option) error | ||||||
| 	Options() Options | 	Options() Options | ||||||
|  | 	NewMessage(topic string, msg interface{}, opts ...MessageOption) Message | ||||||
| 	NewRequest(service string, endpoint string, req interface{}, opts ...RequestOption) Request | 	NewRequest(service string, endpoint string, req interface{}, opts ...RequestOption) Request | ||||||
| 	Call(ctx context.Context, req Request, rsp interface{}, opts ...CallOption) error | 	Call(ctx context.Context, req Request, rsp interface{}, opts ...CallOption) error | ||||||
| 	Stream(ctx context.Context, req Request, opts ...CallOption) (Stream, error) | 	Stream(ctx context.Context, req Request, opts ...CallOption) (Stream, error) | ||||||
|  | 	Publish(ctx context.Context, msg Message, opts ...PublishOption) error | ||||||
|  | 	BatchPublish(ctx context.Context, msg []Message, opts ...PublishOption) error | ||||||
| 	String() string | 	String() string | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -45,8 +49,20 @@ type ( | |||||||
| 	HookCall         func(next FuncCall) FuncCall | 	HookCall         func(next FuncCall) FuncCall | ||||||
| 	FuncStream       func(ctx context.Context, req Request, opts ...CallOption) (Stream, error) | 	FuncStream       func(ctx context.Context, req Request, opts ...CallOption) (Stream, error) | ||||||
| 	HookStream       func(next FuncStream) FuncStream | 	HookStream       func(next FuncStream) FuncStream | ||||||
|  | 	FuncPublish      func(ctx context.Context, msg Message, opts ...PublishOption) error | ||||||
|  | 	HookPublish      func(next FuncPublish) FuncPublish | ||||||
|  | 	FuncBatchPublish func(ctx context.Context, msg []Message, opts ...PublishOption) error | ||||||
|  | 	HookBatchPublish func(next FuncBatchPublish) FuncBatchPublish | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | // Message is the interface for publishing asynchronously | ||||||
|  | type Message interface { | ||||||
|  | 	Topic() string | ||||||
|  | 	Payload() interface{} | ||||||
|  | 	ContentType() string | ||||||
|  | 	Metadata() metadata.Metadata | ||||||
|  | } | ||||||
|  |  | ||||||
| // Request is the interface for a synchronous request used by Call or Stream | // Request is the interface for a synchronous request used by Call or Stream | ||||||
| type Request interface { | type Request interface { | ||||||
| 	// The service to call | 	// The service to call | ||||||
| @@ -105,5 +121,11 @@ type Option func(*Options) | |||||||
| // CallOption used by Call or Stream | // CallOption used by Call or Stream | ||||||
| type CallOption func(*CallOptions) | type CallOption func(*CallOptions) | ||||||
|  |  | ||||||
|  | // PublishOption used by Publish | ||||||
|  | type PublishOption func(*PublishOptions) | ||||||
|  |  | ||||||
|  | // MessageOption used by NewMessage | ||||||
|  | type MessageOption func(*MessageOptions) | ||||||
|  |  | ||||||
| // RequestOption used by NewRequest | // RequestOption used by NewRequest | ||||||
| type RequestOption func(*RequestOptions) | type RequestOption func(*RequestOptions) | ||||||
|   | |||||||
| @@ -32,6 +32,16 @@ func NewContext(ctx context.Context, c Client) context.Context { | |||||||
| 	return context.WithValue(ctx, clientKey{}, c) | 	return context.WithValue(ctx, clientKey{}, c) | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // SetPublishOption returns a function to setup a context with given value | ||||||
|  | func SetPublishOption(k, v interface{}) PublishOption { | ||||||
|  | 	return func(o *PublishOptions) { | ||||||
|  | 		if o.Context == nil { | ||||||
|  | 			o.Context = context.Background() | ||||||
|  | 		} | ||||||
|  | 		o.Context = context.WithValue(o.Context, k, v) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // SetCallOption returns a function to setup a context with given value | // SetCallOption returns a function to setup a context with given value | ||||||
| func SetCallOption(k, v interface{}) CallOption { | func SetCallOption(k, v interface{}) CallOption { | ||||||
| 	return func(o *CallOptions) { | 	return func(o *CallOptions) { | ||||||
|   | |||||||
| @@ -39,6 +39,17 @@ func TestNewNilContext(t *testing.T) { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func TestSetPublishOption(t *testing.T) { | ||||||
|  | 	type key struct{} | ||||||
|  | 	o := SetPublishOption(key{}, "test") | ||||||
|  | 	opts := &PublishOptions{} | ||||||
|  | 	o(opts) | ||||||
|  |  | ||||||
|  | 	if v, ok := opts.Context.Value(key{}).(string); !ok || v == "" { | ||||||
|  | 		t.Fatal("SetPublishOption not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| func TestSetCallOption(t *testing.T) { | func TestSetCallOption(t *testing.T) { | ||||||
| 	type key struct{} | 	type key struct{} | ||||||
| 	o := SetCallOption(key{}, "test") | 	o := SetCallOption(key{}, "test") | ||||||
|   | |||||||
| @@ -4,8 +4,8 @@ import ( | |||||||
| 	"context" | 	"context" | ||||||
| 	"sort" | 	"sort" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/errors" | 	"go.unistack.org/micro/v3/errors" | ||||||
| 	"go.unistack.org/micro/v4/router" | 	"go.unistack.org/micro/v3/router" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // LookupFunc is used to lookup routes for a service | // LookupFunc is used to lookup routes for a service | ||||||
|   | |||||||
							
								
								
									
										149
									
								
								client/noop.go
									
									
									
									
									
								
							
							
						
						
									
										149
									
								
								client/noop.go
									
									
									
									
									
								
							| @@ -3,16 +3,18 @@ package client | |||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"fmt" | 	"fmt" | ||||||
|  | 	"os" | ||||||
| 	"strconv" | 	"strconv" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/codec" | 	"go.unistack.org/micro/v3/broker" | ||||||
| 	"go.unistack.org/micro/v4/errors" | 	"go.unistack.org/micro/v3/codec" | ||||||
| 	"go.unistack.org/micro/v4/metadata" | 	"go.unistack.org/micro/v3/errors" | ||||||
| 	"go.unistack.org/micro/v4/options" | 	"go.unistack.org/micro/v3/metadata" | ||||||
| 	"go.unistack.org/micro/v4/selector" | 	"go.unistack.org/micro/v3/options" | ||||||
| 	"go.unistack.org/micro/v4/semconv" | 	"go.unistack.org/micro/v3/selector" | ||||||
| 	"go.unistack.org/micro/v4/tracer" | 	"go.unistack.org/micro/v3/semconv" | ||||||
|  | 	"go.unistack.org/micro/v3/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // DefaultCodecs will be used to encode/decode data | // DefaultCodecs will be used to encode/decode data | ||||||
| @@ -21,11 +23,19 @@ var DefaultCodecs = map[string]codec.Codec{ | |||||||
| } | } | ||||||
|  |  | ||||||
| type noopClient struct { | type noopClient struct { | ||||||
|  | 	funcPublish      FuncPublish | ||||||
|  | 	funcBatchPublish FuncBatchPublish | ||||||
| 	funcCall         FuncCall | 	funcCall         FuncCall | ||||||
| 	funcStream       FuncStream | 	funcStream       FuncStream | ||||||
| 	opts             Options | 	opts             Options | ||||||
| } | } | ||||||
|  |  | ||||||
|  | type noopMessage struct { | ||||||
|  | 	topic   string | ||||||
|  | 	payload interface{} | ||||||
|  | 	opts    MessageOptions | ||||||
|  | } | ||||||
|  |  | ||||||
| type noopRequest struct { | type noopRequest struct { | ||||||
| 	body        interface{} | 	body        interface{} | ||||||
| 	codec       codec.Codec | 	codec       codec.Codec | ||||||
| @@ -42,6 +52,8 @@ func NewClient(opts ...Option) Client { | |||||||
|  |  | ||||||
| 	n.funcCall = n.fnCall | 	n.funcCall = n.fnCall | ||||||
| 	n.funcStream = n.fnStream | 	n.funcStream = n.fnStream | ||||||
|  | 	n.funcPublish = n.fnPublish | ||||||
|  | 	n.funcBatchPublish = n.fnBatchPublish | ||||||
|  |  | ||||||
| 	return n | 	return n | ||||||
| } | } | ||||||
| @@ -146,6 +158,32 @@ func (n *noopStream) CloseSend() error { | |||||||
| 	return n.err | 	return n.err | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (n *noopMessage) Topic() string { | ||||||
|  | 	return n.topic | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (n *noopMessage) Payload() interface{} { | ||||||
|  | 	return n.payload | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (n *noopMessage) ContentType() string { | ||||||
|  | 	return n.opts.ContentType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (n *noopMessage) Metadata() metadata.Metadata { | ||||||
|  | 	return n.opts.Metadata | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (n *noopClient) newCodec(contentType string) (codec.Codec, error) { | ||||||
|  | 	if cf, ok := n.opts.Codecs[contentType]; ok { | ||||||
|  | 		return cf, nil | ||||||
|  | 	} | ||||||
|  | 	if cf, ok := DefaultCodecs[contentType]; ok { | ||||||
|  | 		return cf, nil | ||||||
|  | 	} | ||||||
|  | 	return nil, codec.ErrUnknownContentType | ||||||
|  | } | ||||||
|  |  | ||||||
| func (n *noopClient) Init(opts ...Option) error { | func (n *noopClient) Init(opts ...Option) error { | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&n.opts) | 		o(&n.opts) | ||||||
| @@ -153,6 +191,8 @@ func (n *noopClient) Init(opts ...Option) error { | |||||||
|  |  | ||||||
| 	n.funcCall = n.fnCall | 	n.funcCall = n.fnCall | ||||||
| 	n.funcStream = n.fnStream | 	n.funcStream = n.fnStream | ||||||
|  | 	n.funcPublish = n.fnPublish | ||||||
|  | 	n.funcBatchPublish = n.fnBatchPublish | ||||||
|  |  | ||||||
| 	n.opts.Hooks.EachPrev(func(hook options.Hook) { | 	n.opts.Hooks.EachPrev(func(hook options.Hook) { | ||||||
| 		switch h := hook.(type) { | 		switch h := hook.(type) { | ||||||
| @@ -160,6 +200,10 @@ func (n *noopClient) Init(opts ...Option) error { | |||||||
| 			n.funcCall = h(n.funcCall) | 			n.funcCall = h(n.funcCall) | ||||||
| 		case HookStream: | 		case HookStream: | ||||||
| 			n.funcStream = h(n.funcStream) | 			n.funcStream = h(n.funcStream) | ||||||
|  | 		case HookPublish: | ||||||
|  | 			n.funcPublish = h(n.funcPublish) | ||||||
|  | 		case HookBatchPublish: | ||||||
|  | 			n.funcBatchPublish = h(n.funcBatchPublish) | ||||||
| 		} | 		} | ||||||
| 	}) | 	}) | ||||||
|  |  | ||||||
| @@ -332,6 +376,11 @@ func (n *noopClient) NewRequest(service, endpoint string, _ interface{}, _ ...Re | |||||||
| 	return &noopRequest{service: service, endpoint: endpoint} | 	return &noopRequest{service: service, endpoint: endpoint} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (n *noopClient) NewMessage(topic string, msg interface{}, opts ...MessageOption) Message { | ||||||
|  | 	options := NewMessageOptions(append([]MessageOption{MessageContentType(n.opts.ContentType)}, opts...)...) | ||||||
|  | 	return &noopMessage{topic: topic, payload: msg, opts: options} | ||||||
|  | } | ||||||
|  |  | ||||||
| func (n *noopClient) Stream(ctx context.Context, req Request, opts ...CallOption) (Stream, error) { | func (n *noopClient) Stream(ctx context.Context, req Request, opts ...CallOption) (Stream, error) { | ||||||
| 	ts := time.Now() | 	ts := time.Now() | ||||||
| 	n.opts.Meter.Counter(semconv.ClientRequestInflight, "endpoint", req.Endpoint()).Inc() | 	n.opts.Meter.Counter(semconv.ClientRequestInflight, "endpoint", req.Endpoint()).Inc() | ||||||
| @@ -500,3 +549,89 @@ func (n *noopClient) fnStream(ctx context.Context, req Request, opts ...CallOpti | |||||||
| func (n *noopClient) stream(ctx context.Context, _ string, _ Request, _ CallOptions) (Stream, error) { | func (n *noopClient) stream(ctx context.Context, _ string, _ Request, _ CallOptions) (Stream, error) { | ||||||
| 	return &noopStream{ctx: ctx}, nil | 	return &noopStream{ctx: ctx}, nil | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (n *noopClient) BatchPublish(ctx context.Context, ps []Message, opts ...PublishOption) error { | ||||||
|  | 	return n.funcBatchPublish(ctx, ps, opts...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (n *noopClient) fnBatchPublish(ctx context.Context, ps []Message, opts ...PublishOption) error { | ||||||
|  | 	return n.publish(ctx, ps, opts...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (n *noopClient) Publish(ctx context.Context, p Message, opts ...PublishOption) error { | ||||||
|  | 	return n.funcPublish(ctx, p, opts...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (n *noopClient) fnPublish(ctx context.Context, p Message, opts ...PublishOption) error { | ||||||
|  | 	return n.publish(ctx, []Message{p}, opts...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (n *noopClient) publish(ctx context.Context, ps []Message, opts ...PublishOption) error { | ||||||
|  | 	options := NewPublishOptions(opts...) | ||||||
|  |  | ||||||
|  | 	msgs := make([]*broker.Message, 0, len(ps)) | ||||||
|  |  | ||||||
|  | 	// get proxy | ||||||
|  | 	exchange := "" | ||||||
|  | 	if v, ok := os.LookupEnv("MICRO_PROXY"); ok { | ||||||
|  | 		exchange = v | ||||||
|  | 	} | ||||||
|  | 	// get the exchange | ||||||
|  | 	if len(options.Exchange) > 0 { | ||||||
|  | 		exchange = options.Exchange | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	omd, ok := metadata.FromOutgoingContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		omd = metadata.New(0) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for _, p := range ps { | ||||||
|  | 		md := metadata.Copy(omd) | ||||||
|  | 		md[metadata.HeaderContentType] = p.ContentType() | ||||||
|  | 		topic := p.Topic() | ||||||
|  | 		if len(exchange) > 0 { | ||||||
|  | 			topic = exchange | ||||||
|  | 		} | ||||||
|  | 		md[metadata.HeaderTopic] = topic | ||||||
|  | 		iter := p.Metadata().Iterator() | ||||||
|  | 		var k, v string | ||||||
|  | 		for iter.Next(&k, &v) { | ||||||
|  | 			md.Set(k, v) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		var body []byte | ||||||
|  |  | ||||||
|  | 		// passed in raw data | ||||||
|  | 		if d, ok := p.Payload().(*codec.Frame); ok { | ||||||
|  | 			body = d.Data | ||||||
|  | 		} else { | ||||||
|  | 			// use codec for payload | ||||||
|  | 			cf, err := n.newCodec(p.ContentType()) | ||||||
|  | 			if err != nil { | ||||||
|  | 				return errors.InternalServerError("go.micro.client", "%s", err) | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			// set the body | ||||||
|  | 			b, err := cf.Marshal(p.Payload()) | ||||||
|  | 			if err != nil { | ||||||
|  | 				return errors.InternalServerError("go.micro.client", "%s", err) | ||||||
|  | 			} | ||||||
|  | 			body = b | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		msgs = append(msgs, &broker.Message{Header: md, Body: body}) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if len(msgs) == 1 { | ||||||
|  | 		return n.opts.Broker.Publish(ctx, msgs[0].Header[metadata.HeaderTopic], msgs[0], | ||||||
|  | 			broker.PublishContext(options.Context), | ||||||
|  | 			broker.PublishBodyOnly(options.BodyOnly), | ||||||
|  | 		) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return n.opts.Broker.BatchPublish(ctx, msgs, | ||||||
|  | 		broker.PublishContext(options.Context), | ||||||
|  | 		broker.PublishBodyOnly(options.BodyOnly), | ||||||
|  | 	) | ||||||
|  | } | ||||||
|   | |||||||
							
								
								
									
										35
									
								
								client/noop_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										35
									
								
								client/noop_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,35 @@ | |||||||
|  | package client | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"testing" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type testHook struct { | ||||||
|  | 	f bool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *testHook) Publish(fn FuncPublish) FuncPublish { | ||||||
|  | 	return func(ctx context.Context, msg Message, opts ...PublishOption) error { | ||||||
|  | 		t.f = true | ||||||
|  | 		return fn(ctx, msg, opts...) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestNoopHook(t *testing.T) { | ||||||
|  | 	h := &testHook{} | ||||||
|  |  | ||||||
|  | 	c := NewClient(Hooks(HookPublish(h.Publish))) | ||||||
|  |  | ||||||
|  | 	if err := c.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if err := c.Publish(context.TODO(), c.NewMessage("", nil, MessageContentType("application/octet-stream"))); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !h.f { | ||||||
|  | 		t.Fatal("hook not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -6,17 +6,17 @@ import ( | |||||||
| 	"net" | 	"net" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/broker" | 	"go.unistack.org/micro/v3/broker" | ||||||
| 	"go.unistack.org/micro/v4/codec" | 	"go.unistack.org/micro/v3/codec" | ||||||
| 	"go.unistack.org/micro/v4/logger" | 	"go.unistack.org/micro/v3/logger" | ||||||
| 	"go.unistack.org/micro/v4/metadata" | 	"go.unistack.org/micro/v3/metadata" | ||||||
| 	"go.unistack.org/micro/v4/meter" | 	"go.unistack.org/micro/v3/meter" | ||||||
| 	"go.unistack.org/micro/v4/options" | 	"go.unistack.org/micro/v3/options" | ||||||
| 	"go.unistack.org/micro/v4/register" | 	"go.unistack.org/micro/v3/register" | ||||||
| 	"go.unistack.org/micro/v4/router" | 	"go.unistack.org/micro/v3/router" | ||||||
| 	"go.unistack.org/micro/v4/selector" | 	"go.unistack.org/micro/v3/selector" | ||||||
| 	"go.unistack.org/micro/v4/selector/random" | 	"go.unistack.org/micro/v3/selector/random" | ||||||
| 	"go.unistack.org/micro/v4/tracer" | 	"go.unistack.org/micro/v3/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Options holds client options | // Options holds client options | ||||||
| @@ -137,6 +137,43 @@ func Context(ctx context.Context) Option { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // NewPublishOptions create new PublishOptions struct from option | ||||||
|  | func NewPublishOptions(opts ...PublishOption) PublishOptions { | ||||||
|  | 	options := PublishOptions{} | ||||||
|  | 	for _, o := range opts { | ||||||
|  | 		o(&options) | ||||||
|  | 	} | ||||||
|  | 	return options | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // PublishOptions holds publish options | ||||||
|  | type PublishOptions struct { | ||||||
|  | 	// Context used for external options | ||||||
|  | 	Context context.Context | ||||||
|  | 	// Exchange topic exchange name | ||||||
|  | 	Exchange string | ||||||
|  | 	// BodyOnly will publish only message body | ||||||
|  | 	BodyOnly bool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewMessageOptions creates message options struct | ||||||
|  | func NewMessageOptions(opts ...MessageOption) MessageOptions { | ||||||
|  | 	options := MessageOptions{Metadata: metadata.New(1)} | ||||||
|  | 	for _, o := range opts { | ||||||
|  | 		o(&options) | ||||||
|  | 	} | ||||||
|  | 	return options | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MessageOptions holds client message options | ||||||
|  | type MessageOptions struct { | ||||||
|  | 	// Metadata additional metadata | ||||||
|  | 	Metadata metadata.Metadata | ||||||
|  | 	// ContentType specify content-type of message | ||||||
|  | 	// deprecated | ||||||
|  | 	ContentType string | ||||||
|  | } | ||||||
|  |  | ||||||
| // NewRequestOptions creates new RequestOptions struct | // NewRequestOptions creates new RequestOptions struct | ||||||
| func NewRequestOptions(opts ...RequestOption) RequestOptions { | func NewRequestOptions(opts ...RequestOption) RequestOptions { | ||||||
| 	options := RequestOptions{} | 	options := RequestOptions{} | ||||||
| @@ -337,6 +374,43 @@ func DialTimeout(d time.Duration) Option { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // WithExchange sets the exchange to route a message through | ||||||
|  | // Deprecated | ||||||
|  | func WithExchange(e string) PublishOption { | ||||||
|  | 	return func(o *PublishOptions) { | ||||||
|  | 		o.Exchange = e | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // PublishExchange sets the exchange to route a message through | ||||||
|  | func PublishExchange(e string) PublishOption { | ||||||
|  | 	return func(o *PublishOptions) { | ||||||
|  | 		o.Exchange = e | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithBodyOnly publish only message body | ||||||
|  | // DERECATED | ||||||
|  | func WithBodyOnly(b bool) PublishOption { | ||||||
|  | 	return func(o *PublishOptions) { | ||||||
|  | 		o.BodyOnly = b | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // PublishBodyOnly publish only message body | ||||||
|  | func PublishBodyOnly(b bool) PublishOption { | ||||||
|  | 	return func(o *PublishOptions) { | ||||||
|  | 		o.BodyOnly = b | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // PublishContext sets the context in publish options | ||||||
|  | func PublishContext(ctx context.Context) PublishOption { | ||||||
|  | 	return func(o *PublishOptions) { | ||||||
|  | 		o.Context = ctx | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // WithContextDialer pass ContextDialer to client call | // WithContextDialer pass ContextDialer to client call | ||||||
| func WithContextDialer(fn func(context.Context, string) (net.Conn, error)) CallOption { | func WithContextDialer(fn func(context.Context, string) (net.Conn, error)) CallOption { | ||||||
| 	return func(o *CallOptions) { | 	return func(o *CallOptions) { | ||||||
| @@ -448,6 +522,30 @@ func WithSelectOptions(sops ...selector.SelectOption) CallOption { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // WithMessageContentType sets the message content type | ||||||
|  | // Deprecated | ||||||
|  | func WithMessageContentType(ct string) MessageOption { | ||||||
|  | 	return func(o *MessageOptions) { | ||||||
|  | 		o.Metadata.Set(metadata.HeaderContentType, ct) | ||||||
|  | 		o.ContentType = ct | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MessageContentType sets the message content type | ||||||
|  | func MessageContentType(ct string) MessageOption { | ||||||
|  | 	return func(o *MessageOptions) { | ||||||
|  | 		o.Metadata.Set(metadata.HeaderContentType, ct) | ||||||
|  | 		o.ContentType = ct | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MessageMetadata sets the message metadata | ||||||
|  | func MessageMetadata(k, v string) MessageOption { | ||||||
|  | 	return func(o *MessageOptions) { | ||||||
|  | 		o.Metadata.Set(k, v) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // StreamingRequest specifies that request is streaming | // StreamingRequest specifies that request is streaming | ||||||
| func StreamingRequest(b bool) RequestOption { | func StreamingRequest(b bool) RequestOption { | ||||||
| 	return func(o *RequestOptions) { | 	return func(o *RequestOptions) { | ||||||
|   | |||||||
| @@ -3,7 +3,7 @@ package client | |||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/errors" | 	"go.unistack.org/micro/v3/errors" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // RetryFunc that returning either false or a non-nil error will result in the call not being retried | // RetryFunc that returning either false or a non-nil error will result in the call not being retried | ||||||
|   | |||||||
| @@ -5,7 +5,7 @@ import ( | |||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"testing" | 	"testing" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/errors" | 	"go.unistack.org/micro/v3/errors" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| func TestRetryAlways(t *testing.T) { | func TestRetryAlways(t *testing.T) { | ||||||
|   | |||||||
| @@ -1,7 +1,7 @@ | |||||||
| package client | package client | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"go.unistack.org/micro/v4/codec" | 	"go.unistack.org/micro/v3/codec" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type testRequest struct { | type testRequest struct { | ||||||
|   | |||||||
| @@ -3,7 +3,7 @@ package cluster | |||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/metadata" | 	"go.unistack.org/micro/v3/metadata" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Message sent to member in cluster | // Message sent to member in cluster | ||||||
|   | |||||||
| @@ -3,8 +3,6 @@ package codec | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"errors" | 	"errors" | ||||||
|  |  | ||||||
| 	"gopkg.in/yaml.v3" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| @@ -56,22 +54,3 @@ func (m *RawMessage) UnmarshalJSON(data []byte) error { | |||||||
| 	*m = append((*m)[0:0], data...) | 	*m = append((*m)[0:0], data...) | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| // MarshalYAML returns m as the JSON encoding of m. |  | ||||||
| func (m *RawMessage) MarshalYAML() ([]byte, error) { |  | ||||||
| 	if m == nil { |  | ||||||
| 		return []byte("null"), nil |  | ||||||
| 	} else if len(*m) == 0 { |  | ||||||
| 		return []byte("null"), nil |  | ||||||
| 	} |  | ||||||
| 	return *m, nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // UnmarshalYAML sets *m to a copy of data. |  | ||||||
| func (m *RawMessage) UnmarshalYAML(n *yaml.Node) error { |  | ||||||
| 	if m == nil { |  | ||||||
| 		return errors.New("RawMessage UnmarshalYAML on nil pointer") |  | ||||||
| 	} |  | ||||||
| 	*m = append((*m)[0:0], []byte(n.Value)...) |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|   | |||||||
| @@ -1,7 +1,5 @@ | |||||||
| package codec | package codec | ||||||
|  |  | ||||||
| import "gopkg.in/yaml.v3" |  | ||||||
|  |  | ||||||
| // Frame gives us the ability to define raw data to send over the pipes | // Frame gives us the ability to define raw data to send over the pipes | ||||||
| type Frame struct { | type Frame struct { | ||||||
| 	Data []byte | 	Data []byte | ||||||
| @@ -22,17 +20,6 @@ func (m *Frame) UnmarshalJSON(data []byte) error { | |||||||
| 	return m.Unmarshal(data) | 	return m.Unmarshal(data) | ||||||
| } | } | ||||||
|  |  | ||||||
| // MarshalYAML returns frame data |  | ||||||
| func (m *Frame) MarshalYAML() ([]byte, error) { |  | ||||||
| 	return m.Marshal() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // UnmarshalYAML set frame data |  | ||||||
| func (m *Frame) UnmarshalYAML(n *yaml.Node) error { |  | ||||||
| 	m.Data = []byte(n.Value) |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // ProtoMessage noop func | // ProtoMessage noop func | ||||||
| func (m *Frame) ProtoMessage() {} | func (m *Frame) ProtoMessage() {} | ||||||
|  |  | ||||||
|   | |||||||
| @@ -17,7 +17,7 @@ syntax = "proto3"; | |||||||
| package micro.codec; | package micro.codec; | ||||||
|  |  | ||||||
| option cc_enable_arenas = true; | option cc_enable_arenas = true; | ||||||
| option go_package = "go.unistack.org/micro/v4/codec;codec"; | option go_package = "go.unistack.org/micro/v3/codec;codec"; | ||||||
| option java_multiple_files = true; | option java_multiple_files = true; | ||||||
| option java_outer_classname = "MicroCodec"; | option java_outer_classname = "MicroCodec"; | ||||||
| option java_package = "micro.codec"; | option java_package = "micro.codec"; | ||||||
|   | |||||||
| @@ -3,7 +3,7 @@ package codec | |||||||
| import ( | import ( | ||||||
| 	"encoding/json" | 	"encoding/json" | ||||||
|  |  | ||||||
| 	codecpb "go.unistack.org/micro-proto/v4/codec" | 	codecpb "go.unistack.org/micro-proto/v3/codec" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type noopCodec struct { | type noopCodec struct { | ||||||
|   | |||||||
| @@ -3,9 +3,9 @@ package codec | |||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/logger" | 	"go.unistack.org/micro/v3/logger" | ||||||
| 	"go.unistack.org/micro/v4/meter" | 	"go.unistack.org/micro/v3/meter" | ||||||
| 	"go.unistack.org/micro/v4/tracer" | 	"go.unistack.org/micro/v3/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Option func | // Option func | ||||||
|   | |||||||
| @@ -9,10 +9,10 @@ import ( | |||||||
|  |  | ||||||
| 	"dario.cat/mergo" | 	"dario.cat/mergo" | ||||||
| 	"github.com/google/uuid" | 	"github.com/google/uuid" | ||||||
| 	"go.unistack.org/micro/v4/options" | 	"go.unistack.org/micro/v3/options" | ||||||
| 	mid "go.unistack.org/micro/v4/util/id" | 	mid "go.unistack.org/micro/v3/util/id" | ||||||
| 	rutil "go.unistack.org/micro/v4/util/reflect" | 	rutil "go.unistack.org/micro/v3/util/reflect" | ||||||
| 	mtime "go.unistack.org/micro/v4/util/time" | 	mtime "go.unistack.org/micro/v3/util/time" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type defaultConfig struct { | type defaultConfig struct { | ||||||
| @@ -210,7 +210,7 @@ func fillValue(value reflect.Value, val string) error { | |||||||
| 				return err | 				return err | ||||||
| 			} | 			} | ||||||
| 			value.Set(reflect.ValueOf(v)) | 			value.Set(reflect.ValueOf(v)) | ||||||
| 		case value.Type().String() == "time.Duration" && value.Type().PkgPath() == "go.unistack.org/micro/v4/util/time": | 		case value.Type().String() == "time.Duration" && value.Type().PkgPath() == "go.unistack.org/micro/v3/util/time": | ||||||
| 			v, err := mtime.ParseDuration(val) | 			v, err := mtime.ParseDuration(val) | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| 				return err | 				return err | ||||||
|   | |||||||
| @@ -7,8 +7,8 @@ import ( | |||||||
| 	"testing" | 	"testing" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/config" | 	"go.unistack.org/micro/v3/config" | ||||||
| 	mtime "go.unistack.org/micro/v4/util/time" | 	mtime "go.unistack.org/micro/v3/util/time" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type cfg struct { | type cfg struct { | ||||||
|   | |||||||
| @@ -4,11 +4,11 @@ import ( | |||||||
| 	"context" | 	"context" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/codec" | 	"go.unistack.org/micro/v3/codec" | ||||||
| 	"go.unistack.org/micro/v4/logger" | 	"go.unistack.org/micro/v3/logger" | ||||||
| 	"go.unistack.org/micro/v4/meter" | 	"go.unistack.org/micro/v3/meter" | ||||||
| 	"go.unistack.org/micro/v4/options" | 	"go.unistack.org/micro/v3/options" | ||||||
| 	"go.unistack.org/micro/v4/tracer" | 	"go.unistack.org/micro/v3/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Options hold the config options | // Options hold the config options | ||||||
|   | |||||||
| @@ -17,7 +17,7 @@ syntax = "proto3"; | |||||||
| package micro.errors; | package micro.errors; | ||||||
|  |  | ||||||
| option cc_enable_arenas = true; | option cc_enable_arenas = true; | ||||||
| option go_package = "go.unistack.org/micro/v4/errors;errors"; | option go_package = "go.unistack.org/micro/v3/errors;errors"; | ||||||
| option java_multiple_files = true; | option java_multiple_files = true; | ||||||
| option java_outer_classname = "MicroErrors"; | option java_outer_classname = "MicroErrors"; | ||||||
| option java_package = "micro.errors"; | option java_package = "micro.errors"; | ||||||
|   | |||||||
							
								
								
									
										27
									
								
								event.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										27
									
								
								event.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,27 @@ | |||||||
|  | package micro | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v3/client" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Event is used to publish messages to a topic | ||||||
|  | type Event interface { | ||||||
|  | 	// Publish publishes a message to the event topic | ||||||
|  | 	Publish(ctx context.Context, msg interface{}, opts ...client.PublishOption) error | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type event struct { | ||||||
|  | 	c     client.Client | ||||||
|  | 	topic string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewEvent creates a new event publisher | ||||||
|  | func NewEvent(topic string, c client.Client) Event { | ||||||
|  | 	return &event{c, topic} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (e *event) Publish(ctx context.Context, msg interface{}, opts ...client.PublishOption) error { | ||||||
|  | 	return e.c.Publish(ctx, e.c.NewMessage(e.topic, msg), opts...) | ||||||
|  | } | ||||||
| @@ -15,6 +15,15 @@ func FromContext(ctx context.Context) (Flow, bool) { | |||||||
| 	return c, ok | 	return c, ok | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // MustContext returns Flow from context | ||||||
|  | func MustContext(ctx context.Context) Flow { | ||||||
|  | 	f, ok := FromContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		panic("missing flow") | ||||||
|  | 	} | ||||||
|  | 	return f | ||||||
|  | } | ||||||
|  |  | ||||||
| // NewContext stores Flow to context | // NewContext stores Flow to context | ||||||
| func NewContext(ctx context.Context, f Flow) context.Context { | func NewContext(ctx context.Context, f Flow) context.Context { | ||||||
| 	if ctx == nil { | 	if ctx == nil { | ||||||
|   | |||||||
| @@ -1,5 +1,3 @@ | |||||||
| //go:build ignore |  | ||||||
|  |  | ||||||
| package flow | package flow | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|   | |||||||
							
								
								
									
										342
									
								
								flow/default.go
									
									
									
									
									
								
							
							
						
						
									
										342
									
								
								flow/default.go
									
									
									
									
									
								
							| @@ -1,19 +1,17 @@ | |||||||
| //go:build ignore |  | ||||||
|  |  | ||||||
| package flow | package flow | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"path/filepath" |  | ||||||
| 	"sync" | 	"sync" | ||||||
|  |  | ||||||
| 	"github.com/heimdalr/dag" | 	"github.com/silas/dag" | ||||||
| 	"go.unistack.org/micro/v4/client" | 	"go.unistack.org/micro/v3/client" | ||||||
| 	"go.unistack.org/micro/v4/codec" | 	"go.unistack.org/micro/v3/codec" | ||||||
| 	"go.unistack.org/micro/v4/metadata" | 	"go.unistack.org/micro/v3/logger" | ||||||
| 	"go.unistack.org/micro/v4/store" | 	"go.unistack.org/micro/v3/metadata" | ||||||
| 	"go.unistack.org/micro/v4/util/id" | 	"go.unistack.org/micro/v3/store" | ||||||
|  | 	"go.unistack.org/micro/v3/util/id" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type microFlow struct { | type microFlow struct { | ||||||
| @@ -22,7 +20,7 @@ type microFlow struct { | |||||||
|  |  | ||||||
| type microWorkflow struct { | type microWorkflow struct { | ||||||
| 	opts   Options | 	opts   Options | ||||||
| 	g      *dag.DAG | 	g      *dag.AcyclicGraph | ||||||
| 	steps  map[string]Step | 	steps  map[string]Step | ||||||
| 	id     string | 	id     string | ||||||
| 	status Status | 	status Status | ||||||
| @@ -34,20 +32,20 @@ func (w *microWorkflow) ID() string { | |||||||
| 	return w.id | 	return w.id | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (w *microWorkflow) Steps() ([][]Step, error) { | ||||||
|  | 	return w.getSteps("", false) | ||||||
|  | } | ||||||
|  |  | ||||||
| func (w *microWorkflow) Status() Status { | func (w *microWorkflow) Status() Status { | ||||||
| 	return w.status | 	return w.status | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *microWorkflow) AppendSteps(steps ...Step) error { | func (w *microWorkflow) AppendSteps(steps ...Step) error { | ||||||
| 	var err error |  | ||||||
| 	w.Lock() | 	w.Lock() | ||||||
| 	defer w.Unlock() |  | ||||||
|  |  | ||||||
| 	for _, s := range steps { | 	for _, s := range steps { | ||||||
| 		w.steps[s.String()] = s | 		w.steps[s.String()] = s | ||||||
| 		if _, err = w.g.AddVertex(s); err != nil { | 		w.g.Add(s) | ||||||
| 			return err |  | ||||||
| 		} |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	for _, dst := range steps { | 	for _, dst := range steps { | ||||||
| @@ -56,13 +54,18 @@ func (w *microWorkflow) AppendSteps(steps ...Step) error { | |||||||
| 			if !ok { | 			if !ok { | ||||||
| 				return ErrStepNotExists | 				return ErrStepNotExists | ||||||
| 			} | 			} | ||||||
| 			if err = w.g.AddEdge(src.String(), dst.String()); err != nil { | 			w.g.Connect(dag.BasicEdge(src, dst)) | ||||||
| 				return err |  | ||||||
| 			} |  | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	w.g.ReduceTransitively() | 	if err := w.g.Validate(); err != nil { | ||||||
|  | 		w.Unlock() | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	w.g.TransitiveReduction() | ||||||
|  |  | ||||||
|  | 	w.Unlock() | ||||||
|  |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
| @@ -71,11 +74,10 @@ func (w *microWorkflow) RemoveSteps(steps ...Step) error { | |||||||
| 	// TODO: handle case when some step requires or required by removed step | 	// TODO: handle case when some step requires or required by removed step | ||||||
|  |  | ||||||
| 	w.Lock() | 	w.Lock() | ||||||
| 	defer w.Unlock() |  | ||||||
|  |  | ||||||
| 	for _, s := range steps { | 	for _, s := range steps { | ||||||
| 		delete(w.steps, s.String()) | 		delete(w.steps, s.String()) | ||||||
| 		w.g.DeleteVertex(s.String()) | 		w.g.Remove(s) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	for _, dst := range steps { | 	for _, dst := range steps { | ||||||
| @@ -84,34 +86,91 @@ func (w *microWorkflow) RemoveSteps(steps ...Step) error { | |||||||
| 			if !ok { | 			if !ok { | ||||||
| 				return ErrStepNotExists | 				return ErrStepNotExists | ||||||
| 			} | 			} | ||||||
| 			w.g.AddEdge(src.String(), dst.String()) | 			w.g.Connect(dag.BasicEdge(src, dst)) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	w.g.ReduceTransitively() | 	if err := w.g.Validate(); err != nil { | ||||||
|  | 		w.Unlock() | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	w.g.TransitiveReduction() | ||||||
|  |  | ||||||
|  | 	w.Unlock() | ||||||
|  |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (w *microWorkflow) getSteps(start string, reverse bool) ([][]Step, error) { | ||||||
|  | 	var steps [][]Step | ||||||
|  | 	var root dag.Vertex | ||||||
|  | 	var err error | ||||||
|  |  | ||||||
|  | 	fn := func(n dag.Vertex, idx int) error { | ||||||
|  | 		if idx == 0 { | ||||||
|  | 			steps = make([][]Step, 1) | ||||||
|  | 			steps[0] = make([]Step, 0, 1) | ||||||
|  | 		} else if idx >= len(steps) { | ||||||
|  | 			tsteps := make([][]Step, idx+1) | ||||||
|  | 			copy(tsteps, steps) | ||||||
|  | 			steps = tsteps | ||||||
|  | 			steps[idx] = make([]Step, 0, 1) | ||||||
|  | 		} | ||||||
|  | 		steps[idx] = append(steps[idx], n.(Step)) | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if start != "" { | ||||||
|  | 		var ok bool | ||||||
|  | 		w.RLock() | ||||||
|  | 		root, ok = w.steps[start] | ||||||
|  | 		w.RUnlock() | ||||||
|  | 		if !ok { | ||||||
|  | 			return nil, ErrStepNotExists | ||||||
|  | 		} | ||||||
|  | 	} else { | ||||||
|  | 		root, err = w.g.Root() | ||||||
|  | 		if err != nil { | ||||||
|  | 			return nil, err | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if reverse { | ||||||
|  | 		err = w.g.SortedReverseDepthFirstWalk([]dag.Vertex{root}, fn) | ||||||
|  | 	} else { | ||||||
|  | 		err = w.g.SortedDepthFirstWalk([]dag.Vertex{root}, fn) | ||||||
|  | 	} | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return steps, nil | ||||||
|  | } | ||||||
|  |  | ||||||
| func (w *microWorkflow) Abort(ctx context.Context, id string) error { | func (w *microWorkflow) Abort(ctx context.Context, id string) error { | ||||||
| 	workflowStore := store.NewNamespaceStore(w.opts.Store, filepath.Join("workflows", id)) | 	workflowStore := store.NewNamespaceStore(w.opts.Store, "workflows"+w.opts.Store.Options().Separator+id) | ||||||
| 	return workflowStore.Write(ctx, "status", &codec.Frame{Data: []byte(StatusAborted.String())}) | 	return workflowStore.Write(ctx, "status", &codec.Frame{Data: []byte(StatusAborted.String())}) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *microWorkflow) Suspend(ctx context.Context, id string) error { | func (w *microWorkflow) Suspend(ctx context.Context, id string) error { | ||||||
| 	workflowStore := store.NewNamespaceStore(w.opts.Store, filepath.Join("workflows", id)) | 	workflowStore := store.NewNamespaceStore(w.opts.Store, "workflows"+w.opts.Store.Options().Separator+id) | ||||||
| 	return workflowStore.Write(ctx, "status", &codec.Frame{Data: []byte(StatusSuspend.String())}) | 	return workflowStore.Write(ctx, "status", &codec.Frame{Data: []byte(StatusSuspend.String())}) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *microWorkflow) Resume(ctx context.Context, id string) error { | func (w *microWorkflow) Resume(ctx context.Context, id string) error { | ||||||
| 	workflowStore := store.NewNamespaceStore(w.opts.Store, filepath.Join("workflows", id)) | 	workflowStore := store.NewNamespaceStore(w.opts.Store, "workflows"+w.opts.Store.Options().Separator+id) | ||||||
| 	return workflowStore.Write(ctx, "status", &codec.Frame{Data: []byte(StatusRunning.String())}) | 	return workflowStore.Write(ctx, "status", &codec.Frame{Data: []byte(StatusRunning.String())}) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...ExecuteOption) (string, error) { | func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...ExecuteOption) (string, error) { | ||||||
| 	w.Lock() | 	w.Lock() | ||||||
| 	if !w.init { | 	if !w.init { | ||||||
| 		w.g.ReduceTransitively() | 		if err := w.g.Validate(); err != nil { | ||||||
|  | 			w.Unlock() | ||||||
|  | 			return "", err | ||||||
|  | 		} | ||||||
|  | 		w.g.TransitiveReduction() | ||||||
| 		w.init = true | 		w.init = true | ||||||
| 	} | 	} | ||||||
| 	w.Unlock() | 	w.Unlock() | ||||||
| @@ -121,11 +180,26 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...Execu | |||||||
| 		return "", err | 		return "", err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	//	stepStore := store.NewNamespaceStore(w.opts.Store, filepath.Join("steps", eid)) | 	stepStore := store.NewNamespaceStore(w.opts.Store, "steps"+w.opts.Store.Options().Separator+eid) | ||||||
| 	workflowStore := store.NewNamespaceStore(w.opts.Store, filepath.Join("workflows", eid)) | 	workflowStore := store.NewNamespaceStore(w.opts.Store, "workflows"+w.opts.Store.Options().Separator+eid) | ||||||
|  |  | ||||||
| 	options := NewExecuteOptions(opts...) | 	options := NewExecuteOptions(opts...) | ||||||
|  |  | ||||||
|  | 	steps, err := w.getSteps(options.Start, options.Reverse) | ||||||
|  | 	if err != nil { | ||||||
|  | 		if werr := workflowStore.Write(w.opts.Context, "status", &codec.Frame{Data: []byte(StatusPending.String())}); werr != nil { | ||||||
|  | 			w.opts.Logger.Error(w.opts.Context, "store write error", werr) | ||||||
|  | 		} | ||||||
|  | 		return "", err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	var wg sync.WaitGroup | ||||||
|  | 	cherr := make(chan error, 1) | ||||||
|  | 	chstatus := make(chan Status, 1) | ||||||
|  |  | ||||||
|  | 	nctx, cancel := context.WithCancel(ctx) | ||||||
|  | 	defer cancel() | ||||||
|  |  | ||||||
| 	nopts := make([]ExecuteOption, 0, len(opts)+5) | 	nopts := make([]ExecuteOption, 0, len(opts)+5) | ||||||
|  |  | ||||||
| 	nopts = append(nopts, | 	nopts = append(nopts, | ||||||
| @@ -135,152 +209,21 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...Execu | |||||||
| 		ExecuteMeter(w.opts.Meter), | 		ExecuteMeter(w.opts.Meter), | ||||||
| 	) | 	) | ||||||
| 	nopts = append(nopts, opts...) | 	nopts = append(nopts, opts...) | ||||||
|  | 	done := make(chan struct{}) | ||||||
|  |  | ||||||
| 	if werr := workflowStore.Write(ctx, "status", &codec.Frame{Data: []byte(StatusRunning.String())}); werr != nil { | 	if werr := workflowStore.Write(w.opts.Context, "status", &codec.Frame{Data: []byte(StatusRunning.String())}); werr != nil { | ||||||
| 		w.opts.Logger.Error(ctx, "store error: %v", werr) | 		w.opts.Logger.Error(w.opts.Context, "store write error", werr) | ||||||
| 		return eid, werr | 		return eid, werr | ||||||
| 	} | 	} | ||||||
|  | 	for idx := range steps { | ||||||
| 	var startID string | 		for nidx := range steps[idx] { | ||||||
| 	if options.Start == "" { | 			cstep := steps[idx][nidx] | ||||||
| 		mp := w.g.GetRoots() | 			if werr := stepStore.Write(ctx, cstep.ID()+w.opts.Store.Options().Separator+"status", &codec.Frame{Data: []byte(StatusPending.String())}); werr != nil { | ||||||
| 		if len(mp) != 1 { | 				return eid, werr | ||||||
| 			return eid, ErrStepNotExists |  | ||||||
| 		} |  | ||||||
| 		for k := range mp { |  | ||||||
| 			startID = k |  | ||||||
| 		} |  | ||||||
| 	} else { |  | ||||||
| 		for k, v := range w.g.GetVertices() { |  | ||||||
| 			if v == options.Start { |  | ||||||
| 				startID = k |  | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if startID == "" { |  | ||||||
| 		return eid, ErrStepNotExists |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if options.Async { |  | ||||||
| 		go w.handleWorkflow(startID, nopts...) |  | ||||||
| 		return eid, nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return eid, w.handleWorkflow(startID, nopts...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *microWorkflow) handleWorkflow(startID string, opts ...ExecuteOption) error { |  | ||||||
| 	w.RLock() |  | ||||||
| 	defer w.RUnlock() |  | ||||||
|  |  | ||||||
| 	//	stepStore := store.NewNamespaceStore(w.opts.Store, filepath.Join("steps", eid)) |  | ||||||
| 	// workflowStore := store.NewNamespaceStore(w.opts.Store, filepath.Join("workflows", eid)) |  | ||||||
|  |  | ||||||
| 	// Get IDs of all descendant vertices. |  | ||||||
| 	flowIDs, errDes := w.g.GetDescendants(startID) |  | ||||||
| 	if errDes != nil { |  | ||||||
| 		return errDes |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// inputChannels provides for input channels for each of the descendant vertices (+ the start-vertex). |  | ||||||
| 	inputChannels := make(map[string]chan FlowResult, len(flowIDs)+1) |  | ||||||
|  |  | ||||||
| 	// Iterate vertex IDs and create an input channel for each of them and a single |  | ||||||
| 	// output channel for leaves. Note, this "pre-flight" is needed to ensure we |  | ||||||
| 	// really have an input channel regardless of how we traverse the tree and spawn |  | ||||||
| 	// workers. |  | ||||||
| 	leafCount := 0 |  | ||||||
|  |  | ||||||
| 	for id := range flowIDs { |  | ||||||
|  |  | ||||||
| 		// Get all parents of this vertex. |  | ||||||
| 		parents, errPar := w.g.GetParents(id) |  | ||||||
| 		if errPar != nil { |  | ||||||
| 			return errPar |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// Create a buffered input channel that has capacity for all parent results. |  | ||||||
| 		inputChannels[id] = make(chan FlowResult, len(parents)) |  | ||||||
|  |  | ||||||
| 		if ok, err := w.g.IsLeaf(id); ok && err == nil { |  | ||||||
| 			leafCount += 1 |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// outputChannel caries the results of leaf vertices. |  | ||||||
| 	outputChannel := make(chan FlowResult, leafCount) |  | ||||||
|  |  | ||||||
| 	// To also process the start vertex and to have its results being passed to its |  | ||||||
| 	// children, add it to the vertex IDs. Also add an input channel for the start |  | ||||||
| 	// vertex and feed the inputs to this channel. |  | ||||||
| 	flowIDs[startID] = struct{}{} |  | ||||||
| 	inputChannels[startID] = make(chan FlowResult, len(inputs)) |  | ||||||
| 	for _, i := range inputs { |  | ||||||
| 		inputChannels[startID] <- i |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	wg := sync.WaitGroup{} |  | ||||||
|  |  | ||||||
| 	// Iterate all vertex IDs (now incl. start vertex) and handle each worker (incl. |  | ||||||
| 	// inputs and outputs) in a separate goroutine. |  | ||||||
| 	for id := range flowIDs { |  | ||||||
|  |  | ||||||
| 		// Get all children of this vertex that later need to be notified. Note, we |  | ||||||
| 		// collect all children before the goroutine to be able to release the read |  | ||||||
| 		// lock as early as possible. |  | ||||||
| 		children, errChildren := w.g.GetChildren(id) |  | ||||||
| 		if errChildren != nil { |  | ||||||
| 			return errChildren |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// Remember to wait for this goroutine. |  | ||||||
| 		wg.Add(1) |  | ||||||
|  |  | ||||||
| 		go func(id string) { |  | ||||||
| 			// Get this vertex's input channel. |  | ||||||
| 			// Note, only concurrent read here, which is fine. |  | ||||||
| 			c := inputChannels[id] |  | ||||||
|  |  | ||||||
| 			// Await all parent inputs and stuff them into a slice. |  | ||||||
| 			parentCount := cap(c) |  | ||||||
| 			parentResults := make([]FlowResult, parentCount) |  | ||||||
| 			for i := 0; i < parentCount; i++ { |  | ||||||
| 				parentResults[i] = <-c |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			// Execute the worker. |  | ||||||
| 			errWorker := callback(w.g, id, parentResults) |  | ||||||
| 			if errWorker != nil { |  | ||||||
| 				return errWorker |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			// Send this worker's FlowResult onto all children's input channels or, if it is |  | ||||||
| 			// a leaf (i.e. no children), send the result onto the output channel. |  | ||||||
| 			if len(children) > 0 { |  | ||||||
| 				for child := range children { |  | ||||||
| 					inputChannels[child] <- flowResult |  | ||||||
| 				} |  | ||||||
| 			} else { |  | ||||||
| 				outputChannel <- flowResult |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			// "Sign off". |  | ||||||
| 			wg.Done() |  | ||||||
| 		}(id) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// Wait for all go routines to finish. |  | ||||||
| 	wg.Wait() |  | ||||||
|  |  | ||||||
| 	// Await all leaf vertex results and stuff them into a slice. |  | ||||||
| 	resultCount := cap(outputChannel) |  | ||||||
| 	results := make([]FlowResult, resultCount) |  | ||||||
| 	for i := 0; i < resultCount; i++ { |  | ||||||
| 		results[i] = <-outputChannel |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	/* |  | ||||||
| 	go func() { | 	go func() { | ||||||
| 		for idx := range steps { | 		for idx := range steps { | ||||||
| 			for nidx := range steps[idx] { | 			for nidx := range steps[idx] { | ||||||
| @@ -294,7 +237,7 @@ func (w *microWorkflow) handleWorkflow(startID string, opts ...ExecuteOption) er | |||||||
| 					return | 					return | ||||||
| 				} | 				} | ||||||
| 				if w.opts.Logger.V(logger.TraceLevel) { | 				if w.opts.Logger.V(logger.TraceLevel) { | ||||||
| 						w.opts.Logger.Tracef(nctx, "will be executed %v", steps[idx][nidx]) | 					w.opts.Logger.Trace(nctx, fmt.Sprintf("will be executed %v", steps[idx][nidx])) | ||||||
| 				} | 				} | ||||||
| 				cstep := steps[idx][nidx] | 				cstep := steps[idx][nidx] | ||||||
| 				// nolint: nestif | 				// nolint: nestif | ||||||
| @@ -302,65 +245,65 @@ func (w *microWorkflow) handleWorkflow(startID string, opts ...ExecuteOption) er | |||||||
| 					wg.Add(1) | 					wg.Add(1) | ||||||
| 					go func(step Step) { | 					go func(step Step) { | ||||||
| 						defer wg.Done() | 						defer wg.Done() | ||||||
| 							if werr := stepStore.Write(ctx, filepath.Join(step.ID(), "req"), req); werr != nil { | 						if werr := stepStore.Write(ctx, step.ID()+w.opts.Store.Options().Separator+"req", req); werr != nil { | ||||||
| 							cherr <- werr | 							cherr <- werr | ||||||
| 							return | 							return | ||||||
| 						} | 						} | ||||||
| 							if werr := stepStore.Write(ctx, filepath.Join(step.ID(), "status"), &codec.Frame{Data: []byte(StatusRunning.String())}); werr != nil { | 						if werr := stepStore.Write(ctx, step.ID()+w.opts.Store.Options().Separator+"status", &codec.Frame{Data: []byte(StatusRunning.String())}); werr != nil { | ||||||
| 							cherr <- werr | 							cherr <- werr | ||||||
| 							return | 							return | ||||||
| 						} | 						} | ||||||
| 						rsp, serr := step.Execute(nctx, req, nopts...) | 						rsp, serr := step.Execute(nctx, req, nopts...) | ||||||
| 						if serr != nil { | 						if serr != nil { | ||||||
| 							step.SetStatus(StatusFailure) | 							step.SetStatus(StatusFailure) | ||||||
| 								if werr := stepStore.Write(ctx, filepath.Join(step.ID(), "rsp"), serr); werr != nil && w.opts.Logger.V(logger.ErrorLevel) { | 							if werr := stepStore.Write(ctx, step.ID()+w.opts.Store.Options().Separator+"rsp", serr); werr != nil && w.opts.Logger.V(logger.ErrorLevel) { | ||||||
| 									w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | 								w.opts.Logger.Error(ctx, "store write error", werr) | ||||||
| 							} | 							} | ||||||
| 								if werr := stepStore.Write(ctx, filepath.Join(step.ID(), "status"), &codec.Frame{Data: []byte(StatusFailure.String())}); werr != nil && w.opts.Logger.V(logger.ErrorLevel) { | 							if werr := stepStore.Write(ctx, step.ID()+w.opts.Store.Options().Separator+"status", &codec.Frame{Data: []byte(StatusFailure.String())}); werr != nil && w.opts.Logger.V(logger.ErrorLevel) { | ||||||
| 									w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | 								w.opts.Logger.Error(ctx, "store write error", werr) | ||||||
| 							} | 							} | ||||||
| 							cherr <- serr | 							cherr <- serr | ||||||
| 							return | 							return | ||||||
| 						} | 						} | ||||||
| 							if werr := stepStore.Write(ctx, filepath.Join(step.ID(), "rsp"), rsp); werr != nil { | 						if werr := stepStore.Write(ctx, step.ID()+w.opts.Store.Options().Separator+"rsp", rsp); werr != nil { | ||||||
| 								w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | 							w.opts.Logger.Error(ctx, "store write error", werr) | ||||||
| 							cherr <- werr | 							cherr <- werr | ||||||
| 							return | 							return | ||||||
| 						} | 						} | ||||||
| 							if werr := stepStore.Write(ctx, filepath.Join(step.ID(), "status"), &codec.Frame{Data: []byte(StatusSuccess.String())}); werr != nil { | 						if werr := stepStore.Write(ctx, step.ID()+w.opts.Store.Options().Separator+"status", &codec.Frame{Data: []byte(StatusSuccess.String())}); werr != nil { | ||||||
| 								w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | 							w.opts.Logger.Error(ctx, "store write error", werr) | ||||||
| 							cherr <- werr | 							cherr <- werr | ||||||
| 							return | 							return | ||||||
| 						} | 						} | ||||||
| 					}(cstep) | 					}(cstep) | ||||||
| 					wg.Wait() | 					wg.Wait() | ||||||
| 				} else { | 				} else { | ||||||
| 						if werr := stepStore.Write(ctx, filepath.Join(cstep.ID(), "req"), req); werr != nil { | 					if werr := stepStore.Write(ctx, cstep.ID()+w.opts.Store.Options().Separator+"req", req); werr != nil { | ||||||
| 						cherr <- werr | 						cherr <- werr | ||||||
| 						return | 						return | ||||||
| 					} | 					} | ||||||
| 						if werr := stepStore.Write(ctx, filepath.Join(cstep.ID(), "status"), &codec.Frame{Data: []byte(StatusRunning.String())}); werr != nil { | 					if werr := stepStore.Write(ctx, cstep.ID()+w.opts.Store.Options().Separator+"status", &codec.Frame{Data: []byte(StatusRunning.String())}); werr != nil { | ||||||
| 						cherr <- werr | 						cherr <- werr | ||||||
| 						return | 						return | ||||||
| 					} | 					} | ||||||
| 					rsp, serr := cstep.Execute(nctx, req, nopts...) | 					rsp, serr := cstep.Execute(nctx, req, nopts...) | ||||||
| 					if serr != nil { | 					if serr != nil { | ||||||
| 						cstep.SetStatus(StatusFailure) | 						cstep.SetStatus(StatusFailure) | ||||||
| 							if werr := stepStore.Write(ctx, filepath.Join(cstep.ID(), "rsp"), serr); werr != nil && w.opts.Logger.V(logger.ErrorLevel) { | 						if werr := stepStore.Write(ctx, cstep.ID()+w.opts.Store.Options().Separator+"rsp", serr); werr != nil && w.opts.Logger.V(logger.ErrorLevel) { | ||||||
| 								w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | 							w.opts.Logger.Error(ctx, "store write error", werr) | ||||||
| 						} | 						} | ||||||
| 							if werr := stepStore.Write(ctx, filepath.Join(cstep.ID(), "status"), &codec.Frame{Data: []byte(StatusFailure.String())}); werr != nil && w.opts.Logger.V(logger.ErrorLevel) { | 						if werr := stepStore.Write(ctx, cstep.ID()+w.opts.Store.Options().Separator+"status", &codec.Frame{Data: []byte(StatusFailure.String())}); werr != nil && w.opts.Logger.V(logger.ErrorLevel) { | ||||||
| 								w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | 							w.opts.Logger.Error(ctx, "store write error", werr) | ||||||
| 						} | 						} | ||||||
| 						cherr <- serr | 						cherr <- serr | ||||||
| 						return | 						return | ||||||
| 					} | 					} | ||||||
| 						if werr := stepStore.Write(ctx, filepath.Join(cstep.ID(), "rsp"), rsp); werr != nil { | 					if werr := stepStore.Write(ctx, cstep.ID()+w.opts.Store.Options().Separator+"rsp", rsp); werr != nil { | ||||||
| 							w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | 						w.opts.Logger.Error(ctx, "store write error", werr) | ||||||
| 						cherr <- werr | 						cherr <- werr | ||||||
| 						return | 						return | ||||||
| 					} | 					} | ||||||
| 						if werr := stepStore.Write(ctx, filepath.Join(cstep.ID(), "status"), &codec.Frame{Data: []byte(StatusSuccess.String())}); werr != nil { | 					if werr := stepStore.Write(ctx, cstep.ID()+w.opts.Store.Options().Separator+"status", &codec.Frame{Data: []byte(StatusSuccess.String())}); werr != nil { | ||||||
| 						cherr <- werr | 						cherr <- werr | ||||||
| 						return | 						return | ||||||
| 					} | 					} | ||||||
| @@ -374,7 +317,7 @@ func (w *microWorkflow) handleWorkflow(startID string, opts ...ExecuteOption) er | |||||||
| 		return eid, nil | 		return eid, nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 		logger.Tracef(ctx, "wait for finish or error") | 	logger.DefaultLogger.Trace(ctx, "wait for finish or error") | ||||||
| 	select { | 	select { | ||||||
| 	case <-nctx.Done(): | 	case <-nctx.Done(): | ||||||
| 		err = nctx.Err() | 		err = nctx.Err() | ||||||
| @@ -390,19 +333,19 @@ func (w *microWorkflow) handleWorkflow(startID string, opts ...ExecuteOption) er | |||||||
| 	switch { | 	switch { | ||||||
| 	case nctx.Err() != nil: | 	case nctx.Err() != nil: | ||||||
| 		if werr := workflowStore.Write(w.opts.Context, "status", &codec.Frame{Data: []byte(StatusAborted.String())}); werr != nil { | 		if werr := workflowStore.Write(w.opts.Context, "status", &codec.Frame{Data: []byte(StatusAborted.String())}); werr != nil { | ||||||
| 				w.opts.Logger.Errorf(w.opts.Context, "store error: %v", werr) | 			w.opts.Logger.Error(w.opts.Context, "store write error", werr) | ||||||
| 		} | 		} | ||||||
| 	case err == nil: | 	case err == nil: | ||||||
| 		if werr := workflowStore.Write(w.opts.Context, "status", &codec.Frame{Data: []byte(StatusSuccess.String())}); werr != nil { | 		if werr := workflowStore.Write(w.opts.Context, "status", &codec.Frame{Data: []byte(StatusSuccess.String())}); werr != nil { | ||||||
| 				w.opts.Logger.Errorf(w.opts.Context, "store error: %v", werr) | 			w.opts.Logger.Error(w.opts.Context, "store write error", werr) | ||||||
| 		} | 		} | ||||||
| 	case err != nil: | 	case err != nil: | ||||||
| 		if werr := workflowStore.Write(w.opts.Context, "status", &codec.Frame{Data: []byte(StatusFailure.String())}); werr != nil { | 		if werr := workflowStore.Write(w.opts.Context, "status", &codec.Frame{Data: []byte(StatusFailure.String())}); werr != nil { | ||||||
| 				w.opts.Logger.Errorf(w.opts.Context, "store error: %v", werr) | 			w.opts.Logger.Error(w.opts.Context, "store write error", werr) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	*/ |  | ||||||
| 	return err | 	return eid, err | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewFlow create new flow | // NewFlow create new flow | ||||||
| @@ -442,11 +385,11 @@ func (f *microFlow) WorkflowList(ctx context.Context) ([]Workflow, error) { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (f *microFlow) WorkflowCreate(ctx context.Context, id string, steps ...Step) (Workflow, error) { | func (f *microFlow) WorkflowCreate(ctx context.Context, id string, steps ...Step) (Workflow, error) { | ||||||
| 	w := µWorkflow{opts: f.opts, id: id, g: &dag.DAG{}, steps: make(map[string]Step, len(steps))} | 	w := µWorkflow{opts: f.opts, id: id, g: &dag.AcyclicGraph{}, steps: make(map[string]Step, len(steps))} | ||||||
|  |  | ||||||
| 	for _, s := range steps { | 	for _, s := range steps { | ||||||
| 		w.steps[s.String()] = s | 		w.steps[s.String()] = s | ||||||
| 		w.g.AddVertex(s) | 		w.g.Add(s) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	for _, dst := range steps { | 	for _, dst := range steps { | ||||||
| @@ -455,11 +398,14 @@ func (f *microFlow) WorkflowCreate(ctx context.Context, id string, steps ...Step | |||||||
| 			if !ok { | 			if !ok { | ||||||
| 				return nil, ErrStepNotExists | 				return nil, ErrStepNotExists | ||||||
| 			} | 			} | ||||||
| 			w.g.AddEdge(src.String(), dst.String()) | 			w.g.Connect(dag.BasicEdge(src, dst)) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	w.g.ReduceTransitively() | 	if err := w.g.Validate(); err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	w.g.TransitiveReduction() | ||||||
|  |  | ||||||
| 	w.init = true | 	w.init = true | ||||||
|  |  | ||||||
|   | |||||||
| @@ -1,5 +1,5 @@ | |||||||
| // Package flow is an interface used for saga pattern microservice workflow | // Package flow is an interface used for saga pattern microservice workflow | ||||||
| package flow // import "go.unistack.org/micro/v4/flow" | package flow | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| @@ -7,7 +7,7 @@ import ( | |||||||
| 	"sync" | 	"sync" | ||||||
| 	"sync/atomic" | 	"sync/atomic" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/metadata" | 	"go.unistack.org/micro/v3/metadata" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| @@ -125,6 +125,8 @@ type Workflow interface { | |||||||
| 	AppendSteps(steps ...Step) error | 	AppendSteps(steps ...Step) error | ||||||
| 	// Status returns workflow status | 	// Status returns workflow status | ||||||
| 	Status() Status | 	Status() Status | ||||||
|  | 	// Steps returns steps slice where parallel steps returned on the same level | ||||||
|  | 	Steps() ([][]Step, error) | ||||||
| 	// Suspend suspends execution | 	// Suspend suspends execution | ||||||
| 	Suspend(ctx context.Context, id string) error | 	Suspend(ctx context.Context, id string) error | ||||||
| 	// Resume resumes execution | 	// Resume resumes execution | ||||||
|   | |||||||
| @@ -4,11 +4,11 @@ import ( | |||||||
| 	"context" | 	"context" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/client" | 	"go.unistack.org/micro/v3/client" | ||||||
| 	"go.unistack.org/micro/v4/logger" | 	"go.unistack.org/micro/v3/logger" | ||||||
| 	"go.unistack.org/micro/v4/meter" | 	"go.unistack.org/micro/v3/meter" | ||||||
| 	"go.unistack.org/micro/v4/store" | 	"go.unistack.org/micro/v3/store" | ||||||
| 	"go.unistack.org/micro/v4/tracer" | 	"go.unistack.org/micro/v3/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Option func | // Option func | ||||||
| @@ -123,6 +123,8 @@ type ExecuteOptions struct { | |||||||
| 	Start string | 	Start string | ||||||
| 	// Timeout for execution | 	// Timeout for execution | ||||||
| 	Timeout time.Duration | 	Timeout time.Duration | ||||||
|  | 	// Reverse execution | ||||||
|  | 	Reverse bool | ||||||
| 	// Async enables async execution | 	// Async enables async execution | ||||||
| 	Async bool | 	Async bool | ||||||
| } | } | ||||||
| @@ -165,6 +167,13 @@ func ExecuteContext(ctx context.Context) ExecuteOption { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // ExecuteReverse says that dag must be run in reverse order | ||||||
|  | func ExecuteReverse(b bool) ExecuteOption { | ||||||
|  | 	return func(o *ExecuteOptions) { | ||||||
|  | 		o.Reverse = b | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // ExecuteTimeout pass timeout time.Duration for execution | // ExecuteTimeout pass timeout time.Duration for execution | ||||||
| func ExecuteTimeout(td time.Duration) ExecuteOption { | func ExecuteTimeout(td time.Duration) ExecuteOption { | ||||||
| 	return func(o *ExecuteOptions) { | 	return func(o *ExecuteOptions) { | ||||||
|   | |||||||
| @@ -5,7 +5,7 @@ import ( | |||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"testing" | 	"testing" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/logger" | 	"go.unistack.org/micro/v3/logger" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| func TestFSMStart(t *testing.T) { | func TestFSMStart(t *testing.T) { | ||||||
|   | |||||||
							
								
								
									
										35
									
								
								go.mod
									
									
									
									
									
								
							
							
						
						
									
										35
									
								
								go.mod
									
									
									
									
									
								
							| @@ -1,34 +1,43 @@ | |||||||
| module go.unistack.org/micro/v4 | module go.unistack.org/micro/v3 | ||||||
|  |  | ||||||
| go 1.22.0 | go 1.23.4 | ||||||
|  |  | ||||||
| require ( | require ( | ||||||
| 	dario.cat/mergo v1.0.1 | 	dario.cat/mergo v1.0.1 | ||||||
| 	github.com/DATA-DOG/go-sqlmock v1.5.2 | 	github.com/DATA-DOG/go-sqlmock v1.5.0 | ||||||
| 	github.com/KimMachineGun/automemlimit v0.7.0 | 	github.com/KimMachineGun/automemlimit v0.6.1 | ||||||
| 	github.com/ash3in/uuidv8 v1.2.0 | 	github.com/ash3in/uuidv8 v1.0.1 | ||||||
| 	github.com/google/uuid v1.6.0 | 	github.com/google/uuid v1.6.0 | ||||||
| 	github.com/matoous/go-nanoid v1.5.1 | 	github.com/matoous/go-nanoid v1.5.1 | ||||||
| 	github.com/patrickmn/go-cache v2.1.0+incompatible | 	github.com/patrickmn/go-cache v2.1.0+incompatible | ||||||
| 	github.com/silas/dag v0.0.0-20220518035006-a7e85ada93c5 | 	github.com/silas/dag v0.0.0-20220518035006-a7e85ada93c5 | ||||||
| 	github.com/spf13/cast v1.7.1 |  | ||||||
| 	go.uber.org/atomic v1.11.0 |  | ||||||
| 	go.uber.org/automaxprocs v1.6.0 | 	go.uber.org/automaxprocs v1.6.0 | ||||||
| 	go.unistack.org/micro-proto/v4 v4.1.0 | 	go.unistack.org/micro-proto/v3 v3.4.1 | ||||||
| 	golang.org/x/sync v0.10.0 | 	golang.org/x/sync v0.10.0 | ||||||
| 	google.golang.org/grpc v1.69.4 | 	google.golang.org/grpc v1.68.1 | ||||||
| 	google.golang.org/protobuf v1.36.3 | 	google.golang.org/protobuf v1.35.2 | ||||||
| 	gopkg.in/yaml.v3 v3.0.1 | 	gopkg.in/yaml.v3 v3.0.1 | ||||||
| ) | ) | ||||||
|  |  | ||||||
| require ( | require ( | ||||||
|  | 	github.com/cilium/ebpf v0.16.0 // indirect | ||||||
|  | 	github.com/containerd/cgroups/v3 v3.0.4 // indirect | ||||||
|  | 	github.com/containerd/log v0.1.0 // indirect | ||||||
|  | 	github.com/coreos/go-systemd/v22 v22.5.0 // indirect | ||||||
| 	github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect | 	github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect | ||||||
|  | 	github.com/docker/go-units v0.5.0 // indirect | ||||||
|  | 	github.com/godbus/dbus/v5 v5.1.0 // indirect | ||||||
|  | 	github.com/moby/sys/userns v0.1.0 // indirect | ||||||
|  | 	github.com/opencontainers/runtime-spec v1.2.0 // indirect | ||||||
| 	github.com/pbnjay/memory v0.0.0-20210728143218-7b4eea64cf58 // indirect | 	github.com/pbnjay/memory v0.0.0-20210728143218-7b4eea64cf58 // indirect | ||||||
| 	github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 // indirect | 	github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 // indirect | ||||||
| 	github.com/rogpeppe/go-internal v1.13.1 // indirect | 	github.com/rogpeppe/go-internal v1.13.1 // indirect | ||||||
|  | 	github.com/sirupsen/logrus v1.9.3 // indirect | ||||||
| 	github.com/stretchr/testify v1.10.0 // indirect | 	github.com/stretchr/testify v1.10.0 // indirect | ||||||
| 	golang.org/x/net v0.34.0 // indirect | 	go.uber.org/goleak v1.3.0 // indirect | ||||||
| 	golang.org/x/sys v0.29.0 // indirect | 	golang.org/x/exp v0.0.0-20241210194714-1829a127f884 // indirect | ||||||
| 	google.golang.org/genproto/googleapis/rpc v0.0.0-20241216192217-9240e9c98484 // indirect | 	golang.org/x/net v0.32.0 // indirect | ||||||
|  | 	golang.org/x/sys v0.28.0 // indirect | ||||||
|  | 	google.golang.org/genproto/googleapis/rpc v0.0.0-20241209162323-e6fa225c2576 // indirect | ||||||
| 	gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c // indirect | 	gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c // indirect | ||||||
| ) | ) | ||||||
|   | |||||||
							
								
								
									
										83
									
								
								go.sum
									
									
									
									
									
								
							
							
						
						
									
										83
									
								
								go.sum
									
									
									
									
									
								
							| @@ -1,22 +1,40 @@ | |||||||
| dario.cat/mergo v1.0.1 h1:Ra4+bf83h2ztPIQYNP99R6m+Y7KfnARDfID+a+vLl4s= | dario.cat/mergo v1.0.1 h1:Ra4+bf83h2ztPIQYNP99R6m+Y7KfnARDfID+a+vLl4s= | ||||||
| dario.cat/mergo v1.0.1/go.mod h1:uNxQE+84aUszobStD9th8a29P2fMDhsBdgRYvZOxGmk= | dario.cat/mergo v1.0.1/go.mod h1:uNxQE+84aUszobStD9th8a29P2fMDhsBdgRYvZOxGmk= | ||||||
| github.com/DATA-DOG/go-sqlmock v1.5.2 h1:OcvFkGmslmlZibjAjaHm3L//6LiuBgolP7OputlJIzU= | github.com/DATA-DOG/go-sqlmock v1.5.0 h1:Shsta01QNfFxHCfpW6YH2STWB0MudeXXEWMr20OEh60= | ||||||
| github.com/DATA-DOG/go-sqlmock v1.5.2/go.mod h1:88MAG/4G7SMwSE3CeA0ZKzrT5CiOU3OJ+JlNzwDqpNU= | github.com/DATA-DOG/go-sqlmock v1.5.0/go.mod h1:f/Ixk793poVmq4qj/V1dPUg2JEAKC73Q5eFN3EC/SaM= | ||||||
| github.com/KimMachineGun/automemlimit v0.7.0 h1:7G06p/dMSf7G8E6oq+f2uOPuVncFyIlDI/pBWK49u88= | github.com/KimMachineGun/automemlimit v0.6.1 h1:ILa9j1onAAMadBsyyUJv5cack8Y1WT26yLj/V+ulKp8= | ||||||
| github.com/KimMachineGun/automemlimit v0.7.0/go.mod h1:QZxpHaGOQoYvFhv/r4u3U0JTC2ZcOwbSr11UZF46UBM= | github.com/KimMachineGun/automemlimit v0.6.1/go.mod h1:T7xYht7B8r6AG/AqFcUdc7fzd2bIdBKmepfP2S1svPY= | ||||||
| github.com/ash3in/uuidv8 v1.2.0 h1:2oogGdtCPwaVtyvPPGin4TfZLtOGE5F+W++E880G6SI= | github.com/ash3in/uuidv8 v1.0.1 h1:dIq1XRkWT8lGA7N5s7WRTB4V3k49WTBLvILz7aCLp80= | ||||||
| github.com/ash3in/uuidv8 v1.2.0/go.mod h1:BnU0wJBxnzdEKmVg4xckBkD+VZuecTFTUP3M0dWgyY4= | github.com/ash3in/uuidv8 v1.0.1/go.mod h1:EoyUgCtxNBnrnpc9efw5rVN1cQ+LFGCoJiFuD6maOMw= | ||||||
|  | github.com/cilium/ebpf v0.16.0 h1:+BiEnHL6Z7lXnlGUsXQPPAE7+kenAd4ES8MQ5min0Ok= | ||||||
|  | github.com/cilium/ebpf v0.16.0/go.mod h1:L7u2Blt2jMM/vLAVgjxluxtBKlz3/GWjB0dMOEngfwE= | ||||||
|  | github.com/containerd/cgroups/v3 v3.0.4 h1:2fs7l3P0Qxb1nKWuJNFiwhp2CqiKzho71DQkDrHJIo4= | ||||||
|  | github.com/containerd/cgroups/v3 v3.0.4/go.mod h1:SA5DLYnXO8pTGYiAHXz94qvLQTKfVM5GEVisn4jpins= | ||||||
|  | github.com/containerd/log v0.1.0 h1:TCJt7ioM2cr/tfR8GPbGf9/VRAX8D2B4PjzCpfX540I= | ||||||
|  | github.com/containerd/log v0.1.0/go.mod h1:VRRf09a7mHDIRezVKTRCrOq78v577GXq3bSa3EhrzVo= | ||||||
|  | github.com/coreos/go-systemd/v22 v22.5.0 h1:RrqgGjYQKalulkV8NGVIfkXQf6YYmOyiJKk8iXXhfZs= | ||||||
|  | github.com/coreos/go-systemd/v22 v22.5.0/go.mod h1:Y58oyj3AT4RCenI/lSvhwexgC+NSVTIJ3seZv2GcEnc= | ||||||
|  | github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | ||||||
|  | github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | ||||||
| github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1VwoXQT9A3Wy9MM3WgvqSxFWenqJduM= | github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1VwoXQT9A3Wy9MM3WgvqSxFWenqJduM= | ||||||
| github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | ||||||
| github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8= | github.com/docker/go-units v0.5.0 h1:69rxXcBk27SvSaaxTtLh/8llcHD8vYHT7WSdRZ/jvr4= | ||||||
| github.com/frankban/quicktest v1.14.6/go.mod h1:4ptaffx2x8+WTWXmUCuVU6aPUX1/Mz7zb5vbUoiM6w0= | github.com/docker/go-units v0.5.0/go.mod h1:fgPhTUdO+D/Jk86RDLlptpiXQzgHJF7gydDDbaIK4Dk= | ||||||
|  | github.com/go-quicktest/qt v1.101.0 h1:O1K29Txy5P2OK0dGo59b7b0LR6wKfIhttaAhHUyn7eI= | ||||||
|  | github.com/go-quicktest/qt v1.101.0/go.mod h1:14Bz/f7NwaXPtdYEgzsx46kqSxVwTbzVZsDC26tQJow= | ||||||
|  | github.com/godbus/dbus/v5 v5.0.4/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA= | ||||||
|  | github.com/godbus/dbus/v5 v5.1.0 h1:4KLkAxT3aOY8Li4FRJe/KvhoNFFxo0m6fNuFUO8QJUk= | ||||||
|  | github.com/godbus/dbus/v5 v5.1.0/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA= | ||||||
| github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek= | github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek= | ||||||
| github.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps= | github.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps= | ||||||
| github.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI= | github.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI= | ||||||
| github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY= | github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY= | ||||||
| github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0= | github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0= | ||||||
| github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= | github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= | ||||||
| github.com/kisielk/sqlstruct v0.0.0-20201105191214-5f3e10d3ab46/go.mod h1:yyMNCyc/Ib3bDTKd379tNMpB/7/H5TjM2Y9QJ5THLbE= | github.com/josharian/native v1.1.0 h1:uuaP0hAbW7Y4l0ZRQ6C9zfb7Mg1mbFKry/xzDAfmtLA= | ||||||
|  | github.com/josharian/native v1.1.0/go.mod h1:7X/raswPFr05uY3HiLlYeyQntB6OO7E/d2Cu7qoaN2w= | ||||||
|  | github.com/jsimonetti/rtnetlink/v2 v2.0.1 h1:xda7qaHDSVOsADNouv7ukSuicKZO7GgVUCXxpaIEIlM= | ||||||
|  | github.com/jsimonetti/rtnetlink/v2 v2.0.1/go.mod h1:7MoNYNbb3UaDHtF8udiJo/RH6VsTKP1pqKLUTVCvToE= | ||||||
| github.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI= | github.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI= | ||||||
| github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE= | github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE= | ||||||
| github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk= | github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk= | ||||||
| @@ -26,10 +44,19 @@ github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY= | |||||||
| github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE= | github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE= | ||||||
| github.com/matoous/go-nanoid v1.5.1 h1:aCjdvTyO9LLnTIi0fgdXhOPPvOHjpXN6Ik9DaNjIct4= | github.com/matoous/go-nanoid v1.5.1 h1:aCjdvTyO9LLnTIi0fgdXhOPPvOHjpXN6Ik9DaNjIct4= | ||||||
| github.com/matoous/go-nanoid v1.5.1/go.mod h1:zyD2a71IubI24efhpvkJz+ZwfwagzgSO6UNiFsZKN7U= | github.com/matoous/go-nanoid v1.5.1/go.mod h1:zyD2a71IubI24efhpvkJz+ZwfwagzgSO6UNiFsZKN7U= | ||||||
|  | github.com/mdlayher/netlink v1.7.2 h1:/UtM3ofJap7Vl4QWCPDGXY8d3GIY2UGSDbK+QWmY8/g= | ||||||
|  | github.com/mdlayher/netlink v1.7.2/go.mod h1:xraEF7uJbxLhc5fpHL4cPe221LI2bdttWlU+ZGLfQSw= | ||||||
|  | github.com/mdlayher/socket v0.4.1 h1:eM9y2/jlbs1M615oshPQOHZzj6R6wMT7bX5NPiQvn2U= | ||||||
|  | github.com/mdlayher/socket v0.4.1/go.mod h1:cAqeGjoufqdxWkD7DkpyS+wcefOtmu5OQ8KuoJGIReA= | ||||||
|  | github.com/moby/sys/userns v0.1.0 h1:tVLXkFOxVu9A64/yh59slHVv9ahO9UIev4JZusOLG/g= | ||||||
|  | github.com/moby/sys/userns v0.1.0/go.mod h1:IHUYgu/kao6N8YZlp9Cf444ySSvCmDlmzUcYfDHOl28= | ||||||
|  | github.com/opencontainers/runtime-spec v1.2.0 h1:z97+pHb3uELt/yiAWD691HNHQIF07bE7dzrbT927iTk= | ||||||
|  | github.com/opencontainers/runtime-spec v1.2.0/go.mod h1:jwyrGlmzljRJv/Fgzds9SsS/C5hL+LL3ko9hs6T5lQ0= | ||||||
| github.com/patrickmn/go-cache v2.1.0+incompatible h1:HRMgzkcYKYpi3C8ajMPV8OFXaaRUnok+kx1WdO15EQc= | github.com/patrickmn/go-cache v2.1.0+incompatible h1:HRMgzkcYKYpi3C8ajMPV8OFXaaRUnok+kx1WdO15EQc= | ||||||
| github.com/patrickmn/go-cache v2.1.0+incompatible/go.mod h1:3Qf8kWWT7OJRJbdiICTKqZju1ZixQ/KpMGzzAfe6+WQ= | github.com/patrickmn/go-cache v2.1.0+incompatible/go.mod h1:3Qf8kWWT7OJRJbdiICTKqZju1ZixQ/KpMGzzAfe6+WQ= | ||||||
| github.com/pbnjay/memory v0.0.0-20210728143218-7b4eea64cf58 h1:onHthvaw9LFnH4t2DcNVpwGmV9E1BkGknEliJkfwQj0= | github.com/pbnjay/memory v0.0.0-20210728143218-7b4eea64cf58 h1:onHthvaw9LFnH4t2DcNVpwGmV9E1BkGknEliJkfwQj0= | ||||||
| github.com/pbnjay/memory v0.0.0-20210728143218-7b4eea64cf58/go.mod h1:DXv8WO4yhMYhSNPKjeNKa5WY9YCIEBRbNzFFPJbWO6Y= | github.com/pbnjay/memory v0.0.0-20210728143218-7b4eea64cf58/go.mod h1:DXv8WO4yhMYhSNPKjeNKa5WY9YCIEBRbNzFFPJbWO6Y= | ||||||
|  | github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= | ||||||
| github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 h1:Jamvg5psRIccs7FGNTlIRMkT8wgtp5eCXdBlqhYGL6U= | github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 h1:Jamvg5psRIccs7FGNTlIRMkT8wgtp5eCXdBlqhYGL6U= | ||||||
| github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= | github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= | ||||||
| github.com/prashantv/gostub v1.1.0 h1:BTyx3RfQjRHnUWaGF9oQos79AlQ5k8WNktv7VGvVH4g= | github.com/prashantv/gostub v1.1.0 h1:BTyx3RfQjRHnUWaGF9oQos79AlQ5k8WNktv7VGvVH4g= | ||||||
| @@ -38,32 +65,38 @@ github.com/rogpeppe/go-internal v1.13.1 h1:KvO1DLK/DRN07sQ1LQKScxyZJuNnedQ5/wKSR | |||||||
| github.com/rogpeppe/go-internal v1.13.1/go.mod h1:uMEvuHeurkdAXX61udpOXGD/AzZDWNMNyH2VO9fmH0o= | github.com/rogpeppe/go-internal v1.13.1/go.mod h1:uMEvuHeurkdAXX61udpOXGD/AzZDWNMNyH2VO9fmH0o= | ||||||
| github.com/silas/dag v0.0.0-20220518035006-a7e85ada93c5 h1:G/FZtUu7a6NTWl3KUHMV9jkLAh/Rvtf03NWMHaEDl+E= | github.com/silas/dag v0.0.0-20220518035006-a7e85ada93c5 h1:G/FZtUu7a6NTWl3KUHMV9jkLAh/Rvtf03NWMHaEDl+E= | ||||||
| github.com/silas/dag v0.0.0-20220518035006-a7e85ada93c5/go.mod h1:7RTUFBdIRC9nZ7/3RyRNH1bdqIShrDejd1YbLwgPS+I= | github.com/silas/dag v0.0.0-20220518035006-a7e85ada93c5/go.mod h1:7RTUFBdIRC9nZ7/3RyRNH1bdqIShrDejd1YbLwgPS+I= | ||||||
| github.com/spf13/cast v1.7.1 h1:cuNEagBQEHWN1FnbGEjCXL2szYEXqfJPbP2HNUaca9Y= | github.com/sirupsen/logrus v1.9.3 h1:dueUQJ1C2q9oE3F7wvmSGAaVtTmUizReu6fjN8uqzbQ= | ||||||
| github.com/spf13/cast v1.7.1/go.mod h1:ancEpBxwJDODSW/UG4rDrAqiKolqNNh2DX3mk86cAdo= | github.com/sirupsen/logrus v1.9.3/go.mod h1:naHLuLoDiP4jHNo9R0sCBMtWGeIprob74mVsIT4qYEQ= | ||||||
|  | github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= | ||||||
|  | github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= | ||||||
| github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA= | github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA= | ||||||
| github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY= | github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY= | ||||||
| go.uber.org/atomic v1.11.0 h1:ZvwS0R+56ePWxUNi+Atn9dWONBPp/AUETXlHW0DxSjE= |  | ||||||
| go.uber.org/atomic v1.11.0/go.mod h1:LUxbIzbOniOlMKjJjyPfpl4v+PKK2cNJn91OQbhoJI0= |  | ||||||
| go.uber.org/automaxprocs v1.6.0 h1:O3y2/QNTOdbF+e/dpXNNW7Rx2hZ4sTIPyybbxyNqTUs= | go.uber.org/automaxprocs v1.6.0 h1:O3y2/QNTOdbF+e/dpXNNW7Rx2hZ4sTIPyybbxyNqTUs= | ||||||
| go.uber.org/automaxprocs v1.6.0/go.mod h1:ifeIMSnPZuznNm6jmdzmU3/bfk01Fe2fotchwEFJ8r8= | go.uber.org/automaxprocs v1.6.0/go.mod h1:ifeIMSnPZuznNm6jmdzmU3/bfk01Fe2fotchwEFJ8r8= | ||||||
| go.unistack.org/micro-proto/v4 v4.1.0 h1:qPwL2n/oqh9RE3RTTDgt28XK3QzV597VugQPaw9lKUk= | go.uber.org/goleak v1.3.0 h1:2K3zAYmnTNqV73imy9J1T3WC+gmCePx2hEGkimedGto= | ||||||
| go.unistack.org/micro-proto/v4 v4.1.0/go.mod h1:ArmK7o+uFvxSY3dbJhKBBX4Pm1rhWdLEFf3LxBrMtec= | go.uber.org/goleak v1.3.0/go.mod h1:CoHD4mav9JJNrW/WLlf7HGZPjdw8EucARQHekz1X6bE= | ||||||
| golang.org/x/net v0.34.0 h1:Mb7Mrk043xzHgnRM88suvJFwzVrRfHEHJEl5/71CKw0= | go.unistack.org/micro-proto/v3 v3.4.1 h1:UTjLSRz2YZuaHk9iSlVqqsA50JQNAEK2ZFboGqtEa9Q= | ||||||
| golang.org/x/net v0.34.0/go.mod h1:di0qlW3YNM5oh6GqDGQr92MyTozJPmybPK4Ev/Gm31k= | go.unistack.org/micro-proto/v3 v3.4.1/go.mod h1:okx/cnOhzuCX0ggl/vToatbCupi0O44diiiLLsZ93Zo= | ||||||
|  | golang.org/x/exp v0.0.0-20241210194714-1829a127f884 h1:Y/Mj/94zIQQGHVSv1tTtQBDaQaJe62U9bkDZKKyhPCU= | ||||||
|  | golang.org/x/exp v0.0.0-20241210194714-1829a127f884/go.mod h1:qj5a5QZpwLU2NLQudwIN5koi3beDhSAlJwa67PuM98c= | ||||||
|  | golang.org/x/net v0.32.0 h1:ZqPmj8Kzc+Y6e0+skZsuACbx+wzMgo5MQsJh9Qd6aYI= | ||||||
|  | golang.org/x/net v0.32.0/go.mod h1:CwU0IoeOlnQQWJ6ioyFrfRuomB8GKF6KbYXZVyeXNfs= | ||||||
| golang.org/x/sync v0.10.0 h1:3NQrjDixjgGwUOCaF8w2+VYHv0Ve/vGYSbdkTa98gmQ= | golang.org/x/sync v0.10.0 h1:3NQrjDixjgGwUOCaF8w2+VYHv0Ve/vGYSbdkTa98gmQ= | ||||||
| golang.org/x/sync v0.10.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk= | golang.org/x/sync v0.10.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk= | ||||||
| golang.org/x/sys v0.29.0 h1:TPYlXGxvx1MGTn2GiZDhnjPA9wZzZeGKHHmKhHYvgaU= | golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||||||
| golang.org/x/sys v0.29.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= | golang.org/x/sys v0.28.0 h1:Fksou7UEQUWlKvIdsqzJmUmCX3cZuD2+P3XyyzwMhlA= | ||||||
|  | golang.org/x/sys v0.28.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= | ||||||
| golang.org/x/text v0.21.0 h1:zyQAAkrwaneQ066sspRyJaG9VNi/YJ1NfzcGB3hZ/qo= | golang.org/x/text v0.21.0 h1:zyQAAkrwaneQ066sspRyJaG9VNi/YJ1NfzcGB3hZ/qo= | ||||||
| golang.org/x/text v0.21.0/go.mod h1:4IBbMaMmOPCJ8SecivzSH54+73PCFmPWxNTLm+vZkEQ= | golang.org/x/text v0.21.0/go.mod h1:4IBbMaMmOPCJ8SecivzSH54+73PCFmPWxNTLm+vZkEQ= | ||||||
| google.golang.org/genproto/googleapis/rpc v0.0.0-20241216192217-9240e9c98484 h1:Z7FRVJPSMaHQxD0uXU8WdgFh8PseLM8Q8NzhnpMrBhQ= | google.golang.org/genproto/googleapis/rpc v0.0.0-20241209162323-e6fa225c2576 h1:8ZmaLZE4XWrtU3MyClkYqqtl6Oegr3235h7jxsDyqCY= | ||||||
| google.golang.org/genproto/googleapis/rpc v0.0.0-20241216192217-9240e9c98484/go.mod h1:lcTa1sDdWEIHMWlITnIczmw5w60CF9ffkb8Z+DVmmjA= | google.golang.org/genproto/googleapis/rpc v0.0.0-20241209162323-e6fa225c2576/go.mod h1:5uTbfoYQed2U9p3KIj2/Zzm02PYhndfdmML0qC3q3FU= | ||||||
| google.golang.org/grpc v1.69.4 h1:MF5TftSMkd8GLw/m0KM6V8CMOCY6NZ1NQDPGFgbTt4A= | google.golang.org/grpc v1.68.1 h1:oI5oTa11+ng8r8XMMN7jAOmWfPZWbYpCFaMUTACxkM0= | ||||||
| google.golang.org/grpc v1.69.4/go.mod h1:vyjdE6jLBI76dgpDojsFGNaHlxdjXN9ghpnd2o7JGZ4= | google.golang.org/grpc v1.68.1/go.mod h1:+q1XYFJjShcqn0QZHvCyeR4CXPA+llXIeUIfIe00waw= | ||||||
| google.golang.org/protobuf v1.36.3 h1:82DV7MYdb8anAVi3qge1wSnMDrnKK7ebr+I0hHRN1BU= | google.golang.org/protobuf v1.35.2 h1:8Ar7bF+apOIoThw1EdZl0p1oWvMqTHmpA2fRTyZO8io= | ||||||
| google.golang.org/protobuf v1.36.3/go.mod h1:9fA7Ob0pmnwhb644+1+CVWFRbNajQ6iRojtC/QF5bRE= | google.golang.org/protobuf v1.35.2/go.mod h1:9fA7Ob0pmnwhb644+1+CVWFRbNajQ6iRojtC/QF5bRE= | ||||||
| gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= | gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= | ||||||
| gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk= | gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk= | ||||||
| gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q= | gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q= | ||||||
|  | gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= | ||||||
| gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= | gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= | ||||||
| gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= | gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= | ||||||
|   | |||||||
| @@ -8,7 +8,7 @@ import ( | |||||||
| 	"slices" | 	"slices" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/meter" | 	"go.unistack.org/micro/v3/meter" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Option func signature | // Option func signature | ||||||
| @@ -99,7 +99,6 @@ func WithAddFields(fields ...interface{}) Option { | |||||||
| 					iv, iok := o.Fields[i].(string) | 					iv, iok := o.Fields[i].(string) | ||||||
| 					jv, jok := fields[j].(string) | 					jv, jok := fields[j].(string) | ||||||
| 					if iok && jok && iv == jv { | 					if iok && jok && iv == jv { | ||||||
| 						o.Fields[i+1] = fields[j+1] |  | ||||||
| 						fields = slices.Delete(fields, j, j+2) | 						fields = slices.Delete(fields, j, j+2) | ||||||
| 					} | 					} | ||||||
| 				} | 				} | ||||||
|   | |||||||
| @@ -13,16 +13,15 @@ import ( | |||||||
| 	"sync/atomic" | 	"sync/atomic" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/logger" | 	"go.unistack.org/micro/v3/logger" | ||||||
| 	"go.unistack.org/micro/v4/semconv" | 	"go.unistack.org/micro/v3/semconv" | ||||||
| 	"go.unistack.org/micro/v4/tracer" | 	"go.unistack.org/micro/v3/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| const ( | const ( | ||||||
| 	badKey = "!BADKEY" | 	badKey = "!BADKEY" | ||||||
| 	// defaultCallerSkipCount used by logger | 	// defaultCallerSkipCount used by logger | ||||||
| 	defaultCallerSkipCount = 3 | 	defaultCallerSkipCount = 3 | ||||||
| 	timeFormat             = "2006-01-02T15:04:05.000000000Z07:00" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var reTrace = regexp.MustCompile(`.*/slog/logger\.go.*\n`) | var reTrace = regexp.MustCompile(`.*/slog/logger\.go.*\n`) | ||||||
| @@ -65,7 +64,6 @@ func (s *slogLogger) renameAttr(_ []string, a slog.Attr) slog.Attr { | |||||||
| 		a.Key = s.opts.SourceKey | 		a.Key = s.opts.SourceKey | ||||||
| 	case slog.TimeKey: | 	case slog.TimeKey: | ||||||
| 		a.Key = s.opts.TimeKey | 		a.Key = s.opts.TimeKey | ||||||
| 		a.Value = slog.StringValue(a.Value.Time().Format(timeFormat)) |  | ||||||
| 	case slog.MessageKey: | 	case slog.MessageKey: | ||||||
| 		a.Key = s.opts.MessageKey | 		a.Key = s.opts.MessageKey | ||||||
| 	case slog.LevelKey: | 	case slog.LevelKey: | ||||||
| @@ -278,7 +276,7 @@ func (s *slogLogger) printLog(ctx context.Context, lvl logger.Level, msg string, | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if s.opts.AddStacktrace && (lvl == logger.FatalLevel || lvl == logger.ErrorLevel) { | 	if (s.opts.AddStacktrace || lvl == logger.FatalLevel) || (s.opts.AddStacktrace && lvl == logger.ErrorLevel) { | ||||||
| 		stackInfo := make([]byte, 1024*1024) | 		stackInfo := make([]byte, 1024*1024) | ||||||
| 		if stackSize := runtime.Stack(stackInfo, false); stackSize > 0 { | 		if stackSize := runtime.Stack(stackInfo, false); stackSize > 0 { | ||||||
| 			traceLines := reTrace.Split(string(stackInfo[:stackSize]), -1) | 			traceLines := reTrace.Split(string(stackInfo[:stackSize]), -1) | ||||||
|   | |||||||
| @@ -9,61 +9,18 @@ import ( | |||||||
| 	"log/slog" | 	"log/slog" | ||||||
| 	"strings" | 	"strings" | ||||||
| 	"testing" | 	"testing" | ||||||
| 	"time" |  | ||||||
|  |  | ||||||
| 	"github.com/google/uuid" | 	"github.com/google/uuid" | ||||||
| 	"go.unistack.org/micro/v4/logger" | 	"go.unistack.org/micro/v3/logger" | ||||||
| 	"go.unistack.org/micro/v4/metadata" | 	"go.unistack.org/micro/v3/metadata" | ||||||
| 	"go.unistack.org/micro/v4/util/buffer" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // always first to have proper check |  | ||||||
| func TestStacktrace(t *testing.T) { | func TestStacktrace(t *testing.T) { | ||||||
| 	ctx := context.TODO() |  | ||||||
| 	buf := bytes.NewBuffer(nil) |  | ||||||
| 	l := NewLogger(logger.WithLevel(logger.DebugLevel), logger.WithOutput(buf), |  | ||||||
| 		WithHandlerFunc(slog.NewTextHandler), |  | ||||||
| 		logger.WithAddStacktrace(true), |  | ||||||
| 	) |  | ||||||
| 	if err := l.Init(logger.WithFields("key1", "val1")); err != nil { |  | ||||||
| 		t.Fatal(err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	l.Error(ctx, "msg1", errors.New("err")) |  | ||||||
|  |  | ||||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`slog_test.go:32`)) { |  | ||||||
| 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestDelayedBuffer(t *testing.T) { |  | ||||||
| 	ctx := context.TODO() |  | ||||||
| 	buf := bytes.NewBuffer(nil) |  | ||||||
| 	dbuf := buffer.NewDelayedBuffer(100, 100*time.Millisecond, buf) |  | ||||||
| 	l := NewLogger(logger.WithLevel(logger.ErrorLevel), logger.WithOutput(dbuf), |  | ||||||
| 		WithHandlerFunc(slog.NewTextHandler), |  | ||||||
| 		logger.WithAddStacktrace(true), |  | ||||||
| 	) |  | ||||||
| 	if err := l.Init(logger.WithFields("key1", "val1")); err != nil { |  | ||||||
| 		t.Fatal(err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	l.Error(ctx, "msg1", errors.New("err")) |  | ||||||
| 	time.Sleep(120 * time.Millisecond) |  | ||||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`key1=val1`)) { |  | ||||||
| 		t.Fatalf("logger delayed buffer not works, buf contains: %s", buf.Bytes()) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestTime(t *testing.T) { |  | ||||||
| 	ctx := context.TODO() | 	ctx := context.TODO() | ||||||
| 	buf := bytes.NewBuffer(nil) | 	buf := bytes.NewBuffer(nil) | ||||||
| 	l := NewLogger(logger.WithLevel(logger.ErrorLevel), logger.WithOutput(buf), | 	l := NewLogger(logger.WithLevel(logger.ErrorLevel), logger.WithOutput(buf), | ||||||
| 		WithHandlerFunc(slog.NewTextHandler), | 		WithHandlerFunc(slog.NewTextHandler), | ||||||
| 		logger.WithAddStacktrace(true), | 		logger.WithAddStacktrace(true), | ||||||
| 		logger.WithTimeFunc(func() time.Time { |  | ||||||
| 			return time.Unix(0, 0) |  | ||||||
| 		}), |  | ||||||
| 	) | 	) | ||||||
| 	if err := l.Init(logger.WithFields("key1", "val1")); err != nil { | 	if err := l.Init(logger.WithFields("key1", "val1")); err != nil { | ||||||
| 		t.Fatal(err) | 		t.Fatal(err) | ||||||
| @@ -71,8 +28,7 @@ func TestTime(t *testing.T) { | |||||||
|  |  | ||||||
| 	l.Error(ctx, "msg1", errors.New("err")) | 	l.Error(ctx, "msg1", errors.New("err")) | ||||||
|  |  | ||||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`timestamp=1970-01-01T03:00:00.000000000+03:00`)) && | 	if !bytes.Contains(buf.Bytes(), []byte(`slog_test.go:29`)) { | ||||||
| 		!bytes.Contains(buf.Bytes(), []byte(`timestamp=1970-01-01T00:00:00.000000000Z`)) { |  | ||||||
| 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| @@ -124,7 +80,7 @@ func TestWithDedupKeysWithAddFields(t *testing.T) { | |||||||
|  |  | ||||||
| 	l.Info(ctx, "msg3") | 	l.Info(ctx, "msg3") | ||||||
|  |  | ||||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`msg=msg3 key1=val4 key2=val3`)) { | 	if !bytes.Contains(buf.Bytes(), []byte(`msg=msg3 key1=val1 key2=val2`)) { | ||||||
| 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| @@ -412,10 +368,9 @@ func Test_WithContextAttrFunc(t *testing.T) { | |||||||
| 			} | 			} | ||||||
| 			attrs := make([]interface{}, 0, 10) | 			attrs := make([]interface{}, 0, 10) | ||||||
| 			for k, v := range md { | 			for k, v := range md { | ||||||
| 				key := strings.ToLower(k) | 				switch k { | ||||||
| 				switch key { | 				case "X-Request-Id", "Phone", "External-Id", "Source-Service", "X-App-Install-Id", "Client-Id", "Client-Ip": | ||||||
| 				case "x-request-id", "phone", "external-Id", "source-service", "x-app-install-id", "client-id", "client-ip": | 					attrs = append(attrs, strings.ToLower(k), v) | ||||||
| 					attrs = append(attrs, key, v[0]) |  | ||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
| 			return attrs | 			return attrs | ||||||
|   | |||||||
| @@ -8,7 +8,7 @@ import ( | |||||||
| 	"strconv" | 	"strconv" | ||||||
| 	"strings" | 	"strings" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/codec" | 	"go.unistack.org/micro/v3/codec" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| const sf = "0-+# " | const sf = "0-+# " | ||||||
|   | |||||||
| @@ -5,7 +5,7 @@ import ( | |||||||
| 	"strings" | 	"strings" | ||||||
| 	"testing" | 	"testing" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/codec" | 	"go.unistack.org/micro/v3/codec" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| func TestUnwrap(t *testing.T) { | func TestUnwrap(t *testing.T) { | ||||||
|   | |||||||
							
								
								
									
										179
									
								
								metadata/context.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										179
									
								
								metadata/context.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,179 @@ | |||||||
|  | // Package metadata is a way of defining message headers | ||||||
|  | package metadata | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type ( | ||||||
|  | 	mdIncomingKey struct{} | ||||||
|  | 	mdOutgoingKey struct{} | ||||||
|  | 	mdKey         struct{} | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // FromIncomingContext returns metadata from incoming ctx | ||||||
|  | // returned metadata shoud not be modified or race condition happens | ||||||
|  | func FromIncomingContext(ctx context.Context) (Metadata, bool) { | ||||||
|  | 	if ctx == nil { | ||||||
|  | 		return nil, false | ||||||
|  | 	} | ||||||
|  | 	md, ok := ctx.Value(mdIncomingKey{}).(*rawMetadata) | ||||||
|  | 	if !ok || md.md == nil { | ||||||
|  | 		return nil, false | ||||||
|  | 	} | ||||||
|  | 	return md.md, ok | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MustIncomingContext returns metadata from incoming ctx | ||||||
|  | // returned metadata shoud not be modified or race condition happens. | ||||||
|  | // If metadata not exists panics. | ||||||
|  | func MustIncomingContext(ctx context.Context) Metadata { | ||||||
|  | 	md, ok := FromIncomingContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		panic("missing metadata") | ||||||
|  | 	} | ||||||
|  | 	return md | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // FromOutgoingContext returns metadata from outgoing ctx | ||||||
|  | // returned metadata shoud not be modified or race condition happens | ||||||
|  | func FromOutgoingContext(ctx context.Context) (Metadata, bool) { | ||||||
|  | 	if ctx == nil { | ||||||
|  | 		return nil, false | ||||||
|  | 	} | ||||||
|  | 	md, ok := ctx.Value(mdOutgoingKey{}).(*rawMetadata) | ||||||
|  | 	if !ok || md.md == nil { | ||||||
|  | 		return nil, false | ||||||
|  | 	} | ||||||
|  | 	return md.md, ok | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MustOutgoingContext returns metadata from outgoing ctx | ||||||
|  | // returned metadata shoud not be modified or race condition happens. | ||||||
|  | // If metadata not exists panics. | ||||||
|  | func MustOutgoingContext(ctx context.Context) Metadata { | ||||||
|  | 	md, ok := FromOutgoingContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		panic("missing metadata") | ||||||
|  | 	} | ||||||
|  | 	return md | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // FromContext returns metadata from the given context | ||||||
|  | // returned metadata shoud not be modified or race condition happens | ||||||
|  | func FromContext(ctx context.Context) (Metadata, bool) { | ||||||
|  | 	if ctx == nil { | ||||||
|  | 		return nil, false | ||||||
|  | 	} | ||||||
|  | 	md, ok := ctx.Value(mdKey{}).(*rawMetadata) | ||||||
|  | 	if !ok || md.md == nil { | ||||||
|  | 		return nil, false | ||||||
|  | 	} | ||||||
|  | 	return md.md, ok | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MustContext returns metadata from the given context | ||||||
|  | // returned metadata shoud not be modified or race condition happens | ||||||
|  | func MustContext(ctx context.Context) Metadata { | ||||||
|  | 	md, ok := FromContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		panic("missing metadata") | ||||||
|  | 	} | ||||||
|  | 	return md | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewContext creates a new context with the given metadata | ||||||
|  | func NewContext(ctx context.Context, md Metadata) context.Context { | ||||||
|  | 	if ctx == nil { | ||||||
|  | 		ctx = context.Background() | ||||||
|  | 	} | ||||||
|  | 	return context.WithValue(ctx, mdKey{}, &rawMetadata{md}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SetOutgoingContext modify outgoing context with given metadata | ||||||
|  | func SetOutgoingContext(ctx context.Context, md Metadata) bool { | ||||||
|  | 	if ctx == nil { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  | 	if omd, ok := ctx.Value(mdOutgoingKey{}).(*rawMetadata); ok { | ||||||
|  | 		omd.md = md | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SetIncomingContext modify incoming context with given metadata | ||||||
|  | func SetIncomingContext(ctx context.Context, md Metadata) bool { | ||||||
|  | 	if ctx == nil { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  | 	if omd, ok := ctx.Value(mdIncomingKey{}).(*rawMetadata); ok { | ||||||
|  | 		omd.md = md | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewIncomingContext creates a new context with incoming metadata attached | ||||||
|  | func NewIncomingContext(ctx context.Context, md Metadata) context.Context { | ||||||
|  | 	if ctx == nil { | ||||||
|  | 		ctx = context.Background() | ||||||
|  | 	} | ||||||
|  | 	return context.WithValue(ctx, mdIncomingKey{}, &rawMetadata{md}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewOutgoingContext creates a new context with outcoming metadata attached | ||||||
|  | func NewOutgoingContext(ctx context.Context, md Metadata) context.Context { | ||||||
|  | 	if ctx == nil { | ||||||
|  | 		ctx = context.Background() | ||||||
|  | 	} | ||||||
|  | 	return context.WithValue(ctx, mdOutgoingKey{}, &rawMetadata{md}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // AppendOutgoingContext apends new md to context | ||||||
|  | func AppendOutgoingContext(ctx context.Context, kv ...string) context.Context { | ||||||
|  | 	md, ok := Pairs(kv...) | ||||||
|  | 	if !ok { | ||||||
|  | 		return ctx | ||||||
|  | 	} | ||||||
|  | 	omd, ok := FromOutgoingContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		return NewOutgoingContext(ctx, md) | ||||||
|  | 	} | ||||||
|  | 	for k, v := range md { | ||||||
|  | 		omd.Set(k, v) | ||||||
|  | 	} | ||||||
|  | 	return ctx | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // AppendIncomingContext apends new md to context | ||||||
|  | func AppendIncomingContext(ctx context.Context, kv ...string) context.Context { | ||||||
|  | 	md, ok := Pairs(kv...) | ||||||
|  | 	if !ok { | ||||||
|  | 		return ctx | ||||||
|  | 	} | ||||||
|  | 	omd, ok := FromIncomingContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		return NewIncomingContext(ctx, md) | ||||||
|  | 	} | ||||||
|  | 	for k, v := range md { | ||||||
|  | 		omd.Set(k, v) | ||||||
|  | 	} | ||||||
|  | 	return ctx | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // AppendContext apends new md to context | ||||||
|  | func AppendContext(ctx context.Context, kv ...string) context.Context { | ||||||
|  | 	md, ok := Pairs(kv...) | ||||||
|  | 	if !ok { | ||||||
|  | 		return ctx | ||||||
|  | 	} | ||||||
|  | 	omd, ok := FromContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		return NewContext(ctx, md) | ||||||
|  | 	} | ||||||
|  | 	for k, v := range md { | ||||||
|  | 		omd.Set(k, v) | ||||||
|  | 	} | ||||||
|  | 	return ctx | ||||||
|  | } | ||||||
							
								
								
									
										140
									
								
								metadata/context_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										140
									
								
								metadata/context_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,140 @@ | |||||||
|  | package metadata | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"testing" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func TestFromNilContext(t *testing.T) { | ||||||
|  | 	// nolint: staticcheck | ||||||
|  | 	c, ok := FromContext(nil) | ||||||
|  | 	if ok || c != nil { | ||||||
|  | 		t.Fatal("FromContext not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestNewNilContext(t *testing.T) { | ||||||
|  | 	// nolint: staticcheck | ||||||
|  | 	ctx := NewContext(nil, New(0)) | ||||||
|  |  | ||||||
|  | 	c, ok := FromContext(ctx) | ||||||
|  | 	if c == nil || !ok { | ||||||
|  | 		t.Fatal("NewContext not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestFromContext(t *testing.T) { | ||||||
|  | 	ctx := context.WithValue(context.TODO(), mdKey{}, &rawMetadata{New(0)}) | ||||||
|  |  | ||||||
|  | 	c, ok := FromContext(ctx) | ||||||
|  | 	if c == nil || !ok { | ||||||
|  | 		t.Fatal("FromContext not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestNewContext(t *testing.T) { | ||||||
|  | 	ctx := NewContext(context.TODO(), New(0)) | ||||||
|  |  | ||||||
|  | 	c, ok := FromContext(ctx) | ||||||
|  | 	if c == nil || !ok { | ||||||
|  | 		t.Fatal("NewContext not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestFromIncomingContext(t *testing.T) { | ||||||
|  | 	ctx := context.WithValue(context.TODO(), mdIncomingKey{}, &rawMetadata{New(0)}) | ||||||
|  |  | ||||||
|  | 	c, ok := FromIncomingContext(ctx) | ||||||
|  | 	if c == nil || !ok { | ||||||
|  | 		t.Fatal("FromIncomingContext not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestFromOutgoingContext(t *testing.T) { | ||||||
|  | 	ctx := context.WithValue(context.TODO(), mdOutgoingKey{}, &rawMetadata{New(0)}) | ||||||
|  |  | ||||||
|  | 	c, ok := FromOutgoingContext(ctx) | ||||||
|  | 	if c == nil || !ok { | ||||||
|  | 		t.Fatal("FromOutgoingContext not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestSetIncomingContext(t *testing.T) { | ||||||
|  | 	md := New(1) | ||||||
|  | 	md.Set("key", "val") | ||||||
|  | 	ctx := context.WithValue(context.TODO(), mdIncomingKey{}, &rawMetadata{}) | ||||||
|  | 	if !SetIncomingContext(ctx, md) { | ||||||
|  | 		t.Fatal("SetIncomingContext not works") | ||||||
|  | 	} | ||||||
|  | 	md, ok := FromIncomingContext(ctx) | ||||||
|  | 	if md == nil || !ok { | ||||||
|  | 		t.Fatal("SetIncomingContext not works") | ||||||
|  | 	} else if v, ok := md.Get("key"); !ok || v != "val" { | ||||||
|  | 		t.Fatal("SetIncomingContext not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestSetOutgoingContext(t *testing.T) { | ||||||
|  | 	md := New(1) | ||||||
|  | 	md.Set("key", "val") | ||||||
|  | 	ctx := context.WithValue(context.TODO(), mdOutgoingKey{}, &rawMetadata{}) | ||||||
|  | 	if !SetOutgoingContext(ctx, md) { | ||||||
|  | 		t.Fatal("SetOutgoingContext not works") | ||||||
|  | 	} | ||||||
|  | 	md, ok := FromOutgoingContext(ctx) | ||||||
|  | 	if md == nil || !ok { | ||||||
|  | 		t.Fatal("SetOutgoingContext not works") | ||||||
|  | 	} else if v, ok := md.Get("key"); !ok || v != "val" { | ||||||
|  | 		t.Fatal("SetOutgoingContext not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestNewIncomingContext(t *testing.T) { | ||||||
|  | 	md := New(1) | ||||||
|  | 	md.Set("key", "val") | ||||||
|  | 	ctx := NewIncomingContext(context.TODO(), md) | ||||||
|  |  | ||||||
|  | 	c, ok := FromIncomingContext(ctx) | ||||||
|  | 	if c == nil || !ok { | ||||||
|  | 		t.Fatal("NewIncomingContext not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestNewOutgoingContext(t *testing.T) { | ||||||
|  | 	md := New(1) | ||||||
|  | 	md.Set("key", "val") | ||||||
|  | 	ctx := NewOutgoingContext(context.TODO(), md) | ||||||
|  |  | ||||||
|  | 	c, ok := FromOutgoingContext(ctx) | ||||||
|  | 	if c == nil || !ok { | ||||||
|  | 		t.Fatal("NewOutgoingContext not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestAppendIncomingContext(t *testing.T) { | ||||||
|  | 	md := New(1) | ||||||
|  | 	md.Set("key1", "val1") | ||||||
|  | 	ctx := AppendIncomingContext(context.TODO(), "key2", "val2") | ||||||
|  |  | ||||||
|  | 	nmd, ok := FromIncomingContext(ctx) | ||||||
|  | 	if nmd == nil || !ok { | ||||||
|  | 		t.Fatal("AppendIncomingContext not works") | ||||||
|  | 	} | ||||||
|  | 	if v, ok := nmd.Get("key2"); !ok || v != "val2" { | ||||||
|  | 		t.Fatal("AppendIncomingContext not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestAppendOutgoingContext(t *testing.T) { | ||||||
|  | 	md := New(1) | ||||||
|  | 	md.Set("key1", "val1") | ||||||
|  | 	ctx := AppendOutgoingContext(context.TODO(), "key2", "val2") | ||||||
|  |  | ||||||
|  | 	nmd, ok := FromOutgoingContext(ctx) | ||||||
|  | 	if nmd == nil || !ok { | ||||||
|  | 		t.Fatal("AppendOutgoingContext not works") | ||||||
|  | 	} | ||||||
|  | 	if v, ok := nmd.Get("key2"); !ok || v != "val2" { | ||||||
|  | 		t.Fatal("AppendOutgoingContext not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -1,19 +0,0 @@ | |||||||
| // Package metadata is a way of defining message headers |  | ||||||
| package metadata |  | ||||||
|  |  | ||||||
| var ( |  | ||||||
| 	// HeaderTopic is the header name that contains topic name |  | ||||||
| 	HeaderTopic = "Micro-Topic" |  | ||||||
| 	// HeaderContentType specifies content type of message |  | ||||||
| 	HeaderContentType = "Content-Type" |  | ||||||
| 	// HeaderEndpoint specifies endpoint in service |  | ||||||
| 	HeaderEndpoint = "Micro-Endpoint" |  | ||||||
| 	// HeaderService specifies service |  | ||||||
| 	HeaderService = "Micro-Service" |  | ||||||
| 	// HeaderTimeout specifies timeout of operation |  | ||||||
| 	HeaderTimeout = "Micro-Timeout" |  | ||||||
| 	// HeaderAuthorization specifies Authorization header |  | ||||||
| 	HeaderAuthorization = "Authorization" |  | ||||||
| 	// HeaderXRequestID specifies request id |  | ||||||
| 	HeaderXRequestID = "X-Request-Id" |  | ||||||
| ) |  | ||||||
| @@ -1,471 +1,41 @@ | |||||||
|  | // Package metadata is a way of defining message headers | ||||||
| package metadata | package metadata | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" |  | ||||||
| 	"fmt" |  | ||||||
| 	"net/textproto" | 	"net/textproto" | ||||||
| 	"sort" | 	"sort" | ||||||
| 	"strings" | 	"strings" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // defaultMetadataSize used when need to init new Metadata | var ( | ||||||
| var defaultMetadataSize = 2 | 	// HeaderTopic is the header name that contains topic name | ||||||
|  | 	HeaderTopic = "Micro-Topic" | ||||||
| // Metadata is a mapping from metadata keys to values. Users should use the following | 	// HeaderContentType specifies content type of message | ||||||
| // two convenience functions New and Pairs to generate Metadata. | 	HeaderContentType = "Content-Type" | ||||||
| type Metadata map[string][]string | 	// HeaderEndpoint specifies endpoint in service | ||||||
|  | 	HeaderEndpoint = "Micro-Endpoint" | ||||||
| // New creates an zero Metadata. | 	// HeaderService specifies service | ||||||
| func New(l int) Metadata { | 	HeaderService = "Micro-Service" | ||||||
| 	if l == 0 { | 	// HeaderTimeout specifies timeout of operation | ||||||
| 		l = defaultMetadataSize | 	HeaderTimeout = "Micro-Timeout" | ||||||
| 	} | 	// HeaderAuthorization specifies Authorization header | ||||||
| 	md := make(Metadata, l) | 	HeaderAuthorization = "Authorization" | ||||||
| 	return md | 	// HeaderXRequestID specifies request id | ||||||
| } | 	HeaderXRequestID = "X-Request-Id" | ||||||
|  |  | ||||||
| // NewWithMetadata creates an Metadata from a given key-value map. |  | ||||||
| func NewWithMetadata(m map[string]string) Metadata { |  | ||||||
| 	md := make(Metadata, len(m)) |  | ||||||
| 	for key, val := range m { |  | ||||||
| 		md[key] = append(md[key], val) |  | ||||||
| 	} |  | ||||||
| 	return md |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Pairs returns an Metadata formed by the mapping of key, value ... |  | ||||||
| // Pairs panics if len(kv) is odd. |  | ||||||
| func Pairs(kv ...string) Metadata { |  | ||||||
| 	if len(kv)%2 == 1 { |  | ||||||
| 		panic(fmt.Sprintf("metadata: Pairs got the odd number of input pairs for metadata: %d", len(kv))) |  | ||||||
| 	} |  | ||||||
| 	md := make(Metadata, len(kv)/2) |  | ||||||
| 	for i := 0; i < len(kv); i += 2 { |  | ||||||
| 		md[kv[i]] = append(md[kv[i]], kv[i+1]) |  | ||||||
| 	} |  | ||||||
| 	return md |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Len returns the number of items in Metadata. |  | ||||||
| func (md Metadata) Len() int { |  | ||||||
| 	return len(md) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Copy returns a copy of Metadata. |  | ||||||
| func Copy(src Metadata) Metadata { |  | ||||||
| 	out := make(Metadata, len(src)) |  | ||||||
| 	for k, v := range src { |  | ||||||
| 		out[k] = copyOf(v) |  | ||||||
| 	} |  | ||||||
| 	return out |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Copy returns a copy of Metadata. |  | ||||||
| func (md Metadata) Copy() Metadata { |  | ||||||
| 	out := make(Metadata, len(md)) |  | ||||||
| 	for k, v := range md { |  | ||||||
| 		out[k] = copyOf(v) |  | ||||||
| 	} |  | ||||||
| 	return out |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // AsMap returns a copy of Metadata with map[string]string. |  | ||||||
| func (md Metadata) AsMap() map[string]string { |  | ||||||
| 	out := make(map[string]string, len(md)) |  | ||||||
| 	for k, v := range md { |  | ||||||
| 		out[k] = strings.Join(v, ",") |  | ||||||
| 	} |  | ||||||
| 	return out |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // AsHTTP1 returns a copy of Metadata |  | ||||||
| // with CanonicalMIMEHeaderKey. |  | ||||||
| func (md Metadata) AsHTTP1() map[string][]string { |  | ||||||
| 	out := make(map[string][]string, len(md)) |  | ||||||
| 	for k, v := range md { |  | ||||||
| 		out[textproto.CanonicalMIMEHeaderKey(k)] = copyOf(v) |  | ||||||
| 	} |  | ||||||
| 	return out |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // AsHTTP1 returns a copy of Metadata |  | ||||||
| // with strings.ToLower. |  | ||||||
| func (md Metadata) AsHTTP2() map[string][]string { |  | ||||||
| 	out := make(map[string][]string, len(md)) |  | ||||||
| 	for k, v := range md { |  | ||||||
| 		out[strings.ToLower(k)] = copyOf(v) |  | ||||||
| 	} |  | ||||||
| 	return out |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // CopyTo copies Metadata to out. |  | ||||||
| func (md Metadata) CopyTo(out Metadata) { |  | ||||||
| 	for k, v := range md { |  | ||||||
| 		out[k] = copyOf(v) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Get obtains the values for a given key. |  | ||||||
| func (md Metadata) MustGet(k string) []string { |  | ||||||
| 	v, ok := md.Get(k) |  | ||||||
| 	if !ok { |  | ||||||
| 		panic("missing metadata key") |  | ||||||
| 	} |  | ||||||
| 	return v |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Get obtains the values for a given key. |  | ||||||
| func (md Metadata) Get(k string) ([]string, bool) { |  | ||||||
| 	v, ok := md[k] |  | ||||||
| 	if !ok { |  | ||||||
| 		v, ok = md[strings.ToLower(k)] |  | ||||||
| 	} |  | ||||||
| 	if !ok { |  | ||||||
| 		v, ok = md[textproto.CanonicalMIMEHeaderKey(k)] |  | ||||||
| 	} |  | ||||||
| 	return v, ok |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // MustGetJoined obtains the values for a given key |  | ||||||
| // with joined values with "," symbol |  | ||||||
| func (md Metadata) MustGetJoined(k string) string { |  | ||||||
| 	v, ok := md.GetJoined(k) |  | ||||||
| 	if !ok { |  | ||||||
| 		panic("missing metadata key") |  | ||||||
| 	} |  | ||||||
| 	return v |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // GetJoined obtains the values for a given key |  | ||||||
| // with joined values with "," symbol |  | ||||||
| func (md Metadata) GetJoined(k string) (string, bool) { |  | ||||||
| 	v, ok := md.Get(k) |  | ||||||
| 	if !ok { |  | ||||||
| 		return "", ok |  | ||||||
| 	} |  | ||||||
| 	return strings.Join(v, ","), true |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Set sets the value of a given key with a slice of values. |  | ||||||
| func (md Metadata) Add(key string, vals ...string) { |  | ||||||
| 	if len(vals) == 0 { |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
| 	md[key] = vals |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Set sets the value of a given key with a slice of values. |  | ||||||
| func (md Metadata) Set(kvs ...string) { |  | ||||||
| 	if len(kvs)%2 == 1 { |  | ||||||
| 		panic(fmt.Sprintf("metadata: Set got an odd number of input pairs for metadata: %d", len(kvs))) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	for i := 0; i < len(kvs); i += 2 { |  | ||||||
| 		md[kvs[i]] = append(md[kvs[i]], kvs[i+1]) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Append adds the values to key k, not overwriting what was already stored at |  | ||||||
| // that key. |  | ||||||
| func (md Metadata) Append(key string, vals ...string) { |  | ||||||
| 	if len(vals) == 0 { |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
| 	md[key] = append(md[key], vals...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Del removes the values for a given keys k. |  | ||||||
| func (md Metadata) Del(k ...string) { |  | ||||||
| 	for i := range k { |  | ||||||
| 		delete(md, k[i]) |  | ||||||
| 		delete(md, strings.ToLower(k[i])) |  | ||||||
| 		delete(md, textproto.CanonicalMIMEHeaderKey(k[i])) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Join joins any number of Metadatas into a single Metadata. |  | ||||||
| // |  | ||||||
| // The order of values for each key is determined by the order in which the Metadatas |  | ||||||
| // containing those values are presented to Join. |  | ||||||
| func Join(mds ...Metadata) Metadata { |  | ||||||
| 	out := Metadata{} |  | ||||||
| 	for _, Metadata := range mds { |  | ||||||
| 		for k, v := range Metadata { |  | ||||||
| 			out[k] = append(out[k], v...) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return out |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type ( |  | ||||||
| 	metadataIncomingKey struct{} |  | ||||||
| 	metadataOutgoingKey struct{} |  | ||||||
| 	metadataCurrentKey  struct{} |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // NewContext creates a new context with Metadata attached. Metadata must | // Metadata is our way of representing request headers internally. | ||||||
| // not be modified after calling this function. | // They're used at the RPC level and translate back and forth | ||||||
| func NewContext(ctx context.Context, md Metadata) context.Context { | // from Transport headers. | ||||||
| 	return context.WithValue(ctx, metadataCurrentKey{}, rawMetadata{md: md}) | type Metadata map[string]string | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewIncomingContext creates a new context with incoming Metadata attached. Metadata must |  | ||||||
| // not be modified after calling this function. |  | ||||||
| func NewIncomingContext(ctx context.Context, md Metadata) context.Context { |  | ||||||
| 	return context.WithValue(ctx, metadataIncomingKey{}, rawMetadata{md: md}) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewOutgoingContext creates a new context with outgoing Metadata attached. If used |  | ||||||
| // in conjunction with AppendOutgoingContext, NewOutgoingContext will |  | ||||||
| // overwrite any previously-appended metadata. Metadata must not be modified after |  | ||||||
| // calling this function. |  | ||||||
| func NewOutgoingContext(ctx context.Context, md Metadata) context.Context { |  | ||||||
| 	return context.WithValue(ctx, metadataOutgoingKey{}, rawMetadata{md: md}) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // AppendContext returns a new context with the provided kv merged |  | ||||||
| // with any existing metadata in the context. Please refer to the documentation |  | ||||||
| // of Pairs for a description of kv. |  | ||||||
| func AppendContext(ctx context.Context, kv ...string) context.Context { |  | ||||||
| 	if len(kv)%2 == 1 { |  | ||||||
| 		panic(fmt.Sprintf("metadata: AppendContext got an odd number of input pairs for metadata: %d", len(kv))) |  | ||||||
| 	} |  | ||||||
| 	md, _ := ctx.Value(metadataCurrentKey{}).(rawMetadata) |  | ||||||
| 	added := make([][]string, len(md.added)+1) |  | ||||||
| 	copy(added, md.added) |  | ||||||
| 	kvCopy := make([]string, 0, len(kv)) |  | ||||||
| 	for i := 0; i < len(kv); i += 2 { |  | ||||||
| 		kvCopy = append(kvCopy, strings.ToLower(kv[i]), kv[i+1]) |  | ||||||
| 	} |  | ||||||
| 	added[len(added)-1] = kvCopy |  | ||||||
| 	return context.WithValue(ctx, metadataCurrentKey{}, rawMetadata{md: md.md, added: added}) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // AppendIncomingContext returns a new context with the provided kv merged |  | ||||||
| // with any existing metadata in the context. Please refer to the documentation |  | ||||||
| // of Pairs for a description of kv. |  | ||||||
| func AppendIncomingContext(ctx context.Context, kv ...string) context.Context { |  | ||||||
| 	if len(kv)%2 == 1 { |  | ||||||
| 		panic(fmt.Sprintf("metadata: AppendIncomingContext got an odd number of input pairs for metadata: %d", len(kv))) |  | ||||||
| 	} |  | ||||||
| 	md, _ := ctx.Value(metadataIncomingKey{}).(rawMetadata) |  | ||||||
| 	added := make([][]string, len(md.added)+1) |  | ||||||
| 	copy(added, md.added) |  | ||||||
| 	kvCopy := make([]string, 0, len(kv)) |  | ||||||
| 	for i := 0; i < len(kv); i += 2 { |  | ||||||
| 		kvCopy = append(kvCopy, strings.ToLower(kv[i]), kv[i+1]) |  | ||||||
| 	} |  | ||||||
| 	added[len(added)-1] = kvCopy |  | ||||||
| 	return context.WithValue(ctx, metadataIncomingKey{}, rawMetadata{md: md.md, added: added}) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // AppendOutgoingContext returns a new context with the provided kv merged |  | ||||||
| // with any existing metadata in the context. Please refer to the documentation |  | ||||||
| // of Pairs for a description of kv. |  | ||||||
| func AppendOutgoingContext(ctx context.Context, kv ...string) context.Context { |  | ||||||
| 	if len(kv)%2 == 1 { |  | ||||||
| 		panic(fmt.Sprintf("metadata: AppendOutgoingContext got an odd number of input pairs for metadata: %d", len(kv))) |  | ||||||
| 	} |  | ||||||
| 	md, _ := ctx.Value(metadataOutgoingKey{}).(rawMetadata) |  | ||||||
| 	added := make([][]string, len(md.added)+1) |  | ||||||
| 	copy(added, md.added) |  | ||||||
| 	kvCopy := make([]string, 0, len(kv)) |  | ||||||
| 	for i := 0; i < len(kv); i += 2 { |  | ||||||
| 		kvCopy = append(kvCopy, strings.ToLower(kv[i]), kv[i+1]) |  | ||||||
| 	} |  | ||||||
| 	added[len(added)-1] = kvCopy |  | ||||||
| 	return context.WithValue(ctx, metadataOutgoingKey{}, rawMetadata{md: md.md, added: added}) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // FromContext returns the metadata in ctx if it exists. |  | ||||||
| func FromContext(ctx context.Context) (Metadata, bool) { |  | ||||||
| 	raw, ok := ctx.Value(metadataCurrentKey{}).(rawMetadata) |  | ||||||
| 	if !ok { |  | ||||||
| 		return nil, false |  | ||||||
| 	} |  | ||||||
| 	metadataSize := len(raw.md) |  | ||||||
| 	for i := range raw.added { |  | ||||||
| 		metadataSize += len(raw.added[i]) / 2 |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	out := make(Metadata, metadataSize) |  | ||||||
| 	for k, v := range raw.md { |  | ||||||
| 		out[k] = copyOf(v) |  | ||||||
| 	} |  | ||||||
| 	for _, added := range raw.added { |  | ||||||
| 		if len(added)%2 == 1 { |  | ||||||
| 			panic(fmt.Sprintf("metadata: FromContext got an odd number of input pairs for metadata: %d", len(added))) |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		for i := 0; i < len(added); i += 2 { |  | ||||||
| 			out[added[i]] = append(out[added[i]], added[i+1]) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return out, true |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // MustContext returns the metadata in ctx. |  | ||||||
| func MustContext(ctx context.Context) Metadata { |  | ||||||
| 	md, ok := FromContext(ctx) |  | ||||||
| 	if !ok { |  | ||||||
| 		panic("missing metadata") |  | ||||||
| 	} |  | ||||||
| 	return md |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // FromIncomingContext returns the incoming metadata in ctx if it exists. |  | ||||||
| func FromIncomingContext(ctx context.Context) (Metadata, bool) { |  | ||||||
| 	raw, ok := ctx.Value(metadataIncomingKey{}).(rawMetadata) |  | ||||||
| 	if !ok { |  | ||||||
| 		return nil, false |  | ||||||
| 	} |  | ||||||
| 	metadataSize := len(raw.md) |  | ||||||
| 	for i := range raw.added { |  | ||||||
| 		metadataSize += len(raw.added[i]) / 2 |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	out := make(Metadata, metadataSize) |  | ||||||
| 	for k, v := range raw.md { |  | ||||||
| 		out[k] = copyOf(v) |  | ||||||
| 	} |  | ||||||
| 	for _, added := range raw.added { |  | ||||||
| 		if len(added)%2 == 1 { |  | ||||||
| 			panic(fmt.Sprintf("metadata: FromIncomingContext got an odd number of input pairs for metadata: %d", len(added))) |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		for i := 0; i < len(added); i += 2 { |  | ||||||
| 			out[added[i]] = append(out[added[i]], added[i+1]) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return out, true |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // MustIncomingContext returns the incoming metadata in ctx. |  | ||||||
| func MustIncomingContext(ctx context.Context) Metadata { |  | ||||||
| 	md, ok := FromIncomingContext(ctx) |  | ||||||
| 	if !ok { |  | ||||||
| 		panic("missing metadata") |  | ||||||
| 	} |  | ||||||
| 	return md |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // ValueFromIncomingContext returns the metadata value corresponding to the metadata |  | ||||||
| // key from the incoming metadata if it exists. Keys are matched in a case insensitive |  | ||||||
| // manner. |  | ||||||
| func ValueFromIncomingContext(ctx context.Context, key string) []string { |  | ||||||
| 	raw, ok := ctx.Value(metadataIncomingKey{}).(rawMetadata) |  | ||||||
| 	if !ok { |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if v, ok := raw.md[key]; ok { |  | ||||||
| 		return copyOf(v) |  | ||||||
| 	} |  | ||||||
| 	for k, v := range raw.md { |  | ||||||
| 		// Case insensitive comparison: Metadata is a map, and there's no guarantee |  | ||||||
| 		// that the Metadata attached to the context is created using our helper |  | ||||||
| 		// functions. |  | ||||||
| 		if strings.EqualFold(k, key) { |  | ||||||
| 			return copyOf(v) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // ValueFromCurrentContext returns the metadata value corresponding to the metadata |  | ||||||
| // key from the incoming metadata if it exists. Keys are matched in a case insensitive |  | ||||||
| // manner. |  | ||||||
| func ValueFromCurrentContext(ctx context.Context, key string) []string { |  | ||||||
| 	md, ok := ctx.Value(metadataCurrentKey{}).(rawMetadata) |  | ||||||
| 	if !ok { |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if v, ok := md.md[key]; ok { |  | ||||||
| 		return copyOf(v) |  | ||||||
| 	} |  | ||||||
| 	for k, v := range md.md { |  | ||||||
| 		// Case insensitive comparison: Metadata is a map, and there's no guarantee |  | ||||||
| 		// that the Metadata attached to the context is created using our helper |  | ||||||
| 		// functions. |  | ||||||
| 		if strings.EqualFold(k, key) { |  | ||||||
| 			return copyOf(v) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // MustOutgoingContext returns the outgoing metadata in ctx. |  | ||||||
| func MustOutgoingContext(ctx context.Context) Metadata { |  | ||||||
| 	md, ok := FromOutgoingContext(ctx) |  | ||||||
| 	if !ok { |  | ||||||
| 		panic("missing metadata") |  | ||||||
| 	} |  | ||||||
| 	return md |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // ValueFromOutgoingContext returns the metadata value corresponding to the metadata |  | ||||||
| // key from the incoming metadata if it exists. Keys are matched in a case insensitive |  | ||||||
| // manner. |  | ||||||
| func ValueFromOutgoingContext(ctx context.Context, key string) []string { |  | ||||||
| 	md, ok := ctx.Value(metadataOutgoingKey{}).(rawMetadata) |  | ||||||
| 	if !ok { |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if v, ok := md.md[key]; ok { |  | ||||||
| 		return copyOf(v) |  | ||||||
| 	} |  | ||||||
| 	for k, v := range md.md { |  | ||||||
| 		// Case insensitive comparison: Metadata is a map, and there's no guarantee |  | ||||||
| 		// that the Metadata attached to the context is created using our helper |  | ||||||
| 		// functions. |  | ||||||
| 		if strings.EqualFold(k, key) { |  | ||||||
| 			return copyOf(v) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func copyOf(v []string) []string { |  | ||||||
| 	vals := make([]string, len(v)) |  | ||||||
| 	copy(vals, v) |  | ||||||
| 	return vals |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // FromOutgoingContext returns the outgoing metadata in ctx if it exists. |  | ||||||
| func FromOutgoingContext(ctx context.Context) (Metadata, bool) { |  | ||||||
| 	raw, ok := ctx.Value(metadataOutgoingKey{}).(rawMetadata) |  | ||||||
| 	if !ok { |  | ||||||
| 		return nil, false |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	metadataSize := len(raw.md) |  | ||||||
| 	for i := range raw.added { |  | ||||||
| 		metadataSize += len(raw.added[i]) / 2 |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	out := make(Metadata, metadataSize) |  | ||||||
| 	for k, v := range raw.md { |  | ||||||
| 		out[k] = copyOf(v) |  | ||||||
| 	} |  | ||||||
| 	for _, added := range raw.added { |  | ||||||
| 		if len(added)%2 == 1 { |  | ||||||
| 			panic(fmt.Sprintf("metadata: FromOutgoingContext got an odd number of input pairs for metadata: %d", len(added))) |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		for i := 0; i < len(added); i += 2 { |  | ||||||
| 			out[added[i]] = append(out[added[i]], added[i+1]) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return out, ok |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type rawMetadata struct { | type rawMetadata struct { | ||||||
| 	md Metadata | 	md Metadata | ||||||
| 	added [][]string |  | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // defaultMetadataSize used when need to init new Metadata | ||||||
|  | var defaultMetadataSize = 2 | ||||||
|  |  | ||||||
| // Iterator used to iterate over metadata with order | // Iterator used to iterate over metadata with order | ||||||
| type Iterator struct { | type Iterator struct { | ||||||
| 	md   Metadata | 	md   Metadata | ||||||
| @@ -475,18 +45,14 @@ type Iterator struct { | |||||||
| } | } | ||||||
|  |  | ||||||
| // Next advance iterator to next element | // Next advance iterator to next element | ||||||
| func (iter *Iterator) Next(k *string, v *[]string) bool { | func (iter *Iterator) Next(k, v *string) bool { | ||||||
| 	if iter.cur+1 > iter.cnt { | 	if iter.cur+1 > iter.cnt { | ||||||
| 		return false | 		return false | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if k != nil && v != nil { |  | ||||||
| 	*k = iter.keys[iter.cur] | 	*k = iter.keys[iter.cur] | ||||||
| 		vv := iter.md[*k] | 	*v = iter.md[*k] | ||||||
| 		*v = make([]string, len(vv)) |  | ||||||
| 		copy(*v, vv) |  | ||||||
| 	iter.cur++ | 	iter.cur++ | ||||||
| 	} |  | ||||||
| 	return true | 	return true | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -500,3 +66,102 @@ func (md Metadata) Iterator() *Iterator { | |||||||
| 	sort.Strings(iter.keys) | 	sort.Strings(iter.keys) | ||||||
| 	return iter | 	return iter | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (md Metadata) MustGet(key string) string { | ||||||
|  | 	val, ok := md.Get(key) | ||||||
|  | 	if !ok { | ||||||
|  | 		panic("missing metadata key") | ||||||
|  | 	} | ||||||
|  | 	return val | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Get returns value from metadata by key | ||||||
|  | func (md Metadata) Get(key string) (string, bool) { | ||||||
|  | 	// fast path | ||||||
|  | 	val, ok := md[key] | ||||||
|  | 	if !ok { | ||||||
|  | 		// slow path | ||||||
|  | 		val, ok = md[textproto.CanonicalMIMEHeaderKey(key)] | ||||||
|  | 		if !ok { | ||||||
|  | 			val, ok = md[strings.ToLower(key)] | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return val, ok | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Set is used to store value in metadata | ||||||
|  | func (md Metadata) Set(kv ...string) { | ||||||
|  | 	if len(kv)%2 == 1 { | ||||||
|  | 		kv = kv[:len(kv)-1] | ||||||
|  | 	} | ||||||
|  | 	for idx := 0; idx < len(kv); idx += 2 { | ||||||
|  | 		md[textproto.CanonicalMIMEHeaderKey(kv[idx])] = kv[idx+1] | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Del is used to remove value from metadata | ||||||
|  | func (md Metadata) Del(keys ...string) { | ||||||
|  | 	for _, key := range keys { | ||||||
|  | 		// fast path | ||||||
|  | 		delete(md, key) | ||||||
|  | 		// slow path | ||||||
|  | 		delete(md, textproto.CanonicalMIMEHeaderKey(key)) | ||||||
|  | 		// very slow path | ||||||
|  | 		delete(md, strings.ToLower(key)) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Copy makes a copy of the metadata | ||||||
|  | func (md Metadata) CopyTo(dst Metadata) { | ||||||
|  | 	for k, v := range md { | ||||||
|  | 		dst[k] = v | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Copy makes a copy of the metadata | ||||||
|  | func Copy(md Metadata, exclude ...string) Metadata { | ||||||
|  | 	nmd := New(len(md)) | ||||||
|  | 	for k, v := range md { | ||||||
|  | 		nmd[k] = v | ||||||
|  | 	} | ||||||
|  | 	nmd.Del(exclude...) | ||||||
|  | 	return nmd | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New return new sized metadata | ||||||
|  | func New(size int) Metadata { | ||||||
|  | 	if size == 0 { | ||||||
|  | 		size = defaultMetadataSize | ||||||
|  | 	} | ||||||
|  | 	return make(Metadata, size) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Merge merges metadata to existing metadata, overwriting if specified | ||||||
|  | func Merge(omd Metadata, mmd Metadata, overwrite bool) Metadata { | ||||||
|  | 	var ok bool | ||||||
|  | 	nmd := Copy(omd) | ||||||
|  | 	for key, val := range mmd { | ||||||
|  | 		_, ok = nmd[key] | ||||||
|  | 		switch { | ||||||
|  | 		case ok && !overwrite: | ||||||
|  | 			continue | ||||||
|  | 		case val != "": | ||||||
|  | 			nmd[key] = val | ||||||
|  | 		case ok && val == "": | ||||||
|  | 			nmd.Del(key) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return nmd | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Pairs from which metadata created | ||||||
|  | func Pairs(kv ...string) (Metadata, bool) { | ||||||
|  | 	if len(kv)%2 == 1 { | ||||||
|  | 		return nil, false | ||||||
|  | 	} | ||||||
|  | 	md := New(len(kv) / 2) | ||||||
|  | 	for idx := 0; idx < len(kv); idx += 2 { | ||||||
|  | 		md[kv[idx]] = kv[idx+1] | ||||||
|  | 	} | ||||||
|  | 	return md, true | ||||||
|  | } | ||||||
|   | |||||||
| @@ -5,21 +5,10 @@ import ( | |||||||
| 	"testing" | 	"testing" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| /* |  | ||||||
| func TestAppendOutgoingContextModify(t *testing.T) { |  | ||||||
| 	md := Pairs("key1", "val1") |  | ||||||
| 	ctx := NewOutgoingContext(context.TODO(), md) |  | ||||||
| 	nctx := AppendOutgoingContext(ctx, "key1", "val3", "key2", "val2") |  | ||||||
| 	_ = nctx |  | ||||||
| 	omd := MustOutgoingContext(nctx) |  | ||||||
| 	fmt.Printf("%#+v\n", omd) |  | ||||||
| } |  | ||||||
| */ |  | ||||||
|  |  | ||||||
| func TestLowercase(t *testing.T) { | func TestLowercase(t *testing.T) { | ||||||
| 	md := New(1) | 	md := New(1) | ||||||
| 	md["x-request-id"] = []string{"12345"} | 	md["x-request-id"] = "12345" | ||||||
| 	v, ok := md.GetJoined("X-Request-Id") | 	v, ok := md.Get("X-Request-Id") | ||||||
| 	if !ok || v == "" { | 	if !ok || v == "" { | ||||||
| 		t.Fatalf("metadata invalid %#+v", md) | 		t.Fatalf("metadata invalid %#+v", md) | ||||||
| 	} | 	} | ||||||
| @@ -49,12 +38,15 @@ func TestMultipleUsage(t *testing.T) { | |||||||
|  |  | ||||||
| func TestMetadataSetMultiple(t *testing.T) { | func TestMetadataSetMultiple(t *testing.T) { | ||||||
| 	md := New(4) | 	md := New(4) | ||||||
| 	md.Set("key1", "val1", "key2", "val2") | 	md.Set("key1", "val1", "key2", "val2", "key3") | ||||||
|  |  | ||||||
| 	if v, ok := md.GetJoined("key1"); !ok || v != "val1" { | 	if v, ok := md.Get("key1"); !ok || v != "val1" { | ||||||
| 		t.Fatalf("invalid kv %#+v", md) | 		t.Fatalf("invalid kv %#+v", md) | ||||||
| 	} | 	} | ||||||
| 	if v, ok := md.GetJoined("key2"); !ok || v != "val2" { | 	if v, ok := md.Get("key2"); !ok || v != "val2" { | ||||||
|  | 		t.Fatalf("invalid kv %#+v", md) | ||||||
|  | 	} | ||||||
|  | 	if _, ok := md.Get("key3"); ok { | ||||||
| 		t.Fatalf("invalid kv %#+v", md) | 		t.Fatalf("invalid kv %#+v", md) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| @@ -72,12 +64,22 @@ func TestAppend(t *testing.T) { | |||||||
| } | } | ||||||
|  |  | ||||||
| func TestPairs(t *testing.T) { | func TestPairs(t *testing.T) { | ||||||
| 	md := Pairs("key1", "val1", "key2", "val2") | 	md, ok := Pairs("key1", "val1", "key2", "val2") | ||||||
| 	if _, ok := md.Get("key1"); !ok { | 	if !ok { | ||||||
|  | 		t.Fatal("odd number of kv") | ||||||
|  | 	} | ||||||
|  | 	if _, ok = md.Get("key1"); !ok { | ||||||
| 		t.Fatal("key1 not found") | 		t.Fatal("key1 not found") | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func testCtx(ctx context.Context) { | ||||||
|  | 	md := New(2) | ||||||
|  | 	md.Set("Key1", "Val1_new") | ||||||
|  | 	md.Set("Key3", "Val3") | ||||||
|  | 	SetOutgoingContext(ctx, md) | ||||||
|  | } | ||||||
|  |  | ||||||
| func TestPassing(t *testing.T) { | func TestPassing(t *testing.T) { | ||||||
| 	ctx := context.TODO() | 	ctx := context.TODO() | ||||||
| 	md1 := New(2) | 	md1 := New(2) | ||||||
| @@ -85,62 +87,69 @@ func TestPassing(t *testing.T) { | |||||||
| 	md1.Set("Key2", "Val2") | 	md1.Set("Key2", "Val2") | ||||||
|  |  | ||||||
| 	ctx = NewIncomingContext(ctx, md1) | 	ctx = NewIncomingContext(ctx, md1) | ||||||
|  | 	testCtx(ctx) | ||||||
| 	_, ok := FromOutgoingContext(ctx) | 	_, ok := FromOutgoingContext(ctx) | ||||||
| 	if ok { | 	if ok { | ||||||
| 		t.Fatalf("create outgoing context") | 		t.Fatalf("create outgoing context") | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	ctx = NewOutgoingContext(ctx, md1) | 	ctx = NewOutgoingContext(ctx, New(1)) | ||||||
|  | 	testCtx(ctx) | ||||||
| 	md, ok := FromOutgoingContext(ctx) | 	md, ok := FromOutgoingContext(ctx) | ||||||
| 	if !ok { | 	if !ok { | ||||||
| 		t.Fatalf("missing metadata from outgoing context") | 		t.Fatalf("missing metadata from outgoing context") | ||||||
| 	} | 	} | ||||||
| 	if v, ok := md.Get("Key1"); !ok || v[0] != "Val1" { | 	if v, ok := md.Get("Key1"); !ok || v != "Val1_new" { | ||||||
| 		t.Fatalf("invalid metadata value %#+v", md) | 		t.Fatalf("invalid metadata value %#+v", md) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func TestIterator(t *testing.T) { | func TestMerge(t *testing.T) { | ||||||
| 	md := Pairs( | 	omd := Metadata{ | ||||||
| 		"1Last", "last", | 		"key1": "val1", | ||||||
| 		"2First", "first", | 	} | ||||||
| 		"3Second", "second", | 	mmd := Metadata{ | ||||||
| 	) | 		"key2": "val2", | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	nmd := Merge(omd, mmd, true) | ||||||
|  | 	if len(nmd) != 2 { | ||||||
|  | 		t.Fatalf("merge failed: %v", nmd) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestIterator(_ *testing.T) { | ||||||
|  | 	md := Metadata{ | ||||||
|  | 		"1Last":   "last", | ||||||
|  | 		"2First":  "first", | ||||||
|  | 		"3Second": "second", | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	iter := md.Iterator() | 	iter := md.Iterator() | ||||||
| 	var k string | 	var k, v string | ||||||
| 	var v []string |  | ||||||
| 	chk := New(3) |  | ||||||
| 	for iter.Next(&k, &v) { |  | ||||||
| 		chk[k] = v |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	for k, v := range chk { | 	for iter.Next(&k, &v) { | ||||||
| 		if cv, ok := md[k]; !ok || len(cv) != len(v) || cv[0] != v[0] { | 		// fmt.Printf("k: %s, v: %s\n", k, v) | ||||||
| 			t.Fatalf("XXXX %#+v %#+v", chk, md) |  | ||||||
| 		} |  | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func TestMedataCanonicalKey(t *testing.T) { | func TestMedataCanonicalKey(t *testing.T) { | ||||||
| 	md := New(1) | 	md := New(1) | ||||||
| 	md.Set("x-request-id", "12345") | 	md.Set("x-request-id", "12345") | ||||||
| 	v, ok := md.GetJoined("x-request-id") | 	v, ok := md.Get("x-request-id") | ||||||
| 	if !ok { | 	if !ok { | ||||||
| 		t.Fatalf("failed to get x-request-id") | 		t.Fatalf("failed to get x-request-id") | ||||||
| 	} else if v != "12345" { | 	} else if v != "12345" { | ||||||
| 		t.Fatalf("invalid metadata value: %s != %s", "12345", v) | 		t.Fatalf("invalid metadata value: %s != %s", "12345", v) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	v, ok = md.GetJoined("X-Request-Id") | 	v, ok = md.Get("X-Request-Id") | ||||||
| 	if !ok { | 	if !ok { | ||||||
| 		t.Fatalf("failed to get x-request-id") | 		t.Fatalf("failed to get x-request-id") | ||||||
| 	} else if v != "12345" { | 	} else if v != "12345" { | ||||||
| 		t.Fatalf("invalid metadata value: %s != %s", "12345", v) | 		t.Fatalf("invalid metadata value: %s != %s", "12345", v) | ||||||
| 	} | 	} | ||||||
| 	v, ok = md.GetJoined("X-Request-ID") | 	v, ok = md.Get("X-Request-ID") | ||||||
| 	if !ok { | 	if !ok { | ||||||
| 		t.Fatalf("failed to get x-request-id") | 		t.Fatalf("failed to get x-request-id") | ||||||
| 	} else if v != "12345" { | 	} else if v != "12345" { | ||||||
| @@ -153,7 +162,7 @@ func TestMetadataSet(t *testing.T) { | |||||||
|  |  | ||||||
| 	md.Set("Key", "val") | 	md.Set("Key", "val") | ||||||
|  |  | ||||||
| 	val, ok := md.GetJoined("Key") | 	val, ok := md.Get("Key") | ||||||
| 	if !ok { | 	if !ok { | ||||||
| 		t.Fatal("key Key not found") | 		t.Fatal("key Key not found") | ||||||
| 	} | 	} | ||||||
| @@ -164,8 +173,8 @@ func TestMetadataSet(t *testing.T) { | |||||||
|  |  | ||||||
| func TestMetadataDelete(t *testing.T) { | func TestMetadataDelete(t *testing.T) { | ||||||
| 	md := Metadata{ | 	md := Metadata{ | ||||||
| 		"Foo": []string{"bar"}, | 		"Foo": "bar", | ||||||
| 		"Baz": []string{"empty"}, | 		"Baz": "empty", | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	md.Del("Baz") | 	md.Del("Baz") | ||||||
| @@ -175,16 +184,25 @@ func TestMetadataDelete(t *testing.T) { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func TestNilContext(t *testing.T) { | ||||||
|  | 	var ctx context.Context | ||||||
|  |  | ||||||
|  | 	_, ok := FromContext(ctx) | ||||||
|  | 	if ok { | ||||||
|  | 		t.Fatal("nil context") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| func TestMetadataCopy(t *testing.T) { | func TestMetadataCopy(t *testing.T) { | ||||||
| 	md := Metadata{ | 	md := Metadata{ | ||||||
| 		"Foo": []string{"bar"}, | 		"Foo": "bar", | ||||||
| 		"Bar": []string{"baz"}, | 		"Bar": "baz", | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	cp := Copy(md) | 	cp := Copy(md) | ||||||
|  |  | ||||||
| 	for k, v := range md { | 	for k, v := range md { | ||||||
| 		if cv := cp[k]; cv[0] != v[0] { | 		if cv := cp[k]; cv != v { | ||||||
| 			t.Fatalf("Got %s:%s for %s:%s", k, cv, k, v) | 			t.Fatalf("Got %s:%s for %s:%s", k, cv, k, v) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| @@ -192,7 +210,7 @@ func TestMetadataCopy(t *testing.T) { | |||||||
|  |  | ||||||
| func TestMetadataContext(t *testing.T) { | func TestMetadataContext(t *testing.T) { | ||||||
| 	md := Metadata{ | 	md := Metadata{ | ||||||
| 		"Foo": []string{"bar"}, | 		"Foo": "bar", | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	ctx := NewContext(context.TODO(), md) | 	ctx := NewContext(context.TODO(), md) | ||||||
| @@ -202,7 +220,7 @@ func TestMetadataContext(t *testing.T) { | |||||||
| 		t.Errorf("Unexpected error retrieving metadata, got %t", ok) | 		t.Errorf("Unexpected error retrieving metadata, got %t", ok) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if emd["Foo"][0] != md["Foo"][0] { | 	if emd["Foo"] != md["Foo"] { | ||||||
| 		t.Errorf("Expected key: %s val: %s, got key: %s val: %s", "Foo", md["Foo"], "Foo", emd["Foo"]) | 		t.Errorf("Expected key: %s val: %s, got key: %s val: %s", "Foo", md["Foo"], "Foo", emd["Foo"]) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -211,88 +229,13 @@ func TestMetadataContext(t *testing.T) { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func TestFromContext(t *testing.T) { | func TestCopy(t *testing.T) { | ||||||
| 	ctx := context.WithValue(context.TODO(), metadataCurrentKey{}, rawMetadata{md: New(0)}) | 	md := New(2) | ||||||
|  | 	md.Set("key1", "val1", "key2", "val2") | ||||||
| 	c, ok := FromContext(ctx) | 	nmd := Copy(md, "key2") | ||||||
| 	if c == nil || !ok { | 	if len(nmd) != 1 { | ||||||
| 		t.Fatal("FromContext not works") | 		t.Fatal("Copy exclude not works") | ||||||
| 	} | 	} else if nmd["Key1"] != "val1" { | ||||||
| } | 		t.Fatal("Copy exclude not works") | ||||||
|  |  | ||||||
| func TestNewContext(t *testing.T) { |  | ||||||
| 	ctx := NewContext(context.TODO(), New(0)) |  | ||||||
|  |  | ||||||
| 	c, ok := FromContext(ctx) |  | ||||||
| 	if c == nil || !ok { |  | ||||||
| 		t.Fatal("NewContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestFromIncomingContext(t *testing.T) { |  | ||||||
| 	ctx := context.WithValue(context.TODO(), metadataIncomingKey{}, rawMetadata{md: New(0)}) |  | ||||||
|  |  | ||||||
| 	c, ok := FromIncomingContext(ctx) |  | ||||||
| 	if c == nil || !ok { |  | ||||||
| 		t.Fatal("FromIncomingContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestFromOutgoingContext(t *testing.T) { |  | ||||||
| 	ctx := context.WithValue(context.TODO(), metadataOutgoingKey{}, rawMetadata{md: New(0)}) |  | ||||||
|  |  | ||||||
| 	c, ok := FromOutgoingContext(ctx) |  | ||||||
| 	if c == nil || !ok { |  | ||||||
| 		t.Fatal("FromOutgoingContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestNewIncomingContext(t *testing.T) { |  | ||||||
| 	md := New(1) |  | ||||||
| 	md.Set("key", "val") |  | ||||||
| 	ctx := NewIncomingContext(context.TODO(), md) |  | ||||||
|  |  | ||||||
| 	c, ok := FromIncomingContext(ctx) |  | ||||||
| 	if c == nil || !ok { |  | ||||||
| 		t.Fatal("NewIncomingContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestNewOutgoingContext(t *testing.T) { |  | ||||||
| 	md := New(1) |  | ||||||
| 	md.Set("key", "val") |  | ||||||
| 	ctx := NewOutgoingContext(context.TODO(), md) |  | ||||||
|  |  | ||||||
| 	c, ok := FromOutgoingContext(ctx) |  | ||||||
| 	if c == nil || !ok { |  | ||||||
| 		t.Fatal("NewOutgoingContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestAppendIncomingContext(t *testing.T) { |  | ||||||
| 	md := New(1) |  | ||||||
| 	md.Set("key1", "val1") |  | ||||||
| 	ctx := AppendIncomingContext(context.TODO(), "key2", "val2") |  | ||||||
|  |  | ||||||
| 	nmd, ok := FromIncomingContext(ctx) |  | ||||||
| 	if nmd == nil || !ok { |  | ||||||
| 		t.Fatal("AppendIncomingContext not works") |  | ||||||
| 	} |  | ||||||
| 	if v, ok := nmd.GetJoined("key2"); !ok || v != "val2" { |  | ||||||
| 		t.Fatal("AppendIncomingContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestAppendOutgoingContext(t *testing.T) { |  | ||||||
| 	md := New(1) |  | ||||||
| 	md.Set("key1", "val1") |  | ||||||
| 	ctx := AppendOutgoingContext(context.TODO(), "key2", "val2") |  | ||||||
|  |  | ||||||
| 	nmd, ok := FromOutgoingContext(ctx) |  | ||||||
| 	if nmd == nil || !ok { |  | ||||||
| 		t.Fatal("AppendOutgoingContext not works") |  | ||||||
| 	} |  | ||||||
| 	if v, ok := nmd.GetJoined("key2"); !ok || v != "val2" { |  | ||||||
| 		t.Fatal("AppendOutgoingContext not works") |  | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										30
									
								
								micro.go
									
									
									
									
									
								
							
							
						
						
									
										30
									
								
								micro.go
									
									
									
									
									
								
							| @@ -3,21 +3,21 @@ package micro | |||||||
| import ( | import ( | ||||||
| 	"reflect" | 	"reflect" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/broker" | 	"go.unistack.org/micro/v3/broker" | ||||||
| 	"go.unistack.org/micro/v4/client" | 	"go.unistack.org/micro/v3/client" | ||||||
| 	"go.unistack.org/micro/v4/codec" | 	"go.unistack.org/micro/v3/codec" | ||||||
| 	"go.unistack.org/micro/v4/flow" | 	"go.unistack.org/micro/v3/flow" | ||||||
| 	"go.unistack.org/micro/v4/fsm" | 	"go.unistack.org/micro/v3/fsm" | ||||||
| 	"go.unistack.org/micro/v4/logger" | 	"go.unistack.org/micro/v3/logger" | ||||||
| 	"go.unistack.org/micro/v4/meter" | 	"go.unistack.org/micro/v3/meter" | ||||||
| 	"go.unistack.org/micro/v4/register" | 	"go.unistack.org/micro/v3/register" | ||||||
| 	"go.unistack.org/micro/v4/resolver" | 	"go.unistack.org/micro/v3/resolver" | ||||||
| 	"go.unistack.org/micro/v4/router" | 	"go.unistack.org/micro/v3/router" | ||||||
| 	"go.unistack.org/micro/v4/selector" | 	"go.unistack.org/micro/v3/selector" | ||||||
| 	"go.unistack.org/micro/v4/server" | 	"go.unistack.org/micro/v3/server" | ||||||
| 	"go.unistack.org/micro/v4/store" | 	"go.unistack.org/micro/v3/store" | ||||||
| 	"go.unistack.org/micro/v4/sync" | 	"go.unistack.org/micro/v3/sync" | ||||||
| 	"go.unistack.org/micro/v4/tracer" | 	"go.unistack.org/micro/v3/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| func As(b any, target any) bool { | func As(b any, target any) bool { | ||||||
|   | |||||||
| @@ -6,9 +6,8 @@ import ( | |||||||
| 	"reflect" | 	"reflect" | ||||||
| 	"testing" | 	"testing" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/broker" | 	"go.unistack.org/micro/v3/broker" | ||||||
| 	"go.unistack.org/micro/v4/fsm" | 	"go.unistack.org/micro/v3/fsm" | ||||||
| 	"go.unistack.org/micro/v4/metadata" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| func TestAs(t *testing.T) { | func TestAs(t *testing.T) { | ||||||
| @@ -62,8 +61,6 @@ func TestAs(t *testing.T) { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| var _ broker.Broker = (*bro)(nil) |  | ||||||
|  |  | ||||||
| type bro struct { | type bro struct { | ||||||
| 	name string | 	name string | ||||||
| } | } | ||||||
| @@ -90,18 +87,23 @@ func (p *bro) Connect(_ context.Context) error { return nil } | |||||||
| // Disconnect disconnect from broker | // Disconnect disconnect from broker | ||||||
| func (p *bro) Disconnect(_ context.Context) error { return nil } | func (p *bro) Disconnect(_ context.Context) error { return nil } | ||||||
|  |  | ||||||
| // NewMessage creates new message |  | ||||||
| func (p *bro) NewMessage(_ context.Context, _ metadata.Metadata, _ interface{}, _ ...broker.PublishOption) (broker.Message, error) { |  | ||||||
| 	return nil, nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Publish message, msg can be single broker.Message or []broker.Message | // Publish message, msg can be single broker.Message or []broker.Message | ||||||
| func (p *bro) Publish(_ context.Context, _ string, _ ...broker.Message) error { | func (p *bro) Publish(_ context.Context, _ string, _ *broker.Message, _ ...broker.PublishOption) error { | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // BatchPublish messages to broker with multiple topics | ||||||
|  | func (p *bro) BatchPublish(_ context.Context, _ []*broker.Message, _ ...broker.PublishOption) error { | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // BatchSubscribe subscribes to topic messages via handler | ||||||
|  | func (p *bro) BatchSubscribe(_ context.Context, _ string, _ broker.BatchHandler, _ ...broker.SubscribeOption) (broker.Subscriber, error) { | ||||||
|  | 	return nil, nil | ||||||
|  | } | ||||||
|  |  | ||||||
| // Subscribe subscribes to topic message via handler | // Subscribe subscribes to topic message via handler | ||||||
| func (p *bro) Subscribe(_ context.Context, _ string, _ interface{}, _ ...broker.SubscribeOption) (broker.Subscriber, error) { | func (p *bro) Subscribe(_ context.Context, _ string, _ broker.Handler, _ ...broker.SubscribeOption) (broker.Subscriber, error) { | ||||||
| 	return nil, nil | 	return nil, nil | ||||||
| } | } | ||||||
|  |  | ||||||
|   | |||||||
							
								
								
									
										55
									
								
								network/network.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										55
									
								
								network/network.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,55 @@ | |||||||
|  | // Package network is for creating internetworks | ||||||
|  | package network | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"go.unistack.org/micro/v3/client" | ||||||
|  | 	"go.unistack.org/micro/v3/server" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Error is network node errors | ||||||
|  | type Error interface { | ||||||
|  | 	// Count is current count of errors | ||||||
|  | 	Count() int | ||||||
|  | 	// Msg is last error message | ||||||
|  | 	Msg() string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Status is node status | ||||||
|  | type Status interface { | ||||||
|  | 	// Error reports error status | ||||||
|  | 	Error() Error | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Node is network node | ||||||
|  | type Node interface { | ||||||
|  | 	// Id is node id | ||||||
|  | 	Id() string | ||||||
|  | 	// Address is node bind address | ||||||
|  | 	Address() string | ||||||
|  | 	// Peers returns node peers | ||||||
|  | 	Peers() []Node | ||||||
|  | 	// Network is the network node is in | ||||||
|  | 	Network() Network | ||||||
|  | 	// Status returns node status | ||||||
|  | 	Status() Status | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Network is micro network | ||||||
|  | type Network interface { | ||||||
|  | 	// Node is network node | ||||||
|  | 	Node | ||||||
|  | 	// Initialise options | ||||||
|  | 	Init(...Option) error | ||||||
|  | 	// Options returns the network options | ||||||
|  | 	Options() Options | ||||||
|  | 	// Name of the network | ||||||
|  | 	Name() string | ||||||
|  | 	// Connect starts the resolver and tunnel server | ||||||
|  | 	Connect() error | ||||||
|  | 	// Close stops the tunnel and resolving | ||||||
|  | 	Close() error | ||||||
|  | 	// Client is micro client | ||||||
|  | 	Client() client.Client | ||||||
|  | 	// Server is micro server | ||||||
|  | 	Server() server.Server | ||||||
|  | } | ||||||
							
								
								
									
										135
									
								
								network/options.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										135
									
								
								network/options.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,135 @@ | |||||||
|  | package network | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"go.unistack.org/micro/v3/logger" | ||||||
|  | 	"go.unistack.org/micro/v3/meter" | ||||||
|  | 	"go.unistack.org/micro/v3/network/tunnel" | ||||||
|  | 	"go.unistack.org/micro/v3/proxy" | ||||||
|  | 	"go.unistack.org/micro/v3/router" | ||||||
|  | 	"go.unistack.org/micro/v3/tracer" | ||||||
|  | 	"go.unistack.org/micro/v3/util/id" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Option func | ||||||
|  | type Option func(*Options) | ||||||
|  |  | ||||||
|  | // Options configure network | ||||||
|  | type Options struct { | ||||||
|  | 	// Router used for routing | ||||||
|  | 	Router router.Router | ||||||
|  | 	// Proxy holds the proxy | ||||||
|  | 	Proxy proxy.Proxy | ||||||
|  | 	// Logger used for logging | ||||||
|  | 	Logger logger.Logger | ||||||
|  | 	// Meter used for metrics | ||||||
|  | 	Meter meter.Meter | ||||||
|  | 	// Tracer used for tracing | ||||||
|  | 	Tracer tracer.Tracer | ||||||
|  | 	// Tunnel used for transfer data | ||||||
|  | 	Tunnel tunnel.Tunnel | ||||||
|  | 	// ID of the node | ||||||
|  | 	ID string | ||||||
|  | 	// Name of the network | ||||||
|  | 	Name string | ||||||
|  | 	// Address to bind to | ||||||
|  | 	Address string | ||||||
|  | 	// Advertise sets the address to advertise | ||||||
|  | 	Advertise string | ||||||
|  | 	// Nodes is a list of nodes to connect to | ||||||
|  | 	Nodes []string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ID sets the id of the network node | ||||||
|  | func ID(id string) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.ID = id | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Name sets the network name | ||||||
|  | func Name(n string) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Name = n | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Address sets the network address | ||||||
|  | func Address(a string) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Address = a | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Advertise sets the address to advertise | ||||||
|  | func Advertise(a string) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Advertise = a | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Nodes is a list of nodes to connect to | ||||||
|  | func Nodes(n ...string) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Nodes = n | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Tunnel sets the network tunnel | ||||||
|  | func Tunnel(t tunnel.Tunnel) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Tunnel = t | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Router sets the network router | ||||||
|  | func Router(r router.Router) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Router = r | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Proxy sets the network proxy | ||||||
|  | func Proxy(p proxy.Proxy) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Proxy = p | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Logger sets the network logger | ||||||
|  | func Logger(l logger.Logger) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Logger = l | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Meter sets the meter | ||||||
|  | func Meter(m meter.Meter) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Meter = m | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Tracer to be used for tracing | ||||||
|  | func Tracer(t tracer.Tracer) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Tracer = t | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewOptions returns network default options | ||||||
|  | func NewOptions(opts ...Option) Options { | ||||||
|  | 	options := Options{ | ||||||
|  | 		ID:      id.MustNew(), | ||||||
|  | 		Name:    "go.micro", | ||||||
|  | 		Address: ":0", | ||||||
|  | 		Logger:  logger.DefaultLogger, | ||||||
|  | 		Meter:   meter.DefaultMeter, | ||||||
|  | 		Tracer:  tracer.DefaultTracer, | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for _, o := range opts { | ||||||
|  | 		o(&options) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return options | ||||||
|  | } | ||||||
							
								
								
									
										34
									
								
								network/transport/context.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										34
									
								
								network/transport/context.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,34 @@ | |||||||
|  | package transport | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type transportKey struct{} | ||||||
|  |  | ||||||
|  | // FromContext get transport from context | ||||||
|  | func FromContext(ctx context.Context) (Transport, bool) { | ||||||
|  | 	if ctx == nil { | ||||||
|  | 		return nil, false | ||||||
|  | 	} | ||||||
|  | 	c, ok := ctx.Value(transportKey{}).(Transport) | ||||||
|  | 	return c, ok | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewContext put transport in context | ||||||
|  | func NewContext(ctx context.Context, c Transport) context.Context { | ||||||
|  | 	if ctx == nil { | ||||||
|  | 		ctx = context.Background() | ||||||
|  | 	} | ||||||
|  | 	return context.WithValue(ctx, transportKey{}, c) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SetOption returns a function to setup a context with given value | ||||||
|  | func SetOption(k, v interface{}) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		if o.Context == nil { | ||||||
|  | 			o.Context = context.Background() | ||||||
|  | 		} | ||||||
|  | 		o.Context = context.WithValue(o.Context, k, v) | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										258
									
								
								network/transport/memory.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										258
									
								
								network/transport/memory.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,258 @@ | |||||||
|  | package transport | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"errors" | ||||||
|  | 	"fmt" | ||||||
|  | 	"net" | ||||||
|  | 	"sync" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
|  | 	maddr "go.unistack.org/micro/v3/util/addr" | ||||||
|  | 	mnet "go.unistack.org/micro/v3/util/net" | ||||||
|  | 	"go.unistack.org/micro/v3/util/rand" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type memorySocket struct { | ||||||
|  | 	ctx     context.Context | ||||||
|  | 	recv    chan *Message | ||||||
|  | 	exit    chan bool | ||||||
|  | 	lexit   chan bool | ||||||
|  | 	send    chan *Message | ||||||
|  | 	local   string | ||||||
|  | 	remote  string | ||||||
|  | 	timeout time.Duration | ||||||
|  | 	sync.RWMutex | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type memoryClient struct { | ||||||
|  | 	*memorySocket | ||||||
|  | 	opts DialOptions | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type memoryListener struct { | ||||||
|  | 	lopts ListenOptions | ||||||
|  | 	ctx   context.Context | ||||||
|  | 	exit  chan bool | ||||||
|  | 	conn  chan *memorySocket | ||||||
|  | 	addr  string | ||||||
|  | 	topts Options | ||||||
|  | 	sync.RWMutex | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type memoryTransport struct { | ||||||
|  | 	listeners map[string]*memoryListener | ||||||
|  | 	opts      Options | ||||||
|  | 	sync.RWMutex | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (ms *memorySocket) Recv(m *Message) error { | ||||||
|  | 	ms.RLock() | ||||||
|  | 	defer ms.RUnlock() | ||||||
|  |  | ||||||
|  | 	ctx := ms.ctx | ||||||
|  | 	if ms.timeout > 0 { | ||||||
|  | 		var cancel context.CancelFunc | ||||||
|  | 		ctx, cancel = context.WithTimeout(ms.ctx, ms.timeout) | ||||||
|  | 		defer cancel() | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	select { | ||||||
|  | 	case <-ctx.Done(): | ||||||
|  | 		return ctx.Err() | ||||||
|  | 	case <-ms.exit: | ||||||
|  | 		return errors.New("connection closed") | ||||||
|  | 	case <-ms.lexit: | ||||||
|  | 		return errors.New("server connection closed") | ||||||
|  | 	case cm := <-ms.recv: | ||||||
|  | 		*m = *cm | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (ms *memorySocket) Local() string { | ||||||
|  | 	return ms.local | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (ms *memorySocket) Remote() string { | ||||||
|  | 	return ms.remote | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (ms *memorySocket) Send(m *Message) error { | ||||||
|  | 	ms.RLock() | ||||||
|  | 	defer ms.RUnlock() | ||||||
|  |  | ||||||
|  | 	ctx := ms.ctx | ||||||
|  | 	if ms.timeout > 0 { | ||||||
|  | 		var cancel context.CancelFunc | ||||||
|  | 		ctx, cancel = context.WithTimeout(ms.ctx, ms.timeout) | ||||||
|  | 		defer cancel() | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	select { | ||||||
|  | 	case <-ctx.Done(): | ||||||
|  | 		return ctx.Err() | ||||||
|  | 	case <-ms.exit: | ||||||
|  | 		return errors.New("connection closed") | ||||||
|  | 	case <-ms.lexit: | ||||||
|  | 		return errors.New("server connection closed") | ||||||
|  | 	case ms.send <- m: | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (ms *memorySocket) Close() error { | ||||||
|  | 	ms.Lock() | ||||||
|  | 	defer ms.Unlock() | ||||||
|  | 	select { | ||||||
|  | 	case <-ms.exit: | ||||||
|  | 		return nil | ||||||
|  | 	default: | ||||||
|  | 		close(ms.exit) | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryListener) Addr() string { | ||||||
|  | 	return m.addr | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryListener) Close() error { | ||||||
|  | 	m.Lock() | ||||||
|  | 	defer m.Unlock() | ||||||
|  | 	select { | ||||||
|  | 	case <-m.exit: | ||||||
|  | 		return nil | ||||||
|  | 	default: | ||||||
|  | 		close(m.exit) | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryListener) Accept(fn func(Socket)) error { | ||||||
|  | 	for { | ||||||
|  | 		select { | ||||||
|  | 		case <-m.exit: | ||||||
|  | 			return nil | ||||||
|  | 		case c := <-m.conn: | ||||||
|  | 			go fn(&memorySocket{ | ||||||
|  | 				lexit:   c.lexit, | ||||||
|  | 				exit:    c.exit, | ||||||
|  | 				send:    c.recv, | ||||||
|  | 				recv:    c.send, | ||||||
|  | 				local:   c.Remote(), | ||||||
|  | 				remote:  c.Local(), | ||||||
|  | 				timeout: m.topts.Timeout, | ||||||
|  | 				ctx:     m.topts.Context, | ||||||
|  | 			}) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryTransport) Dial(ctx context.Context, addr string, opts ...DialOption) (Client, error) { | ||||||
|  | 	m.RLock() | ||||||
|  | 	defer m.RUnlock() | ||||||
|  |  | ||||||
|  | 	listener, ok := m.listeners[addr] | ||||||
|  | 	if !ok { | ||||||
|  | 		return nil, errors.New("could not dial " + addr) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	options := NewDialOptions(opts...) | ||||||
|  |  | ||||||
|  | 	client := &memoryClient{ | ||||||
|  | 		&memorySocket{ | ||||||
|  | 			send:    make(chan *Message), | ||||||
|  | 			recv:    make(chan *Message), | ||||||
|  | 			exit:    make(chan bool), | ||||||
|  | 			lexit:   listener.exit, | ||||||
|  | 			local:   addr, | ||||||
|  | 			remote:  addr, | ||||||
|  | 			timeout: m.opts.Timeout, | ||||||
|  | 			ctx:     m.opts.Context, | ||||||
|  | 		}, | ||||||
|  | 		options, | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// pseudo connect | ||||||
|  | 	select { | ||||||
|  | 	case <-listener.exit: | ||||||
|  | 		return nil, errors.New("connection error") | ||||||
|  | 	case listener.conn <- client.memorySocket: | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return client, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryTransport) Listen(ctx context.Context, addr string, opts ...ListenOption) (Listener, error) { | ||||||
|  | 	m.Lock() | ||||||
|  | 	defer m.Unlock() | ||||||
|  |  | ||||||
|  | 	options := NewListenOptions(opts...) | ||||||
|  |  | ||||||
|  | 	host, port, err := net.SplitHostPort(addr) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	addr, err = maddr.Extract(host) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// if zero port then randomly assign one | ||||||
|  | 	if len(port) > 0 && port == "0" { | ||||||
|  | 		var rng rand.Rand | ||||||
|  | 		i := rng.Intn(20000) | ||||||
|  | 		port = fmt.Sprintf("%d", 10000+i) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// set addr with port | ||||||
|  | 	addr = mnet.HostPort(addr, port) | ||||||
|  |  | ||||||
|  | 	if _, ok := m.listeners[addr]; ok { | ||||||
|  | 		return nil, errors.New("already listening on " + addr) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	listener := &memoryListener{ | ||||||
|  | 		lopts: options, | ||||||
|  | 		topts: m.opts, | ||||||
|  | 		addr:  addr, | ||||||
|  | 		conn:  make(chan *memorySocket), | ||||||
|  | 		exit:  make(chan bool), | ||||||
|  | 		ctx:   m.opts.Context, | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	m.listeners[addr] = listener | ||||||
|  |  | ||||||
|  | 	return listener, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryTransport) Init(opts ...Option) error { | ||||||
|  | 	for _, o := range opts { | ||||||
|  | 		o(&m.opts) | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryTransport) Options() Options { | ||||||
|  | 	return m.opts | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryTransport) String() string { | ||||||
|  | 	return "memory" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryTransport) Name() string { | ||||||
|  | 	return m.opts.Name | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewTransport returns new memory transport with options | ||||||
|  | func NewTransport(opts ...Option) Transport { | ||||||
|  | 	options := NewOptions(opts...) | ||||||
|  |  | ||||||
|  | 	return &memoryTransport{ | ||||||
|  | 		opts:      options, | ||||||
|  | 		listeners: make(map[string]*memoryListener), | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										100
									
								
								network/transport/memory_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										100
									
								
								network/transport/memory_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,100 @@ | |||||||
|  | package transport | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"os" | ||||||
|  | 	"testing" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func TestMemoryTransport(t *testing.T) { | ||||||
|  | 	tr := NewTransport() | ||||||
|  | 	ctx := context.Background() | ||||||
|  | 	// bind / listen | ||||||
|  | 	l, err := tr.Listen(ctx, "127.0.0.1:8080") | ||||||
|  | 	if err != nil { | ||||||
|  | 		t.Fatalf("Unexpected error listening %v", err) | ||||||
|  | 	} | ||||||
|  | 	defer l.Close() | ||||||
|  |  | ||||||
|  | 	cherr := make(chan error, 1) | ||||||
|  | 	// accept | ||||||
|  | 	go func() { | ||||||
|  | 		if nerr := l.Accept(func(sock Socket) { | ||||||
|  | 			for { | ||||||
|  | 				var m Message | ||||||
|  | 				if rerr := sock.Recv(&m); rerr != nil { | ||||||
|  | 					cherr <- rerr | ||||||
|  | 					return | ||||||
|  | 				} | ||||||
|  | 				if len(os.Getenv("INTEGRATION_TESTS")) == 0 { | ||||||
|  | 					t.Logf("Server Received %s", string(m.Body)) | ||||||
|  | 				} | ||||||
|  | 				if cerr := sock.Send(&Message{ | ||||||
|  | 					Body: []byte(`pong`), | ||||||
|  | 				}); cerr != nil { | ||||||
|  | 					cherr <- cerr | ||||||
|  | 					return | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		}); nerr != nil { | ||||||
|  | 			cherr <- err | ||||||
|  | 		} | ||||||
|  | 	}() | ||||||
|  |  | ||||||
|  | 	// dial | ||||||
|  | 	c, err := tr.Dial(ctx, "127.0.0.1:8080") | ||||||
|  | 	if err != nil { | ||||||
|  | 		t.Fatalf("Unexpected error dialing %v", err) | ||||||
|  | 	} | ||||||
|  | 	defer c.Close() | ||||||
|  |  | ||||||
|  | 	select { | ||||||
|  | 	case err := <-cherr: | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	default: | ||||||
|  | 		// send <=> receive | ||||||
|  | 		for i := 0; i < 3; i++ { | ||||||
|  | 			if err := c.Send(&Message{ | ||||||
|  | 				Body: []byte(`ping`), | ||||||
|  | 			}); err != nil { | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
|  | 			var m Message | ||||||
|  | 			if err := c.Recv(&m); err != nil { | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
|  | 			if len(os.Getenv("INTEGRATION_TESTS")) == 0 { | ||||||
|  | 				t.Logf("Client Received %s", string(m.Body)) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestListener(t *testing.T) { | ||||||
|  | 	tr := NewTransport() | ||||||
|  | 	ctx := context.Background() | ||||||
|  | 	// bind / listen on random port | ||||||
|  | 	l, err := tr.Listen(ctx, ":0") | ||||||
|  | 	if err != nil { | ||||||
|  | 		t.Fatalf("Unexpected error listening %v", err) | ||||||
|  | 	} | ||||||
|  | 	defer l.Close() | ||||||
|  |  | ||||||
|  | 	// try again | ||||||
|  | 	l2, err := tr.Listen(ctx, ":0") | ||||||
|  | 	if err != nil { | ||||||
|  | 		t.Fatalf("Unexpected error listening %v", err) | ||||||
|  | 	} | ||||||
|  | 	defer l2.Close() | ||||||
|  |  | ||||||
|  | 	// now make sure it still fails | ||||||
|  | 	l3, err := tr.Listen(ctx, ":8080") | ||||||
|  | 	if err != nil { | ||||||
|  | 		t.Fatalf("Unexpected error listening %v", err) | ||||||
|  | 	} | ||||||
|  | 	defer l3.Close() | ||||||
|  |  | ||||||
|  | 	if _, err := tr.Listen(ctx, ":8080"); err == nil { | ||||||
|  | 		t.Fatal("Expected error binding to :8080 got nil") | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										175
									
								
								network/transport/options.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										175
									
								
								network/transport/options.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,175 @@ | |||||||
|  | package transport | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"crypto/tls" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v3/codec" | ||||||
|  | 	"go.unistack.org/micro/v3/logger" | ||||||
|  | 	"go.unistack.org/micro/v3/meter" | ||||||
|  | 	"go.unistack.org/micro/v3/tracer" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Options struct holds the transport options | ||||||
|  | type Options struct { | ||||||
|  | 	// Meter used for metrics | ||||||
|  | 	Meter meter.Meter | ||||||
|  | 	// Tracer used for tracing | ||||||
|  | 	Tracer tracer.Tracer | ||||||
|  | 	// Codec used for marshal/unmarshal messages | ||||||
|  | 	Codec codec.Codec | ||||||
|  | 	// Logger used for logging | ||||||
|  | 	Logger logger.Logger | ||||||
|  | 	// Context holds external options | ||||||
|  | 	Context context.Context | ||||||
|  | 	// TLSConfig holds tls.TLSConfig options | ||||||
|  | 	TLSConfig *tls.Config | ||||||
|  | 	// Name holds the transport name | ||||||
|  | 	Name string | ||||||
|  | 	// Addrs holds the transport addrs | ||||||
|  | 	Addrs []string | ||||||
|  | 	// Timeout holds the timeout | ||||||
|  | 	Timeout time.Duration | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewOptions returns new options | ||||||
|  | func NewOptions(opts ...Option) Options { | ||||||
|  | 	options := Options{ | ||||||
|  | 		Logger:  logger.DefaultLogger, | ||||||
|  | 		Meter:   meter.DefaultMeter, | ||||||
|  | 		Tracer:  tracer.DefaultTracer, | ||||||
|  | 		Context: context.Background(), | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for _, o := range opts { | ||||||
|  | 		o(&options) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return options | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // DialOptions struct | ||||||
|  | type DialOptions struct { | ||||||
|  | 	// Context holds the external options | ||||||
|  | 	Context context.Context | ||||||
|  | 	// Timeout holds the timeout | ||||||
|  | 	Timeout time.Duration | ||||||
|  | 	// Stream flag | ||||||
|  | 	Stream bool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewDialOptions returns new DialOptions | ||||||
|  | func NewDialOptions(opts ...DialOption) DialOptions { | ||||||
|  | 	options := DialOptions{ | ||||||
|  | 		Timeout: DefaultDialTimeout, | ||||||
|  | 		Context: context.Background(), | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for _, o := range opts { | ||||||
|  | 		o(&options) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return options | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ListenOptions struct | ||||||
|  | type ListenOptions struct { | ||||||
|  | 	// TODO: add tls options when listening | ||||||
|  | 	// Currently set in global options | ||||||
|  | 	// Context holds the external options | ||||||
|  | 	Context context.Context | ||||||
|  | 	// TLSConfig holds the *tls.Config options | ||||||
|  | 	TLSConfig *tls.Config | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewListenOptions returns new ListenOptions | ||||||
|  | func NewListenOptions(opts ...ListenOption) ListenOptions { | ||||||
|  | 	options := ListenOptions{ | ||||||
|  | 		Context: context.Background(), | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for _, o := range opts { | ||||||
|  | 		o(&options) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return options | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Addrs to use for transport | ||||||
|  | func Addrs(addrs ...string) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Addrs = addrs | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Logger sets the logger | ||||||
|  | func Logger(l logger.Logger) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Logger = l | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Meter sets the meter | ||||||
|  | func Meter(m meter.Meter) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Meter = m | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Context sets the context | ||||||
|  | func Context(ctx context.Context) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Context = ctx | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Codec sets the codec used for encoding where the transport | ||||||
|  | // does not support message headers | ||||||
|  | func Codec(c codec.Codec) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Codec = c | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Timeout sets the timeout for Send/Recv execution | ||||||
|  | func Timeout(t time.Duration) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Timeout = t | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // TLSConfig to be used for the transport. | ||||||
|  | func TLSConfig(t *tls.Config) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.TLSConfig = t | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithStream indicates whether this is a streaming connection | ||||||
|  | func WithStream() DialOption { | ||||||
|  | 	return func(o *DialOptions) { | ||||||
|  | 		o.Stream = true | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithTimeout used when dialling the remote side | ||||||
|  | func WithTimeout(d time.Duration) DialOption { | ||||||
|  | 	return func(o *DialOptions) { | ||||||
|  | 		o.Timeout = d | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Tracer to be used for tracing | ||||||
|  | func Tracer(t tracer.Tracer) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Tracer = t | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Name sets the name | ||||||
|  | func Name(n string) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Name = n | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										63
									
								
								network/transport/transport.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										63
									
								
								network/transport/transport.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,63 @@ | |||||||
|  | // Package transport is an interface for synchronous connection based communication | ||||||
|  | package transport | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v3/metadata" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | var ( | ||||||
|  | 	// DefaultTransport is the global default transport | ||||||
|  | 	DefaultTransport = NewTransport() | ||||||
|  | 	// DefaultDialTimeout the default dial timeout | ||||||
|  | 	DefaultDialTimeout = time.Second * 5 | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Transport is an interface which is used for communication between | ||||||
|  | // services. It uses connection based socket send/recv semantics and | ||||||
|  | // has various implementations; http, grpc, quic. | ||||||
|  | type Transport interface { | ||||||
|  | 	Init(...Option) error | ||||||
|  | 	Options() Options | ||||||
|  | 	Dial(ctx context.Context, addr string, opts ...DialOption) (Client, error) | ||||||
|  | 	Listen(ctx context.Context, addr string, opts ...ListenOption) (Listener, error) | ||||||
|  | 	String() string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Message is used to transfer data | ||||||
|  | type Message struct { | ||||||
|  | 	Header metadata.Metadata | ||||||
|  | 	Body   []byte | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Socket bastraction interface | ||||||
|  | type Socket interface { | ||||||
|  | 	Recv(*Message) error | ||||||
|  | 	Send(*Message) error | ||||||
|  | 	Close() error | ||||||
|  | 	Local() string | ||||||
|  | 	Remote() string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Client is the socket owner | ||||||
|  | type Client interface { | ||||||
|  | 	Socket | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Listener is the interface for stream oriented messaging | ||||||
|  | type Listener interface { | ||||||
|  | 	Addr() string | ||||||
|  | 	Close() error | ||||||
|  | 	Accept(func(Socket)) error | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Option is the option signature | ||||||
|  | type Option func(*Options) | ||||||
|  |  | ||||||
|  | // DialOption is the option signature | ||||||
|  | type DialOption func(*DialOptions) | ||||||
|  |  | ||||||
|  | // ListenOption is the option signature | ||||||
|  | type ListenOption func(*ListenOptions) | ||||||
							
								
								
									
										372
									
								
								network/tunnel/broker/broker.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										372
									
								
								network/tunnel/broker/broker.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,372 @@ | |||||||
|  | // Package broker is a tunnel broker | ||||||
|  | package broker | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"fmt" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v3/broker" | ||||||
|  | 	"go.unistack.org/micro/v3/logger" | ||||||
|  | 	"go.unistack.org/micro/v3/metadata" | ||||||
|  | 	"go.unistack.org/micro/v3/network/transport" | ||||||
|  | 	"go.unistack.org/micro/v3/network/tunnel" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type tunBroker struct { | ||||||
|  | 	tunnel tunnel.Tunnel | ||||||
|  | 	opts   broker.Options | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type tunSubscriber struct { | ||||||
|  | 	listener tunnel.Listener | ||||||
|  | 	handler  broker.Handler | ||||||
|  | 	closed   chan bool | ||||||
|  | 	topic    string | ||||||
|  | 	opts     broker.SubscribeOptions | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type tunBatchSubscriber struct { | ||||||
|  | 	listener tunnel.Listener | ||||||
|  | 	handler  broker.BatchHandler | ||||||
|  | 	closed   chan bool | ||||||
|  | 	topic    string | ||||||
|  | 	opts     broker.SubscribeOptions | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type tunEvent struct { | ||||||
|  | 	err     error | ||||||
|  | 	message *broker.Message | ||||||
|  | 	topic   string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // used to access tunnel from options context | ||||||
|  | type ( | ||||||
|  | 	tunnelKey  struct{} | ||||||
|  | 	tunnelAddr struct{} | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func (t *tunBroker) Live() bool { | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunBroker) Ready() bool { | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunBroker) Health() bool { | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunBroker) Init(opts ...broker.Option) error { | ||||||
|  | 	for _, o := range opts { | ||||||
|  | 		o(&t.opts) | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunBroker) Name() string { | ||||||
|  | 	return t.opts.Name | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunBroker) Options() broker.Options { | ||||||
|  | 	return t.opts | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunBroker) Address() string { | ||||||
|  | 	return t.tunnel.Address() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunBroker) Connect(ctx context.Context) error { | ||||||
|  | 	return t.tunnel.Connect(ctx) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunBroker) Disconnect(ctx context.Context) error { | ||||||
|  | 	return t.tunnel.Close(ctx) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunBroker) BatchPublish(ctx context.Context, msgs []*broker.Message, _ ...broker.PublishOption) error { | ||||||
|  | 	// TODO: this is probably inefficient, we might want to just maintain an open connection | ||||||
|  | 	// it may be easier to add broadcast to the tunnel | ||||||
|  | 	topicMap := make(map[string]tunnel.Session) | ||||||
|  |  | ||||||
|  | 	var err error | ||||||
|  | 	for _, msg := range msgs { | ||||||
|  | 		topic, _ := msg.Header.Get(metadata.HeaderTopic) | ||||||
|  | 		c, ok := topicMap[topic] | ||||||
|  | 		if !ok { | ||||||
|  | 			c, err = t.tunnel.Dial(ctx, topic, tunnel.DialMode(tunnel.Multicast)) | ||||||
|  | 			if err != nil { | ||||||
|  | 				return err | ||||||
|  | 			} | ||||||
|  | 			defer c.Close() | ||||||
|  | 			topicMap[topic] = c | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if err = c.Send(&transport.Message{ | ||||||
|  | 			Header: msg.Header, | ||||||
|  | 			Body:   msg.Body, | ||||||
|  | 		}); err != nil { | ||||||
|  | 			//	msg.SetError(err) | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunBroker) Publish(ctx context.Context, topic string, m *broker.Message, _ ...broker.PublishOption) error { | ||||||
|  | 	// TODO: this is probably inefficient, we might want to just maintain an open connection | ||||||
|  | 	// it may be easier to add broadcast to the tunnel | ||||||
|  | 	c, err := t.tunnel.Dial(ctx, topic, tunnel.DialMode(tunnel.Multicast)) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	defer c.Close() | ||||||
|  |  | ||||||
|  | 	return c.Send(&transport.Message{ | ||||||
|  | 		Header: m.Header, | ||||||
|  | 		Body:   m.Body, | ||||||
|  | 	}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunBroker) BatchSubscribe(ctx context.Context, topic string, h broker.BatchHandler, opts ...broker.SubscribeOption) (broker.Subscriber, error) { | ||||||
|  | 	l, err := t.tunnel.Listen(ctx, topic, tunnel.ListenMode(tunnel.Multicast)) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	tunSub := &tunBatchSubscriber{ | ||||||
|  | 		topic:    topic, | ||||||
|  | 		handler:  h, | ||||||
|  | 		opts:     broker.NewSubscribeOptions(opts...), | ||||||
|  | 		closed:   make(chan bool), | ||||||
|  | 		listener: l, | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// start processing | ||||||
|  | 	go tunSub.run() | ||||||
|  |  | ||||||
|  | 	return tunSub, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunBroker) Subscribe(ctx context.Context, topic string, h broker.Handler, opts ...broker.SubscribeOption) (broker.Subscriber, error) { | ||||||
|  | 	l, err := t.tunnel.Listen(ctx, topic, tunnel.ListenMode(tunnel.Multicast)) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	tunSub := &tunSubscriber{ | ||||||
|  | 		topic:    topic, | ||||||
|  | 		handler:  h, | ||||||
|  | 		opts:     broker.NewSubscribeOptions(opts...), | ||||||
|  | 		closed:   make(chan bool), | ||||||
|  | 		listener: l, | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// start processing | ||||||
|  | 	go tunSub.run() | ||||||
|  |  | ||||||
|  | 	return tunSub, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunBroker) String() string { | ||||||
|  | 	return "tunnel" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunBatchSubscriber) run() { | ||||||
|  | 	for { | ||||||
|  | 		// accept a new connection | ||||||
|  | 		c, err := t.listener.Accept() | ||||||
|  | 		if err != nil { | ||||||
|  | 			select { | ||||||
|  | 			case <-t.closed: | ||||||
|  | 				return | ||||||
|  | 			default: | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// receive message | ||||||
|  | 		m := new(transport.Message) | ||||||
|  | 		if err := c.Recv(m); err != nil { | ||||||
|  | 			if logger.DefaultLogger.V(logger.ErrorLevel) { | ||||||
|  | 				logger.DefaultLogger.Error(t.opts.Context, err.Error(), err) | ||||||
|  | 			} | ||||||
|  | 			if err = c.Close(); err != nil { | ||||||
|  | 				if logger.DefaultLogger.V(logger.ErrorLevel) { | ||||||
|  | 					logger.DefaultLogger.Error(t.opts.Context, err.Error(), err) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// close the connection | ||||||
|  | 		c.Close() | ||||||
|  |  | ||||||
|  | 		evts := broker.Events{&tunEvent{ | ||||||
|  | 			topic: t.topic, | ||||||
|  | 			message: &broker.Message{ | ||||||
|  | 				Header: m.Header, | ||||||
|  | 				Body:   m.Body, | ||||||
|  | 			}, | ||||||
|  | 		}} | ||||||
|  | 		// handle the message | ||||||
|  | 		go func() { | ||||||
|  | 			_ = t.handler(evts) | ||||||
|  | 		}() | ||||||
|  |  | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunSubscriber) run() { | ||||||
|  | 	for { | ||||||
|  | 		// accept a new connection | ||||||
|  | 		c, err := t.listener.Accept() | ||||||
|  | 		if err != nil { | ||||||
|  | 			select { | ||||||
|  | 			case <-t.closed: | ||||||
|  | 				return | ||||||
|  | 			default: | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// receive message | ||||||
|  | 		m := new(transport.Message) | ||||||
|  | 		if err := c.Recv(m); err != nil { | ||||||
|  | 			if logger.DefaultLogger.V(logger.ErrorLevel) { | ||||||
|  | 				logger.DefaultLogger.Error(t.opts.Context, err.Error(), err) | ||||||
|  | 			} | ||||||
|  | 			if err = c.Close(); err != nil { | ||||||
|  | 				if logger.DefaultLogger.V(logger.ErrorLevel) { | ||||||
|  | 					logger.DefaultLogger.Error(t.opts.Context, err.Error(), err) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// close the connection | ||||||
|  | 		c.Close() | ||||||
|  |  | ||||||
|  | 		// handle the message | ||||||
|  | 		go func() { | ||||||
|  | 			_ = t.handler(&tunEvent{ | ||||||
|  | 				topic: t.topic, | ||||||
|  | 				message: &broker.Message{ | ||||||
|  | 					Header: m.Header, | ||||||
|  | 					Body:   m.Body, | ||||||
|  | 				}, | ||||||
|  | 			}) | ||||||
|  | 		}() | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunBatchSubscriber) Options() broker.SubscribeOptions { | ||||||
|  | 	return t.opts | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunBatchSubscriber) Topic() string { | ||||||
|  | 	return t.topic | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunBatchSubscriber) Unsubscribe(ctx context.Context) error { | ||||||
|  | 	select { | ||||||
|  | 	case <-t.closed: | ||||||
|  | 		return nil | ||||||
|  | 	default: | ||||||
|  | 		close(t.closed) | ||||||
|  | 		return t.listener.Close() | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunSubscriber) Options() broker.SubscribeOptions { | ||||||
|  | 	return t.opts | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunSubscriber) Topic() string { | ||||||
|  | 	return t.topic | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunSubscriber) Unsubscribe(ctx context.Context) error { | ||||||
|  | 	select { | ||||||
|  | 	case <-t.closed: | ||||||
|  | 		return nil | ||||||
|  | 	default: | ||||||
|  | 		close(t.closed) | ||||||
|  | 		return t.listener.Close() | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunEvent) Topic() string { | ||||||
|  | 	return t.topic | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunEvent) Message() *broker.Message { | ||||||
|  | 	return t.message | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunEvent) Ack() error { | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunEvent) Error() error { | ||||||
|  | 	return t.err | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunEvent) SetError(err error) { | ||||||
|  | 	t.err = err | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunEvent) Context() context.Context { | ||||||
|  | 	return context.TODO() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewBroker returns new tunnel broker | ||||||
|  | func NewBroker(opts ...broker.Option) (broker.Broker, error) { | ||||||
|  | 	options := broker.NewOptions(opts...) | ||||||
|  |  | ||||||
|  | 	t, ok := options.Context.Value(tunnelKey{}).(tunnel.Tunnel) | ||||||
|  | 	if !ok { | ||||||
|  | 		return nil, fmt.Errorf("tunnel not set") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	a, ok := options.Context.Value(tunnelAddr{}).(string) | ||||||
|  | 	if ok { | ||||||
|  | 		// initialise address | ||||||
|  | 		if err := t.Init(tunnel.Address(a)); err != nil { | ||||||
|  | 			return nil, err | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if len(options.Addrs) > 0 { | ||||||
|  | 		// initialise nodes | ||||||
|  | 		if err := t.Init(tunnel.Nodes(options.Addrs...)); err != nil { | ||||||
|  | 			return nil, err | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return &tunBroker{ | ||||||
|  | 		opts:   options, | ||||||
|  | 		tunnel: t, | ||||||
|  | 	}, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithAddress sets the tunnel address | ||||||
|  | func WithAddress(a string) broker.Option { | ||||||
|  | 	return func(o *broker.Options) { | ||||||
|  | 		if o.Context == nil { | ||||||
|  | 			o.Context = context.Background() | ||||||
|  | 		} | ||||||
|  | 		o.Context = context.WithValue(o.Context, tunnelAddr{}, a) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithTunnel sets the internal tunnel | ||||||
|  | func WithTunnel(t tunnel.Tunnel) broker.Option { | ||||||
|  | 	return func(o *broker.Options) { | ||||||
|  | 		if o.Context == nil { | ||||||
|  | 			o.Context = context.Background() | ||||||
|  | 		} | ||||||
|  | 		o.Context = context.WithValue(o.Context, tunnelKey{}, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										192
									
								
								network/tunnel/options.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										192
									
								
								network/tunnel/options.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,192 @@ | |||||||
|  | package tunnel | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v3/logger" | ||||||
|  | 	"go.unistack.org/micro/v3/meter" | ||||||
|  | 	"go.unistack.org/micro/v3/network/transport" | ||||||
|  | 	"go.unistack.org/micro/v3/tracer" | ||||||
|  | 	"go.unistack.org/micro/v3/util/id" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | var ( | ||||||
|  | 	// DefaultAddress is default tunnel bind address | ||||||
|  | 	DefaultAddress = ":0" | ||||||
|  | 	// DefaultToken the shared default token | ||||||
|  | 	DefaultToken = "go.micro.tunnel" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Option func signature | ||||||
|  | type Option func(*Options) | ||||||
|  |  | ||||||
|  | // Options provides network configuration options | ||||||
|  | type Options struct { | ||||||
|  | 	// Logger used for logging | ||||||
|  | 	Logger logger.Logger | ||||||
|  | 	// Meter used for metrics | ||||||
|  | 	Meter meter.Meter | ||||||
|  | 	// Tracer used for tracing | ||||||
|  | 	Tracer tracer.Tracer | ||||||
|  | 	// Transport used for communication | ||||||
|  | 	Transport transport.Transport | ||||||
|  | 	// Token the shared auth token | ||||||
|  | 	Token string | ||||||
|  | 	// Name holds the tunnel name | ||||||
|  | 	Name string | ||||||
|  | 	// ID holds the tunnel id | ||||||
|  | 	ID string | ||||||
|  | 	// Address holds the tunnel address | ||||||
|  | 	Address string | ||||||
|  | 	// Nodes holds the tunnel nodes | ||||||
|  | 	Nodes []string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // DialOption func | ||||||
|  | type DialOption func(*DialOptions) | ||||||
|  |  | ||||||
|  | // DialOptions provides dial options | ||||||
|  | type DialOptions struct { | ||||||
|  | 	// Link specifies the link to use | ||||||
|  | 	Link string | ||||||
|  | 	// specify mode of the session | ||||||
|  | 	Mode Mode | ||||||
|  | 	// Wait for connection to be accepted | ||||||
|  | 	Wait bool | ||||||
|  | 	// the dial timeout | ||||||
|  | 	Timeout time.Duration | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ListenOption func | ||||||
|  | type ListenOption func(*ListenOptions) | ||||||
|  |  | ||||||
|  | // ListenOptions provides listen options | ||||||
|  | type ListenOptions struct { | ||||||
|  | 	// Mode specify mode of the session | ||||||
|  | 	Mode Mode | ||||||
|  | 	// Timeout the read timeout | ||||||
|  | 	Timeout time.Duration | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ID sets the tunnel id | ||||||
|  | func ID(id string) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.ID = id | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Logger sets the logger | ||||||
|  | func Logger(l logger.Logger) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Logger = l | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Meter sets the meter | ||||||
|  | func Meter(m meter.Meter) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Meter = m | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Address sets the tunnel address | ||||||
|  | func Address(a string) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Address = a | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Nodes specify remote network nodes | ||||||
|  | func Nodes(n ...string) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Nodes = n | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Token sets the shared token for auth | ||||||
|  | func Token(t string) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Token = t | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Transport listens for incoming connections | ||||||
|  | func Transport(t transport.Transport) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Transport = t | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ListenMode option | ||||||
|  | func ListenMode(m Mode) ListenOption { | ||||||
|  | 	return func(o *ListenOptions) { | ||||||
|  | 		o.Mode = m | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ListenTimeout for reads and writes on the listener session | ||||||
|  | func ListenTimeout(t time.Duration) ListenOption { | ||||||
|  | 	return func(o *ListenOptions) { | ||||||
|  | 		o.Timeout = t | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // DialMode multicast sets the multicast option to send only to those mapped | ||||||
|  | func DialMode(m Mode) DialOption { | ||||||
|  | 	return func(o *DialOptions) { | ||||||
|  | 		o.Mode = m | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // DialTimeout sets the dial timeout of the connection | ||||||
|  | func DialTimeout(t time.Duration) DialOption { | ||||||
|  | 	return func(o *DialOptions) { | ||||||
|  | 		o.Timeout = t | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // DialLink specifies the link to pin this connection to. | ||||||
|  | // This is not applicable if the multicast option is set. | ||||||
|  | func DialLink(id string) DialOption { | ||||||
|  | 	return func(o *DialOptions) { | ||||||
|  | 		o.Link = id | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // DialWait specifies whether to wait for the connection | ||||||
|  | // to be accepted before returning the session | ||||||
|  | func DialWait(b bool) DialOption { | ||||||
|  | 	return func(o *DialOptions) { | ||||||
|  | 		o.Wait = b | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewOptions returns router default options with filled values | ||||||
|  | func NewOptions(opts ...Option) Options { | ||||||
|  | 	options := Options{ | ||||||
|  | 		ID:      id.MustNew(), | ||||||
|  | 		Address: DefaultAddress, | ||||||
|  | 		Token:   DefaultToken, | ||||||
|  | 		Logger:  logger.DefaultLogger, | ||||||
|  | 		Meter:   meter.DefaultMeter, | ||||||
|  | 		Tracer:  tracer.DefaultTracer, | ||||||
|  | 	} | ||||||
|  | 	for _, o := range opts { | ||||||
|  | 		o(&options) | ||||||
|  | 	} | ||||||
|  | 	return options | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Tracer to be used for tracing | ||||||
|  | func Tracer(t tracer.Tracer) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Tracer = t | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Name sets the name | ||||||
|  | func Name(n string) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Name = n | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										30
									
								
								network/tunnel/transport/listener.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										30
									
								
								network/tunnel/transport/listener.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,30 @@ | |||||||
|  | package transport | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"go.unistack.org/micro/v3/network/transport" | ||||||
|  | 	"go.unistack.org/micro/v3/network/tunnel" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type tunListener struct { | ||||||
|  | 	l tunnel.Listener | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunListener) Addr() string { | ||||||
|  | 	return t.l.Channel() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunListener) Close() error { | ||||||
|  | 	return t.l.Close() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunListener) Accept(fn func(socket transport.Socket)) error { | ||||||
|  | 	for { | ||||||
|  | 		// accept connection | ||||||
|  | 		c, err := t.l.Accept() | ||||||
|  | 		if err != nil { | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  | 		// execute the function | ||||||
|  | 		go fn(c) | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										113
									
								
								network/tunnel/transport/transport.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										113
									
								
								network/tunnel/transport/transport.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,113 @@ | |||||||
|  | // Package transport provides a tunnel transport | ||||||
|  | package transport | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"fmt" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v3/network/transport" | ||||||
|  | 	"go.unistack.org/micro/v3/network/tunnel" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type tunTransport struct { | ||||||
|  | 	tunnel  tunnel.Tunnel | ||||||
|  | 	options transport.Options | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type tunnelKey struct{} | ||||||
|  |  | ||||||
|  | type transportKey struct{} | ||||||
|  |  | ||||||
|  | func (t *tunTransport) Init(opts ...transport.Option) error { | ||||||
|  | 	for _, o := range opts { | ||||||
|  | 		o(&t.options) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// close the existing tunnel | ||||||
|  | 	if t.tunnel != nil { | ||||||
|  | 		t.tunnel.Close(context.TODO()) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// get the tunnel | ||||||
|  | 	tun, ok := t.options.Context.Value(tunnelKey{}).(tunnel.Tunnel) | ||||||
|  | 	if !ok { | ||||||
|  | 		return fmt.Errorf("tunnel not set") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// get the transport | ||||||
|  | 	tr, ok := t.options.Context.Value(transportKey{}).(transport.Transport) | ||||||
|  | 	if ok { | ||||||
|  | 		_ = tun.Init(tunnel.Transport(tr)) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// set the tunnel | ||||||
|  | 	t.tunnel = tun | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunTransport) Dial(ctx context.Context, addr string, opts ...transport.DialOption) (transport.Client, error) { | ||||||
|  | 	if err := t.tunnel.Connect(ctx); err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c, err := t.tunnel.Dial(ctx, addr) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return c, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunTransport) Listen(ctx context.Context, addr string, opts ...transport.ListenOption) (transport.Listener, error) { | ||||||
|  | 	if err := t.tunnel.Connect(ctx); err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l, err := t.tunnel.Listen(ctx, addr) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return &tunListener{l}, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunTransport) Options() transport.Options { | ||||||
|  | 	return t.options | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *tunTransport) String() string { | ||||||
|  | 	return "tunnel" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewTransport honours the initialiser used in | ||||||
|  | func NewTransport(opts ...transport.Option) transport.Transport { | ||||||
|  | 	t := &tunTransport{ | ||||||
|  | 		options: transport.Options{}, | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// initialise | ||||||
|  | 	// t.Init(opts...) | ||||||
|  |  | ||||||
|  | 	return t | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithTunnel sets the internal tunnel | ||||||
|  | func WithTunnel(t tunnel.Tunnel) transport.Option { | ||||||
|  | 	return func(o *transport.Options) { | ||||||
|  | 		if o.Context == nil { | ||||||
|  | 			o.Context = context.Background() | ||||||
|  | 		} | ||||||
|  | 		o.Context = context.WithValue(o.Context, tunnelKey{}, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithTransport sets the internal transport | ||||||
|  | func WithTransport(t transport.Transport) transport.Option { | ||||||
|  | 	return func(o *transport.Options) { | ||||||
|  | 		if o.Context == nil { | ||||||
|  | 			o.Context = context.Background() | ||||||
|  | 		} | ||||||
|  | 		o.Context = context.WithValue(o.Context, transportKey{}, t) | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										106
									
								
								network/tunnel/tunnel.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										106
									
								
								network/tunnel/tunnel.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,106 @@ | |||||||
|  | // Package tunnel provides gre network tunnelling | ||||||
|  | package tunnel | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"errors" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v3/network/transport" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // DefaultTunnel contains default tunnel implementation | ||||||
|  | var DefaultTunnel Tunnel | ||||||
|  |  | ||||||
|  | const ( | ||||||
|  | 	// Unicast send over one link | ||||||
|  | 	Unicast Mode = iota | ||||||
|  | 	// Multicast send to all channel listeners | ||||||
|  | 	Multicast | ||||||
|  | 	// Broadcast send to all links | ||||||
|  | 	Broadcast | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | var ( | ||||||
|  | 	// DefaultDialTimeout is the dial timeout if none is specified | ||||||
|  | 	DefaultDialTimeout = time.Second * 5 | ||||||
|  | 	// ErrDialTimeout is returned by a call to Dial where the timeout occurs | ||||||
|  | 	ErrDialTimeout = errors.New("dial timeout") | ||||||
|  | 	// ErrDiscoverChan is returned when we failed to receive the "announce" back from a discovery | ||||||
|  | 	ErrDiscoverChan = errors.New("failed to discover channel") | ||||||
|  | 	// ErrLinkNotFound is returned when a link is specified at dial time and does not exist | ||||||
|  | 	ErrLinkNotFound = errors.New("link not found") | ||||||
|  | 	// ErrLinkDisconnected is returned when a link we attempt to send to is disconnected | ||||||
|  | 	ErrLinkDisconnected = errors.New("link not connected") | ||||||
|  | 	// ErrLinkLoopback is returned when attempting to send an outbound message over loopback link | ||||||
|  | 	ErrLinkLoopback = errors.New("link is loopback") | ||||||
|  | 	// ErrLinkRemote is returned when attempting to send a loopback message over remote link | ||||||
|  | 	ErrLinkRemote = errors.New("link is remote") | ||||||
|  | 	// ErrReadTimeout is a timeout on session.Recv | ||||||
|  | 	ErrReadTimeout = errors.New("read timeout") | ||||||
|  | 	// ErrDecryptingData is for when theres a nonce error | ||||||
|  | 	ErrDecryptingData = errors.New("error decrypting data") | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Mode of the session | ||||||
|  | type Mode uint8 | ||||||
|  |  | ||||||
|  | // Tunnel creates a gre tunnel on top of the micro/transport. | ||||||
|  | // It establishes multiple streams using the Micro-Tunnel-Channel header | ||||||
|  | // and Micro-Tunnel-Session header. The tunnel id is a hash of | ||||||
|  | // the address being requested. | ||||||
|  | type Tunnel interface { | ||||||
|  | 	// Init initializes tunnel with options | ||||||
|  | 	Init(opts ...Option) error | ||||||
|  | 	// Address returns the address the tunnel is listening on | ||||||
|  | 	Address() string | ||||||
|  | 	// Connect connects the tunnel | ||||||
|  | 	Connect(ctx context.Context) error | ||||||
|  | 	// Close closes the tunnel | ||||||
|  | 	Close(ctx context.Context) error | ||||||
|  | 	// Links returns all the links the tunnel is connected to | ||||||
|  | 	Links() []Link | ||||||
|  | 	// Dial allows a client to connect to a channel | ||||||
|  | 	Dial(ctx context.Context, channel string, opts ...DialOption) (Session, error) | ||||||
|  | 	// Listen allows to accept connections on a channel | ||||||
|  | 	Listen(ctx context.Context, channel string, opts ...ListenOption) (Listener, error) | ||||||
|  | 	// String returns the name of the tunnel implementation | ||||||
|  | 	String() string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Link represents internal links to the tunnel | ||||||
|  | type Link interface { | ||||||
|  | 	// Id returns the link unique Id | ||||||
|  | 	Id() string | ||||||
|  | 	// Delay is the current load on the link (lower is better) | ||||||
|  | 	Delay() int64 | ||||||
|  | 	// Length returns the roundtrip time as nanoseconds (lower is better) | ||||||
|  | 	Length() int64 | ||||||
|  | 	// Current transfer rate as bits per second (lower is better) | ||||||
|  | 	Rate() float64 | ||||||
|  | 	// Is this a loopback link | ||||||
|  | 	Loopback() bool | ||||||
|  | 	// State of the link: connected/closed/error | ||||||
|  | 	State() string | ||||||
|  | 	// honours transport socket | ||||||
|  | 	transport.Socket | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Listener provides similar constructs to the transport.Listener | ||||||
|  | type Listener interface { | ||||||
|  | 	Accept() (Session, error) | ||||||
|  | 	Channel() string | ||||||
|  | 	Close() error | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Session is a unique session created when dialling or accepting connections on the tunnel | ||||||
|  | type Session interface { | ||||||
|  | 	// The unique session id | ||||||
|  | 	Id() string | ||||||
|  | 	// The channel name | ||||||
|  | 	Channel() string | ||||||
|  | 	// The link the session is on | ||||||
|  | 	Link() string | ||||||
|  | 	// a transport socket | ||||||
|  | 	transport.Socket | ||||||
|  | } | ||||||
							
								
								
									
										22
									
								
								options.go
									
									
									
									
									
								
							
							
						
						
									
										22
									
								
								options.go
									
									
									
									
									
								
							| @@ -5,17 +5,17 @@ import ( | |||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/broker" | 	"go.unistack.org/micro/v3/broker" | ||||||
| 	"go.unistack.org/micro/v4/client" | 	"go.unistack.org/micro/v3/client" | ||||||
| 	"go.unistack.org/micro/v4/config" | 	"go.unistack.org/micro/v3/config" | ||||||
| 	"go.unistack.org/micro/v4/logger" | 	"go.unistack.org/micro/v3/logger" | ||||||
| 	"go.unistack.org/micro/v4/metadata" | 	"go.unistack.org/micro/v3/metadata" | ||||||
| 	"go.unistack.org/micro/v4/meter" | 	"go.unistack.org/micro/v3/meter" | ||||||
| 	"go.unistack.org/micro/v4/register" | 	"go.unistack.org/micro/v3/register" | ||||||
| 	"go.unistack.org/micro/v4/router" | 	"go.unistack.org/micro/v3/router" | ||||||
| 	"go.unistack.org/micro/v4/server" | 	"go.unistack.org/micro/v3/server" | ||||||
| 	"go.unistack.org/micro/v4/store" | 	"go.unistack.org/micro/v3/store" | ||||||
| 	"go.unistack.org/micro/v4/tracer" | 	"go.unistack.org/micro/v3/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Options for micro service | // Options for micro service | ||||||
|   | |||||||
| @@ -1,222 +0,0 @@ | |||||||
| package options |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"reflect" |  | ||||||
| 	"strings" |  | ||||||
| 	"time" |  | ||||||
|  |  | ||||||
| 	"github.com/spf13/cast" |  | ||||||
| 	mreflect "go.unistack.org/micro/v4/util/reflect" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // Options interface must be used by all options |  | ||||||
| type Validator interface { |  | ||||||
| 	// Validate returns nil, if all options are correct, |  | ||||||
| 	// otherwise returns an error explaining the mistake |  | ||||||
| 	Validate() error |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Option func signature |  | ||||||
| type Option func(interface{}) error |  | ||||||
|  |  | ||||||
| // Apply assign options to struct src |  | ||||||
| func Apply(src interface{}, opts ...Option) error { |  | ||||||
| 	for _, opt := range opts { |  | ||||||
| 		if err := opt(src); err != nil { |  | ||||||
| 			return err |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // SetValueByPath set src struct field to val dst via path |  | ||||||
| func SetValueByPath(src interface{}, dst interface{}, path string) error { |  | ||||||
| 	var err error |  | ||||||
|  |  | ||||||
| 	switch v := dst.(type) { |  | ||||||
| 	case []interface{}: |  | ||||||
| 		if len(v) == 1 { |  | ||||||
| 			dst = v[0] |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	var sv reflect.Value |  | ||||||
| 	switch t := src.(type) { |  | ||||||
| 	case reflect.Value: |  | ||||||
| 		sv = t |  | ||||||
| 	default: |  | ||||||
| 		sv = reflect.ValueOf(src) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	parts := strings.Split(path, ".") |  | ||||||
|  |  | ||||||
| 	for _, p := range parts { |  | ||||||
|  |  | ||||||
| 		if sv.Kind() == reflect.Ptr { |  | ||||||
| 			sv = sv.Elem() |  | ||||||
| 		} |  | ||||||
| 		if sv.Kind() != reflect.Struct { |  | ||||||
| 			return mreflect.ErrInvalidStruct |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		typ := sv.Type() |  | ||||||
| 		for idx := 0; idx < typ.NumField(); idx++ { |  | ||||||
| 			fld := typ.Field(idx) |  | ||||||
| 			val := sv.Field(idx) |  | ||||||
|  |  | ||||||
| 			/* |  | ||||||
| 				if len(fld.PkgPath) != 0 { |  | ||||||
| 					continue |  | ||||||
| 				} |  | ||||||
| 			*/ |  | ||||||
|  |  | ||||||
| 			if fld.Anonymous { |  | ||||||
| 				if len(parts) == 1 && val.Kind() == reflect.Struct { |  | ||||||
| 					if err = SetValueByPath(val, dst, p); err != nil { |  | ||||||
| 						return err |  | ||||||
| 					} |  | ||||||
| 				} |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			if fld.Name != p && !strings.EqualFold(strings.ToLower(fld.Name), strings.ToLower(p)) { |  | ||||||
| 				continue |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			switch val.Interface().(type) { |  | ||||||
| 			case []time.Duration: |  | ||||||
| 				dst, err = cast.ToDurationSliceE(dst) |  | ||||||
| 				if err != nil { |  | ||||||
| 					return err |  | ||||||
| 				} |  | ||||||
| 				reflect.Copy(val, reflect.ValueOf(dst)) |  | ||||||
| 				return nil |  | ||||||
| 			case time.Duration: |  | ||||||
| 				dst, err = cast.ToDurationE(dst) |  | ||||||
| 				if err != nil { |  | ||||||
| 					return err |  | ||||||
| 				} |  | ||||||
| 				val.Set(reflect.ValueOf(dst)) |  | ||||||
| 				return nil |  | ||||||
| 			case time.Time: |  | ||||||
| 				dst, err = cast.ToTimeE(dst) |  | ||||||
| 				if err != nil { |  | ||||||
| 					return err |  | ||||||
| 				} |  | ||||||
| 				val.Set(reflect.ValueOf(dst)) |  | ||||||
| 				return nil |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			switch val.Kind() { |  | ||||||
| 			case reflect.Map: |  | ||||||
| 				if val.IsZero() { |  | ||||||
| 					val.Set(reflect.MakeMap(val.Type())) |  | ||||||
| 				} |  | ||||||
|  |  | ||||||
| 				return setMap(val.Interface(), dst) |  | ||||||
| 			case reflect.Array, reflect.Slice: |  | ||||||
| 				switch val.Type().Elem().Kind() { |  | ||||||
| 				case reflect.Bool: |  | ||||||
| 					dst, err = cast.ToBoolSliceE(dst) |  | ||||||
| 				case reflect.String: |  | ||||||
| 					dst, err = cast.ToStringSliceE(dst) |  | ||||||
| 				case reflect.Float32: |  | ||||||
| 					dst, err = toFloat32SliceE(dst) |  | ||||||
| 				case reflect.Float64: |  | ||||||
| 					dst, err = toFloat64SliceE(dst) |  | ||||||
| 				case reflect.Int8: |  | ||||||
| 					dst, err = toInt8SliceE(dst) |  | ||||||
| 				case reflect.Int: |  | ||||||
| 					dst, err = cast.ToIntSliceE(dst) |  | ||||||
| 				case reflect.Int16: |  | ||||||
| 					dst, err = toInt16SliceE(dst) |  | ||||||
| 				case reflect.Int32: |  | ||||||
| 					dst, err = toInt32SliceE(dst) |  | ||||||
| 				case reflect.Int64: |  | ||||||
| 					dst, err = toInt64SliceE(dst) |  | ||||||
| 				case reflect.Uint8: |  | ||||||
| 					dst, err = toUint8SliceE(dst) |  | ||||||
| 				case reflect.Uint: |  | ||||||
| 					dst, err = toUintSliceE(dst) |  | ||||||
| 				case reflect.Uint16: |  | ||||||
| 					dst, err = toUint16SliceE(dst) |  | ||||||
| 				case reflect.Uint32: |  | ||||||
| 					dst, err = toUint32SliceE(dst) |  | ||||||
| 				case reflect.Uint64: |  | ||||||
| 					dst, err = toUint64SliceE(dst) |  | ||||||
| 				} |  | ||||||
| 				if err != nil { |  | ||||||
| 					return err |  | ||||||
| 				} |  | ||||||
| 				if val.Kind() == reflect.Slice { |  | ||||||
| 					val.Set(reflect.ValueOf(dst)) |  | ||||||
| 				} else { |  | ||||||
| 					reflect.Copy(val, reflect.ValueOf(dst)) |  | ||||||
| 				} |  | ||||||
| 				return nil |  | ||||||
| 			case reflect.Float32: |  | ||||||
| 				dst, err = toFloat32SliceE(dst) |  | ||||||
| 			case reflect.Float64: |  | ||||||
| 				dst, err = toFloat64SliceE(dst) |  | ||||||
| 			case reflect.Bool: |  | ||||||
| 				dst, err = cast.ToBoolE(dst) |  | ||||||
| 			case reflect.String: |  | ||||||
| 				dst, err = cast.ToStringE(dst) |  | ||||||
| 			case reflect.Int8: |  | ||||||
| 				dst, err = cast.ToInt8E(dst) |  | ||||||
| 			case reflect.Int: |  | ||||||
| 				dst, err = cast.ToIntE(dst) |  | ||||||
| 			case reflect.Int16: |  | ||||||
| 				dst, err = cast.ToInt16E(dst) |  | ||||||
| 			case reflect.Int32: |  | ||||||
| 				dst, err = cast.ToInt32E(dst) |  | ||||||
| 			case reflect.Int64: |  | ||||||
| 				dst, err = cast.ToInt64E(dst) |  | ||||||
| 			case reflect.Uint8: |  | ||||||
| 				dst, err = cast.ToUint8E(dst) |  | ||||||
| 			case reflect.Uint: |  | ||||||
| 				dst, err = cast.ToUintE(dst) |  | ||||||
| 			case reflect.Uint16: |  | ||||||
| 				dst, err = cast.ToUint16E(dst) |  | ||||||
| 			case reflect.Uint32: |  | ||||||
| 				dst, err = cast.ToUint32E(dst) |  | ||||||
| 			case reflect.Uint64: |  | ||||||
| 				dst, err = cast.ToUint64E(dst) |  | ||||||
| 			default: |  | ||||||
| 			} |  | ||||||
| 			if err != nil { |  | ||||||
| 				return err |  | ||||||
| 			} |  | ||||||
| 			val.Set(reflect.ValueOf(dst)) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewOption create new option with name |  | ||||||
| func NewOption(name string) func(...interface{}) Option { |  | ||||||
| 	return func(dst ...interface{}) Option { |  | ||||||
| 		return func(src interface{}) error { |  | ||||||
| 			return SetValueByPath(src, dst, name) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| var ( |  | ||||||
| 	Address   = NewOption("Address") |  | ||||||
| 	Name      = NewOption("Name") |  | ||||||
| 	Broker    = NewOption("Broker") |  | ||||||
| 	Logger    = NewOption("Logger") |  | ||||||
| 	Meter     = NewOption("Meter") |  | ||||||
| 	Tracer    = NewOption("Tracer") |  | ||||||
| 	Store     = NewOption("Store") |  | ||||||
| 	Register  = NewOption("Register") |  | ||||||
| 	Router    = NewOption("Router") |  | ||||||
| 	Codec     = NewOption("Codec") |  | ||||||
| 	Codecs    = NewOption("Codecs") |  | ||||||
| 	Client    = NewOption("Client") |  | ||||||
| 	Context   = NewOption("Context") |  | ||||||
| 	TLSConfig = NewOption("TLSConfig") |  | ||||||
| 	Metadata  = NewOption("Metadata") |  | ||||||
| 	Timeout   = NewOption("Timeout") |  | ||||||
| ) |  | ||||||
| @@ -1,181 +0,0 @@ | |||||||
| package options_test |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"crypto/tls" |  | ||||||
| 	"sync" |  | ||||||
| 	"testing" |  | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/options" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| type codec interface { |  | ||||||
| 	Marshal(v interface{}, opts ...options.Option) ([]byte, error) |  | ||||||
| 	Unmarshal(b []byte, v interface{}, opts ...options.Option) error |  | ||||||
| 	String() string |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestCodecs(t *testing.T) { |  | ||||||
| 	type s struct { |  | ||||||
| 		Codecs map[string]codec |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	wg := &sync.WaitGroup{} |  | ||||||
| 	tc := &tls.Config{InsecureSkipVerify: true} |  | ||||||
| 	opts := []options.Option{ |  | ||||||
| 		options.NewOption("Codecs")(wg), |  | ||||||
| 		options.NewOption("TLSConfig")(tc), |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	src := &s{} |  | ||||||
|  |  | ||||||
| 	if err := options.Apply(src, opts...); err != nil { |  | ||||||
| 		t.Fatal(err) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestSpecial(t *testing.T) { |  | ||||||
| 	type s struct { |  | ||||||
| 		Wait      *sync.WaitGroup |  | ||||||
| 		TLSConfig *tls.Config |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	wg := &sync.WaitGroup{} |  | ||||||
| 	tc := &tls.Config{InsecureSkipVerify: true} |  | ||||||
| 	opts := []options.Option{ |  | ||||||
| 		options.NewOption("Wait")(wg), |  | ||||||
| 		options.NewOption("TLSConfig")(tc), |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	src := &s{} |  | ||||||
|  |  | ||||||
| 	if err := options.Apply(src, opts...); err != nil { |  | ||||||
| 		t.Fatal(err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if src.Wait == nil { |  | ||||||
| 		t.Fatalf("failed to set Wait %#+v", src) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if src.TLSConfig == nil { |  | ||||||
| 		t.Fatalf("failed to set TLSConfig %#+v", src) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if src.TLSConfig.InsecureSkipVerify != true { |  | ||||||
| 		t.Fatalf("failed to set TLSConfig %#+v", src) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestNested(t *testing.T) { |  | ||||||
| 	type server struct { |  | ||||||
| 		Address []string |  | ||||||
| 	} |  | ||||||
| 	type ownserver struct { |  | ||||||
| 		server |  | ||||||
| 		OwnField string |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	opts := []options.Option{ |  | ||||||
| 		options.Address("host:port"), |  | ||||||
| 		options.NewOption("OwnField")("fieldval"), |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	src := &ownserver{} |  | ||||||
|  |  | ||||||
| 	if err := options.Apply(src, opts...); err != nil { |  | ||||||
| 		t.Fatal(err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if src.Address[0] != "host:port" { |  | ||||||
| 		t.Fatalf("failed to set Address %#+v", src) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if src.OwnField != "fieldval" { |  | ||||||
| 		t.Fatalf("failed to set OwnField %#+v", src) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestAddress(t *testing.T) { |  | ||||||
| 	type s struct { |  | ||||||
| 		Address []string |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	opts := []options.Option{options.Address("host:port")} |  | ||||||
|  |  | ||||||
| 	src := &s{} |  | ||||||
|  |  | ||||||
| 	if err := options.Apply(src, opts...); err != nil { |  | ||||||
| 		t.Fatal(err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if src.Address[0] != "host:port" { |  | ||||||
| 		t.Fatalf("failed to set Address %#+v", src) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestNewOption(t *testing.T) { |  | ||||||
| 	type s struct { |  | ||||||
| 		Address []string |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	opts := []options.Option{options.NewOption("Address")("host1:port1", "host2:port2")} |  | ||||||
|  |  | ||||||
| 	src := &s{} |  | ||||||
|  |  | ||||||
| 	if err := options.Apply(src, opts...); err != nil { |  | ||||||
| 		t.Fatal(err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if src.Address[0] != "host1:port1" { |  | ||||||
| 		t.Fatalf("failed to set Address %#+v", src) |  | ||||||
| 	} |  | ||||||
| 	if src.Address[1] != "host2:port2" { |  | ||||||
| 		t.Fatalf("failed to set Address %#+v", src) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestArray(t *testing.T) { |  | ||||||
| 	type s struct { |  | ||||||
| 		Address [1]string |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	opts := []options.Option{options.NewOption("Address")("host:port", "host1:port1")} |  | ||||||
|  |  | ||||||
| 	src := &s{} |  | ||||||
|  |  | ||||||
| 	if err := options.Apply(src, opts...); err != nil { |  | ||||||
| 		t.Fatal(err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if src.Address[0] != "host:port" { |  | ||||||
| 		t.Fatalf("failed to set Address %#+v", src) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestMap(t *testing.T) { |  | ||||||
| 	type s struct { |  | ||||||
| 		Metadata map[string]string |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	opts := []options.Option{ |  | ||||||
| 		options.NewOption("Metadata")("key1", "val1"), |  | ||||||
| 		options.NewOption("Metadata")(map[string]string{"key2": "val2"}), |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	src := &s{} |  | ||||||
|  |  | ||||||
| 	if err := options.Apply(src, opts...); err != nil { |  | ||||||
| 		t.Fatal(err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if len(src.Metadata) != 2 { |  | ||||||
| 		t.Fatalf("failed to set Metadata %#+v", src) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if src.Metadata["key1"] != "val1" { |  | ||||||
| 		t.Fatalf("failed to set Metadata %#+v", src) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if src.Metadata["key2"] != "val2" { |  | ||||||
| 		t.Fatalf("failed to set Metadata %#+v", src) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
							
								
								
									
										577
									
								
								options/util.go
									
									
									
									
									
								
							
							
						
						
									
										577
									
								
								options/util.go
									
									
									
									
									
								
							| @@ -1,577 +0,0 @@ | |||||||
| package options |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"fmt" |  | ||||||
| 	"reflect" |  | ||||||
|  |  | ||||||
| 	"github.com/spf13/cast" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| func toInt8SliceE(i interface{}) ([]int8, error) { |  | ||||||
| 	if i == nil { |  | ||||||
| 		return []int8{}, fmt.Errorf("unable to cast %#v of type %T to []int8", i, i) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	switch v := i.(type) { |  | ||||||
| 	case []int8: |  | ||||||
| 		return v, nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	kind := reflect.TypeOf(i).Kind() |  | ||||||
| 	switch kind { |  | ||||||
| 	case reflect.Slice, reflect.Array: |  | ||||||
| 		s := reflect.ValueOf(i) |  | ||||||
| 		a := make([]int8, s.Len()) |  | ||||||
| 		for j := 0; j < s.Len(); j++ { |  | ||||||
| 			val, err := cast.ToInt8E(s.Index(j).Interface()) |  | ||||||
| 			if err != nil { |  | ||||||
| 				return []int8{}, fmt.Errorf("unable to cast %#v of type %T to []int8", i, i) |  | ||||||
| 			} |  | ||||||
| 			a[j] = val |  | ||||||
| 		} |  | ||||||
| 		return a, nil |  | ||||||
| 	default: |  | ||||||
| 		return []int8{}, fmt.Errorf("unable to cast %#v of type %T to []int8", i, i) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func toInt16SliceE(i interface{}) ([]int16, error) { |  | ||||||
| 	if i == nil { |  | ||||||
| 		return []int16{}, fmt.Errorf("unable to cast %#v of type %T to []int16", i, i) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	switch v := i.(type) { |  | ||||||
| 	case []int16: |  | ||||||
| 		return v, nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	kind := reflect.TypeOf(i).Kind() |  | ||||||
| 	switch kind { |  | ||||||
| 	case reflect.Slice, reflect.Array: |  | ||||||
| 		s := reflect.ValueOf(i) |  | ||||||
| 		a := make([]int16, s.Len()) |  | ||||||
| 		for j := 0; j < s.Len(); j++ { |  | ||||||
| 			val, err := cast.ToInt16E(s.Index(j).Interface()) |  | ||||||
| 			if err != nil { |  | ||||||
| 				return []int16{}, fmt.Errorf("unable to cast %#v of type %T to []int16", i, i) |  | ||||||
| 			} |  | ||||||
| 			a[j] = val |  | ||||||
| 		} |  | ||||||
| 		return a, nil |  | ||||||
| 	default: |  | ||||||
| 		return []int16{}, fmt.Errorf("unable to cast %#v of type %T to []int16", i, i) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func toInt32SliceE(i interface{}) ([]int32, error) { |  | ||||||
| 	if i == nil { |  | ||||||
| 		return []int32{}, fmt.Errorf("unable to cast %#v of type %T to []int32", i, i) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	switch v := i.(type) { |  | ||||||
| 	case []int32: |  | ||||||
| 		return v, nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	kind := reflect.TypeOf(i).Kind() |  | ||||||
| 	switch kind { |  | ||||||
| 	case reflect.Slice, reflect.Array: |  | ||||||
| 		s := reflect.ValueOf(i) |  | ||||||
| 		a := make([]int32, s.Len()) |  | ||||||
| 		for j := 0; j < s.Len(); j++ { |  | ||||||
| 			val, err := cast.ToInt32E(s.Index(j).Interface()) |  | ||||||
| 			if err != nil { |  | ||||||
| 				return []int32{}, fmt.Errorf("unable to cast %#v of type %T to []int32", i, i) |  | ||||||
| 			} |  | ||||||
| 			a[j] = val |  | ||||||
| 		} |  | ||||||
| 		return a, nil |  | ||||||
| 	default: |  | ||||||
| 		return []int32{}, fmt.Errorf("unable to cast %#v of type %T to []int32", i, i) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func toInt64SliceE(i interface{}) ([]int64, error) { |  | ||||||
| 	if i == nil { |  | ||||||
| 		return []int64{}, fmt.Errorf("unable to cast %#v of type %T to []int64", i, i) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	switch v := i.(type) { |  | ||||||
| 	case []int64: |  | ||||||
| 		return v, nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	kind := reflect.TypeOf(i).Kind() |  | ||||||
| 	switch kind { |  | ||||||
| 	case reflect.Slice, reflect.Array: |  | ||||||
| 		s := reflect.ValueOf(i) |  | ||||||
| 		a := make([]int64, s.Len()) |  | ||||||
| 		for j := 0; j < s.Len(); j++ { |  | ||||||
| 			val, err := cast.ToInt64E(s.Index(j).Interface()) |  | ||||||
| 			if err != nil { |  | ||||||
| 				return []int64{}, fmt.Errorf("unable to cast %#v of type %T to []int64", i, i) |  | ||||||
| 			} |  | ||||||
| 			a[j] = val |  | ||||||
| 		} |  | ||||||
| 		return a, nil |  | ||||||
| 	default: |  | ||||||
| 		return []int64{}, fmt.Errorf("unable to cast %#v of type %T to []int64", i, i) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func toUintSliceE(i interface{}) ([]uint, error) { |  | ||||||
| 	if i == nil { |  | ||||||
| 		return []uint{}, fmt.Errorf("unable to cast %#v of type %T to []uint", i, i) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	switch v := i.(type) { |  | ||||||
| 	case []uint: |  | ||||||
| 		return v, nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	kind := reflect.TypeOf(i).Kind() |  | ||||||
| 	switch kind { |  | ||||||
| 	case reflect.Slice, reflect.Array: |  | ||||||
| 		s := reflect.ValueOf(i) |  | ||||||
| 		a := make([]uint, s.Len()) |  | ||||||
| 		for j := 0; j < s.Len(); j++ { |  | ||||||
| 			val, err := cast.ToUintE(s.Index(j).Interface()) |  | ||||||
| 			if err != nil { |  | ||||||
| 				return []uint{}, fmt.Errorf("unable to cast %#v of type %T to []uint", i, i) |  | ||||||
| 			} |  | ||||||
| 			a[j] = val |  | ||||||
| 		} |  | ||||||
| 		return a, nil |  | ||||||
| 	default: |  | ||||||
| 		return []uint{}, fmt.Errorf("unable to cast %#v of type %T to []uint", i, i) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func toUint8SliceE(i interface{}) ([]uint8, error) { |  | ||||||
| 	if i == nil { |  | ||||||
| 		return []uint8{}, fmt.Errorf("unable to cast %#v of type %T to []uint8", i, i) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	switch v := i.(type) { |  | ||||||
| 	case []uint8: |  | ||||||
| 		return v, nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	kind := reflect.TypeOf(i).Kind() |  | ||||||
| 	switch kind { |  | ||||||
| 	case reflect.Slice, reflect.Array: |  | ||||||
| 		s := reflect.ValueOf(i) |  | ||||||
| 		a := make([]uint8, s.Len()) |  | ||||||
| 		for j := 0; j < s.Len(); j++ { |  | ||||||
| 			val, err := cast.ToUint8E(s.Index(j).Interface()) |  | ||||||
| 			if err != nil { |  | ||||||
| 				return []uint8{}, fmt.Errorf("unable to cast %#v of type %T to []uint8", i, i) |  | ||||||
| 			} |  | ||||||
| 			a[j] = val |  | ||||||
| 		} |  | ||||||
| 		return a, nil |  | ||||||
| 	default: |  | ||||||
| 		return []uint8{}, fmt.Errorf("unable to cast %#v of type %T to []uint8", i, i) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func toUint16SliceE(i interface{}) ([]uint16, error) { |  | ||||||
| 	if i == nil { |  | ||||||
| 		return []uint16{}, fmt.Errorf("unable to cast %#v of type %T to []uint16", i, i) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	switch v := i.(type) { |  | ||||||
| 	case []uint16: |  | ||||||
| 		return v, nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	kind := reflect.TypeOf(i).Kind() |  | ||||||
| 	switch kind { |  | ||||||
| 	case reflect.Slice, reflect.Array: |  | ||||||
| 		s := reflect.ValueOf(i) |  | ||||||
| 		a := make([]uint16, s.Len()) |  | ||||||
| 		for j := 0; j < s.Len(); j++ { |  | ||||||
| 			val, err := cast.ToUint16E(s.Index(j).Interface()) |  | ||||||
| 			if err != nil { |  | ||||||
| 				return []uint16{}, fmt.Errorf("unable to cast %#v of type %T to []uint16", i, i) |  | ||||||
| 			} |  | ||||||
| 			a[j] = val |  | ||||||
| 		} |  | ||||||
| 		return a, nil |  | ||||||
| 	default: |  | ||||||
| 		return []uint16{}, fmt.Errorf("unable to cast %#v of type %T to []uint16", i, i) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func toUint32SliceE(i interface{}) ([]uint32, error) { |  | ||||||
| 	if i == nil { |  | ||||||
| 		return []uint32{}, fmt.Errorf("unable to cast %#v of type %T to []uint32", i, i) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	switch v := i.(type) { |  | ||||||
| 	case []uint32: |  | ||||||
| 		return v, nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	kind := reflect.TypeOf(i).Kind() |  | ||||||
| 	switch kind { |  | ||||||
| 	case reflect.Slice, reflect.Array: |  | ||||||
| 		s := reflect.ValueOf(i) |  | ||||||
| 		a := make([]uint32, s.Len()) |  | ||||||
| 		for j := 0; j < s.Len(); j++ { |  | ||||||
| 			val, err := cast.ToUint32E(s.Index(j).Interface()) |  | ||||||
| 			if err != nil { |  | ||||||
| 				return []uint32{}, fmt.Errorf("unable to cast %#v of type %T to []uint32", i, i) |  | ||||||
| 			} |  | ||||||
| 			a[j] = val |  | ||||||
| 		} |  | ||||||
| 		return a, nil |  | ||||||
| 	default: |  | ||||||
| 		return []uint32{}, fmt.Errorf("unable to cast %#v of type %T to []uint32", i, i) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func toUint64SliceE(i interface{}) ([]uint64, error) { |  | ||||||
| 	if i == nil { |  | ||||||
| 		return []uint64{}, fmt.Errorf("unable to cast %#v of type %T to []uint64", i, i) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	switch v := i.(type) { |  | ||||||
| 	case []uint64: |  | ||||||
| 		return v, nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	kind := reflect.TypeOf(i).Kind() |  | ||||||
| 	switch kind { |  | ||||||
| 	case reflect.Slice, reflect.Array: |  | ||||||
| 		s := reflect.ValueOf(i) |  | ||||||
| 		a := make([]uint64, s.Len()) |  | ||||||
| 		for j := 0; j < s.Len(); j++ { |  | ||||||
| 			val, err := cast.ToUint64E(s.Index(j).Interface()) |  | ||||||
| 			if err != nil { |  | ||||||
| 				return []uint64{}, fmt.Errorf("unable to cast %#v of type %T to []uint64", i, i) |  | ||||||
| 			} |  | ||||||
| 			a[j] = val |  | ||||||
| 		} |  | ||||||
| 		return a, nil |  | ||||||
| 	default: |  | ||||||
| 		return []uint64{}, fmt.Errorf("unable to cast %#v of type %T to []uint64", i, i) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func toFloat32SliceE(i interface{}) ([]float32, error) { |  | ||||||
| 	if i == nil { |  | ||||||
| 		return []float32{}, fmt.Errorf("unable to cast %#v of type %T to []float32", i, i) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	switch v := i.(type) { |  | ||||||
| 	case []float32: |  | ||||||
| 		return v, nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	kind := reflect.TypeOf(i).Kind() |  | ||||||
| 	switch kind { |  | ||||||
| 	case reflect.Slice, reflect.Array: |  | ||||||
| 		s := reflect.ValueOf(i) |  | ||||||
| 		a := make([]float32, s.Len()) |  | ||||||
| 		for j := 0; j < s.Len(); j++ { |  | ||||||
| 			val, err := cast.ToFloat32E(s.Index(j).Interface()) |  | ||||||
| 			if err != nil { |  | ||||||
| 				return []float32{}, fmt.Errorf("unable to cast %#v of type %T to []float32", i, i) |  | ||||||
| 			} |  | ||||||
| 			a[j] = val |  | ||||||
| 		} |  | ||||||
| 		return a, nil |  | ||||||
| 	default: |  | ||||||
| 		return []float32{}, fmt.Errorf("unable to cast %#v of type %T to []float32", i, i) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func toFloat64SliceE(i interface{}) ([]float64, error) { |  | ||||||
| 	if i == nil { |  | ||||||
| 		return []float64{}, fmt.Errorf("unable to cast %#v of type %T to []float64", i, i) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	switch v := i.(type) { |  | ||||||
| 	case []float64: |  | ||||||
| 		return v, nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	kind := reflect.TypeOf(i).Kind() |  | ||||||
| 	switch kind { |  | ||||||
| 	case reflect.Slice, reflect.Array: |  | ||||||
| 		s := reflect.ValueOf(i) |  | ||||||
| 		a := make([]float64, s.Len()) |  | ||||||
| 		for j := 0; j < s.Len(); j++ { |  | ||||||
| 			val, err := cast.ToFloat64E(s.Index(j).Interface()) |  | ||||||
| 			if err != nil { |  | ||||||
| 				return []float64{}, fmt.Errorf("unable to cast %#v of type %T to []float64", i, i) |  | ||||||
| 			} |  | ||||||
| 			a[j] = val |  | ||||||
| 		} |  | ||||||
| 		return a, nil |  | ||||||
| 	default: |  | ||||||
| 		return []float64{}, fmt.Errorf("unable to cast %#v of type %T to []float32", i, i) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func setMap(src interface{}, dst interface{}) error { |  | ||||||
| 	var err error |  | ||||||
|  |  | ||||||
| 	if src == nil { |  | ||||||
| 		return fmt.Errorf("unable to cast %#v of type %T", src, src) |  | ||||||
| 	} |  | ||||||
| 	if dst == nil { |  | ||||||
| 		return fmt.Errorf("unable to cast %#v of type %T", dst, dst) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	val := reflect.ValueOf(src) |  | ||||||
|  |  | ||||||
| 	keyKind := val.Type().Key().Kind() |  | ||||||
| 	valKind := val.Type().Elem().Kind() |  | ||||||
|  |  | ||||||
| 	switch v := dst.(type) { |  | ||||||
| 	case []interface{}: |  | ||||||
| 		if len(v) == 1 { |  | ||||||
| 			dstVal := reflect.ValueOf(v[0]) |  | ||||||
| 			if dstVal.Kind() != reflect.Map { |  | ||||||
| 				return nil |  | ||||||
| 			} |  | ||||||
| 			mapIter := dstVal.MapRange() |  | ||||||
| 			for mapIter.Next() { |  | ||||||
| 				var ( |  | ||||||
| 					keyVal interface{} |  | ||||||
| 					valVal interface{} |  | ||||||
| 				) |  | ||||||
| 				switch keyKind { |  | ||||||
| 				case reflect.Bool: |  | ||||||
| 					keyVal, err = cast.ToBoolE(mapIter.Key()) |  | ||||||
| 				case reflect.String: |  | ||||||
| 					keyVal, err = cast.ToStringE(mapIter.Key()) |  | ||||||
| 				case reflect.Float32: |  | ||||||
| 					keyVal, err = cast.ToFloat32E(mapIter.Key()) |  | ||||||
| 				case reflect.Float64: |  | ||||||
| 					keyVal, err = cast.ToFloat64E(mapIter.Key()) |  | ||||||
| 				case reflect.Int8: |  | ||||||
| 					keyVal, err = cast.ToInt8E(mapIter.Key()) |  | ||||||
| 				case reflect.Int: |  | ||||||
| 					keyVal, err = cast.ToIntE(mapIter.Key()) |  | ||||||
| 				case reflect.Int16: |  | ||||||
| 					keyVal, err = cast.ToInt16E(mapIter.Key()) |  | ||||||
| 				case reflect.Int32: |  | ||||||
| 					keyVal, err = cast.ToInt32E(mapIter.Key()) |  | ||||||
| 				case reflect.Int64: |  | ||||||
| 					keyVal, err = cast.ToInt64E(mapIter.Key()) |  | ||||||
| 				case reflect.Uint8: |  | ||||||
| 					keyVal, err = cast.ToUint8E(mapIter.Key()) |  | ||||||
| 				case reflect.Uint: |  | ||||||
| 					keyVal, err = cast.ToUintE(mapIter.Key()) |  | ||||||
| 				case reflect.Uint16: |  | ||||||
| 					keyVal, err = cast.ToUint16E(mapIter.Key()) |  | ||||||
| 				case reflect.Uint32: |  | ||||||
| 					keyVal, err = cast.ToUint32E(mapIter.Key()) |  | ||||||
| 				case reflect.Uint64: |  | ||||||
| 					keyVal, err = cast.ToUint64E(mapIter.Key()) |  | ||||||
| 				} |  | ||||||
| 				if err != nil { |  | ||||||
| 					return err |  | ||||||
| 				} |  | ||||||
| 				switch valKind { |  | ||||||
| 				case reflect.Bool: |  | ||||||
| 					valVal, err = cast.ToBoolE(mapIter.Value()) |  | ||||||
| 				case reflect.String: |  | ||||||
| 					valVal, err = cast.ToStringE(mapIter.Value()) |  | ||||||
| 				case reflect.Float32: |  | ||||||
| 					valVal, err = cast.ToFloat32E(mapIter.Value()) |  | ||||||
| 				case reflect.Float64: |  | ||||||
| 					valVal, err = cast.ToFloat64E(mapIter.Value()) |  | ||||||
| 				case reflect.Int8: |  | ||||||
| 					valVal, err = cast.ToInt8E(mapIter.Value()) |  | ||||||
| 				case reflect.Int: |  | ||||||
| 					valVal, err = cast.ToIntE(mapIter.Value()) |  | ||||||
| 				case reflect.Int16: |  | ||||||
| 					valVal, err = cast.ToInt16E(mapIter.Value()) |  | ||||||
| 				case reflect.Int32: |  | ||||||
| 					valVal, err = cast.ToInt32E(mapIter.Value()) |  | ||||||
| 				case reflect.Int64: |  | ||||||
| 					valVal, err = cast.ToInt64E(mapIter.Value()) |  | ||||||
| 				case reflect.Uint8: |  | ||||||
| 					valVal, err = cast.ToUint8E(mapIter.Value()) |  | ||||||
| 				case reflect.Uint: |  | ||||||
| 					valVal, err = cast.ToUintE(mapIter.Value()) |  | ||||||
| 				case reflect.Uint16: |  | ||||||
| 					valVal, err = cast.ToUint16E(mapIter.Value()) |  | ||||||
| 				case reflect.Uint32: |  | ||||||
| 					valVal, err = cast.ToUint32E(mapIter.Value()) |  | ||||||
| 				case reflect.Uint64: |  | ||||||
| 					valVal, err = cast.ToUint64E(mapIter.Value()) |  | ||||||
| 				} |  | ||||||
| 				if err != nil { |  | ||||||
| 					return err |  | ||||||
| 				} |  | ||||||
|  |  | ||||||
| 				val.SetMapIndex(reflect.ValueOf(keyVal), reflect.ValueOf(valVal)) |  | ||||||
| 			} |  | ||||||
| 			return nil |  | ||||||
| 		} |  | ||||||
| 		if l := len(v) % 2; l == 1 { |  | ||||||
| 			v = v[:len(v)-1] |  | ||||||
| 		} |  | ||||||
| 		var ( |  | ||||||
| 			keyVal interface{} |  | ||||||
| 			valVal interface{} |  | ||||||
| 		) |  | ||||||
| 		for i := 0; i < len(v); i += 2 { |  | ||||||
| 			switch keyKind { |  | ||||||
| 			case reflect.Bool: |  | ||||||
| 				keyVal, err = cast.ToBoolE(v[i]) |  | ||||||
| 			case reflect.String: |  | ||||||
| 				keyVal, err = cast.ToStringE(v[i]) |  | ||||||
| 			case reflect.Float32: |  | ||||||
| 				keyVal, err = cast.ToFloat32E(v[i]) |  | ||||||
| 			case reflect.Float64: |  | ||||||
| 				keyVal, err = cast.ToFloat64E(v[i]) |  | ||||||
| 			case reflect.Int8: |  | ||||||
| 				keyVal, err = cast.ToInt8E(v[i]) |  | ||||||
| 			case reflect.Int: |  | ||||||
| 				keyVal, err = cast.ToIntE(v[i]) |  | ||||||
| 			case reflect.Int16: |  | ||||||
| 				keyVal, err = cast.ToInt16E(v[i]) |  | ||||||
| 			case reflect.Int32: |  | ||||||
| 				keyVal, err = cast.ToInt32E(v[i]) |  | ||||||
| 			case reflect.Int64: |  | ||||||
| 				keyVal, err = cast.ToInt64E(v[i]) |  | ||||||
| 			case reflect.Uint8: |  | ||||||
| 				keyVal, err = cast.ToUint8E(v[i]) |  | ||||||
| 			case reflect.Uint: |  | ||||||
| 				keyVal, err = cast.ToUintE(v[i]) |  | ||||||
| 			case reflect.Uint16: |  | ||||||
| 				keyVal, err = cast.ToUint16E(v[i]) |  | ||||||
| 			case reflect.Uint32: |  | ||||||
| 				keyVal, err = cast.ToUint32E(v[i]) |  | ||||||
| 			case reflect.Uint64: |  | ||||||
| 				keyVal, err = cast.ToUint64E(v[i]) |  | ||||||
| 			} |  | ||||||
| 			if err != nil { |  | ||||||
| 				return err |  | ||||||
| 			} |  | ||||||
| 			switch valKind { |  | ||||||
| 			case reflect.Bool: |  | ||||||
| 				valVal, err = cast.ToBoolE(v[i+1]) |  | ||||||
| 			case reflect.String: |  | ||||||
| 				valVal, err = cast.ToStringE(v[i+1]) |  | ||||||
| 			case reflect.Float32: |  | ||||||
| 				valVal, err = cast.ToFloat32E(v[i+1]) |  | ||||||
| 			case reflect.Float64: |  | ||||||
| 				valVal, err = cast.ToFloat64E(v[i+1]) |  | ||||||
| 			case reflect.Int8: |  | ||||||
| 				valVal, err = cast.ToInt8E(v[i+1]) |  | ||||||
| 			case reflect.Int: |  | ||||||
| 				valVal, err = cast.ToIntE(v[i+1]) |  | ||||||
| 			case reflect.Int16: |  | ||||||
| 				valVal, err = cast.ToInt16E(v[i+1]) |  | ||||||
| 			case reflect.Int32: |  | ||||||
| 				valVal, err = cast.ToInt32E(v[i+1]) |  | ||||||
| 			case reflect.Int64: |  | ||||||
| 				valVal, err = cast.ToInt64E(v[i+1]) |  | ||||||
| 			case reflect.Uint8: |  | ||||||
| 				valVal, err = cast.ToUint8E(v[i+1]) |  | ||||||
| 			case reflect.Uint: |  | ||||||
| 				valVal, err = cast.ToUintE(v[i+1]) |  | ||||||
| 			case reflect.Uint16: |  | ||||||
| 				valVal, err = cast.ToUint16E(v[i+1]) |  | ||||||
| 			case reflect.Uint32: |  | ||||||
| 				valVal, err = cast.ToUint32E(v[i+1]) |  | ||||||
| 			case reflect.Uint64: |  | ||||||
| 				valVal, err = cast.ToUint64E(v[i+1]) |  | ||||||
| 			} |  | ||||||
| 			if err != nil { |  | ||||||
| 				return err |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			val.SetMapIndex(reflect.ValueOf(keyVal), reflect.ValueOf(valVal)) |  | ||||||
| 		} |  | ||||||
| 	default: |  | ||||||
| 		dstVal := reflect.ValueOf(dst) |  | ||||||
| 		if dstVal.Kind() != reflect.Map { |  | ||||||
| 			return nil |  | ||||||
| 		} |  | ||||||
| 		mapIter := dstVal.MapRange() |  | ||||||
| 		for mapIter.Next() { |  | ||||||
| 			var ( |  | ||||||
| 				keyVal interface{} |  | ||||||
| 				valVal interface{} |  | ||||||
| 			) |  | ||||||
| 			switch keyKind { |  | ||||||
| 			case reflect.Bool: |  | ||||||
| 				keyVal, err = cast.ToBoolE(mapIter.Key()) |  | ||||||
| 			case reflect.String: |  | ||||||
| 				keyVal, err = cast.ToStringE(mapIter.Key()) |  | ||||||
| 			case reflect.Float32: |  | ||||||
| 				keyVal, err = cast.ToFloat32E(mapIter.Key()) |  | ||||||
| 			case reflect.Float64: |  | ||||||
| 				keyVal, err = cast.ToFloat64E(mapIter.Key()) |  | ||||||
| 			case reflect.Int8: |  | ||||||
| 				keyVal, err = cast.ToInt8E(mapIter.Key()) |  | ||||||
| 			case reflect.Int: |  | ||||||
| 				keyVal, err = cast.ToIntE(mapIter.Key()) |  | ||||||
| 			case reflect.Int16: |  | ||||||
| 				keyVal, err = cast.ToInt16E(mapIter.Key()) |  | ||||||
| 			case reflect.Int32: |  | ||||||
| 				keyVal, err = cast.ToInt32E(mapIter.Key()) |  | ||||||
| 			case reflect.Int64: |  | ||||||
| 				keyVal, err = cast.ToInt64E(mapIter.Key()) |  | ||||||
| 			case reflect.Uint8: |  | ||||||
| 				keyVal, err = cast.ToUint8E(mapIter.Key()) |  | ||||||
| 			case reflect.Uint: |  | ||||||
| 				keyVal, err = cast.ToUintE(mapIter.Key()) |  | ||||||
| 			case reflect.Uint16: |  | ||||||
| 				keyVal, err = cast.ToUint16E(mapIter.Key()) |  | ||||||
| 			case reflect.Uint32: |  | ||||||
| 				keyVal, err = cast.ToUint32E(mapIter.Key()) |  | ||||||
| 			case reflect.Uint64: |  | ||||||
| 				keyVal, err = cast.ToUint64E(mapIter.Key()) |  | ||||||
| 			} |  | ||||||
| 			if err != nil { |  | ||||||
| 				return err |  | ||||||
| 			} |  | ||||||
| 			switch valKind { |  | ||||||
| 			case reflect.Bool: |  | ||||||
| 				valVal, err = cast.ToBoolE(mapIter.Value()) |  | ||||||
| 			case reflect.String: |  | ||||||
| 				valVal, err = cast.ToStringE(mapIter.Value()) |  | ||||||
| 			case reflect.Float32: |  | ||||||
| 				valVal, err = cast.ToFloat32E(mapIter.Value()) |  | ||||||
| 			case reflect.Float64: |  | ||||||
| 				valVal, err = cast.ToFloat64E(mapIter.Value()) |  | ||||||
| 			case reflect.Int8: |  | ||||||
| 				valVal, err = cast.ToInt8E(mapIter.Value()) |  | ||||||
| 			case reflect.Int: |  | ||||||
| 				valVal, err = cast.ToIntE(mapIter.Value()) |  | ||||||
| 			case reflect.Int16: |  | ||||||
| 				valVal, err = cast.ToInt16E(mapIter.Value()) |  | ||||||
| 			case reflect.Int32: |  | ||||||
| 				valVal, err = cast.ToInt32E(mapIter.Value()) |  | ||||||
| 			case reflect.Int64: |  | ||||||
| 				valVal, err = cast.ToInt64E(mapIter.Value()) |  | ||||||
| 			case reflect.Uint8: |  | ||||||
| 				valVal, err = cast.ToUint8E(mapIter.Value()) |  | ||||||
| 			case reflect.Uint: |  | ||||||
| 				valVal, err = cast.ToUintE(mapIter.Value()) |  | ||||||
| 			case reflect.Uint16: |  | ||||||
| 				valVal, err = cast.ToUint16E(mapIter.Value()) |  | ||||||
| 			case reflect.Uint32: |  | ||||||
| 				valVal, err = cast.ToUint32E(mapIter.Value()) |  | ||||||
| 			case reflect.Uint64: |  | ||||||
| 				valVal, err = cast.ToUint64E(mapIter.Value()) |  | ||||||
| 			} |  | ||||||
| 			if err != nil { |  | ||||||
| 				return err |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			val.SetMapIndex(reflect.ValueOf(keyVal), reflect.ValueOf(valVal)) |  | ||||||
| 		} |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
| @@ -7,7 +7,7 @@ import ( | |||||||
| 	"net/http/pprof" | 	"net/http/pprof" | ||||||
| 	"sync" | 	"sync" | ||||||
|  |  | ||||||
| 	profile "go.unistack.org/micro/v4/profiler" | 	profile "go.unistack.org/micro/v3/profiler" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type httpProfile struct { | type httpProfile struct { | ||||||
|   | |||||||
| @@ -9,7 +9,7 @@ import ( | |||||||
| 	"sync" | 	"sync" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	profile "go.unistack.org/micro/v4/profiler" | 	profile "go.unistack.org/micro/v3/profiler" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type profiler struct { | type profiler struct { | ||||||
|   | |||||||
							
								
								
									
										98
									
								
								proxy/options.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										98
									
								
								proxy/options.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,98 @@ | |||||||
|  | // Package proxy is a transparent proxy built on the micro/server | ||||||
|  | package proxy | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"go.unistack.org/micro/v3/client" | ||||||
|  | 	"go.unistack.org/micro/v3/logger" | ||||||
|  | 	"go.unistack.org/micro/v3/meter" | ||||||
|  | 	"go.unistack.org/micro/v3/router" | ||||||
|  | 	"go.unistack.org/micro/v3/tracer" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Options for proxy | ||||||
|  | type Options struct { | ||||||
|  | 	// Tracer used for tracing | ||||||
|  | 	Tracer tracer.Tracer | ||||||
|  | 	// Client for communication | ||||||
|  | 	Client client.Client | ||||||
|  | 	// Router for routing | ||||||
|  | 	Router router.Router | ||||||
|  | 	// Logger used for logging | ||||||
|  | 	Logger logger.Logger | ||||||
|  | 	// Meter used for metrics | ||||||
|  | 	Meter meter.Meter | ||||||
|  | 	// Links holds the communication links | ||||||
|  | 	Links map[string]client.Client | ||||||
|  | 	// Endpoint holds the destination address | ||||||
|  | 	Endpoint string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Option func signature | ||||||
|  | type Option func(o *Options) | ||||||
|  |  | ||||||
|  | // NewOptions returns new options struct that filled by opts | ||||||
|  | func NewOptions(opts ...Option) Options { | ||||||
|  | 	options := Options{ | ||||||
|  | 		Logger: logger.DefaultLogger, | ||||||
|  | 		Meter:  meter.DefaultMeter, | ||||||
|  | 		Tracer: tracer.DefaultTracer, | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for _, o := range opts { | ||||||
|  | 		o(&options) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return options | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithEndpoint sets a proxy endpoint | ||||||
|  | func WithEndpoint(e string) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Endpoint = e | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithClient sets the client | ||||||
|  | func WithClient(c client.Client) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Client = c | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithRouter specifies the router to use | ||||||
|  | func WithRouter(r router.Router) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Router = r | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithLogger specifies the logger to use | ||||||
|  | func WithLogger(l logger.Logger) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Logger = l | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithMeter specifies the meter to use | ||||||
|  | func WithMeter(m meter.Meter) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Meter = m | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithLink sets a link for outbound requests | ||||||
|  | func WithLink(name string, c client.Client) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		if o.Links == nil { | ||||||
|  | 			o.Links = make(map[string]client.Client) | ||||||
|  | 		} | ||||||
|  | 		o.Links[name] = c | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Tracer to be used for tracing | ||||||
|  | func Tracer(t tracer.Tracer) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Tracer = t | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										21
									
								
								proxy/proxy.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										21
									
								
								proxy/proxy.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,21 @@ | |||||||
|  | // Package proxy is a transparent proxy built on the micro/server | ||||||
|  | package proxy | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v3/server" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // DefaultEndpoint holds default proxy address | ||||||
|  | var DefaultEndpoint = "localhost:9090" | ||||||
|  |  | ||||||
|  | // Proxy can be used as a proxy server for micro services | ||||||
|  | type Proxy interface { | ||||||
|  | 	// ProcessMessage handles inbound messages | ||||||
|  | 	ProcessMessage(context.Context, server.Message) error | ||||||
|  | 	// ServeRequest handles inbound requests | ||||||
|  | 	ServeRequest(context.Context, server.Request, server.Response) error | ||||||
|  | 	// Name of the proxy protocol | ||||||
|  | 	String() string | ||||||
|  | } | ||||||
| @@ -1,9 +1,12 @@ | |||||||
| package register | package register | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"fmt" | ||||||
| 	"reflect" | 	"reflect" | ||||||
| 	"unicode" | 	"unicode" | ||||||
| 	"unicode/utf8" | 	"unicode/utf8" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v3/metadata" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // ExtractValue from reflect.Type from specified depth | // ExtractValue from reflect.Type from specified depth | ||||||
| @@ -35,6 +38,53 @@ func ExtractValue(v reflect.Type, d int) string { | |||||||
| 	return v.Name() | 	return v.Name() | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // ExtractEndpoint extract *Endpoint from reflect.Method | ||||||
|  | func ExtractEndpoint(method reflect.Method) *Endpoint { | ||||||
|  | 	if method.PkgPath != "" { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	var rspType, reqType reflect.Type | ||||||
|  | 	var stream bool | ||||||
|  | 	mt := method.Type | ||||||
|  |  | ||||||
|  | 	switch mt.NumIn() { | ||||||
|  | 	case 3: | ||||||
|  | 		reqType = mt.In(1) | ||||||
|  | 		rspType = mt.In(2) | ||||||
|  | 	case 4: | ||||||
|  | 		reqType = mt.In(2) | ||||||
|  | 		rspType = mt.In(3) | ||||||
|  | 	default: | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// are we dealing with a stream? | ||||||
|  | 	switch rspType.Kind() { | ||||||
|  | 	case reflect.Func, reflect.Interface: | ||||||
|  | 		stream = true | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	request := ExtractValue(reqType, 0) | ||||||
|  | 	response := ExtractValue(rspType, 0) | ||||||
|  | 	if request == "" || response == "" { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	ep := &Endpoint{ | ||||||
|  | 		Name:     method.Name, | ||||||
|  | 		Request:  request, | ||||||
|  | 		Response: response, | ||||||
|  | 		Metadata: metadata.New(0), | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if stream { | ||||||
|  | 		ep.Metadata.Set("stream", fmt.Sprintf("%v", stream)) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return ep | ||||||
|  | } | ||||||
|  |  | ||||||
| // ExtractSubValue exctact *Value from reflect.Type | // ExtractSubValue exctact *Value from reflect.Type | ||||||
| func ExtractSubValue(typ reflect.Type) string { | func ExtractSubValue(typ reflect.Type) string { | ||||||
| 	var reqType reflect.Type | 	var reqType reflect.Type | ||||||
|   | |||||||
| @@ -2,6 +2,8 @@ package register | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
|  | 	"reflect" | ||||||
|  | 	"testing" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type TestHandler struct{} | type TestHandler struct{} | ||||||
| @@ -13,3 +15,40 @@ type TestResponse struct{} | |||||||
| func (t *TestHandler) Test(ctx context.Context, req *TestRequest, rsp *TestResponse) error { | func (t *TestHandler) Test(ctx context.Context, req *TestRequest, rsp *TestResponse) error { | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func TestExtractEndpoint(t *testing.T) { | ||||||
|  | 	handler := &TestHandler{} | ||||||
|  | 	typ := reflect.TypeOf(handler) | ||||||
|  |  | ||||||
|  | 	var endpoints []*Endpoint | ||||||
|  |  | ||||||
|  | 	for m := 0; m < typ.NumMethod(); m++ { | ||||||
|  | 		if e := ExtractEndpoint(typ.Method(m)); e != nil { | ||||||
|  | 			endpoints = append(endpoints, e) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if i := len(endpoints); i != 1 { | ||||||
|  | 		t.Fatalf("Expected 1 endpoint, have %d", i) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if endpoints[0].Name != "Test" { | ||||||
|  | 		t.Fatalf("Expected handler Test, got %s", endpoints[0].Name) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if endpoints[0].Request == "" { | ||||||
|  | 		t.Fatal("Expected non nil Request") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if endpoints[0].Response == "" { | ||||||
|  | 		t.Fatal("Expected non nil Request") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if endpoints[0].Request != "TestRequest" { | ||||||
|  | 		t.Fatalf("Expected TestRequest got %s", endpoints[0].Request) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if endpoints[0].Response != "TestResponse" { | ||||||
|  | 		t.Fatalf("Expected TestResponse got %s", endpoints[0].Response) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -6,10 +6,9 @@ import ( | |||||||
| 	"sync" | 	"sync" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/logger" | 	"go.unistack.org/micro/v3/logger" | ||||||
| 	"go.unistack.org/micro/v4/metadata" | 	"go.unistack.org/micro/v3/register" | ||||||
| 	"go.unistack.org/micro/v4/register" | 	"go.unistack.org/micro/v3/util/id" | ||||||
| 	"go.unistack.org/micro/v4/util/id" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| @@ -26,7 +25,9 @@ type node struct { | |||||||
| type record struct { | type record struct { | ||||||
| 	Name      string | 	Name      string | ||||||
| 	Version   string | 	Version   string | ||||||
|  | 	Metadata  map[string]string | ||||||
| 	Nodes     map[string]*node | 	Nodes     map[string]*node | ||||||
|  | 	Endpoints []*register.Endpoint | ||||||
| } | } | ||||||
|  |  | ||||||
| type memory struct { | type memory struct { | ||||||
| @@ -58,7 +59,7 @@ func (m *memory) ttlPrune() { | |||||||
|  |  | ||||||
| 	for range prune.C { | 	for range prune.C { | ||||||
| 		m.Lock() | 		m.Lock() | ||||||
| 		for namespace, services := range m.records { | 		for domain, services := range m.records { | ||||||
| 			for service, versions := range services { | 			for service, versions := range services { | ||||||
| 				for version, record := range versions { | 				for version, record := range versions { | ||||||
| 					for id, n := range record.Nodes { | 					for id, n := range record.Nodes { | ||||||
| @@ -66,7 +67,7 @@ func (m *memory) ttlPrune() { | |||||||
| 							if m.opts.Logger.V(logger.DebugLevel) { | 							if m.opts.Logger.V(logger.DebugLevel) { | ||||||
| 								m.opts.Logger.Debug(m.opts.Context, fmt.Sprintf("Register TTL expired for node %s of service %s", n.ID, service)) | 								m.opts.Logger.Debug(m.opts.Context, fmt.Sprintf("Register TTL expired for node %s of service %s", n.ID, service)) | ||||||
| 							} | 							} | ||||||
| 							delete(m.records[namespace][service][version].Nodes, id) | 							delete(m.records[domain][service][version].Nodes, id) | ||||||
| 						} | 						} | ||||||
| 					} | 					} | ||||||
| 				} | 				} | ||||||
| @@ -130,12 +131,17 @@ func (m *memory) Register(_ context.Context, s *register.Service, opts ...regist | |||||||
| 	options := register.NewRegisterOptions(opts...) | 	options := register.NewRegisterOptions(opts...) | ||||||
|  |  | ||||||
| 	// get the services for this domain from the register | 	// get the services for this domain from the register | ||||||
| 	srvs, ok := m.records[options.Namespace] | 	srvs, ok := m.records[options.Domain] | ||||||
| 	if !ok { | 	if !ok { | ||||||
| 		srvs = make(services) | 		srvs = make(services) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	s.Namespace = options.Namespace | 	// domain is set in metadata so it can be passed to watchers | ||||||
|  | 	if s.Metadata == nil { | ||||||
|  | 		s.Metadata = map[string]string{"domain": options.Domain} | ||||||
|  | 	} else { | ||||||
|  | 		s.Metadata["domain"] = options.Domain | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	// ensure the service name exists | 	// ensure the service name exists | ||||||
| 	r := serviceToRecord(s, options.TTL) | 	r := serviceToRecord(s, options.TTL) | ||||||
| @@ -148,8 +154,8 @@ func (m *memory) Register(_ context.Context, s *register.Service, opts ...regist | |||||||
| 		if m.opts.Logger.V(logger.DebugLevel) { | 		if m.opts.Logger.V(logger.DebugLevel) { | ||||||
| 			m.opts.Logger.Debug(m.opts.Context, fmt.Sprintf("Register added new service: %s, version: %s", s.Name, s.Version)) | 			m.opts.Logger.Debug(m.opts.Context, fmt.Sprintf("Register added new service: %s, version: %s", s.Name, s.Version)) | ||||||
| 		} | 		} | ||||||
| 		m.records[options.Namespace] = srvs | 		m.records[options.Domain] = srvs | ||||||
| 		go m.sendEvent(®ister.Result{Action: register.EventCreate, Service: s}) | 		go m.sendEvent(®ister.Result{Action: "create", Service: s}) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	var addedNodes bool | 	var addedNodes bool | ||||||
| @@ -160,14 +166,22 @@ func (m *memory) Register(_ context.Context, s *register.Service, opts ...regist | |||||||
| 			continue | 			continue | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		md := metadata.Copy(n.Metadata) | 		metadata := make(map[string]string, len(n.Metadata)) | ||||||
|  |  | ||||||
|  | 		// make copy of metadata | ||||||
|  | 		for k, v := range n.Metadata { | ||||||
|  | 			metadata[k] = v | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// set the domain | ||||||
|  | 		metadata["domain"] = options.Domain | ||||||
|  |  | ||||||
| 		// add the node | 		// add the node | ||||||
| 		srvs[s.Name][s.Version].Nodes[n.ID] = &node{ | 		srvs[s.Name][s.Version].Nodes[n.ID] = &node{ | ||||||
| 			Node: ®ister.Node{ | 			Node: ®ister.Node{ | ||||||
| 				ID:       n.ID, | 				ID:       n.ID, | ||||||
| 				Address:  n.Address, | 				Address:  n.Address, | ||||||
| 				Metadata: md, | 				Metadata: metadata, | ||||||
| 			}, | 			}, | ||||||
| 			TTL:      options.TTL, | 			TTL:      options.TTL, | ||||||
| 			LastSeen: time.Now(), | 			LastSeen: time.Now(), | ||||||
| @@ -180,7 +194,7 @@ func (m *memory) Register(_ context.Context, s *register.Service, opts ...regist | |||||||
| 		if m.opts.Logger.V(logger.DebugLevel) { | 		if m.opts.Logger.V(logger.DebugLevel) { | ||||||
| 			m.opts.Logger.Debug(m.opts.Context, fmt.Sprintf("Register added new node to service: %s, version: %s", s.Name, s.Version)) | 			m.opts.Logger.Debug(m.opts.Context, fmt.Sprintf("Register added new node to service: %s, version: %s", s.Name, s.Version)) | ||||||
| 		} | 		} | ||||||
| 		go m.sendEvent(®ister.Result{Action: register.EventUpdate, Service: s}) | 		go m.sendEvent(®ister.Result{Action: "update", Service: s}) | ||||||
| 	} else { | 	} else { | ||||||
| 		// refresh TTL and timestamp | 		// refresh TTL and timestamp | ||||||
| 		for _, n := range s.Nodes { | 		for _, n := range s.Nodes { | ||||||
| @@ -192,7 +206,7 @@ func (m *memory) Register(_ context.Context, s *register.Service, opts ...regist | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	m.records[options.Namespace] = srvs | 	m.records[options.Domain] = srvs | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -202,8 +216,15 @@ func (m *memory) Deregister(ctx context.Context, s *register.Service, opts ...re | |||||||
|  |  | ||||||
| 	options := register.NewDeregisterOptions(opts...) | 	options := register.NewDeregisterOptions(opts...) | ||||||
|  |  | ||||||
|  | 	// domain is set in metadata so it can be passed to watchers | ||||||
|  | 	if s.Metadata == nil { | ||||||
|  | 		s.Metadata = map[string]string{"domain": options.Domain} | ||||||
|  | 	} else { | ||||||
|  | 		s.Metadata["domain"] = options.Domain | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	// if the domain doesn't exist, there is nothing to deregister | 	// if the domain doesn't exist, there is nothing to deregister | ||||||
| 	services, ok := m.records[options.Namespace] | 	services, ok := m.records[options.Domain] | ||||||
| 	if !ok { | 	if !ok { | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
| @@ -232,16 +253,16 @@ func (m *memory) Deregister(ctx context.Context, s *register.Service, opts ...re | |||||||
| 	// if the nodes not empty, we replace the version in the store and exist, the rest of the logic | 	// if the nodes not empty, we replace the version in the store and exist, the rest of the logic | ||||||
| 	// is cleanup | 	// is cleanup | ||||||
| 	if len(version.Nodes) > 0 { | 	if len(version.Nodes) > 0 { | ||||||
| 		m.records[options.Namespace][s.Name][s.Version] = version | 		m.records[options.Domain][s.Name][s.Version] = version | ||||||
| 		go m.sendEvent(®ister.Result{Action: register.EventUpdate, Service: s}) | 		go m.sendEvent(®ister.Result{Action: "update", Service: s}) | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// if this version was the only version of the service, we can remove the whole service from the | 	// if this version was the only version of the service, we can remove the whole service from the | ||||||
| 	// register and exit | 	// register and exit | ||||||
| 	if len(versions) == 1 { | 	if len(versions) == 1 { | ||||||
| 		delete(m.records[options.Namespace], s.Name) | 		delete(m.records[options.Domain], s.Name) | ||||||
| 		go m.sendEvent(®ister.Result{Action: register.EventDelete, Service: s}) | 		go m.sendEvent(®ister.Result{Action: "delete", Service: s}) | ||||||
|  |  | ||||||
| 		if m.opts.Logger.V(logger.DebugLevel) { | 		if m.opts.Logger.V(logger.DebugLevel) { | ||||||
| 			m.opts.Logger.Debug(m.opts.Context, fmt.Sprintf("Register removed service: %s", s.Name)) | 			m.opts.Logger.Debug(m.opts.Context, fmt.Sprintf("Register removed service: %s", s.Name)) | ||||||
| @@ -250,8 +271,8 @@ func (m *memory) Deregister(ctx context.Context, s *register.Service, opts ...re | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// there are other versions of the service running, so only remove this version of it | 	// there are other versions of the service running, so only remove this version of it | ||||||
| 	delete(m.records[options.Namespace][s.Name], s.Version) | 	delete(m.records[options.Domain][s.Name], s.Version) | ||||||
| 	go m.sendEvent(®ister.Result{Action: register.EventDelete, Service: s}) | 	go m.sendEvent(®ister.Result{Action: "delete", Service: s}) | ||||||
| 	if m.opts.Logger.V(logger.DebugLevel) { | 	if m.opts.Logger.V(logger.DebugLevel) { | ||||||
| 		m.opts.Logger.Debug(m.opts.Context, fmt.Sprintf("Register removed service: %s, version: %s", s.Name, s.Version)) | 		m.opts.Logger.Debug(m.opts.Context, fmt.Sprintf("Register removed service: %s, version: %s", s.Name, s.Version)) | ||||||
| 	} | 	} | ||||||
| @@ -263,15 +284,15 @@ func (m *memory) LookupService(ctx context.Context, name string, opts ...registe | |||||||
| 	options := register.NewLookupOptions(opts...) | 	options := register.NewLookupOptions(opts...) | ||||||
|  |  | ||||||
| 	// if it's a wildcard domain, return from all domains | 	// if it's a wildcard domain, return from all domains | ||||||
| 	if options.Namespace == register.WildcardNamespace { | 	if options.Domain == register.WildcardDomain { | ||||||
| 		m.RLock() | 		m.RLock() | ||||||
| 		recs := m.records | 		recs := m.records | ||||||
| 		m.RUnlock() | 		m.RUnlock() | ||||||
|  |  | ||||||
| 		var services []*register.Service | 		var services []*register.Service | ||||||
|  |  | ||||||
| 		for namespace := range recs { | 		for domain := range recs { | ||||||
| 			srvs, err := m.LookupService(ctx, name, append(opts, register.LookupNamespace(namespace))...) | 			srvs, err := m.LookupService(ctx, name, append(opts, register.LookupDomain(domain))...) | ||||||
| 			if err == register.ErrNotFound { | 			if err == register.ErrNotFound { | ||||||
| 				continue | 				continue | ||||||
| 			} else if err != nil { | 			} else if err != nil { | ||||||
| @@ -290,7 +311,7 @@ func (m *memory) LookupService(ctx context.Context, name string, opts ...registe | |||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
|  |  | ||||||
| 	// check the domain exists | 	// check the domain exists | ||||||
| 	services, ok := m.records[options.Namespace] | 	services, ok := m.records[options.Domain] | ||||||
| 	if !ok { | 	if !ok { | ||||||
| 		return nil, register.ErrNotFound | 		return nil, register.ErrNotFound | ||||||
| 	} | 	} | ||||||
| @@ -307,7 +328,7 @@ func (m *memory) LookupService(ctx context.Context, name string, opts ...registe | |||||||
| 	var i int | 	var i int | ||||||
|  |  | ||||||
| 	for _, r := range versions { | 	for _, r := range versions { | ||||||
| 		result[i] = recordToService(r, options.Namespace) | 		result[i] = recordToService(r, options.Domain) | ||||||
| 		i++ | 		i++ | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -318,15 +339,15 @@ func (m *memory) ListServices(ctx context.Context, opts ...register.ListOption) | |||||||
| 	options := register.NewListOptions(opts...) | 	options := register.NewListOptions(opts...) | ||||||
|  |  | ||||||
| 	// if it's a wildcard domain, list from all domains | 	// if it's a wildcard domain, list from all domains | ||||||
| 	if options.Namespace == register.WildcardNamespace { | 	if options.Domain == register.WildcardDomain { | ||||||
| 		m.RLock() | 		m.RLock() | ||||||
| 		recs := m.records | 		recs := m.records | ||||||
| 		m.RUnlock() | 		m.RUnlock() | ||||||
|  |  | ||||||
| 		var services []*register.Service | 		var services []*register.Service | ||||||
|  |  | ||||||
| 		for namespace := range recs { | 		for domain := range recs { | ||||||
| 			srvs, err := m.ListServices(ctx, append(opts, register.ListNamespace(namespace))...) | 			srvs, err := m.ListServices(ctx, append(opts, register.ListDomain(domain))...) | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| 				return nil, err | 				return nil, err | ||||||
| 			} | 			} | ||||||
| @@ -340,7 +361,7 @@ func (m *memory) ListServices(ctx context.Context, opts ...register.ListOption) | |||||||
| 	defer m.RUnlock() | 	defer m.RUnlock() | ||||||
|  |  | ||||||
| 	// ensure the domain exists | 	// ensure the domain exists | ||||||
| 	services, ok := m.records[options.Namespace] | 	services, ok := m.records[options.Domain] | ||||||
| 	if !ok { | 	if !ok { | ||||||
| 		return make([]*register.Service, 0), nil | 		return make([]*register.Service, 0), nil | ||||||
| 	} | 	} | ||||||
| @@ -350,7 +371,7 @@ func (m *memory) ListServices(ctx context.Context, opts ...register.ListOption) | |||||||
|  |  | ||||||
| 	for _, service := range services { | 	for _, service := range services { | ||||||
| 		for _, version := range service { | 		for _, version := range service { | ||||||
| 			result = append(result, recordToService(version, options.Namespace)) | 			result = append(result, recordToService(version, options.Domain)) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -405,13 +426,16 @@ func (m *watcher) Next() (*register.Result, error) { | |||||||
| 				continue | 				continue | ||||||
| 			} | 			} | ||||||
|  |  | ||||||
| 			namespace := register.DefaultNamespace | 			// extract domain from service metadata | ||||||
| 			if r.Service.Namespace != "" { | 			var domain string | ||||||
| 				namespace = r.Service.Namespace | 			if r.Service.Metadata != nil && len(r.Service.Metadata["domain"]) > 0 { | ||||||
|  | 				domain = r.Service.Metadata["domain"] | ||||||
|  | 			} else { | ||||||
|  | 				domain = register.DefaultDomain | ||||||
| 			} | 			} | ||||||
|  |  | ||||||
| 			// only send the event if watching the wildcard or this specific domain | 			// only send the event if watching the wildcard or this specific domain | ||||||
| 			if m.wo.Namespace == register.WildcardNamespace || m.wo.Namespace == namespace { | 			if m.wo.Domain == register.WildcardDomain || m.wo.Domain == domain { | ||||||
| 				return r, nil | 				return r, nil | ||||||
| 			} | 			} | ||||||
| 		case <-m.exit: | 		case <-m.exit: | ||||||
| @@ -430,6 +454,11 @@ func (m *watcher) Stop() { | |||||||
| } | } | ||||||
|  |  | ||||||
| func serviceToRecord(s *register.Service, ttl time.Duration) *record { | func serviceToRecord(s *register.Service, ttl time.Duration) *record { | ||||||
|  | 	metadata := make(map[string]string, len(s.Metadata)) | ||||||
|  | 	for k, v := range s.Metadata { | ||||||
|  | 		metadata[k] = v | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	nodes := make(map[string]*node, len(s.Nodes)) | 	nodes := make(map[string]*node, len(s.Nodes)) | ||||||
| 	for _, n := range s.Nodes { | 	for _, n := range s.Nodes { | ||||||
| 		nodes[n.ID] = &node{ | 		nodes[n.ID] = &node{ | ||||||
| @@ -439,23 +468,54 @@ func serviceToRecord(s *register.Service, ttl time.Duration) *record { | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	endpoints := make([]*register.Endpoint, len(s.Endpoints)) | ||||||
|  | 	copy(endpoints, s.Endpoints) | ||||||
|  |  | ||||||
| 	return &record{ | 	return &record{ | ||||||
| 		Name:      s.Name, | 		Name:      s.Name, | ||||||
| 		Version:   s.Version, | 		Version:   s.Version, | ||||||
|  | 		Metadata:  metadata, | ||||||
| 		Nodes:     nodes, | 		Nodes:     nodes, | ||||||
|  | 		Endpoints: endpoints, | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func recordToService(r *record, domain string) *register.Service { | ||||||
|  | 	metadata := make(map[string]string, len(r.Metadata)) | ||||||
|  | 	for k, v := range r.Metadata { | ||||||
|  | 		metadata[k] = v | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// set the domain in metadata so it can be determined when a wildcard query is performed | ||||||
|  | 	metadata["domain"] = domain | ||||||
|  |  | ||||||
|  | 	endpoints := make([]*register.Endpoint, len(r.Endpoints)) | ||||||
|  | 	for i, e := range r.Endpoints { | ||||||
|  | 		md := make(map[string]string, len(e.Metadata)) | ||||||
|  | 		for k, v := range e.Metadata { | ||||||
|  | 			md[k] = v | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		endpoints[i] = ®ister.Endpoint{ | ||||||
|  | 			Name:     e.Name, | ||||||
|  | 			Request:  e.Request, | ||||||
|  | 			Response: e.Response, | ||||||
|  | 			Metadata: md, | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| func recordToService(r *record, namespace string) *register.Service { |  | ||||||
| 	nodes := make([]*register.Node, len(r.Nodes)) | 	nodes := make([]*register.Node, len(r.Nodes)) | ||||||
| 	i := 0 | 	i := 0 | ||||||
| 	for _, n := range r.Nodes { | 	for _, n := range r.Nodes { | ||||||
| 		nmd := metadata.Copy(n.Metadata) | 		md := make(map[string]string, len(n.Metadata)) | ||||||
|  | 		for k, v := range n.Metadata { | ||||||
|  | 			md[k] = v | ||||||
|  | 		} | ||||||
|  |  | ||||||
| 		nodes[i] = ®ister.Node{ | 		nodes[i] = ®ister.Node{ | ||||||
| 			ID:       n.ID, | 			ID:       n.ID, | ||||||
| 			Address:  n.Address, | 			Address:  n.Address, | ||||||
| 			Metadata: nmd, | 			Metadata: md, | ||||||
| 		} | 		} | ||||||
| 		i++ | 		i++ | ||||||
| 	} | 	} | ||||||
| @@ -463,7 +523,8 @@ func recordToService(r *record, namespace string) *register.Service { | |||||||
| 	return ®ister.Service{ | 	return ®ister.Service{ | ||||||
| 		Name:      r.Name, | 		Name:      r.Name, | ||||||
| 		Version:   r.Version, | 		Version:   r.Version, | ||||||
|  | 		Metadata:  metadata, | ||||||
|  | 		Endpoints: endpoints, | ||||||
| 		Nodes:     nodes, | 		Nodes:     nodes, | ||||||
| 		Namespace: namespace, |  | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -7,7 +7,7 @@ import ( | |||||||
| 	"testing" | 	"testing" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/register" | 	"go.unistack.org/micro/v3/register" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var testData = map[string][]*register.Service{ | var testData = map[string][]*register.Service{ | ||||||
| @@ -253,32 +253,32 @@ func TestMemoryWildcard(t *testing.T) { | |||||||
|  |  | ||||||
| 	testSrv := ®ister.Service{Name: "foo", Version: "1.0.0"} | 	testSrv := ®ister.Service{Name: "foo", Version: "1.0.0"} | ||||||
|  |  | ||||||
| 	if err := m.Register(ctx, testSrv, register.RegisterNamespace("one")); err != nil { | 	if err := m.Register(ctx, testSrv, register.RegisterDomain("one")); err != nil { | ||||||
| 		t.Fatalf("Register err: %v", err) | 		t.Fatalf("Register err: %v", err) | ||||||
| 	} | 	} | ||||||
| 	if err := m.Register(ctx, testSrv, register.RegisterNamespace("two")); err != nil { | 	if err := m.Register(ctx, testSrv, register.RegisterDomain("two")); err != nil { | ||||||
| 		t.Fatalf("Register err: %v", err) | 		t.Fatalf("Register err: %v", err) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if recs, err := m.ListServices(ctx, register.ListNamespace("one")); err != nil { | 	if recs, err := m.ListServices(ctx, register.ListDomain("one")); err != nil { | ||||||
| 		t.Errorf("List err: %v", err) | 		t.Errorf("List err: %v", err) | ||||||
| 	} else if len(recs) != 1 { | 	} else if len(recs) != 1 { | ||||||
| 		t.Errorf("Expected 1 record, got %v", len(recs)) | 		t.Errorf("Expected 1 record, got %v", len(recs)) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if recs, err := m.ListServices(ctx, register.ListNamespace("*")); err != nil { | 	if recs, err := m.ListServices(ctx, register.ListDomain("*")); err != nil { | ||||||
| 		t.Errorf("List err: %v", err) | 		t.Errorf("List err: %v", err) | ||||||
| 	} else if len(recs) != 2 { | 	} else if len(recs) != 2 { | ||||||
| 		t.Errorf("Expected 2 records, got %v", len(recs)) | 		t.Errorf("Expected 2 records, got %v", len(recs)) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if recs, err := m.LookupService(ctx, testSrv.Name, register.LookupNamespace("one")); err != nil { | 	if recs, err := m.LookupService(ctx, testSrv.Name, register.LookupDomain("one")); err != nil { | ||||||
| 		t.Errorf("Lookup err: %v", err) | 		t.Errorf("Lookup err: %v", err) | ||||||
| 	} else if len(recs) != 1 { | 	} else if len(recs) != 1 { | ||||||
| 		t.Errorf("Expected 1 record, got %v", len(recs)) | 		t.Errorf("Expected 1 record, got %v", len(recs)) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if recs, err := m.LookupService(ctx, testSrv.Name, register.LookupNamespace("*")); err != nil { | 	if recs, err := m.LookupService(ctx, testSrv.Name, register.LookupDomain("*")); err != nil { | ||||||
| 		t.Errorf("Lookup err: %v", err) | 		t.Errorf("Lookup err: %v", err) | ||||||
| 	} else if len(recs) != 2 { | 	} else if len(recs) != 2 { | ||||||
| 		t.Errorf("Expected 2 records, got %v", len(recs)) | 		t.Errorf("Expected 2 records, got %v", len(recs)) | ||||||
|   | |||||||
| @@ -5,10 +5,9 @@ import ( | |||||||
| 	"crypto/tls" | 	"crypto/tls" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/codec" | 	"go.unistack.org/micro/v3/logger" | ||||||
| 	"go.unistack.org/micro/v4/logger" | 	"go.unistack.org/micro/v3/meter" | ||||||
| 	"go.unistack.org/micro/v4/meter" | 	"go.unistack.org/micro/v3/tracer" | ||||||
| 	"go.unistack.org/micro/v4/tracer" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Options holds options for register | // Options holds options for register | ||||||
| @@ -27,8 +26,6 @@ type Options struct { | |||||||
| 	Name string | 	Name string | ||||||
| 	// Addrs specifies register addrs | 	// Addrs specifies register addrs | ||||||
| 	Addrs []string | 	Addrs []string | ||||||
| 	// Codec used to marshal/unmarshal data in register |  | ||||||
| 	Codec codec.Codec |  | ||||||
| 	// Timeout specifies timeout | 	// Timeout specifies timeout | ||||||
| 	Timeout time.Duration | 	Timeout time.Duration | ||||||
| } | } | ||||||
| @@ -40,7 +37,6 @@ func NewOptions(opts ...Option) Options { | |||||||
| 		Meter:   meter.DefaultMeter, | 		Meter:   meter.DefaultMeter, | ||||||
| 		Tracer:  tracer.DefaultTracer, | 		Tracer:  tracer.DefaultTracer, | ||||||
| 		Context: context.Background(), | 		Context: context.Background(), | ||||||
| 		Codec:   codec.NewCodec(), |  | ||||||
| 	} | 	} | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&options) | 		o(&options) | ||||||
| @@ -51,7 +47,7 @@ func NewOptions(opts ...Option) Options { | |||||||
| // RegisterOptions holds options for register method | // RegisterOptions holds options for register method | ||||||
| type RegisterOptions struct { // nolint: golint,revive | type RegisterOptions struct { // nolint: golint,revive | ||||||
| 	Context  context.Context | 	Context  context.Context | ||||||
| 	Namespace string | 	Domain   string | ||||||
| 	TTL      time.Duration | 	TTL      time.Duration | ||||||
| 	Attempts int | 	Attempts int | ||||||
| } | } | ||||||
| @@ -59,7 +55,7 @@ type RegisterOptions struct { // nolint: golint,revive | |||||||
| // NewRegisterOptions returns register options struct filled by opts | // NewRegisterOptions returns register options struct filled by opts | ||||||
| func NewRegisterOptions(opts ...RegisterOption) RegisterOptions { | func NewRegisterOptions(opts ...RegisterOption) RegisterOptions { | ||||||
| 	options := RegisterOptions{ | 	options := RegisterOptions{ | ||||||
| 		Namespace: DefaultNamespace, | 		Domain:  DefaultDomain, | ||||||
| 		Context: context.Background(), | 		Context: context.Background(), | ||||||
| 	} | 	} | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| @@ -76,14 +72,14 @@ type WatchOptions struct { | |||||||
| 	// Other options for implementations of the interface | 	// Other options for implementations of the interface | ||||||
| 	// can be stored in a context | 	// can be stored in a context | ||||||
| 	Context context.Context | 	Context context.Context | ||||||
| 	// Namespace to watch | 	// Domain to watch | ||||||
| 	Namespace string | 	Domain string | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewWatchOptions returns watch options filled by opts | // NewWatchOptions returns watch options filled by opts | ||||||
| func NewWatchOptions(opts ...WatchOption) WatchOptions { | func NewWatchOptions(opts ...WatchOption) WatchOptions { | ||||||
| 	options := WatchOptions{ | 	options := WatchOptions{ | ||||||
| 		Namespace: DefaultNamespace, | 		Domain:  DefaultDomain, | ||||||
| 		Context: context.Background(), | 		Context: context.Background(), | ||||||
| 	} | 	} | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| @@ -95,8 +91,8 @@ func NewWatchOptions(opts ...WatchOption) WatchOptions { | |||||||
| // DeregisterOptions holds options for deregister method | // DeregisterOptions holds options for deregister method | ||||||
| type DeregisterOptions struct { | type DeregisterOptions struct { | ||||||
| 	Context context.Context | 	Context context.Context | ||||||
| 	// Namespace the service was registered in | 	// Domain the service was registered in | ||||||
| 	Namespace string | 	Domain string | ||||||
| 	// Atempts specify max attempts for deregister | 	// Atempts specify max attempts for deregister | ||||||
| 	Attempts int | 	Attempts int | ||||||
| } | } | ||||||
| @@ -104,7 +100,7 @@ type DeregisterOptions struct { | |||||||
| // NewDeregisterOptions returns options for deregister filled by opts | // NewDeregisterOptions returns options for deregister filled by opts | ||||||
| func NewDeregisterOptions(opts ...DeregisterOption) DeregisterOptions { | func NewDeregisterOptions(opts ...DeregisterOption) DeregisterOptions { | ||||||
| 	options := DeregisterOptions{ | 	options := DeregisterOptions{ | ||||||
| 		Namespace: DefaultNamespace, | 		Domain:  DefaultDomain, | ||||||
| 		Context: context.Background(), | 		Context: context.Background(), | ||||||
| 	} | 	} | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| @@ -116,14 +112,14 @@ func NewDeregisterOptions(opts ...DeregisterOption) DeregisterOptions { | |||||||
| // LookupOptions holds lookup options | // LookupOptions holds lookup options | ||||||
| type LookupOptions struct { | type LookupOptions struct { | ||||||
| 	Context context.Context | 	Context context.Context | ||||||
| 	// Namespace to scope the request to | 	// Domain to scope the request to | ||||||
| 	Namespace string | 	Domain string | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewLookupOptions returns lookup options filled by opts | // NewLookupOptions returns lookup options filled by opts | ||||||
| func NewLookupOptions(opts ...LookupOption) LookupOptions { | func NewLookupOptions(opts ...LookupOption) LookupOptions { | ||||||
| 	options := LookupOptions{ | 	options := LookupOptions{ | ||||||
| 		Namespace: DefaultNamespace, | 		Domain:  DefaultDomain, | ||||||
| 		Context: context.Background(), | 		Context: context.Background(), | ||||||
| 	} | 	} | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| @@ -134,16 +130,15 @@ func NewLookupOptions(opts ...LookupOption) LookupOptions { | |||||||
|  |  | ||||||
| // ListOptions holds the list options for list method | // ListOptions holds the list options for list method | ||||||
| type ListOptions struct { | type ListOptions struct { | ||||||
| 	// Context used to store additional options |  | ||||||
| 	Context context.Context | 	Context context.Context | ||||||
| 	// Namespace to scope the request to | 	// Domain to scope the request to | ||||||
| 	Namespace string | 	Domain string | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewListOptions returns list options filled by opts | // NewListOptions returns list options filled by opts | ||||||
| func NewListOptions(opts ...ListOption) ListOptions { | func NewListOptions(opts ...ListOption) ListOptions { | ||||||
| 	options := ListOptions{ | 	options := ListOptions{ | ||||||
| 		Namespace: DefaultNamespace, | 		Domain:  DefaultDomain, | ||||||
| 		Context: context.Background(), | 		Context: context.Background(), | ||||||
| 	} | 	} | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| @@ -222,10 +217,10 @@ func RegisterContext(ctx context.Context) RegisterOption { // nolint: golint,rev | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // RegisterNamespace secifies register Namespace | // RegisterDomain secifies register domain | ||||||
| func RegisterNamespace(d string) RegisterOption { // nolint: golint,revive | func RegisterDomain(d string) RegisterOption { // nolint: golint,revive | ||||||
| 	return func(o *RegisterOptions) { | 	return func(o *RegisterOptions) { | ||||||
| 		o.Namespace = d | 		o.Domain = d | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -243,10 +238,10 @@ func WatchContext(ctx context.Context) WatchOption { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // WatchNamespace sets the Namespace for watch | // WatchDomain sets the domain for watch | ||||||
| func WatchNamespace(d string) WatchOption { | func WatchDomain(d string) WatchOption { | ||||||
| 	return func(o *WatchOptions) { | 	return func(o *WatchOptions) { | ||||||
| 		o.Namespace = d | 		o.Domain = d | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -264,10 +259,10 @@ func DeregisterContext(ctx context.Context) DeregisterOption { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // DeregisterNamespace specifies deregister Namespace | // DeregisterDomain specifies deregister domain | ||||||
| func DeregisterNamespace(d string) DeregisterOption { | func DeregisterDomain(d string) DeregisterOption { | ||||||
| 	return func(o *DeregisterOptions) { | 	return func(o *DeregisterOptions) { | ||||||
| 		o.Namespace = d | 		o.Domain = d | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -278,10 +273,10 @@ func LookupContext(ctx context.Context) LookupOption { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // LookupNamespace sets the Namespace for lookup | // LookupDomain sets the domain for lookup | ||||||
| func LookupNamespace(d string) LookupOption { | func LookupDomain(d string) LookupOption { | ||||||
| 	return func(o *LookupOptions) { | 	return func(o *LookupOptions) { | ||||||
| 		o.Namespace = d | 		o.Domain = d | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -292,10 +287,10 @@ func ListContext(ctx context.Context) ListOption { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // ListNamespace sets the Namespace for list method | // ListDomain sets the domain for list method | ||||||
| func ListNamespace(d string) ListOption { | func ListDomain(d string) ListOption { | ||||||
| 	return func(o *ListOptions) { | 	return func(o *ListOptions) { | ||||||
| 		o.Namespace = d | 		o.Domain = d | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -305,9 +300,3 @@ func Name(n string) Option { | |||||||
| 		o.Name = n | 		o.Name = n | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func Codec(c codec.Codec) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.Codec = c |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|   | |||||||
| @@ -5,16 +5,16 @@ import ( | |||||||
| 	"context" | 	"context" | ||||||
| 	"errors" | 	"errors" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/metadata" | 	"go.unistack.org/micro/v3/metadata" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| const ( | const ( | ||||||
| 	// WildcardNamespace indicates any Namespace | 	// WildcardDomain indicates any domain | ||||||
| 	WildcardNamespace = "*" | 	WildcardDomain = "*" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // DefaultNamespace to use if none was provided in options | // DefaultDomain to use if none was provided in options | ||||||
| var DefaultNamespace = "micro" | var DefaultDomain = "micro" | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| 	// DefaultRegister is the global default register | 	// DefaultRegister is the global default register | ||||||
| @@ -59,17 +59,26 @@ type Register interface { | |||||||
|  |  | ||||||
| // Service holds service register info | // Service holds service register info | ||||||
| type Service struct { | type Service struct { | ||||||
| 	Name      string  `json:"name,omitempty"` | 	Name      string            `json:"name"` | ||||||
| 	Version   string  `json:"version,omitempty"` | 	Version   string            `json:"version"` | ||||||
| 	Nodes     []*Node `json:"nodes,omitempty"` | 	Metadata  metadata.Metadata `json:"metadata"` | ||||||
| 	Namespace string  `json:"namespace,omitempty"` | 	Endpoints []*Endpoint       `json:"endpoints"` | ||||||
|  | 	Nodes     []*Node           `json:"nodes"` | ||||||
| } | } | ||||||
|  |  | ||||||
| // Node holds node register info | // Node holds node register info | ||||||
| type Node struct { | type Node struct { | ||||||
| 	Metadata metadata.Metadata `json:"metadata,omitempty"` | 	Metadata metadata.Metadata `json:"metadata"` | ||||||
| 	ID       string            `json:"id,omitempty"` | 	ID       string            `json:"id"` | ||||||
| 	Address  string            `json:"address,omitempty"` | 	Address  string            `json:"address"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Endpoint holds endpoint register info | ||||||
|  | type Endpoint struct { | ||||||
|  | 	Request  string            `json:"request"` | ||||||
|  | 	Response string            `json:"response"` | ||||||
|  | 	Metadata metadata.Metadata `json:"metadata"` | ||||||
|  | 	Name     string            `json:"name"` | ||||||
| } | } | ||||||
|  |  | ||||||
| // Option func signature | // Option func signature | ||||||
|   | |||||||
| @@ -15,31 +15,31 @@ type Watcher interface { | |||||||
| // the watcher. Actions can be create, update, delete | // the watcher. Actions can be create, update, delete | ||||||
| type Result struct { | type Result struct { | ||||||
| 	// Service holds register service | 	// Service holds register service | ||||||
| 	Service *Service `json:"service,omitempty"` | 	Service *Service | ||||||
| 	// Action holds the action | 	// Action holds the action | ||||||
| 	Action EventType `json:"action,omitempty"` | 	Action string | ||||||
| } | } | ||||||
|  |  | ||||||
| // EventType defines register event type | // EventType defines register event type | ||||||
| type EventType int | type EventType int | ||||||
|  |  | ||||||
| const ( | const ( | ||||||
| 	// EventCreate is emitted when a new service is registered | 	// Create is emitted when a new service is registered | ||||||
| 	EventCreate EventType = iota | 	Create EventType = iota | ||||||
| 	// EventDelete is emitted when an existing service is deregistered | 	// Delete is emitted when an existing service is deregistered | ||||||
| 	EventDelete | 	Delete | ||||||
| 	// EventUpdate is emitted when an existing service is updated | 	// Update is emitted when an existing service is updated | ||||||
| 	EventUpdate | 	Update | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // String returns human readable event type | // String returns human readable event type | ||||||
| func (t EventType) String() string { | func (t EventType) String() string { | ||||||
| 	switch t { | 	switch t { | ||||||
| 	case EventCreate: | 	case Create: | ||||||
| 		return "create" | 		return "create" | ||||||
| 	case EventDelete: | 	case Delete: | ||||||
| 		return "delete" | 		return "delete" | ||||||
| 	case EventUpdate: | 	case Update: | ||||||
| 		return "update" | 		return "update" | ||||||
| 	default: | 	default: | ||||||
| 		return "unknown" | 		return "unknown" | ||||||
| @@ -49,11 +49,11 @@ func (t EventType) String() string { | |||||||
| // Event is register event | // Event is register event | ||||||
| type Event struct { | type Event struct { | ||||||
| 	// Timestamp is event timestamp | 	// Timestamp is event timestamp | ||||||
| 	Timestamp time.Time `json:"timestamp,omitempty"` | 	Timestamp time.Time | ||||||
| 	// Service is register service | 	// Service is register service | ||||||
| 	Service *Service `json:"service,omitempty"` | 	Service *Service | ||||||
| 	// ID is register id | 	// ID is register id | ||||||
| 	ID string `json:"id,omitempty"` | 	ID string | ||||||
| 	// Type defines type of event | 	// Type defines type of event | ||||||
| 	Type EventType `json:"type,omitempty"` | 	Type EventType | ||||||
| } | } | ||||||
|   | |||||||
| @@ -7,7 +7,7 @@ import ( | |||||||
| 	"sync" | 	"sync" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/resolver" | 	"go.unistack.org/micro/v3/resolver" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Resolver is a DNS network resolve | // Resolver is a DNS network resolve | ||||||
|   | |||||||
| @@ -5,7 +5,7 @@ import ( | |||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"net" | 	"net" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/resolver" | 	"go.unistack.org/micro/v3/resolver" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Resolver is a DNS network resolve | // Resolver is a DNS network resolve | ||||||
|   | |||||||
| @@ -8,7 +8,7 @@ import ( | |||||||
| 	"net/http" | 	"net/http" | ||||||
| 	"net/url" | 	"net/url" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/resolver" | 	"go.unistack.org/micro/v3/resolver" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // nolint: golint,revive | // nolint: golint,revive | ||||||
|   | |||||||
| @@ -2,7 +2,7 @@ | |||||||
| package noop | package noop | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"go.unistack.org/micro/v4/resolver" | 	"go.unistack.org/micro/v3/resolver" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Resolver contains noop resolver | // Resolver contains noop resolver | ||||||
|   | |||||||
| @@ -4,8 +4,8 @@ package register | |||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/register" | 	"go.unistack.org/micro/v3/register" | ||||||
| 	"go.unistack.org/micro/v4/resolver" | 	"go.unistack.org/micro/v3/resolver" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Resolver is a register network resolver | // Resolver is a register network resolver | ||||||
|   | |||||||
| @@ -2,7 +2,7 @@ | |||||||
| package static | package static | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"go.unistack.org/micro/v4/resolver" | 	"go.unistack.org/micro/v3/resolver" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Resolver returns a static list of nodes. In the event the node list | // Resolver returns a static list of nodes. In the event the node list | ||||||
|   | |||||||
| @@ -3,9 +3,9 @@ package router | |||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/logger" | 	"go.unistack.org/micro/v3/logger" | ||||||
| 	"go.unistack.org/micro/v4/register" | 	"go.unistack.org/micro/v3/register" | ||||||
| 	"go.unistack.org/micro/v4/util/id" | 	"go.unistack.org/micro/v3/util/id" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Options are router options | // Options are router options | ||||||
|   | |||||||
| @@ -3,7 +3,7 @@ package router | |||||||
| import ( | import ( | ||||||
| 	"hash/fnv" | 	"hash/fnv" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/metadata" | 	"go.unistack.org/micro/v3/metadata" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
|   | |||||||
| @@ -1,8 +1,8 @@ | |||||||
| package random | package random | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"go.unistack.org/micro/v4/selector" | 	"go.unistack.org/micro/v3/selector" | ||||||
| 	"go.unistack.org/micro/v4/util/rand" | 	"go.unistack.org/micro/v3/util/rand" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type random struct{} | type random struct{} | ||||||
|   | |||||||
| @@ -3,7 +3,7 @@ package random | |||||||
| import ( | import ( | ||||||
| 	"testing" | 	"testing" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/selector" | 	"go.unistack.org/micro/v3/selector" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| func TestRandom(t *testing.T) { | func TestRandom(t *testing.T) { | ||||||
|   | |||||||
| @@ -1,8 +1,8 @@ | |||||||
| package roundrobin | package roundrobin | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"go.unistack.org/micro/v4/selector" | 	"go.unistack.org/micro/v3/selector" | ||||||
| 	"go.unistack.org/micro/v4/util/rand" | 	"go.unistack.org/micro/v3/util/rand" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // NewSelector returns an initialised round robin selector | // NewSelector returns an initialised round robin selector | ||||||
|   | |||||||
| @@ -3,7 +3,7 @@ package roundrobin | |||||||
| import ( | import ( | ||||||
| 	"testing" | 	"testing" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/selector" | 	"go.unistack.org/micro/v3/selector" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| func TestRoundRobin(t *testing.T) { | func TestRoundRobin(t *testing.T) { | ||||||
|   | |||||||
| @@ -1,6 +1,6 @@ | |||||||
| package server | package server | ||||||
|  |  | ||||||
| import "go.unistack.org/micro/v4/errors" | import "go.unistack.org/micro/v3/errors" | ||||||
|  |  | ||||||
| type Error struct { | type Error struct { | ||||||
| 	id string | 	id string | ||||||
|   | |||||||
| @@ -3,7 +3,7 @@ package server | |||||||
| import ( | import ( | ||||||
| 	"testing" | 	"testing" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/errors" | 	"go.unistack.org/micro/v3/errors" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| func TestError(t *testing.T) { | func TestError(t *testing.T) { | ||||||
|   | |||||||
							
								
								
									
										446
									
								
								server/noop.go
									
									
									
									
									
								
							
							
						
						
									
										446
									
								
								server/noop.go
									
									
									
									
									
								
							| @@ -1,17 +1,25 @@ | |||||||
| package server | package server | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"context" | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"reflect" | 	"reflect" | ||||||
|  | 	"runtime/debug" | ||||||
|  | 	"sort" | ||||||
|  | 	"strings" | ||||||
| 	"sync" | 	"sync" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/codec" | 	"go.unistack.org/micro/v3/broker" | ||||||
| 	"go.unistack.org/micro/v4/logger" | 	"go.unistack.org/micro/v3/codec" | ||||||
| 	"go.unistack.org/micro/v4/register" | 	"go.unistack.org/micro/v3/errors" | ||||||
| 	maddr "go.unistack.org/micro/v4/util/addr" | 	"go.unistack.org/micro/v3/logger" | ||||||
| 	mnet "go.unistack.org/micro/v4/util/net" | 	"go.unistack.org/micro/v3/metadata" | ||||||
| 	"go.unistack.org/micro/v4/util/rand" | 	"go.unistack.org/micro/v3/options" | ||||||
|  | 	"go.unistack.org/micro/v3/register" | ||||||
|  | 	maddr "go.unistack.org/micro/v3/util/addr" | ||||||
|  | 	mnet "go.unistack.org/micro/v3/util/net" | ||||||
|  | 	"go.unistack.org/micro/v3/util/rand" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // DefaultCodecs will be used to encode/decode | // DefaultCodecs will be used to encode/decode | ||||||
| @@ -19,21 +27,42 @@ var DefaultCodecs = map[string]codec.Codec{ | |||||||
| 	"application/octet-stream": codec.NewCodec(), | 	"application/octet-stream": codec.NewCodec(), | ||||||
| } | } | ||||||
|  |  | ||||||
|  | const ( | ||||||
|  | 	defaultContentType = "application/json" | ||||||
|  | ) | ||||||
|  |  | ||||||
| type rpcHandler struct { | type rpcHandler struct { | ||||||
| 	opts      HandlerOptions | 	opts      HandlerOptions | ||||||
| 	handler   interface{} | 	handler   interface{} | ||||||
| 	name      string | 	name      string | ||||||
|  | 	endpoints []*register.Endpoint | ||||||
| } | } | ||||||
|  |  | ||||||
| func newRPCHandler(handler interface{}, opts ...HandlerOption) Handler { | func newRPCHandler(handler interface{}, opts ...HandlerOption) Handler { | ||||||
| 	options := NewHandlerOptions(opts...) | 	options := NewHandlerOptions(opts...) | ||||||
|  |  | ||||||
|  | 	typ := reflect.TypeOf(handler) | ||||||
| 	hdlr := reflect.ValueOf(handler) | 	hdlr := reflect.ValueOf(handler) | ||||||
| 	name := reflect.Indirect(hdlr).Type().Name() | 	name := reflect.Indirect(hdlr).Type().Name() | ||||||
|  |  | ||||||
|  | 	var endpoints []*register.Endpoint | ||||||
|  |  | ||||||
|  | 	for m := 0; m < typ.NumMethod(); m++ { | ||||||
|  | 		if e := register.ExtractEndpoint(typ.Method(m)); e != nil { | ||||||
|  | 			e.Name = name + "." + e.Name | ||||||
|  |  | ||||||
|  | 			for k, v := range options.Metadata[e.Name] { | ||||||
|  | 				e.Metadata[k] = v | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			endpoints = append(endpoints, e) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	return &rpcHandler{ | 	return &rpcHandler{ | ||||||
| 		name:      name, | 		name:      name, | ||||||
| 		handler:   handler, | 		handler:   handler, | ||||||
|  | 		endpoints: endpoints, | ||||||
| 		opts:      options, | 		opts:      options, | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| @@ -46,6 +75,10 @@ func (r *rpcHandler) Handler() interface{} { | |||||||
| 	return r.handler | 	return r.handler | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (r *rpcHandler) Endpoints() []*register.Endpoint { | ||||||
|  | 	return r.endpoints | ||||||
|  | } | ||||||
|  |  | ||||||
| func (r *rpcHandler) Options() HandlerOptions { | func (r *rpcHandler) Options() HandlerOptions { | ||||||
| 	return r.opts | 	return r.opts | ||||||
| } | } | ||||||
| @@ -55,6 +88,7 @@ type noopServer struct { | |||||||
| 	wg          *sync.WaitGroup | 	wg          *sync.WaitGroup | ||||||
| 	rsvc        *register.Service | 	rsvc        *register.Service | ||||||
| 	handlers    map[string]Handler | 	handlers    map[string]Handler | ||||||
|  | 	subscribers map[*subscriber][]broker.Subscriber | ||||||
| 	exit        chan chan error | 	exit        chan chan error | ||||||
| 	opts        Options | 	opts        Options | ||||||
| 	sync.RWMutex | 	sync.RWMutex | ||||||
| @@ -68,12 +102,25 @@ func NewServer(opts ...Option) Server { | |||||||
| 	if n.handlers == nil { | 	if n.handlers == nil { | ||||||
| 		n.handlers = make(map[string]Handler) | 		n.handlers = make(map[string]Handler) | ||||||
| 	} | 	} | ||||||
|  | 	if n.subscribers == nil { | ||||||
|  | 		n.subscribers = make(map[*subscriber][]broker.Subscriber) | ||||||
|  | 	} | ||||||
| 	if n.exit == nil { | 	if n.exit == nil { | ||||||
| 		n.exit = make(chan chan error) | 		n.exit = make(chan chan error) | ||||||
| 	} | 	} | ||||||
| 	return n | 	return n | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (n *noopServer) newCodec(contentType string) (codec.Codec, error) { | ||||||
|  | 	if cf, ok := n.opts.Codecs[contentType]; ok { | ||||||
|  | 		return cf, nil | ||||||
|  | 	} | ||||||
|  | 	if cf, ok := DefaultCodecs[contentType]; ok { | ||||||
|  | 		return cf, nil | ||||||
|  | 	} | ||||||
|  | 	return nil, codec.ErrUnknownContentType | ||||||
|  | } | ||||||
|  |  | ||||||
| func (n *noopServer) Live() bool { | func (n *noopServer) Live() bool { | ||||||
| 	return true | 	return true | ||||||
| } | } | ||||||
| @@ -95,10 +142,65 @@ func (n *noopServer) Name() string { | |||||||
| 	return n.opts.Name | 	return n.opts.Name | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (n *noopServer) Subscribe(sb Subscriber) error { | ||||||
|  | 	sub, ok := sb.(*subscriber) | ||||||
|  | 	if !ok { | ||||||
|  | 		return fmt.Errorf("invalid subscriber: expected *subscriber") | ||||||
|  | 	} else if len(sub.handlers) == 0 { | ||||||
|  | 		return fmt.Errorf("invalid subscriber: no handler functions") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if err := ValidateSubscriber(sb); err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	n.Lock() | ||||||
|  | 	if _, ok = n.subscribers[sub]; ok { | ||||||
|  | 		n.Unlock() | ||||||
|  | 		return fmt.Errorf("subscriber %v already exists", sub) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	n.subscribers[sub] = nil | ||||||
|  | 	n.Unlock() | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type rpcMessage struct { | ||||||
|  | 	payload     interface{} | ||||||
|  | 	codec       codec.Codec | ||||||
|  | 	header      metadata.Metadata | ||||||
|  | 	topic       string | ||||||
|  | 	contentType string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (r *rpcMessage) ContentType() string { | ||||||
|  | 	return r.contentType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (r *rpcMessage) Topic() string { | ||||||
|  | 	return r.topic | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (r *rpcMessage) Body() interface{} { | ||||||
|  | 	return r.payload | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (r *rpcMessage) Header() metadata.Metadata { | ||||||
|  | 	return r.header | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (r *rpcMessage) Codec() codec.Codec { | ||||||
|  | 	return r.codec | ||||||
|  | } | ||||||
|  |  | ||||||
| func (n *noopServer) NewHandler(h interface{}, opts ...HandlerOption) Handler { | func (n *noopServer) NewHandler(h interface{}, opts ...HandlerOption) Handler { | ||||||
| 	return newRPCHandler(h, opts...) | 	return newRPCHandler(h, opts...) | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (n *noopServer) NewSubscriber(topic string, sb interface{}, opts ...SubscriberOption) Subscriber { | ||||||
|  | 	return newSubscriber(topic, sb, opts...) | ||||||
|  | } | ||||||
|  |  | ||||||
| func (n *noopServer) Init(opts ...Option) error { | func (n *noopServer) Init(opts ...Option) error { | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&n.opts) | 		o(&n.opts) | ||||||
| @@ -107,6 +209,9 @@ func (n *noopServer) Init(opts ...Option) error { | |||||||
| 	if n.handlers == nil { | 	if n.handlers == nil { | ||||||
| 		n.handlers = make(map[string]Handler, 1) | 		n.handlers = make(map[string]Handler, 1) | ||||||
| 	} | 	} | ||||||
|  | 	if n.subscribers == nil { | ||||||
|  | 		n.subscribers = make(map[*subscriber][]broker.Subscriber, 1) | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	if n.exit == nil { | 	if n.exit == nil { | ||||||
| 		n.exit = make(chan chan error) | 		n.exit = make(chan chan error) | ||||||
| @@ -144,6 +249,35 @@ func (n *noopServer) Register() error { | |||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	n.RLock() | ||||||
|  | 	handlerList := make([]string, 0, len(n.handlers)) | ||||||
|  | 	for n := range n.handlers { | ||||||
|  | 		handlerList = append(handlerList, n) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	sort.Strings(handlerList) | ||||||
|  |  | ||||||
|  | 	subscriberList := make([]*subscriber, 0, len(n.subscribers)) | ||||||
|  | 	for e := range n.subscribers { | ||||||
|  | 		subscriberList = append(subscriberList, e) | ||||||
|  | 	} | ||||||
|  | 	sort.Slice(subscriberList, func(i, j int) bool { | ||||||
|  | 		return subscriberList[i].topic > subscriberList[j].topic | ||||||
|  | 	}) | ||||||
|  |  | ||||||
|  | 	endpoints := make([]*register.Endpoint, 0, len(handlerList)+len(subscriberList)) | ||||||
|  | 	for _, h := range handlerList { | ||||||
|  | 		endpoints = append(endpoints, n.handlers[h].Endpoints()...) | ||||||
|  | 	} | ||||||
|  | 	for _, e := range subscriberList { | ||||||
|  | 		endpoints = append(endpoints, e.Endpoints()...) | ||||||
|  | 	} | ||||||
|  | 	n.RUnlock() | ||||||
|  |  | ||||||
|  | 	service.Nodes[0].Metadata["protocol"] = "noop" | ||||||
|  | 	service.Nodes[0].Metadata["transport"] = service.Nodes[0].Metadata["protocol"] | ||||||
|  | 	service.Endpoints = endpoints | ||||||
|  |  | ||||||
| 	n.RLock() | 	n.RLock() | ||||||
| 	registered := n.registered | 	registered := n.registered | ||||||
| 	n.RUnlock() | 	n.RUnlock() | ||||||
| @@ -205,6 +339,33 @@ func (n *noopServer) Deregister() error { | |||||||
|  |  | ||||||
| 	n.registered = false | 	n.registered = false | ||||||
|  |  | ||||||
|  | 	cx := config.Context | ||||||
|  |  | ||||||
|  | 	wg := sync.WaitGroup{} | ||||||
|  | 	for sb, subs := range n.subscribers { | ||||||
|  | 		for idx := range subs { | ||||||
|  | 			if sb.Options().Context != nil { | ||||||
|  | 				cx = sb.Options().Context | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			ncx := cx | ||||||
|  | 			wg.Add(1) | ||||||
|  | 			go func(s broker.Subscriber) { | ||||||
|  | 				defer wg.Done() | ||||||
|  | 				if config.Logger.V(logger.InfoLevel) { | ||||||
|  | 					config.Logger.Info(n.opts.Context, "unsubscribing from topic: "+s.Topic()) | ||||||
|  | 				} | ||||||
|  | 				if err := s.Unsubscribe(ncx); err != nil { | ||||||
|  | 					if config.Logger.V(logger.ErrorLevel) { | ||||||
|  | 						config.Logger.Error(n.opts.Context, "unsubscribing from topic: "+s.Topic(), err) | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			}(subs[idx]) | ||||||
|  | 		} | ||||||
|  | 		n.subscribers[sb] = nil | ||||||
|  | 	} | ||||||
|  | 	wg.Wait() | ||||||
|  |  | ||||||
| 	n.Unlock() | 	n.Unlock() | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
| @@ -241,6 +402,21 @@ func (n *noopServer) Start() error { | |||||||
| 	} | 	} | ||||||
| 	n.Unlock() | 	n.Unlock() | ||||||
|  |  | ||||||
|  | 	// only connect if we're subscribed | ||||||
|  | 	if len(n.subscribers) > 0 { | ||||||
|  | 		// connect to the broker | ||||||
|  | 		if err := config.Broker.Connect(config.Context); err != nil { | ||||||
|  | 			if config.Logger.V(logger.ErrorLevel) { | ||||||
|  | 				config.Logger.Error(n.opts.Context, fmt.Sprintf("broker [%s] connect error", config.Broker.String()), err) | ||||||
|  | 			} | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if config.Logger.V(logger.InfoLevel) { | ||||||
|  | 			config.Logger.Info(n.opts.Context, fmt.Sprintf("broker [%s] Connected to %s", config.Broker.String(), config.Broker.Address())) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	// use RegisterCheck func before register | 	// use RegisterCheck func before register | ||||||
| 	// nolint: nestif | 	// nolint: nestif | ||||||
| 	if err := config.RegisterCheck(config.Context); err != nil { | 	if err := config.RegisterCheck(config.Context); err != nil { | ||||||
| @@ -256,6 +432,10 @@ func (n *noopServer) Start() error { | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	if err := n.subscribe(); err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	go func() { | 	go func() { | ||||||
| 		t := new(time.Ticker) | 		t := new(time.Ticker) | ||||||
|  |  | ||||||
| @@ -339,6 +519,42 @@ func (n *noopServer) Start() error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (n *noopServer) subscribe() error { | ||||||
|  | 	config := n.Options() | ||||||
|  |  | ||||||
|  | 	subCtx := config.Context | ||||||
|  |  | ||||||
|  | 	for sb := range n.subscribers { | ||||||
|  |  | ||||||
|  | 		if cx := sb.Options().Context; cx != nil { | ||||||
|  | 			subCtx = cx | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		opts := []broker.SubscribeOption{ | ||||||
|  | 			broker.SubscribeContext(subCtx), | ||||||
|  | 			broker.SubscribeAutoAck(sb.Options().AutoAck), | ||||||
|  | 			broker.SubscribeBodyOnly(sb.Options().BodyOnly), | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if queue := sb.Options().Queue; len(queue) > 0 { | ||||||
|  | 			opts = append(opts, broker.SubscribeGroup(queue)) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if config.Logger.V(logger.InfoLevel) { | ||||||
|  | 			config.Logger.Info(n.opts.Context, "subscribing to topic: "+sb.Topic()) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		sub, err := config.Broker.Subscribe(subCtx, sb.Topic(), n.createSubHandler(sb, config), opts...) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		n.subscribers[sb] = []broker.Subscriber{sub} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
| func (n *noopServer) Stop() error { | func (n *noopServer) Stop() error { | ||||||
| 	n.RLock() | 	n.RLock() | ||||||
| 	if !n.started { | 	if !n.started { | ||||||
| @@ -358,3 +574,221 @@ func (n *noopServer) Stop() error { | |||||||
|  |  | ||||||
| 	return err | 	return err | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func newSubscriber(topic string, sub interface{}, opts ...SubscriberOption) Subscriber { | ||||||
|  | 	var endpoints []*register.Endpoint | ||||||
|  | 	var handlers []*handler | ||||||
|  |  | ||||||
|  | 	options := NewSubscriberOptions(opts...) | ||||||
|  |  | ||||||
|  | 	if typ := reflect.TypeOf(sub); typ.Kind() == reflect.Func { | ||||||
|  | 		h := &handler{ | ||||||
|  | 			method: reflect.ValueOf(sub), | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		switch typ.NumIn() { | ||||||
|  | 		case 1: | ||||||
|  | 			h.reqType = typ.In(0) | ||||||
|  | 		case 2: | ||||||
|  | 			h.ctxType = typ.In(0) | ||||||
|  | 			h.reqType = typ.In(1) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		handlers = append(handlers, h) | ||||||
|  | 		ep := ®ister.Endpoint{ | ||||||
|  | 			Name:     "Func", | ||||||
|  | 			Request:  register.ExtractSubValue(typ), | ||||||
|  | 			Metadata: metadata.New(2), | ||||||
|  | 		} | ||||||
|  | 		ep.Metadata.Set("topic", topic) | ||||||
|  | 		ep.Metadata.Set("subscriber", "true") | ||||||
|  | 		endpoints = append(endpoints, ep) | ||||||
|  | 	} else { | ||||||
|  | 		hdlr := reflect.ValueOf(sub) | ||||||
|  | 		name := reflect.Indirect(hdlr).Type().Name() | ||||||
|  |  | ||||||
|  | 		for m := 0; m < typ.NumMethod(); m++ { | ||||||
|  | 			method := typ.Method(m) | ||||||
|  | 			h := &handler{ | ||||||
|  | 				method: method.Func, | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			switch method.Type.NumIn() { | ||||||
|  | 			case 2: | ||||||
|  | 				h.reqType = method.Type.In(1) | ||||||
|  | 			case 3: | ||||||
|  | 				h.ctxType = method.Type.In(1) | ||||||
|  | 				h.reqType = method.Type.In(2) | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			handlers = append(handlers, h) | ||||||
|  | 			ep := ®ister.Endpoint{ | ||||||
|  | 				Name:     name + "." + method.Name, | ||||||
|  | 				Request:  register.ExtractSubValue(method.Type), | ||||||
|  | 				Metadata: metadata.New(2), | ||||||
|  | 			} | ||||||
|  | 			ep.Metadata.Set("topic", topic) | ||||||
|  | 			ep.Metadata.Set("subscriber", "true") | ||||||
|  | 			endpoints = append(endpoints, ep) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return &subscriber{ | ||||||
|  | 		rcvr:       reflect.ValueOf(sub), | ||||||
|  | 		typ:        reflect.TypeOf(sub), | ||||||
|  | 		topic:      topic, | ||||||
|  | 		subscriber: sub, | ||||||
|  | 		handlers:   handlers, | ||||||
|  | 		endpoints:  endpoints, | ||||||
|  | 		opts:       options, | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | //nolint:gocyclo | ||||||
|  | func (n *noopServer) createSubHandler(sb *subscriber, opts Options) broker.Handler { | ||||||
|  | 	return func(p broker.Event) (err error) { | ||||||
|  | 		defer func() { | ||||||
|  | 			if r := recover(); r != nil { | ||||||
|  | 				n.RLock() | ||||||
|  | 				config := n.opts | ||||||
|  | 				n.RUnlock() | ||||||
|  | 				if config.Logger.V(logger.ErrorLevel) { | ||||||
|  | 					config.Logger.Error(n.opts.Context, "panic recovered: ", r) | ||||||
|  | 					config.Logger.Error(n.opts.Context, string(debug.Stack())) | ||||||
|  | 				} | ||||||
|  | 				err = errors.InternalServerError(n.opts.Name+".subscriber", "panic recovered: %v", r) | ||||||
|  | 			} | ||||||
|  | 		}() | ||||||
|  |  | ||||||
|  | 		msg := p.Message() | ||||||
|  | 		// if we don't have headers, create empty map | ||||||
|  | 		if msg.Header == nil { | ||||||
|  | 			msg.Header = metadata.New(2) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		ct := msg.Header["Content-Type"] | ||||||
|  | 		if len(ct) == 0 { | ||||||
|  | 			msg.Header.Set(metadata.HeaderContentType, defaultContentType) | ||||||
|  | 			ct = defaultContentType | ||||||
|  | 		} | ||||||
|  | 		cf, err := n.newCodec(ct) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		hdr := metadata.New(len(msg.Header)) | ||||||
|  | 		for k, v := range msg.Header { | ||||||
|  | 			hdr.Set(k, v) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		ctx := metadata.NewIncomingContext(sb.opts.Context, hdr) | ||||||
|  |  | ||||||
|  | 		results := make(chan error, len(sb.handlers)) | ||||||
|  |  | ||||||
|  | 		for i := 0; i < len(sb.handlers); i++ { | ||||||
|  | 			handler := sb.handlers[i] | ||||||
|  |  | ||||||
|  | 			var isVal bool | ||||||
|  | 			var req reflect.Value | ||||||
|  |  | ||||||
|  | 			if handler.reqType.Kind() == reflect.Ptr { | ||||||
|  | 				req = reflect.New(handler.reqType.Elem()) | ||||||
|  | 			} else { | ||||||
|  | 				req = reflect.New(handler.reqType) | ||||||
|  | 				isVal = true | ||||||
|  | 			} | ||||||
|  | 			if isVal { | ||||||
|  | 				req = req.Elem() | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			if err = cf.Unmarshal(msg.Body, req.Interface()); err != nil { | ||||||
|  | 				return err | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			fn := func(ctx context.Context, msg Message) error { | ||||||
|  | 				var vals []reflect.Value | ||||||
|  | 				if sb.typ.Kind() != reflect.Func { | ||||||
|  | 					vals = append(vals, sb.rcvr) | ||||||
|  | 				} | ||||||
|  | 				if handler.ctxType != nil { | ||||||
|  | 					vals = append(vals, reflect.ValueOf(ctx)) | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 				vals = append(vals, reflect.ValueOf(msg.Body())) | ||||||
|  |  | ||||||
|  | 				returnValues := handler.method.Call(vals) | ||||||
|  | 				if rerr := returnValues[0].Interface(); rerr != nil { | ||||||
|  | 					return rerr.(error) | ||||||
|  | 				} | ||||||
|  | 				return nil | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			opts.Hooks.EachPrev(func(hook options.Hook) { | ||||||
|  | 				if h, ok := hook.(HookSubHandler); ok { | ||||||
|  | 					fn = h(fn) | ||||||
|  | 				} | ||||||
|  | 			}) | ||||||
|  |  | ||||||
|  | 			if n.wg != nil { | ||||||
|  | 				n.wg.Add(1) | ||||||
|  | 			} | ||||||
|  | 			go func() { | ||||||
|  | 				if n.wg != nil { | ||||||
|  | 					defer n.wg.Done() | ||||||
|  | 				} | ||||||
|  | 				cerr := fn(ctx, &rpcMessage{ | ||||||
|  | 					topic:       sb.topic, | ||||||
|  | 					contentType: ct, | ||||||
|  | 					payload:     req.Interface(), | ||||||
|  | 					header:      msg.Header, | ||||||
|  | 				}) | ||||||
|  | 				results <- cerr | ||||||
|  | 			}() | ||||||
|  | 		} | ||||||
|  | 		var errors []string | ||||||
|  | 		for i := 0; i < len(sb.handlers); i++ { | ||||||
|  | 			if rerr := <-results; rerr != nil { | ||||||
|  | 				errors = append(errors, rerr.Error()) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		if len(errors) > 0 { | ||||||
|  | 			err = fmt.Errorf("subscriber error: %s", strings.Join(errors, "\n")) | ||||||
|  | 		} | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *subscriber) Topic() string { | ||||||
|  | 	return s.topic | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *subscriber) Subscriber() interface{} { | ||||||
|  | 	return s.subscriber | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *subscriber) Endpoints() []*register.Endpoint { | ||||||
|  | 	return s.endpoints | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *subscriber) Options() SubscriberOptions { | ||||||
|  | 	return s.opts | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type subscriber struct { | ||||||
|  | 	topic string | ||||||
|  |  | ||||||
|  | 	typ        reflect.Type | ||||||
|  | 	subscriber interface{} | ||||||
|  |  | ||||||
|  | 	endpoints []*register.Endpoint | ||||||
|  | 	handlers  []*handler | ||||||
|  |  | ||||||
|  | 	rcvr reflect.Value | ||||||
|  | 	opts SubscriberOptions | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type handler struct { | ||||||
|  | 	reqType reflect.Type | ||||||
|  | 	ctxType reflect.Type | ||||||
|  | 	method  reflect.Value | ||||||
|  | } | ||||||
|   | |||||||
							
								
								
									
										124
									
								
								server/noop_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										124
									
								
								server/noop_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,124 @@ | |||||||
|  | package server_test | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"fmt" | ||||||
|  | 	"testing" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v3/broker" | ||||||
|  | 	"go.unistack.org/micro/v3/client" | ||||||
|  | 	"go.unistack.org/micro/v3/codec" | ||||||
|  | 	"go.unistack.org/micro/v3/logger" | ||||||
|  | 	"go.unistack.org/micro/v3/options" | ||||||
|  | 	"go.unistack.org/micro/v3/server" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type TestHandler struct { | ||||||
|  | 	t *testing.T | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type TestMessage struct { | ||||||
|  | 	Name string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (h *TestHandler) SingleSubHandler(ctx context.Context, msg *codec.Frame) error { | ||||||
|  | 	// fmt.Printf("msg %s\n", msg.Data) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestNoopSub(t *testing.T) { | ||||||
|  | 	ctx := context.Background() | ||||||
|  |  | ||||||
|  | 	b := broker.NewBroker() | ||||||
|  |  | ||||||
|  | 	if err := b.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if err := b.Connect(ctx); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if err := logger.DefaultLogger.Init(logger.WithLevel(logger.ErrorLevel)); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  | 	s := server.NewServer( | ||||||
|  | 		server.Broker(b), | ||||||
|  | 		server.Codec("application/octet-stream", codec.NewCodec()), | ||||||
|  | 	) | ||||||
|  | 	if err := s.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	c := client.NewClient( | ||||||
|  | 		client.Broker(b), | ||||||
|  | 		client.Codec("application/octet-stream", codec.NewCodec()), | ||||||
|  | 		client.ContentType("application/octet-stream"), | ||||||
|  | 	) | ||||||
|  | 	if err := c.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  | 	h := &TestHandler{t: t} | ||||||
|  |  | ||||||
|  | 	if err := s.Subscribe(s.NewSubscriber("single_topic", h.SingleSubHandler, | ||||||
|  | 		server.SubscriberQueue("queue"), | ||||||
|  | 	)); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if err := s.Start(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	msgs := make([]client.Message, 0, 8) | ||||||
|  | 	for i := 0; i < 8; i++ { | ||||||
|  | 		msgs = append(msgs, c.NewMessage("batch_topic", &codec.Frame{Data: []byte(fmt.Sprintf(`{"name": "test_name %d"}`, i))})) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if err := c.BatchPublish(ctx, msgs); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	defer func() { | ||||||
|  | 		if err := s.Stop(); err != nil { | ||||||
|  | 			t.Fatal(err) | ||||||
|  | 		} | ||||||
|  | 	}() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestHooks_Wrap(t *testing.T) { | ||||||
|  | 	n := 5 | ||||||
|  | 	fn1 := func(next server.FuncSubHandler) server.FuncSubHandler { | ||||||
|  | 		return func(ctx context.Context, msg server.Message) (err error) { | ||||||
|  | 			n *= 2 | ||||||
|  | 			return next(ctx, msg) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	fn2 := func(next server.FuncSubHandler) server.FuncSubHandler { | ||||||
|  | 		return func(ctx context.Context, msg server.Message) (err error) { | ||||||
|  | 			n -= 10 | ||||||
|  | 			return next(ctx, msg) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	hs := &options.Hooks{} | ||||||
|  | 	hs.Append(server.HookSubHandler(fn1), server.HookSubHandler(fn2)) | ||||||
|  |  | ||||||
|  | 	var fn = func(ctx context.Context, msg server.Message) error { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	hs.EachPrev(func(hook options.Hook) { | ||||||
|  | 		if h, ok := hook.(server.HookSubHandler); ok { | ||||||
|  | 			fn = h(fn) | ||||||
|  | 		} | ||||||
|  | 	}) | ||||||
|  |  | ||||||
|  | 	if err := fn(nil, nil); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if n != 0 { | ||||||
|  | 		t.Fatalf("uncorrected hooks call") | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -7,16 +7,16 @@ import ( | |||||||
| 	"sync" | 	"sync" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/broker" | 	"go.unistack.org/micro/v3/broker" | ||||||
| 	"go.unistack.org/micro/v4/codec" | 	"go.unistack.org/micro/v3/codec" | ||||||
| 	"go.unistack.org/micro/v4/logger" | 	"go.unistack.org/micro/v3/logger" | ||||||
| 	"go.unistack.org/micro/v4/metadata" | 	"go.unistack.org/micro/v3/metadata" | ||||||
| 	"go.unistack.org/micro/v4/meter" | 	"go.unistack.org/micro/v3/meter" | ||||||
| 	"go.unistack.org/micro/v4/options" | 	"go.unistack.org/micro/v3/options" | ||||||
| 	"go.unistack.org/micro/v4/register" | 	"go.unistack.org/micro/v3/register" | ||||||
| 	msync "go.unistack.org/micro/v4/sync" | 	msync "go.unistack.org/micro/v3/sync" | ||||||
| 	"go.unistack.org/micro/v4/tracer" | 	"go.unistack.org/micro/v3/tracer" | ||||||
| 	"go.unistack.org/micro/v4/util/id" | 	"go.unistack.org/micro/v3/util/id" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Option func | // Option func | ||||||
|   | |||||||
| @@ -4,10 +4,10 @@ import ( | |||||||
| 	"net" | 	"net" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/metadata" | 	"go.unistack.org/micro/v3/metadata" | ||||||
| 	"go.unistack.org/micro/v4/register" | 	"go.unistack.org/micro/v3/register" | ||||||
| 	"go.unistack.org/micro/v4/util/addr" | 	"go.unistack.org/micro/v3/util/addr" | ||||||
| 	"go.unistack.org/micro/v4/util/backoff" | 	"go.unistack.org/micro/v3/util/backoff" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| @@ -17,7 +17,7 @@ var ( | |||||||
|  |  | ||||||
| 		opts := []register.RegisterOption{ | 		opts := []register.RegisterOption{ | ||||||
| 			register.RegisterTTL(config.RegisterTTL), | 			register.RegisterTTL(config.RegisterTTL), | ||||||
| 			register.RegisterNamespace(config.Namespace), | 			register.RegisterDomain(config.Namespace), | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		for i := 0; i <= config.RegisterAttempts; i++ { | 		for i := 0; i <= config.RegisterAttempts; i++ { | ||||||
| @@ -36,7 +36,7 @@ var ( | |||||||
| 		var err error | 		var err error | ||||||
|  |  | ||||||
| 		opts := []register.DeregisterOption{ | 		opts := []register.DeregisterOption{ | ||||||
| 			register.DeregisterNamespace(config.Namespace), | 			register.DeregisterDomain(config.Namespace), | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		for i := 0; i <= config.DeregisterAttempts; i++ { | 		for i := 0; i <= config.DeregisterAttempts; i++ { | ||||||
| @@ -77,9 +77,14 @@ func NewRegisterService(s Server) (*register.Service, error) { | |||||||
| 	} | 	} | ||||||
| 	node.Metadata = metadata.Copy(opts.Metadata) | 	node.Metadata = metadata.Copy(opts.Metadata) | ||||||
|  |  | ||||||
|  | 	node.Metadata["server"] = s.String() | ||||||
|  | 	node.Metadata["broker"] = opts.Broker.String() | ||||||
|  | 	node.Metadata["register"] = opts.Register.String() | ||||||
|  |  | ||||||
| 	return ®ister.Service{ | 	return ®ister.Service{ | ||||||
| 		Name:     opts.Name, | 		Name:     opts.Name, | ||||||
| 		Version:  opts.Version, | 		Version:  opts.Version, | ||||||
| 		Nodes:    []*register.Node{node}, | 		Nodes:    []*register.Node{node}, | ||||||
|  | 		Metadata: metadata.New(0), | ||||||
| 	}, nil | 	}, nil | ||||||
| } | } | ||||||
|   | |||||||
| @@ -3,11 +3,11 @@ package server | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"errors" |  | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/codec" | 	"go.unistack.org/micro/v3/codec" | ||||||
| 	"go.unistack.org/micro/v4/metadata" | 	"go.unistack.org/micro/v3/metadata" | ||||||
|  | 	"go.unistack.org/micro/v3/register" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // DefaultServer default server | // DefaultServer default server | ||||||
| @@ -52,6 +52,10 @@ type Server interface { | |||||||
| 	Handle(h Handler) error | 	Handle(h Handler) error | ||||||
| 	// Create a new handler | 	// Create a new handler | ||||||
| 	NewHandler(h interface{}, opts ...HandlerOption) Handler | 	NewHandler(h interface{}, opts ...HandlerOption) Handler | ||||||
|  | 	// Create a new subscriber | ||||||
|  | 	NewSubscriber(topic string, h interface{}, opts ...SubscriberOption) Subscriber | ||||||
|  | 	// Register a subscriber | ||||||
|  | 	Subscribe(s Subscriber) error | ||||||
| 	// Start the server | 	// Start the server | ||||||
| 	Start() error | 	Start() error | ||||||
| 	// Stop the server | 	// Stop the server | ||||||
| @@ -67,10 +71,36 @@ type Server interface { | |||||||
| } | } | ||||||
|  |  | ||||||
| type ( | type ( | ||||||
|  | 	FuncSubHandler func(ctx context.Context, ms Message) error | ||||||
|  | 	HookSubHandler func(next FuncSubHandler) FuncSubHandler | ||||||
| 	FuncHandler    func(ctx context.Context, req Request, rsp interface{}) error | 	FuncHandler    func(ctx context.Context, req Request, rsp interface{}) error | ||||||
| 	HookHandler    func(next FuncHandler) FuncHandler | 	HookHandler    func(next FuncHandler) FuncHandler | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | /* | ||||||
|  | // Router handle serving messages | ||||||
|  | type Router interface { | ||||||
|  | 	// ProcessMessage processes a message | ||||||
|  | 	ProcessMessage(ctx context.Context, msg Message) error | ||||||
|  | 	// ServeRequest processes a request to completion | ||||||
|  | 	ServeRequest(ctx context.Context, req Request, rsp Response) error | ||||||
|  | } | ||||||
|  | */ | ||||||
|  |  | ||||||
|  | // Message is an async message interface | ||||||
|  | type Message interface { | ||||||
|  | 	// Topic of the message | ||||||
|  | 	Topic() string | ||||||
|  | 	// The decoded payload value | ||||||
|  | 	Body() interface{} | ||||||
|  | 	// The content type of the payload | ||||||
|  | 	ContentType() string | ||||||
|  | 	// The raw headers of the message | ||||||
|  | 	Header() metadata.Metadata | ||||||
|  | 	// Codec used to decode the message | ||||||
|  | 	Codec() codec.Codec | ||||||
|  | } | ||||||
|  |  | ||||||
| // Request is a synchronous request interface | // Request is a synchronous request interface | ||||||
| type Request interface { | type Request interface { | ||||||
| 	// Service name requested | 	// Service name requested | ||||||
| @@ -140,23 +170,16 @@ type Stream interface { | |||||||
| type Handler interface { | type Handler interface { | ||||||
| 	Name() string | 	Name() string | ||||||
| 	Handler() interface{} | 	Handler() interface{} | ||||||
|  | 	Endpoints() []*register.Endpoint | ||||||
| 	Options() HandlerOptions | 	Options() HandlerOptions | ||||||
| } | } | ||||||
|  |  | ||||||
| type serverHeaderKey struct{} | // Subscriber interface represents a subscription to a given topic using | ||||||
|  | // a specific subscriber function or object with endpoints. It mirrors | ||||||
| func ResponseMetadata(ctx context.Context, md *metadata.Metadata) context.Context { | // the handler in its behaviour. | ||||||
| 	return context.WithValue(ctx, serverHeaderKey{}, md) | type Subscriber interface { | ||||||
| } | 	Topic() string | ||||||
|  | 	Subscriber() interface{} | ||||||
| func SetResponseMetadata(ctx context.Context, md metadata.Metadata) error { | 	Endpoints() []*register.Endpoint | ||||||
| 	if md.Len() == 0 { | 	Options() SubscriberOptions | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
| 	h, ok := ctx.Value(serverHeaderKey{}).(*metadata.Metadata) |  | ||||||
| 	if !ok || h == nil { |  | ||||||
| 		return errors.New("missing metadata") |  | ||||||
| 	} |  | ||||||
| 	md.CopyTo(*h) |  | ||||||
| 	return nil |  | ||||||
| } | } | ||||||
|   | |||||||
Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user