Compare commits
	
		
			219 Commits
		
	
	
		
			v4.0.7
			...
			vtolstov-p
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
|  | 629087ec7e | ||
| c576abf26a | |||
| 70d6a79274 | |||
| 664b1586af | |||
| 8d747c64a8 | |||
| 94beb5ed3b | |||
| 98981ba86c | |||
| 1013f50d0e | |||
| 0b190997b1 | |||
| 69a44eb190 | |||
| 0476028f69 | |||
| 330d8b149a | |||
| 19b04fe070 | |||
| 4cd55875c6 | |||
| a7896cc728 | |||
| ff991bf49c | |||
| 5a6551b703 | |||
| 9406a33d60 | |||
| 8f185abd9d | |||
| 86492e0644 | |||
| b21972964a | |||
| f5ee065d09 | |||
| 8cb02f2b08 | |||
| bc926cd6bd | |||
| 356abfd818 | |||
| 18444d3f98 | |||
| d5f07922e8 | |||
| 675e717410 | |||
| 7b6aea235a | |||
| 2cb7200467 | |||
| f430f97a97 | |||
| 0060c4377a | |||
| e46579fe9a | |||
| ca52973194 | |||
| 5bb33c7e1d | |||
| b71fc25328 | |||
| 9345dd075a | |||
| 1c1b9c0a28 | |||
| 2969494c5a | |||
| cbd3fa38ba | |||
| 569a36383d | |||
| 90bed77526 | |||
| ba4478a5e0 | |||
| 6dc76cdfea | |||
| e266683d96 | |||
| 2b62ad04f2 | |||
| 275b0a64e5 | |||
| 38c5fe8b5a | |||
| b6a0e4d983 | |||
| d9b822deff | |||
| 0e66688f8f | |||
| 9213fd212f | |||
| aa360dcf51 | |||
| 2df259b5b8 | |||
| 15e9310368 | |||
|  | 16d8cf3434 | ||
| 9704ef2e5e | |||
| 94e8f90f00 | |||
| 34d1587881 | |||
| bf4143cde5 | |||
| 36b7b9f5fb | |||
| ae97023092 | |||
| 115ca6a018 | |||
| 89cf4ef8af | |||
| 2a6ce6d4da | |||
| ad19fe2b90 | |||
| 49055a28ea | |||
| d1c6e121c1 | |||
| 7cd7fb0c0a | |||
| 77eb5b5264 | |||
| 929e46c087 | |||
| 1fb5673d27 | |||
| 3bbb0cbc72 | |||
| 71fe0df73f | |||
| f1b8ecbdb3 | |||
| fd2b2762e9 | |||
| 82d269cfb4 | |||
| 6641463eed | |||
| faf2454f0a | |||
| de9e4d73f5 | |||
| 4ae7277140 | |||
| a98618ed5b | |||
| 3aaf1182cb | |||
| eb1482d789 | |||
| a305f7553f | |||
|  | d9b2f2a45d | ||
| 3ace7657dc | |||
| 53b40617e2 | |||
| 1a9236caad | |||
| 6c68d39081 | |||
| 35e62fbeb0 | |||
| 00b3ceb468 | |||
| 7dc8f088c9 | |||
| c65afcea1b | |||
| 3eebfb5b11 | |||
| fa1427014c | |||
| 62074965ee | |||
| 9c8fbb2202 | |||
| 7c0a5f5e2a | |||
| b08f5321b0 | |||
| cc0f24e012 | |||
| 307a08f50c | |||
| edc93e8c37 | |||
| 391813c260 | |||
| 1a1459dd0e | |||
| 4e99680c30 | |||
| 92a3a547b8 | |||
| 849c462037 | |||
| 54a55c83e2 | |||
| 781dee03db | |||
| 26dd2eb405 | |||
| 3a21069b86 | |||
| add3ce478c | |||
|  | c3de003e4a | ||
| 7b7cf18a65 | |||
| 1bcf71c189 | |||
| c320d8e518 | |||
| b5f8316b57 | |||
| d7ddd912a8 | |||
| c020d90cb4 | |||
| db47b62159 | |||
| 8254456c8b | |||
| c2808679c3 | |||
| f418235c16 | |||
| 67ba7b3753 | |||
| e48d7cadf9 | |||
| c906186011 | |||
| dc0ff91b83 | |||
| e739c2d438 | |||
| bf4a036652 | |||
| f83a29eb67 | |||
| aef7f53d88 | |||
| 02c8e4fb7f | |||
| f5693bd940 | |||
| 701afb7bea | |||
| 019b407e74 | |||
| f29a346434 | |||
| 27db1876c0 | |||
| f66ac9736b | |||
| ed7972a1fa | |||
| 2cc004b01c | |||
| df951e5daf | |||
| 5bec0cef03 | |||
| 34940b68d7 | |||
| 1c57127128 | |||
| a4dd1a494c | |||
| 60e5e42167 | |||
| b519b61fff | |||
| f62b26eda3 | |||
| 13eda451da | |||
| 89cad06121 | |||
| 0bebf3d59f | |||
| 01e05e8df6 | |||
| 2b69a4f51c | |||
| 4af2b077dd | |||
| de4418189d | |||
| 2c44550897 | |||
| 99b8a3c950 | |||
| 4c7e1607d4 | |||
| 897be419b4 | |||
| 81b9a4341f | |||
| d3bb2f7236 | |||
| 97fd62cb21 | |||
| 3cd8bc33d6 | |||
| f6f67af8d0 | |||
| 2d5acaca2f | |||
| 0674df3d9f | |||
| 2c282825ce | |||
| e87ff942bb | |||
| 0459ea0613 | |||
| d44a75d074 | |||
|  | ccf92eb84d | ||
| 6baf1f2744 | |||
| 8e2eafde9c | |||
| c2b97b0f20 | |||
| 1db017d966 | |||
| debf8cb03d | |||
| 1dc9c1891f | |||
| 930859a537 | |||
| 3141f1ed8b | |||
| 47943cfb05 | |||
| ed4e9d54b1 | |||
| b4b8583594 | |||
| fb43e8c58c | |||
| 8863c10ef4 | |||
| 8058095bcc | |||
| 092f5d96b1 | |||
| 84552513f7 | |||
| 80a2db264e | |||
| 0be09c8b3e | |||
| 047f479e1b | |||
| 8f757c953e | |||
| 5f1c673a24 | |||
| 6794ea9871 | |||
| 089e7b6812 | |||
| 1c703f0f0c | |||
| d167c8c67c | |||
| df4f96a2d8 | |||
| fac3b20bd4 | |||
| 7c6bd98498 | |||
| 23e1174f25 | |||
| 52bed214cf | |||
| 64c4f5f47e | |||
| 036c612137 | |||
| ca80e3ecf2 | |||
| 18e7bb41ca | |||
| 8e72fb1c35 | |||
| 17f21a03f4 | |||
| a076d43a26 | |||
| de6efaee0b | |||
| 9e0e657003 | |||
| be5f9ab77f | |||
| 144dca0cae | |||
| 75173560e3 | |||
| 9b3bccd1f1 | |||
| ce125b77c1 | |||
| 2ee8d4ed46 | |||
| f58781d076 | |||
| e1af4aa3a4 | 
							
								
								
									
										51
									
								
								.gitea/workflows/job_coverage.yml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										51
									
								
								.gitea/workflows/job_coverage.yml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,51 @@ | |||||||
|  | name: coverage | ||||||
|  |  | ||||||
|  | on: | ||||||
|  |   push: | ||||||
|  |     branches: [ main, v3, v4 ] | ||||||
|  |   pull_request: | ||||||
|  |     branches: [ main, v3, v4 ] | ||||||
|  |   # Allows you to run this workflow manually from the Actions tab | ||||||
|  |   workflow_dispatch: | ||||||
|  |  | ||||||
|  | jobs: | ||||||
|  |  | ||||||
|  |   build: | ||||||
|  |     runs-on: ubuntu-latest | ||||||
|  |     steps: | ||||||
|  |     - name: checkout code | ||||||
|  |       uses: actions/checkout@v4 | ||||||
|  |       with: | ||||||
|  |         filter: 'blob:none' | ||||||
|  |  | ||||||
|  |     - name: setup go | ||||||
|  |       uses: actions/setup-go@v5 | ||||||
|  |       with: | ||||||
|  |         cache-dependency-path: "**/*.sum" | ||||||
|  |         go-version: 'stable'  | ||||||
|  |  | ||||||
|  |     - name: test coverage | ||||||
|  |       run: | | ||||||
|  |         go test -v -cover ./... -coverprofile coverage.out -coverpkg ./... | ||||||
|  |         go tool cover -func coverage.out -o coverage.out | ||||||
|  |  | ||||||
|  |     - name: coverage badge | ||||||
|  |       uses: tj-actions/coverage-badge-go@v1 | ||||||
|  |       with: | ||||||
|  |         green: 80 | ||||||
|  |         filename: coverage.out | ||||||
|  |  | ||||||
|  |     - uses: stefanzweifel/git-auto-commit-action@v4 | ||||||
|  |       id: auto-commit-action | ||||||
|  |       with: | ||||||
|  |         commit_message: Apply Code Coverage Badge | ||||||
|  |         skip_fetch: true | ||||||
|  |         skip_checkout: true | ||||||
|  |         file_pattern: ./README.md | ||||||
|  |  | ||||||
|  |     - name: Push Changes | ||||||
|  |       if: steps.auto-commit-action.outputs.changes_detected == 'true' | ||||||
|  |       uses: ad-m/github-push-action@master | ||||||
|  |       with: | ||||||
|  |         github_token: ${{ github.token }} | ||||||
|  |         branch: ${{ github.ref }} | ||||||
							
								
								
									
										29
									
								
								.gitea/workflows/job_lint.yml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										29
									
								
								.gitea/workflows/job_lint.yml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,29 @@ | |||||||
|  | name: lint | ||||||
|  |  | ||||||
|  | on: | ||||||
|  |   pull_request: | ||||||
|  |     types: [opened, reopened, synchronize] | ||||||
|  |     branches: | ||||||
|  |     - master | ||||||
|  |     - v3 | ||||||
|  |     - v4 | ||||||
|  |  | ||||||
|  | jobs: | ||||||
|  |   lint: | ||||||
|  |     runs-on: ubuntu-latest | ||||||
|  |     steps: | ||||||
|  |     - name: checkout code | ||||||
|  |       uses: actions/checkout@v4 | ||||||
|  |       with: | ||||||
|  |         filter: 'blob:none' | ||||||
|  |     - name: setup go | ||||||
|  |       uses: actions/setup-go@v5 | ||||||
|  |       with: | ||||||
|  |         cache-dependency-path: "**/*.sum" | ||||||
|  |         go-version: 'stable'  | ||||||
|  |     - name: setup deps | ||||||
|  |       run: go get -v ./... | ||||||
|  |     - name: run lint | ||||||
|  |       uses: https://github.com/golangci/golangci-lint-action@v6 | ||||||
|  |       with: | ||||||
|  |         version: 'latest' | ||||||
							
								
								
									
										34
									
								
								.gitea/workflows/job_test.yml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										34
									
								
								.gitea/workflows/job_test.yml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,34 @@ | |||||||
|  | name: test | ||||||
|  |  | ||||||
|  | on: | ||||||
|  |   pull_request: | ||||||
|  |     types: [opened, reopened, synchronize] | ||||||
|  |     branches: | ||||||
|  |     - master | ||||||
|  |     - v3 | ||||||
|  |     - v4 | ||||||
|  |   push: | ||||||
|  |     branches: | ||||||
|  |     - master | ||||||
|  |     - v3 | ||||||
|  |     - v4 | ||||||
|  |  | ||||||
|  | jobs: | ||||||
|  |   test: | ||||||
|  |     runs-on: ubuntu-latest | ||||||
|  |     steps: | ||||||
|  |     - name: checkout code | ||||||
|  |       uses: actions/checkout@v4 | ||||||
|  |       with: | ||||||
|  |         filter: 'blob:none' | ||||||
|  |     - name: setup go | ||||||
|  |       uses: actions/setup-go@v5 | ||||||
|  |       with: | ||||||
|  |         cache-dependency-path: "**/*.sum" | ||||||
|  |         go-version: 'stable' | ||||||
|  |     - name: setup deps | ||||||
|  |       run: go get -v ./... | ||||||
|  |     - name: run test | ||||||
|  |       env: | ||||||
|  |         INTEGRATION_TESTS: yes | ||||||
|  |       run: go test -mod readonly -v ./... | ||||||
							
								
								
									
										53
									
								
								.gitea/workflows/job_tests.yml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										53
									
								
								.gitea/workflows/job_tests.yml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,53 @@ | |||||||
|  | name: test | ||||||
|  |  | ||||||
|  | on: | ||||||
|  |   pull_request: | ||||||
|  |     types: [opened, reopened, synchronize] | ||||||
|  |     branches: | ||||||
|  |     - master | ||||||
|  |     - v3 | ||||||
|  |     - v4 | ||||||
|  |   push: | ||||||
|  |     branches: | ||||||
|  |     - master | ||||||
|  |     - v3 | ||||||
|  |     - v4 | ||||||
|  |  | ||||||
|  | jobs: | ||||||
|  |   test: | ||||||
|  |     runs-on: ubuntu-latest | ||||||
|  |     steps: | ||||||
|  |     - name: checkout code | ||||||
|  |       uses: actions/checkout@v4 | ||||||
|  |       with: | ||||||
|  |         filter: 'blob:none' | ||||||
|  |     - name: checkout tests | ||||||
|  |       uses: actions/checkout@v4 | ||||||
|  |       with: | ||||||
|  |         ref: master | ||||||
|  |         filter: 'blob:none' | ||||||
|  |         repository: unistack-org/micro-tests | ||||||
|  |         path: micro-tests | ||||||
|  |     - name: setup go | ||||||
|  |       uses: actions/setup-go@v5 | ||||||
|  |       with: | ||||||
|  |         cache-dependency-path: "**/*.sum" | ||||||
|  |         go-version: 'stable' | ||||||
|  |     - name: setup go work | ||||||
|  |       env: | ||||||
|  |         GOWORK: /workspace/${{ github.repository_owner }}/go.work | ||||||
|  |       run: | | ||||||
|  |         go work init | ||||||
|  |         go work use . | ||||||
|  |         go work use micro-tests | ||||||
|  |     - name: setup deps | ||||||
|  |       env: | ||||||
|  |         GOWORK: /workspace/${{ github.repository_owner }}/go.work | ||||||
|  |       run: go get -v ./... | ||||||
|  |     - name: run tests | ||||||
|  |       env: | ||||||
|  |         INTEGRATION_TESTS: yes | ||||||
|  |         GOWORK: /workspace/${{ github.repository_owner }}/go.work | ||||||
|  |       run: | | ||||||
|  |         cd micro-tests | ||||||
|  |         go test -mod readonly -v ./... || true | ||||||
| @@ -1,24 +0,0 @@ | |||||||
| name: lint |  | ||||||
| on: |  | ||||||
|   pull_request: |  | ||||||
|     branches: |  | ||||||
|     - master |  | ||||||
|     - v3 |  | ||||||
| jobs: |  | ||||||
|   lint: |  | ||||||
|     name: lint |  | ||||||
|     runs-on: ubuntu-latest |  | ||||||
|     steps: |  | ||||||
|     - name: setup-go |  | ||||||
|       uses: https://gitea.com/actions/setup-go@v3 |  | ||||||
|       with: |  | ||||||
|         go-version: 1.18 |  | ||||||
|     - name: checkout |  | ||||||
|       uses: https://gitea.com/actions/checkout@v3 |  | ||||||
|     - name: deps |  | ||||||
|       run: go get -v -d ./... |  | ||||||
|     - name: lint |  | ||||||
|       uses: https://github.com/golangci/golangci-lint-action@v3.4.0 |  | ||||||
|       continue-on-error: true |  | ||||||
|       with: |  | ||||||
|         version: v1.52 |  | ||||||
| @@ -1,23 +0,0 @@ | |||||||
| name: pr |  | ||||||
| on: |  | ||||||
|   pull_request: |  | ||||||
|     branches: |  | ||||||
|     - master |  | ||||||
|     - v3 |  | ||||||
| jobs: |  | ||||||
|   test: |  | ||||||
|     name: test |  | ||||||
|     runs-on: ubuntu-latest |  | ||||||
|     steps: |  | ||||||
|     - name: checkout |  | ||||||
|       uses: https://gitea.com/actions/checkout@v3 |  | ||||||
|     - name: setup-go |  | ||||||
|       uses: https://gitea.com/actions/setup-go@v3 |  | ||||||
|       with: |  | ||||||
|         go-version: 1.18 |  | ||||||
|     - name: deps |  | ||||||
|       run: go get -v -t -d ./... |  | ||||||
|     - name: test |  | ||||||
|       env: |  | ||||||
|         INTEGRATION_TESTS: yes |  | ||||||
|       run: go test -mod readonly -v ./... |  | ||||||
							
								
								
									
										3
									
								
								.gitignore
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										3
									
								
								.gitignore
									
									
									
									
										vendored
									
									
								
							| @@ -1,6 +1,8 @@ | |||||||
| # Develop tools | # Develop tools | ||||||
| /.vscode/ | /.vscode/ | ||||||
| /.idea/ | /.idea/ | ||||||
|  | .idea | ||||||
|  | .vscode | ||||||
|  |  | ||||||
| # Binaries for programs and plugins | # Binaries for programs and plugins | ||||||
| *.exe | *.exe | ||||||
| @@ -13,6 +15,7 @@ | |||||||
| _obj | _obj | ||||||
| _test | _test | ||||||
| _build | _build | ||||||
|  | .DS_Store | ||||||
|  |  | ||||||
| # Architecture specific extensions/prefixes | # Architecture specific extensions/prefixes | ||||||
| *.[568vq] | *.[568vq] | ||||||
|   | |||||||
| @@ -1,44 +1,5 @@ | |||||||
| run: | run: | ||||||
|   concurrency: 4 |   concurrency: 8 | ||||||
|   deadline: 5m |   deadline: 5m | ||||||
|   issues-exit-code: 1 |   issues-exit-code: 1 | ||||||
|   tests: true |   tests: true | ||||||
|  |  | ||||||
| linters-settings: |  | ||||||
|   govet: |  | ||||||
|     check-shadowing: true |  | ||||||
|     enable: |  | ||||||
|       - fieldalignment |  | ||||||
|  |  | ||||||
| linters: |  | ||||||
|   enable: |  | ||||||
|     - govet |  | ||||||
|     - deadcode |  | ||||||
|     - errcheck |  | ||||||
|     - govet |  | ||||||
|     - ineffassign |  | ||||||
|     - staticcheck |  | ||||||
|     - structcheck |  | ||||||
|     - typecheck |  | ||||||
|     - unused |  | ||||||
|     - varcheck |  | ||||||
|     - bodyclose |  | ||||||
|     - gci |  | ||||||
|     - goconst |  | ||||||
|     - gocritic |  | ||||||
|     - gosimple |  | ||||||
|     - gofmt |  | ||||||
|     - gofumpt |  | ||||||
|     - goimports |  | ||||||
|     - revive |  | ||||||
|     - gosec |  | ||||||
|     - makezero |  | ||||||
|     - misspell |  | ||||||
|     - nakedret |  | ||||||
|     - nestif |  | ||||||
|     - nilerr |  | ||||||
|     - noctx |  | ||||||
|     - prealloc |  | ||||||
|     - unconvert |  | ||||||
|     - unparam |  | ||||||
|   disable-all: false |  | ||||||
|   | |||||||
| @@ -1,4 +1,5 @@ | |||||||
| # Micro [](https://opensource.org/licenses/Apache-2.0) [](https://pkg.go.dev/go.unistack.org/micro/v4?tab=overview) [](https://github.com/unistack-org/micro/actions?query=workflow%3Abuild+branch%3Amaster+event%3Apush) [](https://goreportcard.com/report/go.unistack.org/micro/v4) [](https://codecov.io/gh/unistack-org/micro) | # Micro  | ||||||
|  |  | ||||||
|  |  | ||||||
| Micro is a standard library for microservices. | Micro is a standard library for microservices. | ||||||
|  |  | ||||||
|   | |||||||
							
								
								
									
										105
									
								
								broker/broker.go
									
									
									
									
									
								
							
							
						
						
									
										105
									
								
								broker/broker.go
									
									
									
									
									
								
							| @@ -1,12 +1,13 @@ | |||||||
| // Package broker is an interface used for asynchronous messaging | // Package broker is an interface used for asynchronous messaging | ||||||
| package broker // import "go.unistack.org/micro/v4/broker" | package broker | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"errors" | 	"errors" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/metadata" | 	"go.unistack.org/micro/v3/codec" | ||||||
| 	"go.unistack.org/micro/v4/options" | 	"go.unistack.org/micro/v3/metadata" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // DefaultBroker default memory broker | // DefaultBroker default memory broker | ||||||
| @@ -17,8 +18,8 @@ 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 message has nvalid format | 	// DefaultGracefulTimeout | ||||||
| 	ErrInvalidMessage = errors.New("broker message has invalid format") | 	DefaultGracefulTimeout = 5 * time.Second | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Broker is an interface used for asynchronous messaging. | // Broker is an interface used for asynchronous messaging. | ||||||
| @@ -26,7 +27,7 @@ type Broker interface { | |||||||
| 	// Name returns broker instance name | 	// Name returns broker instance name | ||||||
| 	Name() string | 	Name() string | ||||||
| 	// Init initilize broker | 	// Init initilize broker | ||||||
| 	Init(opts ...options.Option) error | 	Init(opts ...Option) error | ||||||
| 	// Options returns broker options | 	// Options returns broker options | ||||||
| 	Options() Options | 	Options() Options | ||||||
| 	// Address return configured address | 	// Address return configured address | ||||||
| @@ -35,29 +36,91 @@ 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 | ||||||
| 	// Publish message, msg can be single broker.Message or []broker.Message | 	// Publish message to broker topic | ||||||
| 	Publish(ctx context.Context, msg interface{}, opts ...options.Option) 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 ...options.Option) (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() bool | ||||||
|  | 	// Ready returns broker readiness | ||||||
|  | 	Ready() bool | ||||||
|  | 	// Health returns broker health | ||||||
|  | 	Health() bool | ||||||
| } | } | ||||||
|  |  | ||||||
| // Message is given to a subscription handler for processing | type ( | ||||||
| type Message interface { | 	FuncPublish        func(ctx context.Context, topic string, msg *Message, opts ...PublishOption) error | ||||||
| 	// Context for the message | 	HookPublish        func(next FuncPublish) FuncPublish | ||||||
|  | 	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 | ||||||
|  | 	FuncBatchSubscribe func(ctx context.Context, topic string, h BatchHandler, opts ...SubscribeOption) (Subscriber, error) | ||||||
|  | 	HookBatchSubscribe func(next FuncBatchSubscribe) FuncBatchSubscribe | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Handler is used to process messages via a subscription of a topic. | ||||||
|  | type Handler func(Event) error | ||||||
|  |  | ||||||
|  | // 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 | 	// 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 may be []byte slice or some go struct |  | ||||||
| 	Body() interface{} |  | ||||||
| 	// Ack acknowledge message | 	// Ack acknowledge message | ||||||
| 	Ack() error | 	Ack() error | ||||||
| 	// Error returns message error (like decoding errors or some other) | 	// Error returns message error (like decoding errors or some other) | ||||||
| 	// In this case Body contains raw []byte from broker |  | ||||||
| 	Error() error | 	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 | ||||||
| @@ -69,9 +132,3 @@ type Subscriber interface { | |||||||
| 	// Unsubscribe from topic | 	// Unsubscribe from topic | ||||||
| 	Unsubscribe(ctx context.Context) error | 	Unsubscribe(ctx context.Context) error | ||||||
| } | } | ||||||
|  |  | ||||||
| // MessageHandler func signature for single message processing |  | ||||||
| type MessageHandler func(Message) error |  | ||||||
|  |  | ||||||
| // MessagesHandler func signature for batch message processing |  | ||||||
| type MessagesHandler func([]Message) error |  | ||||||
|   | |||||||
| @@ -15,6 +15,15 @@ func FromContext(ctx context.Context) (Broker, bool) { | |||||||
| 	return c, ok | 	return c, ok | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // MustContext returns broker from passed context | ||||||
|  | func MustContext(ctx context.Context) Broker { | ||||||
|  | 	b, ok := FromContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		panic("missing broker") | ||||||
|  | 	} | ||||||
|  | 	return b | ||||||
|  | } | ||||||
|  |  | ||||||
| // NewContext savess broker in context | // NewContext savess broker in context | ||||||
| func NewContext(ctx context.Context, s Broker) context.Context { | func NewContext(ctx context.Context, s Broker) context.Context { | ||||||
| 	if ctx == nil { | 	if ctx == nil { | ||||||
| @@ -22,3 +31,33 @@ func NewContext(ctx context.Context, s Broker) context.Context { | |||||||
| 	} | 	} | ||||||
| 	return context.WithValue(ctx, brokerKey{}, s) | 	return context.WithValue(ctx, brokerKey{}, s) | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // SetSubscribeOption returns a function to setup a context with given value | ||||||
|  | func SetSubscribeOption(k, v interface{}) SubscribeOption { | ||||||
|  | 	return func(o *SubscribeOptions) { | ||||||
|  | 		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 | ||||||
|  | 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) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // 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) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -37,3 +37,36 @@ func TestNewNilContext(t *testing.T) { | |||||||
| 		t.Fatal("NewContext not works") | 		t.Fatal("NewContext not works") | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func TestSetSubscribeOption(t *testing.T) { | ||||||
|  | 	type key struct{} | ||||||
|  | 	o := SetSubscribeOption(key{}, "test") | ||||||
|  | 	opts := &SubscribeOptions{} | ||||||
|  | 	o(opts) | ||||||
|  |  | ||||||
|  | 	if v, ok := opts.Context.Value(key{}).(string); !ok || v == "" { | ||||||
|  | 		t.Fatal("SetSubscribeOption not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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) { | ||||||
|  | 	type key struct{} | ||||||
|  | 	o := SetOption(key{}, "test") | ||||||
|  | 	opts := &Options{} | ||||||
|  | 	o(opts) | ||||||
|  |  | ||||||
|  | 	if v, ok := opts.Context.Value(key{}).(string); !ok || v == "" { | ||||||
|  | 		t.Fatal("SetOption not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
							
								
								
									
										334
									
								
								broker/memory.go
									
									
									
									
									
								
							
							
						
						
									
										334
									
								
								broker/memory.go
									
									
									
									
									
								
							| @@ -1,334 +0,0 @@ | |||||||
| package broker |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"context" |  | ||||||
| 	"fmt" |  | ||||||
| 	"sync" |  | ||||||
| 	"time" |  | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/codec" |  | ||||||
| 	"go.unistack.org/micro/v4/logger" |  | ||||||
| 	"go.unistack.org/micro/v4/metadata" |  | ||||||
| 	"go.unistack.org/micro/v4/options" |  | ||||||
| 	"go.unistack.org/micro/v4/semconv" |  | ||||||
| 	maddr "go.unistack.org/micro/v4/util/addr" |  | ||||||
| 	"go.unistack.org/micro/v4/util/id" |  | ||||||
| 	mnet "go.unistack.org/micro/v4/util/net" |  | ||||||
| 	"go.unistack.org/micro/v4/util/rand" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| type MemoryBroker struct { |  | ||||||
| 	subscribers map[string][]*memorySubscriber |  | ||||||
| 	addr        string |  | ||||||
| 	opts        Options |  | ||||||
| 	sync.RWMutex |  | ||||||
| 	connected bool |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *MemoryBroker) Options() Options { |  | ||||||
| 	return m.opts |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *MemoryBroker) Address() string { |  | ||||||
| 	return m.addr |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *MemoryBroker) Connect(ctx context.Context) error { |  | ||||||
| 	m.Lock() |  | ||||||
| 	defer m.Unlock() |  | ||||||
|  |  | ||||||
| 	if m.connected { |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// use 127.0.0.1 to avoid scan of all network interfaces |  | ||||||
| 	addr, err := maddr.Extract("127.0.0.1") |  | ||||||
| 	if err != nil { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
| 	var rng rand.Rand |  | ||||||
| 	i := rng.Intn(20000) |  | ||||||
| 	// set addr with port |  | ||||||
| 	addr = mnet.HostPort(addr, 10000+i) |  | ||||||
|  |  | ||||||
| 	m.addr = addr |  | ||||||
| 	m.connected = true |  | ||||||
|  |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *MemoryBroker) Disconnect(ctx context.Context) error { |  | ||||||
| 	m.Lock() |  | ||||||
| 	defer m.Unlock() |  | ||||||
|  |  | ||||||
| 	select { |  | ||||||
| 	case <-ctx.Done(): |  | ||||||
| 		return ctx.Err() |  | ||||||
| 	default: |  | ||||||
| 		if m.connected { |  | ||||||
| 			m.connected = false |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *MemoryBroker) Init(opts ...options.Option) error { |  | ||||||
| 	var err error |  | ||||||
| 	for _, o := range opts { |  | ||||||
| 		if err = o(&m.opts); err != nil { |  | ||||||
| 			return err |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *MemoryBroker) Publish(ctx context.Context, message interface{}, opts ...options.Option) error { |  | ||||||
| 	m.RLock() |  | ||||||
| 	if !m.connected { |  | ||||||
| 		m.RUnlock() |  | ||||||
| 		return ErrNotConnected |  | ||||||
| 	} |  | ||||||
| 	m.RUnlock() |  | ||||||
|  |  | ||||||
| 	var err error |  | ||||||
|  |  | ||||||
| 	select { |  | ||||||
| 	case <-ctx.Done(): |  | ||||||
| 		return ctx.Err() |  | ||||||
| 	default: |  | ||||||
| 		options := NewPublishOptions(opts...) |  | ||||||
| 		var msgs []Message |  | ||||||
| 		switch v := message.(type) { |  | ||||||
| 		case []Message: |  | ||||||
| 			msgs = v |  | ||||||
| 		case Message: |  | ||||||
| 			msgs = append(msgs, v) |  | ||||||
| 		default: |  | ||||||
| 			return ErrInvalidMessage |  | ||||||
| 		} |  | ||||||
| 		msgTopicMap := make(map[string][]*memoryMessage) |  | ||||||
| 		for _, msg := range msgs { |  | ||||||
| 			p := &memoryMessage{opts: options} |  | ||||||
| 			p.topic, _ = msg.Header().Get(metadata.HeaderTopic) |  | ||||||
| 			if v, ok := msg.Body().(*codec.Frame); ok { |  | ||||||
| 				p.body = msg.Body() |  | ||||||
| 			} else if len(m.opts.Codecs) == 0 { |  | ||||||
| 				p.body = msg.Body() |  | ||||||
| 			} else { |  | ||||||
| 				cf, ok := m.opts.Codecs[options.ContentType] |  | ||||||
| 				if !ok { |  | ||||||
| 					return fmt.Errorf("%s: %s", codec.ErrUnknownContentType, options.ContentType) |  | ||||||
| 				} |  | ||||||
| 				p.body, err = cf.Marshal(v) |  | ||||||
| 				if err != nil { |  | ||||||
| 					return err |  | ||||||
| 				} |  | ||||||
| 			} |  | ||||||
| 			msgTopicMap[p.topic] = append(msgTopicMap[p.topic], p) |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		eh := m.opts.ErrorHandler |  | ||||||
|  |  | ||||||
| 		for t, ms := range msgTopicMap { |  | ||||||
| 			ts := time.Now() |  | ||||||
|  |  | ||||||
| 			m.opts.Meter.Counter(semconv.PublishMessageInflight, "endpoint", t).Add(len(ms)) |  | ||||||
| 			m.opts.Meter.Counter(semconv.SubscribeMessageInflight, "endpoint", t).Add(len(ms)) |  | ||||||
|  |  | ||||||
| 			m.RLock() |  | ||||||
| 			subs, ok := m.subscribers[t] |  | ||||||
| 			m.RUnlock() |  | ||||||
| 			if !ok { |  | ||||||
| 				m.opts.Meter.Counter(semconv.PublishMessageTotal, "endpoint", t, "status", "failure").Add(len(ms)) |  | ||||||
| 				m.opts.Meter.Counter(semconv.PublishMessageInflight, "endpoint", t).Add(-len(ms)) |  | ||||||
| 				m.opts.Meter.Counter(semconv.SubscribeMessageInflight, "endpoint", t).Add(-len(ms)) |  | ||||||
| 				continue |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			m.opts.Meter.Counter(semconv.PublishMessageTotal, "endpoint", t, "status", "success").Add(len(ms)) |  | ||||||
| 			for _, sub := range subs { |  | ||||||
| 				if sub.opts.ErrorHandler != nil { |  | ||||||
| 					eh = sub.opts.ErrorHandler |  | ||||||
| 				} |  | ||||||
|  |  | ||||||
| 				switch mh := sub.handler.(type) { |  | ||||||
| 				case MessagesHandler: |  | ||||||
| 					mhs := make([]Message, 0, len(ms)) |  | ||||||
| 					for _, m := range ms { |  | ||||||
| 						mhs = append(mhs, m) |  | ||||||
| 					} |  | ||||||
| 					if err = mh(mhs); err != nil { |  | ||||||
| 						m.opts.Meter.Counter(semconv.SubscribeMessageTotal, "endpoint", t, "status", "failure").Add(len(ms)) |  | ||||||
| 						if eh != nil { |  | ||||||
| 							switch meh := eh.(type) { |  | ||||||
| 							case MessagesHandler: |  | ||||||
| 								_ = meh(mhs) |  | ||||||
| 							case MessageHandler: |  | ||||||
| 								for _, me := range mhs { |  | ||||||
| 									_ = meh(me) |  | ||||||
| 								} |  | ||||||
| 							} |  | ||||||
| 						} else if m.opts.Logger.V(logger.ErrorLevel) { |  | ||||||
| 							m.opts.Logger.Error(m.opts.Context, err.Error()) |  | ||||||
| 						} |  | ||||||
| 					} |  | ||||||
| 				case MessageHandler: |  | ||||||
| 					for _, p := range ms { |  | ||||||
| 						if err = mh(p); err != nil { |  | ||||||
| 							m.opts.Meter.Counter(semconv.SubscribeMessageTotal, "endpoint", t, "status", "failure").Inc() |  | ||||||
| 							if eh != nil { |  | ||||||
| 								switch meh := eh.(type) { |  | ||||||
| 								case MessageHandler: |  | ||||||
| 									_ = meh(p) |  | ||||||
| 								case MessagesHandler: |  | ||||||
| 									_ = meh([]Message{p}) |  | ||||||
| 								} |  | ||||||
| 							} else if m.opts.Logger.V(logger.ErrorLevel) { |  | ||||||
| 								m.opts.Logger.Error(m.opts.Context, err.Error()) |  | ||||||
| 							} |  | ||||||
| 						} else { |  | ||||||
| 							if sub.opts.AutoAck { |  | ||||||
| 								if err = p.Ack(); err != nil { |  | ||||||
| 									m.opts.Logger.Errorf(m.opts.Context, "ack failed: %v", err) |  | ||||||
| 									m.opts.Meter.Counter(semconv.SubscribeMessageTotal, "endpoint", t, "status", "failure").Inc() |  | ||||||
| 								} else { |  | ||||||
| 									m.opts.Meter.Counter(semconv.SubscribeMessageTotal, "endpoint", t, "status", "success").Inc() |  | ||||||
| 								} |  | ||||||
| 							} else { |  | ||||||
| 								m.opts.Meter.Counter(semconv.SubscribeMessageTotal, "endpoint", t, "status", "success").Inc() |  | ||||||
| 							} |  | ||||||
| 						} |  | ||||||
| 						m.opts.Meter.Counter(semconv.PublishMessageInflight, "endpoint", t).Add(-1) |  | ||||||
| 						m.opts.Meter.Counter(semconv.SubscribeMessageInflight, "endpoint", t).Add(-1) |  | ||||||
| 					} |  | ||||||
| 				} |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			te := time.Since(ts) |  | ||||||
| 			m.opts.Meter.Summary(semconv.PublishMessageLatencyMicroseconds, "endpoint", t).Update(te.Seconds()) |  | ||||||
| 			m.opts.Meter.Histogram(semconv.PublishMessageDurationSeconds, "endpoint", t).Update(te.Seconds()) |  | ||||||
| 			m.opts.Meter.Summary(semconv.SubscribeMessageLatencyMicroseconds, "endpoint", t).Update(te.Seconds()) |  | ||||||
| 			m.opts.Meter.Histogram(semconv.SubscribeMessageDurationSeconds, "endpoint", t).Update(te.Seconds()) |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *MemoryBroker) Subscribe(ctx context.Context, topic string, handler interface{}, opts ...options.Option) (Subscriber, error) { |  | ||||||
| 	m.RLock() |  | ||||||
| 	if !m.connected { |  | ||||||
| 		m.RUnlock() |  | ||||||
| 		return nil, ErrNotConnected |  | ||||||
| 	} |  | ||||||
| 	m.RUnlock() |  | ||||||
|  |  | ||||||
| 	sid, err := id.New() |  | ||||||
| 	if err != nil { |  | ||||||
| 		return nil, err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	options := NewSubscribeOptions(opts...) |  | ||||||
|  |  | ||||||
| 	sub := &memorySubscriber{ |  | ||||||
| 		exit:    make(chan bool, 1), |  | ||||||
| 		id:      sid, |  | ||||||
| 		topic:   topic, |  | ||||||
| 		handler: 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) String() string { |  | ||||||
| 	return "memory" |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *MemoryBroker) Name() string { |  | ||||||
| 	return m.opts.Name |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type memoryMessage struct { |  | ||||||
| 	err    error |  | ||||||
| 	body   interface{} |  | ||||||
| 	topic  string |  | ||||||
| 	header metadata.Metadata |  | ||||||
| 	opts   PublishOptions |  | ||||||
| 	ctx    context.Context |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryMessage) Topic() string { |  | ||||||
| 	return m.topic |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryMessage) Header() metadata.Metadata { |  | ||||||
| 	return m.header |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryMessage) Body() interface{} { |  | ||||||
| 	return m.body |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryMessage) Ack() error { |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryMessage) Error() error { |  | ||||||
| 	return m.err |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryMessage) Context() context.Context { |  | ||||||
| 	return m.ctx |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type memorySubscriber struct { |  | ||||||
| 	ctx     context.Context |  | ||||||
| 	exit    chan bool |  | ||||||
| 	handler interface{} |  | ||||||
| 	id      string |  | ||||||
| 	topic   string |  | ||||||
| 	opts    SubscribeOptions |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memorySubscriber) Options() SubscribeOptions { |  | ||||||
| 	return m.opts |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memorySubscriber) Topic() string { |  | ||||||
| 	return m.topic |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memorySubscriber) Unsubscribe(ctx context.Context) error { |  | ||||||
| 	m.exit <- true |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewBroker return new memory broker |  | ||||||
| func NewBroker(opts ...options.Option) *MemoryBroker { |  | ||||||
| 	return &MemoryBroker{ |  | ||||||
| 		opts:        NewOptions(opts...), |  | ||||||
| 		subscribers: make(map[string][]*memorySubscriber), |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
							
								
								
									
										411
									
								
								broker/memory/memory.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										411
									
								
								broker/memory/memory.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,411 @@ | |||||||
|  | package broker | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"sync" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v3/broker" | ||||||
|  | 	"go.unistack.org/micro/v3/logger" | ||||||
|  | 	"go.unistack.org/micro/v3/metadata" | ||||||
|  | 	"go.unistack.org/micro/v3/options" | ||||||
|  | 	maddr "go.unistack.org/micro/v3/util/addr" | ||||||
|  | 	"go.unistack.org/micro/v3/util/id" | ||||||
|  | 	mnet "go.unistack.org/micro/v3/util/net" | ||||||
|  | 	"go.unistack.org/micro/v3/util/rand" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type memoryBroker struct { | ||||||
|  | 	funcPublish        broker.FuncPublish | ||||||
|  | 	funcBatchPublish   broker.FuncBatchPublish | ||||||
|  | 	funcSubscribe      broker.FuncSubscribe | ||||||
|  | 	funcBatchSubscribe broker.FuncBatchSubscribe | ||||||
|  | 	subscribers        map[string][]*memorySubscriber | ||||||
|  | 	addr               string | ||||||
|  | 	opts               broker.Options | ||||||
|  | 	sync.RWMutex | ||||||
|  | 	connected bool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type memoryEvent struct { | ||||||
|  | 	err     error | ||||||
|  | 	message interface{} | ||||||
|  | 	topic   string | ||||||
|  | 	opts    broker.Options | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type memorySubscriber struct { | ||||||
|  | 	ctx          context.Context | ||||||
|  | 	exit         chan bool | ||||||
|  | 	handler      broker.Handler | ||||||
|  | 	batchhandler broker.BatchHandler | ||||||
|  | 	id           string | ||||||
|  | 	topic        string | ||||||
|  | 	opts         broker.SubscribeOptions | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryBroker) Options() broker.Options { | ||||||
|  | 	return m.opts | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryBroker) Address() string { | ||||||
|  | 	return m.addr | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryBroker) Connect(ctx context.Context) error { | ||||||
|  | 	select { | ||||||
|  | 	case <-ctx.Done(): | ||||||
|  | 		return ctx.Err() | ||||||
|  | 	default: | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	m.Lock() | ||||||
|  | 	defer m.Unlock() | ||||||
|  |  | ||||||
|  | 	if m.connected { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// use 127.0.0.1 to avoid scan of all network interfaces | ||||||
|  | 	addr, err := maddr.Extract("127.0.0.1") | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	var rng rand.Rand | ||||||
|  | 	i := rng.Intn(20000) | ||||||
|  | 	// set addr with port | ||||||
|  | 	addr = mnet.HostPort(addr, 10000+i) | ||||||
|  |  | ||||||
|  | 	m.addr = addr | ||||||
|  | 	m.connected = true | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryBroker) Disconnect(ctx context.Context) error { | ||||||
|  | 	select { | ||||||
|  | 	case <-ctx.Done(): | ||||||
|  | 		return ctx.Err() | ||||||
|  | 	default: | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	m.Lock() | ||||||
|  | 	defer m.Unlock() | ||||||
|  |  | ||||||
|  | 	if !m.connected { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	m.connected = false | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryBroker) Init(opts ...broker.Option) error { | ||||||
|  | 	for _, o := range opts { | ||||||
|  | 		o(&m.opts) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	m.funcPublish = m.fnPublish | ||||||
|  | 	m.funcBatchPublish = m.fnBatchPublish | ||||||
|  | 	m.funcSubscribe = m.fnSubscribe | ||||||
|  | 	m.funcBatchSubscribe = m.fnBatchSubscribe | ||||||
|  |  | ||||||
|  | 	m.opts.Hooks.EachPrev(func(hook options.Hook) { | ||||||
|  | 		switch h := hook.(type) { | ||||||
|  | 		case broker.HookPublish: | ||||||
|  | 			m.funcPublish = h(m.funcPublish) | ||||||
|  | 		case broker.HookBatchPublish: | ||||||
|  | 			m.funcBatchPublish = h(m.funcBatchPublish) | ||||||
|  | 		case broker.HookSubscribe: | ||||||
|  | 			m.funcSubscribe = h(m.funcSubscribe) | ||||||
|  | 		case broker.HookBatchSubscribe: | ||||||
|  | 			m.funcBatchSubscribe = h(m.funcBatchSubscribe) | ||||||
|  | 		} | ||||||
|  | 	}) | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryBroker) Publish(ctx context.Context, topic string, msg *broker.Message, opts ...broker.PublishOption) error { | ||||||
|  | 	return m.funcPublish(ctx, topic, msg, opts...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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 (m *memoryBroker) BatchPublish(ctx context.Context, msgs []*broker.Message, opts ...broker.PublishOption) error { | ||||||
|  | 	return m.funcBatchPublish(ctx, msgs, opts...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryBroker) fnBatchPublish(ctx context.Context, msgs []*broker.Message, opts ...broker.PublishOption) error { | ||||||
|  | 	return m.publish(ctx, msgs, opts...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryBroker) publish(ctx context.Context, msgs []*broker.Message, opts ...broker.PublishOption) error { | ||||||
|  | 	m.RLock() | ||||||
|  | 	if !m.connected { | ||||||
|  | 		m.RUnlock() | ||||||
|  | 		return broker.ErrNotConnected | ||||||
|  | 	} | ||||||
|  | 	m.RUnlock() | ||||||
|  |  | ||||||
|  | 	var err error | ||||||
|  |  | ||||||
|  | 	select { | ||||||
|  | 	case <-ctx.Done(): | ||||||
|  | 		return ctx.Err() | ||||||
|  | 	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) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		beh := m.opts.BatchErrorHandler | ||||||
|  | 		eh := m.opts.ErrorHandler | ||||||
|  |  | ||||||
|  | 		for t, ms := range msgTopicMap { | ||||||
|  | 			m.RLock() | ||||||
|  | 			subs, ok := m.subscribers[t] | ||||||
|  | 			m.RUnlock() | ||||||
|  | 			if !ok { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			for _, sub := range subs { | ||||||
|  | 				if sub.opts.BatchErrorHandler != nil { | ||||||
|  | 					beh = sub.opts.BatchErrorHandler | ||||||
|  | 				} | ||||||
|  | 				if sub.opts.ErrorHandler != nil { | ||||||
|  | 					eh = sub.opts.ErrorHandler | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 				switch { | ||||||
|  | 				// 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) | ||||||
|  | 						} | ||||||
|  | 					} | ||||||
|  | 					// single processing | ||||||
|  | 				case sub.handler != nil: | ||||||
|  | 					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) | ||||||
|  | 							} | ||||||
|  | 						} else if sub.opts.AutoAck { | ||||||
|  | 							if err = p.Ack(); err != nil { | ||||||
|  | 								m.opts.Logger.Error(m.opts.Context, "broker ack error", err) | ||||||
|  | 							} | ||||||
|  | 						} | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryBroker) BatchSubscribe(ctx context.Context, topic string, handler broker.BatchHandler, opts ...broker.SubscribeOption) (broker.Subscriber, error) { | ||||||
|  | 	return m.funcBatchSubscribe(ctx, topic, handler, opts...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryBroker) fnBatchSubscribe(ctx context.Context, topic string, handler broker.BatchHandler, 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), | ||||||
|  | 		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), | ||||||
|  | 		id:      sid, | ||||||
|  | 		topic:   topic, | ||||||
|  | 		handler: 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) String() string { | ||||||
|  | 	return "memory" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryBroker) Name() string { | ||||||
|  | 	return m.opts.Name | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryBroker) Live() bool { | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryBroker) Ready() bool { | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryBroker) Health() bool { | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryEvent) Topic() string { | ||||||
|  | 	return m.topic | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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 | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewBroker return new memory broker | ||||||
|  | func NewBroker(opts ...broker.Option) broker.Broker { | ||||||
|  | 	return &memoryBroker{ | ||||||
|  | 		opts:        broker.NewOptions(opts...), | ||||||
|  | 		subscribers: make(map[string][]*memorySubscriber), | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -5,13 +5,18 @@ import ( | |||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"testing" | 	"testing" | ||||||
| 
 | 
 | ||||||
| 	"go.unistack.org/micro/v4/metadata" | 	"go.unistack.org/micro/v3/broker" | ||||||
|  | 	"go.unistack.org/micro/v3/metadata" | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
| func TestMemoryBatchBroker(t *testing.T) { | func TestMemoryBatchBroker(t *testing.T) { | ||||||
| 	b := NewBroker() | 	b := NewBroker() | ||||||
| 	ctx := context.Background() | 	ctx := context.Background() | ||||||
| 
 | 
 | ||||||
|  | 	if err := b.Init(); err != nil { | ||||||
|  | 		t.Fatalf("Unexpected init error %v", err) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
| 	if err := b.Connect(ctx); err != nil { | 	if err := b.Connect(ctx); err != nil { | ||||||
| 		t.Fatalf("Unexpected connect error %v", err) | 		t.Fatalf("Unexpected connect error %v", err) | ||||||
| 	} | 	} | ||||||
| @@ -19,35 +24,29 @@ func TestMemoryBatchBroker(t *testing.T) { | |||||||
| 	topic := "test" | 	topic := "test" | ||||||
| 	count := 10 | 	count := 10 | ||||||
| 
 | 
 | ||||||
| 	fn := func(evts []Message) error { | 	fn := func(evts broker.Events) error { | ||||||
| 		var err error | 		return evts.Ack() | ||||||
| 		for _, evt := range evts { |  | ||||||
| 			if err = evt.Ack(); err != nil { |  | ||||||
| 				return err |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 		return nil |  | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	sub, err := b.Subscribe(ctx, topic, fn) | 	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) | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	msgs := make([]Message, 0, count) | 	msgs := make([]*broker.Message, 0, count) | ||||||
| 	for i := 0; i < count; i++ { | 	for i := 0; i < count; i++ { | ||||||
| 		message := &memoryMessage{ | 		message := &broker.Message{ | ||||||
| 			header: map[string]string{ | 			Header: map[string]string{ | ||||||
| 				metadata.HeaderTopic: topic, | 				metadata.HeaderTopic: topic, | ||||||
| 				"foo":                "bar", | 				"foo":                "bar", | ||||||
| 				"id":                 fmt.Sprintf("%d", i), | 				"id":                 fmt.Sprintf("%d", i), | ||||||
| 			}, | 			}, | ||||||
| 			body: []byte(`"hello world"`), | 			Body: []byte(`"hello world"`), | ||||||
| 		} | 		} | ||||||
| 		msgs = append(msgs, message) | 		msgs = append(msgs, message) | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	if err := b.Publish(ctx, msgs); err != nil { | 	if err := b.BatchPublish(ctx, msgs); err != nil { | ||||||
| 		t.Fatalf("Unexpected error publishing %v", err) | 		t.Fatalf("Unexpected error publishing %v", err) | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| @@ -64,6 +63,10 @@ func TestMemoryBroker(t *testing.T) { | |||||||
| 	b := NewBroker() | 	b := NewBroker() | ||||||
| 	ctx := context.Background() | 	ctx := context.Background() | ||||||
| 
 | 
 | ||||||
|  | 	if err := b.Init(); err != nil { | ||||||
|  | 		t.Fatalf("Unexpected init error %v", err) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
| 	if err := b.Connect(ctx); err != nil { | 	if err := b.Connect(ctx); err != nil { | ||||||
| 		t.Fatalf("Unexpected connect error %v", err) | 		t.Fatalf("Unexpected connect error %v", err) | ||||||
| 	} | 	} | ||||||
| @@ -71,8 +74,8 @@ func TestMemoryBroker(t *testing.T) { | |||||||
| 	topic := "test" | 	topic := "test" | ||||||
| 	count := 10 | 	count := 10 | ||||||
| 
 | 
 | ||||||
| 	fn := func(p Message) error { | 	fn := func(_ broker.Event) error { | ||||||
| 		return p.Ack() | 		return nil | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	sub, err := b.Subscribe(ctx, topic, fn) | 	sub, err := b.Subscribe(ctx, topic, fn) | ||||||
| @@ -80,20 +83,24 @@ func TestMemoryBroker(t *testing.T) { | |||||||
| 		t.Fatalf("Unexpected error subscribing %v", err) | 		t.Fatalf("Unexpected error subscribing %v", err) | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	msgs := make([]Message, 0, count) | 	msgs := make([]*broker.Message, 0, count) | ||||||
| 	for i := 0; i < count; i++ { | 	for i := 0; i < count; i++ { | ||||||
| 		message := &memoryMessage{ | 		message := &broker.Message{ | ||||||
| 			header: map[string]string{ | 			Header: map[string]string{ | ||||||
| 				metadata.HeaderTopic: topic, | 				metadata.HeaderTopic: topic, | ||||||
| 				"foo":                "bar", | 				"foo":                "bar", | ||||||
| 				"id":                 fmt.Sprintf("%d", i), | 				"id":                 fmt.Sprintf("%d", i), | ||||||
| 			}, | 			}, | ||||||
| 			body: []byte(`"hello world"`), | 			Body: []byte(`"hello world"`), | ||||||
| 		} | 		} | ||||||
| 		msgs = append(msgs, message) | 		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.Publish(ctx, msgs); err != nil { | 	if err := b.BatchPublish(ctx, msgs); err != nil { | ||||||
| 		t.Fatalf("Unexpected error publishing %v", err) | 		t.Fatalf("Unexpected error publishing %v", err) | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
							
								
								
									
										140
									
								
								broker/noop.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										140
									
								
								broker/noop.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,140 @@ | |||||||
|  | package broker | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"strings" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v3/options" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type NoopBroker struct { | ||||||
|  | 	funcPublish        FuncPublish | ||||||
|  | 	funcBatchPublish   FuncBatchPublish | ||||||
|  | 	funcSubscribe      FuncSubscribe | ||||||
|  | 	funcBatchSubscribe FuncBatchSubscribe | ||||||
|  | 	opts               Options | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func NewBroker(opts ...Option) *NoopBroker { | ||||||
|  | 	b := &NoopBroker{opts: NewOptions(opts...)} | ||||||
|  | 	b.funcPublish = b.fnPublish | ||||||
|  | 	b.funcBatchPublish = b.fnBatchPublish | ||||||
|  | 	b.funcSubscribe = b.fnSubscribe | ||||||
|  | 	b.funcBatchSubscribe = b.fnBatchSubscribe | ||||||
|  |  | ||||||
|  | 	return b | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) Health() bool { | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) Live() bool { | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) Ready() bool { | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) Name() string { | ||||||
|  | 	return b.opts.Name | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) String() string { | ||||||
|  | 	return "noop" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) Options() Options { | ||||||
|  | 	return b.opts | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) Init(opts ...Option) error { | ||||||
|  | 	for _, opt := range opts { | ||||||
|  | 		opt(&b.opts) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	b.funcPublish = b.fnPublish | ||||||
|  | 	b.funcBatchPublish = b.fnBatchPublish | ||||||
|  | 	b.funcSubscribe = b.fnSubscribe | ||||||
|  | 	b.funcBatchSubscribe = b.fnBatchSubscribe | ||||||
|  |  | ||||||
|  | 	b.opts.Hooks.EachPrev(func(hook options.Hook) { | ||||||
|  | 		switch h := hook.(type) { | ||||||
|  | 		case HookPublish: | ||||||
|  | 			b.funcPublish = h(b.funcPublish) | ||||||
|  | 		case HookBatchPublish: | ||||||
|  | 			b.funcBatchPublish = h(b.funcBatchPublish) | ||||||
|  | 		case HookSubscribe: | ||||||
|  | 			b.funcSubscribe = h(b.funcSubscribe) | ||||||
|  | 		case HookBatchSubscribe: | ||||||
|  | 			b.funcBatchSubscribe = h(b.funcBatchSubscribe) | ||||||
|  | 		} | ||||||
|  | 	}) | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) Connect(_ context.Context) error { | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) Disconnect(_ context.Context) error { | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) Address() string { | ||||||
|  | 	return strings.Join(b.opts.Addrs, ",") | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) fnBatchPublish(_ context.Context, _ []*Message, _ ...PublishOption) error { | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) BatchPublish(ctx context.Context, msgs []*Message, opts ...PublishOption) error { | ||||||
|  | 	return b.funcBatchPublish(ctx, msgs, opts...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) fnPublish(_ context.Context, _ string, _ *Message, _ ...PublishOption) error { | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) Publish(ctx context.Context, topic string, msg *Message, opts ...PublishOption) error { | ||||||
|  | 	return b.funcPublish(ctx, topic, msg, opts...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type NoopSubscriber struct { | ||||||
|  | 	ctx          context.Context | ||||||
|  | 	topic        string | ||||||
|  | 	handler      Handler | ||||||
|  | 	batchHandler BatchHandler | ||||||
|  | 	opts         SubscribeOptions | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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 | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) Subscribe(ctx context.Context, topic string, handler Handler, opts ...SubscribeOption) (Subscriber, error) { | ||||||
|  | 	return b.funcSubscribe(ctx, topic, handler, opts...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *NoopSubscriber) Options() SubscribeOptions { | ||||||
|  | 	return s.opts | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *NoopSubscriber) Topic() string { | ||||||
|  | 	return s.topic | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *NoopSubscriber) Unsubscribe(_ context.Context) error { | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
							
								
								
									
										35
									
								
								broker/noop_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										35
									
								
								broker/noop_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,35 @@ | |||||||
|  | package broker | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"testing" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type testHook struct { | ||||||
|  | 	f bool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *testHook) Publish1(fn FuncPublish) FuncPublish { | ||||||
|  | 	return func(ctx context.Context, topic string, msg *Message, opts ...PublishOption) error { | ||||||
|  | 		t.f = true | ||||||
|  | 		return fn(ctx, topic, msg, opts...) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestNoopHook(t *testing.T) { | ||||||
|  | 	h := &testHook{} | ||||||
|  |  | ||||||
|  | 	b := NewBroker(Hooks(HookPublish(h.Publish1))) | ||||||
|  |  | ||||||
|  | 	if err := b.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if err := b.Publish(context.TODO(), "", nil); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !h.f { | ||||||
|  | 		t.Fatal("hook not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -5,48 +5,63 @@ 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/metadata" | 	"go.unistack.org/micro/v3/meter" | ||||||
| 	"go.unistack.org/micro/v4/meter" | 	"go.unistack.org/micro/v3/options" | ||||||
| 	"go.unistack.org/micro/v4/options" | 	"go.unistack.org/micro/v3/register" | ||||||
| 	"go.unistack.org/micro/v4/register" | 	"go.unistack.org/micro/v3/sync" | ||||||
| 	"go.unistack.org/micro/v4/tracer" | 	"go.unistack.org/micro/v3/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Options struct | // Options struct | ||||||
| type Options struct { | type Options struct { | ||||||
|  | 	// Name holds the broker name | ||||||
|  | 	Name string | ||||||
|  |  | ||||||
| 	// Tracer used for tracing | 	// Tracer used for tracing | ||||||
| 	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 codec for marshal/unmarshal | 	// 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 | ||||||
| 	Meter meter.Meter | 	Meter meter.Meter | ||||||
| 	// Context holds external options | 	// Context holds external options | ||||||
| 	Context context.Context | 	Context context.Context | ||||||
|  |  | ||||||
|  | 	// Wait waits for a collection of goroutines to finish | ||||||
|  | 	Wait *sync.WaitGroup | ||||||
| 	// TLSConfig holds tls.TLSConfig options | 	// TLSConfig holds tls.TLSConfig options | ||||||
| 	TLSConfig *tls.Config | 	TLSConfig *tls.Config | ||||||
| 	// ErrorHandler used when broker have error while processing message |  | ||||||
| 	ErrorHandler interface{} | 	// ErrorHandler used when broker can't unmarshal incoming message | ||||||
| 	// Name holds the broker name | 	ErrorHandler Handler | ||||||
| 	Name string | 	// BatchErrorHandler used when broker can't unmashal incoming messages | ||||||
| 	// Address holds the broker address | 	BatchErrorHandler BatchHandler | ||||||
| 	Address []string |  | ||||||
|  | 	// Addrs holds the broker address | ||||||
|  | 	Addrs []string | ||||||
|  | 	// Hooks can be run before broker Publish/BatchPublish and | ||||||
|  | 	// Subscribe/BatchSubscribe methods | ||||||
|  | 	Hooks options.Hooks | ||||||
|  |  | ||||||
|  | 	// GracefulTimeout contains time to wait to finish in flight requests | ||||||
|  | 	GracefulTimeout time.Duration | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewOptions create new Options | // NewOptions create new Options | ||||||
| func NewOptions(opts ...options.Option) Options { | func NewOptions(opts ...Option) Options { | ||||||
| 	options := Options{ | 	options := Options{ | ||||||
| 		Register:        register.DefaultRegister, | 		Register:        register.DefaultRegister, | ||||||
| 		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, | ||||||
| 	} | 	} | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&options) | 		o(&options) | ||||||
| @@ -54,22 +69,23 @@ func NewOptions(opts ...options.Option) Options { | |||||||
| 	return options | 	return options | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // Context sets the context option | ||||||
|  | func Context(ctx context.Context) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Context = ctx | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // PublishOptions struct | // PublishOptions struct | ||||||
| type PublishOptions struct { | type PublishOptions struct { | ||||||
| 	// Context holds external options | 	// Context holds external options | ||||||
| 	Context context.Context | 	Context context.Context | ||||||
| 	// BodyOnly flag says the message contains raw body bytes | 	// BodyOnly flag says the message contains raw body bytes | ||||||
| 	BodyOnly bool | 	BodyOnly bool | ||||||
| 	// Message metadata usually passed as message headers |  | ||||||
| 	Metadata metadata.Metadata |  | ||||||
| 	// Content-Type of message for marshal |  | ||||||
| 	ContentType string |  | ||||||
| 	// Topic destination |  | ||||||
| 	Topic string |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewPublishOptions creates PublishOptions struct | // NewPublishOptions creates PublishOptions struct | ||||||
| func NewPublishOptions(opts ...options.Option) PublishOptions { | func NewPublishOptions(opts ...PublishOption) PublishOptions { | ||||||
| 	options := PublishOptions{ | 	options := PublishOptions{ | ||||||
| 		Context: context.Background(), | 		Context: context.Background(), | ||||||
| 	} | 	} | ||||||
| @@ -79,21 +95,16 @@ func NewPublishOptions(opts ...options.Option) PublishOptions { | |||||||
| 	return options | 	return options | ||||||
| } | } | ||||||
|  |  | ||||||
| // PublishTopic pass topic for messages |  | ||||||
| func PublishTopic(t string) options.Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return options.Set(src, t, ".Topic") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // SubscribeOptions struct | // SubscribeOptions struct | ||||||
| type SubscribeOptions struct { | type SubscribeOptions struct { | ||||||
| 	// Context holds external options | 	// Context holds external options | ||||||
| 	Context context.Context | 	Context context.Context | ||||||
| 	// ErrorHandler used when broker have error while processing message | 	// ErrorHandler used when broker can't unmarshal incoming message | ||||||
| 	ErrorHandler interface{} | 	ErrorHandler Handler | ||||||
| 	// QueueGroup holds consumer group | 	// BatchErrorHandler used when broker can't unmashal incoming messages | ||||||
| 	QueueGroup string | 	BatchErrorHandler BatchHandler | ||||||
|  | 	// Group holds consumer group | ||||||
|  | 	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 | ||||||
| 	AutoAck bool | 	AutoAck bool | ||||||
| 	// BodyOnly flag specifies that message contains only body bytes without header | 	// BodyOnly flag specifies that message contains only body bytes without header | ||||||
| @@ -104,16 +115,186 @@ type SubscribeOptions struct { | |||||||
| 	BatchWait time.Duration | 	BatchWait time.Duration | ||||||
| } | } | ||||||
|  |  | ||||||
| // ErrorHandler will catch all broker errors that cant be handled | // Option func | ||||||
| // in normal way, for example Codec errors | type Option func(*Options) | ||||||
| func ErrorHandler(h interface{}) options.Option { |  | ||||||
| 	return func(src interface{}) error { | // PublishOption func | ||||||
| 		return options.Set(src, h, ".ErrorHandler") | type PublishOption func(*PublishOptions) | ||||||
|  |  | ||||||
|  | // PublishBodyOnly publish only body of the message | ||||||
|  | func PublishBodyOnly(b bool) PublishOption { | ||||||
|  | 	return func(o *PublishOptions) { | ||||||
|  | 		o.BodyOnly = b | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // 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 | ||||||
|  | func Addrs(addrs ...string) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Addrs = addrs | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Codec sets the codec used for encoding/decoding used where | ||||||
|  | // a broker does not support headers | ||||||
|  | func Codec(c codec.Codec) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		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 | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SubscribeGroup sets the name of the queue to share messages on | ||||||
|  | func SubscribeGroup(name string) SubscribeOption { | ||||||
|  | 	return func(o *SubscribeOptions) { | ||||||
|  | 		o.Group = name | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Register sets register option | ||||||
|  | func Register(r register.Register) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Register = r | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // TLSConfig sets the TLS Config | ||||||
|  | func TLSConfig(t *tls.Config) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.TLSConfig = t | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Logger sets the logger | ||||||
|  | func Logger(l logger.Logger) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Logger = l | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Tracer to be used for tracing | ||||||
|  | func Tracer(t tracer.Tracer) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Tracer = t | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Meter sets the meter | ||||||
|  | func Meter(m meter.Meter) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Meter = m | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Name sets the name | ||||||
|  | func Name(n string) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Name = n | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Hooks sets hook runs before action | ||||||
|  | func Hooks(h ...options.Hook) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Hooks = append(o.Hooks, h...) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SubscribeContext set context | ||||||
|  | func SubscribeContext(ctx context.Context) SubscribeOption { | ||||||
|  | 	return func(o *SubscribeOptions) { | ||||||
|  | 		o.Context = ctx | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // DisableAutoAck disables auto ack | ||||||
|  | // Deprecated | ||||||
|  | func DisableAutoAck() SubscribeOption { | ||||||
|  | 	return func(o *SubscribeOptions) { | ||||||
|  | 		o.AutoAck = false | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SubscribeAutoAck contol auto acking of messages | ||||||
|  | // after they have been handled. | ||||||
|  | func SubscribeAutoAck(b bool) SubscribeOption { | ||||||
|  | 	return func(o *SubscribeOptions) { | ||||||
|  | 		o.AutoAck = b | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SubscribeBodyOnly consumes only body of the message | ||||||
|  | func SubscribeBodyOnly(b bool) SubscribeOption { | ||||||
|  | 	return func(o *SubscribeOptions) { | ||||||
|  | 		o.BodyOnly = b | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SubscribeBatchSize specifies max batch size | ||||||
|  | func SubscribeBatchSize(n int) SubscribeOption { | ||||||
|  | 	return func(o *SubscribeOptions) { | ||||||
|  | 		o.BatchSize = n | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SubscribeBatchWait specifies max batch wait time | ||||||
|  | func SubscribeBatchWait(td time.Duration) SubscribeOption { | ||||||
|  | 	return func(o *SubscribeOptions) { | ||||||
|  | 		o.BatchWait = td | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SubscribeOption func | ||||||
|  | type SubscribeOption func(*SubscribeOptions) | ||||||
|  |  | ||||||
| // NewSubscribeOptions creates new SubscribeOptions | // NewSubscribeOptions creates new SubscribeOptions | ||||||
| func NewSubscribeOptions(opts ...options.Option) SubscribeOptions { | func NewSubscribeOptions(opts ...SubscribeOption) SubscribeOptions { | ||||||
| 	options := SubscribeOptions{ | 	options := SubscribeOptions{ | ||||||
| 		AutoAck: true, | 		AutoAck: true, | ||||||
| 		Context: context.Background(), | 		Context: context.Background(), | ||||||
| @@ -123,39 +304,3 @@ func NewSubscribeOptions(opts ...options.Option) SubscribeOptions { | |||||||
| 	} | 	} | ||||||
| 	return options | 	return options | ||||||
| } | } | ||||||
|  |  | ||||||
| // SubscribeAutoAck contol auto acking of messages |  | ||||||
| // after they have been handled. |  | ||||||
| func SubscribeAutoAck(b bool) options.Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return options.Set(src, b, ".AutoAck") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // BodyOnly transfer only body without |  | ||||||
| func BodyOnly(b bool) options.Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return options.Set(src, b, ".BodyOnly") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // SubscribeBatchSize specifies max batch size |  | ||||||
| func SubscribeBatchSize(n int) options.Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return options.Set(src, n, ".BatchSize") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // SubscribeBatchWait specifies max batch wait time |  | ||||||
| func SubscribeBatchWait(td time.Duration) options.Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return options.Set(src, td, ".BatchWait") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // SubscribeQueueGroup sets the shared queue name distributed messages across subscribers |  | ||||||
| func SubscribeQueueGroup(n string) options.Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return options.Set(src, n, ".QueueGroup") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|   | |||||||
| @@ -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 | ||||||
| @@ -17,13 +17,13 @@ func BackoffExp(_ context.Context, _ Request, attempts int) (time.Duration, erro | |||||||
| } | } | ||||||
|  |  | ||||||
| // BackoffInterval specifies randomization interval for backoff func | // BackoffInterval specifies randomization interval for backoff func | ||||||
| func BackoffInterval(min time.Duration, max time.Duration) BackoffFunc { | func BackoffInterval(minTime time.Duration, maxTime time.Duration) BackoffFunc { | ||||||
| 	return func(_ context.Context, _ Request, attempts int) (time.Duration, error) { | 	return func(_ context.Context, _ Request, attempts int) (time.Duration, error) { | ||||||
| 		td := time.Duration(math.Pow(float64(attempts), math.E)) * time.Millisecond * 100 | 		td := time.Duration(math.Pow(float64(attempts), math.E)) * time.Millisecond * 100 | ||||||
| 		if td < min { | 		if td < minTime { | ||||||
| 			return min, nil | 			return minTime, nil | ||||||
| 		} else if td > max { | 		} else if td > maxTime { | ||||||
| 			return max, nil | 			return maxTime, nil | ||||||
| 		} | 		} | ||||||
| 		return td, nil | 		return td, nil | ||||||
| 	} | 	} | ||||||
|   | |||||||
| @@ -34,23 +34,23 @@ func TestBackoffExp(t *testing.T) { | |||||||
| } | } | ||||||
|  |  | ||||||
| func TestBackoffInterval(t *testing.T) { | func TestBackoffInterval(t *testing.T) { | ||||||
| 	min := 100 * time.Millisecond | 	minTime := 100 * time.Millisecond | ||||||
| 	max := 300 * time.Millisecond | 	maxTime := 300 * time.Millisecond | ||||||
|  |  | ||||||
| 	r := &testRequest{ | 	r := &testRequest{ | ||||||
| 		service: "test", | 		service: "test", | ||||||
| 		method:  "test", | 		method:  "test", | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	fn := BackoffInterval(min, max) | 	fn := BackoffInterval(minTime, maxTime) | ||||||
| 	for i := 0; i < 5; i++ { | 	for i := 0; i < 5; i++ { | ||||||
| 		d, err := fn(context.TODO(), r, i) | 		d, err := fn(context.TODO(), r, i) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			t.Fatal(err) | 			t.Fatal(err) | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		if d < min || d > max { | 		if d < minTime || d > maxTime { | ||||||
| 			t.Fatalf("Expected %v < %v < %v", min, d, max) | 			t.Fatalf("Expected %v < %v < %v", minTime, d, maxTime) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,12 +1,12 @@ | |||||||
| // Package client is an interface for an RPC client | // Package client is an interface for an RPC client | ||||||
| package client // import "go.unistack.org/micro/v4/client" | package client | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/codec" | 	"go.unistack.org/micro/v3/codec" | ||||||
| 	"go.unistack.org/micro/v4/options" | 	"go.unistack.org/micro/v3/metadata" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| @@ -22,8 +22,6 @@ var ( | |||||||
| 	DefaultRetries = 0 | 	DefaultRetries = 0 | ||||||
| 	// DefaultRequestTimeout is the default request timeout | 	// DefaultRequestTimeout is the default request timeout | ||||||
| 	DefaultRequestTimeout = time.Second * 5 | 	DefaultRequestTimeout = time.Second * 5 | ||||||
| 	// DefaultDialTimeout the default dial timeout |  | ||||||
| 	DefaultDialTimeout = time.Second * 5 |  | ||||||
| 	// DefaultPoolSize sets the connection pool size | 	// DefaultPoolSize sets the connection pool size | ||||||
| 	DefaultPoolSize = 100 | 	DefaultPoolSize = 100 | ||||||
| 	// DefaultPoolTTL sets the connection pool ttl | 	// DefaultPoolTTL sets the connection pool ttl | ||||||
| @@ -35,14 +33,36 @@ var ( | |||||||
| // 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 ...options.Option) error | 	Init(opts ...Option) error | ||||||
| 	Options() Options | 	Options() Options | ||||||
| 	NewRequest(service string, endpoint string, req interface{}, opts ...options.Option) Request | 	NewMessage(topic string, msg interface{}, opts ...MessageOption) Message | ||||||
| 	Call(ctx context.Context, req Request, rsp interface{}, opts ...options.Option) error | 	NewRequest(service string, endpoint string, req interface{}, opts ...RequestOption) Request | ||||||
| 	Stream(ctx context.Context, req Request, opts ...options.Option) (Stream, error) | 	Call(ctx context.Context, req Request, rsp interface{}, opts ...CallOption) 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 | ||||||
| } | } | ||||||
|  |  | ||||||
|  | type ( | ||||||
|  | 	FuncCall         func(ctx context.Context, req Request, rsp interface{}, opts ...CallOption) error | ||||||
|  | 	HookCall         func(next FuncCall) FuncCall | ||||||
|  | 	FuncStream       func(ctx context.Context, req Request, opts ...CallOption) (Stream, error) | ||||||
|  | 	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 | ||||||
| @@ -59,22 +79,16 @@ type Request interface { | |||||||
| 	Codec() codec.Codec | 	Codec() codec.Codec | ||||||
| 	// indicates whether the request will be a streaming one rather than unary | 	// indicates whether the request will be a streaming one rather than unary | ||||||
| 	Stream() bool | 	Stream() bool | ||||||
| 	// Header data |  | ||||||
| 	// Header() metadata.Metadata |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // Response is the response received from a service | // Response is the response received from a service | ||||||
| type Response interface { | type Response interface { | ||||||
| 	// Read the response | 	// Read the response | ||||||
| 	Codec() codec.Codec | 	Codec() codec.Codec | ||||||
| 	// The content type |  | ||||||
| 	// ContentType() string |  | ||||||
| 	// Header data | 	// Header data | ||||||
| 	// Header() metadata.Metadata | 	Header() metadata.Metadata | ||||||
| 	// Read the undecoded response | 	// Read the undecoded response | ||||||
| 	Read() ([]byte, error) | 	Read() ([]byte, error) | ||||||
| 	// The unencoded request body |  | ||||||
| 	// Body() interface{} |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // Stream is the interface for a bidirectional synchronous stream | // Stream is the interface for a bidirectional synchronous stream | ||||||
| @@ -100,3 +114,18 @@ type Stream interface { | |||||||
| 	// CloseSend closes the send direction of the stream | 	// CloseSend closes the send direction of the stream | ||||||
| 	CloseSend() error | 	CloseSend() error | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // Option used by the Client | ||||||
|  | type Option func(*Options) | ||||||
|  |  | ||||||
|  | // CallOption used by Call or Stream | ||||||
|  | type CallOption func(*CallOptions) | ||||||
|  |  | ||||||
|  | // PublishOption used by Publish | ||||||
|  | type PublishOption func(*PublishOptions) | ||||||
|  |  | ||||||
|  | // MessageOption used by NewMessage | ||||||
|  | type MessageOption func(*MessageOptions) | ||||||
|  |  | ||||||
|  | // RequestOption used by NewRequest | ||||||
|  | type RequestOption func(*RequestOptions) | ||||||
|   | |||||||
| @@ -2,24 +2,22 @@ package client | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/options" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type clientCallOptions struct { | type clientCallOptions struct { | ||||||
| 	Client | 	Client | ||||||
| 	opts []options.Option | 	opts []CallOption | ||||||
| } | } | ||||||
|  |  | ||||||
| func (s *clientCallOptions) Call(ctx context.Context, req Request, rsp interface{}, opts ...options.Option) error { | func (s *clientCallOptions) Call(ctx context.Context, req Request, rsp interface{}, opts ...CallOption) error { | ||||||
| 	return s.Client.Call(ctx, req, rsp, append(s.opts, opts...)...) | 	return s.Client.Call(ctx, req, rsp, append(s.opts, opts...)...) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (s *clientCallOptions) Stream(ctx context.Context, req Request, opts ...options.Option) (Stream, error) { | func (s *clientCallOptions) Stream(ctx context.Context, req Request, opts ...CallOption) (Stream, error) { | ||||||
| 	return s.Client.Stream(ctx, req, append(s.opts, opts...)...) | 	return s.Client.Stream(ctx, req, append(s.opts, opts...)...) | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewClientCallOptions add CallOption to every call | // NewClientCallOptions add CallOption to every call | ||||||
| func NewClientCallOptions(c Client, opts ...options.Option) Client { | func NewClientCallOptions(c Client, opts ...CallOption) Client { | ||||||
| 	return &clientCallOptions{c, opts} | 	return &clientCallOptions{c, opts} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,28 +0,0 @@ | |||||||
| package client |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"context" |  | ||||||
| 	"testing" |  | ||||||
| 	"time" |  | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/options" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| func TestNewClientCallOptions(t *testing.T) { |  | ||||||
| 	var flag bool |  | ||||||
| 	w := func(fn CallFunc) CallFunc { |  | ||||||
| 		flag = true |  | ||||||
| 		return fn |  | ||||||
| 	} |  | ||||||
| 	c := NewClientCallOptions(NewClient(), |  | ||||||
| 		options.Address("127.0.0.1"), |  | ||||||
| 		WithCallWrapper(w), |  | ||||||
| 		RequestTimeout(1*time.Millisecond), |  | ||||||
| 		Retries(0), |  | ||||||
| 		Backoff(BackoffInterval(10*time.Millisecond, 100*time.Millisecond)), |  | ||||||
| 	) |  | ||||||
| 	_ = c.Call(context.TODO(), c.NewRequest("service", "endpoint", nil), nil) |  | ||||||
| 	if !flag { |  | ||||||
| 		t.Fatalf("NewClientCallOptions not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -15,6 +15,15 @@ func FromContext(ctx context.Context) (Client, bool) { | |||||||
| 	return c, ok | 	return c, ok | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // MustContext get client from context | ||||||
|  | func MustContext(ctx context.Context) Client { | ||||||
|  | 	c, ok := FromContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		panic("missing client") | ||||||
|  | 	} | ||||||
|  | 	return c | ||||||
|  | } | ||||||
|  |  | ||||||
| // NewContext put client in context | // NewContext put client in context | ||||||
| func NewContext(ctx context.Context, c Client) context.Context { | func NewContext(ctx context.Context, c Client) context.Context { | ||||||
| 	if ctx == nil { | 	if ctx == nil { | ||||||
| @@ -22,3 +31,33 @@ 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 | ||||||
|  | func SetCallOption(k, v interface{}) CallOption { | ||||||
|  | 	return func(o *CallOptions) { | ||||||
|  | 		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 | ||||||
|  | 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) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -38,3 +38,36 @@ func TestNewNilContext(t *testing.T) { | |||||||
| 		t.Fatal("NewContext not works") | 		t.Fatal("NewContext not works") | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | 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) { | ||||||
|  | 	type key struct{} | ||||||
|  | 	o := SetCallOption(key{}, "test") | ||||||
|  | 	opts := &CallOptions{} | ||||||
|  | 	o(opts) | ||||||
|  |  | ||||||
|  | 	if v, ok := opts.Context.Value(key{}).(string); !ok || v == "" { | ||||||
|  | 		t.Fatal("SetCallOption not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestSetOption(t *testing.T) { | ||||||
|  | 	type key struct{} | ||||||
|  | 	o := SetOption(key{}, "test") | ||||||
|  | 	opts := &Options{} | ||||||
|  | 	o(opts) | ||||||
|  |  | ||||||
|  | 	if v, ok := opts.Context.Value(key{}).(string); !ok || v == "" { | ||||||
|  | 		t.Fatal("SetOption not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -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 | ||||||
|   | |||||||
							
								
								
									
										289
									
								
								client/noop.go
									
									
									
									
									
								
							
							
						
						
									
										289
									
								
								client/noop.go
									
									
									
									
									
								
							| @@ -3,14 +3,18 @@ package client | |||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"fmt" | 	"fmt" | ||||||
|  | 	"os" | ||||||
|  | 	"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/v3/semconv" | ||||||
|  | 	"go.unistack.org/micro/v3/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // DefaultCodecs will be used to encode/decode data | // DefaultCodecs will be used to encode/decode data | ||||||
| @@ -19,9 +23,19 @@ var DefaultCodecs = map[string]codec.Codec{ | |||||||
| } | } | ||||||
|  |  | ||||||
| type noopClient struct { | type noopClient struct { | ||||||
|  | 	funcPublish      FuncPublish | ||||||
|  | 	funcBatchPublish FuncBatchPublish | ||||||
|  | 	funcCall         FuncCall | ||||||
|  | 	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 | ||||||
| @@ -33,13 +47,15 @@ type noopRequest struct { | |||||||
| } | } | ||||||
|  |  | ||||||
| // NewClient returns new noop client | // NewClient returns new noop client | ||||||
| func NewClient(opts ...options.Option) Client { | func NewClient(opts ...Option) Client { | ||||||
| 	nc := &noopClient{opts: NewOptions(opts...)} | 	n := &noopClient{opts: NewOptions(opts...)} | ||||||
| 	// wrap in reverse |  | ||||||
|  |  | ||||||
| 	c := Client(nc) | 	n.funcCall = n.fnCall | ||||||
|  | 	n.funcStream = n.fnStream | ||||||
|  | 	n.funcPublish = n.fnPublish | ||||||
|  | 	n.funcBatchPublish = n.fnBatchPublish | ||||||
|  |  | ||||||
| 	return c | 	return n | ||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopClient) Name() string { | func (n *noopClient) Name() string { | ||||||
| @@ -91,10 +107,13 @@ func (n *noopResponse) Read() ([]byte, error) { | |||||||
| 	return nil, nil | 	return nil, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| type noopStream struct{} | type noopStream struct { | ||||||
|  | 	err error | ||||||
|  | 	ctx context.Context | ||||||
|  | } | ||||||
|  |  | ||||||
| func (n *noopStream) Context() context.Context { | func (n *noopStream) Context() context.Context { | ||||||
| 	return context.Background() | 	return n.ctx | ||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopStream) Request() Request { | func (n *noopStream) Request() Request { | ||||||
| @@ -122,15 +141,37 @@ func (n *noopStream) RecvMsg(interface{}) error { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopStream) Error() error { | func (n *noopStream) Error() error { | ||||||
| 	return nil | 	return n.err | ||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopStream) Close() error { | func (n *noopStream) Close() error { | ||||||
| 	return nil | 	if sp, ok := tracer.SpanFromContext(n.ctx); ok && sp != nil { | ||||||
|  | 		if n.err != nil { | ||||||
|  | 			sp.SetStatus(tracer.SpanStatusError, n.err.Error()) | ||||||
|  | 		} | ||||||
|  | 		sp.Finish() | ||||||
|  | 	} | ||||||
|  | 	return n.err | ||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopStream) CloseSend() error { | func (n *noopStream) CloseSend() error { | ||||||
| 	return nil | 	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) { | func (n *noopClient) newCodec(contentType string) (codec.Codec, error) { | ||||||
| @@ -143,10 +184,29 @@ func (n *noopClient) newCodec(contentType string) (codec.Codec, error) { | |||||||
| 	return nil, codec.ErrUnknownContentType | 	return nil, codec.ErrUnknownContentType | ||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopClient) Init(opts ...options.Option) error { | func (n *noopClient) Init(opts ...Option) error { | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&n.opts) | 		o(&n.opts) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	n.funcCall = n.fnCall | ||||||
|  | 	n.funcStream = n.fnStream | ||||||
|  | 	n.funcPublish = n.fnPublish | ||||||
|  | 	n.funcBatchPublish = n.fnBatchPublish | ||||||
|  |  | ||||||
|  | 	n.opts.Hooks.EachPrev(func(hook options.Hook) { | ||||||
|  | 		switch h := hook.(type) { | ||||||
|  | 		case HookCall: | ||||||
|  | 			n.funcCall = h(n.funcCall) | ||||||
|  | 		case HookStream: | ||||||
|  | 			n.funcStream = h(n.funcStream) | ||||||
|  | 		case HookPublish: | ||||||
|  | 			n.funcPublish = h(n.funcPublish) | ||||||
|  | 		case HookBatchPublish: | ||||||
|  | 			n.funcBatchPublish = h(n.funcBatchPublish) | ||||||
|  | 		} | ||||||
|  | 	}) | ||||||
|  |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -158,7 +218,32 @@ func (n *noopClient) String() string { | |||||||
| 	return "noop" | 	return "noop" | ||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopClient) Call(ctx context.Context, req Request, rsp interface{}, opts ...options.Option) error { | func (n *noopClient) Call(ctx context.Context, req Request, rsp interface{}, opts ...CallOption) error { | ||||||
|  | 	ts := time.Now() | ||||||
|  | 	n.opts.Meter.Counter(semconv.ClientRequestInflight, "endpoint", req.Endpoint()).Inc() | ||||||
|  | 	var sp tracer.Span | ||||||
|  | 	ctx, sp = n.opts.Tracer.Start(ctx, "rpc-client", | ||||||
|  | 		tracer.WithSpanKind(tracer.SpanKindClient), | ||||||
|  | 		tracer.WithSpanLabels("endpoint", req.Endpoint()), | ||||||
|  | 	) | ||||||
|  | 	err := n.funcCall(ctx, req, rsp, opts...) | ||||||
|  | 	n.opts.Meter.Counter(semconv.ClientRequestInflight, "endpoint", req.Endpoint()).Dec() | ||||||
|  | 	te := time.Since(ts) | ||||||
|  | 	n.opts.Meter.Summary(semconv.ClientRequestLatencyMicroseconds, "endpoint", req.Endpoint()).Update(te.Seconds()) | ||||||
|  | 	n.opts.Meter.Histogram(semconv.ClientRequestDurationSeconds, "endpoint", req.Endpoint()).Update(te.Seconds()) | ||||||
|  |  | ||||||
|  | 	if me := errors.FromError(err); me == nil { | ||||||
|  | 		sp.Finish() | ||||||
|  | 		n.opts.Meter.Counter(semconv.ClientRequestTotal, "endpoint", req.Endpoint(), "status", "success", "code", strconv.Itoa(int(200))).Inc() | ||||||
|  | 	} else { | ||||||
|  | 		sp.SetStatus(tracer.SpanStatusError, err.Error()) | ||||||
|  | 		n.opts.Meter.Counter(semconv.ClientRequestTotal, "endpoint", req.Endpoint(), "status", "failure", "code", strconv.Itoa(int(me.Code))).Inc() | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return err | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (n *noopClient) fnCall(ctx context.Context, req Request, rsp interface{}, opts ...CallOption) error { | ||||||
| 	// make a copy of call opts | 	// make a copy of call opts | ||||||
| 	callOpts := n.opts.CallOptions | 	callOpts := n.opts.CallOptions | ||||||
| 	for _, opt := range opts { | 	for _, opt := range opts { | ||||||
| @@ -175,7 +260,7 @@ func (n *noopClient) Call(ctx context.Context, req Request, rsp interface{}, opt | |||||||
| 	} else { | 	} else { | ||||||
| 		// got a deadline so no need to setup context | 		// got a deadline so no need to setup context | ||||||
| 		// but we need to set the timeout we pass along | 		// but we need to set the timeout we pass along | ||||||
| 		opt := RequestTimeout(time.Until(d)) | 		opt := WithRequestTimeout(time.Until(d)) | ||||||
| 		opt(&callOpts) | 		opt(&callOpts) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -187,11 +272,8 @@ func (n *noopClient) Call(ctx context.Context, req Request, rsp interface{}, opt | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// make copy of call method | 	// make copy of call method | ||||||
| 	hcall := n.call | 	hcall := func(ctx context.Context, addr string, req Request, rsp interface{}, opts CallOptions) error { | ||||||
|  | 		return nil | ||||||
| 	// wrap the call in reverse |  | ||||||
| 	for i := len(callOpts.CallWrappers); i > 0; i-- { |  | ||||||
| 		hcall = callOpts.CallWrappers[i-1](hcall) |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// use the router passed as a call option, or fallback to the rpc clients router | 	// use the router passed as a call option, or fallback to the rpc clients router | ||||||
| @@ -216,7 +298,7 @@ func (n *noopClient) Call(ctx context.Context, req Request, rsp interface{}, opt | |||||||
| 		// call backoff first. Someone may want an initial start delay | 		// call backoff first. Someone may want an initial start delay | ||||||
| 		t, err := callOpts.Backoff(ctx, req, i) | 		t, err := callOpts.Backoff(ctx, req, i) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			return errors.InternalServerError("go.micro.client", err.Error()) | 			return errors.InternalServerError("go.micro.client", "%s", err) | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		// only sleep if greater than 0 | 		// only sleep if greater than 0 | ||||||
| @@ -230,7 +312,7 @@ func (n *noopClient) Call(ctx context.Context, req Request, rsp interface{}, opt | |||||||
| 			// TODO apply any filtering here | 			// TODO apply any filtering here | ||||||
| 			routes, err = n.opts.Lookup(ctx, req, callOpts) | 			routes, err = n.opts.Lookup(ctx, req, callOpts) | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| 				return errors.InternalServerError("go.micro.client", err.Error()) | 				return errors.InternalServerError("go.micro.client", "%s", err) | ||||||
| 			} | 			} | ||||||
|  |  | ||||||
| 			// balance the list of nodes | 			// balance the list of nodes | ||||||
| @@ -260,9 +342,6 @@ func (n *noopClient) Call(ctx context.Context, req Request, rsp interface{}, opt | |||||||
| 	ch := make(chan error, callOpts.Retries) | 	ch := make(chan error, callOpts.Retries) | ||||||
| 	var gerr error | 	var gerr error | ||||||
|  |  | ||||||
| 	ts := time.Now() |  | ||||||
| 	endpoint := fmt.Sprintf("%s.%s", req.Service(), req.Endpoint()) |  | ||||||
| 	n.opts.Meter.Counter(semconv.ClientRequestInflight, "endpoint", endpoint).Inc() |  | ||||||
| 	for i := 0; i <= callOpts.Retries; i++ { | 	for i := 0; i <= callOpts.Retries; i++ { | ||||||
| 		go func() { | 		go func() { | ||||||
| 			ch <- call(i) | 			ch <- call(i) | ||||||
| @@ -290,28 +369,44 @@ func (n *noopClient) Call(ctx context.Context, req Request, rsp interface{}, opt | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if gerr != nil { |  | ||||||
| 		n.opts.Meter.Counter(semconv.ClientRequestTotal, "endpoint", endpoint, "status", "failure").Inc() |  | ||||||
| 	} else { |  | ||||||
| 		n.opts.Meter.Counter(semconv.ClientRequestTotal, "endpoint", endpoint, "status", "success").Inc() |  | ||||||
| 	} |  | ||||||
| 	n.opts.Meter.Counter(semconv.ClientRequestInflight, "endpoint", endpoint).Dec() |  | ||||||
| 	te := time.Since(ts) |  | ||||||
| 	n.opts.Meter.Summary(semconv.ClientRequestLatencyMicroseconds, "endpoint", endpoint).Update(te.Seconds()) |  | ||||||
| 	n.opts.Meter.Histogram(semconv.ClientRequestDurationSeconds, "endpoint", endpoint).Update(te.Seconds()) |  | ||||||
|  |  | ||||||
| 	return gerr | 	return gerr | ||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopClient) call(ctx context.Context, addr string, req Request, rsp interface{}, opts CallOptions) error { | func (n *noopClient) NewRequest(service, endpoint string, _ interface{}, _ ...RequestOption) Request { | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (n *noopClient) NewRequest(service, endpoint string, req interface{}, opts ...options.Option) Request { |  | ||||||
| 	return &noopRequest{service: service, endpoint: endpoint} | 	return &noopRequest{service: service, endpoint: endpoint} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopClient) Stream(ctx context.Context, req Request, opts ...options.Option) (Stream, error) { | 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) { | ||||||
|  | 	ts := time.Now() | ||||||
|  | 	n.opts.Meter.Counter(semconv.ClientRequestInflight, "endpoint", req.Endpoint()).Inc() | ||||||
|  | 	var sp tracer.Span | ||||||
|  | 	ctx, sp = n.opts.Tracer.Start(ctx, "rpc-client", | ||||||
|  | 		tracer.WithSpanKind(tracer.SpanKindClient), | ||||||
|  | 		tracer.WithSpanLabels("endpoint", req.Endpoint()), | ||||||
|  | 	) | ||||||
|  | 	stream, err := n.funcStream(ctx, req, opts...) | ||||||
|  | 	n.opts.Meter.Counter(semconv.ClientRequestInflight, "endpoint", req.Endpoint()).Dec() | ||||||
|  | 	te := time.Since(ts) | ||||||
|  | 	n.opts.Meter.Summary(semconv.ClientRequestLatencyMicroseconds, "endpoint", req.Endpoint()).Update(te.Seconds()) | ||||||
|  | 	n.opts.Meter.Histogram(semconv.ClientRequestDurationSeconds, "endpoint", req.Endpoint()).Update(te.Seconds()) | ||||||
|  |  | ||||||
|  | 	if me := errors.FromError(err); me == nil { | ||||||
|  | 		sp.Finish() | ||||||
|  | 		n.opts.Meter.Counter(semconv.ClientRequestTotal, "endpoint", req.Endpoint(), "status", "success", "code", strconv.Itoa(int(200))).Inc() | ||||||
|  | 	} else { | ||||||
|  | 		sp.SetStatus(tracer.SpanStatusError, err.Error()) | ||||||
|  | 		n.opts.Meter.Counter(semconv.ClientRequestTotal, "endpoint", req.Endpoint(), "status", "failure", "code", strconv.Itoa(int(me.Code))).Inc() | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return stream, err | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (n *noopClient) fnStream(ctx context.Context, req Request, opts ...CallOption) (Stream, error) { | ||||||
| 	var err error | 	var err error | ||||||
|  |  | ||||||
| 	// make a copy of call opts | 	// make a copy of call opts | ||||||
| @@ -330,7 +425,7 @@ func (n *noopClient) Stream(ctx context.Context, req Request, opts ...options.Op | |||||||
| 	} else { | 	} else { | ||||||
| 		// got a deadline so no need to setup context | 		// got a deadline so no need to setup context | ||||||
| 		// but we need to set the timeout we pass along | 		// but we need to set the timeout we pass along | ||||||
| 		o := StreamTimeout(time.Until(d)) | 		o := WithStreamTimeout(time.Until(d)) | ||||||
| 		o(&callOpts) | 		o(&callOpts) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -371,7 +466,7 @@ func (n *noopClient) Stream(ctx context.Context, req Request, opts ...options.Op | |||||||
| 		// call backoff first. Someone may want an initial start delay | 		// call backoff first. Someone may want an initial start delay | ||||||
| 		t, cerr := callOpts.Backoff(ctx, req, i) | 		t, cerr := callOpts.Backoff(ctx, req, i) | ||||||
| 		if cerr != nil { | 		if cerr != nil { | ||||||
| 			return nil, errors.InternalServerError("go.micro.client", cerr.Error()) | 			return nil, errors.InternalServerError("go.micro.client", "%s", cerr) | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		// only sleep if greater than 0 | 		// only sleep if greater than 0 | ||||||
| @@ -385,7 +480,7 @@ func (n *noopClient) Stream(ctx context.Context, req Request, opts ...options.Op | |||||||
| 			// TODO apply any filtering here | 			// TODO apply any filtering here | ||||||
| 			routes, err = n.opts.Lookup(ctx, req, callOpts) | 			routes, err = n.opts.Lookup(ctx, req, callOpts) | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| 				return nil, errors.InternalServerError("go.micro.client", err.Error()) | 				return nil, errors.InternalServerError("go.micro.client", "%s", err) | ||||||
| 			} | 			} | ||||||
|  |  | ||||||
| 			// balance the list of nodes | 			// balance the list of nodes | ||||||
| @@ -397,15 +492,7 @@ func (n *noopClient) Stream(ctx context.Context, req Request, opts ...options.Op | |||||||
|  |  | ||||||
| 		node := next() | 		node := next() | ||||||
|  |  | ||||||
| 		// ts := time.Now() |  | ||||||
| 		endpoint := fmt.Sprintf("%s.%s", req.Service(), req.Endpoint()) |  | ||||||
| 		n.opts.Meter.Counter(semconv.ClientRequestInflight, "endpoint", endpoint).Inc() |  | ||||||
| 		stream, cerr := n.stream(ctx, node, req, callOpts) | 		stream, cerr := n.stream(ctx, node, req, callOpts) | ||||||
| 		if cerr != nil { |  | ||||||
| 			n.opts.Meter.Counter(semconv.ClientRequestTotal, "endpoint", endpoint, "status", "failure").Inc() |  | ||||||
| 		} else { |  | ||||||
| 			n.opts.Meter.Counter(semconv.ClientRequestTotal, "endpoint", endpoint, "status", "success").Inc() |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		// record the result of the call to inform future routing decisions | 		// record the result of the call to inform future routing decisions | ||||||
| 		if verr := n.opts.Selector.Record(node, cerr); verr != nil { | 		if verr := n.opts.Selector.Record(node, cerr); verr != nil { | ||||||
| @@ -459,6 +546,92 @@ func (n *noopClient) Stream(ctx context.Context, req Request, opts ...options.Op | |||||||
| 	return nil, grr | 	return nil, grr | ||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopClient) stream(ctx context.Context, addr string, req Request, opts CallOptions) (*noopStream, error) { | func (n *noopClient) stream(ctx context.Context, _ string, _ Request, _ CallOptions) (Stream, error) { | ||||||
| 	return &noopStream{}, 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,57 +6,71 @@ import ( | |||||||
| 	"net" | 	"net" | ||||||
| 	"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/metadata" | 	"go.unistack.org/micro/v3/logger" | ||||||
| 	"go.unistack.org/micro/v4/meter" | 	"go.unistack.org/micro/v3/metadata" | ||||||
| 	"go.unistack.org/micro/v4/options" | 	"go.unistack.org/micro/v3/meter" | ||||||
| 	"go.unistack.org/micro/v4/router" | 	"go.unistack.org/micro/v3/options" | ||||||
| 	"go.unistack.org/micro/v4/selector" | 	"go.unistack.org/micro/v3/register" | ||||||
| 	"go.unistack.org/micro/v4/selector/random" | 	"go.unistack.org/micro/v3/router" | ||||||
| 	"go.unistack.org/micro/v4/tracer" | 	"go.unistack.org/micro/v3/selector" | ||||||
|  | 	"go.unistack.org/micro/v3/selector/random" | ||||||
|  | 	"go.unistack.org/micro/v3/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Options holds client options | // Options holds client options | ||||||
| type Options struct { | type Options struct { | ||||||
| 	// Selector used to select needed address |  | ||||||
| 	Selector selector.Selector |  | ||||||
| 	// Logger used to log messages |  | ||||||
| 	Logger logger.Logger |  | ||||||
| 	// Tracer used for tracing |  | ||||||
| 	Tracer tracer.Tracer |  | ||||||
| 	// Meter used for metrics |  | ||||||
| 	Meter meter.Meter |  | ||||||
| 	// Context is used for external options |  | ||||||
| 	Context context.Context |  | ||||||
| 	// Router used to get route |  | ||||||
| 	Router router.Router |  | ||||||
| 	// TLSConfig specifies tls.Config for secure connection |  | ||||||
| 	TLSConfig *tls.Config |  | ||||||
| 	// Codecs map | 	// Codecs map | ||||||
| 	Codecs map[string]codec.Codec | 	Codecs map[string]codec.Codec | ||||||
| 	// Lookup func used to get destination addr |  | ||||||
| 	Lookup LookupFunc |  | ||||||
| 	// Proxy is used for proxy requests | 	// Proxy is used for proxy requests | ||||||
| 	Proxy string | 	Proxy string | ||||||
| 	// ContentType is used to select codec | 	// ContentType is used to select codec | ||||||
| 	ContentType string | 	ContentType string | ||||||
| 	// Name is the client name | 	// Name is the client name | ||||||
| 	Name string | 	Name string | ||||||
|  |  | ||||||
|  | 	// Selector used to select needed address | ||||||
|  | 	Selector selector.Selector | ||||||
|  | 	// Logger used to log messages | ||||||
|  | 	Logger logger.Logger | ||||||
|  | 	// Tracer used for tracing | ||||||
|  | 	Tracer tracer.Tracer | ||||||
|  | 	// Broker used to publish messages | ||||||
|  | 	Broker broker.Broker | ||||||
|  | 	// Meter used for metrics | ||||||
|  | 	Meter meter.Meter | ||||||
|  | 	// Context is used for external options | ||||||
|  | 	Context context.Context | ||||||
|  | 	// Router used to get route | ||||||
|  | 	Router router.Router | ||||||
|  |  | ||||||
|  | 	// TLSConfig specifies tls.Config for secure connection | ||||||
|  | 	TLSConfig *tls.Config | ||||||
|  |  | ||||||
|  | 	// Lookup func used to get destination addr | ||||||
|  | 	Lookup LookupFunc | ||||||
|  | 	// ContextDialer used to connect | ||||||
|  | 	ContextDialer func(context.Context, string) (net.Conn, error) | ||||||
|  |  | ||||||
|  | 	// Wrappers contains wrappers | ||||||
|  | 	Wrappers []Wrapper | ||||||
|  | 	// Hooks can be run before broker Publish/BatchPublish and | ||||||
|  | 	// Subscribe/BatchSubscribe methods | ||||||
|  | 	Hooks options.Hooks | ||||||
|  |  | ||||||
| 	// CallOptions contains default CallOptions | 	// CallOptions contains default CallOptions | ||||||
| 	CallOptions CallOptions | 	CallOptions CallOptions | ||||||
|  |  | ||||||
| 	// PoolSize connection pool size | 	// PoolSize connection pool size | ||||||
| 	PoolSize int | 	PoolSize int | ||||||
| 	// PoolTTL connection pool ttl | 	// PoolTTL connection pool ttl | ||||||
| 	PoolTTL time.Duration | 	PoolTTL time.Duration | ||||||
| 	// ContextDialer used to connect |  | ||||||
| 	ContextDialer func(context.Context, string) (net.Conn, error) |  | ||||||
| 	// Hooks may contains Client func wrapper |  | ||||||
| 	Hooks options.Hooks |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewCallOptions creates new call options struct | // NewCallOptions creates new call options struct | ||||||
| func NewCallOptions(opts ...options.Option) CallOptions { | func NewCallOptions(opts ...CallOption) CallOptions { | ||||||
| 	options := CallOptions{} | 	options := CallOptions{} | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&options) | 		o(&options) | ||||||
| @@ -66,6 +80,16 @@ func NewCallOptions(opts ...options.Option) CallOptions { | |||||||
|  |  | ||||||
| // CallOptions holds client call options | // CallOptions holds client call options | ||||||
| type CallOptions struct { | type CallOptions struct { | ||||||
|  | 	// RequestMetadata holds additional metadata for call | ||||||
|  | 	RequestMetadata metadata.Metadata | ||||||
|  |  | ||||||
|  | 	// Network name | ||||||
|  | 	Network string | ||||||
|  | 	// Content-Type | ||||||
|  | 	ContentType string | ||||||
|  | 	// AuthToken string | ||||||
|  | 	AuthToken string | ||||||
|  |  | ||||||
| 	// Selector selects addr | 	// Selector selects addr | ||||||
| 	Selector selector.Selector | 	Selector selector.Selector | ||||||
| 	// Context used for deadline | 	// Context used for deadline | ||||||
| @@ -73,46 +97,85 @@ type CallOptions struct { | |||||||
| 	// Router used for route | 	// Router used for route | ||||||
| 	Router router.Router | 	Router router.Router | ||||||
| 	// Retry func used for retries | 	// Retry func used for retries | ||||||
|  |  | ||||||
|  | 	// ResponseMetadata holds additional metadata from call | ||||||
|  | 	ResponseMetadata *metadata.Metadata | ||||||
|  |  | ||||||
| 	Retry RetryFunc | 	Retry RetryFunc | ||||||
| 	// Backoff func used for backoff when retry | 	// Backoff func used for backoff when retry | ||||||
| 	Backoff BackoffFunc | 	Backoff BackoffFunc | ||||||
| 	// Network name | 	// ContextDialer used to connect | ||||||
| 	Network string | 	ContextDialer func(context.Context, string) (net.Conn, error) | ||||||
| 	// Content-Type |  | ||||||
| 	ContentType string |  | ||||||
| 	// AuthToken string |  | ||||||
| 	AuthToken string |  | ||||||
| 	// Address specifies static addr list | 	// Address specifies static addr list | ||||||
| 	Address []string | 	Address []string | ||||||
| 	// SelectOptions selector options | 	// SelectOptions selector options | ||||||
| 	SelectOptions []selector.SelectOption | 	SelectOptions []selector.SelectOption | ||||||
| 	// CallWrappers call wrappers |  | ||||||
| 	CallWrappers []CallWrapper |  | ||||||
| 	// StreamTimeout stream timeout | 	// StreamTimeout stream timeout | ||||||
| 	StreamTimeout time.Duration | 	StreamTimeout time.Duration | ||||||
| 	// RequestTimeout request timeout | 	// RequestTimeout request timeout | ||||||
| 	RequestTimeout time.Duration | 	RequestTimeout time.Duration | ||||||
| 	// RequestMetadata holds additional metadata for call |  | ||||||
| 	RequestMetadata metadata.Metadata |  | ||||||
| 	// ResponseMetadata holds additional metadata from call |  | ||||||
| 	ResponseMetadata *metadata.Metadata |  | ||||||
| 	// DialTimeout dial timeout | 	// DialTimeout dial timeout | ||||||
| 	DialTimeout time.Duration | 	DialTimeout time.Duration | ||||||
| 	// Retries specifies retries num | 	// Retries specifies retries num | ||||||
| 	Retries int | 	Retries int | ||||||
| 	// ContextDialer used to connect |  | ||||||
| 	ContextDialer func(context.Context, string) (net.Conn, error) |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // ContextDialer pass ContextDialer to client | // ContextDialer pass ContextDialer to client | ||||||
| func ContextDialer(fn func(context.Context, string) (net.Conn, error)) options.Option { | func ContextDialer(fn func(context.Context, string) (net.Conn, error)) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, fn, ".ContextDialer") | 		o.ContextDialer = fn | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // Context pass context to client | ||||||
|  | func Context(ctx context.Context) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Context = ctx | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // 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 ...options.Option) RequestOptions { | func NewRequestOptions(opts ...RequestOption) RequestOptions { | ||||||
| 	options := RequestOptions{} | 	options := RequestOptions{} | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&options) | 		o(&options) | ||||||
| @@ -131,24 +194,24 @@ type RequestOptions struct { | |||||||
| } | } | ||||||
|  |  | ||||||
| // NewOptions creates new options struct | // NewOptions creates new options struct | ||||||
| func NewOptions(opts ...options.Option) Options { | func NewOptions(opts ...Option) Options { | ||||||
| 	options := Options{ | 	options := Options{ | ||||||
| 		Context:     context.Background(), | 		Context:     context.Background(), | ||||||
| 		ContentType: DefaultContentType, | 		ContentType: DefaultContentType, | ||||||
| 		Codecs:      make(map[string]codec.Codec), | 		Codecs:      DefaultCodecs, | ||||||
| 		CallOptions: CallOptions{ | 		CallOptions: CallOptions{ | ||||||
| 			Context:        context.Background(), | 			Context:        context.Background(), | ||||||
| 			Backoff:        DefaultBackoff, | 			Backoff:        DefaultBackoff, | ||||||
| 			Retry:          DefaultRetry, | 			Retry:          DefaultRetry, | ||||||
| 			Retries:        DefaultRetries, | 			Retries:        DefaultRetries, | ||||||
| 			RequestTimeout: DefaultRequestTimeout, | 			RequestTimeout: DefaultRequestTimeout, | ||||||
| 			DialTimeout:    DefaultDialTimeout, |  | ||||||
| 		}, | 		}, | ||||||
| 		Lookup:   LookupRoute, | 		Lookup:   LookupRoute, | ||||||
| 		PoolSize: DefaultPoolSize, | 		PoolSize: DefaultPoolSize, | ||||||
| 		PoolTTL:  DefaultPoolTTL, | 		PoolTTL:  DefaultPoolTTL, | ||||||
| 		Selector: random.NewSelector(), | 		Selector: random.NewSelector(), | ||||||
| 		Logger:   logger.DefaultLogger, | 		Logger:   logger.DefaultLogger, | ||||||
|  | 		Broker:   broker.DefaultBroker, | ||||||
| 		Meter:    meter.DefaultMeter, | 		Meter:    meter.DefaultMeter, | ||||||
| 		Tracer:   tracer.DefaultTracer, | 		Tracer:   tracer.DefaultTracer, | ||||||
| 		Router:   router.DefaultRouter, | 		Router:   router.DefaultRouter, | ||||||
| @@ -161,131 +224,345 @@ func NewOptions(opts ...options.Option) Options { | |||||||
| 	return options | 	return options | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // Broker to be used for pub/sub | ||||||
|  | func Broker(b broker.Broker) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Broker = b | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Tracer to be used for tracing | ||||||
|  | func Tracer(t tracer.Tracer) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Tracer = t | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Logger to be used for log mesages | ||||||
|  | func Logger(l logger.Logger) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Logger = l | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Meter to be used for metrics | ||||||
|  | func Meter(m meter.Meter) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Meter = m | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Codec to be used to encode/decode requests for a given content type | ||||||
|  | func Codec(contentType string, c codec.Codec) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Codecs[contentType] = c | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ContentType used by default if not specified | ||||||
|  | func ContentType(ct string) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.ContentType = ct | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // Proxy sets the proxy address | // Proxy sets the proxy address | ||||||
| func Proxy(addr string) options.Option { | func Proxy(addr string) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, addr, ".Proxy") | 		o.Proxy = addr | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // PoolSize sets the connection pool size | // PoolSize sets the connection pool size | ||||||
| func PoolSize(d int) options.Option { | func PoolSize(d int) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, d, ".PoolSize") | 		o.PoolSize = d | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // PoolTTL sets the connection pool ttl | // PoolTTL sets the connection pool ttl | ||||||
| func PoolTTL(td time.Duration) options.Option { | func PoolTTL(d time.Duration) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, td, ".PoolTTL") | 		o.PoolTTL = d | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Register sets the routers register | ||||||
|  | func Register(r register.Register) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		if o.Router != nil { | ||||||
|  | 			_ = o.Router.Init(router.Register(r)) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Router is used to lookup routes for a service | ||||||
|  | func Router(r router.Router) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Router = r | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // Selector is used to select a route | // Selector is used to select a route | ||||||
| func Selector(s selector.Selector) options.Option { | func Selector(s selector.Selector) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, s, ".Selector") | 		o.Selector = s | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // Backoff is used to set the backoff function used when retrying Calls | // Backoff is used to set the backoff function used when retrying Calls | ||||||
| func Backoff(fn BackoffFunc) options.Option { | func Backoff(fn BackoffFunc) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, fn, ".Backoff") | 		o.CallOptions.Backoff = fn | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Name sets the client name | ||||||
|  | func Name(n string) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Name = n | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // Lookup sets the lookup function to use for resolving service names | // Lookup sets the lookup function to use for resolving service names | ||||||
| func Lookup(fn LookupFunc) options.Option { | func Lookup(l LookupFunc) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, fn, ".Lookup") | 		o.Lookup = l | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // WithCallWrapper sets the retry function to be used when re-trying. | // TLSConfig specifies a *tls.Config | ||||||
| func WithCallWrapper(fn CallWrapper) options.Option { | func TLSConfig(t *tls.Config) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, fn, ".CallWrappers") | 		// set the internal tls | ||||||
|  | 		o.TLSConfig = t | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // Retries sets the retry count when making the request. | // Retries sets the retry count when making the request. | ||||||
| func Retries(n int) options.Option { | func Retries(i int) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, n, ".Retries") | 		o.CallOptions.Retries = i | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // Retry sets the retry function to be used when re-trying. | // Retry sets the retry function to be used when re-trying. | ||||||
| func Retry(fn RetryFunc) options.Option { | func Retry(fn RetryFunc) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, fn, ".Retry") | 		o.CallOptions.Retry = fn | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // RequestTimeout is the request timeout. | // RequestTimeout is the request timeout. | ||||||
| func RequestTimeout(td time.Duration) options.Option { | func RequestTimeout(d time.Duration) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, td, ".RequestTimeout") | 		o.CallOptions.RequestTimeout = d | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // StreamTimeout sets the stream timeout | // StreamTimeout sets the stream timeout | ||||||
| func StreamTimeout(td time.Duration) options.Option { | func StreamTimeout(d time.Duration) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, td, ".StreamTimeout") | 		o.CallOptions.StreamTimeout = d | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // DialTimeout sets the dial timeout | // DialTimeout sets the dial timeout | ||||||
| func DialTimeout(td time.Duration) options.Option { | func DialTimeout(d time.Duration) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, td, ".DialTimeout") | 		o.CallOptions.DialTimeout = d | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // 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 | ||||||
|  | func WithContextDialer(fn func(context.Context, string) (net.Conn, error)) CallOption { | ||||||
|  | 	return func(o *CallOptions) { | ||||||
|  | 		o.ContextDialer = fn | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithContentType specifies call content type | ||||||
|  | func WithContentType(ct string) CallOption { | ||||||
|  | 	return func(o *CallOptions) { | ||||||
|  | 		o.ContentType = ct | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithAddress sets the remote addresses to use rather than using service discovery | ||||||
|  | func WithAddress(a ...string) CallOption { | ||||||
|  | 	return func(o *CallOptions) { | ||||||
|  | 		o.Address = a | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithBackoff is a CallOption which overrides that which | ||||||
|  | // set in Options.CallOptions | ||||||
|  | func WithBackoff(fn BackoffFunc) CallOption { | ||||||
|  | 	return func(o *CallOptions) { | ||||||
|  | 		o.Backoff = fn | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithRetry is a CallOption which overrides that which | ||||||
|  | // set in Options.CallOptions | ||||||
|  | func WithRetry(fn RetryFunc) CallOption { | ||||||
|  | 	return func(o *CallOptions) { | ||||||
|  | 		o.Retry = fn | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithRetries is a CallOption which overrides that which | ||||||
|  | // set in Options.CallOptions | ||||||
|  | func WithRetries(i int) CallOption { | ||||||
|  | 	return func(o *CallOptions) { | ||||||
|  | 		o.Retries = i | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // WithResponseMetadata is a CallOption which adds metadata.Metadata to Options.CallOptions | // WithResponseMetadata is a CallOption which adds metadata.Metadata to Options.CallOptions | ||||||
| func ResponseMetadata(md *metadata.Metadata) options.Option { | func WithResponseMetadata(md *metadata.Metadata) CallOption { | ||||||
| 	return func(src interface{}) error { | 	return func(o *CallOptions) { | ||||||
| 		return options.Set(src, md, ".ResponseMetadata") | 		o.ResponseMetadata = md | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // WithRequestMetadata is a CallOption which adds metadata.Metadata to Options.CallOptions | // WithRequestMetadata is a CallOption which adds metadata.Metadata to Options.CallOptions | ||||||
| func RequestMetadata(md metadata.Metadata) options.Option { | func WithRequestMetadata(md metadata.Metadata) CallOption { | ||||||
| 	return func(src interface{}) error { | 	return func(o *CallOptions) { | ||||||
| 		return options.Set(src, metadata.Copy(md), ".RequestMetadata") | 		o.RequestMetadata = md | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // AuthToken is a CallOption which overrides the | // WithRequestTimeout is a CallOption which overrides that which | ||||||
|  | // set in Options.CallOptions | ||||||
|  | func WithRequestTimeout(d time.Duration) CallOption { | ||||||
|  | 	return func(o *CallOptions) { | ||||||
|  | 		o.RequestTimeout = d | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithStreamTimeout sets the stream timeout | ||||||
|  | func WithStreamTimeout(d time.Duration) CallOption { | ||||||
|  | 	return func(o *CallOptions) { | ||||||
|  | 		o.StreamTimeout = d | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithDialTimeout is a CallOption which overrides that which | ||||||
|  | // set in Options.CallOptions | ||||||
|  | func WithDialTimeout(d time.Duration) CallOption { | ||||||
|  | 	return func(o *CallOptions) { | ||||||
|  | 		o.DialTimeout = d | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithAuthToken is a CallOption which overrides the | ||||||
| // authorization header with the services own auth token | // authorization header with the services own auth token | ||||||
| func AuthToken(t string) options.Option { | func WithAuthToken(t string) CallOption { | ||||||
| 	return func(src interface{}) error { | 	return func(o *CallOptions) { | ||||||
| 		return options.Set(src, t, ".AuthToken") | 		o.AuthToken = t | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // Network is a CallOption which sets the network attribute | // WithRouter sets the router to use for this call | ||||||
| func Network(n string) options.Option { | func WithRouter(r router.Router) CallOption { | ||||||
| 	return func(src interface{}) error { | 	return func(o *CallOptions) { | ||||||
| 		return options.Set(src, n, ".Network") | 		o.Router = r | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithSelector sets the selector to use for this call | ||||||
|  | func WithSelector(s selector.Selector) CallOption { | ||||||
|  | 	return func(o *CallOptions) { | ||||||
|  | 		o.Selector = s | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| /* |  | ||||||
| // WithSelectOptions sets the options to pass to the selector for this call | // WithSelectOptions sets the options to pass to the selector for this call | ||||||
| func WithSelectOptions(sops ...selector.SelectOption) options.Option { | func WithSelectOptions(sops ...selector.SelectOption) CallOption { | ||||||
| 	return func(o *CallOptions) { | 	return func(o *CallOptions) { | ||||||
| 		o.SelectOptions = sops | 		o.SelectOptions = sops | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| */ |  | ||||||
|  | // 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) options.Option { | func StreamingRequest(b bool) RequestOption { | ||||||
| 	return func(src interface{}) error { | 	return func(o *RequestOptions) { | ||||||
| 		return options.Set(src, b, ".Stream") | 		o.Stream = b | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // RequestContentType specifies request content type | ||||||
|  | func RequestContentType(ct string) RequestOption { | ||||||
|  | 	return func(o *RequestOptions) { | ||||||
|  | 		o.ContentType = ct | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Hooks sets hook runs before action | ||||||
|  | func Hooks(h ...options.Hook) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Hooks = append(o.Hooks, h...) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -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 { | ||||||
|   | |||||||
							
								
								
									
										47
									
								
								cluster/cluster.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										47
									
								
								cluster/cluster.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,47 @@ | |||||||
|  | package cluster | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v3/metadata" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Message sent to member in cluster | ||||||
|  | type Message interface { | ||||||
|  | 	// Header returns message headers | ||||||
|  | 	Header() metadata.Metadata | ||||||
|  | 	// Body returns broker message may be []byte slice or some go struct or interface | ||||||
|  | 	Body() interface{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type Node interface { | ||||||
|  | 	// Name returns node name | ||||||
|  | 	Name() string | ||||||
|  | 	// Address returns node address | ||||||
|  | 	Address() string | ||||||
|  | 	// Metadata returns node metadata | ||||||
|  | 	Metadata() metadata.Metadata | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Cluster interface used for cluster communication across nodes | ||||||
|  | type Cluster interface { | ||||||
|  | 	// Join is used to take an existing members and performing state sync | ||||||
|  | 	Join(ctx context.Context, addr ...string) error | ||||||
|  | 	// Leave broadcast a leave message and stop listeners | ||||||
|  | 	Leave(ctx context.Context) error | ||||||
|  | 	// Ping is used to probe live status of the node | ||||||
|  | 	Ping(ctx context.Context, node Node, payload []byte) error | ||||||
|  | 	// Members returns the cluster members | ||||||
|  | 	Members() ([]Node, error) | ||||||
|  | 	// Broadcast send message for all members in cluster, if filter is not nil, nodes may be filtered | ||||||
|  | 	// by key/value pairs | ||||||
|  | 	Broadcast(ctx context.Context, msg Message, filter ...string) error | ||||||
|  | 	// Unicast send message to single member in cluster | ||||||
|  | 	Unicast(ctx context.Context, node Node, msg Message) error | ||||||
|  | 	// Live returns cluster liveness | ||||||
|  | 	Live() bool | ||||||
|  | 	// Ready returns cluster readiness | ||||||
|  | 	Ready() bool | ||||||
|  | 	// Health returns cluster health | ||||||
|  | 	Health() bool | ||||||
|  | } | ||||||
| @@ -1,19 +1,8 @@ | |||||||
| // Package codec is an interface for encoding messages | // Package codec is an interface for encoding messages | ||||||
| package codec // import "go.unistack.org/micro/v4/codec" | package codec | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"errors" | 	"errors" | ||||||
| 	"io" |  | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/metadata" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // Message types |  | ||||||
| const ( |  | ||||||
| 	Error MessageType = iota |  | ||||||
| 	Request |  | ||||||
| 	Response |  | ||||||
| 	Event |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| @@ -24,65 +13,23 @@ var ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| 	// DefaultMaxMsgSize specifies how much data codec can handle |  | ||||||
| 	DefaultMaxMsgSize = 1024 * 1024 * 4 // 4Mb |  | ||||||
| 	// DefaultCodec is the global default codec | 	// DefaultCodec is the global default codec | ||||||
| 	DefaultCodec = NewCodec() | 	DefaultCodec = NewCodec() | ||||||
| 	// DefaultTagName specifies struct tag name to control codec Marshal/Unmarshal | 	// DefaultTagName specifies struct tag name to control codec Marshal/Unmarshal | ||||||
| 	DefaultTagName = "codec" | 	DefaultTagName = "codec" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // MessageType specifies message type for codec | // Codec encodes/decodes various types of messages. | ||||||
| type MessageType int |  | ||||||
|  |  | ||||||
| // Codec encodes/decodes various types of messages used within micro. |  | ||||||
| // ReadHeader and ReadBody are called in pairs to read requests/responses |  | ||||||
| // from the connection. Close is called when finished with the |  | ||||||
| // connection. ReadBody may be called with a nil argument to force the |  | ||||||
| // body to be read and discarded. |  | ||||||
| type Codec interface { | type Codec interface { | ||||||
| 	ReadHeader(r io.Reader, m *Message, mt MessageType) error |  | ||||||
| 	ReadBody(r io.Reader, v interface{}) error |  | ||||||
| 	Write(w io.Writer, m *Message, v interface{}) error |  | ||||||
| 	Marshal(v interface{}, opts ...Option) ([]byte, error) | 	Marshal(v interface{}, opts ...Option) ([]byte, error) | ||||||
| 	Unmarshal(b []byte, v interface{}, opts ...Option) error | 	Unmarshal(b []byte, v interface{}, opts ...Option) error | ||||||
| 	String() string | 	String() string | ||||||
| } | } | ||||||
|  |  | ||||||
| // Message represents detailed information about | type CodecV2 interface { | ||||||
| // the communication, likely followed by the body. | 	Marshal(buf []byte, v interface{}, opts ...Option) ([]byte, error) | ||||||
| // In the case of an error, body may be nil. | 	Unmarshal(buf []byte, v interface{}, opts ...Option) error | ||||||
| type Message struct { | 	String() string | ||||||
| 	Header   metadata.Metadata |  | ||||||
| 	Target   string |  | ||||||
| 	Method   string |  | ||||||
| 	Endpoint string |  | ||||||
| 	Error    string |  | ||||||
| 	ID       string |  | ||||||
| 	Body     []byte |  | ||||||
| 	Type     MessageType |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewMessage creates new codec message |  | ||||||
| func NewMessage(t MessageType) *Message { |  | ||||||
| 	return &Message{Type: t, Header: metadata.New(0)} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // MarshalAppend calls codec.Marshal(v) and returns the data appended to buf. |  | ||||||
| // If codec implements MarshalAppend, that is called instead. |  | ||||||
| func MarshalAppend(buf []byte, c Codec, v interface{}, opts ...Option) ([]byte, error) { |  | ||||||
| 	if nc, ok := c.(interface { |  | ||||||
| 		MarshalAppend([]byte, interface{}, ...Option) ([]byte, error) |  | ||||||
| 	}); ok { |  | ||||||
| 		return nc.MarshalAppend(buf, v, opts...) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	mbuf, err := c.Marshal(v, opts...) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return nil, err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return append(buf, mbuf...), nil |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // RawMessage is a raw encoded JSON value. | // RawMessage is a raw encoded JSON value. | ||||||
| @@ -93,6 +40,8 @@ type RawMessage []byte | |||||||
| func (m *RawMessage) MarshalJSON() ([]byte, error) { | func (m *RawMessage) MarshalJSON() ([]byte, error) { | ||||||
| 	if m == nil { | 	if m == nil { | ||||||
| 		return []byte("null"), nil | 		return []byte("null"), nil | ||||||
|  | 	} else if len(*m) == 0 { | ||||||
|  | 		return []byte("null"), nil | ||||||
| 	} | 	} | ||||||
| 	return *m, nil | 	return *m, nil | ||||||
| } | } | ||||||
|   | |||||||
| @@ -15,6 +15,15 @@ func FromContext(ctx context.Context) (Codec, bool) { | |||||||
| 	return c, ok | 	return c, ok | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // MustContext returns codec from context | ||||||
|  | func MustContext(ctx context.Context) Codec { | ||||||
|  | 	c, ok := FromContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		panic("missing codec") | ||||||
|  | 	} | ||||||
|  | 	return c | ||||||
|  | } | ||||||
|  |  | ||||||
| // NewContext put codec in context | // NewContext put codec in context | ||||||
| func NewContext(ctx context.Context, c Codec) context.Context { | func NewContext(ctx context.Context, c Codec) context.Context { | ||||||
| 	if ctx == nil { | 	if ctx == nil { | ||||||
|   | |||||||
| @@ -1,4 +1,4 @@ | |||||||
| // Copyright 2021-2023 Unistack LLC | // Copyright 2021 Unistack LLC | ||||||
| // | // | ||||||
| // Licensed under the Apache License, Version 2.0 (the "License"); | // Licensed under the Apache License, Version 2.0 (the "License"); | ||||||
| // you may not use this file except in compliance with the License. | // you may not use this file except in compliance with the License. | ||||||
| @@ -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"; | ||||||
|   | |||||||
| @@ -2,70 +2,14 @@ package codec | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"encoding/json" | 	"encoding/json" | ||||||
| 	"io" |  | ||||||
|  | 	codecpb "go.unistack.org/micro-proto/v3/codec" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type noopCodec struct { | type noopCodec struct { | ||||||
| 	opts Options | 	opts Options | ||||||
| } | } | ||||||
|  |  | ||||||
| func (c *noopCodec) ReadHeader(conn io.Reader, m *Message, t MessageType) error { |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (c *noopCodec) ReadBody(conn io.Reader, b interface{}) error { |  | ||||||
| 	// read bytes |  | ||||||
| 	buf, err := io.ReadAll(conn) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if b == nil { |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	switch v := b.(type) { |  | ||||||
| 	case *string: |  | ||||||
| 		*v = string(buf) |  | ||||||
| 	case *[]byte: |  | ||||||
| 		*v = buf |  | ||||||
| 	case *Frame: |  | ||||||
| 		v.Data = buf |  | ||||||
| 	default: |  | ||||||
| 		return json.Unmarshal(buf, v) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (c *noopCodec) Write(conn io.Writer, m *Message, b interface{}) error { |  | ||||||
| 	if b == nil { |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	var v []byte |  | ||||||
| 	switch vb := b.(type) { |  | ||||||
| 	case *Frame: |  | ||||||
| 		v = vb.Data |  | ||||||
| 	case string: |  | ||||||
| 		v = []byte(vb) |  | ||||||
| 	case *string: |  | ||||||
| 		v = []byte(*vb) |  | ||||||
| 	case *[]byte: |  | ||||||
| 		v = *vb |  | ||||||
| 	case []byte: |  | ||||||
| 		v = vb |  | ||||||
| 	default: |  | ||||||
| 		var err error |  | ||||||
| 		v, err = json.Marshal(vb) |  | ||||||
| 		if err != nil { |  | ||||||
| 			return err |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	_, err := conn.Write(v) |  | ||||||
| 	return err |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (c *noopCodec) String() string { | func (c *noopCodec) String() string { | ||||||
| 	return "noop" | 	return "noop" | ||||||
| } | } | ||||||
| @@ -91,8 +35,8 @@ func (c *noopCodec) Marshal(v interface{}, opts ...Option) ([]byte, error) { | |||||||
| 		return ve, nil | 		return ve, nil | ||||||
| 	case *Frame: | 	case *Frame: | ||||||
| 		return ve.Data, nil | 		return ve.Data, nil | ||||||
| 	case *Message: | 	case *codecpb.Frame: | ||||||
| 		return ve.Body, nil | 		return ve.Data, nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return json.Marshal(v) | 	return json.Marshal(v) | ||||||
| @@ -115,8 +59,8 @@ func (c *noopCodec) Unmarshal(d []byte, v interface{}, opts ...Option) error { | |||||||
| 	case *Frame: | 	case *Frame: | ||||||
| 		ve.Data = d | 		ve.Data = d | ||||||
| 		return nil | 		return nil | ||||||
| 	case *Message: | 	case *codecpb.Frame: | ||||||
| 		ve.Body = d | 		ve.Data = d | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|   | |||||||
| @@ -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 | ||||||
| @@ -23,15 +23,8 @@ type Options struct { | |||||||
| 	Context context.Context | 	Context context.Context | ||||||
| 	// TagName specifies tag name in struct to control codec | 	// TagName specifies tag name in struct to control codec | ||||||
| 	TagName string | 	TagName string | ||||||
| 	// MaxMsgSize specifies max messages size that reads by codec | 	// Flatten specifies that struct must be analyzed for flatten tag | ||||||
| 	MaxMsgSize int | 	Flatten bool | ||||||
| } |  | ||||||
|  |  | ||||||
| // MaxMsgSize sets the max message size |  | ||||||
| func MaxMsgSize(n int) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.MaxMsgSize = n |  | ||||||
| 	} |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // TagName sets the codec tag name in struct | // TagName sets the codec tag name in struct | ||||||
| @@ -41,6 +34,13 @@ func TagName(n string) Option { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // Flatten enables checking for flatten tag name | ||||||
|  | func Flatten(b bool) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Flatten = b | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // Logger sets the logger | // Logger sets the logger | ||||||
| func Logger(l logger.Logger) Option { | func Logger(l logger.Logger) Option { | ||||||
| 	return func(o *Options) { | 	return func(o *Options) { | ||||||
| @@ -69,8 +69,8 @@ func NewOptions(opts ...Option) Options { | |||||||
| 		Logger:  logger.DefaultLogger, | 		Logger:  logger.DefaultLogger, | ||||||
| 		Meter:   meter.DefaultMeter, | 		Meter:   meter.DefaultMeter, | ||||||
| 		Tracer:  tracer.DefaultTracer, | 		Tracer:  tracer.DefaultTracer, | ||||||
| 		MaxMsgSize: DefaultMaxMsgSize, |  | ||||||
| 		TagName: DefaultTagName, | 		TagName: DefaultTagName, | ||||||
|  | 		Flatten: false, | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
|   | |||||||
| @@ -1,13 +1,11 @@ | |||||||
| // Package config is an interface for dynamic configuration. | // Package config is an interface for dynamic configuration. | ||||||
| package config // import "go.unistack.org/micro/v4/config" | package config | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"errors" | 	"errors" | ||||||
| 	"reflect" | 	"reflect" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/options" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type Validator interface { | type Validator interface { | ||||||
| @@ -39,19 +37,26 @@ type Config interface { | |||||||
| 	// Name returns name of config | 	// Name returns name of config | ||||||
| 	Name() string | 	Name() string | ||||||
| 	// Init the config | 	// Init the config | ||||||
| 	Init(opts ...options.Option) error | 	Init(opts ...Option) error | ||||||
| 	// Options in the config | 	// Options in the config | ||||||
| 	Options() Options | 	Options() Options | ||||||
| 	// Load config from sources | 	// Load config from sources | ||||||
| 	Load(context.Context, ...options.Option) error | 	Load(context.Context, ...LoadOption) error | ||||||
| 	// Save config to sources | 	// Save config to sources | ||||||
| 	Save(context.Context, ...options.Option) error | 	Save(context.Context, ...SaveOption) error | ||||||
| 	// Watch a config for changes | 	// Watch a config for changes | ||||||
| 	Watch(context.Context, ...options.Option) (Watcher, error) | 	Watch(context.Context, ...WatchOption) (Watcher, error) | ||||||
| 	// String returns config type name | 	// String returns config type name | ||||||
| 	String() string | 	String() string | ||||||
| } | } | ||||||
|  |  | ||||||
|  | type ( | ||||||
|  | 	FuncLoad func(ctx context.Context, opts ...LoadOption) error | ||||||
|  | 	HookLoad func(next FuncLoad) FuncLoad | ||||||
|  | 	FuncSave func(ctx context.Context, opts ...SaveOption) error | ||||||
|  | 	HookSave func(next FuncSave) FuncSave | ||||||
|  | ) | ||||||
|  |  | ||||||
| // Watcher is the config watcher | // Watcher is the config watcher | ||||||
| type Watcher interface { | type Watcher interface { | ||||||
| 	// Next blocks until update happens or error returned | 	// Next blocks until update happens or error returned | ||||||
| @@ -61,7 +66,7 @@ type Watcher interface { | |||||||
| } | } | ||||||
|  |  | ||||||
| // Load loads config from config sources | // Load loads config from config sources | ||||||
| func Load(ctx context.Context, cs []Config, opts ...options.Option) error { | func Load(ctx context.Context, cs []Config, opts ...LoadOption) error { | ||||||
| 	var err error | 	var err error | ||||||
| 	for _, c := range cs { | 	for _, c := range cs { | ||||||
| 		if err = c.Init(); err != nil { | 		if err = c.Init(); err != nil { | ||||||
| @@ -133,7 +138,7 @@ var ( | |||||||
| 				return nil | 				return nil | ||||||
| 			} | 			} | ||||||
| 			if err := fn(ctx, c); err != nil { | 			if err := fn(ctx, c); err != nil { | ||||||
| 				c.Options().Logger.Errorf(ctx, "%s BeforeLoad err: %v", c.String(), err) | 				c.Options().Logger.Error(ctx, c.String()+" BeforeLoad error", err) | ||||||
| 				if !c.Options().AllowFail { | 				if !c.Options().AllowFail { | ||||||
| 					return err | 					return err | ||||||
| 				} | 				} | ||||||
| @@ -148,7 +153,7 @@ var ( | |||||||
| 				return nil | 				return nil | ||||||
| 			} | 			} | ||||||
| 			if err := fn(ctx, c); err != nil { | 			if err := fn(ctx, c); err != nil { | ||||||
| 				c.Options().Logger.Errorf(ctx, "%s AfterLoad err: %v", c.String(), err) | 				c.Options().Logger.Error(ctx, c.String()+" AfterLoad error", err) | ||||||
| 				if !c.Options().AllowFail { | 				if !c.Options().AllowFail { | ||||||
| 					return err | 					return err | ||||||
| 				} | 				} | ||||||
| @@ -163,7 +168,7 @@ var ( | |||||||
| 				return nil | 				return nil | ||||||
| 			} | 			} | ||||||
| 			if err := fn(ctx, c); err != nil { | 			if err := fn(ctx, c); err != nil { | ||||||
| 				c.Options().Logger.Errorf(ctx, "%s BeforeSave err: %v", c.String(), err) | 				c.Options().Logger.Error(ctx, c.String()+" BeforeSave error", err) | ||||||
| 				if !c.Options().AllowFail { | 				if !c.Options().AllowFail { | ||||||
| 					return err | 					return err | ||||||
| 				} | 				} | ||||||
| @@ -178,7 +183,7 @@ var ( | |||||||
| 				return nil | 				return nil | ||||||
| 			} | 			} | ||||||
| 			if err := fn(ctx, c); err != nil { | 			if err := fn(ctx, c); err != nil { | ||||||
| 				c.Options().Logger.Errorf(ctx, "%s AfterSave err: %v", c.String(), err) | 				c.Options().Logger.Error(ctx, c.String()+" AfterSave error", err) | ||||||
| 				if !c.Options().AllowFail { | 				if !c.Options().AllowFail { | ||||||
| 					return err | 					return err | ||||||
| 				} | 				} | ||||||
| @@ -193,7 +198,7 @@ var ( | |||||||
| 				return nil | 				return nil | ||||||
| 			} | 			} | ||||||
| 			if err := fn(ctx, c); err != nil { | 			if err := fn(ctx, c); err != nil { | ||||||
| 				c.Options().Logger.Errorf(ctx, "%s BeforeInit err: %v", c.String(), err) | 				c.Options().Logger.Error(ctx, c.String()+" BeforeInit error", err) | ||||||
| 				if !c.Options().AllowFail { | 				if !c.Options().AllowFail { | ||||||
| 					return err | 					return err | ||||||
| 				} | 				} | ||||||
| @@ -208,7 +213,7 @@ var ( | |||||||
| 				return nil | 				return nil | ||||||
| 			} | 			} | ||||||
| 			if err := fn(ctx, c); err != nil { | 			if err := fn(ctx, c); err != nil { | ||||||
| 				c.Options().Logger.Errorf(ctx, "%s AfterInit err: %v", c.String(), err) | 				c.Options().Logger.Error(ctx, c.String()+" AfterInit error", err) | ||||||
| 				if !c.Options().AllowFail { | 				if !c.Options().AllowFail { | ||||||
| 					return err | 					return err | ||||||
| 				} | 				} | ||||||
|   | |||||||
| @@ -15,6 +15,15 @@ func FromContext(ctx context.Context) (Config, bool) { | |||||||
| 	return c, ok | 	return c, ok | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // MustContext returns store from context | ||||||
|  | func MustContext(ctx context.Context) Config { | ||||||
|  | 	c, ok := FromContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		panic("missing config") | ||||||
|  | 	} | ||||||
|  | 	return c | ||||||
|  | } | ||||||
|  |  | ||||||
| // NewContext put store in context | // NewContext put store in context | ||||||
| func NewContext(ctx context.Context, c Config) context.Context { | func NewContext(ctx context.Context, c Config) context.Context { | ||||||
| 	if ctx == nil { | 	if ctx == nil { | ||||||
| @@ -22,3 +31,43 @@ func NewContext(ctx context.Context, c Config) context.Context { | |||||||
| 	} | 	} | ||||||
| 	return context.WithValue(ctx, configKey{}, c) | 	return context.WithValue(ctx, configKey{}, 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) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SetSaveOption returns a function to setup a context with given value | ||||||
|  | func SetSaveOption(k, v interface{}) SaveOption { | ||||||
|  | 	return func(o *SaveOptions) { | ||||||
|  | 		if o.Context == nil { | ||||||
|  | 			o.Context = context.Background() | ||||||
|  | 		} | ||||||
|  | 		o.Context = context.WithValue(o.Context, k, v) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SetLoadOption returns a function to setup a context with given value | ||||||
|  | func SetLoadOption(k, v interface{}) LoadOption { | ||||||
|  | 	return func(o *LoadOptions) { | ||||||
|  | 		if o.Context == nil { | ||||||
|  | 			o.Context = context.Background() | ||||||
|  | 		} | ||||||
|  | 		o.Context = context.WithValue(o.Context, k, v) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SetWatchOption returns a function to setup a context with given value | ||||||
|  | func SetWatchOption(k, v interface{}) WatchOption { | ||||||
|  | 	return func(o *WatchOptions) { | ||||||
|  | 		if o.Context == nil { | ||||||
|  | 			o.Context = context.Background() | ||||||
|  | 		} | ||||||
|  | 		o.Context = context.WithValue(o.Context, k, v) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -40,3 +40,47 @@ func TestNewContext(t *testing.T) { | |||||||
| 		t.Fatal("NewContext not works") | 		t.Fatal("NewContext not works") | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func TestSetOption(t *testing.T) { | ||||||
|  | 	type key struct{} | ||||||
|  | 	o := SetOption(key{}, "test") | ||||||
|  | 	opts := &Options{} | ||||||
|  | 	o(opts) | ||||||
|  |  | ||||||
|  | 	if v, ok := opts.Context.Value(key{}).(string); !ok || v == "" { | ||||||
|  | 		t.Fatal("SetOption not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestSetSaveOption(t *testing.T) { | ||||||
|  | 	type key struct{} | ||||||
|  | 	o := SetSaveOption(key{}, "test") | ||||||
|  | 	opts := &SaveOptions{} | ||||||
|  | 	o(opts) | ||||||
|  |  | ||||||
|  | 	if v, ok := opts.Context.Value(key{}).(string); !ok || v == "" { | ||||||
|  | 		t.Fatal("SetSaveOption not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestSetLoadOption(t *testing.T) { | ||||||
|  | 	type key struct{} | ||||||
|  | 	o := SetLoadOption(key{}, "test") | ||||||
|  | 	opts := &LoadOptions{} | ||||||
|  | 	o(opts) | ||||||
|  |  | ||||||
|  | 	if v, ok := opts.Context.Value(key{}).(string); !ok || v == "" { | ||||||
|  | 		t.Fatal("SetLoadOption not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestSetWatchOption(t *testing.T) { | ||||||
|  | 	type key struct{} | ||||||
|  | 	o := SetWatchOption(key{}, "test") | ||||||
|  | 	opts := &WatchOptions{} | ||||||
|  | 	o(opts) | ||||||
|  |  | ||||||
|  | 	if v, ok := opts.Context.Value(key{}).(string); !ok || v == "" { | ||||||
|  | 		t.Fatal("SetWatchOption not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -7,15 +7,17 @@ import ( | |||||||
| 	"strings" | 	"strings" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
|  | 	"dario.cat/mergo" | ||||||
| 	"github.com/google/uuid" | 	"github.com/google/uuid" | ||||||
| 	"github.com/imdario/mergo" | 	"go.unistack.org/micro/v3/options" | ||||||
| 	"go.unistack.org/micro/v4/options" | 	mid "go.unistack.org/micro/v3/util/id" | ||||||
| 	mid "go.unistack.org/micro/v4/util/id" | 	rutil "go.unistack.org/micro/v3/util/reflect" | ||||||
| 	rutil "go.unistack.org/micro/v4/util/reflect" | 	mtime "go.unistack.org/micro/v3/util/time" | ||||||
| 	mtime "go.unistack.org/micro/v4/util/time" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type defaultConfig struct { | type defaultConfig struct { | ||||||
|  | 	funcLoad FuncLoad | ||||||
|  | 	funcSave FuncSave | ||||||
| 	opts     Options | 	opts     Options | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -23,7 +25,7 @@ func (c *defaultConfig) Options() Options { | |||||||
| 	return c.opts | 	return c.opts | ||||||
| } | } | ||||||
|  |  | ||||||
| func (c *defaultConfig) Init(opts ...options.Option) error { | func (c *defaultConfig) Init(opts ...Option) error { | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&c.opts) | 		o(&c.opts) | ||||||
| 	} | 	} | ||||||
| @@ -32,6 +34,18 @@ func (c *defaultConfig) Init(opts ...options.Option) error { | |||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	c.funcLoad = c.fnLoad | ||||||
|  | 	c.funcSave = c.fnSave | ||||||
|  |  | ||||||
|  | 	c.opts.Hooks.EachPrev(func(hook options.Hook) { | ||||||
|  | 		switch h := hook.(type) { | ||||||
|  | 		case HookLoad: | ||||||
|  | 			c.funcLoad = h(c.funcLoad) | ||||||
|  | 		case HookSave: | ||||||
|  | 			c.funcSave = h(c.funcSave) | ||||||
|  | 		} | ||||||
|  | 	}) | ||||||
|  |  | ||||||
| 	if err := DefaultAfterInit(c.opts.Context, c); err != nil && !c.opts.AllowFail { | 	if err := DefaultAfterInit(c.opts.Context, c); err != nil && !c.opts.AllowFail { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| @@ -39,8 +53,18 @@ func (c *defaultConfig) Init(opts ...options.Option) error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (c *defaultConfig) Load(ctx context.Context, opts ...options.Option) error { | func (c *defaultConfig) Load(ctx context.Context, opts ...LoadOption) error { | ||||||
| 	if err := DefaultBeforeLoad(ctx, c); err != nil && !c.opts.AllowFail { | 	return c.funcLoad(ctx, opts...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *defaultConfig) fnLoad(ctx context.Context, opts ...LoadOption) error { | ||||||
|  | 	var err error | ||||||
|  |  | ||||||
|  | 	if c.opts.SkipLoad != nil && c.opts.SkipLoad(ctx, c) { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if err = DefaultBeforeLoad(ctx, c); err != nil && !c.opts.AllowFail { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -186,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 | ||||||
| @@ -230,6 +254,7 @@ func fillValue(value reflect.Value, val string) error { | |||||||
| 		} | 		} | ||||||
| 		value.Set(reflect.ValueOf(v)) | 		value.Set(reflect.ValueOf(v)) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -292,7 +317,15 @@ func fillValues(valueOf reflect.Value, tname string) error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (c *defaultConfig) Save(ctx context.Context, opts ...options.Option) error { | func (c *defaultConfig) Save(ctx context.Context, opts ...SaveOption) error { | ||||||
|  | 	return c.funcSave(ctx, opts...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *defaultConfig) fnSave(ctx context.Context, opts ...SaveOption) error { | ||||||
|  | 	if c.opts.SkipSave != nil && c.opts.SkipSave(ctx, c) { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	if err := DefaultBeforeSave(ctx, c); err != nil { | 	if err := DefaultBeforeSave(ctx, c); err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| @@ -312,15 +345,19 @@ func (c *defaultConfig) Name() string { | |||||||
| 	return c.opts.Name | 	return c.opts.Name | ||||||
| } | } | ||||||
|  |  | ||||||
| func (c *defaultConfig) Watch(ctx context.Context, opts ...options.Option) (Watcher, error) { | func (c *defaultConfig) Watch(_ context.Context, _ ...WatchOption) (Watcher, error) { | ||||||
| 	return nil, ErrWatcherNotImplemented | 	return nil, ErrWatcherNotImplemented | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewConfig returns new default config source | // NewConfig returns new default config source | ||||||
| func NewConfig(opts ...options.Option) Config { | func NewConfig(opts ...Option) Config { | ||||||
| 	options := NewOptions(opts...) | 	options := NewOptions(opts...) | ||||||
| 	if len(options.StructTag) == 0 { | 	if len(options.StructTag) == 0 { | ||||||
| 		options.StructTag = "default" | 		options.StructTag = "default" | ||||||
| 	} | 	} | ||||||
| 	return &defaultConfig{opts: options} | 	c := &defaultConfig{opts: options} | ||||||
|  | 	c.funcLoad = c.fnLoad | ||||||
|  | 	c.funcSave = c.fnSave | ||||||
|  |  | ||||||
|  | 	return c | ||||||
| } | } | ||||||
|   | |||||||
| @@ -3,24 +3,26 @@ package config_test | |||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"fmt" | 	"fmt" | ||||||
|  | 	"reflect" | ||||||
| 	"testing" | 	"testing" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/config" | 	"go.unistack.org/micro/v3/config" | ||||||
| 	mid "go.unistack.org/micro/v4/util/id" | 	mtime "go.unistack.org/micro/v3/util/time" | ||||||
| 	mtime "go.unistack.org/micro/v4/util/time" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type cfg struct { | type cfg struct { | ||||||
|  | 	MapValue    map[string]bool `default:"key1=true,key2=false"` | ||||||
|  | 	StructValue *cfgStructValue | ||||||
|  |  | ||||||
| 	StringValue string `default:"string_value"` | 	StringValue string `default:"string_value"` | ||||||
| 	IgnoreValue string `json:"-"` | 	IgnoreValue string `json:"-"` | ||||||
| 	StructValue    *cfgStructValue |  | ||||||
| 	IntValue       int             `default:"99"` |  | ||||||
| 	DurationValue  time.Duration   `default:"10s"` |  | ||||||
| 	MDurationValue mtime.Duration  `default:"10s"` |  | ||||||
| 	MapValue       map[string]bool `default:"key1=true,key2=false"` |  | ||||||
| 	UUIDValue   string `default:"micro:generate uuid"` | 	UUIDValue   string `default:"micro:generate uuid"` | ||||||
| 	IDValue     string `default:"micro:generate id"` | 	IDValue     string `default:"micro:generate id"` | ||||||
|  |  | ||||||
|  | 	DurationValue  time.Duration  `default:"10s"` | ||||||
|  | 	MDurationValue mtime.Duration `default:"10s"` | ||||||
|  | 	IntValue       int            `default:"99"` | ||||||
| } | } | ||||||
|  |  | ||||||
| type cfgStructValue struct { | type cfgStructValue struct { | ||||||
| @@ -41,6 +43,35 @@ func (c *cfgStructValue) Validate() error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
|  | type testHook struct { | ||||||
|  | 	f bool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *testHook) Load(fn config.FuncLoad) config.FuncLoad { | ||||||
|  | 	return func(ctx context.Context, opts ...config.LoadOption) error { | ||||||
|  | 		t.f = true | ||||||
|  | 		return fn(ctx, opts...) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestHook(t *testing.T) { | ||||||
|  | 	h := &testHook{} | ||||||
|  |  | ||||||
|  | 	c := config.NewConfig(config.Struct(h), config.Hooks(config.HookLoad(h.Load))) | ||||||
|  |  | ||||||
|  | 	if err := c.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if err := c.Load(context.TODO()); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !h.f { | ||||||
|  | 		t.Fatal("hook not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| func TestDefault(t *testing.T) { | func TestDefault(t *testing.T) { | ||||||
| 	ctx := context.Background() | 	ctx := context.Background() | ||||||
| 	conf := &cfg{IntValue: 10} | 	conf := &cfg{IntValue: 10} | ||||||
| @@ -83,8 +114,6 @@ func TestDefault(t *testing.T) { | |||||||
|  |  | ||||||
| 	if conf.IDValue == "" { | 	if conf.IDValue == "" { | ||||||
| 		t.Fatalf("id value empty") | 		t.Fatalf("id value empty") | ||||||
| 	} else if len(conf.IDValue) != mid.DefaultSize { |  | ||||||
| 		t.Fatalf("id value invalid: %s", conf.IDValue) |  | ||||||
| 	} | 	} | ||||||
| 	_ = conf | 	_ = conf | ||||||
| 	// t.Logf("%#+v\n", conf) | 	// t.Logf("%#+v\n", conf) | ||||||
| @@ -105,3 +134,13 @@ func TestValidate(t *testing.T) { | |||||||
| 		t.Fatal(err) | 		t.Fatal(err) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func Test_SizeOf(t *testing.T) { | ||||||
|  | 	st := cfg{} | ||||||
|  |  | ||||||
|  | 	tVal := reflect.TypeOf(st) | ||||||
|  | 	for i := 0; i < tVal.NumField(); i++ { | ||||||
|  | 		field := tVal.Field(i) | ||||||
|  | 		fmt.Printf("Field: %s, Offset: %d, Size: %d\n", field.Name, field.Offset, field.Type.Size()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -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 | ||||||
| @@ -41,12 +41,23 @@ type Options struct { | |||||||
| 	BeforeInit []func(context.Context, Config) error | 	BeforeInit []func(context.Context, Config) error | ||||||
| 	// AfterInit contains slice of funcs that runs after Init | 	// AfterInit contains slice of funcs that runs after Init | ||||||
| 	AfterInit []func(context.Context, Config) error | 	AfterInit []func(context.Context, Config) error | ||||||
|  |  | ||||||
|  | 	// SkipLoad runs only if condition returns true | ||||||
|  | 	SkipLoad func(context.Context, Config) bool | ||||||
|  | 	// SkipSave runs only if condition returns true | ||||||
|  | 	SkipSave func(context.Context, Config) bool | ||||||
|  | 	// Hooks can be run before/after config Save/Load | ||||||
|  | 	Hooks options.Hooks | ||||||
|  |  | ||||||
| 	// AllowFail flag to allow fail in config source | 	// AllowFail flag to allow fail in config source | ||||||
| 	AllowFail bool | 	AllowFail bool | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // Option function signature | ||||||
|  | type Option func(o *Options) | ||||||
|  |  | ||||||
| // NewOptions new options struct with filed values | // NewOptions new options struct with filed values | ||||||
| func NewOptions(opts ...options.Option) Options { | func NewOptions(opts ...Option) Options { | ||||||
| 	options := Options{ | 	options := Options{ | ||||||
| 		Logger:  logger.DefaultLogger, | 		Logger:  logger.DefaultLogger, | ||||||
| 		Meter:   meter.DefaultMeter, | 		Meter:   meter.DefaultMeter, | ||||||
| @@ -60,16 +71,19 @@ func NewOptions(opts ...options.Option) Options { | |||||||
| 	return options | 	return options | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // LoadOption function signature | ||||||
|  | type LoadOption func(o *LoadOptions) | ||||||
|  |  | ||||||
| // LoadOptions struct | // LoadOptions struct | ||||||
| type LoadOptions struct { | type LoadOptions struct { | ||||||
| 	Struct   interface{} | 	Struct   interface{} | ||||||
|  | 	Context  context.Context | ||||||
| 	Override bool | 	Override bool | ||||||
| 	Append   bool | 	Append   bool | ||||||
| 	Context  context.Context |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewLoadOptions create LoadOptions struct with provided opts | // NewLoadOptions create LoadOptions struct with provided opts | ||||||
| func NewLoadOptions(opts ...options.Option) LoadOptions { | func NewLoadOptions(opts ...LoadOption) LoadOptions { | ||||||
| 	options := LoadOptions{} | 	options := LoadOptions{} | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&options) | 		o(&options) | ||||||
| @@ -78,27 +92,44 @@ func NewLoadOptions(opts ...options.Option) LoadOptions { | |||||||
| } | } | ||||||
|  |  | ||||||
| // LoadOverride override values when load | // LoadOverride override values when load | ||||||
| func LoadOverride(b bool) options.Option { | func LoadOverride(b bool) LoadOption { | ||||||
| 	return func(src interface{}) error { | 	return func(o *LoadOptions) { | ||||||
| 		return options.Set(src, b, ".Override") | 		o.Override = b | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // LoadAppend override values when load | // LoadAppend override values when load | ||||||
| func LoadAppend(b bool) options.Option { | func LoadAppend(b bool) LoadOption { | ||||||
| 	return func(src interface{}) error { | 	return func(o *LoadOptions) { | ||||||
| 		return options.Set(src, b, ".Append") | 		o.Append = b | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // LoadStruct override struct for loading | ||||||
|  | func LoadStruct(src interface{}) LoadOption { | ||||||
|  | 	return func(o *LoadOptions) { | ||||||
|  | 		o.Struct = src | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SaveOption function signature | ||||||
|  | type SaveOption func(o *SaveOptions) | ||||||
|  |  | ||||||
| // SaveOptions struct | // SaveOptions struct | ||||||
| type SaveOptions struct { | type SaveOptions struct { | ||||||
| 	Struct  interface{} | 	Struct  interface{} | ||||||
| 	Context context.Context | 	Context context.Context | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // SaveStruct override struct for save to config | ||||||
|  | func SaveStruct(src interface{}) SaveOption { | ||||||
|  | 	return func(o *SaveOptions) { | ||||||
|  | 		o.Struct = src | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // NewSaveOptions fill SaveOptions struct | // NewSaveOptions fill SaveOptions struct | ||||||
| func NewSaveOptions(opts ...options.Option) SaveOptions { | func NewSaveOptions(opts ...SaveOption) SaveOptions { | ||||||
| 	options := SaveOptions{} | 	options := SaveOptions{} | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&options) | 		o(&options) | ||||||
| @@ -107,65 +138,100 @@ func NewSaveOptions(opts ...options.Option) SaveOptions { | |||||||
| } | } | ||||||
|  |  | ||||||
| // AllowFail allows config source to fail | // AllowFail allows config source to fail | ||||||
| func AllowFail(b bool) options.Option { | func AllowFail(b bool) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, b, ".AllowFail") | 		o.AllowFail = b | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // BeforeInit run funcs before config Init | // BeforeInit run funcs before config Init | ||||||
| func BeforeInit(fn ...func(context.Context, Config) error) options.Option { | func BeforeInit(fn ...func(context.Context, Config) error) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, fn, ".BeforeInit") | 		o.BeforeInit = fn | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // AfterInit run funcs after config Init | // AfterInit run funcs after config Init | ||||||
| func AfterInit(fn ...func(context.Context, Config) error) options.Option { | func AfterInit(fn ...func(context.Context, Config) error) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, fn, ".AfterInit") | 		o.AfterInit = fn | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // BeforeLoad run funcs before config load | // BeforeLoad run funcs before config load | ||||||
| func BeforeLoad(fn ...func(context.Context, Config) error) options.Option { | func BeforeLoad(fn ...func(context.Context, Config) error) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, fn, ".BeforeLoad") | 		o.BeforeLoad = fn | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // AfterLoad run funcs after config load | // AfterLoad run funcs after config load | ||||||
| func AfterLoad(fn ...func(context.Context, Config) error) options.Option { | func AfterLoad(fn ...func(context.Context, Config) error) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, fn, ".AfterLoad") | 		o.AfterLoad = fn | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // BeforeSave run funcs before save | // BeforeSave run funcs before save | ||||||
| func BeforeSave(fn ...func(context.Context, Config) error) options.Option { | func BeforeSave(fn ...func(context.Context, Config) error) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, fn, ".BeforeSave") | 		o.BeforeSave = fn | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // AfterSave run fncs after save | // AfterSave run fncs after save | ||||||
| func AfterSave(fn ...func(context.Context, Config) error) options.Option { | func AfterSave(fn ...func(context.Context, Config) error) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, fn, ".AfterSave") | 		o.AfterSave = fn | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Context pass context | ||||||
|  | func Context(ctx context.Context) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Context = ctx | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Codec sets the source codec | ||||||
|  | func Codec(c codec.Codec) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Codec = c | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Logger sets the logger | ||||||
|  | func Logger(l logger.Logger) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Logger = l | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Tracer to be used for tracing | ||||||
|  | func Tracer(t tracer.Tracer) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Tracer = t | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // Struct used as config | // Struct used as config | ||||||
| func Struct(v interface{}) options.Option { | func Struct(v interface{}) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, v, ".Struct") | 		o.Struct = v | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // StructTag sets the struct tag that used for filling | // StructTag sets the struct tag that used for filling | ||||||
| func StructTag(name string) options.Option { | func StructTag(name string) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, name, ".StructTag") | 		o.StructTag = name | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Name sets the name | ||||||
|  | func Name(n string) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Name = n | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -183,8 +249,11 @@ type WatchOptions struct { | |||||||
| 	Coalesce bool | 	Coalesce bool | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // WatchOption func signature | ||||||
|  | type WatchOption func(*WatchOptions) | ||||||
|  |  | ||||||
| // NewWatchOptions create WatchOptions struct with provided opts | // NewWatchOptions create WatchOptions struct with provided opts | ||||||
| func NewWatchOptions(opts ...options.Option) WatchOptions { | func NewWatchOptions(opts ...WatchOption) WatchOptions { | ||||||
| 	options := WatchOptions{ | 	options := WatchOptions{ | ||||||
| 		Context:     context.Background(), | 		Context:     context.Background(), | ||||||
| 		MinInterval: DefaultWatcherMinInterval, | 		MinInterval: DefaultWatcherMinInterval, | ||||||
| @@ -196,20 +265,38 @@ func NewWatchOptions(opts ...options.Option) WatchOptions { | |||||||
| 	return options | 	return options | ||||||
| } | } | ||||||
|  |  | ||||||
| // Coalesce controls watch event combining | // WatchContext pass context | ||||||
| func WatchCoalesce(b bool) options.Option { | func WatchContext(ctx context.Context) WatchOption { | ||||||
| 	return func(src interface{}) error { | 	return func(o *WatchOptions) { | ||||||
| 		return options.Set(src, b, ".Coalesce") | 		o.Context = ctx | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WatchCoalesce controls watch event combining | ||||||
|  | func WatchCoalesce(b bool) WatchOption { | ||||||
|  | 	return func(o *WatchOptions) { | ||||||
|  | 		o.Coalesce = b | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // WatchInterval specifies min and max time.Duration for pulling changes | // WatchInterval specifies min and max time.Duration for pulling changes | ||||||
| func WatchInterval(min, max time.Duration) options.Option { | func WatchInterval(minTime, maxTime time.Duration) WatchOption { | ||||||
| 	return func(src interface{}) error { | 	return func(o *WatchOptions) { | ||||||
| 		var err error | 		o.MinInterval = minTime | ||||||
| 		if err = options.Set(src, min, ".MinInterval"); err == nil { | 		o.MaxInterval = maxTime | ||||||
| 			err = options.Set(src, max, ".MaxInterval") | 	} | ||||||
| 		} | } | ||||||
| 		return err |  | ||||||
|  | // WatchStruct overrides struct for fill | ||||||
|  | func WatchStruct(src interface{}) WatchOption { | ||||||
|  | 	return func(o *WatchOptions) { | ||||||
|  | 		o.Struct = src | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Hooks sets hook runs before action | ||||||
|  | func Hooks(h ...options.Hook) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Hooks = append(o.Hooks, h...) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										157
									
								
								database/dsn.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										157
									
								
								database/dsn.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,157 @@ | |||||||
|  | package database | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"crypto/tls" | ||||||
|  | 	"errors" | ||||||
|  | 	"fmt" | ||||||
|  | 	"net/url" | ||||||
|  | 	"strings" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | var ( | ||||||
|  | 	ErrInvalidDSNAddr      = errors.New("invalid dsn addr") | ||||||
|  | 	ErrInvalidDSNUnescaped = errors.New("dsn must be escaped") | ||||||
|  | 	ErrInvalidDSNNoSlash   = errors.New("dsn must contains slash") | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type Config struct { | ||||||
|  | 	TLSConfig *tls.Config | ||||||
|  | 	Username  string | ||||||
|  | 	Password  string | ||||||
|  | 	Scheme    string | ||||||
|  | 	Host      string | ||||||
|  | 	Port      string | ||||||
|  | 	Database  string | ||||||
|  | 	Params    []string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (cfg *Config) FormatDSN() string { | ||||||
|  | 	var s strings.Builder | ||||||
|  |  | ||||||
|  | 	if len(cfg.Scheme) > 0 { | ||||||
|  | 		s.WriteString(cfg.Scheme + "://") | ||||||
|  | 	} | ||||||
|  | 	// [username[:password]@] | ||||||
|  | 	if len(cfg.Username) > 0 { | ||||||
|  | 		s.WriteString(cfg.Username) | ||||||
|  | 		if len(cfg.Password) > 0 { | ||||||
|  | 			s.WriteByte(':') | ||||||
|  | 			s.WriteString(url.PathEscape(cfg.Password)) | ||||||
|  | 		} | ||||||
|  | 		s.WriteByte('@') | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// [host:port] | ||||||
|  | 	if len(cfg.Host) > 0 { | ||||||
|  | 		s.WriteString(cfg.Host) | ||||||
|  | 		if len(cfg.Port) > 0 { | ||||||
|  | 			s.WriteByte(':') | ||||||
|  | 			s.WriteString(cfg.Port) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// /dbname | ||||||
|  | 	s.WriteByte('/') | ||||||
|  | 	s.WriteString(url.PathEscape(cfg.Database)) | ||||||
|  |  | ||||||
|  | 	for i := 0; i < len(cfg.Params); i += 2 { | ||||||
|  | 		if i == 0 { | ||||||
|  | 			s.WriteString("?") | ||||||
|  | 		} else { | ||||||
|  | 			s.WriteString("&") | ||||||
|  | 		} | ||||||
|  | 		s.WriteString(cfg.Params[i]) | ||||||
|  | 		s.WriteString("=") | ||||||
|  | 		s.WriteString(cfg.Params[i+1]) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return s.String() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func ParseDSN(dsn string) (*Config, error) { | ||||||
|  | 	cfg := &Config{} | ||||||
|  |  | ||||||
|  | 	// [user[:password]@][net[(addr)]]/dbname[?param1=value1¶mN=valueN] | ||||||
|  | 	// Find last '/' that goes before dbname | ||||||
|  | 	foundSlash := false | ||||||
|  | 	for i := len(dsn) - 1; i >= 0; i-- { | ||||||
|  | 		if dsn[i] == '/' { | ||||||
|  | 			foundSlash = true | ||||||
|  | 			var j, k int | ||||||
|  |  | ||||||
|  | 			// left part is empty if i <= 0 | ||||||
|  | 			if i > 0 { | ||||||
|  | 				// Find the first ':' in dsn | ||||||
|  | 				for j = i; j >= 0; j-- { | ||||||
|  | 					if dsn[j] == ':' { | ||||||
|  | 						cfg.Scheme = dsn[0:j] | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 				// [username[:password]@][host] | ||||||
|  | 				// Find the last '@' in dsn[:i] | ||||||
|  | 				for j = i; j >= 0; j-- { | ||||||
|  | 					if dsn[j] == '@' { | ||||||
|  | 						// username[:password] | ||||||
|  | 						// Find the second ':' in dsn[:j] | ||||||
|  | 						for k = 0; k < j; k++ { | ||||||
|  | 							if dsn[k] == ':' { | ||||||
|  | 								if cfg.Scheme == dsn[:k] { | ||||||
|  | 									continue | ||||||
|  | 								} | ||||||
|  | 								var err error | ||||||
|  | 								cfg.Password, err = url.PathUnescape(dsn[k+1 : j]) | ||||||
|  | 								if err != nil { | ||||||
|  | 									return nil, err | ||||||
|  | 								} | ||||||
|  | 								break | ||||||
|  | 							} | ||||||
|  | 						} | ||||||
|  | 						cfg.Username = dsn[len(cfg.Scheme)+3 : k] | ||||||
|  | 						break | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 				for k = j + 1; k < i; k++ { | ||||||
|  | 					if dsn[k] == ':' { | ||||||
|  | 						cfg.Host = dsn[j+1 : k] | ||||||
|  | 						cfg.Port = dsn[k+1 : i] | ||||||
|  | 						break | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			// dbname[?param1=value1&...¶mN=valueN] | ||||||
|  | 			// Find the first '?' in dsn[i+1:] | ||||||
|  | 			for j = i + 1; j < len(dsn); j++ { | ||||||
|  | 				if dsn[j] == '?' { | ||||||
|  | 					parts := strings.Split(dsn[j+1:], "&") | ||||||
|  | 					cfg.Params = make([]string, 0, len(parts)*2) | ||||||
|  | 					for _, p := range parts { | ||||||
|  | 						k, v, found := strings.Cut(p, "=") | ||||||
|  | 						if !found { | ||||||
|  | 							continue | ||||||
|  | 						} | ||||||
|  | 						cfg.Params = append(cfg.Params, k, v) | ||||||
|  | 					} | ||||||
|  |  | ||||||
|  | 					break | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			var err error | ||||||
|  | 			dbname := dsn[i+1 : j] | ||||||
|  | 			if cfg.Database, err = url.PathUnescape(dbname); err != nil { | ||||||
|  | 				return nil, fmt.Errorf("invalid dbname %q: %w", dbname, err) | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			break | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !foundSlash && len(dsn) > 0 { | ||||||
|  | 		return nil, ErrInvalidDSNNoSlash | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return cfg, nil | ||||||
|  | } | ||||||
							
								
								
									
										31
									
								
								database/dsn_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										31
									
								
								database/dsn_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,31 @@ | |||||||
|  | package database | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"net/url" | ||||||
|  | 	"testing" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func TestParseDSN(t *testing.T) { | ||||||
|  | 	cfg, err := ParseDSN("postgres://username:p@ssword#@host:12345/dbname?key1=val2&key2=val2") | ||||||
|  | 	if err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  | 	if cfg.Password != "p@ssword#" { | ||||||
|  | 		t.Fatalf("parsing error") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestFormatDSN(t *testing.T) { | ||||||
|  | 	src := "postgres://username:p@ssword#@host:12345/dbname?key1=val2&key2=val2" | ||||||
|  | 	cfg, err := ParseDSN(src) | ||||||
|  | 	if err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  | 	dst, err := url.PathUnescape(cfg.FormatDSN()) | ||||||
|  | 	if err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  | 	if src != dst { | ||||||
|  | 		t.Fatalf("\n%s\n%s", src, dst) | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										158
									
								
								errors/errors.go
									
									
									
									
									
								
							
							
						
						
									
										158
									
								
								errors/errors.go
									
									
									
									
									
								
							| @@ -1,14 +1,20 @@ | |||||||
| // Package errors provides a way to return detailed information | // Package errors provides a way to return detailed information | ||||||
| // for an RPC request error. The error is normally JSON encoded. | // for an RPC request error. The error is normally JSON encoded. | ||||||
| package errors // import "go.unistack.org/micro/v4/errors" | package errors | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"bytes" | 	"bytes" | ||||||
|  | 	"context" | ||||||
| 	"encoding/json" | 	"encoding/json" | ||||||
|  | 	"errors" | ||||||
| 	"fmt" | 	"fmt" | ||||||
|  | 	"io" | ||||||
| 	"net/http" | 	"net/http" | ||||||
| 	"strconv" | 	"strconv" | ||||||
| 	"strings" | 	"strings" | ||||||
|  |  | ||||||
|  | 	"google.golang.org/grpc/codes" | ||||||
|  | 	"google.golang.org/grpc/status" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| @@ -38,6 +44,20 @@ var ( | |||||||
| 	ErrGatewayTimeout = &Error{Code: 504} | 	ErrGatewayTimeout = &Error{Code: 504} | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | const ProblemContentType = "application/problem+json" | ||||||
|  |  | ||||||
|  | type Problem struct { | ||||||
|  | 	Type     string `json:"type,omitempty"` | ||||||
|  | 	Title    string `json:"title,omitempty"` | ||||||
|  | 	Detail   string `json:"detail,omitempty"` | ||||||
|  | 	Instance string `json:"instance,omitempty"` | ||||||
|  | 	Errors   []struct { | ||||||
|  | 		Title  string `json:"title,omitempty"` | ||||||
|  | 		Detail string `json:"detail,omitempty"` | ||||||
|  | 	} `json:"errors,omitempty"` | ||||||
|  | 	Status int `json:"status,omitempty"` | ||||||
|  | } | ||||||
|  |  | ||||||
| // Error type | // Error type | ||||||
| type Error struct { | type Error struct { | ||||||
| 	// ID holds error id or service, usually someting like my_service or id | 	// ID holds error id or service, usually someting like my_service or id | ||||||
| @@ -256,6 +276,10 @@ func CodeIn(err interface{}, codes ...int32) bool { | |||||||
|  |  | ||||||
| // FromError try to convert go error to *Error | // FromError try to convert go error to *Error | ||||||
| func FromError(err error) *Error { | func FromError(err error) *Error { | ||||||
|  | 	if err == nil { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	if verr, ok := err.(*Error); ok && verr != nil { | 	if verr, ok := err.(*Error); ok && verr != nil { | ||||||
| 		return verr | 		return verr | ||||||
| 	} | 	} | ||||||
| @@ -340,3 +364,135 @@ func addslashes(str string) string { | |||||||
| 	} | 	} | ||||||
| 	return buf.String() | 	return buf.String() | ||||||
| } | } | ||||||
|  |  | ||||||
|  | type retryableError struct { | ||||||
|  | 	err error | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Retryable returns error that can be retried later | ||||||
|  | func Retryable(err error) error { | ||||||
|  | 	return &retryableError{err: err} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type IsRetryableFunc func(error) bool | ||||||
|  |  | ||||||
|  | var ( | ||||||
|  | 	RetrayableOracleErrors = []IsRetryableFunc{ | ||||||
|  | 		func(err error) bool { | ||||||
|  | 			errmsg := err.Error() | ||||||
|  | 			switch { | ||||||
|  | 			case strings.Contains(errmsg, `ORA-`): | ||||||
|  | 				return true | ||||||
|  | 			case strings.Contains(errmsg, `can not assign`): | ||||||
|  | 				return true | ||||||
|  | 			case strings.Contains(errmsg, `can't assign`): | ||||||
|  | 				return true | ||||||
|  | 			} | ||||||
|  | 			return false | ||||||
|  | 		}, | ||||||
|  | 	} | ||||||
|  | 	RetrayablePostgresErrors = []IsRetryableFunc{ | ||||||
|  | 		func(err error) bool { | ||||||
|  | 			errmsg := err.Error() | ||||||
|  | 			switch { | ||||||
|  | 			case strings.Contains(errmsg, `number of field descriptions must equal number of`): | ||||||
|  | 				return true | ||||||
|  | 			case strings.Contains(errmsg, `not a pointer`): | ||||||
|  | 				return true | ||||||
|  | 			case strings.Contains(errmsg, `values, but dst struct has only`): | ||||||
|  | 				return true | ||||||
|  | 			case strings.Contains(errmsg, `struct doesn't have corresponding row field`): | ||||||
|  | 				return true | ||||||
|  | 			case strings.Contains(errmsg, `cannot find field`): | ||||||
|  | 				return true | ||||||
|  | 			case strings.Contains(errmsg, `cannot scan`) || strings.Contains(errmsg, `cannot convert`): | ||||||
|  | 				return true | ||||||
|  | 			case strings.Contains(errmsg, `failed to connect to`): | ||||||
|  | 				return true | ||||||
|  | 			} | ||||||
|  | 			return false | ||||||
|  | 		}, | ||||||
|  | 	} | ||||||
|  | 	RetryableMicroErrors = []IsRetryableFunc{ | ||||||
|  | 		func(err error) bool { | ||||||
|  | 			switch verr := err.(type) { | ||||||
|  | 			case *Error: | ||||||
|  | 				switch verr.Code { | ||||||
|  | 				case 401, 403, 408, 500, 501, 502, 503, 504: | ||||||
|  | 					return true | ||||||
|  | 				default: | ||||||
|  | 					return false | ||||||
|  | 				} | ||||||
|  | 			case *retryableError: | ||||||
|  | 				return true | ||||||
|  | 			} | ||||||
|  | 			return false | ||||||
|  | 		}, | ||||||
|  | 	} | ||||||
|  | 	RetryableGoErrors = []IsRetryableFunc{ | ||||||
|  | 		func(err error) bool { | ||||||
|  | 			switch verr := err.(type) { | ||||||
|  | 			case interface{ SafeToRetry() bool }: | ||||||
|  | 				return verr.SafeToRetry() | ||||||
|  | 			case interface{ Timeout() bool }: | ||||||
|  | 				return verr.Timeout() | ||||||
|  | 			} | ||||||
|  | 			switch { | ||||||
|  | 			case errors.Is(err, io.EOF), errors.Is(err, io.ErrUnexpectedEOF): | ||||||
|  | 				return true | ||||||
|  | 			case errors.Is(err, context.DeadlineExceeded): | ||||||
|  | 				return true | ||||||
|  | 			case errors.Is(err, io.ErrClosedPipe), errors.Is(err, io.ErrShortBuffer), errors.Is(err, io.ErrShortWrite): | ||||||
|  | 				return true | ||||||
|  | 			} | ||||||
|  | 			return false | ||||||
|  | 		}, | ||||||
|  | 	} | ||||||
|  | 	RetryableGrpcErrors = []IsRetryableFunc{ | ||||||
|  | 		func(err error) bool { | ||||||
|  | 			st, ok := status.FromError(err) | ||||||
|  | 			if !ok { | ||||||
|  | 				return false | ||||||
|  | 			} | ||||||
|  | 			switch st.Code() { | ||||||
|  | 			case codes.Unavailable, codes.ResourceExhausted: | ||||||
|  | 				return true | ||||||
|  | 			case codes.DeadlineExceeded: | ||||||
|  | 				return true | ||||||
|  | 			case codes.Internal: | ||||||
|  | 				switch { | ||||||
|  | 				case strings.Contains(st.Message(), `transport: received the unexpected content-type "text/html; charset=UTF-8"`): | ||||||
|  | 					return true | ||||||
|  | 				case strings.Contains(st.Message(), io.ErrUnexpectedEOF.Error()): | ||||||
|  | 					return true | ||||||
|  | 				case strings.Contains(st.Message(), `stream terminated by RST_STREAM with error code: INTERNAL_ERROR`): | ||||||
|  | 					return true | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			return false | ||||||
|  | 		}, | ||||||
|  | 	} | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Unwrap provides error wrapping | ||||||
|  | func (e *retryableError) Unwrap() error { | ||||||
|  | 	return e.err | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Error returns the error string | ||||||
|  | func (e *retryableError) Error() string { | ||||||
|  | 	if e.err == nil { | ||||||
|  | 		return "" | ||||||
|  | 	} | ||||||
|  | 	return e.err.Error() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // IsRetryable checks error for ability to retry later | ||||||
|  | func IsRetryable(err error, fns ...IsRetryableFunc) bool { | ||||||
|  | 	for _, fn := range fns { | ||||||
|  | 		if ok := fn(err); ok { | ||||||
|  | 			return true | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|   | |||||||
| @@ -1,4 +1,4 @@ | |||||||
| // Copyright 2021-2023 Unistack LLC | // Copyright 2021 Unistack LLC | ||||||
| // | // | ||||||
| // Licensed under the Apache License, Version 2.0 (the "License"); | // Licensed under the Apache License, Version 2.0 (the "License"); | ||||||
| // you may not use this file except in compliance with the License. | // you may not use this file except in compliance with the License. | ||||||
| @@ -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"; | ||||||
|   | |||||||
| @@ -2,12 +2,19 @@ package errors | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"encoding/json" | 	"encoding/json" | ||||||
| 	er "errors" | 	"errors" | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"net/http" | 	"net/http" | ||||||
| 	"testing" | 	"testing" | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | func TestIsRetrayable(t *testing.T) { | ||||||
|  | 	err := fmt.Errorf("ORA-") | ||||||
|  | 	if !IsRetryable(err, RetrayableOracleErrors...) { | ||||||
|  | 		t.Fatalf("IsRetrayable not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| func TestMarshalJSON(t *testing.T) { | func TestMarshalJSON(t *testing.T) { | ||||||
| 	e := InternalServerError("id", "err: %v", fmt.Errorf("err: %v", `xxx: "UNIX_TIMESTAMP": invalid identifier`)) | 	e := InternalServerError("id", "err: %v", fmt.Errorf("err: %v", `xxx: "UNIX_TIMESTAMP": invalid identifier`)) | ||||||
| 	_, err := json.Marshal(e) | 	_, err := json.Marshal(e) | ||||||
| @@ -19,7 +26,7 @@ func TestMarshalJSON(t *testing.T) { | |||||||
| func TestEmpty(t *testing.T) { | func TestEmpty(t *testing.T) { | ||||||
| 	msg := "test" | 	msg := "test" | ||||||
| 	var err *Error | 	var err *Error | ||||||
| 	err = FromError(fmt.Errorf(msg)) | 	err = FromError(errors.New(msg)) | ||||||
| 	if err.Detail != msg { | 	if err.Detail != msg { | ||||||
| 		t.Fatalf("invalid error %v", err) | 		t.Fatalf("invalid error %v", err) | ||||||
| 	} | 	} | ||||||
| @@ -35,7 +42,7 @@ func TestFromError(t *testing.T) { | |||||||
| 	if merr.ID != "go.micro.test" || merr.Code != 404 { | 	if merr.ID != "go.micro.test" || merr.Code != 404 { | ||||||
| 		t.Fatalf("invalid conversation %v != %v", err, merr) | 		t.Fatalf("invalid conversation %v != %v", err, merr) | ||||||
| 	} | 	} | ||||||
| 	err = er.New(err.Error()) | 	err = errors.New(err.Error()) | ||||||
| 	merr = FromError(err) | 	merr = FromError(err) | ||||||
| 	if merr.ID != "go.micro.test" || merr.Code != 404 { | 	if merr.ID != "go.micro.test" || merr.Code != 404 { | ||||||
| 		t.Fatalf("invalid conversation %v != %v", err, merr) | 		t.Fatalf("invalid conversation %v != %v", err, merr) | ||||||
| @@ -50,7 +57,7 @@ func TestEqual(t *testing.T) { | |||||||
| 		t.Fatal("errors must be equal") | 		t.Fatal("errors must be equal") | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	err3 := er.New("my test err") | 	err3 := errors.New("my test err") | ||||||
| 	if Equal(err1, err3) { | 	if Equal(err1, err3) { | ||||||
| 		t.Fatal("errors must be not equal") | 		t.Fatal("errors must be not equal") | ||||||
| 	} | 	} | ||||||
|   | |||||||
							
								
								
									
										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 { | ||||||
| @@ -22,3 +31,13 @@ func NewContext(ctx context.Context, f Flow) context.Context { | |||||||
| 	} | 	} | ||||||
| 	return context.WithValue(ctx, flowKey{}, f) | 	return context.WithValue(ctx, flowKey{}, f) | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // 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) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -40,3 +40,14 @@ func TestNewContext(t *testing.T) { | |||||||
| 		t.Fatal("NewContext not works") | 		t.Fatal("NewContext not works") | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func TestSetOption(t *testing.T) { | ||||||
|  | 	type key struct{} | ||||||
|  | 	o := SetOption(key{}, "test") | ||||||
|  | 	opts := &Options{} | ||||||
|  | 	o(opts) | ||||||
|  |  | ||||||
|  | 	if v, ok := opts.Context.Value(key{}).(string); !ok || v == "" { | ||||||
|  | 		t.Fatal("SetOption not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -6,14 +6,12 @@ import ( | |||||||
| 	"sync" | 	"sync" | ||||||
|  |  | ||||||
| 	"github.com/silas/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/logger" | 	"go.unistack.org/micro/v3/logger" | ||||||
| 	"go.unistack.org/micro/v4/metadata" | 	"go.unistack.org/micro/v3/metadata" | ||||||
| 	"go.unistack.org/micro/v4/options" | 	"go.unistack.org/micro/v3/store" | ||||||
| 	moptions "go.unistack.org/micro/v4/options" | 	"go.unistack.org/micro/v3/util/id" | ||||||
| 	"go.unistack.org/micro/v4/store" |  | ||||||
| 	"go.unistack.org/micro/v4/util/id" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type microFlow struct { | type microFlow struct { | ||||||
| @@ -165,7 +163,7 @@ func (w *microWorkflow) Resume(ctx context.Context, id string) error { | |||||||
| 	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 ...options.Option) (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 { | ||||||
| 		if err := w.g.Validate(); err != nil { | 		if err := w.g.Validate(); err != nil { | ||||||
| @@ -190,7 +188,7 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...optio | |||||||
| 	steps, err := w.getSteps(options.Start, options.Reverse) | 	steps, err := w.getSteps(options.Start, options.Reverse) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		if werr := workflowStore.Write(w.opts.Context, "status", &codec.Frame{Data: []byte(StatusPending.String())}); werr != nil { | 		if werr := workflowStore.Write(w.opts.Context, "status", &codec.Frame{Data: []byte(StatusPending.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 "", err | ||||||
| 	} | 	} | ||||||
| @@ -202,19 +200,19 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...optio | |||||||
| 	nctx, cancel := context.WithCancel(ctx) | 	nctx, cancel := context.WithCancel(ctx) | ||||||
| 	defer cancel() | 	defer cancel() | ||||||
|  |  | ||||||
| 	nopts := make([]moptions.Option, 0, len(opts)+5) | 	nopts := make([]ExecuteOption, 0, len(opts)+5) | ||||||
|  |  | ||||||
| 	nopts = append(nopts, | 	nopts = append(nopts, | ||||||
| 		moptions.Client(w.opts.Client), | 		ExecuteClient(w.opts.Client), | ||||||
| 		moptions.Tracer(w.opts.Tracer), | 		ExecuteTracer(w.opts.Tracer), | ||||||
| 		moptions.Logger(w.opts.Logger), | 		ExecuteLogger(w.opts.Logger), | ||||||
| 		moptions.Meter(w.opts.Meter), | 		ExecuteMeter(w.opts.Meter), | ||||||
| 	) | 	) | ||||||
| 	nopts = append(nopts, opts...) | 	nopts = append(nopts, opts...) | ||||||
| 	done := make(chan struct{}) | 	done := make(chan struct{}) | ||||||
|  |  | ||||||
| 	if werr := workflowStore.Write(w.opts.Context, "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.Errorf(w.opts.Context, "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 { | 	for idx := range steps { | ||||||
| @@ -239,7 +237,7 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...optio | |||||||
| 					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 | ||||||
| @@ -259,21 +257,21 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...optio | |||||||
| 						if serr != nil { | 						if serr != nil { | ||||||
| 							step.SetStatus(StatusFailure) | 							step.SetStatus(StatusFailure) | ||||||
| 							if werr := stepStore.Write(ctx, step.ID()+w.opts.Store.Options().Separator+"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, step.ID()+w.opts.Store.Options().Separator+"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, step.ID()+w.opts.Store.Options().Separator+"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, step.ID()+w.opts.Store.Options().Separator+"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 | ||||||
| 						} | 						} | ||||||
| @@ -292,16 +290,16 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...optio | |||||||
| 					if serr != nil { | 					if serr != nil { | ||||||
| 						cstep.SetStatus(StatusFailure) | 						cstep.SetStatus(StatusFailure) | ||||||
| 						if werr := stepStore.Write(ctx, cstep.ID()+w.opts.Store.Options().Separator+"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, cstep.ID()+w.opts.Store.Options().Separator+"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, cstep.ID()+w.opts.Store.Options().Separator+"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 | ||||||
| 					} | 					} | ||||||
| @@ -319,7 +317,7 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...optio | |||||||
| 		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() | ||||||
| @@ -335,15 +333,15 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...optio | |||||||
| 	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) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -351,7 +349,7 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...optio | |||||||
| } | } | ||||||
|  |  | ||||||
| // NewFlow create new flow | // NewFlow create new flow | ||||||
| func NewFlow(opts ...options.Option) Flow { | func NewFlow(opts ...Option) Flow { | ||||||
| 	options := NewOptions(opts...) | 	options := NewOptions(opts...) | ||||||
| 	return µFlow{opts: options} | 	return µFlow{opts: options} | ||||||
| } | } | ||||||
| @@ -360,7 +358,7 @@ func (f *microFlow) Options() Options { | |||||||
| 	return f.opts | 	return f.opts | ||||||
| } | } | ||||||
|  |  | ||||||
| func (f *microFlow) Init(opts ...options.Option) error { | func (f *microFlow) Init(opts ...Option) error { | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&f.opts) | 		o(&f.opts) | ||||||
| 	} | 	} | ||||||
| @@ -489,17 +487,17 @@ func (s *microCallStep) SetStatus(status Status) { | |||||||
| 	s.status = status | 	s.status = status | ||||||
| } | } | ||||||
|  |  | ||||||
| func (s *microCallStep) Execute(ctx context.Context, req *Message, opts ...options.Option) (*Message, error) { | func (s *microCallStep) Execute(ctx context.Context, req *Message, opts ...ExecuteOption) (*Message, error) { | ||||||
| 	options := NewExecuteOptions(opts...) | 	options := NewExecuteOptions(opts...) | ||||||
| 	if options.Client == nil { | 	if options.Client == nil { | ||||||
| 		return nil, ErrMissingClient | 		return nil, ErrMissingClient | ||||||
| 	} | 	} | ||||||
| 	rsp := &codec.Frame{} | 	rsp := &codec.Frame{} | ||||||
| 	copts := []moptions.Option{client.Retries(0)} | 	copts := []client.CallOption{client.WithRetries(0)} | ||||||
| 	if options.Timeout > 0 { | 	if options.Timeout > 0 { | ||||||
| 		copts = append(copts, | 		copts = append(copts, | ||||||
| 			client.RequestTimeout(options.Timeout), | 			client.WithRequestTimeout(options.Timeout), | ||||||
| 			client.DialTimeout(options.Timeout)) | 			client.WithDialTimeout(options.Timeout)) | ||||||
| 	} | 	} | ||||||
| 	nctx := metadata.NewOutgoingContext(ctx, req.Header) | 	nctx := metadata.NewOutgoingContext(ctx, req.Header) | ||||||
| 	err := options.Client.Call(nctx, options.Client.NewRequest(s.service, s.method, &codec.Frame{Data: req.Body}), rsp, copts...) | 	err := options.Client.Call(nctx, options.Client.NewRequest(s.service, s.method, &codec.Frame{Data: req.Body}), rsp, copts...) | ||||||
| @@ -572,18 +570,18 @@ func (s *microPublishStep) SetStatus(status Status) { | |||||||
| 	s.status = status | 	s.status = status | ||||||
| } | } | ||||||
|  |  | ||||||
| func (s *microPublishStep) Execute(ctx context.Context, req *Message, opts ...options.Option) (*Message, error) { | func (s *microPublishStep) Execute(ctx context.Context, req *Message, opts ...ExecuteOption) (*Message, error) { | ||||||
| 	return nil, nil | 	return nil, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewCallStep create new step with client.Call | // NewCallStep create new step with client.Call | ||||||
| func NewCallStep(service string, name string, method string, opts ...options.Option) Step { | func NewCallStep(service string, name string, method string, opts ...StepOption) Step { | ||||||
| 	options := NewStepOptions(opts...) | 	options := NewStepOptions(opts...) | ||||||
| 	return µCallStep{service: service, method: name + "." + method, opts: options} | 	return µCallStep{service: service, method: name + "." + method, opts: options} | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewPublishStep create new step with client.Publish | // NewPublishStep create new step with client.Publish | ||||||
| func NewPublishStep(topic string, opts ...options.Option) Step { | func NewPublishStep(topic string, opts ...StepOption) Step { | ||||||
| 	options := NewStepOptions(opts...) | 	options := NewStepOptions(opts...) | ||||||
| 	return µPublishStep{topic: topic, opts: options} | 	return µPublishStep{topic: topic, opts: options} | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										11
									
								
								flow/flow.go
									
									
									
									
									
								
							
							
						
						
									
										11
									
								
								flow/flow.go
									
									
									
									
									
								
							| @@ -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,8 +7,7 @@ import ( | |||||||
| 	"sync" | 	"sync" | ||||||
| 	"sync/atomic" | 	"sync/atomic" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/metadata" | 	"go.unistack.org/micro/v3/metadata" | ||||||
| 	"go.unistack.org/micro/v4/options" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| @@ -52,7 +51,7 @@ type Step interface { | |||||||
| 	// Endpoint returns rpc endpoint service_name.service_method or broker topic | 	// Endpoint returns rpc endpoint service_name.service_method or broker topic | ||||||
| 	Endpoint() string | 	Endpoint() string | ||||||
| 	// Execute step run | 	// Execute step run | ||||||
| 	Execute(ctx context.Context, req *Message, opts ...options.Option) (*Message, error) | 	Execute(ctx context.Context, req *Message, opts ...ExecuteOption) (*Message, error) | ||||||
| 	// Requires returns dependent steps | 	// Requires returns dependent steps | ||||||
| 	Requires() []string | 	Requires() []string | ||||||
| 	// Options returns step options | 	// Options returns step options | ||||||
| @@ -119,7 +118,7 @@ type Workflow interface { | |||||||
| 	// ID returns id of the workflow | 	// ID returns id of the workflow | ||||||
| 	ID() string | 	ID() string | ||||||
| 	// Execute workflow with args, return execution id and error | 	// Execute workflow with args, return execution id and error | ||||||
| 	Execute(ctx context.Context, req *Message, opts ...options.Option) (string, error) | 	Execute(ctx context.Context, req *Message, opts ...ExecuteOption) (string, error) | ||||||
| 	// RemoveSteps remove steps from workflow | 	// RemoveSteps remove steps from workflow | ||||||
| 	RemoveSteps(steps ...Step) error | 	RemoveSteps(steps ...Step) error | ||||||
| 	// AppendSteps append steps to workflow | 	// AppendSteps append steps to workflow | ||||||
| @@ -141,7 +140,7 @@ type Flow interface { | |||||||
| 	// Options returns options | 	// Options returns options | ||||||
| 	Options() Options | 	Options() Options | ||||||
| 	// Init initialize | 	// Init initialize | ||||||
| 	Init(...options.Option) error | 	Init(...Option) error | ||||||
| 	// WorkflowCreate creates new workflow with specific id and steps | 	// WorkflowCreate creates new workflow with specific id and steps | ||||||
| 	WorkflowCreate(ctx context.Context, id string, steps ...Step) (Workflow, error) | 	WorkflowCreate(ctx context.Context, id string, steps ...Step) (Workflow, error) | ||||||
| 	// WorkflowSave saves workflow | 	// WorkflowSave saves workflow | ||||||
|   | |||||||
							
								
								
									
										162
									
								
								flow/options.go
									
									
									
									
									
								
							
							
						
						
									
										162
									
								
								flow/options.go
									
									
									
									
									
								
							| @@ -4,14 +4,16 @@ 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/options" | 	"go.unistack.org/micro/v3/store" | ||||||
| 	"go.unistack.org/micro/v4/store" | 	"go.unistack.org/micro/v3/tracer" | ||||||
| 	"go.unistack.org/micro/v4/tracer" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | // Option func | ||||||
|  | type Option func(*Options) | ||||||
|  |  | ||||||
| // Options server struct | // Options server struct | ||||||
| type Options struct { | type Options struct { | ||||||
| 	// Context holds the external options and can be used for flow shutdown | 	// Context holds the external options and can be used for flow shutdown | ||||||
| @@ -29,7 +31,7 @@ type Options struct { | |||||||
| } | } | ||||||
|  |  | ||||||
| // NewOptions returns new options struct with default or passed values | // NewOptions returns new options struct with default or passed values | ||||||
| func NewOptions(opts ...options.Option) Options { | func NewOptions(opts ...Option) Options { | ||||||
| 	options := Options{ | 	options := Options{ | ||||||
| 		Context: context.Background(), | 		Context: context.Background(), | ||||||
| 		Logger:  logger.DefaultLogger, | 		Logger:  logger.DefaultLogger, | ||||||
| @@ -45,12 +47,66 @@ func NewOptions(opts ...options.Option) Options { | |||||||
| 	return options | 	return options | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // Logger sets the logger option | ||||||
|  | func Logger(l logger.Logger) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Logger = l | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Meter sets the meter option | ||||||
|  | func Meter(m meter.Meter) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Meter = m | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Client to use for sync/async communication | ||||||
|  | func Client(c client.Client) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Client = c | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Context specifies a context for the service. | ||||||
|  | // Can be used to signal shutdown of the flow | ||||||
|  | // or can be used for extra option values. | ||||||
|  | func Context(ctx context.Context) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Context = ctx | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Tracer mechanism for distributed tracking | ||||||
|  | func Tracer(t tracer.Tracer) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Tracer = t | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Store used for intermediate results | ||||||
|  | func Store(s store.Store) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Store = s | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WorkflowOption func signature | ||||||
|  | type WorkflowOption func(*WorkflowOptions) | ||||||
|  |  | ||||||
| // WorkflowOptions holds workflow options | // WorkflowOptions holds workflow options | ||||||
| type WorkflowOptions struct { | type WorkflowOptions struct { | ||||||
| 	Context context.Context | 	Context context.Context | ||||||
| 	ID      string | 	ID      string | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // WorkflowID set workflow id | ||||||
|  | func WorkflowID(id string) WorkflowOption { | ||||||
|  | 	return func(o *WorkflowOptions) { | ||||||
|  | 		o.ID = id | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // ExecuteOptions holds execute options | // ExecuteOptions holds execute options | ||||||
| type ExecuteOptions struct { | type ExecuteOptions struct { | ||||||
| 	// Client holds the client.Client | 	// Client holds the client.Client | ||||||
| @@ -73,22 +129,67 @@ type ExecuteOptions struct { | |||||||
| 	Async bool | 	Async bool | ||||||
| } | } | ||||||
|  |  | ||||||
| // Reverse says that dag must be run in reverse order | // ExecuteOption func signature | ||||||
| func Reverse(b bool) options.Option { | type ExecuteOption func(*ExecuteOptions) | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return options.Set(src, b, ".Reverse") | // ExecuteClient pass client.Client to ExecuteOption | ||||||
|  | func ExecuteClient(c client.Client) ExecuteOption { | ||||||
|  | 	return func(o *ExecuteOptions) { | ||||||
|  | 		o.Client = c | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // Async says that caller does not wait for execution complete | // ExecuteTracer pass tracer.Tracer to ExecuteOption | ||||||
| func Async(b bool) options.Option { | func ExecuteTracer(t tracer.Tracer) ExecuteOption { | ||||||
| 	return func(src interface{}) error { | 	return func(o *ExecuteOptions) { | ||||||
| 		return options.Set(src, b, ".Async") | 		o.Tracer = t | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ExecuteLogger pass logger.Logger to ExecuteOption | ||||||
|  | func ExecuteLogger(l logger.Logger) ExecuteOption { | ||||||
|  | 	return func(o *ExecuteOptions) { | ||||||
|  | 		o.Logger = l | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ExecuteMeter pass meter.Meter to ExecuteOption | ||||||
|  | func ExecuteMeter(m meter.Meter) ExecuteOption { | ||||||
|  | 	return func(o *ExecuteOptions) { | ||||||
|  | 		o.Meter = m | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ExecuteContext pass context.Context ot ExecuteOption | ||||||
|  | func ExecuteContext(ctx context.Context) ExecuteOption { | ||||||
|  | 	return func(o *ExecuteOptions) { | ||||||
|  | 		o.Context = ctx | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // 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 | ||||||
|  | func ExecuteTimeout(td time.Duration) ExecuteOption { | ||||||
|  | 	return func(o *ExecuteOptions) { | ||||||
|  | 		o.Timeout = td | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ExecuteAsync says that caller does not wait for execution complete | ||||||
|  | func ExecuteAsync(b bool) ExecuteOption { | ||||||
|  | 	return func(o *ExecuteOptions) { | ||||||
|  | 		o.Async = b | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewExecuteOptions create new ExecuteOptions struct | // NewExecuteOptions create new ExecuteOptions struct | ||||||
| func NewExecuteOptions(opts ...options.Option) ExecuteOptions { | func NewExecuteOptions(opts ...ExecuteOption) ExecuteOptions { | ||||||
| 	options := ExecuteOptions{ | 	options := ExecuteOptions{ | ||||||
| 		Client:  client.DefaultClient, | 		Client:  client.DefaultClient, | ||||||
| 		Logger:  logger.DefaultLogger, | 		Logger:  logger.DefaultLogger, | ||||||
| @@ -110,8 +211,11 @@ type StepOptions struct { | |||||||
| 	Requires []string | 	Requires []string | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // StepOption func signature | ||||||
|  | type StepOption func(*StepOptions) | ||||||
|  |  | ||||||
| // NewStepOptions create new StepOptions struct | // NewStepOptions create new StepOptions struct | ||||||
| func NewStepOptions(opts ...options.Option) StepOptions { | func NewStepOptions(opts ...StepOption) StepOptions { | ||||||
| 	options := StepOptions{ | 	options := StepOptions{ | ||||||
| 		Context: context.Background(), | 		Context: context.Background(), | ||||||
| 	} | 	} | ||||||
| @@ -121,23 +225,23 @@ func NewStepOptions(opts ...options.Option) StepOptions { | |||||||
| 	return options | 	return options | ||||||
| } | } | ||||||
|  |  | ||||||
| // Requires specifies required steps | // StepID sets the step id for dag | ||||||
| func Requires(steps ...string) options.Option { | func StepID(id string) StepOption { | ||||||
| 	return func(src interface{}) error { | 	return func(o *StepOptions) { | ||||||
| 		return options.Set(src, steps, ".Requires") | 		o.ID = id | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // Fallback set the step to run on error | // StepRequires specifies required steps | ||||||
| func Fallback(step string) options.Option { | func StepRequires(steps ...string) StepOption { | ||||||
| 	return func(src interface{}) error { | 	return func(o *StepOptions) { | ||||||
| 		return options.Set(src, step, ".Fallback") | 		o.Requires = steps | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // ID sets the step ID | // StepFallback set the step to run on error | ||||||
| func StepID(id string) options.Option { | func StepFallback(step string) StepOption { | ||||||
| 	return func(src interface{}) error { | 	return func(o *StepOptions) { | ||||||
| 		return options.Set(src, id, ".ID") | 		o.Fallback = step | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -32,7 +32,7 @@ type fsm struct { | |||||||
|  |  | ||||||
| // NewFSM creates a new finite state machine having the specified initial state | // NewFSM creates a new finite state machine having the specified initial state | ||||||
| // with specified options | // with specified options | ||||||
| func NewFSM(opts ...Option) *fsm { | func NewFSM(opts ...Option) FSM { | ||||||
| 	return &fsm{ | 	return &fsm{ | ||||||
| 		statesMap: map[string]StateFunc{}, | 		statesMap: map[string]StateFunc{}, | ||||||
| 		opts:      NewOptions(opts...), | 		opts:      NewOptions(opts...), | ||||||
|   | |||||||
| @@ -1,4 +1,4 @@ | |||||||
| package fsm // import "go.unistack.org/micro/v4/fsm" | package fsm | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
|   | |||||||
| @@ -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) { | ||||||
| @@ -17,7 +17,7 @@ func TestFSMStart(t *testing.T) { | |||||||
|  |  | ||||||
| 	wrapper := func(next StateFunc) StateFunc { | 	wrapper := func(next StateFunc) StateFunc { | ||||||
| 		return func(sctx context.Context, s State, opts ...StateOption) (State, error) { | 		return func(sctx context.Context, s State, opts ...StateOption) (State, error) { | ||||||
| 			sctx = logger.NewContext(sctx, logger.Fields("state", s.Name())) | 			sctx = logger.NewContext(sctx, logger.DefaultLogger.Fields("state", s.Name())) | ||||||
| 			return next(sctx, s, opts...) | 			return next(sctx, s, opts...) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|   | |||||||
							
								
								
									
										43
									
								
								go.mod
									
									
									
									
									
								
							
							
						
						
									
										43
									
								
								go.mod
									
									
									
									
									
								
							| @@ -1,20 +1,43 @@ | |||||||
| module go.unistack.org/micro/v4 | module go.unistack.org/micro/v3 | ||||||
|  |  | ||||||
| go 1.20 | go 1.23.4 | ||||||
|  |  | ||||||
| require ( | require ( | ||||||
|  | 	dario.cat/mergo v1.0.1 | ||||||
| 	github.com/DATA-DOG/go-sqlmock v1.5.0 | 	github.com/DATA-DOG/go-sqlmock v1.5.0 | ||||||
| 	github.com/google/uuid v1.3.0 | 	github.com/KimMachineGun/automemlimit v0.6.1 | ||||||
| 	github.com/imdario/mergo v0.3.15 | 	github.com/ash3in/uuidv8 v1.0.1 | ||||||
|  | 	github.com/google/uuid v1.6.0 | ||||||
|  | 	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 | ||||||
| 	golang.org/x/sync v0.1.0 | 	go.uber.org/automaxprocs v1.6.0 | ||||||
| 	golang.org/x/sys v0.7.0 | 	go.unistack.org/micro-proto/v3 v3.4.1 | ||||||
| 	google.golang.org/grpc v1.54.0 | 	golang.org/x/sync v0.10.0 | ||||||
| 	google.golang.org/protobuf v1.30.0 | 	google.golang.org/grpc v1.68.1 | ||||||
|  | 	google.golang.org/protobuf v1.35.2 | ||||||
|  | 	gopkg.in/yaml.v3 v3.0.1 | ||||||
| ) | ) | ||||||
|  |  | ||||||
| require ( | require ( | ||||||
| 	github.com/golang/protobuf v1.5.3 // indirect | 	github.com/cilium/ebpf v0.16.0 // indirect | ||||||
| 	google.golang.org/genproto v0.0.0-20230410155749-daa745c078e1 // 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/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/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 // 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 | ||||||
|  | 	go.uber.org/goleak v1.3.0 // indirect | ||||||
|  | 	golang.org/x/exp v0.0.0-20241210194714-1829a127f884 // 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 | ||||||
| ) | ) | ||||||
|   | |||||||
							
								
								
									
										117
									
								
								go.sum
									
									
									
									
									
								
							
							
						
						
									
										117
									
								
								go.sum
									
									
									
									
									
								
							| @@ -1,33 +1,102 @@ | |||||||
|  | dario.cat/mergo v1.0.1 h1:Ra4+bf83h2ztPIQYNP99R6m+Y7KfnARDfID+a+vLl4s= | ||||||
|  | dario.cat/mergo v1.0.1/go.mod h1:uNxQE+84aUszobStD9th8a29P2fMDhsBdgRYvZOxGmk= | ||||||
| github.com/DATA-DOG/go-sqlmock v1.5.0 h1:Shsta01QNfFxHCfpW6YH2STWB0MudeXXEWMr20OEh60= | github.com/DATA-DOG/go-sqlmock v1.5.0 h1:Shsta01QNfFxHCfpW6YH2STWB0MudeXXEWMr20OEh60= | ||||||
| github.com/DATA-DOG/go-sqlmock v1.5.0/go.mod h1:f/Ixk793poVmq4qj/V1dPUg2JEAKC73Q5eFN3EC/SaM= | github.com/DATA-DOG/go-sqlmock v1.5.0/go.mod h1:f/Ixk793poVmq4qj/V1dPUg2JEAKC73Q5eFN3EC/SaM= | ||||||
| github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk= | github.com/KimMachineGun/automemlimit v0.6.1 h1:ILa9j1onAAMadBsyyUJv5cack8Y1WT26yLj/V+ulKp8= | ||||||
| github.com/golang/protobuf v1.5.3 h1:KhyjKVUg7Usr/dYsdSqoFveMYd5ko72D+zANwlG1mmg= | github.com/KimMachineGun/automemlimit v0.6.1/go.mod h1:T7xYht7B8r6AG/AqFcUdc7fzd2bIdBKmepfP2S1svPY= | ||||||
| github.com/golang/protobuf v1.5.3/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY= | github.com/ash3in/uuidv8 v1.0.1 h1:dIq1XRkWT8lGA7N5s7WRTB4V3k49WTBLvILz7aCLp80= | ||||||
| github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= | github.com/ash3in/uuidv8 v1.0.1/go.mod h1:EoyUgCtxNBnrnpc9efw5rVN1cQ+LFGCoJiFuD6maOMw= | ||||||
| github.com/google/go-cmp v0.5.9 h1:O2Tfq5qg4qc4AmwVlvv0oLiVAGB7enBSJ2x2DqQFi38= | github.com/cilium/ebpf v0.16.0 h1:+BiEnHL6Z7lXnlGUsXQPPAE7+kenAd4ES8MQ5min0Ok= | ||||||
| github.com/google/uuid v1.3.0 h1:t6JiXgmwXMjEs8VusXIJk2BXHsn+wx8BZdTaoZ5fu7I= | github.com/cilium/ebpf v0.16.0/go.mod h1:L7u2Blt2jMM/vLAVgjxluxtBKlz3/GWjB0dMOEngfwE= | ||||||
| github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= | github.com/containerd/cgroups/v3 v3.0.4 h1:2fs7l3P0Qxb1nKWuJNFiwhp2CqiKzho71DQkDrHJIo4= | ||||||
| github.com/imdario/mergo v0.3.15 h1:M8XP7IuFNsqUx6VPK2P9OSmsYsI/YFaGil0uD21V3dM= | github.com/containerd/cgroups/v3 v3.0.4/go.mod h1:SA5DLYnXO8pTGYiAHXz94qvLQTKfVM5GEVisn4jpins= | ||||||
| github.com/imdario/mergo v0.3.15/go.mod h1:WBLT9ZmE3lPoWsEzCh9LPo3TiwVN+ZKEjmz+hD27ysY= | 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/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | ||||||
|  | github.com/docker/go-units v0.5.0 h1:69rxXcBk27SvSaaxTtLh/8llcHD8vYHT7WSdRZ/jvr4= | ||||||
|  | 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/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps= | ||||||
|  | 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/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0= | ||||||
|  | github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= | ||||||
|  | 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.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE= | ||||||
|  | github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk= | ||||||
|  | github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= | ||||||
|  | github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= | ||||||
|  | github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY= | ||||||
|  | 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/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/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/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= | ||||||
|  | github.com/prashantv/gostub v1.1.0 h1:BTyx3RfQjRHnUWaGF9oQos79AlQ5k8WNktv7VGvVH4g= | ||||||
|  | github.com/prashantv/gostub v1.1.0/go.mod h1:A5zLQHz7ieHGG7is6LLXLz7I8+3LZzsrV0P1IAHhP5U= | ||||||
|  | github.com/rogpeppe/go-internal v1.13.1 h1:KvO1DLK/DRN07sQ1LQKScxyZJuNnedQ5/wKSR38lUII= | ||||||
|  | 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= | ||||||
| golang.org/x/net v0.8.0 h1:Zrh2ngAOFYneWTAIAPethzeaQLuHwhuBkuV6ZiRnUaQ= | github.com/sirupsen/logrus v1.9.3 h1:dueUQJ1C2q9oE3F7wvmSGAaVtTmUizReu6fjN8uqzbQ= | ||||||
| golang.org/x/sync v0.1.0 h1:wsuoTGHzEhffawBOhz5CYhcrV4IdKZbEyZjBMuTp12o= | github.com/sirupsen/logrus v1.9.3/go.mod h1:naHLuLoDiP4jHNo9R0sCBMtWGeIprob74mVsIT4qYEQ= | ||||||
| golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= | github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= | ||||||
| golang.org/x/sys v0.7.0 h1:3jlCCIQZPdOYu1h8BkNvLz8Kgwtae2cagcG/VamtZRU= | github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= | ||||||
| golang.org/x/sys v0.7.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA= | ||||||
| golang.org/x/text v0.8.0 h1:57P1ETyNKtuIjB4SRd15iJxuhj8Gc416Y78H3qgMh68= | github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY= | ||||||
| golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= | go.uber.org/automaxprocs v1.6.0 h1:O3y2/QNTOdbF+e/dpXNNW7Rx2hZ4sTIPyybbxyNqTUs= | ||||||
| google.golang.org/genproto v0.0.0-20230410155749-daa745c078e1 h1:KpwkzHKEF7B9Zxg18WzOa7djJ+Ha5DzthMyZYQfEn2A= | go.uber.org/automaxprocs v1.6.0/go.mod h1:ifeIMSnPZuznNm6jmdzmU3/bfk01Fe2fotchwEFJ8r8= | ||||||
| google.golang.org/genproto v0.0.0-20230410155749-daa745c078e1/go.mod h1:nKE/iIaLqn2bQwXBg8f1g2Ylh6r5MN5CmZvuzZCgsCU= | go.uber.org/goleak v1.3.0 h1:2K3zAYmnTNqV73imy9J1T3WC+gmCePx2hEGkimedGto= | ||||||
| google.golang.org/grpc v1.54.0 h1:EhTqbhiYeixwWQtAEZAxmV9MGqcjEU2mFx52xCzNyag= | go.uber.org/goleak v1.3.0/go.mod h1:CoHD4mav9JJNrW/WLlf7HGZPjdw8EucARQHekz1X6bE= | ||||||
| google.golang.org/grpc v1.54.0/go.mod h1:PUSEXI6iWghWaB6lXM4knEgpJNu2qUcKfDtNci3EC2g= | go.unistack.org/micro-proto/v3 v3.4.1 h1:UTjLSRz2YZuaHk9iSlVqqsA50JQNAEK2ZFboGqtEa9Q= | ||||||
| google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw= | go.unistack.org/micro-proto/v3 v3.4.1/go.mod h1:okx/cnOhzuCX0ggl/vToatbCupi0O44diiiLLsZ93Zo= | ||||||
| google.golang.org/protobuf v1.26.0/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc= | golang.org/x/exp v0.0.0-20241210194714-1829a127f884 h1:Y/Mj/94zIQQGHVSv1tTtQBDaQaJe62U9bkDZKKyhPCU= | ||||||
| google.golang.org/protobuf v1.30.0 h1:kPPoIgf3TsEvrm0PFe15JQ+570QVxYzEvvHqChK+cng= | golang.org/x/exp v0.0.0-20241210194714-1829a127f884/go.mod h1:qj5a5QZpwLU2NLQudwIN5koi3beDhSAlJwa67PuM98c= | ||||||
| google.golang.org/protobuf v1.30.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I= | 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/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk= | ||||||
|  | golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||||||
|  | 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/go.mod h1:4IBbMaMmOPCJ8SecivzSH54+73PCFmPWxNTLm+vZkEQ= | ||||||
|  | google.golang.org/genproto/googleapis/rpc v0.0.0-20241209162323-e6fa225c2576 h1:8ZmaLZE4XWrtU3MyClkYqqtl6Oegr3235h7jxsDyqCY= | ||||||
|  | google.golang.org/genproto/googleapis/rpc v0.0.0-20241209162323-e6fa225c2576/go.mod h1:5uTbfoYQed2U9p3KIj2/Zzm02PYhndfdmML0qC3q3FU= | ||||||
|  | google.golang.org/grpc v1.68.1 h1:oI5oTa11+ng8r8XMMN7jAOmWfPZWbYpCFaMUTACxkM0= | ||||||
|  | google.golang.org/grpc v1.68.1/go.mod h1:+q1XYFJjShcqn0QZHvCyeR4CXPA+llXIeUIfIe00waw= | ||||||
|  | google.golang.org/protobuf v1.35.2 h1:8Ar7bF+apOIoThw1EdZl0p1oWvMqTHmpA2fRTyZO8io= | ||||||
|  | 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/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= | ||||||
|   | |||||||
| @@ -13,6 +13,15 @@ func FromContext(ctx context.Context) (Logger, bool) { | |||||||
| 	return l, ok | 	return l, ok | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // MustContext returns logger from passed context or DefaultLogger if empty | ||||||
|  | func MustContext(ctx context.Context) Logger { | ||||||
|  | 	l, ok := FromContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		panic("missing logger") | ||||||
|  | 	} | ||||||
|  | 	return l | ||||||
|  | } | ||||||
|  |  | ||||||
| // NewContext stores logger into passed context | // NewContext stores logger into passed context | ||||||
| func NewContext(ctx context.Context, l Logger) context.Context { | func NewContext(ctx context.Context, l Logger) context.Context { | ||||||
| 	if ctx == nil { | 	if ctx == nil { | ||||||
| @@ -20,3 +29,13 @@ func NewContext(ctx context.Context, l Logger) context.Context { | |||||||
| 	} | 	} | ||||||
| 	return context.WithValue(ctx, loggerKey{}, l) | 	return context.WithValue(ctx, loggerKey{}, l) | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // 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) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -40,3 +40,14 @@ func TestNewContext(t *testing.T) { | |||||||
| 		t.Fatal("NewContext not works") | 		t.Fatal("NewContext not works") | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func TestSetOption(t *testing.T) { | ||||||
|  | 	type key struct{} | ||||||
|  | 	o := SetOption(key{}, "test") | ||||||
|  | 	opts := &Options{} | ||||||
|  | 	o(opts) | ||||||
|  |  | ||||||
|  | 	if v, ok := opts.Context.Value(key{}).(string); !ok || v == "" { | ||||||
|  | 		t.Fatal("SetOption not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -1,232 +0,0 @@ | |||||||
| package logger |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"context" |  | ||||||
| 	"encoding/json" |  | ||||||
| 	"fmt" |  | ||||||
| 	"os" |  | ||||||
| 	"runtime" |  | ||||||
| 	"strings" |  | ||||||
| 	"sync" |  | ||||||
| 	"time" |  | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/options" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| type defaultLogger struct { |  | ||||||
| 	enc  *json.Encoder |  | ||||||
| 	opts Options |  | ||||||
| 	sync.RWMutex |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Init(opts...) should only overwrite provided options |  | ||||||
| func (l *defaultLogger) Init(opts ...options.Option) error { |  | ||||||
| 	l.Lock() |  | ||||||
| 	for _, o := range opts { |  | ||||||
| 		o(&l.opts) |  | ||||||
| 	} |  | ||||||
| 	l.enc = json.NewEncoder(l.opts.Out) |  | ||||||
| 	// wrap the Log func |  | ||||||
| 	l.Unlock() |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) String() string { |  | ||||||
| 	return "micro" |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) Clone(opts ...options.Option) Logger { |  | ||||||
| 	newopts := NewOptions(opts...) |  | ||||||
| 	oldopts := l.opts |  | ||||||
| 	for _, o := range opts { |  | ||||||
| 		o(&newopts) |  | ||||||
| 		o(&oldopts) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	l.Lock() |  | ||||||
| 	cl := &defaultLogger{opts: oldopts, enc: json.NewEncoder(l.opts.Out)} |  | ||||||
| 	l.Unlock() |  | ||||||
|  |  | ||||||
| 	return cl |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) V(level Level) bool { |  | ||||||
| 	l.RLock() |  | ||||||
| 	ok := l.opts.Level.Enabled(level) |  | ||||||
| 	l.RUnlock() |  | ||||||
| 	return ok |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) Level(level Level) { |  | ||||||
| 	l.Lock() |  | ||||||
| 	l.opts.Level = level |  | ||||||
| 	l.Unlock() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) Fields(fields ...interface{}) Logger { |  | ||||||
| 	l.RLock() |  | ||||||
| 	nl := &defaultLogger{opts: l.opts, enc: l.enc} |  | ||||||
| 	if len(fields) == 0 { |  | ||||||
| 		l.RUnlock() |  | ||||||
| 		return nl |  | ||||||
| 	} else if len(fields)%2 != 0 { |  | ||||||
| 		fields = fields[:len(fields)-1] |  | ||||||
| 	} |  | ||||||
| 	nl.opts.Fields = copyFields(l.opts.Fields) |  | ||||||
| 	nl.opts.Fields = append(nl.opts.Fields, fields...) |  | ||||||
| 	l.RUnlock() |  | ||||||
| 	return nl |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func copyFields(src []interface{}) []interface{} { |  | ||||||
| 	dst := make([]interface{}, len(src)) |  | ||||||
| 	copy(dst, src) |  | ||||||
| 	return dst |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // logCallerfilePath returns a package/file:line description of the caller, |  | ||||||
| // preserving only the leaf directory name and file name. |  | ||||||
| func logCallerfilePath(loggingFilePath string) string { |  | ||||||
| 	// To make sure we trim the path correctly on Windows too, we |  | ||||||
| 	// counter-intuitively need to use '/' and *not* os.PathSeparator here, |  | ||||||
| 	// because the path given originates from Go stdlib, specifically |  | ||||||
| 	// runtime.Caller() which (as of Mar/17) returns forward slashes even on |  | ||||||
| 	// Windows. |  | ||||||
| 	// |  | ||||||
| 	// See https://github.com/golang/go/issues/3335 |  | ||||||
| 	// and https://github.com/golang/go/issues/18151 |  | ||||||
| 	// |  | ||||||
| 	// for discussion on the issue on Go side. |  | ||||||
| 	idx := strings.LastIndexByte(loggingFilePath, '/') |  | ||||||
| 	if idx == -1 { |  | ||||||
| 		return loggingFilePath |  | ||||||
| 	} |  | ||||||
| 	idx = strings.LastIndexByte(loggingFilePath[:idx], '/') |  | ||||||
| 	if idx == -1 { |  | ||||||
| 		return loggingFilePath |  | ||||||
| 	} |  | ||||||
| 	return loggingFilePath[idx+1:] |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) Info(ctx context.Context, args ...interface{}) { |  | ||||||
| 	l.Log(ctx, InfoLevel, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) Error(ctx context.Context, args ...interface{}) { |  | ||||||
| 	l.Log(ctx, ErrorLevel, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) Debug(ctx context.Context, args ...interface{}) { |  | ||||||
| 	l.Log(ctx, DebugLevel, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) Warn(ctx context.Context, args ...interface{}) { |  | ||||||
| 	l.Log(ctx, WarnLevel, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) Trace(ctx context.Context, args ...interface{}) { |  | ||||||
| 	l.Log(ctx, TraceLevel, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) Fatal(ctx context.Context, args ...interface{}) { |  | ||||||
| 	l.Log(ctx, FatalLevel, args...) |  | ||||||
| 	os.Exit(1) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) Infof(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	l.Logf(ctx, InfoLevel, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) Errorf(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	l.Logf(ctx, ErrorLevel, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) Debugf(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	l.Logf(ctx, DebugLevel, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) Warnf(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	l.Logf(ctx, WarnLevel, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) Tracef(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	l.Logf(ctx, TraceLevel, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) Fatalf(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	l.Logf(ctx, FatalLevel, msg, args...) |  | ||||||
| 	os.Exit(1) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) Log(ctx context.Context, level Level, args ...interface{}) { |  | ||||||
| 	if !l.V(level) { |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	l.RLock() |  | ||||||
| 	fields := copyFields(l.opts.Fields) |  | ||||||
| 	l.RUnlock() |  | ||||||
|  |  | ||||||
| 	fields = append(fields, "level", level.String()) |  | ||||||
|  |  | ||||||
| 	if _, file, line, ok := runtime.Caller(l.opts.CallerSkipCount); ok { |  | ||||||
| 		fields = append(fields, "caller", fmt.Sprintf("%s:%d", logCallerfilePath(file), line)) |  | ||||||
| 	} |  | ||||||
| 	fields = append(fields, "timestamp", time.Now().Format("2006-01-02 15:04:05")) |  | ||||||
|  |  | ||||||
| 	if len(args) > 0 { |  | ||||||
| 		fields = append(fields, "msg", fmt.Sprint(args...)) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	out := make(map[string]interface{}, len(fields)/2) |  | ||||||
| 	for i := 0; i < len(fields); i += 2 { |  | ||||||
| 		out[fields[i].(string)] = fields[i+1] |  | ||||||
| 	} |  | ||||||
| 	l.RLock() |  | ||||||
| 	_ = l.enc.Encode(out) |  | ||||||
| 	l.RUnlock() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) Logf(ctx context.Context, level Level, msg string, args ...interface{}) { |  | ||||||
| 	if !l.V(level) { |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	l.RLock() |  | ||||||
| 	fields := copyFields(l.opts.Fields) |  | ||||||
| 	l.RUnlock() |  | ||||||
|  |  | ||||||
| 	fields = append(fields, "level", level.String()) |  | ||||||
|  |  | ||||||
| 	if _, file, line, ok := runtime.Caller(l.opts.CallerSkipCount); ok { |  | ||||||
| 		fields = append(fields, "caller", fmt.Sprintf("%s:%d", logCallerfilePath(file), line)) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	fields = append(fields, "timestamp", time.Now().Format("2006-01-02 15:04:05")) |  | ||||||
| 	if len(args) > 0 { |  | ||||||
| 		fields = append(fields, "msg", fmt.Sprintf(msg, args...)) |  | ||||||
| 	} else if msg != "" { |  | ||||||
| 		fields = append(fields, "msg", msg) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	out := make(map[string]interface{}, len(fields)/2) |  | ||||||
| 	for i := 0; i < len(fields); i += 2 { |  | ||||||
| 		out[fields[i].(string)] = fields[i+1] |  | ||||||
| 	} |  | ||||||
| 	l.RLock() |  | ||||||
| 	_ = l.enc.Encode(out) |  | ||||||
| 	l.RUnlock() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) Options() Options { |  | ||||||
| 	return l.opts |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewLogger builds a new logger based on options |  | ||||||
| func NewLogger(opts ...options.Option) Logger { |  | ||||||
| 	l := &defaultLogger{ |  | ||||||
| 		opts: NewOptions(opts...), |  | ||||||
| 	} |  | ||||||
| 	l.enc = json.NewEncoder(l.opts.Out) |  | ||||||
| 	return l |  | ||||||
| } |  | ||||||
							
								
								
									
										124
									
								
								logger/logger.go
									
									
									
									
									
								
							
							
						
						
									
										124
									
								
								logger/logger.go
									
									
									
									
									
								
							| @@ -1,28 +1,27 @@ | |||||||
| // Package logger provides a log interface | // Package logger provides a log interface | ||||||
| package logger // import "go.unistack.org/micro/v4/logger" | package logger | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"os" |  | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/options" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | type ContextAttrFunc func(ctx context.Context) []interface{} | ||||||
|  |  | ||||||
|  | var DefaultContextAttrFuncs []ContextAttrFunc | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| 	// DefaultLogger variable | 	// DefaultLogger variable | ||||||
| 	DefaultLogger = NewLogger(WithLevel(ParseLevel(os.Getenv("MICRO_LOG_LEVEL")))) | 	DefaultLogger = NewLogger() | ||||||
| 	// DefaultLevel used by logger | 	// DefaultLevel used by logger | ||||||
| 	DefaultLevel = InfoLevel | 	DefaultLevel = InfoLevel | ||||||
| 	// DefaultCallerSkipCount used by logger |  | ||||||
| 	DefaultCallerSkipCount = 2 |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Logger is a generic logging interface | // Logger is a generic logging interface | ||||||
| type Logger interface { | type Logger interface { | ||||||
| 	// Init initialises options | 	// Init initialises options | ||||||
| 	Init(opts ...options.Option) error | 	Init(opts ...Option) error | ||||||
| 	// Clone create logger copy with new options | 	// Clone create logger copy with new options | ||||||
| 	Clone(opts ...options.Option) Logger | 	Clone(opts ...Option) Logger | ||||||
| 	// V compare provided verbosity level with current log level | 	// V compare provided verbosity level with current log level | ||||||
| 	V(level Level) bool | 	V(level Level) bool | ||||||
| 	// Level sets the log level for logger | 	// Level sets the log level for logger | ||||||
| @@ -32,111 +31,24 @@ type Logger interface { | |||||||
| 	// Fields set fields to always be logged with keyval pairs | 	// Fields set fields to always be logged with keyval pairs | ||||||
| 	Fields(fields ...interface{}) Logger | 	Fields(fields ...interface{}) Logger | ||||||
| 	// Info level message | 	// Info level message | ||||||
| 	Info(ctx context.Context, args ...interface{}) | 	Info(ctx context.Context, msg string, args ...interface{}) | ||||||
| 	// Trace level message | 	// Trace level message | ||||||
| 	Trace(ctx context.Context, args ...interface{}) | 	Trace(ctx context.Context, msg string, args ...interface{}) | ||||||
| 	// Debug level message | 	// Debug level message | ||||||
| 	Debug(ctx context.Context, args ...interface{}) | 	Debug(ctx context.Context, msg string, args ...interface{}) | ||||||
| 	// Warn level message | 	// Warn level message | ||||||
| 	Warn(ctx context.Context, args ...interface{}) | 	Warn(ctx context.Context, msg string, args ...interface{}) | ||||||
| 	// Error level message | 	// Error level message | ||||||
| 	Error(ctx context.Context, args ...interface{}) | 	Error(ctx context.Context, msg string, args ...interface{}) | ||||||
| 	// Fatal level message | 	// Fatal level message | ||||||
| 	Fatal(ctx context.Context, args ...interface{}) | 	Fatal(ctx context.Context, msg string, args ...interface{}) | ||||||
| 	// Infof level message |  | ||||||
| 	Infof(ctx context.Context, msg string, args ...interface{}) |  | ||||||
| 	// Tracef level message |  | ||||||
| 	Tracef(ctx context.Context, msg string, args ...interface{}) |  | ||||||
| 	// Debug level message |  | ||||||
| 	Debugf(ctx context.Context, msg string, args ...interface{}) |  | ||||||
| 	// Warn level message |  | ||||||
| 	Warnf(ctx context.Context, msg string, args ...interface{}) |  | ||||||
| 	// Error level message |  | ||||||
| 	Errorf(ctx context.Context, msg string, args ...interface{}) |  | ||||||
| 	// Fatal level message |  | ||||||
| 	Fatalf(ctx context.Context, msg string, args ...interface{}) |  | ||||||
| 	// Log logs message with needed level | 	// Log logs message with needed level | ||||||
| 	Log(ctx context.Context, level Level, args ...interface{}) | 	Log(ctx context.Context, level Level, msg string, args ...interface{}) | ||||||
| 	// Logf logs message with needed level | 	// Name returns broker instance name | ||||||
| 	Logf(ctx context.Context, level Level, msg string, args ...interface{}) | 	Name() string | ||||||
| 	// String returns the name of logger | 	// String returns the type of logger | ||||||
| 	String() string | 	String() string | ||||||
| } | } | ||||||
|  |  | ||||||
| // Field contains keyval pair | // Field contains keyval pair | ||||||
| type Field interface{} | type Field interface{} | ||||||
|  |  | ||||||
| // Info writes msg to default logger on info level |  | ||||||
| func Info(ctx context.Context, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Info(ctx, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Error writes msg to default logger on error level |  | ||||||
| func Error(ctx context.Context, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Error(ctx, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Debug writes msg to default logger on debug level |  | ||||||
| func Debug(ctx context.Context, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Debug(ctx, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Warn writes msg to default logger on warn level |  | ||||||
| func Warn(ctx context.Context, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Warn(ctx, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Trace writes msg to default logger on trace level |  | ||||||
| func Trace(ctx context.Context, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Trace(ctx, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Fatal writes msg to default logger on fatal level |  | ||||||
| func Fatal(ctx context.Context, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Fatal(ctx, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Infof writes formatted msg to default logger on info level |  | ||||||
| func Infof(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Infof(ctx, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Errorf writes formatted msg to default logger on error level |  | ||||||
| func Errorf(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Errorf(ctx, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Debugf writes formatted msg to default logger on debug level |  | ||||||
| func Debugf(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Debugf(ctx, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Warnf writes formatted msg to default logger on warn level |  | ||||||
| func Warnf(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Warnf(ctx, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Tracef writes formatted msg to default logger on trace level |  | ||||||
| func Tracef(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Tracef(ctx, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Fatalf writes formatted msg to default logger on fatal level |  | ||||||
| func Fatalf(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Fatalf(ctx, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // V returns true if passed level enabled in default logger |  | ||||||
| func V(level Level) bool { |  | ||||||
| 	return DefaultLogger.V(level) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Init initialize logger |  | ||||||
| func Init(opts ...options.Option) error { |  | ||||||
| 	return DefaultLogger.Init(opts...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Fields create logger with specific fields |  | ||||||
| func Fields(fields ...interface{}) Logger { |  | ||||||
| 	return DefaultLogger.Fields(fields...) |  | ||||||
| } |  | ||||||
|   | |||||||
| @@ -1,138 +0,0 @@ | |||||||
| package logger |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"bytes" |  | ||||||
| 	"context" |  | ||||||
| 	"log" |  | ||||||
| 	"testing" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| func TestContext(t *testing.T) { |  | ||||||
| 	ctx := context.TODO() |  | ||||||
| 	buf := bytes.NewBuffer(nil) |  | ||||||
| 	l := NewLogger(WithLevel(TraceLevel), WithOutput(buf)) |  | ||||||
| 	if err := l.Init(); err != nil { |  | ||||||
| 		t.Fatal(err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	nl, ok := FromContext(NewContext(ctx, l.Fields("key", "val"))) |  | ||||||
| 	if !ok { |  | ||||||
| 		t.Fatal("context without logger") |  | ||||||
| 	} |  | ||||||
| 	nl.Info(ctx, "message") |  | ||||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"key":"val"`)) { |  | ||||||
| 		t.Fatalf("logger fields not works, buf contains: %s", buf.Bytes()) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestFields(t *testing.T) { |  | ||||||
| 	ctx := context.TODO() |  | ||||||
| 	buf := bytes.NewBuffer(nil) |  | ||||||
| 	l := NewLogger(WithLevel(TraceLevel), WithOutput(buf)) |  | ||||||
| 	if err := l.Init(); err != nil { |  | ||||||
| 		t.Fatal(err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	nl := l.Fields("key", "val") |  | ||||||
|  |  | ||||||
| 	nl.Info(ctx, "message") |  | ||||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"key":"val"`)) { |  | ||||||
| 		t.Fatalf("logger fields not works, buf contains: %s", buf.Bytes()) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestFromContextWithFields(t *testing.T) { |  | ||||||
| 	ctx := context.TODO() |  | ||||||
| 	buf := bytes.NewBuffer(nil) |  | ||||||
| 	var ok bool |  | ||||||
| 	l := NewLogger(WithLevel(TraceLevel), WithOutput(buf)) |  | ||||||
| 	if err := l.Init(); err != nil { |  | ||||||
| 		t.Fatal(err) |  | ||||||
| 	} |  | ||||||
| 	nl := l.Fields("key", "val") |  | ||||||
|  |  | ||||||
| 	ctx = NewContext(ctx, nl) |  | ||||||
|  |  | ||||||
| 	l, ok = FromContext(ctx) |  | ||||||
| 	if !ok { |  | ||||||
| 		t.Fatalf("context does not have logger") |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	l.Info(ctx, "message") |  | ||||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"key":"val"`)) { |  | ||||||
| 		t.Fatalf("logger fields not works, buf contains: %s", buf.Bytes()) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestClone(t *testing.T) { |  | ||||||
| 	ctx := context.TODO() |  | ||||||
| 	buf := bytes.NewBuffer(nil) |  | ||||||
| 	l := NewLogger(WithLevel(TraceLevel), WithOutput(buf)) |  | ||||||
| 	if err := l.Init(); err != nil { |  | ||||||
| 		t.Fatal(err) |  | ||||||
| 	} |  | ||||||
| 	nl := l.Clone(WithLevel(ErrorLevel)) |  | ||||||
| 	if err := nl.Init(); err != nil { |  | ||||||
| 		t.Fatal(err) |  | ||||||
| 	} |  | ||||||
| 	nl.Info(ctx, "info message") |  | ||||||
| 	if len(buf.Bytes()) != 0 { |  | ||||||
| 		t.Fatal("message must not be logged") |  | ||||||
| 	} |  | ||||||
| 	l.Info(ctx, "info message") |  | ||||||
| 	if len(buf.Bytes()) == 0 { |  | ||||||
| 		t.Fatal("message must be logged") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestRedirectStdLogger(t *testing.T) { |  | ||||||
| 	buf := bytes.NewBuffer(nil) |  | ||||||
| 	l := NewLogger(WithLevel(TraceLevel), WithOutput(buf)) |  | ||||||
| 	if err := l.Init(); err != nil { |  | ||||||
| 		t.Fatal(err) |  | ||||||
| 	} |  | ||||||
| 	fn := RedirectStdLogger(l, ErrorLevel) |  | ||||||
| 	defer fn() |  | ||||||
| 	log.Print("test") |  | ||||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"level":"error","msg":"test","timestamp"`)) { |  | ||||||
| 		t.Fatalf("logger error, buf %s", buf.Bytes()) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestStdLogger(t *testing.T) { |  | ||||||
| 	buf := bytes.NewBuffer(nil) |  | ||||||
| 	l := NewLogger(WithLevel(TraceLevel), WithOutput(buf)) |  | ||||||
| 	if err := l.Init(); err != nil { |  | ||||||
| 		t.Fatal(err) |  | ||||||
| 	} |  | ||||||
| 	lg := NewStdLogger(l, ErrorLevel) |  | ||||||
| 	lg.Print("test") |  | ||||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"level":"error","msg":"test","timestamp"`)) { |  | ||||||
| 		t.Fatalf("logger error, buf %s", buf.Bytes()) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestLogger(t *testing.T) { |  | ||||||
| 	ctx := context.TODO() |  | ||||||
| 	buf := bytes.NewBuffer(nil) |  | ||||||
| 	l := NewLogger(WithLevel(TraceLevel), WithOutput(buf)) |  | ||||||
| 	if err := l.Init(); err != nil { |  | ||||||
| 		t.Fatal(err) |  | ||||||
| 	} |  | ||||||
| 	l.Trace(ctx, "trace_msg1") |  | ||||||
| 	l.Warn(ctx, "warn_msg1") |  | ||||||
| 	l.Fields("error", "test").Info(ctx, "error message") |  | ||||||
| 	l.Warn(ctx, "first", " ", "second") |  | ||||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"level":"trace","msg":"trace_msg1"`)) { |  | ||||||
| 		t.Fatalf("logger error, buf %s", buf.Bytes()) |  | ||||||
| 	} |  | ||||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"warn","msg":"warn_msg1"`)) { |  | ||||||
| 		t.Fatalf("logger error, buf %s", buf.Bytes()) |  | ||||||
| 	} |  | ||||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"error":"test","level":"info","msg":"error message"`)) { |  | ||||||
| 		t.Fatalf("logger error, buf %s", buf.Bytes()) |  | ||||||
| 	} |  | ||||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"level":"warn","msg":"first second"`)) { |  | ||||||
| 		t.Fatalf("logger error, buf %s", buf.Bytes()) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
							
								
								
									
										78
									
								
								logger/noop.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										78
									
								
								logger/noop.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,78 @@ | |||||||
|  | package logger | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | const ( | ||||||
|  | 	defaultCallerSkipCount = 2 | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type noopLogger struct { | ||||||
|  | 	opts Options | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func NewLogger(opts ...Option) Logger { | ||||||
|  | 	options := NewOptions(opts...) | ||||||
|  | 	options.CallerSkipCount = defaultCallerSkipCount | ||||||
|  | 	return &noopLogger{opts: options} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (l *noopLogger) V(_ Level) bool { | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (l *noopLogger) Level(_ Level) { | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (l *noopLogger) Name() string { | ||||||
|  | 	return l.opts.Name | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (l *noopLogger) Init(opts ...Option) error { | ||||||
|  | 	for _, o := range opts { | ||||||
|  | 		o(&l.opts) | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (l *noopLogger) Clone(opts ...Option) Logger { | ||||||
|  | 	nl := &noopLogger{opts: l.opts} | ||||||
|  | 	for _, o := range opts { | ||||||
|  | 		o(&nl.opts) | ||||||
|  | 	} | ||||||
|  | 	return nl | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (l *noopLogger) Fields(_ ...interface{}) Logger { | ||||||
|  | 	return l | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (l *noopLogger) Options() Options { | ||||||
|  | 	return l.opts | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (l *noopLogger) String() string { | ||||||
|  | 	return "noop" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (l *noopLogger) Log(ctx context.Context, lvl Level, msg string, attrs ...interface{}) { | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (l *noopLogger) Info(ctx context.Context, msg string, attrs ...interface{}) { | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (l *noopLogger) Debug(ctx context.Context, msg string, attrs ...interface{}) { | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (l *noopLogger) Error(ctx context.Context, msg string, attrs ...interface{}) { | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (l *noopLogger) Trace(ctx context.Context, msg string, attrs ...interface{}) { | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (l *noopLogger) Warn(ctx context.Context, msg string, attrs ...interface{}) { | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (l *noopLogger) Fatal(ctx context.Context, msg string, attrs ...interface{}) { | ||||||
|  | } | ||||||
| @@ -3,66 +3,227 @@ package logger | |||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"io" | 	"io" | ||||||
|  | 	"log/slog" | ||||||
| 	"os" | 	"os" | ||||||
|  | 	"slices" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/options" | 	"go.unistack.org/micro/v3/meter" | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | // Option func signature | ||||||
|  | type Option func(*Options) | ||||||
|  |  | ||||||
| // Options holds logger options | // Options holds logger options | ||||||
| type Options struct { | type Options struct { | ||||||
|  | 	// TimeKey is the key used for the time of the log call | ||||||
|  | 	TimeKey string | ||||||
|  | 	// LevelKey is the key used for the level of the log call | ||||||
|  | 	LevelKey string | ||||||
|  | 	// ErroreKey is the key used for the error of the log call | ||||||
|  | 	ErrorKey string | ||||||
|  | 	// MessageKey is the key used for the message of the log call | ||||||
|  | 	MessageKey string | ||||||
|  | 	// SourceKey is the key used for the source file and line of the log call | ||||||
|  | 	SourceKey string | ||||||
|  | 	// StacktraceKey is the key used for the stacktrace | ||||||
|  | 	StacktraceKey string | ||||||
|  | 	// Name holds the logger name | ||||||
|  | 	Name string | ||||||
| 	// Out holds the output writer | 	// Out holds the output writer | ||||||
| 	Out io.Writer | 	Out io.Writer | ||||||
| 	// Context holds exernal options | 	// Context holds exernal options | ||||||
| 	Context context.Context | 	Context context.Context | ||||||
|  | 	// Meter used to count logs for specific level | ||||||
|  | 	Meter meter.Meter | ||||||
|  | 	// TimeFunc used to obtain current time | ||||||
|  | 	TimeFunc func() time.Time | ||||||
| 	// Fields holds additional metadata | 	// Fields holds additional metadata | ||||||
| 	Fields []interface{} | 	Fields []interface{} | ||||||
| 	// Name holds the logger name | 	// ContextAttrFuncs contains funcs that executed before log func on context | ||||||
| 	Name string | 	ContextAttrFuncs []ContextAttrFunc | ||||||
|  | 	// callerSkipCount number of frmaes to skip | ||||||
|  | 	CallerSkipCount int | ||||||
| 	// The logging level the logger should log | 	// The logging level the logger should log | ||||||
| 	Level Level | 	Level Level | ||||||
| 	// CallerSkipCount number of frmaes to skip | 	// AddSource enabled writing source file and position in log | ||||||
| 	CallerSkipCount int | 	AddSource bool | ||||||
|  | 	// AddStacktrace controls writing of stacktaces on error | ||||||
|  | 	AddStacktrace bool | ||||||
|  | 	// DedupKeys deduplicate keys in log output | ||||||
|  | 	DedupKeys bool | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewOptions creates new options struct | // NewOptions creates new options struct | ||||||
| func NewOptions(opts ...options.Option) Options { | func NewOptions(opts ...Option) Options { | ||||||
| 	options := Options{ | 	options := Options{ | ||||||
| 		Level:            DefaultLevel, | 		Level:            DefaultLevel, | ||||||
| 		Fields:           make([]interface{}, 0, 6), | 		Fields:           make([]interface{}, 0, 6), | ||||||
| 		Out:              os.Stderr, | 		Out:              os.Stderr, | ||||||
| 		CallerSkipCount: DefaultCallerSkipCount, |  | ||||||
| 		Context:          context.Background(), | 		Context:          context.Background(), | ||||||
|  | 		ContextAttrFuncs: DefaultContextAttrFuncs, | ||||||
|  | 		AddSource:        true, | ||||||
|  | 		TimeFunc:         time.Now, | ||||||
|  | 		Meter:            meter.DefaultMeter, | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	WithMicroKeys()(&options) | ||||||
|  |  | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&options) | 		o(&options) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return options | 	return options | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // WithContextAttrFuncs appends default funcs for the context attrs filler | ||||||
|  | func WithContextAttrFuncs(fncs ...ContextAttrFunc) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.ContextAttrFuncs = append(o.ContextAttrFuncs, fncs...) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithDedupKeys dont log duplicate keys | ||||||
|  | func WithDedupKeys(b bool) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.DedupKeys = b | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithAddFields add fields for the logger | ||||||
|  | func WithAddFields(fields ...interface{}) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		if o.DedupKeys { | ||||||
|  | 			for i := 0; i < len(o.Fields); i += 2 { | ||||||
|  | 				for j := 0; j < len(fields); j += 2 { | ||||||
|  | 					iv, iok := o.Fields[i].(string) | ||||||
|  | 					jv, jok := fields[j].(string) | ||||||
|  | 					if iok && jok && iv == jv { | ||||||
|  | 						fields = slices.Delete(fields, j, j+2) | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			if len(fields) > 0 { | ||||||
|  | 				o.Fields = append(o.Fields, fields...) | ||||||
|  | 			} | ||||||
|  | 		} else { | ||||||
|  | 			o.Fields = append(o.Fields, fields...) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // WithFields set default fields for the logger | // WithFields set default fields for the logger | ||||||
| func WithFields(fields ...interface{}) options.Option { | func WithFields(fields ...interface{}) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, fields, ".Fields") | 		o.Fields = fields | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // WithLevel set default level for the logger | // WithLevel set default level for the logger | ||||||
| func WithLevel(lvl Level) options.Option { | func WithLevel(level Level) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, lvl, ".Level") | 		o.Level = level | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // WithOutput set default output writer for the logger | // WithOutput set default output writer for the logger | ||||||
| func WithOutput(out io.Writer) options.Option { | func WithOutput(out io.Writer) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, out, ".Out") | 		o.Out = out | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // WithCallerSkipCount set frame count to skip | // WithAddStacktrace controls writing stacktrace on error | ||||||
| func WithCallerSkipCount(c int) options.Option { | func WithAddStacktrace(v bool) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, c, ".CallerSkipCount") | 		o.AddStacktrace = v | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithAddSource controls writing source file and pos in log | ||||||
|  | func WithAddSource(v bool) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.AddSource = v | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithContext set context | ||||||
|  | func WithContext(ctx context.Context) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Context = ctx | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithName sets the name | ||||||
|  | func WithName(n string) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Name = n | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithMeter sets the meter | ||||||
|  | func WithMeter(m meter.Meter) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Meter = m | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithTimeFunc sets the func to obtain current time | ||||||
|  | func WithTimeFunc(fn func() time.Time) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.TimeFunc = fn | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func WithZapKeys() Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.TimeKey = "@timestamp" | ||||||
|  | 		o.LevelKey = slog.LevelKey | ||||||
|  | 		o.MessageKey = slog.MessageKey | ||||||
|  | 		o.SourceKey = "caller" | ||||||
|  | 		o.StacktraceKey = "stacktrace" | ||||||
|  | 		o.ErrorKey = "error" | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func WithZerologKeys() Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.TimeKey = slog.TimeKey | ||||||
|  | 		o.LevelKey = slog.LevelKey | ||||||
|  | 		o.MessageKey = "message" | ||||||
|  | 		o.SourceKey = "caller" | ||||||
|  | 		o.StacktraceKey = "stacktrace" | ||||||
|  | 		o.ErrorKey = "error" | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func WithSlogKeys() Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.TimeKey = slog.TimeKey | ||||||
|  | 		o.LevelKey = slog.LevelKey | ||||||
|  | 		o.MessageKey = slog.MessageKey | ||||||
|  | 		o.SourceKey = slog.SourceKey | ||||||
|  | 		o.StacktraceKey = "stacktrace" | ||||||
|  | 		o.ErrorKey = "error" | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func WithMicroKeys() Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.TimeKey = "timestamp" | ||||||
|  | 		o.LevelKey = slog.LevelKey | ||||||
|  | 		o.MessageKey = slog.MessageKey | ||||||
|  | 		o.SourceKey = "caller" | ||||||
|  | 		o.StacktraceKey = "stacktrace" | ||||||
|  | 		o.ErrorKey = "error" | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithAddCallerSkipCount add skip count for copy logger | ||||||
|  | func WithAddCallerSkipCount(n int) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		if n > 0 { | ||||||
|  | 			o.CallerSkipCount += n | ||||||
|  | 		} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										373
									
								
								logger/slog/slog.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										373
									
								
								logger/slog/slog.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,373 @@ | |||||||
|  | package slog | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"io" | ||||||
|  | 	"log/slog" | ||||||
|  | 	"os" | ||||||
|  | 	"reflect" | ||||||
|  | 	"regexp" | ||||||
|  | 	"runtime" | ||||||
|  | 	"strconv" | ||||||
|  | 	"sync" | ||||||
|  | 	"sync/atomic" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v3/logger" | ||||||
|  | 	"go.unistack.org/micro/v3/semconv" | ||||||
|  | 	"go.unistack.org/micro/v3/tracer" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | const ( | ||||||
|  | 	badKey = "!BADKEY" | ||||||
|  | 	// defaultCallerSkipCount used by logger | ||||||
|  | 	defaultCallerSkipCount = 3 | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | var reTrace = regexp.MustCompile(`.*/slog/logger\.go.*\n`) | ||||||
|  |  | ||||||
|  | var ( | ||||||
|  | 	traceValue = slog.StringValue("trace") | ||||||
|  | 	debugValue = slog.StringValue("debug") | ||||||
|  | 	infoValue  = slog.StringValue("info") | ||||||
|  | 	warnValue  = slog.StringValue("warn") | ||||||
|  | 	errorValue = slog.StringValue("error") | ||||||
|  | 	fatalValue = slog.StringValue("fatal") | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type wrapper struct { | ||||||
|  | 	h     slog.Handler | ||||||
|  | 	level atomic.Int64 | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (h *wrapper) Enabled(ctx context.Context, level slog.Level) bool { | ||||||
|  | 	return level >= slog.Level(int(h.level.Load())) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (h *wrapper) Handle(ctx context.Context, rec slog.Record) error { | ||||||
|  | 	return h.h.Handle(ctx, rec) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (h *wrapper) WithAttrs(attrs []slog.Attr) slog.Handler { | ||||||
|  | 	return h.h.WithAttrs(attrs) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (h *wrapper) WithGroup(name string) slog.Handler { | ||||||
|  | 	return h.h.WithGroup(name) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *slogLogger) renameAttr(_ []string, a slog.Attr) slog.Attr { | ||||||
|  | 	switch a.Key { | ||||||
|  | 	case slog.SourceKey: | ||||||
|  | 		source := a.Value.Any().(*slog.Source) | ||||||
|  | 		a.Value = slog.StringValue(source.File + ":" + strconv.Itoa(source.Line)) | ||||||
|  | 		a.Key = s.opts.SourceKey | ||||||
|  | 	case slog.TimeKey: | ||||||
|  | 		a.Key = s.opts.TimeKey | ||||||
|  | 	case slog.MessageKey: | ||||||
|  | 		a.Key = s.opts.MessageKey | ||||||
|  | 	case slog.LevelKey: | ||||||
|  | 		level := a.Value.Any().(slog.Level) | ||||||
|  | 		lvl := slogToLoggerLevel(level) | ||||||
|  | 		a.Key = s.opts.LevelKey | ||||||
|  | 		switch { | ||||||
|  | 		case lvl < logger.DebugLevel: | ||||||
|  | 			a.Value = traceValue | ||||||
|  | 		case lvl < logger.InfoLevel: | ||||||
|  | 			a.Value = debugValue | ||||||
|  | 		case lvl < logger.WarnLevel: | ||||||
|  | 			a.Value = infoValue | ||||||
|  | 		case lvl < logger.ErrorLevel: | ||||||
|  | 			a.Value = warnValue | ||||||
|  | 		case lvl < logger.FatalLevel: | ||||||
|  | 			a.Value = errorValue | ||||||
|  | 		case lvl >= logger.FatalLevel: | ||||||
|  | 			a.Value = fatalValue | ||||||
|  | 		default: | ||||||
|  | 			a.Value = infoValue | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return a | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type slogLogger struct { | ||||||
|  | 	handler *wrapper | ||||||
|  | 	opts    logger.Options | ||||||
|  | 	mu      sync.RWMutex | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *slogLogger) Clone(opts ...logger.Option) logger.Logger { | ||||||
|  | 	s.mu.RLock() | ||||||
|  | 	options := s.opts | ||||||
|  | 	s.mu.RUnlock() | ||||||
|  |  | ||||||
|  | 	for _, o := range opts { | ||||||
|  | 		o(&options) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if len(options.ContextAttrFuncs) == 0 { | ||||||
|  | 		options.ContextAttrFuncs = logger.DefaultContextAttrFuncs | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	attrs, _ := s.argsAttrs(options.Fields) | ||||||
|  | 	l := &slogLogger{ | ||||||
|  | 		handler: &wrapper{h: s.handler.h.WithAttrs(attrs)}, | ||||||
|  | 		opts:    options, | ||||||
|  | 	} | ||||||
|  | 	l.handler.level.Store(int64(loggerToSlogLevel(options.Level))) | ||||||
|  |  | ||||||
|  | 	return l | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *slogLogger) V(level logger.Level) bool { | ||||||
|  | 	s.mu.Lock() | ||||||
|  | 	v := s.opts.Level.Enabled(level) | ||||||
|  | 	s.mu.Unlock() | ||||||
|  | 	return v | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *slogLogger) Level(level logger.Level) { | ||||||
|  | 	s.mu.Lock() | ||||||
|  | 	s.opts.Level = level | ||||||
|  | 	s.handler.level.Store(int64(loggerToSlogLevel(level))) | ||||||
|  | 	s.mu.Unlock() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *slogLogger) Options() logger.Options { | ||||||
|  | 	return s.opts | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *slogLogger) Fields(fields ...interface{}) logger.Logger { | ||||||
|  | 	s.mu.RLock() | ||||||
|  | 	options := s.opts | ||||||
|  | 	s.mu.RUnlock() | ||||||
|  |  | ||||||
|  | 	l := &slogLogger{opts: options} | ||||||
|  | 	logger.WithAddFields(fields...)(&l.opts) | ||||||
|  |  | ||||||
|  | 	if len(options.ContextAttrFuncs) == 0 { | ||||||
|  | 		options.ContextAttrFuncs = logger.DefaultContextAttrFuncs | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	attrs, _ := s.argsAttrs(fields) | ||||||
|  | 	l.handler = &wrapper{h: s.handler.h.WithAttrs(attrs)} | ||||||
|  | 	l.handler.level.Store(int64(loggerToSlogLevel(l.opts.Level))) | ||||||
|  |  | ||||||
|  | 	return l | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *slogLogger) Init(opts ...logger.Option) error { | ||||||
|  | 	s.mu.Lock() | ||||||
|  |  | ||||||
|  | 	for _, o := range opts { | ||||||
|  | 		o(&s.opts) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if len(s.opts.ContextAttrFuncs) == 0 { | ||||||
|  | 		s.opts.ContextAttrFuncs = logger.DefaultContextAttrFuncs | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	handleOpt := &slog.HandlerOptions{ | ||||||
|  | 		ReplaceAttr: s.renameAttr, | ||||||
|  | 		Level:       loggerToSlogLevel(logger.TraceLevel), | ||||||
|  | 		AddSource:   s.opts.AddSource, | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	attrs, _ := s.argsAttrs(s.opts.Fields) | ||||||
|  |  | ||||||
|  | 	var h slog.Handler | ||||||
|  | 	if s.opts.Context != nil { | ||||||
|  | 		if v, ok := s.opts.Context.Value(handlerKey{}).(slog.Handler); ok && v != nil { | ||||||
|  | 			h = v | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if fn := s.opts.Context.Value(handlerFnKey{}); fn != nil { | ||||||
|  | 			if rfn := reflect.ValueOf(fn); rfn.Kind() == reflect.Func { | ||||||
|  | 				if ret := rfn.Call([]reflect.Value{reflect.ValueOf(s.opts.Out), reflect.ValueOf(handleOpt)}); len(ret) == 1 { | ||||||
|  | 					if iface, ok := ret[0].Interface().(slog.Handler); ok && iface != nil { | ||||||
|  | 						h = iface | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if h == nil { | ||||||
|  | 		h = slog.NewJSONHandler(s.opts.Out, handleOpt) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	s.handler = &wrapper{h: h.WithAttrs(attrs)} | ||||||
|  | 	s.handler.level.Store(int64(loggerToSlogLevel(s.opts.Level))) | ||||||
|  | 	s.mu.Unlock() | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *slogLogger) Log(ctx context.Context, lvl logger.Level, msg string, attrs ...interface{}) { | ||||||
|  | 	s.printLog(ctx, lvl, msg, attrs...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *slogLogger) Info(ctx context.Context, msg string, attrs ...interface{}) { | ||||||
|  | 	s.printLog(ctx, logger.InfoLevel, msg, attrs...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *slogLogger) Debug(ctx context.Context, msg string, attrs ...interface{}) { | ||||||
|  | 	s.printLog(ctx, logger.DebugLevel, msg, attrs...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *slogLogger) Trace(ctx context.Context, msg string, attrs ...interface{}) { | ||||||
|  | 	s.printLog(ctx, logger.TraceLevel, msg, attrs...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *slogLogger) Error(ctx context.Context, msg string, attrs ...interface{}) { | ||||||
|  | 	s.printLog(ctx, logger.ErrorLevel, msg, attrs...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *slogLogger) Fatal(ctx context.Context, msg string, attrs ...interface{}) { | ||||||
|  | 	s.printLog(ctx, logger.FatalLevel, msg, attrs...) | ||||||
|  | 	if closer, ok := s.opts.Out.(io.Closer); ok { | ||||||
|  | 		closer.Close() | ||||||
|  | 	} | ||||||
|  | 	time.Sleep(1 * time.Second) | ||||||
|  | 	os.Exit(1) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *slogLogger) Warn(ctx context.Context, msg string, attrs ...interface{}) { | ||||||
|  | 	s.printLog(ctx, logger.WarnLevel, msg, attrs...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *slogLogger) Name() string { | ||||||
|  | 	return s.opts.Name | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *slogLogger) String() string { | ||||||
|  | 	return "slog" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *slogLogger) printLog(ctx context.Context, lvl logger.Level, msg string, args ...interface{}) { | ||||||
|  | 	if !s.V(lvl) { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	var argError error | ||||||
|  |  | ||||||
|  | 	s.opts.Meter.Counter(semconv.LoggerMessageTotal, "level", lvl.String()).Inc() | ||||||
|  |  | ||||||
|  | 	attrs, err := s.argsAttrs(args) | ||||||
|  | 	if err != nil { | ||||||
|  | 		argError = err | ||||||
|  | 	} | ||||||
|  | 	if argError != nil { | ||||||
|  | 		if span, ok := tracer.SpanFromContext(ctx); ok { | ||||||
|  | 			span.SetStatus(tracer.SpanStatusError, argError.Error()) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for _, fn := range s.opts.ContextAttrFuncs { | ||||||
|  | 		ctxAttrs, err := s.argsAttrs(fn(ctx)) | ||||||
|  | 		if err != nil { | ||||||
|  | 			argError = err | ||||||
|  | 		} | ||||||
|  | 		attrs = append(attrs, ctxAttrs...) | ||||||
|  | 	} | ||||||
|  | 	if argError != nil { | ||||||
|  | 		if span, ok := tracer.SpanFromContext(ctx); ok { | ||||||
|  | 			span.SetStatus(tracer.SpanStatusError, argError.Error()) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if (s.opts.AddStacktrace || lvl == logger.FatalLevel) || (s.opts.AddStacktrace && lvl == logger.ErrorLevel) { | ||||||
|  | 		stackInfo := make([]byte, 1024*1024) | ||||||
|  | 		if stackSize := runtime.Stack(stackInfo, false); stackSize > 0 { | ||||||
|  | 			traceLines := reTrace.Split(string(stackInfo[:stackSize]), -1) | ||||||
|  | 			if len(traceLines) != 0 { | ||||||
|  | 				attrs = append(attrs, slog.String(s.opts.StacktraceKey, traceLines[len(traceLines)-1])) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	var pcs [1]uintptr | ||||||
|  | 	runtime.Callers(s.opts.CallerSkipCount, pcs[:]) // skip [Callers, printLog, LogLvlMethod] | ||||||
|  | 	r := slog.NewRecord(s.opts.TimeFunc(), loggerToSlogLevel(lvl), msg, pcs[0]) | ||||||
|  | 	r.AddAttrs(attrs...) | ||||||
|  | 	_ = s.handler.Handle(ctx, r) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func NewLogger(opts ...logger.Option) logger.Logger { | ||||||
|  | 	s := &slogLogger{ | ||||||
|  | 		opts: logger.NewOptions(opts...), | ||||||
|  | 	} | ||||||
|  | 	s.opts.CallerSkipCount = defaultCallerSkipCount | ||||||
|  |  | ||||||
|  | 	return s | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func loggerToSlogLevel(level logger.Level) slog.Level { | ||||||
|  | 	switch level { | ||||||
|  | 	case logger.DebugLevel: | ||||||
|  | 		return slog.LevelDebug | ||||||
|  | 	case logger.WarnLevel: | ||||||
|  | 		return slog.LevelWarn | ||||||
|  | 	case logger.ErrorLevel: | ||||||
|  | 		return slog.LevelError | ||||||
|  | 	case logger.TraceLevel: | ||||||
|  | 		return slog.LevelDebug - 1 | ||||||
|  | 	case logger.FatalLevel: | ||||||
|  | 		return slog.LevelError + 1 | ||||||
|  | 	default: | ||||||
|  | 		return slog.LevelInfo | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func slogToLoggerLevel(level slog.Level) logger.Level { | ||||||
|  | 	switch level { | ||||||
|  | 	case slog.LevelDebug: | ||||||
|  | 		return logger.DebugLevel | ||||||
|  | 	case slog.LevelWarn: | ||||||
|  | 		return logger.WarnLevel | ||||||
|  | 	case slog.LevelError: | ||||||
|  | 		return logger.ErrorLevel | ||||||
|  | 	case slog.LevelDebug - 1: | ||||||
|  | 		return logger.TraceLevel | ||||||
|  | 	case slog.LevelError + 1: | ||||||
|  | 		return logger.FatalLevel | ||||||
|  | 	default: | ||||||
|  | 		return logger.InfoLevel | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *slogLogger) argsAttrs(args []interface{}) ([]slog.Attr, error) { | ||||||
|  | 	attrs := make([]slog.Attr, 0, len(args)) | ||||||
|  | 	var err error | ||||||
|  |  | ||||||
|  | 	for idx := 0; idx < len(args); idx++ { | ||||||
|  | 		switch arg := args[idx].(type) { | ||||||
|  | 		case slog.Attr: | ||||||
|  | 			attrs = append(attrs, arg) | ||||||
|  | 		case string: | ||||||
|  | 			if idx+1 < len(args) { | ||||||
|  | 				attrs = append(attrs, slog.Any(arg, args[idx+1])) | ||||||
|  | 				idx++ | ||||||
|  | 			} else { | ||||||
|  | 				attrs = append(attrs, slog.String(badKey, arg)) | ||||||
|  | 			} | ||||||
|  | 		case error: | ||||||
|  | 			attrs = append(attrs, slog.String(s.opts.ErrorKey, arg.Error())) | ||||||
|  | 			err = arg | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return attrs, err | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type handlerKey struct{} | ||||||
|  |  | ||||||
|  | func WithHandler(h slog.Handler) logger.Option { | ||||||
|  | 	return logger.SetOption(handlerKey{}, h) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type handlerFnKey struct{} | ||||||
|  |  | ||||||
|  | func WithHandlerFunc(fn any) logger.Option { | ||||||
|  | 	return logger.SetOption(handlerFnKey{}, fn) | ||||||
|  | } | ||||||
							
								
								
									
										409
									
								
								logger/slog/slog_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										409
									
								
								logger/slog/slog_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,409 @@ | |||||||
|  | package slog | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"bytes" | ||||||
|  | 	"context" | ||||||
|  | 	"errors" | ||||||
|  | 	"fmt" | ||||||
|  | 	"log" | ||||||
|  | 	"log/slog" | ||||||
|  | 	"strings" | ||||||
|  | 	"testing" | ||||||
|  |  | ||||||
|  | 	"github.com/google/uuid" | ||||||
|  | 	"go.unistack.org/micro/v3/logger" | ||||||
|  | 	"go.unistack.org/micro/v3/metadata" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func TestStacktrace(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), | ||||||
|  | 	) | ||||||
|  | 	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:29`)) { | ||||||
|  | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestWithFields(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.InfoLevel), logger.WithOutput(buf), | ||||||
|  | 		WithHandlerFunc(slog.NewTextHandler), | ||||||
|  | 		logger.WithDedupKeys(true), | ||||||
|  | 	) | ||||||
|  | 	if err := l.Init(logger.WithFields("key1", "val1")); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l.Info(ctx, "msg1") | ||||||
|  |  | ||||||
|  | 	l = l.Fields("key1", "val2") | ||||||
|  |  | ||||||
|  | 	l.Info(ctx, "msg2") | ||||||
|  |  | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`msg=msg2 key1=val1`)) { | ||||||
|  | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestWithDedupKeysWithAddFields(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.InfoLevel), logger.WithOutput(buf), | ||||||
|  | 		WithHandlerFunc(slog.NewTextHandler), | ||||||
|  | 		logger.WithDedupKeys(true), | ||||||
|  | 	) | ||||||
|  | 	if err := l.Init(logger.WithFields("key1", "val1")); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l.Info(ctx, "msg1") | ||||||
|  |  | ||||||
|  | 	if err := l.Init(logger.WithAddFields("key2", "val2")); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l.Info(ctx, "msg2") | ||||||
|  |  | ||||||
|  | 	if err := l.Init(logger.WithAddFields("key2", "val3", "key1", "val4")); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l.Info(ctx, "msg3") | ||||||
|  |  | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`msg=msg3 key1=val1 key2=val2`)) { | ||||||
|  | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestWithHandlerFunc(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.InfoLevel), logger.WithOutput(buf), | ||||||
|  | 		WithHandlerFunc(slog.NewTextHandler), | ||||||
|  | 	) | ||||||
|  | 	if err := l.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l.Info(ctx, "msg1") | ||||||
|  |  | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`msg=msg1`)) { | ||||||
|  | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestWithAddFields(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.InfoLevel), logger.WithOutput(buf)) | ||||||
|  | 	if err := l.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l.Info(ctx, "msg1") | ||||||
|  |  | ||||||
|  | 	if err := l.Init(logger.WithAddFields("key1", "val1")); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  | 	l.Info(ctx, "msg2") | ||||||
|  |  | ||||||
|  | 	if err := l.Init(logger.WithAddFields("key2", "val2")); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  | 	l.Info(ctx, "msg3") | ||||||
|  |  | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`"key1"`)) { | ||||||
|  | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`"key2"`)) { | ||||||
|  | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestMultipleFieldsWithLevel(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.InfoLevel), logger.WithOutput(buf)) | ||||||
|  | 	if err := l.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l = l.Fields("key", "val") | ||||||
|  |  | ||||||
|  | 	l.Info(ctx, "msg1") | ||||||
|  | 	nl := l.Clone(logger.WithLevel(logger.DebugLevel)) | ||||||
|  | 	nl.Debug(ctx, "msg2") | ||||||
|  | 	l.Debug(ctx, "msg3") | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`"key":"val"`)) { | ||||||
|  | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`"msg1"`)) { | ||||||
|  | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`"msg2"`)) { | ||||||
|  | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | 	if bytes.Contains(buf.Bytes(), []byte(`"msg3"`)) { | ||||||
|  | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestMultipleFields(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.InfoLevel), logger.WithOutput(buf)) | ||||||
|  | 	if err := l.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l = l.Fields("key", "val") | ||||||
|  |  | ||||||
|  | 	l = l.Fields("key1", "val1") | ||||||
|  |  | ||||||
|  | 	l.Info(ctx, "msg") | ||||||
|  |  | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`"key":"val"`)) { | ||||||
|  | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`"key1":"val1"`)) { | ||||||
|  | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestError(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.ErrorLevel), logger.WithOutput(buf), logger.WithAddStacktrace(true)) | ||||||
|  | 	if err := l.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l.Error(ctx, "message", fmt.Errorf("error message")) | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`"stacktrace":"`)) { | ||||||
|  | 		t.Fatalf("logger stacktrace not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`"error":"`)) { | ||||||
|  | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestErrorf(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  |  | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.ErrorLevel), logger.WithOutput(buf), logger.WithAddStacktrace(true)) | ||||||
|  | 	if err := l.Init(logger.WithContextAttrFuncs(func(_ context.Context) []interface{} { | ||||||
|  | 		return nil | ||||||
|  | 	})); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l.Log(ctx, logger.ErrorLevel, "message", errors.New("error msg")) | ||||||
|  |  | ||||||
|  | 	l.Log(ctx, logger.ErrorLevel, "", errors.New("error msg")) | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`"error":"error msg"`)) { | ||||||
|  | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`"stacktrace":"`)) { | ||||||
|  | 		t.Fatalf("logger stacktrace not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`"error":"`)) { | ||||||
|  | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestContext(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.TraceLevel), logger.WithOutput(buf)) | ||||||
|  | 	if err := l.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	nl, ok := logger.FromContext(logger.NewContext(ctx, l.Fields("key", "val"))) | ||||||
|  | 	if !ok { | ||||||
|  | 		t.Fatal("context without logger") | ||||||
|  | 	} | ||||||
|  | 	nl.Info(ctx, "message") | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`"key":"val"`)) { | ||||||
|  | 		t.Fatalf("logger fields not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestFields(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.TraceLevel), logger.WithOutput(buf)) | ||||||
|  | 	if err := l.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	nl := l.Fields("key", "val") | ||||||
|  |  | ||||||
|  | 	nl.Info(ctx, "message") | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`"key":"val"`)) { | ||||||
|  | 		t.Fatalf("logger fields not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestFromContextWithFields(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	var ok bool | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.TraceLevel), logger.WithOutput(buf)) | ||||||
|  | 	if err := l.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  | 	nl := l.Fields("key", "val") | ||||||
|  |  | ||||||
|  | 	ctx = logger.NewContext(ctx, nl) | ||||||
|  |  | ||||||
|  | 	l, ok = logger.FromContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		t.Fatalf("context does not have logger") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l.Info(ctx, "message") | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`"key":"val"`)) { | ||||||
|  | 		t.Fatalf("logger fields not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l.Info(ctx, "test", "uncorrected number attributes") | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`"!BADKEY":"uncorrected number attributes"`)) { | ||||||
|  | 		t.Fatalf("logger fields not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestClone(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.TraceLevel), logger.WithOutput(buf)) | ||||||
|  | 	if err := l.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  | 	nl := l.Clone(logger.WithLevel(logger.ErrorLevel)) | ||||||
|  | 	if err := nl.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  | 	nl.Info(ctx, "info message") | ||||||
|  | 	if len(buf.Bytes()) != 0 { | ||||||
|  | 		t.Fatal("message must not be logged") | ||||||
|  | 	} | ||||||
|  | 	l.Info(ctx, "info message") | ||||||
|  | 	if len(buf.Bytes()) == 0 { | ||||||
|  | 		t.Fatal("message must be logged") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestRedirectStdLogger(t *testing.T) { | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.ErrorLevel), logger.WithOutput(buf)) | ||||||
|  | 	if err := l.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  | 	fn := logger.RedirectStdLogger(l, logger.ErrorLevel) | ||||||
|  | 	defer fn() | ||||||
|  | 	log.Print("test") | ||||||
|  | 	if !(bytes.Contains(buf.Bytes(), []byte(`"level":"error"`)) && bytes.Contains(buf.Bytes(), []byte(`"msg":"test"`))) { | ||||||
|  | 		t.Fatalf("logger error, buf %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestStdLogger(t *testing.T) { | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.TraceLevel), logger.WithOutput(buf)) | ||||||
|  | 	if err := l.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  | 	lg := logger.NewStdLogger(l, logger.ErrorLevel) | ||||||
|  | 	lg.Print("test") | ||||||
|  | 	if !(bytes.Contains(buf.Bytes(), []byte(`"level":"error"`)) && bytes.Contains(buf.Bytes(), []byte(`"msg":"test"`))) { | ||||||
|  | 		t.Fatalf("logger error, buf %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestLogger(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.TraceLevel), logger.WithOutput(buf)) | ||||||
|  | 	if err := l.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  | 	l.Trace(ctx, "trace_msg1") | ||||||
|  | 	l.Warn(ctx, "warn_msg1") | ||||||
|  | 	l.Fields("error", "test").Info(ctx, "error message") | ||||||
|  | 	l.Warn(ctx, "first second") | ||||||
|  |  | ||||||
|  | 	if !(bytes.Contains(buf.Bytes(), []byte(`"level":"trace"`)) && bytes.Contains(buf.Bytes(), []byte(`"msg":"trace_msg1"`))) { | ||||||
|  | 		t.Fatalf("logger tracer, buf %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | 	if !(bytes.Contains(buf.Bytes(), []byte(`"level":"warn"`)) && bytes.Contains(buf.Bytes(), []byte(`"msg":"warn_msg1"`))) { | ||||||
|  | 		t.Fatalf("logger warn, buf %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | 	if !(bytes.Contains(buf.Bytes(), []byte(`"level":"info"`)) && bytes.Contains(buf.Bytes(), []byte(`"msg":"error message","error":"test"`))) { | ||||||
|  | 		t.Fatalf("logger info, buf %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | 	if !(bytes.Contains(buf.Bytes(), []byte(`"level":"warn"`)) && bytes.Contains(buf.Bytes(), []byte(`"msg":"first second"`))) { | ||||||
|  | 		t.Fatalf("logger warn, buf %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func Test_WithContextAttrFunc(t *testing.T) { | ||||||
|  | 	loggerContextAttrFuncs := []logger.ContextAttrFunc{ | ||||||
|  | 		func(ctx context.Context) []interface{} { | ||||||
|  | 			md, ok := metadata.FromIncomingContext(ctx) | ||||||
|  | 			if !ok { | ||||||
|  | 				return nil | ||||||
|  | 			} | ||||||
|  | 			attrs := make([]interface{}, 0, 10) | ||||||
|  | 			for k, v := range md { | ||||||
|  | 				switch k { | ||||||
|  | 				case "X-Request-Id", "Phone", "External-Id", "Source-Service", "X-App-Install-Id", "Client-Id", "Client-Ip": | ||||||
|  | 					attrs = append(attrs, strings.ToLower(k), v) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			return attrs | ||||||
|  | 		}, | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	logger.DefaultContextAttrFuncs = append(logger.DefaultContextAttrFuncs, loggerContextAttrFuncs...) | ||||||
|  |  | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	ctx = metadata.AppendIncomingContext(ctx, "X-Request-Id", uuid.New().String(), | ||||||
|  | 		"Source-Service", "Test-System") | ||||||
|  |  | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.TraceLevel), logger.WithOutput(buf)) | ||||||
|  | 	if err := l.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l.Info(ctx, "test message") | ||||||
|  | 	if !(bytes.Contains(buf.Bytes(), []byte(`"level":"info"`)) && bytes.Contains(buf.Bytes(), []byte(`"msg":"test message"`))) { | ||||||
|  | 		t.Fatalf("logger info, buf %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | 	if !(bytes.Contains(buf.Bytes(), []byte(`"x-request-id":"`))) { | ||||||
|  | 		t.Fatalf("logger info, buf %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | 	if !(bytes.Contains(buf.Bytes(), []byte(`"source-service":"Test-System"`))) { | ||||||
|  | 		t.Fatalf("logger info, buf %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | 	buf.Reset() | ||||||
|  | 	imd, _ := metadata.FromIncomingContext(ctx) | ||||||
|  | 	l.Info(ctx, "test message1") | ||||||
|  | 	imd.Set("Source-Service", "Test-System2") | ||||||
|  | 	l.Info(ctx, "test message2") | ||||||
|  |  | ||||||
|  | 	// t.Logf("xxx %s", buf.Bytes()) | ||||||
|  | } | ||||||
| @@ -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-+# " | ||||||
| @@ -36,8 +36,8 @@ var ( | |||||||
| 	circularShortBytes = []byte("<shown>") | 	circularShortBytes = []byte("<shown>") | ||||||
| 	invalidAngleBytes  = []byte("<invalid>") | 	invalidAngleBytes  = []byte("<invalid>") | ||||||
| 	filteredBytes      = []byte("<filtered>") | 	filteredBytes      = []byte("<filtered>") | ||||||
| 	openBracketBytes   = []byte("[") | 	// openBracketBytes   = []byte("[") | ||||||
| 	closeBracketBytes  = []byte("]") | 	// closeBracketBytes  = []byte("]") | ||||||
| 	percentBytes    = []byte("%") | 	percentBytes    = []byte("%") | ||||||
| 	precisionBytes  = []byte(".") | 	precisionBytes  = []byte(".") | ||||||
| 	openAngleBytes  = []byte("<") | 	openAngleBytes  = []byte("<") | ||||||
| @@ -52,13 +52,15 @@ type protoMessage interface { | |||||||
| } | } | ||||||
|  |  | ||||||
| type Wrapper struct { | type Wrapper struct { | ||||||
|  | 	pointers map[uintptr]int | ||||||
|  | 	takeMap  map[int]bool | ||||||
|  |  | ||||||
| 	val  interface{} | 	val  interface{} | ||||||
| 	s    fmt.State | 	s    fmt.State | ||||||
| 	pointers         map[uintptr]int |  | ||||||
| 	opts *Options | 	opts *Options | ||||||
|  |  | ||||||
| 	depth            int | 	depth            int | ||||||
| 	ignoreNextType   bool | 	ignoreNextType   bool | ||||||
| 	takeMap          map[int]bool |  | ||||||
| 	protoWrapperType bool | 	protoWrapperType bool | ||||||
| 	sqlWrapperType   bool | 	sqlWrapperType   bool | ||||||
| } | } | ||||||
|   | |||||||
| @@ -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) { | ||||||
| @@ -82,11 +82,11 @@ func TestTagged(t *testing.T) { | |||||||
| func TestTaggedNested(t *testing.T) { | func TestTaggedNested(t *testing.T) { | ||||||
| 	type val struct { | 	type val struct { | ||||||
| 		key string `logger:"take"` | 		key string `logger:"take"` | ||||||
| 		val string `logger:"omit"` | 		// val string `logger:"omit"` | ||||||
| 		unk string | 		unk string | ||||||
| 	} | 	} | ||||||
| 	type str struct { | 	type str struct { | ||||||
| 		key string `logger:"omit"` | 		// key string `logger:"omit"` | ||||||
| 		val *val `logger:"take"` | 		val *val `logger:"take"` | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|   | |||||||
| @@ -1,297 +0,0 @@ | |||||||
| // Package wrapper provides wrapper for Logger |  | ||||||
| package wrapper |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"context" |  | ||||||
| 	"fmt" |  | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/client" |  | ||||||
| 	"go.unistack.org/micro/v4/logger" |  | ||||||
| 	"go.unistack.org/micro/v4/server" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| var ( |  | ||||||
| 	// DefaultClientCallObserver called by wrapper in client Call |  | ||||||
| 	DefaultClientCallObserver = func(ctx context.Context, req client.Request, rsp interface{}, opts []client.CallOption, err error) []string { |  | ||||||
| 		labels := []string{"service", req.Service(), "endpoint", req.Endpoint()} |  | ||||||
| 		if err != nil { |  | ||||||
| 			labels = append(labels, "error", err.Error()) |  | ||||||
| 		} |  | ||||||
| 		return labels |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// DefaultClientStreamObserver called by wrapper in client Stream |  | ||||||
| 	DefaultClientStreamObserver = func(ctx context.Context, req client.Request, opts []client.CallOption, stream client.Stream, err error) []string { |  | ||||||
| 		labels := []string{"service", req.Service(), "endpoint", req.Endpoint()} |  | ||||||
| 		if err != nil { |  | ||||||
| 			labels = append(labels, "error", err.Error()) |  | ||||||
| 		} |  | ||||||
| 		return labels |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// DefaultServerHandlerObserver called by wrapper in server Handler |  | ||||||
| 	DefaultServerHandlerObserver = func(ctx context.Context, req server.Request, rsp interface{}, err error) []string { |  | ||||||
| 		labels := []string{"service", req.Service(), "endpoint", req.Endpoint()} |  | ||||||
| 		if err != nil { |  | ||||||
| 			labels = append(labels, "error", err.Error()) |  | ||||||
| 		} |  | ||||||
| 		return labels |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// DefaultClientCallFuncObserver called by wrapper in client CallFunc |  | ||||||
| 	DefaultClientCallFuncObserver = func(ctx context.Context, addr string, req client.Request, rsp interface{}, opts client.CallOptions, err error) []string { |  | ||||||
| 		labels := []string{"service", req.Service(), "endpoint", req.Endpoint()} |  | ||||||
| 		if err != nil { |  | ||||||
| 			labels = append(labels, "error", err.Error()) |  | ||||||
| 		} |  | ||||||
| 		return labels |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// DefaultSkipEndpoints wrapper not called for this endpoints |  | ||||||
| 	DefaultSkipEndpoints = []string{"Meter.Metrics", "Health.Live", "Health.Ready", "Health.Version"} |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| type lWrapper struct { |  | ||||||
| 	client.Client |  | ||||||
| 	serverHandler  server.HandlerFunc |  | ||||||
| 	clientCallFunc client.CallFunc |  | ||||||
| 	opts           Options |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type ( |  | ||||||
| 	// ClientCallObserver func signature |  | ||||||
| 	ClientCallObserver func(context.Context, client.Request, interface{}, []client.CallOption, error) []string |  | ||||||
| 	// ClientStreamObserver func signature |  | ||||||
| 	ClientStreamObserver func(context.Context, client.Request, []client.CallOption, client.Stream, error) []string |  | ||||||
| 	// ClientCallFuncObserver func signature |  | ||||||
| 	ClientCallFuncObserver func(context.Context, string, client.Request, interface{}, client.CallOptions, error) []string |  | ||||||
| 	// ServerHandlerObserver func signature |  | ||||||
| 	ServerHandlerObserver func(context.Context, server.Request, interface{}, error) []string |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // Options struct for wrapper |  | ||||||
| type Options struct { |  | ||||||
| 	// Logger that used for log |  | ||||||
| 	Logger logger.Logger |  | ||||||
| 	// ServerHandlerObservers funcs |  | ||||||
| 	ServerHandlerObservers []ServerHandlerObserver |  | ||||||
| 	// ClientCallObservers funcs |  | ||||||
| 	ClientCallObservers []ClientCallObserver |  | ||||||
| 	// ClientStreamObservers funcs |  | ||||||
| 	ClientStreamObservers []ClientStreamObserver |  | ||||||
| 	// ClientCallFuncObservers funcs |  | ||||||
| 	ClientCallFuncObservers []ClientCallFuncObserver |  | ||||||
| 	// SkipEndpoints |  | ||||||
| 	SkipEndpoints []string |  | ||||||
| 	// Level for logger |  | ||||||
| 	Level logger.Level |  | ||||||
| 	// Enabled flag |  | ||||||
| 	Enabled bool |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Option func signature |  | ||||||
| type Option func(*Options) |  | ||||||
|  |  | ||||||
| // NewOptions creates Options from Option slice |  | ||||||
| func NewOptions(opts ...Option) Options { |  | ||||||
| 	options := Options{ |  | ||||||
| 		Logger:                  logger.DefaultLogger, |  | ||||||
| 		Level:                   logger.TraceLevel, |  | ||||||
| 		ClientCallObservers:     []ClientCallObserver{DefaultClientCallObserver}, |  | ||||||
| 		ClientStreamObservers:   []ClientStreamObserver{DefaultClientStreamObserver}, |  | ||||||
| 		ClientCallFuncObservers: []ClientCallFuncObserver{DefaultClientCallFuncObserver}, |  | ||||||
| 		ServerHandlerObservers:  []ServerHandlerObserver{DefaultServerHandlerObserver}, |  | ||||||
| 		SkipEndpoints:           DefaultSkipEndpoints, |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	for _, o := range opts { |  | ||||||
| 		o(&options) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return options |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // WithEnabled enable/diable flag |  | ||||||
| func WithEnabled(b bool) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.Enabled = b |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // WithLevel log level |  | ||||||
| func WithLevel(l logger.Level) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.Level = l |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // WithLogger logger |  | ||||||
| func WithLogger(l logger.Logger) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.Logger = l |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // WithClientCallObservers funcs |  | ||||||
| func WithClientCallObservers(ob ...ClientCallObserver) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.ClientCallObservers = ob |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // WithClientStreamObservers funcs |  | ||||||
| func WithClientStreamObservers(ob ...ClientStreamObserver) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.ClientStreamObservers = ob |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // WithClientCallFuncObservers funcs |  | ||||||
| func WithClientCallFuncObservers(ob ...ClientCallFuncObserver) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.ClientCallFuncObservers = ob |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // WithServerHandlerObservers funcs |  | ||||||
| func WithServerHandlerObservers(ob ...ServerHandlerObserver) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.ServerHandlerObservers = ob |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // SkipEndpoins |  | ||||||
| func SkipEndpoints(eps ...string) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.SkipEndpoints = append(o.SkipEndpoints, eps...) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *lWrapper) Call(ctx context.Context, req client.Request, rsp interface{}, opts ...client.CallOption) error { |  | ||||||
| 	err := l.Client.Call(ctx, req, rsp, opts...) |  | ||||||
|  |  | ||||||
| 	endpoint := fmt.Sprintf("%s.%s", req.Service(), req.Endpoint()) |  | ||||||
| 	for _, ep := range l.opts.SkipEndpoints { |  | ||||||
| 		if ep == endpoint { |  | ||||||
| 			return err |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if !l.opts.Enabled { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	var labels []string |  | ||||||
| 	for _, o := range l.opts.ClientCallObservers { |  | ||||||
| 		labels = append(labels, o(ctx, req, rsp, opts, err)...) |  | ||||||
| 	} |  | ||||||
| 	l.opts.Logger.Fields(labels).Log(ctx, l.opts.Level) |  | ||||||
|  |  | ||||||
| 	return err |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *lWrapper) Stream(ctx context.Context, req client.Request, opts ...client.CallOption) (client.Stream, error) { |  | ||||||
| 	stream, err := l.Client.Stream(ctx, req, opts...) |  | ||||||
|  |  | ||||||
| 	endpoint := fmt.Sprintf("%s.%s", req.Service(), req.Endpoint()) |  | ||||||
| 	for _, ep := range l.opts.SkipEndpoints { |  | ||||||
| 		if ep == endpoint { |  | ||||||
| 			return stream, err |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if !l.opts.Enabled { |  | ||||||
| 		return stream, err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	var labels []string |  | ||||||
| 	for _, o := range l.opts.ClientStreamObservers { |  | ||||||
| 		labels = append(labels, o(ctx, req, opts, stream, err)...) |  | ||||||
| 	} |  | ||||||
| 	l.opts.Logger.Fields(labels).Log(ctx, l.opts.Level) |  | ||||||
|  |  | ||||||
| 	return stream, err |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *lWrapper) ServerHandler(ctx context.Context, req server.Request, rsp interface{}) error { |  | ||||||
| 	err := l.serverHandler(ctx, req, rsp) |  | ||||||
|  |  | ||||||
| 	endpoint := req.Endpoint() |  | ||||||
| 	for _, ep := range l.opts.SkipEndpoints { |  | ||||||
| 		if ep == endpoint { |  | ||||||
| 			return err |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if !l.opts.Enabled { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	var labels []string |  | ||||||
| 	for _, o := range l.opts.ServerHandlerObservers { |  | ||||||
| 		labels = append(labels, o(ctx, req, rsp, err)...) |  | ||||||
| 	} |  | ||||||
| 	l.opts.Logger.Fields(labels).Log(ctx, l.opts.Level) |  | ||||||
|  |  | ||||||
| 	return err |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewClientWrapper accepts an open options and returns a Client Wrapper |  | ||||||
| func NewClientWrapper(opts ...Option) client.Wrapper { |  | ||||||
| 	return func(c client.Client) client.Client { |  | ||||||
| 		options := NewOptions() |  | ||||||
| 		for _, o := range opts { |  | ||||||
| 			o(&options) |  | ||||||
| 		} |  | ||||||
| 		return &lWrapper{opts: options, Client: c} |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewClientCallWrapper accepts an options and returns a Call Wrapper |  | ||||||
| func NewClientCallWrapper(opts ...Option) client.CallWrapper { |  | ||||||
| 	return func(h client.CallFunc) client.CallFunc { |  | ||||||
| 		options := NewOptions() |  | ||||||
| 		for _, o := range opts { |  | ||||||
| 			o(&options) |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		l := &lWrapper{opts: options, clientCallFunc: h} |  | ||||||
| 		return l.ClientCallFunc |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *lWrapper) ClientCallFunc(ctx context.Context, addr string, req client.Request, rsp interface{}, opts client.CallOptions) error { |  | ||||||
| 	err := l.clientCallFunc(ctx, addr, req, rsp, opts) |  | ||||||
|  |  | ||||||
| 	endpoint := fmt.Sprintf("%s.%s", req.Service(), req.Endpoint()) |  | ||||||
| 	for _, ep := range l.opts.SkipEndpoints { |  | ||||||
| 		if ep == endpoint { |  | ||||||
| 			return err |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if !l.opts.Enabled { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	var labels []string |  | ||||||
| 	for _, o := range l.opts.ClientCallFuncObservers { |  | ||||||
| 		labels = append(labels, o(ctx, addr, req, rsp, opts, err)...) |  | ||||||
| 	} |  | ||||||
| 	l.opts.Logger.Fields(labels).Log(ctx, l.opts.Level) |  | ||||||
|  |  | ||||||
| 	return err |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewServerHandlerWrapper accepts an options and returns a Handler Wrapper |  | ||||||
| func NewServerHandlerWrapper(opts ...Option) server.HandlerWrapper { |  | ||||||
| 	return func(h server.HandlerFunc) server.HandlerFunc { |  | ||||||
| 		options := NewOptions() |  | ||||||
| 		for _, o := range opts { |  | ||||||
| 			o(&options) |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		l := &lWrapper{opts: options, serverHandler: h} |  | ||||||
| 		return l.ServerHandler |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -24,6 +24,17 @@ func FromIncomingContext(ctx context.Context) (Metadata, bool) { | |||||||
| 	return md.md, ok | 	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 | // FromOutgoingContext returns metadata from outgoing ctx | ||||||
| // returned metadata shoud not be modified or race condition happens | // returned metadata shoud not be modified or race condition happens | ||||||
| func FromOutgoingContext(ctx context.Context) (Metadata, bool) { | func FromOutgoingContext(ctx context.Context) (Metadata, bool) { | ||||||
| @@ -37,6 +48,17 @@ func FromOutgoingContext(ctx context.Context) (Metadata, bool) { | |||||||
| 	return md.md, ok | 	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 | // FromContext returns metadata from the given context | ||||||
| // returned metadata shoud not be modified or race condition happens | // returned metadata shoud not be modified or race condition happens | ||||||
| func FromContext(ctx context.Context) (Metadata, bool) { | func FromContext(ctx context.Context) (Metadata, bool) { | ||||||
| @@ -50,15 +72,22 @@ func FromContext(ctx context.Context) (Metadata, bool) { | |||||||
| 	return md.md, ok | 	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 | // NewContext creates a new context with the given metadata | ||||||
| func NewContext(ctx context.Context, md Metadata) context.Context { | func NewContext(ctx context.Context, md Metadata) context.Context { | ||||||
| 	if ctx == nil { | 	if ctx == nil { | ||||||
| 		ctx = context.Background() | 		ctx = context.Background() | ||||||
| 	} | 	} | ||||||
| 	ctx = context.WithValue(ctx, mdKey{}, &rawMetadata{md}) | 	return context.WithValue(ctx, mdKey{}, &rawMetadata{md}) | ||||||
| 	ctx = context.WithValue(ctx, mdIncomingKey{}, &rawMetadata{}) |  | ||||||
| 	ctx = context.WithValue(ctx, mdOutgoingKey{}, &rawMetadata{}) |  | ||||||
| 	return ctx |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // SetOutgoingContext modify outgoing context with given metadata | // SetOutgoingContext modify outgoing context with given metadata | ||||||
| @@ -90,11 +119,7 @@ func NewIncomingContext(ctx context.Context, md Metadata) context.Context { | |||||||
| 	if ctx == nil { | 	if ctx == nil { | ||||||
| 		ctx = context.Background() | 		ctx = context.Background() | ||||||
| 	} | 	} | ||||||
| 	ctx = context.WithValue(ctx, mdIncomingKey{}, &rawMetadata{md}) | 	return context.WithValue(ctx, mdIncomingKey{}, &rawMetadata{md}) | ||||||
| 	if v, ok := ctx.Value(mdOutgoingKey{}).(*rawMetadata); !ok || v == nil { |  | ||||||
| 		ctx = context.WithValue(ctx, mdOutgoingKey{}, &rawMetadata{}) |  | ||||||
| 	} |  | ||||||
| 	return ctx |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewOutgoingContext creates a new context with outcoming metadata attached | // NewOutgoingContext creates a new context with outcoming metadata attached | ||||||
| @@ -102,11 +127,7 @@ func NewOutgoingContext(ctx context.Context, md Metadata) context.Context { | |||||||
| 	if ctx == nil { | 	if ctx == nil { | ||||||
| 		ctx = context.Background() | 		ctx = context.Background() | ||||||
| 	} | 	} | ||||||
| 	ctx = context.WithValue(ctx, mdOutgoingKey{}, &rawMetadata{md}) | 	return context.WithValue(ctx, mdOutgoingKey{}, &rawMetadata{md}) | ||||||
| 	if v, ok := ctx.Value(mdIncomingKey{}).(*rawMetadata); !ok || v == nil { |  | ||||||
| 		ctx = context.WithValue(ctx, mdIncomingKey{}, &rawMetadata{}) |  | ||||||
| 	} |  | ||||||
| 	return ctx |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // AppendOutgoingContext apends new md to context | // AppendOutgoingContext apends new md to context | ||||||
| @@ -122,7 +143,7 @@ func AppendOutgoingContext(ctx context.Context, kv ...string) context.Context { | |||||||
| 	for k, v := range md { | 	for k, v := range md { | ||||||
| 		omd.Set(k, v) | 		omd.Set(k, v) | ||||||
| 	} | 	} | ||||||
| 	return NewOutgoingContext(ctx, omd) | 	return ctx | ||||||
| } | } | ||||||
|  |  | ||||||
| // AppendIncomingContext apends new md to context | // AppendIncomingContext apends new md to context | ||||||
| @@ -138,5 +159,21 @@ func AppendIncomingContext(ctx context.Context, kv ...string) context.Context { | |||||||
| 	for k, v := range md { | 	for k, v := range md { | ||||||
| 		omd.Set(k, v) | 		omd.Set(k, v) | ||||||
| 	} | 	} | ||||||
| 	return NewIncomingContext(ctx, omd) | 	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 | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,9 +1,10 @@ | |||||||
| // Package metadata is a way of defining message headers | // Package metadata is a way of defining message headers | ||||||
| package metadata // import "go.unistack.org/micro/v4/metadata" | package metadata | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"net/textproto" | 	"net/textproto" | ||||||
| 	"sort" | 	"sort" | ||||||
|  | 	"strings" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| @@ -19,6 +20,8 @@ var ( | |||||||
| 	HeaderTimeout = "Micro-Timeout" | 	HeaderTimeout = "Micro-Timeout" | ||||||
| 	// HeaderAuthorization specifies Authorization header | 	// HeaderAuthorization specifies Authorization header | ||||||
| 	HeaderAuthorization = "Authorization" | 	HeaderAuthorization = "Authorization" | ||||||
|  | 	// HeaderXRequestID specifies request id | ||||||
|  | 	HeaderXRequestID = "X-Request-Id" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Metadata is our way of representing request headers internally. | // Metadata is our way of representing request headers internally. | ||||||
| @@ -64,6 +67,14 @@ func (md Metadata) Iterator() *Iterator { | |||||||
| 	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 | // Get returns value from metadata by key | ||||||
| func (md Metadata) Get(key string) (string, bool) { | func (md Metadata) Get(key string) (string, bool) { | ||||||
| 	// fast path | 	// fast path | ||||||
| @@ -71,6 +82,9 @@ func (md Metadata) Get(key string) (string, bool) { | |||||||
| 	if !ok { | 	if !ok { | ||||||
| 		// slow path | 		// slow path | ||||||
| 		val, ok = md[textproto.CanonicalMIMEHeaderKey(key)] | 		val, ok = md[textproto.CanonicalMIMEHeaderKey(key)] | ||||||
|  | 		if !ok { | ||||||
|  | 			val, ok = md[strings.ToLower(key)] | ||||||
|  | 		} | ||||||
| 	} | 	} | ||||||
| 	return val, ok | 	return val, ok | ||||||
| } | } | ||||||
| @@ -92,15 +106,25 @@ func (md Metadata) Del(keys ...string) { | |||||||
| 		delete(md, key) | 		delete(md, key) | ||||||
| 		// slow path | 		// slow path | ||||||
| 		delete(md, textproto.CanonicalMIMEHeaderKey(key)) | 		delete(md, textproto.CanonicalMIMEHeaderKey(key)) | ||||||
|  | 		// very slow path | ||||||
|  | 		delete(md, strings.ToLower(key)) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // Copy makes a copy of the metadata | // Copy makes a copy of the metadata | ||||||
| func Copy(md Metadata) Metadata { | func (md Metadata) CopyTo(dst Metadata) { | ||||||
| 	nmd := New(len(md)) | 	for k, v := range md { | ||||||
| 	for key, val := range md { | 		dst[k] = v | ||||||
| 		nmd.Set(key, val) |  | ||||||
| 	} | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // 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 | 	return nmd | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -122,7 +146,7 @@ func Merge(omd Metadata, mmd Metadata, overwrite bool) Metadata { | |||||||
| 		case ok && !overwrite: | 		case ok && !overwrite: | ||||||
| 			continue | 			continue | ||||||
| 		case val != "": | 		case val != "": | ||||||
| 			nmd.Set(key, val) | 			nmd[key] = val | ||||||
| 		case ok && val == "": | 		case ok && val == "": | ||||||
| 			nmd.Del(key) | 			nmd.Del(key) | ||||||
| 		} | 		} | ||||||
| @@ -136,6 +160,8 @@ func Pairs(kv ...string) (Metadata, bool) { | |||||||
| 		return nil, false | 		return nil, false | ||||||
| 	} | 	} | ||||||
| 	md := New(len(kv) / 2) | 	md := New(len(kv) / 2) | ||||||
| 	md.Set(kv...) | 	for idx := 0; idx < len(kv); idx += 2 { | ||||||
|  | 		md[kv[idx]] = kv[idx+1] | ||||||
|  | 	} | ||||||
| 	return md, true | 	return md, true | ||||||
| } | } | ||||||
|   | |||||||
| @@ -5,6 +5,37 @@ import ( | |||||||
| 	"testing" | 	"testing" | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | func TestLowercase(t *testing.T) { | ||||||
|  | 	md := New(1) | ||||||
|  | 	md["x-request-id"] = "12345" | ||||||
|  | 	v, ok := md.Get("X-Request-Id") | ||||||
|  | 	if !ok || v == "" { | ||||||
|  | 		t.Fatalf("metadata invalid %#+v", md) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestMultipleUsage(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	md := New(0) | ||||||
|  | 	md.Set("key1_1", "val1_1", "key1_2", "val1_2", "key1_3", "val1_3") | ||||||
|  | 	ctx = NewIncomingContext(ctx, Copy(md)) | ||||||
|  | 	ctx = NewOutgoingContext(ctx, Copy(md)) | ||||||
|  | 	imd, _ := FromIncomingContext(ctx) | ||||||
|  | 	omd, _ := FromOutgoingContext(ctx) | ||||||
|  | 	_ = func(x context.Context) context.Context { | ||||||
|  | 		m, _ := FromIncomingContext(x) | ||||||
|  | 		m.Del("key1_2") | ||||||
|  | 		return ctx | ||||||
|  | 	}(ctx) | ||||||
|  | 	_ = func(x context.Context) context.Context { | ||||||
|  | 		m, _ := FromIncomingContext(x) | ||||||
|  | 		m.Del("key1_3") | ||||||
|  | 		return ctx | ||||||
|  | 	}(ctx) | ||||||
|  | 	_ = imd | ||||||
|  | 	_ = omd | ||||||
|  | } | ||||||
|  |  | ||||||
| func TestMetadataSetMultiple(t *testing.T) { | func TestMetadataSetMultiple(t *testing.T) { | ||||||
| 	md := New(4) | 	md := New(4) | ||||||
| 	md.Set("key1", "val1", "key2", "val2", "key3") | 	md.Set("key1", "val1", "key2", "val2", "key3") | ||||||
| @@ -57,6 +88,13 @@ func TestPassing(t *testing.T) { | |||||||
|  |  | ||||||
| 	ctx = NewIncomingContext(ctx, md1) | 	ctx = NewIncomingContext(ctx, md1) | ||||||
| 	testCtx(ctx) | 	testCtx(ctx) | ||||||
|  | 	_, ok := FromOutgoingContext(ctx) | ||||||
|  | 	if ok { | ||||||
|  | 		t.Fatalf("create outgoing context") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	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") | ||||||
| @@ -80,7 +118,7 @@ func TestMerge(t *testing.T) { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func TestIterator(t *testing.T) { | func TestIterator(_ *testing.T) { | ||||||
| 	md := Metadata{ | 	md := Metadata{ | ||||||
| 		"1Last":   "last", | 		"1Last":   "last", | ||||||
| 		"2First":  "first", | 		"2First":  "first", | ||||||
| @@ -190,3 +228,14 @@ func TestMetadataContext(t *testing.T) { | |||||||
| 		t.Errorf("Expected metadata length 1 got %d", i) | 		t.Errorf("Expected metadata length 1 got %d", i) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func TestCopy(t *testing.T) { | ||||||
|  | 	md := New(2) | ||||||
|  | 	md.Set("key1", "val1", "key2", "val2") | ||||||
|  | 	nmd := Copy(md, "key2") | ||||||
|  | 	if len(nmd) != 1 { | ||||||
|  | 		t.Fatal("Copy exclude not works") | ||||||
|  | 	} else if nmd["Key1"] != "val1" { | ||||||
|  | 		t.Fatal("Copy exclude not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -15,6 +15,15 @@ func FromContext(ctx context.Context) (Meter, bool) { | |||||||
| 	return c, ok | 	return c, ok | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // MustContext get meter from context | ||||||
|  | func MustContext(ctx context.Context) Meter { | ||||||
|  | 	m, ok := FromContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		panic("missing meter") | ||||||
|  | 	} | ||||||
|  | 	return m | ||||||
|  | } | ||||||
|  |  | ||||||
| // NewContext put meter in context | // NewContext put meter in context | ||||||
| func NewContext(ctx context.Context, c Meter) context.Context { | func NewContext(ctx context.Context, c Meter) context.Context { | ||||||
| 	if ctx == nil { | 	if ctx == nil { | ||||||
| @@ -22,3 +31,13 @@ func NewContext(ctx context.Context, c Meter) context.Context { | |||||||
| 	} | 	} | ||||||
| 	return context.WithValue(ctx, meterKey{}, c) | 	return context.WithValue(ctx, meterKey{}, 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) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -40,3 +40,14 @@ func TestNewContext(t *testing.T) { | |||||||
| 		t.Fatal("NewContext not works") | 		t.Fatal("NewContext not works") | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func TestSetOption(t *testing.T) { | ||||||
|  | 	type key struct{} | ||||||
|  | 	o := SetOption(key{}, "test") | ||||||
|  | 	opts := &Options{} | ||||||
|  | 	o(opts) | ||||||
|  |  | ||||||
|  | 	if v, ok := opts.Context.Value(key{}).(string); !ok || v == "" { | ||||||
|  | 		t.Fatal("SetOption not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -1,5 +1,5 @@ | |||||||
| // Package meter is for instrumentation | // Package meter is for instrumentation | ||||||
| package meter // import "go.unistack.org/micro/v4/meter" | package meter | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"io" | 	"io" | ||||||
| @@ -7,8 +7,6 @@ import ( | |||||||
| 	"strconv" | 	"strconv" | ||||||
| 	"strings" | 	"strings" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/options" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| @@ -18,14 +16,19 @@ var ( | |||||||
| 	DefaultAddress = ":9090" | 	DefaultAddress = ":9090" | ||||||
| 	// DefaultPath the meter endpoint where the Meter data will be made available | 	// DefaultPath the meter endpoint where the Meter data will be made available | ||||||
| 	DefaultPath = "/metrics" | 	DefaultPath = "/metrics" | ||||||
| 	// DefaultMetricPrefix holds the string that prepends to all metrics | 	// DefaultMeterStatsInterval specifies interval for meter updating | ||||||
| 	DefaultMetricPrefix = "micro_" | 	DefaultMeterStatsInterval = 5 * time.Second | ||||||
| 	// DefaultLabelPrefix holds the string that prepends to all labels |  | ||||||
| 	DefaultLabelPrefix = "micro_" |  | ||||||
| 	// DefaultSummaryQuantiles is the default spread of stats for summary | 	// DefaultSummaryQuantiles is the default spread of stats for summary | ||||||
| 	DefaultSummaryQuantiles = []float64{0.5, 0.9, 0.97, 0.99, 1} | 	DefaultSummaryQuantiles = []float64{0.5, 0.9, 0.97, 0.99, 1} | ||||||
| 	// DefaultSummaryWindow is the default window for summary | 	// DefaultSummaryWindow is the default window for summary | ||||||
| 	DefaultSummaryWindow = 5 * time.Minute | 	DefaultSummaryWindow = 5 * time.Minute | ||||||
|  | 	// DefaultSkipEndpoints is the slice of endpoint that must not be metered | ||||||
|  | 	DefaultSkipEndpoints = []string{ | ||||||
|  | 		"MeterService.Metrics", | ||||||
|  | 		"HealthService.Live", | ||||||
|  | 		"HealthService.Ready", | ||||||
|  | 		"HealthService.Version", | ||||||
|  | 	} | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Meter is an interface for collecting and instrumenting metrics | // Meter is an interface for collecting and instrumenting metrics | ||||||
| @@ -33,9 +36,9 @@ type Meter interface { | |||||||
| 	// Name returns meter name | 	// Name returns meter name | ||||||
| 	Name() string | 	Name() string | ||||||
| 	// Init initialize meter | 	// Init initialize meter | ||||||
| 	Init(opts ...options.Option) error | 	Init(opts ...Option) error | ||||||
| 	// Clone create meter copy with new options | 	// Clone create meter copy with new options | ||||||
| 	Clone(opts ...options.Option) Meter | 	Clone(opts ...Option) Meter | ||||||
| 	// Counter get or create counter | 	// Counter get or create counter | ||||||
| 	Counter(name string, labels ...string) Counter | 	Counter(name string, labels ...string) Counter | ||||||
| 	// FloatCounter get or create float counter | 	// FloatCounter get or create float counter | ||||||
| @@ -43,7 +46,7 @@ type Meter interface { | |||||||
| 	// Gauge get or create gauge | 	// Gauge get or create gauge | ||||||
| 	Gauge(name string, fn func() float64, labels ...string) Gauge | 	Gauge(name string, fn func() float64, labels ...string) Gauge | ||||||
| 	// Set create new meter metrics set | 	// Set create new meter metrics set | ||||||
| 	Set(opts ...options.Option) Meter | 	Set(opts ...Option) Meter | ||||||
| 	// Histogram get or create histogram | 	// Histogram get or create histogram | ||||||
| 	Histogram(name string, labels ...string) Histogram | 	Histogram(name string, labels ...string) Histogram | ||||||
| 	// Summary get or create summary | 	// Summary get or create summary | ||||||
| @@ -51,7 +54,7 @@ type Meter interface { | |||||||
| 	// SummaryExt get or create summary with spcified quantiles and window time | 	// SummaryExt get or create summary with spcified quantiles and window time | ||||||
| 	SummaryExt(name string, window time.Duration, quantiles []float64, labels ...string) Summary | 	SummaryExt(name string, window time.Duration, quantiles []float64, labels ...string) Summary | ||||||
| 	// Write writes metrics to io.Writer | 	// Write writes metrics to io.Writer | ||||||
| 	Write(w io.Writer, opts ...options.Option) error | 	Write(w io.Writer, opts ...Option) error | ||||||
| 	// Options returns meter options | 	// Options returns meter options | ||||||
| 	Options() Options | 	Options() Options | ||||||
| 	// String return meter type | 	// String return meter type | ||||||
|   | |||||||
| @@ -3,8 +3,6 @@ package meter | |||||||
| import ( | import ( | ||||||
| 	"io" | 	"io" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/options" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // NoopMeter is an noop implementation of Meter | // NoopMeter is an noop implementation of Meter | ||||||
| @@ -13,12 +11,12 @@ type noopMeter struct { | |||||||
| } | } | ||||||
|  |  | ||||||
| // NewMeter returns a configured noop reporter: | // NewMeter returns a configured noop reporter: | ||||||
| func NewMeter(opts ...options.Option) Meter { | func NewMeter(opts ...Option) Meter { | ||||||
| 	return &noopMeter{opts: NewOptions(opts...)} | 	return &noopMeter{opts: NewOptions(opts...)} | ||||||
| } | } | ||||||
|  |  | ||||||
| // Clone return old meter with new options | // Clone return old meter with new options | ||||||
| func (r *noopMeter) Clone(opts ...options.Option) Meter { | func (r *noopMeter) Clone(opts ...Option) Meter { | ||||||
| 	options := r.opts | 	options := r.opts | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&options) | 		o(&options) | ||||||
| @@ -31,7 +29,7 @@ func (r *noopMeter) Name() string { | |||||||
| } | } | ||||||
|  |  | ||||||
| // Init initialize options | // Init initialize options | ||||||
| func (r *noopMeter) Init(opts ...options.Option) error { | func (r *noopMeter) Init(opts ...Option) error { | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&r.opts) | 		o(&r.opts) | ||||||
| 	} | 	} | ||||||
| @@ -39,37 +37,37 @@ func (r *noopMeter) Init(opts ...options.Option) error { | |||||||
| } | } | ||||||
|  |  | ||||||
| // Counter implements the Meter interface | // Counter implements the Meter interface | ||||||
| func (r *noopMeter) Counter(name string, labels ...string) Counter { | func (r *noopMeter) Counter(_ string, labels ...string) Counter { | ||||||
| 	return &noopCounter{labels: labels} | 	return &noopCounter{labels: labels} | ||||||
| } | } | ||||||
|  |  | ||||||
| // FloatCounter implements the Meter interface | // FloatCounter implements the Meter interface | ||||||
| func (r *noopMeter) FloatCounter(name string, labels ...string) FloatCounter { | func (r *noopMeter) FloatCounter(_ string, labels ...string) FloatCounter { | ||||||
| 	return &noopFloatCounter{labels: labels} | 	return &noopFloatCounter{labels: labels} | ||||||
| } | } | ||||||
|  |  | ||||||
| // Gauge implements the Meter interface | // Gauge implements the Meter interface | ||||||
| func (r *noopMeter) Gauge(name string, f func() float64, labels ...string) Gauge { | func (r *noopMeter) Gauge(_ string, _ func() float64, labels ...string) Gauge { | ||||||
| 	return &noopGauge{labels: labels} | 	return &noopGauge{labels: labels} | ||||||
| } | } | ||||||
|  |  | ||||||
| // Summary implements the Meter interface | // Summary implements the Meter interface | ||||||
| func (r *noopMeter) Summary(name string, labels ...string) Summary { | func (r *noopMeter) Summary(_ string, labels ...string) Summary { | ||||||
| 	return &noopSummary{labels: labels} | 	return &noopSummary{labels: labels} | ||||||
| } | } | ||||||
|  |  | ||||||
| // SummaryExt implements the Meter interface | // SummaryExt implements the Meter interface | ||||||
| func (r *noopMeter) SummaryExt(name string, window time.Duration, quantiles []float64, labels ...string) Summary { | func (r *noopMeter) SummaryExt(_ string, _ time.Duration, _ []float64, labels ...string) Summary { | ||||||
| 	return &noopSummary{labels: labels} | 	return &noopSummary{labels: labels} | ||||||
| } | } | ||||||
|  |  | ||||||
| // Histogram implements the Meter interface | // Histogram implements the Meter interface | ||||||
| func (r *noopMeter) Histogram(name string, labels ...string) Histogram { | func (r *noopMeter) Histogram(_ string, labels ...string) Histogram { | ||||||
| 	return &noopHistogram{labels: labels} | 	return &noopHistogram{labels: labels} | ||||||
| } | } | ||||||
|  |  | ||||||
| // Set implements the Meter interface | // Set implements the Meter interface | ||||||
| func (r *noopMeter) Set(opts ...options.Option) Meter { | func (r *noopMeter) Set(opts ...Option) Meter { | ||||||
| 	m := &noopMeter{opts: r.opts} | 	m := &noopMeter{opts: r.opts} | ||||||
|  |  | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| @@ -79,7 +77,7 @@ func (r *noopMeter) Set(opts ...options.Option) Meter { | |||||||
| 	return m | 	return m | ||||||
| } | } | ||||||
|  |  | ||||||
| func (r *noopMeter) Write(_ io.Writer, _ ...options.Option) error { | func (r *noopMeter) Write(_ io.Writer, _ ...Option) error { | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
|   | |||||||
| @@ -2,17 +2,13 @@ package meter | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"reflect" |  | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/logger" |  | ||||||
| 	"go.unistack.org/micro/v4/options" |  | ||||||
| 	rutil "go.unistack.org/micro/v4/util/reflect" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | // Option powers the configuration for metrics implementations: | ||||||
|  | type Option func(*Options) | ||||||
|  |  | ||||||
| // Options for metrics implementations | // Options for metrics implementations | ||||||
| type Options struct { | type Options struct { | ||||||
| 	// Logger used for logging |  | ||||||
| 	Logger logger.Logger |  | ||||||
| 	// Context holds external options | 	// Context holds external options | ||||||
| 	Context context.Context | 	Context context.Context | ||||||
| 	// Name holds the meter name | 	// Name holds the meter name | ||||||
| @@ -21,10 +17,6 @@ type Options struct { | |||||||
| 	Address string | 	Address string | ||||||
| 	// Path holds the path for metrics | 	// Path holds the path for metrics | ||||||
| 	Path string | 	Path string | ||||||
| 	// MetricPrefix holds the prefix for all metrics |  | ||||||
| 	MetricPrefix string |  | ||||||
| 	// LabelPrefix holds the prefix for all labels |  | ||||||
| 	LabelPrefix string |  | ||||||
| 	// Labels holds the default labels | 	// Labels holds the default labels | ||||||
| 	Labels []string | 	Labels []string | ||||||
| 	// WriteProcessMetrics flag to write process metrics | 	// WriteProcessMetrics flag to write process metrics | ||||||
| @@ -34,14 +26,11 @@ type Options struct { | |||||||
| } | } | ||||||
|  |  | ||||||
| // NewOptions prepares a set of options: | // NewOptions prepares a set of options: | ||||||
| func NewOptions(opt ...options.Option) Options { | func NewOptions(opt ...Option) Options { | ||||||
| 	opts := Options{ | 	opts := Options{ | ||||||
| 		Address: DefaultAddress, | 		Address: DefaultAddress, | ||||||
| 		Path:    DefaultPath, | 		Path:    DefaultPath, | ||||||
| 		Context: context.Background(), | 		Context: context.Background(), | ||||||
| 		Logger:       logger.DefaultLogger, |  | ||||||
| 		MetricPrefix: DefaultMetricPrefix, |  | ||||||
| 		LabelPrefix:  DefaultLabelPrefix, |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	for _, o := range opt { | 	for _, o := range opt { | ||||||
| @@ -51,24 +40,24 @@ func NewOptions(opt ...options.Option) Options { | |||||||
| 	return opts | 	return opts | ||||||
| } | } | ||||||
|  |  | ||||||
| // LabelPrefix sets the labels prefix | // Context sets the metrics context | ||||||
| func LabelPrefix(pref string) options.Option { | func Context(ctx context.Context) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, pref, ".LabelPrefix") | 		o.Context = ctx | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // MetricPrefix sets the metric prefix |  | ||||||
| func MetricPrefix(pref string) options.Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return options.Set(src, pref, ".MetricPrefix") |  | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // Path used to serve metrics over HTTP | // Path used to serve metrics over HTTP | ||||||
| func Path(path string) options.Option { | func Path(value string) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, path, ".Path") | 		o.Path = value | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Address is the listen address to serve metrics | ||||||
|  | func Address(value string) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Address = value | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -81,34 +70,30 @@ func TimingObjectives(value map[float64]float64) Option { | |||||||
| } | } | ||||||
| */ | */ | ||||||
|  |  | ||||||
| // Labels sets the meter labels | // Labels add the meter labels | ||||||
| func Labels(ls ...string) options.Option { | func Labels(ls ...string) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		v, err := options.Get(src, ".Labels") | 		o.Labels = append(o.Labels, ls...) | ||||||
| 		if err != nil { |  | ||||||
| 			return err |  | ||||||
| 		} else if rutil.IsZero(v) { |  | ||||||
| 			v = reflect.MakeSlice(reflect.TypeOf(v), 0, len(ls)).Interface() |  | ||||||
| 	} | 	} | ||||||
| 		cv := reflect.ValueOf(v) |  | ||||||
| 		for _, l := range ls { |  | ||||||
| 			reflect.Append(cv, reflect.ValueOf(l)) |  | ||||||
| } | } | ||||||
| 		err = options.Set(src, cv, ".Labels") |  | ||||||
| 		return err | // Name sets the name | ||||||
|  | func Name(n string) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Name = n | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // WriteProcessMetrics enable process metrics output for write | // WriteProcessMetrics enable process metrics output for write | ||||||
| func WriteProcessMetrics(b bool) options.Option { | func WriteProcessMetrics(b bool) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, b, ".WriteProcessMetrics") | 		o.WriteProcessMetrics = b | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // WriteFDMetrics enable fd metrics output for write | // WriteFDMetrics enable fd metrics output for write | ||||||
| func WriteFDMetrics(b bool) options.Option { | func WriteFDMetrics(b bool) Option { | ||||||
| 	return func(src interface{}) error { | 	return func(o *Options) { | ||||||
| 		return options.Set(src, b, ".WriteFDMetrics") | 		o.WriteFDMetrics = b | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,238 +0,0 @@ | |||||||
| package wrapper // import "go.unistack.org/micro/v4/meter/wrapper" |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"context" |  | ||||||
| 	"fmt" |  | ||||||
| 	"time" |  | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/client" |  | ||||||
| 	"go.unistack.org/micro/v4/meter" |  | ||||||
| 	"go.unistack.org/micro/v4/semconv" |  | ||||||
| 	"go.unistack.org/micro/v4/server" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| var ( |  | ||||||
| 	labelSuccess  = "success" |  | ||||||
| 	labelFailure  = "failure" |  | ||||||
| 	labelStatus   = "status" |  | ||||||
| 	labelEndpoint = "endpoint" |  | ||||||
|  |  | ||||||
| 	// DefaultSkipEndpoints contains list of endpoints that not evaluted by wrapper |  | ||||||
| 	DefaultSkipEndpoints = []string{"Meter.Metrics", "Health.Live", "Health.Ready", "Health.Version"} |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // Options struct |  | ||||||
| type Options struct { |  | ||||||
| 	Meter         meter.Meter |  | ||||||
| 	lopts         []meter.Option |  | ||||||
| 	SkipEndpoints []string |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Option func signature |  | ||||||
| type Option func(*Options) |  | ||||||
|  |  | ||||||
| // NewOptions creates new Options struct |  | ||||||
| func NewOptions(opts ...Option) Options { |  | ||||||
| 	options := Options{ |  | ||||||
| 		Meter:         meter.DefaultMeter, |  | ||||||
| 		lopts:         make([]meter.Option, 0, 5), |  | ||||||
| 		SkipEndpoints: DefaultSkipEndpoints, |  | ||||||
| 	} |  | ||||||
| 	for _, o := range opts { |  | ||||||
| 		o(&options) |  | ||||||
| 	} |  | ||||||
| 	return options |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // ServiceName passes service name to meter label |  | ||||||
| func ServiceName(name string) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.lopts = append(o.lopts, meter.Labels("name", name)) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // ServiceVersion passes service version to meter label |  | ||||||
| func ServiceVersion(version string) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.lopts = append(o.lopts, meter.Labels("version", version)) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // ServiceID passes service id to meter label |  | ||||||
| func ServiceID(id string) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.lopts = append(o.lopts, meter.Labels("id", id)) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Meter passes meter |  | ||||||
| func Meter(m meter.Meter) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.Meter = m |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // SkipEndoints add endpoint to skip |  | ||||||
| func SkipEndoints(eps ...string) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.SkipEndpoints = append(o.SkipEndpoints, eps...) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type wrapper struct { |  | ||||||
| 	client.Client |  | ||||||
| 	callFunc client.CallFunc |  | ||||||
| 	opts     Options |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewClientWrapper create new client wrapper |  | ||||||
| func NewClientWrapper(opts ...Option) client.Wrapper { |  | ||||||
| 	return func(c client.Client) client.Client { |  | ||||||
| 		handler := &wrapper{ |  | ||||||
| 			opts:   NewOptions(opts...), |  | ||||||
| 			Client: c, |  | ||||||
| 		} |  | ||||||
| 		return handler |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewCallWrapper create new call wrapper |  | ||||||
| func NewCallWrapper(opts ...Option) client.CallWrapper { |  | ||||||
| 	return func(fn client.CallFunc) client.CallFunc { |  | ||||||
| 		handler := &wrapper{ |  | ||||||
| 			opts:     NewOptions(opts...), |  | ||||||
| 			callFunc: fn, |  | ||||||
| 		} |  | ||||||
| 		return handler.CallFunc |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *wrapper) CallFunc(ctx context.Context, addr string, req client.Request, rsp interface{}, opts client.CallOptions) error { |  | ||||||
| 	endpoint := fmt.Sprintf("%s.%s", req.Service(), req.Endpoint()) |  | ||||||
| 	for _, ep := range w.opts.SkipEndpoints { |  | ||||||
| 		if ep == endpoint { |  | ||||||
| 			return w.callFunc(ctx, addr, req, rsp, opts) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	labels := make([]string, 0, 4) |  | ||||||
| 	labels = append(labels, labelEndpoint, endpoint) |  | ||||||
|  |  | ||||||
| 	w.opts.Meter.Counter(semconv.ClientRequestInflight, labels...).Inc() |  | ||||||
| 	ts := time.Now() |  | ||||||
| 	err := w.callFunc(ctx, addr, req, rsp, opts) |  | ||||||
| 	te := time.Since(ts) |  | ||||||
| 	w.opts.Meter.Counter(semconv.ClientRequestInflight, labels...).Dec() |  | ||||||
|  |  | ||||||
| 	w.opts.Meter.Summary(semconv.ClientRequestLatencyMicroseconds, labels...).Update(te.Seconds()) |  | ||||||
| 	w.opts.Meter.Histogram(semconv.ClientRequestDurationSeconds, labels...).Update(te.Seconds()) |  | ||||||
|  |  | ||||||
| 	if err == nil { |  | ||||||
| 		labels = append(labels, labelStatus, labelSuccess) |  | ||||||
| 	} else { |  | ||||||
| 		labels = append(labels, labelStatus, labelFailure) |  | ||||||
| 	} |  | ||||||
| 	w.opts.Meter.Counter(semconv.ClientRequestTotal, labels...).Inc() |  | ||||||
|  |  | ||||||
| 	return err |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *wrapper) Call(ctx context.Context, req client.Request, rsp interface{}, opts ...client.CallOption) error { |  | ||||||
| 	endpoint := fmt.Sprintf("%s.%s", req.Service(), req.Endpoint()) |  | ||||||
| 	for _, ep := range w.opts.SkipEndpoints { |  | ||||||
| 		if ep == endpoint { |  | ||||||
| 			return w.Client.Call(ctx, req, rsp, opts...) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	labels := make([]string, 0, 4) |  | ||||||
| 	labels = append(labels, labelEndpoint, endpoint) |  | ||||||
|  |  | ||||||
| 	w.opts.Meter.Counter(semconv.ClientRequestInflight, labels...).Inc() |  | ||||||
| 	ts := time.Now() |  | ||||||
| 	err := w.Client.Call(ctx, req, rsp, opts...) |  | ||||||
| 	te := time.Since(ts) |  | ||||||
| 	w.opts.Meter.Counter(semconv.ClientRequestInflight, labels...).Dec() |  | ||||||
|  |  | ||||||
| 	w.opts.Meter.Summary(semconv.ClientRequestLatencyMicroseconds, labels...).Update(te.Seconds()) |  | ||||||
| 	w.opts.Meter.Histogram(semconv.ClientRequestDurationSeconds, labels...).Update(te.Seconds()) |  | ||||||
|  |  | ||||||
| 	if err == nil { |  | ||||||
| 		labels = append(labels, labelStatus, labelSuccess) |  | ||||||
| 	} else { |  | ||||||
| 		labels = append(labels, labelStatus, labelFailure) |  | ||||||
| 	} |  | ||||||
| 	w.opts.Meter.Counter(semconv.ClientRequestTotal, labels...).Inc() |  | ||||||
|  |  | ||||||
| 	return err |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *wrapper) Stream(ctx context.Context, req client.Request, opts ...client.CallOption) (client.Stream, error) { |  | ||||||
| 	endpoint := fmt.Sprintf("%s.%s", req.Service(), req.Endpoint()) |  | ||||||
| 	for _, ep := range w.opts.SkipEndpoints { |  | ||||||
| 		if ep == endpoint { |  | ||||||
| 			return w.Client.Stream(ctx, req, opts...) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	labels := make([]string, 0, 4) |  | ||||||
| 	labels = append(labels, labelEndpoint, endpoint) |  | ||||||
|  |  | ||||||
| 	w.opts.Meter.Counter(semconv.ClientRequestInflight, labels...).Inc() |  | ||||||
| 	ts := time.Now() |  | ||||||
| 	stream, err := w.Client.Stream(ctx, req, opts...) |  | ||||||
| 	te := time.Since(ts) |  | ||||||
| 	w.opts.Meter.Counter(semconv.ClientRequestInflight, labels...).Dec() |  | ||||||
|  |  | ||||||
| 	w.opts.Meter.Summary(semconv.ClientRequestLatencyMicroseconds, labels...).Update(te.Seconds()) |  | ||||||
| 	w.opts.Meter.Histogram(semconv.ClientRequestDurationSeconds, labels...).Update(te.Seconds()) |  | ||||||
|  |  | ||||||
| 	if err == nil { |  | ||||||
| 		labels = append(labels, labelStatus, labelSuccess) |  | ||||||
| 	} else { |  | ||||||
| 		labels = append(labels, labelStatus, labelFailure) |  | ||||||
| 	} |  | ||||||
| 	w.opts.Meter.Counter(semconv.ClientRequestTotal, labels...).Inc() |  | ||||||
|  |  | ||||||
| 	return stream, err |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewServerHandlerWrapper create new server handler wrapper |  | ||||||
| func NewServerHandlerWrapper(opts ...Option) server.HandlerWrapper { |  | ||||||
| 	handler := &wrapper{ |  | ||||||
| 		opts: NewOptions(opts...), |  | ||||||
| 	} |  | ||||||
| 	return handler.HandlerFunc |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *wrapper) HandlerFunc(fn server.HandlerFunc) server.HandlerFunc { |  | ||||||
| 	return func(ctx context.Context, req server.Request, rsp interface{}) error { |  | ||||||
| 		endpoint := req.Service() + "." + req.Endpoint() |  | ||||||
| 		for _, ep := range w.opts.SkipEndpoints { |  | ||||||
| 			if ep == endpoint { |  | ||||||
| 				return fn(ctx, req, rsp) |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		labels := make([]string, 0, 4) |  | ||||||
| 		labels = append(labels, labelEndpoint, endpoint) |  | ||||||
|  |  | ||||||
| 		w.opts.Meter.Counter(semconv.ServerRequestInflight, labels...).Inc() |  | ||||||
| 		ts := time.Now() |  | ||||||
| 		err := fn(ctx, req, rsp) |  | ||||||
| 		te := time.Since(ts) |  | ||||||
| 		w.opts.Meter.Counter(semconv.ServerRequestInflight, labels...).Dec() |  | ||||||
|  |  | ||||||
| 		w.opts.Meter.Summary(semconv.ServerRequestLatencyMicroseconds, labels...).Update(te.Seconds()) |  | ||||||
| 		w.opts.Meter.Histogram(semconv.ServerRequestDurationSeconds, labels...).Update(te.Seconds()) |  | ||||||
|  |  | ||||||
| 		if err == nil { |  | ||||||
| 			labels = append(labels, labelStatus, labelSuccess) |  | ||||||
| 		} else { |  | ||||||
| 			labels = append(labels, labelStatus, labelFailure) |  | ||||||
| 		} |  | ||||||
| 		w.opts.Meter.Counter(semconv.ServerRequestTotal, labels...).Inc() |  | ||||||
|  |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
							
								
								
									
										98
									
								
								micro.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										98
									
								
								micro.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,98 @@ | |||||||
|  | package micro | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"reflect" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v3/broker" | ||||||
|  | 	"go.unistack.org/micro/v3/client" | ||||||
|  | 	"go.unistack.org/micro/v3/codec" | ||||||
|  | 	"go.unistack.org/micro/v3/flow" | ||||||
|  | 	"go.unistack.org/micro/v3/fsm" | ||||||
|  | 	"go.unistack.org/micro/v3/logger" | ||||||
|  | 	"go.unistack.org/micro/v3/meter" | ||||||
|  | 	"go.unistack.org/micro/v3/register" | ||||||
|  | 	"go.unistack.org/micro/v3/resolver" | ||||||
|  | 	"go.unistack.org/micro/v3/router" | ||||||
|  | 	"go.unistack.org/micro/v3/selector" | ||||||
|  | 	"go.unistack.org/micro/v3/server" | ||||||
|  | 	"go.unistack.org/micro/v3/store" | ||||||
|  | 	"go.unistack.org/micro/v3/sync" | ||||||
|  | 	"go.unistack.org/micro/v3/tracer" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func As(b any, target any) bool { | ||||||
|  | 	if b == nil { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  | 	if target == nil { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  | 	val := reflect.ValueOf(target) | ||||||
|  | 	typ := val.Type() | ||||||
|  | 	if typ.Kind() != reflect.Ptr || val.IsNil() { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  | 	targetType := typ.Elem() | ||||||
|  | 	if targetType.Kind() != reflect.Interface { | ||||||
|  | 		switch { | ||||||
|  | 		case targetType.Implements(brokerType): | ||||||
|  | 			break | ||||||
|  | 		case targetType.Implements(loggerType): | ||||||
|  | 			break | ||||||
|  | 		case targetType.Implements(clientType): | ||||||
|  | 			break | ||||||
|  | 		case targetType.Implements(serverType): | ||||||
|  | 			break | ||||||
|  | 		case targetType.Implements(codecType): | ||||||
|  | 			break | ||||||
|  | 		case targetType.Implements(flowType): | ||||||
|  | 			break | ||||||
|  | 		case targetType.Implements(fsmType): | ||||||
|  | 			break | ||||||
|  | 		case targetType.Implements(meterType): | ||||||
|  | 			break | ||||||
|  | 		case targetType.Implements(registerType): | ||||||
|  | 			break | ||||||
|  | 		case targetType.Implements(resolverType): | ||||||
|  | 			break | ||||||
|  | 		case targetType.Implements(selectorType): | ||||||
|  | 			break | ||||||
|  | 		case targetType.Implements(storeType): | ||||||
|  | 			break | ||||||
|  | 		case targetType.Implements(syncType): | ||||||
|  | 			break | ||||||
|  | 		case targetType.Implements(serviceType): | ||||||
|  | 			break | ||||||
|  | 		case targetType.Implements(routerType): | ||||||
|  | 			break | ||||||
|  | 		case targetType.Implements(tracerType): | ||||||
|  | 			break | ||||||
|  | 		default: | ||||||
|  | 			return false | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	if reflect.TypeOf(b).AssignableTo(targetType) { | ||||||
|  | 		val.Elem().Set(reflect.ValueOf(b)) | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|  |  | ||||||
|  | var ( | ||||||
|  | 	brokerType   = reflect.TypeOf((*broker.Broker)(nil)).Elem() | ||||||
|  | 	loggerType   = reflect.TypeOf((*logger.Logger)(nil)).Elem() | ||||||
|  | 	clientType   = reflect.TypeOf((*client.Client)(nil)).Elem() | ||||||
|  | 	serverType   = reflect.TypeOf((*server.Server)(nil)).Elem() | ||||||
|  | 	codecType    = reflect.TypeOf((*codec.Codec)(nil)).Elem() | ||||||
|  | 	flowType     = reflect.TypeOf((*flow.Flow)(nil)).Elem() | ||||||
|  | 	fsmType      = reflect.TypeOf((*fsm.FSM)(nil)).Elem() | ||||||
|  | 	meterType    = reflect.TypeOf((*meter.Meter)(nil)).Elem() | ||||||
|  | 	registerType = reflect.TypeOf((*register.Register)(nil)).Elem() | ||||||
|  | 	resolverType = reflect.TypeOf((*resolver.Resolver)(nil)).Elem() | ||||||
|  | 	routerType   = reflect.TypeOf((*router.Router)(nil)).Elem() | ||||||
|  | 	selectorType = reflect.TypeOf((*selector.Selector)(nil)).Elem() | ||||||
|  | 	storeType    = reflect.TypeOf((*store.Store)(nil)).Elem() | ||||||
|  | 	syncType     = reflect.TypeOf((*sync.Sync)(nil)).Elem() | ||||||
|  | 	tracerType   = reflect.TypeOf((*tracer.Tracer)(nil)).Elem() | ||||||
|  | 	serviceType  = reflect.TypeOf((*Service)(nil)).Elem() | ||||||
|  | ) | ||||||
							
								
								
									
										122
									
								
								micro_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										122
									
								
								micro_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,122 @@ | |||||||
|  | package micro | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"fmt" | ||||||
|  | 	"reflect" | ||||||
|  | 	"testing" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v3/broker" | ||||||
|  | 	"go.unistack.org/micro/v3/fsm" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func TestAs(t *testing.T) { | ||||||
|  | 	var b *bro | ||||||
|  | 	broTarget := &bro{name: "kafka"} | ||||||
|  | 	fsmTarget := &fsmT{name: "fsm"} | ||||||
|  |  | ||||||
|  | 	testCases := []struct { | ||||||
|  | 		b      any | ||||||
|  | 		target any | ||||||
|  | 		want   any | ||||||
|  |  | ||||||
|  | 		match bool | ||||||
|  | 	}{ | ||||||
|  | 		{ | ||||||
|  | 			b:      broTarget, | ||||||
|  | 			target: &b, | ||||||
|  | 			match:  true, | ||||||
|  | 			want:   broTarget, | ||||||
|  | 		}, | ||||||
|  | 		{ | ||||||
|  | 			b:      nil, | ||||||
|  | 			target: &b, | ||||||
|  | 			match:  false, | ||||||
|  | 			want:   nil, | ||||||
|  | 		}, | ||||||
|  | 		{ | ||||||
|  | 			b:      fsmTarget, | ||||||
|  | 			target: &b, | ||||||
|  | 			match:  false, | ||||||
|  | 			want:   nil, | ||||||
|  | 		}, | ||||||
|  | 	} | ||||||
|  | 	for i, tc := range testCases { | ||||||
|  | 		name := fmt.Sprintf("%d:As(Errorf(..., %v), %v)", i, tc.b, tc.target) | ||||||
|  | 		// Clear the target pointer, in case it was set in a previous test. | ||||||
|  | 		rtarget := reflect.ValueOf(tc.target) | ||||||
|  | 		rtarget.Elem().Set(reflect.Zero(reflect.TypeOf(tc.target).Elem())) | ||||||
|  | 		t.Run(name, func(t *testing.T) { | ||||||
|  | 			match := As(tc.b, tc.target) | ||||||
|  | 			if match != tc.match { | ||||||
|  | 				t.Fatalf("match: got %v; want %v", match, tc.match) | ||||||
|  | 			} | ||||||
|  | 			if !match { | ||||||
|  | 				return | ||||||
|  | 			} | ||||||
|  | 			if got := rtarget.Elem().Interface(); got != tc.want { | ||||||
|  | 				t.Fatalf("got %#v, want %#v", got, tc.want) | ||||||
|  | 			} | ||||||
|  | 		}) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type bro struct { | ||||||
|  | 	name string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (p *bro) Name() string { return p.name } | ||||||
|  |  | ||||||
|  | func (p *bro) Live() bool { return true } | ||||||
|  |  | ||||||
|  | func (p *bro) Ready() bool { return true } | ||||||
|  |  | ||||||
|  | func (p *bro) Health() bool { return true } | ||||||
|  |  | ||||||
|  | func (p *bro) Init(_ ...broker.Option) error { return nil } | ||||||
|  |  | ||||||
|  | // Options returns broker options | ||||||
|  | func (p *bro) Options() broker.Options { return broker.Options{} } | ||||||
|  |  | ||||||
|  | // Address return configured address | ||||||
|  | func (p *bro) Address() string { return "" } | ||||||
|  |  | ||||||
|  | // Connect connects to broker | ||||||
|  | func (p *bro) Connect(_ context.Context) error { return nil } | ||||||
|  |  | ||||||
|  | // Disconnect disconnect from broker | ||||||
|  | func (p *bro) Disconnect(_ context.Context) error { return nil } | ||||||
|  |  | ||||||
|  | // Publish message, msg can be single broker.Message or []broker.Message | ||||||
|  | func (p *bro) Publish(_ context.Context, _ string, _ *broker.Message, _ ...broker.PublishOption) error { | ||||||
|  | 	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 | ||||||
|  | func (p *bro) Subscribe(_ context.Context, _ string, _ broker.Handler, _ ...broker.SubscribeOption) (broker.Subscriber, error) { | ||||||
|  | 	return nil, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // String type of broker | ||||||
|  | func (p *bro) String() string { return p.name } | ||||||
|  |  | ||||||
|  | type fsmT struct { | ||||||
|  | 	name string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (f *fsmT) Start(_ context.Context, _ interface{}, _ ...Option) (interface{}, error) { | ||||||
|  | 	return nil, nil | ||||||
|  | } | ||||||
|  | func (f *fsmT) Current() string                 { return f.name } | ||||||
|  | func (f *fsmT) Reset()                          {} | ||||||
|  | func (f *fsmT) State(_ string, _ fsm.StateFunc) {} | ||||||
| @@ -1,4 +1,4 @@ | |||||||
| package mtls // import "go.unistack.org/micro/v4/mtls" | package mtls | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"bytes" | 	"bytes" | ||||||
|   | |||||||
| @@ -8,17 +8,18 @@ import ( | |||||||
|  |  | ||||||
| // CertificateOptions holds options for x509.CreateCertificate | // CertificateOptions holds options for x509.CreateCertificate | ||||||
| type CertificateOptions struct { | type CertificateOptions struct { | ||||||
| 	Organization          []string |  | ||||||
| 	OrganizationalUnit    []string |  | ||||||
| 	CommonName            string |  | ||||||
| 	OCSPServer            []string |  | ||||||
| 	IssuingCertificateURL []string |  | ||||||
| 	SerialNumber          *big.Int | 	SerialNumber          *big.Int | ||||||
| 	NotAfter              time.Time | 	NotAfter              time.Time | ||||||
| 	NotBefore             time.Time | 	NotBefore             time.Time | ||||||
|  | 	CommonName            string | ||||||
|  | 	Organization          []string | ||||||
|  | 	OrganizationalUnit    []string | ||||||
|  | 	OCSPServer            []string | ||||||
|  | 	IssuingCertificateURL []string | ||||||
|  | 	ExtKeyUsage           []x509.ExtKeyUsage | ||||||
|  |  | ||||||
| 	SignatureAlgorithm x509.SignatureAlgorithm | 	SignatureAlgorithm x509.SignatureAlgorithm | ||||||
| 	PublicKeyAlgorithm x509.PublicKeyAlgorithm | 	PublicKeyAlgorithm x509.PublicKeyAlgorithm | ||||||
| 	ExtKeyUsage           []x509.ExtKeyUsage |  | ||||||
| 	KeyUsage           x509.KeyUsage | 	KeyUsage           x509.KeyUsage | ||||||
| 	IsCA               bool | 	IsCA               bool | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										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 | ||||||
|  | } | ||||||
							
								
								
									
										89
									
								
								options.go
									
									
									
									
									
								
							
							
						
						
									
										89
									
								
								options.go
									
									
									
									
									
								
							| @@ -5,18 +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/options" | 	"go.unistack.org/micro/v3/register" | ||||||
| 	"go.unistack.org/micro/v4/register" | 	"go.unistack.org/micro/v3/router" | ||||||
| 	"go.unistack.org/micro/v4/router" | 	"go.unistack.org/micro/v3/server" | ||||||
| 	"go.unistack.org/micro/v4/server" | 	"go.unistack.org/micro/v3/store" | ||||||
| 	"go.unistack.org/micro/v4/store" | 	"go.unistack.org/micro/v3/tracer" | ||||||
| 	"go.unistack.org/micro/v4/tracer" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Options for micro service | // Options for micro service | ||||||
| @@ -91,7 +90,33 @@ type Option func(*Options) error | |||||||
| // Broker to be used for client and server | // Broker to be used for client and server | ||||||
| func Broker(b broker.Broker, opts ...BrokerOption) Option { | func Broker(b broker.Broker, opts ...BrokerOption) Option { | ||||||
| 	return func(o *Options) error { | 	return func(o *Options) error { | ||||||
| 		o.Brokers = []broker.Broker{b} | 		var err error | ||||||
|  | 		bopts := brokerOptions{} | ||||||
|  | 		for _, opt := range opts { | ||||||
|  | 			opt(&bopts) | ||||||
|  | 		} | ||||||
|  | 		all := false | ||||||
|  | 		if len(opts) == 0 { | ||||||
|  | 			all = true | ||||||
|  | 		} | ||||||
|  | 		for _, srv := range o.Servers { | ||||||
|  | 			for _, os := range bopts.servers { | ||||||
|  | 				if srv.Name() == os || all { | ||||||
|  | 					if err = srv.Init(server.Broker(b)); err != nil { | ||||||
|  | 						return err | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		for _, cli := range o.Clients { | ||||||
|  | 			for _, oc := range bopts.clients { | ||||||
|  | 				if cli.Name() == oc || all { | ||||||
|  | 					if err = cli.Init(client.Broker(b)); err != nil { | ||||||
|  | 						return err | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| @@ -202,7 +227,7 @@ func Logger(l logger.Logger, opts ...LoggerOption) Option { | |||||||
| 		for _, srv := range o.Servers { | 		for _, srv := range o.Servers { | ||||||
| 			for _, os := range lopts.servers { | 			for _, os := range lopts.servers { | ||||||
| 				if srv.Name() == os || all { | 				if srv.Name() == os || all { | ||||||
| 					if err = srv.Init(options.Logger(l)); err != nil { | 					if err = srv.Init(server.Logger(l)); err != nil { | ||||||
| 						return err | 						return err | ||||||
| 					} | 					} | ||||||
| 				} | 				} | ||||||
| @@ -211,7 +236,7 @@ func Logger(l logger.Logger, opts ...LoggerOption) Option { | |||||||
| 		for _, cli := range o.Clients { | 		for _, cli := range o.Clients { | ||||||
| 			for _, oc := range lopts.clients { | 			for _, oc := range lopts.clients { | ||||||
| 				if cli.Name() == oc || all { | 				if cli.Name() == oc || all { | ||||||
| 					if err = cli.Init(options.Logger(l)); err != nil { | 					if err = cli.Init(client.Logger(l)); err != nil { | ||||||
| 						return err | 						return err | ||||||
| 					} | 					} | ||||||
| 				} | 				} | ||||||
| @@ -220,7 +245,7 @@ func Logger(l logger.Logger, opts ...LoggerOption) Option { | |||||||
| 		for _, brk := range o.Brokers { | 		for _, brk := range o.Brokers { | ||||||
| 			for _, ob := range lopts.brokers { | 			for _, ob := range lopts.brokers { | ||||||
| 				if brk.Name() == ob || all { | 				if brk.Name() == ob || all { | ||||||
| 					if err = brk.Init(options.Logger(l)); err != nil { | 					if err = brk.Init(broker.Logger(l)); err != nil { | ||||||
| 						return err | 						return err | ||||||
| 					} | 					} | ||||||
| 				} | 				} | ||||||
| @@ -238,25 +263,17 @@ func Logger(l logger.Logger, opts ...LoggerOption) Option { | |||||||
| 		for _, str := range o.Stores { | 		for _, str := range o.Stores { | ||||||
| 			for _, or := range lopts.stores { | 			for _, or := range lopts.stores { | ||||||
| 				if str.Name() == or || all { | 				if str.Name() == or || all { | ||||||
| 					if err = str.Init(options.Logger(l)); err != nil { | 					if err = str.Init(store.Logger(l)); err != nil { | ||||||
| 						return err |  | ||||||
| 					} |  | ||||||
| 				} |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 		for _, mtr := range o.Meters { |  | ||||||
| 			for _, or := range lopts.meters { |  | ||||||
| 				if mtr.Name() == or || all { |  | ||||||
| 					if err = mtr.Init(options.Logger(l)); err != nil { |  | ||||||
| 						return err | 						return err | ||||||
| 					} | 					} | ||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		for _, trc := range o.Tracers { | 		for _, trc := range o.Tracers { | ||||||
| 			for _, ot := range lopts.tracers { | 			for _, ot := range lopts.tracers { | ||||||
| 				if trc.Name() == ot || all { | 				if trc.Name() == ot || all { | ||||||
| 					if err = trc.Init(options.Logger(l)); err != nil { | 					if err = trc.Init(tracer.Logger(l)); err != nil { | ||||||
| 						return err | 						return err | ||||||
| 					} | 					} | ||||||
| 				} | 				} | ||||||
| @@ -277,7 +294,7 @@ type loggerOptions struct { | |||||||
| 	brokers   []string | 	brokers   []string | ||||||
| 	registers []string | 	registers []string | ||||||
| 	stores    []string | 	stores    []string | ||||||
| 	meters    []string | 	// meters    []string | ||||||
| 	tracers []string | 	tracers []string | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -330,7 +347,7 @@ func Register(r register.Register, opts ...RegisterOption) Option { | |||||||
| 		for _, srv := range o.Servers { | 		for _, srv := range o.Servers { | ||||||
| 			for _, os := range ropts.servers { | 			for _, os := range ropts.servers { | ||||||
| 				if srv.Name() == os || all { | 				if srv.Name() == os || all { | ||||||
| 					if err = srv.Init(options.Register(r)); err != nil { | 					if err = srv.Init(server.Register(r)); err != nil { | ||||||
| 						return err | 						return err | ||||||
| 					} | 					} | ||||||
| 				} | 				} | ||||||
| @@ -339,7 +356,7 @@ func Register(r register.Register, opts ...RegisterOption) Option { | |||||||
| 		for _, brk := range o.Brokers { | 		for _, brk := range o.Brokers { | ||||||
| 			for _, os := range ropts.brokers { | 			for _, os := range ropts.brokers { | ||||||
| 				if brk.Name() == os || all { | 				if brk.Name() == os || all { | ||||||
| 					if err = brk.Init(options.Register(r)); err != nil { | 					if err = brk.Init(broker.Register(r)); err != nil { | ||||||
| 						return err | 						return err | ||||||
| 					} | 					} | ||||||
| 				} | 				} | ||||||
| @@ -396,7 +413,7 @@ func Tracer(t tracer.Tracer, opts ...TracerOption) Option { | |||||||
| 		for _, srv := range o.Servers { | 		for _, srv := range o.Servers { | ||||||
| 			for _, os := range topts.servers { | 			for _, os := range topts.servers { | ||||||
| 				if srv.Name() == os || all { | 				if srv.Name() == os || all { | ||||||
| 					if err = srv.Init(options.Tracer(t)); err != nil { | 					if err = srv.Init(server.Tracer(t)); err != nil { | ||||||
| 						return err | 						return err | ||||||
| 					} | 					} | ||||||
| 				} | 				} | ||||||
| @@ -405,7 +422,7 @@ func Tracer(t tracer.Tracer, opts ...TracerOption) Option { | |||||||
| 		for _, cli := range o.Clients { | 		for _, cli := range o.Clients { | ||||||
| 			for _, os := range topts.clients { | 			for _, os := range topts.clients { | ||||||
| 				if cli.Name() == os || all { | 				if cli.Name() == os || all { | ||||||
| 					if err = cli.Init(options.Tracer(t)); err != nil { | 					if err = cli.Init(client.Tracer(t)); err != nil { | ||||||
| 						return err | 						return err | ||||||
| 					} | 					} | ||||||
| 				} | 				} | ||||||
| @@ -414,7 +431,7 @@ func Tracer(t tracer.Tracer, opts ...TracerOption) Option { | |||||||
| 		for _, str := range o.Stores { | 		for _, str := range o.Stores { | ||||||
| 			for _, os := range topts.stores { | 			for _, os := range topts.stores { | ||||||
| 				if str.Name() == os || all { | 				if str.Name() == os || all { | ||||||
| 					if err = str.Init(options.Tracer(t)); err != nil { | 					if err = str.Init(store.Tracer(t)); err != nil { | ||||||
| 						return err | 						return err | ||||||
| 					} | 					} | ||||||
| 				} | 				} | ||||||
| @@ -423,7 +440,7 @@ func Tracer(t tracer.Tracer, opts ...TracerOption) Option { | |||||||
| 		for _, brk := range o.Brokers { | 		for _, brk := range o.Brokers { | ||||||
| 			for _, os := range topts.brokers { | 			for _, os := range topts.brokers { | ||||||
| 				if brk.Name() == os || all { | 				if brk.Name() == os || all { | ||||||
| 					if err = brk.Init(options.Tracer(t)); err != nil { | 					if err = brk.Init(broker.Tracer(t)); err != nil { | ||||||
| 						return err | 						return err | ||||||
| 					} | 					} | ||||||
| 				} | 				} | ||||||
| @@ -522,7 +539,7 @@ func Router(r router.Router, opts ...RouterOption) Option { | |||||||
| 		for _, cli := range o.Clients { | 		for _, cli := range o.Clients { | ||||||
| 			for _, os := range ropts.clients { | 			for _, os := range ropts.clients { | ||||||
| 				if cli.Name() == os || all { | 				if cli.Name() == os || all { | ||||||
| 					if err = cli.Init(options.Router(r)); err != nil { | 					if err = cli.Init(client.Router(r)); err != nil { | ||||||
| 						return err | 						return err | ||||||
| 					} | 					} | ||||||
| 				} | 				} | ||||||
| @@ -557,7 +574,7 @@ func Address(addr string) Option { | |||||||
| 		default: | 		default: | ||||||
| 			return fmt.Errorf("cant set same address for multiple servers") | 			return fmt.Errorf("cant set same address for multiple servers") | ||||||
| 		} | 		} | ||||||
| 		return o.Servers[0].Init(options.Address(addr)) | 		return o.Servers[0].Init(server.Address(addr)) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|   | |||||||
| @@ -1,4 +1,4 @@ | |||||||
| package options // import "go.unistack.org/micro/v4/options" | package options | ||||||
|  |  | ||||||
| // Hook func interface | // Hook func interface | ||||||
| type Hook interface{} | type Hook interface{} | ||||||
|   | |||||||
| @@ -1,189 +0,0 @@ | |||||||
| package options |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"context" |  | ||||||
| 	"crypto/tls" |  | ||||||
| 	"reflect" |  | ||||||
| 	"time" |  | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v4/metadata" |  | ||||||
| 	rutil "go.unistack.org/micro/v4/util/reflect" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // Option func signature |  | ||||||
| type Option func(interface{}) error |  | ||||||
|  |  | ||||||
| // Set assign value to struct by its path |  | ||||||
| func Set(src interface{}, dst interface{}, path string) error { |  | ||||||
| 	return rutil.SetFieldByPath(src, dst, path) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Get returns value from struct by its path |  | ||||||
| func Get(src interface{}, path string) (interface{}, error) { |  | ||||||
| 	return rutil.StructFieldByPath(src, path) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Name set Name value |  | ||||||
| func Name(v ...string) Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return Set(src, v, ".Name") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Address set Address value to single string or slice of strings |  | ||||||
| func Address(v ...string) Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return Set(src, v, ".Address") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Broker set Broker value |  | ||||||
| func Broker(v interface{}) Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return Set(src, v, ".Broker") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Logger set Logger value |  | ||||||
| func Logger(v interface{}) Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return Set(src, v, ".Logger") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Meter set Meter value |  | ||||||
| func Meter(v interface{}) Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return Set(src, v, ".Meter") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Tracer set Tracer value |  | ||||||
| func Tracer(v interface{}) Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return Set(src, v, ".Tracer") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Store set Store value |  | ||||||
| func Store(v interface{}) Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return Set(src, v, ".Store") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Register set Register value |  | ||||||
| func Register(v interface{}) Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return Set(src, v, ".Register") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Router set Router value |  | ||||||
| func Router(v interface{}) Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return Set(src, v, ".Router") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Codec set Codec value |  | ||||||
| func Codec(v interface{}) Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return Set(src, v, ".Codec") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Client set Client value |  | ||||||
| func Client(v interface{}) Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return Set(src, v, ".Client") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Codecs to be used to encode/decode requests for a given content type |  | ||||||
| func Codecs(ct string, v interface{}) Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		cm, err := Get(src, ".Codecs") |  | ||||||
| 		if err != nil { |  | ||||||
| 			return err |  | ||||||
| 		} else if rutil.IsZero(cm) { |  | ||||||
| 			cm = reflect.MakeMap(reflect.TypeOf(cm)).Interface() |  | ||||||
| 		} |  | ||||||
| 		cv := reflect.ValueOf(cm) |  | ||||||
| 		cv.SetMapIndex(reflect.ValueOf(ct), reflect.ValueOf(v)) |  | ||||||
| 		return Set(src, cv.Interface(), ".Codecs") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Context set Context value |  | ||||||
| func Context(v context.Context) Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return Set(src, v, ".Context") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // TLSConfig set TLSConfig value |  | ||||||
| func TLSConfig(v *tls.Config) Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return Set(src, v, ".TLSConfig") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func ContextOption(k, v interface{}) Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		ctx, err := Get(src, ".Context") |  | ||||||
| 		if err != nil { |  | ||||||
| 			return err |  | ||||||
| 		} |  | ||||||
| 		if ctx == nil { |  | ||||||
| 			ctx = context.Background() |  | ||||||
| 		} |  | ||||||
| 		err = Set(src, context.WithValue(ctx.(context.Context), k, v), ".Context") |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // ContentType pass ContentType for message data |  | ||||||
| func ContentType(ct string) Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return Set(src, ct, ".ContentType") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Metadata pass additional metadata |  | ||||||
| func Metadata(md metadata.Metadata) Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return Set(src, metadata.Copy(md), ".Metadata") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Namespace to use |  | ||||||
| func Namespace(ns string) Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return Set(src, ns, ".Namespace") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Labels sets the labels |  | ||||||
| func Labels(ls ...interface{}) Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		v, err := Get(src, ".Labels") |  | ||||||
| 		if err != nil { |  | ||||||
| 			return err |  | ||||||
| 		} else if rutil.IsZero(v) { |  | ||||||
| 			v = reflect.MakeSlice(reflect.TypeOf(v), 0, len(ls)).Interface() |  | ||||||
| 		} |  | ||||||
| 		cv := reflect.ValueOf(v) |  | ||||||
| 		for _, l := range ls { |  | ||||||
| 			cv = reflect.Append(cv, reflect.ValueOf(l)) |  | ||||||
| 		} |  | ||||||
| 		return Set(src, cv.Interface(), ".Labels") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Timeout pass timeout time.Duration |  | ||||||
| func Timeout(td time.Duration) Option { |  | ||||||
| 	return func(src interface{}) error { |  | ||||||
| 		return Set(src, td, ".Timeout") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user