Compare commits
	
		
			2 Commits
		
	
	
		
			v3.11.39
			...
			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/v3?tab=overview) |  | ||||||
| [](https://git.unistack.org/unistack-org/micro/actions?query=workflow%3Abuild+branch%3Av3+event%3Apush) |  | ||||||
| [](https://goreportcard.com/report/go.unistack.org/micro/v3) |  | ||||||
|  |  | ||||||
| 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. | ||||||
|   | |||||||
| @@ -588,6 +588,7 @@ func (n *noopClient) publish(ctx context.Context, ps []Message, opts ...PublishO | |||||||
|  |  | ||||||
| 	for _, p := range ps { | 	for _, p := range ps { | ||||||
| 		md := metadata.Copy(omd) | 		md := metadata.Copy(omd) | ||||||
|  | 		md[metadata.HeaderContentType] = p.ContentType() | ||||||
| 		topic := p.Topic() | 		topic := p.Topic() | ||||||
| 		if len(exchange) > 0 { | 		if len(exchange) > 0 { | ||||||
| 			topic = exchange | 			topic = exchange | ||||||
| @@ -599,8 +600,6 @@ func (n *noopClient) publish(ctx context.Context, ps []Message, opts ...PublishO | |||||||
| 			md.Set(k, v) | 			md.Set(k, v) | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		md[metadata.HeaderContentType] = p.ContentType() |  | ||||||
|  |  | ||||||
| 		var body []byte | 		var body []byte | ||||||
|  |  | ||||||
| 		// passed in raw data | 		// passed in raw data | ||||||
|   | |||||||
| @@ -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() {} | ||||||
|  |  | ||||||
|   | |||||||
							
								
								
									
										14
									
								
								go.mod
									
									
									
									
									
								
							
							
						
						
									
										14
									
								
								go.mod
									
									
									
									
									
								
							| @@ -1,12 +1,12 @@ | |||||||
| module go.unistack.org/micro/v3 | 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.6.1 | 	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 | ||||||
| @@ -14,8 +14,8 @@ require ( | |||||||
| 	go.uber.org/automaxprocs v1.6.0 | 	go.uber.org/automaxprocs v1.6.0 | ||||||
| 	go.unistack.org/micro-proto/v3 v3.4.1 | 	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.2 | 	google.golang.org/grpc v1.68.1 | ||||||
| 	google.golang.org/protobuf v1.36.1 | 	google.golang.org/protobuf v1.35.2 | ||||||
| 	gopkg.in/yaml.v3 v3.0.1 | 	gopkg.in/yaml.v3 v3.0.1 | ||||||
| ) | ) | ||||||
|  |  | ||||||
| @@ -36,8 +36,8 @@ require ( | |||||||
| 	github.com/stretchr/testify v1.10.0 // indirect | 	github.com/stretchr/testify v1.10.0 // indirect | ||||||
| 	go.uber.org/goleak v1.3.0 // indirect | 	go.uber.org/goleak v1.3.0 // indirect | ||||||
| 	golang.org/x/exp v0.0.0-20241210194714-1829a127f884 // indirect | 	golang.org/x/exp v0.0.0-20241210194714-1829a127f884 // indirect | ||||||
| 	golang.org/x/net v0.33.0 // indirect | 	golang.org/x/net v0.32.0 // indirect | ||||||
| 	golang.org/x/sys v0.28.0 // indirect | 	golang.org/x/sys v0.28.0 // indirect | ||||||
| 	google.golang.org/genproto/googleapis/rpc v0.0.0-20241216192217-9240e9c98484 // 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 | ||||||
| ) | ) | ||||||
|   | |||||||
							
								
								
									
										25
									
								
								go.sum
									
									
									
									
									
								
							
							
						
						
									
										25
									
								
								go.sum
									
									
									
									
									
								
							| @@ -1,11 +1,11 @@ | |||||||
| 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.6.1 h1:ILa9j1onAAMadBsyyUJv5cack8Y1WT26yLj/V+ulKp8= | github.com/KimMachineGun/automemlimit v0.6.1 h1:ILa9j1onAAMadBsyyUJv5cack8Y1WT26yLj/V+ulKp8= | ||||||
| github.com/KimMachineGun/automemlimit v0.6.1/go.mod h1:T7xYht7B8r6AG/AqFcUdc7fzd2bIdBKmepfP2S1svPY= | 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 h1:+BiEnHL6Z7lXnlGUsXQPPAE7+kenAd4ES8MQ5min0Ok= | ||||||
| github.com/cilium/ebpf v0.16.0/go.mod h1:L7u2Blt2jMM/vLAVgjxluxtBKlz3/GWjB0dMOEngfwE= | 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 h1:2fs7l3P0Qxb1nKWuJNFiwhp2CqiKzho71DQkDrHJIo4= | ||||||
| @@ -35,7 +35,6 @@ github.com/josharian/native v1.1.0 h1:uuaP0hAbW7Y4l0ZRQ6C9zfb7Mg1mbFKry/xzDAfmtL | |||||||
| github.com/josharian/native v1.1.0/go.mod h1:7X/raswPFr05uY3HiLlYeyQntB6OO7E/d2Cu7qoaN2w= | 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 h1:xda7qaHDSVOsADNouv7ukSuicKZO7GgVUCXxpaIEIlM= | ||||||
| github.com/jsimonetti/rtnetlink/v2 v2.0.1/go.mod h1:7MoNYNbb3UaDHtF8udiJo/RH6VsTKP1pqKLUTVCvToE= | github.com/jsimonetti/rtnetlink/v2 v2.0.1/go.mod h1:7MoNYNbb3UaDHtF8udiJo/RH6VsTKP1pqKLUTVCvToE= | ||||||
| github.com/kisielk/sqlstruct v0.0.0-20201105191214-5f3e10d3ab46/go.mod h1:yyMNCyc/Ib3bDTKd379tNMpB/7/H5TjM2Y9QJ5THLbE= |  | ||||||
| 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= | ||||||
| @@ -80,8 +79,8 @@ go.unistack.org/micro-proto/v3 v3.4.1 h1:UTjLSRz2YZuaHk9iSlVqqsA50JQNAEK2ZFboGqt | |||||||
| go.unistack.org/micro-proto/v3 v3.4.1/go.mod h1:okx/cnOhzuCX0ggl/vToatbCupi0O44diiiLLsZ93Zo= | 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 h1:Y/Mj/94zIQQGHVSv1tTtQBDaQaJe62U9bkDZKKyhPCU= | ||||||
| golang.org/x/exp v0.0.0-20241210194714-1829a127f884/go.mod h1:qj5a5QZpwLU2NLQudwIN5koi3beDhSAlJwa67PuM98c= | golang.org/x/exp v0.0.0-20241210194714-1829a127f884/go.mod h1:qj5a5QZpwLU2NLQudwIN5koi3beDhSAlJwa67PuM98c= | ||||||
| golang.org/x/net v0.33.0 h1:74SYHlV8BIgHIFC/LrYkOGIwL19eTYXQ5wc6TBuO36I= | golang.org/x/net v0.32.0 h1:ZqPmj8Kzc+Y6e0+skZsuACbx+wzMgo5MQsJh9Qd6aYI= | ||||||
| golang.org/x/net v0.33.0/go.mod h1:HXLR5J+9DxmrqMwG9qjGCxZ+zKXxBru04zlTvWlWuN4= | 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.0.0-20220715151400-c0bba94af5f8/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||||||
| @@ -89,12 +88,12 @@ 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/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.2 h1:U3S9QEtbXC0bYNvRtcoklF3xGtLViumSYxWykJS+7AU= | google.golang.org/grpc v1.68.1 h1:oI5oTa11+ng8r8XMMN7jAOmWfPZWbYpCFaMUTACxkM0= | ||||||
| google.golang.org/grpc v1.69.2/go.mod h1:vyjdE6jLBI76dgpDojsFGNaHlxdjXN9ghpnd2o7JGZ4= | google.golang.org/grpc v1.68.1/go.mod h1:+q1XYFJjShcqn0QZHvCyeR4CXPA+llXIeUIfIe00waw= | ||||||
| google.golang.org/protobuf v1.36.1 h1:yBPeRvTftaleIgM3PZ/WBIZ7XM/eEYAaEyCwvyjq/gk= | google.golang.org/protobuf v1.35.2 h1:8Ar7bF+apOIoThw1EdZl0p1oWvMqTHmpA2fRTyZO8io= | ||||||
| google.golang.org/protobuf v1.36.1/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= | ||||||
|   | |||||||
| @@ -22,7 +22,6 @@ 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: | ||||||
|   | |||||||
| @@ -9,15 +9,12 @@ import ( | |||||||
| 	"log/slog" | 	"log/slog" | ||||||
| 	"strings" | 	"strings" | ||||||
| 	"testing" | 	"testing" | ||||||
| 	"time" |  | ||||||
|  |  | ||||||
| 	"github.com/google/uuid" | 	"github.com/google/uuid" | ||||||
| 	"go.unistack.org/micro/v3/logger" | 	"go.unistack.org/micro/v3/logger" | ||||||
| 	"go.unistack.org/micro/v3/metadata" | 	"go.unistack.org/micro/v3/metadata" | ||||||
| 	"go.unistack.org/micro/v3/util/buffer" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // always first to have proper check |  | ||||||
| func TestStacktrace(t *testing.T) { | func TestStacktrace(t *testing.T) { | ||||||
| 	ctx := context.TODO() | 	ctx := context.TODO() | ||||||
| 	buf := bytes.NewBuffer(nil) | 	buf := bytes.NewBuffer(nil) | ||||||
| @@ -31,48 +28,7 @@ func TestStacktrace(t *testing.T) { | |||||||
|  |  | ||||||
| 	l.Error(ctx, "msg1", errors.New("err")) | 	l.Error(ctx, "msg1", errors.New("err")) | ||||||
|  |  | ||||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`slog_test.go:32`)) { | 	if !bytes.Contains(buf.Bytes(), []byte(`slog_test.go:29`)) { | ||||||
| 		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() |  | ||||||
| 	buf := bytes.NewBuffer(nil) |  | ||||||
| 	l := NewLogger(logger.WithLevel(logger.ErrorLevel), logger.WithOutput(buf), |  | ||||||
| 		WithHandlerFunc(slog.NewTextHandler), |  | ||||||
| 		logger.WithAddStacktrace(true), |  | ||||||
| 		logger.WithTimeFunc(func() time.Time { |  | ||||||
| 			return time.Unix(0, 0) |  | ||||||
| 		}), |  | ||||||
| 	) |  | ||||||
| 	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(`timestamp=1970-01-01T03:00:00.000000000+03:00`)) && |  | ||||||
| 		!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()) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -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) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -27,6 +27,7 @@ type record struct { | |||||||
| 	Version   string | 	Version   string | ||||||
| 	Metadata  map[string]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 | ||||||
| @@ -167,6 +173,9 @@ func (m *memory) Register(_ context.Context, s *register.Service, opts ...regist | |||||||
| 			metadata[k] = v | 			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{ | ||||||
| @@ -185,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 { | ||||||
| @@ -197,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 | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -207,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 | ||||||
| 	} | 	} | ||||||
| @@ -237,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)) | ||||||
| @@ -255,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)) | ||||||
| 	} | 	} | ||||||
| @@ -268,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 { | ||||||
| @@ -295,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 | ||||||
| 	} | 	} | ||||||
| @@ -312,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++ | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -323,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 | ||||||
| 			} | 			} | ||||||
| @@ -345,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 | ||||||
| 	} | 	} | ||||||
| @@ -355,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)) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -410,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: | ||||||
| @@ -435,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{ | ||||||
| @@ -444,19 +468,42 @@ 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, namespace string) *register.Service { | func recordToService(r *record, domain string) *register.Service { | ||||||
| 	metadata := make(map[string]string, len(r.Metadata)) | 	metadata := make(map[string]string, len(r.Metadata)) | ||||||
| 	for k, v := range r.Metadata { | 	for k, v := range r.Metadata { | ||||||
| 		metadata[k] = v | 		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, | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	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 { | ||||||
| @@ -476,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, |  | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -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,7 +5,6 @@ import ( | |||||||
| 	"crypto/tls" | 	"crypto/tls" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/codec" |  | ||||||
| 	"go.unistack.org/micro/v3/logger" | 	"go.unistack.org/micro/v3/logger" | ||||||
| 	"go.unistack.org/micro/v3/meter" | 	"go.unistack.org/micro/v3/meter" | ||||||
| 	"go.unistack.org/micro/v3/tracer" | 	"go.unistack.org/micro/v3/tracer" | ||||||
| @@ -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 |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|   | |||||||
| @@ -9,12 +9,12 @@ import ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
| 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 | ||||||
| } | } | ||||||
|   | |||||||
| @@ -5,6 +5,7 @@ import ( | |||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"reflect" | 	"reflect" | ||||||
| 	"runtime/debug" | 	"runtime/debug" | ||||||
|  | 	"sort" | ||||||
| 	"strings" | 	"strings" | ||||||
| 	"sync" | 	"sync" | ||||||
| 	"time" | 	"time" | ||||||
| @@ -34,17 +35,34 @@ 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, | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| @@ -57,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 | ||||||
| } | } | ||||||
| @@ -227,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() | ||||||
| @@ -525,6 +576,7 @@ func (n *noopServer) Stop() error { | |||||||
| } | } | ||||||
|  |  | ||||||
| func newSubscriber(topic string, sub interface{}, opts ...SubscriberOption) Subscriber { | func newSubscriber(topic string, sub interface{}, opts ...SubscriberOption) Subscriber { | ||||||
|  | 	var endpoints []*register.Endpoint | ||||||
| 	var handlers []*handler | 	var handlers []*handler | ||||||
|  |  | ||||||
| 	options := NewSubscriberOptions(opts...) | 	options := NewSubscriberOptions(opts...) | ||||||
| @@ -543,7 +595,18 @@ func newSubscriber(topic string, sub interface{}, opts ...SubscriberOption) Subs | |||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		handlers = append(handlers, h) | 		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 { | 	} else { | ||||||
|  | 		hdlr := reflect.ValueOf(sub) | ||||||
|  | 		name := reflect.Indirect(hdlr).Type().Name() | ||||||
|  |  | ||||||
| 		for m := 0; m < typ.NumMethod(); m++ { | 		for m := 0; m < typ.NumMethod(); m++ { | ||||||
| 			method := typ.Method(m) | 			method := typ.Method(m) | ||||||
| 			h := &handler{ | 			h := &handler{ | ||||||
| @@ -559,6 +622,14 @@ func newSubscriber(topic string, sub interface{}, opts ...SubscriberOption) Subs | |||||||
| 			} | 			} | ||||||
|  |  | ||||||
| 			handlers = append(handlers, h) | 			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) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -568,6 +639,7 @@ func newSubscriber(topic string, sub interface{}, opts ...SubscriberOption) Subs | |||||||
| 		topic:      topic, | 		topic:      topic, | ||||||
| 		subscriber: sub, | 		subscriber: sub, | ||||||
| 		handlers:   handlers, | 		handlers:   handlers, | ||||||
|  | 		endpoints:  endpoints, | ||||||
| 		opts:       options, | 		opts:       options, | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| @@ -694,6 +766,10 @@ func (s *subscriber) Subscriber() interface{} { | |||||||
| 	return s.subscriber | 	return s.subscriber | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (s *subscriber) Endpoints() []*register.Endpoint { | ||||||
|  | 	return s.endpoints | ||||||
|  | } | ||||||
|  |  | ||||||
| func (s *subscriber) Options() SubscriberOptions { | func (s *subscriber) Options() SubscriberOptions { | ||||||
| 	return s.opts | 	return s.opts | ||||||
| } | } | ||||||
| @@ -704,6 +780,7 @@ type subscriber struct { | |||||||
| 	typ        reflect.Type | 	typ        reflect.Type | ||||||
| 	subscriber interface{} | 	subscriber interface{} | ||||||
|  |  | ||||||
|  | 	endpoints []*register.Endpoint | ||||||
| 	handlers  []*handler | 	handlers  []*handler | ||||||
|  |  | ||||||
| 	rcvr reflect.Value | 	rcvr reflect.Value | ||||||
|   | |||||||
| @@ -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++ { | ||||||
| @@ -85,5 +85,6 @@ func NewRegisterService(s Server) (*register.Service, error) { | |||||||
| 		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 | ||||||
| } | } | ||||||
|   | |||||||
| @@ -7,6 +7,7 @@ import ( | |||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/codec" | 	"go.unistack.org/micro/v3/codec" | ||||||
| 	"go.unistack.org/micro/v3/metadata" | 	"go.unistack.org/micro/v3/metadata" | ||||||
|  | 	"go.unistack.org/micro/v3/register" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // DefaultServer default server | // DefaultServer default server | ||||||
| @@ -169,6 +170,7 @@ type Stream interface { | |||||||
| type Handler interface { | type Handler interface { | ||||||
| 	Name() string | 	Name() string | ||||||
| 	Handler() interface{} | 	Handler() interface{} | ||||||
|  | 	Endpoints() []*register.Endpoint | ||||||
| 	Options() HandlerOptions | 	Options() HandlerOptions | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -178,5 +180,6 @@ type Handler interface { | |||||||
| type Subscriber interface { | type Subscriber interface { | ||||||
| 	Topic() string | 	Topic() string | ||||||
| 	Subscriber() interface{} | 	Subscriber() interface{} | ||||||
|  | 	Endpoints() []*register.Endpoint | ||||||
| 	Options() SubscriberOptions | 	Options() SubscriberOptions | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										27
									
								
								util/buf/buf.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										27
									
								
								util/buf/buf.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,27 @@ | |||||||
|  | package buf | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"bytes" | ||||||
|  | 	"io" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | var _ io.Closer = &Buffer{} | ||||||
|  |  | ||||||
|  | // Buffer bytes.Buffer wrapper to satisfie io.Closer interface | ||||||
|  | type Buffer struct { | ||||||
|  | 	*bytes.Buffer | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Close reset buffer contents | ||||||
|  | func (b *Buffer) Close() error { | ||||||
|  | 	b.Buffer.Reset() | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // New creates new buffer that satisfies Closer interface | ||||||
|  | func New(b *bytes.Buffer) *Buffer { | ||||||
|  | 	if b == nil { | ||||||
|  | 		b = bytes.NewBuffer(nil) | ||||||
|  | 	} | ||||||
|  | 	return &Buffer{b} | ||||||
|  | } | ||||||
| @@ -1,85 +0,0 @@ | |||||||
| package buffer |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"io" |  | ||||||
| 	"sync" |  | ||||||
| 	"time" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| var _ io.WriteCloser = (*DelayedBuffer)(nil) |  | ||||||
|  |  | ||||||
| // DelayedBuffer is the buffer that holds items until either the buffer filled or a specified time limit is reached |  | ||||||
| type DelayedBuffer struct { |  | ||||||
| 	mu        sync.Mutex |  | ||||||
| 	maxWait   time.Duration |  | ||||||
| 	flushTime time.Time |  | ||||||
| 	buffer    chan []byte |  | ||||||
| 	ticker    *time.Ticker |  | ||||||
| 	w         io.Writer |  | ||||||
| 	err       error |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func NewDelayedBuffer(size int, maxWait time.Duration, w io.Writer) *DelayedBuffer { |  | ||||||
| 	b := &DelayedBuffer{ |  | ||||||
| 		buffer:    make(chan []byte, size), |  | ||||||
| 		ticker:    time.NewTicker(maxWait), |  | ||||||
| 		w:         w, |  | ||||||
| 		flushTime: time.Now(), |  | ||||||
| 		maxWait:   maxWait, |  | ||||||
| 	} |  | ||||||
| 	b.loop() |  | ||||||
| 	return b |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (b *DelayedBuffer) loop() { |  | ||||||
| 	go func() { |  | ||||||
| 		for range b.ticker.C { |  | ||||||
| 			b.mu.Lock() |  | ||||||
| 			if time.Since(b.flushTime) > b.maxWait { |  | ||||||
| 				b.flush() |  | ||||||
| 			} |  | ||||||
| 			b.mu.Unlock() |  | ||||||
| 		} |  | ||||||
| 	}() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (b *DelayedBuffer) flush() { |  | ||||||
| 	bufLen := len(b.buffer) |  | ||||||
| 	if bufLen > 0 { |  | ||||||
| 		tmp := make([][]byte, bufLen) |  | ||||||
| 		for i := 0; i < bufLen; i++ { |  | ||||||
| 			tmp[i] = <-b.buffer |  | ||||||
| 		} |  | ||||||
| 		for _, t := range tmp { |  | ||||||
| 			_, b.err = b.w.Write(t) |  | ||||||
| 		} |  | ||||||
| 		b.flushTime = time.Now() |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (b *DelayedBuffer) Put(items ...[]byte) { |  | ||||||
| 	b.mu.Lock() |  | ||||||
| 	for _, item := range items { |  | ||||||
| 		select { |  | ||||||
| 		case b.buffer <- item: |  | ||||||
| 		default: |  | ||||||
| 			b.flush() |  | ||||||
| 			b.buffer <- item |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	b.mu.Unlock() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (b *DelayedBuffer) Close() error { |  | ||||||
| 	b.mu.Lock() |  | ||||||
| 	b.flush() |  | ||||||
| 	close(b.buffer) |  | ||||||
| 	b.ticker.Stop() |  | ||||||
| 	b.mu.Unlock() |  | ||||||
| 	return b.err |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (b *DelayedBuffer) Write(data []byte) (int, error) { |  | ||||||
| 	b.Put(data) |  | ||||||
| 	return len(data), b.err |  | ||||||
| } |  | ||||||
| @@ -1,22 +0,0 @@ | |||||||
| package buffer |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"bytes" |  | ||||||
| 	"testing" |  | ||||||
| 	"time" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| func TestTimedBuffer(t *testing.T) { |  | ||||||
| 	buf := bytes.NewBuffer(nil) |  | ||||||
| 	b := NewDelayedBuffer(100, 300*time.Millisecond, buf) |  | ||||||
| 	for i := 0; i < 100; i++ { |  | ||||||
| 		_, _ = b.Write([]byte(`test`)) |  | ||||||
| 	} |  | ||||||
| 	if buf.Len() != 0 { |  | ||||||
| 		t.Fatal("delayed write not worked") |  | ||||||
| 	} |  | ||||||
| 	time.Sleep(400 * time.Millisecond) |  | ||||||
| 	if buf.Len() == 0 { |  | ||||||
| 		t.Fatal("delayed write not worked") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -1,78 +0,0 @@ | |||||||
| package buffer |  | ||||||
|  |  | ||||||
| import "io" |  | ||||||
|  |  | ||||||
| var _ interface { |  | ||||||
| 	io.ReadCloser |  | ||||||
| 	io.ReadSeeker |  | ||||||
| } = (*SeekerBuffer)(nil) |  | ||||||
|  |  | ||||||
| // Buffer is a ReadWriteCloser that supports seeking. It's intended to |  | ||||||
| // replicate the functionality of bytes.Buffer that I use in my projects. |  | ||||||
| // |  | ||||||
| // Note that the seeking is limited to the read marker; all writes are |  | ||||||
| // append-only. |  | ||||||
| type SeekerBuffer struct { |  | ||||||
| 	data []byte |  | ||||||
| 	pos  int64 |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func NewSeekerBuffer(data []byte) *SeekerBuffer { |  | ||||||
| 	return &SeekerBuffer{ |  | ||||||
| 		data: data, |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (b *SeekerBuffer) Read(p []byte) (int, error) { |  | ||||||
| 	if b.pos >= int64(len(b.data)) { |  | ||||||
| 		return 0, io.EOF |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	n := copy(p, b.data[b.pos:]) |  | ||||||
| 	b.pos += int64(n) |  | ||||||
| 	return n, nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (b *SeekerBuffer) Write(p []byte) (int, error) { |  | ||||||
| 	b.data = append(b.data, p...) |  | ||||||
| 	return len(p), nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Seek sets the read pointer to pos. |  | ||||||
| func (b *SeekerBuffer) Seek(offset int64, whence int) (int64, error) { |  | ||||||
| 	switch whence { |  | ||||||
| 	case io.SeekStart: |  | ||||||
| 		b.pos = offset |  | ||||||
| 	case io.SeekEnd: |  | ||||||
| 		b.pos = int64(len(b.data)) + offset |  | ||||||
| 	case io.SeekCurrent: |  | ||||||
| 		b.pos += offset |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return b.pos, nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Rewind resets the read pointer to 0. |  | ||||||
| func (b *SeekerBuffer) Rewind() error { |  | ||||||
| 	if _, err := b.Seek(0, io.SeekStart); err != nil { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Close clears all the data out of the buffer and sets the read position to 0. |  | ||||||
| func (b *SeekerBuffer) Close() error { |  | ||||||
| 	b.data = nil |  | ||||||
| 	b.pos = 0 |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Len returns the length of data remaining to be read. |  | ||||||
| func (b *SeekerBuffer) Len() int { |  | ||||||
| 	return len(b.data[b.pos:]) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Bytes returns the underlying bytes from the current position. |  | ||||||
| func (b *SeekerBuffer) Bytes() []byte { |  | ||||||
| 	return b.data[b.pos:] |  | ||||||
| } |  | ||||||
| @@ -1,55 +0,0 @@ | |||||||
| package buffer |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"fmt" |  | ||||||
| 	"strings" |  | ||||||
| 	"testing" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| func noErrorT(t *testing.T, err error) { |  | ||||||
| 	if nil != err { |  | ||||||
| 		t.Fatalf("%s", err) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func boolT(t *testing.T, cond bool, s ...string) { |  | ||||||
| 	if !cond { |  | ||||||
| 		what := strings.Join(s, ", ") |  | ||||||
| 		if len(what) > 0 { |  | ||||||
| 			what = ": " + what |  | ||||||
| 		} |  | ||||||
| 		t.Fatalf("assert.Bool failed%s", what) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestSeeking(t *testing.T) { |  | ||||||
| 	partA := []byte("hello, ") |  | ||||||
| 	partB := []byte("world!") |  | ||||||
|  |  | ||||||
| 	buf := NewSeekerBuffer(partA) |  | ||||||
|  |  | ||||||
| 	boolT(t, buf.Len() == len(partA), fmt.Sprintf("on init: have length %d, want length %d", buf.Len(), len(partA))) |  | ||||||
|  |  | ||||||
| 	b := make([]byte, 32) |  | ||||||
|  |  | ||||||
| 	n, err := buf.Read(b) |  | ||||||
| 	noErrorT(t, err) |  | ||||||
| 	boolT(t, buf.Len() == 0, fmt.Sprintf("after reading 1: have length %d, want length 0", buf.Len())) |  | ||||||
| 	boolT(t, n == len(partA), fmt.Sprintf("after reading 2: have length %d, want length %d", n, len(partA))) |  | ||||||
|  |  | ||||||
| 	n, err = buf.Write(partB) |  | ||||||
| 	noErrorT(t, err) |  | ||||||
| 	boolT(t, n == len(partB), fmt.Sprintf("after writing: have length %d, want length %d", n, len(partB))) |  | ||||||
|  |  | ||||||
| 	n, err = buf.Read(b) |  | ||||||
| 	noErrorT(t, err) |  | ||||||
| 	boolT(t, buf.Len() == 0, fmt.Sprintf("after rereading 1: have length %d, want length 0", buf.Len())) |  | ||||||
| 	boolT(t, n == len(partB), fmt.Sprintf("after rereading 2: have length %d, want length %d", n, len(partB))) |  | ||||||
|  |  | ||||||
| 	partsLen := len(partA) + len(partB) |  | ||||||
| 	_ = buf.Rewind() |  | ||||||
| 	boolT(t, buf.Len() == partsLen, fmt.Sprintf("after rewinding: have length %d, want length %d", buf.Len(), partsLen)) |  | ||||||
|  |  | ||||||
| 	buf.Close() |  | ||||||
| 	boolT(t, buf.Len() == 0, fmt.Sprintf("after closing, have length %d, want length 0", buf.Len())) |  | ||||||
| } |  | ||||||
| @@ -8,7 +8,6 @@ import ( | |||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	uuidv8 "github.com/ash3in/uuidv8" | 	uuidv8 "github.com/ash3in/uuidv8" | ||||||
| 	"github.com/google/uuid" |  | ||||||
| 	nanoid "github.com/matoous/go-nanoid" | 	nanoid "github.com/matoous/go-nanoid" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| @@ -97,10 +96,6 @@ func New(opts ...Option) (string, error) { | |||||||
| 	return "", errors.New("invalid option, Type unspecified") | 	return "", errors.New("invalid option, Type unspecified") | ||||||
| } | } | ||||||
|  |  | ||||||
| func ToUUID(s string) uuid.UUID { |  | ||||||
| 	return uuid.MustParse(s) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Must is the same as New but fatals on error | // Must is the same as New but fatals on error | ||||||
| func MustNew(opts ...Option) string { | func MustNew(opts ...Option) string { | ||||||
| 	id, err := New(opts...) | 	id, err := New(opts...) | ||||||
|   | |||||||
| @@ -7,14 +7,5 @@ func TestUUIDv8(t *testing.T) { | |||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		t.Fatal(err) | 		t.Fatal(err) | ||||||
| 	} | 	} | ||||||
| 	_ = id | 	t.Logf("xxx %s\n", id) | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestToUUID(t *testing.T) { |  | ||||||
| 	id, err := New() |  | ||||||
| 	if err != nil { |  | ||||||
| 		t.Fatal(err) |  | ||||||
| 	} |  | ||||||
| 	u := ToUUID(id) |  | ||||||
| 	_ = u |  | ||||||
| } | } | ||||||
|   | |||||||
| @@ -71,6 +71,14 @@ func CopyService(service *register.Service) *register.Service { | |||||||
| 	} | 	} | ||||||
| 	s.Nodes = nodes | 	s.Nodes = nodes | ||||||
|  |  | ||||||
|  | 	// copy endpoints | ||||||
|  | 	eps := make([]*register.Endpoint, len(service.Endpoints)) | ||||||
|  | 	for j, ep := range service.Endpoints { | ||||||
|  | 		e := ®ister.Endpoint{} | ||||||
|  | 		*e = *ep | ||||||
|  | 		eps[j] = e | ||||||
|  | 	} | ||||||
|  | 	s.Endpoints = eps | ||||||
| 	return s | 	return s | ||||||
| } | } | ||||||
|  |  | ||||||
|   | |||||||
| @@ -35,11 +35,11 @@ func TestUnmarshalYAML(t *testing.T) { | |||||||
| 		t.Fatalf("invalid duration %v != 10000000", v.TTL) | 		t.Fatalf("invalid duration %v != 10000000", v.TTL) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	err = yaml.Unmarshal([]byte(`{"ttl":"1d"}`), v) | 	err = yaml.Unmarshal([]byte(`{"ttl":"1y"}`), v) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		t.Fatal(err) | 		t.Fatal(err) | ||||||
| 	} else if *(v.TTL) != 86400000000000 { | 	} else if *(v.TTL) != 31622400000000000 { | ||||||
| 		t.Fatalf("invalid duration %v != 86400000000000", *v.TTL) | 		t.Fatalf("invalid duration %v != 31622400000000000", v.TTL) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -68,11 +68,11 @@ func TestUnmarshalJSON(t *testing.T) { | |||||||
| 		t.Fatalf("invalid duration %v != 10000000", v.TTL) | 		t.Fatalf("invalid duration %v != 10000000", v.TTL) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	err = json.Unmarshal([]byte(`{"ttl":"1d"}`), v) | 	err = json.Unmarshal([]byte(`{"ttl":"1y"}`), v) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		t.Fatal(err) | 		t.Fatal(err) | ||||||
| 	} else if v.TTL != 86400000000000 { | 	} else if v.TTL != 31622400000000000 { | ||||||
| 		t.Fatalf("invalid duration %v != 86400000000000", v.TTL) | 		t.Fatalf("invalid duration %v != 31622400000000000", v.TTL) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -87,11 +87,11 @@ func TestParseDuration(t *testing.T) { | |||||||
| 	if td.String() != "340h0m0s" { | 	if td.String() != "340h0m0s" { | ||||||
| 		t.Fatalf("ParseDuration 14d != 340h0m0s : %s", td.String()) | 		t.Fatalf("ParseDuration 14d != 340h0m0s : %s", td.String()) | ||||||
| 	} | 	} | ||||||
| 	td, err = ParseDuration("1d") | 	td, err = ParseDuration("1y") | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		t.Fatalf("ParseDuration error: %v", err) | 		t.Fatalf("ParseDuration error: %v", err) | ||||||
| 	} | 	} | ||||||
| 	if td.String() != "24h0m0s" { | 	if td.String() != "8784h0m0s" { | ||||||
| 		t.Fatalf("ParseDuration 1d != 24h0m0s : %s", td.String()) | 		t.Fatalf("ParseDuration 1y != 8784h0m0s : %s", td.String()) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
		Reference in New Issue
	
	Block a user