Compare commits
	
		
			154 Commits
		
	
	
		
			v3.11.1
			...
			6c9dbc77dd
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 6c9dbc77dd | |||
| 70f0ace92e | |||
| 3f21bafc2f | |||
| a9ed8b16c1 | |||
|  | 740cd5931d | ||
| 85a78063d0 | |||
| 604ad9cd9d | |||
| 91137537a2 | |||
| 950e2352fd | |||
| 0bb29b29cf | |||
| 17bcd0b0ab | |||
| 20f9f4da3b | |||
| 66fa04b8dc | |||
| 1ef3ad6531 | |||
| c95a91349d | |||
| fdcf8e6ca4 | |||
| 8cb2d9db4a | |||
| 04da4388ac | |||
| 79fb23e644 | |||
| f8fe923ab1 | |||
| 105f56dbfe | |||
| 9fed5a368b | |||
| 7374d41cf8 | |||
| a4a8935c1f | |||
| 5f498c8232 | |||
| a00fdf679b | |||
| dc9ebe4155 | |||
| 87ced484b7 | |||
| af99b11a59 | |||
| 2724b51f7c | |||
| 5b5d0e02b9 | |||
| afc2de6819 | |||
| 32a8ab9c05 | |||
|  | 7e5401bded | ||
| 64b91cea06 | |||
|  | 0f59fdcbde | ||
| 50979e6708 | |||
| 46f3108870 | |||
|  | 5fed91a65f | ||
| 1c5bba908d | |||
|  | bc8ebdcad5 | ||
| fc24f3af92 | |||
| 1058177d1c | |||
|  | fa53fac085 | ||
| 8c060df5e3 | |||
| e1f8c62685 | |||
| 562b1ab9b7 | |||
|  | f3c877a37b | ||
| 0999b2ad78 | |||
| a365513177 | |||
|  | d1e3f3cab2 | ||
| ec94a09417 | |||
| 1728b88e06 | |||
| d3c31da9db | |||
| 59095681be | |||
| f11ceba225 | |||
| ffa01de78f | |||
| 816abc2bbc | |||
| f3f2a9b737 | |||
| 3f82cb3ba4 | |||
|  | 306b7a3962 | ||
| a8eda9d58d | |||
| 7e4477dcb4 | |||
|  | d846044fc6 | ||
| 29d956e74e | |||
| fcc4faff8a | |||
| 5df8f83f45 | |||
|  | 27fa6e9173 | ||
| bd55a35dc3 | |||
| 653bd386cc | |||
|  | 558c6f4d7c | ||
| d7dd6fbeb2 | |||
| a00cf2c8d9 | |||
|  | a3e8ab2492 | ||
| 06da500ef4 | |||
| 277f04ba19 | |||
|  | 470263ff5f | ||
| b8232e02be | |||
|  | f8c5e10c1d | ||
| 397e71f815 | |||
| 74e31d99f6 | |||
| f39de15d93 | |||
| d291102877 | |||
| 37ffbb18d8 | |||
| 9a85dead86 | |||
| a489aab1c3 | |||
| d160664ef1 | |||
| fa868edcaa | |||
|  | 6ed0b0e090 | ||
| 533b265d19 | |||
| 1ace2631a4 | |||
| 3dd5ca68d1 | |||
| 66ccd6021f | |||
| e5346f7e4f | |||
|  | daf19f031a | ||
| 5989fd54ca | |||
| ed30c26324 | |||
| 0f8f93d09a | |||
|  | f460e2f8dd | ||
| 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 | 
							
								
								
									
										54
									
								
								.gitea/workflows/job_coverage.yml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										54
									
								
								.gitea/workflows/job_coverage.yml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,54 @@ | |||||||
|  | name: coverage | ||||||
|  |  | ||||||
|  | on: | ||||||
|  |   push: | ||||||
|  |     branches: [ main, v3, v4 ] | ||||||
|  |     paths-ignore: | ||||||
|  |       - '.github/**' | ||||||
|  |       - '.gitea/**' | ||||||
|  |   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 ./... -covermode=count -coverprofile coverage.out -coverpkg ./... | ||||||
|  |         go tool cover -func coverage.out -o coverage.out | ||||||
|  |  | ||||||
|  |     - name: coverage badge | ||||||
|  |       uses: tj-actions/coverage-badge-go@v2 | ||||||
|  |       with: | ||||||
|  |         green: 80 | ||||||
|  |         filename: coverage.out | ||||||
|  |  | ||||||
|  |     - uses: stefanzweifel/git-auto-commit-action@v4 | ||||||
|  |       name: autocommit | ||||||
|  |       with: | ||||||
|  |         commit_message: Apply Code Coverage Badge | ||||||
|  |         skip_fetch: true | ||||||
|  |         skip_checkout: true | ||||||
|  |         file_pattern: ./README.md | ||||||
|  |  | ||||||
|  |     - name: push | ||||||
|  |       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 ] | ||||||
|  |     paths-ignore: | ||||||
|  |       - '.github/**' | ||||||
|  |       - '.gitea/**' | ||||||
|  |  | ||||||
|  | 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' | ||||||
							
								
								
									
										54
									
								
								.gitea/workflows/job_syncpull.yml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										54
									
								
								.gitea/workflows/job_syncpull.yml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,54 @@ | |||||||
|  | name: syncpull | ||||||
|  |  | ||||||
|  | on: | ||||||
|  |   schedule: | ||||||
|  |     - cron: '* * * * *' | ||||||
|  |   # Allows you to run this workflow manually from the Actions tab | ||||||
|  |   workflow_dispatch: | ||||||
|  |  | ||||||
|  | jobs: | ||||||
|  |   pull: | ||||||
|  |     runs-on: ubuntu-latest | ||||||
|  |     steps: | ||||||
|  |     - name: init | ||||||
|  |       run: | | ||||||
|  |         git config --global user.email "vtolstov <vtolstov@users.noreply.github.com>" | ||||||
|  |         git config --global user.name "github-actions[bot]" | ||||||
|  |         echo "machine git.unistack.org login vtolstov password ${{ secrets.TOKEN_GITEA }}" | tee -a /root/.netrc | ||||||
|  |         echo "machine github.com login vtolstov password ${{ secrets.TOKEN_GITHUB }}" | tee -a /root/.netrc | ||||||
|  |  | ||||||
|  |     - name: track master | ||||||
|  |       run: | | ||||||
|  |         git clone --depth=10 --branch master --single-branch ${GITHUB_SERVER_URL}/${GITHUB_REPOSITORY} repo | ||||||
|  |         cd repo | ||||||
|  |         git remote add --no-tags --fetch --track master upstream https://github.com/${GITHUB_REPOSITORY} | ||||||
|  |         git pull --rebase upstream master | ||||||
|  |         git push upstream master --progress | ||||||
|  |         git merge --allow-unrelated-histories "upstream/master" | ||||||
|  |         git push origin master --progress | ||||||
|  |         cd ../ | ||||||
|  |         rm -rf repo | ||||||
|  |  | ||||||
|  |     - name: track v3 | ||||||
|  |       run: | | ||||||
|  |         git clone --depth=10 --branch v3 --single-branch ${GITHUB_SERVER_URL}/${GITHUB_REPOSITORY} repo | ||||||
|  |         cd repo | ||||||
|  |         git remote add --no-tags --fetch --track v3 upstream https://github.com/${GITHUB_REPOSITORY} | ||||||
|  |         git pull --rebase upstream v3 | ||||||
|  |         git push upstream v3 | ||||||
|  |         git merge --allow-unrelated-histories "upstream/v3" | ||||||
|  |         git push origin v3 --progress | ||||||
|  |         cd ../ | ||||||
|  |         rm -rf repo | ||||||
|  |  | ||||||
|  |     - name: track v4 | ||||||
|  |       run: | | ||||||
|  |         git clone --depth=10 --branch v4 --single-branch ${GITHUB_SERVER_URL}/${GITHUB_REPOSITORY} repo | ||||||
|  |         cd repo | ||||||
|  |         git remote add --no-tags --fetch --track v4 upstream https://github.com/${GITHUB_REPOSITORY} | ||||||
|  |         git pull --rebase upstream v4 | ||||||
|  |         git push upstream v4 | ||||||
|  |         git merge --allow-unrelated-histories "upstream/v4" | ||||||
|  |         git push origin v4 --progress | ||||||
|  |         cd ../ | ||||||
|  |         rm -rf repo | ||||||
							
								
								
									
										31
									
								
								.gitea/workflows/job_test.yml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										31
									
								
								.gitea/workflows/job_test.yml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,31 @@ | |||||||
|  | name: test | ||||||
|  |  | ||||||
|  | on: | ||||||
|  |   pull_request: | ||||||
|  |     types: [opened, reopened, synchronize] | ||||||
|  |     branches: [ master, v3, v4 ] | ||||||
|  |   push: | ||||||
|  |     branches: [ master, v3, v4 ] | ||||||
|  |     paths-ignore: | ||||||
|  |       - '.github/**' | ||||||
|  |       - '.gitea/**' | ||||||
|  |  | ||||||
|  | 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 ./... | ||||||
							
								
								
									
										50
									
								
								.gitea/workflows/job_tests.yml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										50
									
								
								.gitea/workflows/job_tests.yml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,50 @@ | |||||||
|  | name: test | ||||||
|  |  | ||||||
|  | on: | ||||||
|  |   pull_request: | ||||||
|  |     types: [opened, reopened, synchronize] | ||||||
|  |     branches: [ master, v3, v4 ] | ||||||
|  |   push: | ||||||
|  |     branches: [ master, v3, v4 ] | ||||||
|  |     paths-ignore: | ||||||
|  |       - '.github/**' | ||||||
|  |       - '.gitea/**' | ||||||
|  |  | ||||||
|  | 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: actions/setup-go@v3 |  | ||||||
|       with: |  | ||||||
|         go-version: 1.21 |  | ||||||
|     - name: checkout |  | ||||||
|       uses: 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: actions/checkout@v3 |  | ||||||
|     - name: setup-go |  | ||||||
|       uses: actions/setup-go@v3 |  | ||||||
|       with: |  | ||||||
|         go-version: 1.21 |  | ||||||
|     - name: deps |  | ||||||
|       run: go get -v -t -d ./... |  | ||||||
|     - name: test |  | ||||||
|       env: |  | ||||||
|         INTEGRATION_TESTS: yes |  | ||||||
|       run: go test -mod readonly -v ./... |  | ||||||
| @@ -1,44 +1,5 @@ | |||||||
| run: | run: | ||||||
|   concurrency: 4 |   concurrency: 8 | ||||||
|   deadline: 5m |   timeout: 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 |  | ||||||
|   | |||||||
							
								
								
									
										27
									
								
								README.md
									
									
									
									
									
								
							
							
						
						
									
										27
									
								
								README.md
									
									
									
									
									
								
							| @@ -1,4 +1,9 @@ | |||||||
| # Micro [](https://opensource.org/licenses/Apache-2.0) [](https://pkg.go.dev/github.com/unistack-org/micro/v3?tab=overview) [](https://github.com/unistack-org/micro/actions?query=workflow%3Abuild+branch%3Amaster+event%3Apush) [](https://goreportcard.com/report/go.unistack.org/micro/v3) [](https://codecov.io/gh/unistack-org/micro) | # Micro | ||||||
|  |  | ||||||
|  | [](https://opensource.org/licenses/Apache-2.0) | ||||||
|  | [](https://pkg.go.dev/go.unistack.org/micro/v4?tab=overview) | ||||||
|  | [](https://git.unistack.org/unistack-org/micro/actions?query=workflow%3Abuild+branch%3Av4+event%3Apush) | ||||||
|  | [](https://goreportcard.com/report/go.unistack.org/micro/v4) | ||||||
|  |  | ||||||
| Micro is a standard library for microservices. | Micro is a standard library for microservices. | ||||||
|  |  | ||||||
| @@ -10,30 +15,20 @@ Micro provides the core requirements for distributed systems development includi | |||||||
|  |  | ||||||
| Micro abstracts away the details of distributed systems. Here are the main features. | Micro abstracts away the details of distributed systems. Here are the main features. | ||||||
|  |  | ||||||
| - **Authentication** - Auth is built in as a first class citizen. Authentication and authorization enable secure  |  | ||||||
| zero trust networking by providing every service an identity and certificates. This additionally includes rule  |  | ||||||
| based access control. |  | ||||||
|  |  | ||||||
| - **Dynamic Config** - Load and hot reload dynamic config from anywhere. The config interface provides a way to load application  | - **Dynamic Config** - Load and hot reload dynamic config from anywhere. The config interface provides a way to load application  | ||||||
| level config from any source such as env vars, file, etcd. You can merge the sources and even define fallbacks. | level config from any source such as env vars, cmdline, file, consul, vault... You can merge the sources and even define fallbacks. | ||||||
|  |  | ||||||
| - **Data Storage** - A simple data store interface to read, write and delete records. It includes support for memory, file and  | - **Data Storage** - A simple data store interface to read, write and delete records. It includes support for memory, file and  | ||||||
| CockroachDB by default. State and persistence becomes a core requirement beyond prototyping and Micro looks to build that into the framework. | s3. State and persistence becomes a core requirement beyond prototyping and Micro looks to build that into the framework. | ||||||
|  |  | ||||||
| - **Service Discovery** - Automatic service registration and name resolution. Service discovery is at the core of micro service  | - **Service Discovery** - Automatic service registration and name resolution. Service discovery is at the core of micro service  | ||||||
| development. When service A needs to speak to service B it needs the location of that service. | development. When service A needs to speak to service B it needs the location of that service. | ||||||
|  |  | ||||||
| - **Load Balancing** - Client side load balancing built on service discovery. Once we have the addresses of any number of instances  |  | ||||||
| of a service we now need a way to decide which node to route to. We use random hashed load balancing to provide even distribution  |  | ||||||
| across the services and retry a different node if there's a problem.  |  | ||||||
|  |  | ||||||
| - **Message Encoding** - Dynamic message encoding based on content-type. The client and server will use codecs along with content-type  | - **Message Encoding** - Dynamic message encoding based on content-type. The client and server will use codecs along with content-type  | ||||||
| to seamlessly encode and decode Go types for you. Any variety of messages could be encoded and sent from different clients. The client  | to seamlessly encode and decode Go types for you. Any variety of messages could be encoded and sent from different clients. The client  | ||||||
| and server handle this by default. | and server handle this by default. | ||||||
|  |  | ||||||
| - **Transport** - gRPC or http based request/response with support for bidirectional streaming. We provide an abstraction for synchronous communication. A request made to a service will be automatically resolved, load balanced, dialled and streamed. | - **Async Messaging** - Pub/Sub is built in as a first class citizen for asynchronous communication and event driven architectures. | ||||||
|  |  | ||||||
| - **Async Messaging** - PubSub is built in as a first class citizen for asynchronous communication and event driven architectures. |  | ||||||
| Event notifications are a core pattern in micro service development. | Event notifications are a core pattern in micro service development. | ||||||
|  |  | ||||||
| - **Synchronization** - Distributed systems are often built in an eventually consistent manner. Support for distributed locking and  | - **Synchronization** - Distributed systems are often built in an eventually consistent manner. Support for distributed locking and  | ||||||
| @@ -42,10 +37,6 @@ leadership are built in as a Sync interface. When using an eventually consistent | |||||||
| - **Pluggable Interfaces** - Micro makes use of Go interfaces for each system abstraction. Because of this these interfaces  | - **Pluggable Interfaces** - Micro makes use of Go interfaces for each system abstraction. Because of this these interfaces  | ||||||
| are pluggable and allows Micro to be runtime agnostic. | are pluggable and allows Micro to be runtime agnostic. | ||||||
|  |  | ||||||
| ## Getting Started |  | ||||||
|  |  | ||||||
| To be created. |  | ||||||
|  |  | ||||||
| ## License | ## License | ||||||
|  |  | ||||||
| Micro is Apache 2.0 licensed. | Micro is Apache 2.0 licensed. | ||||||
|   | |||||||
							
								
								
									
										15
									
								
								SECURITY.md
									
									
									
									
									
								
							
							
						
						
									
										15
									
								
								SECURITY.md
									
									
									
									
									
								
							| @@ -1,15 +0,0 @@ | |||||||
| # Security Policy |  | ||||||
|  |  | ||||||
| ## Supported Versions |  | ||||||
|  |  | ||||||
| Use this section to tell people about which versions of your project are |  | ||||||
| currently being supported with security updates. |  | ||||||
|  |  | ||||||
| | Version | Supported          | |  | ||||||
| | ------- | ------------------ | |  | ||||||
| | 3.7.x   | :white_check_mark: | |  | ||||||
| | < 3.7.0 | :x:                | |  | ||||||
|  |  | ||||||
| ## Reporting a Vulnerability |  | ||||||
|  |  | ||||||
| If you find any issue, please create github issue in this repo |  | ||||||
| @@ -6,8 +6,8 @@ import ( | |||||||
| 	"errors" | 	"errors" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/codec" | 	"go.unistack.org/micro/v4/codec" | ||||||
| 	"go.unistack.org/micro/v3/metadata" | 	"go.unistack.org/micro/v4/metadata" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // DefaultBroker default memory broker | // DefaultBroker default memory broker | ||||||
| @@ -18,6 +18,10 @@ var ( | |||||||
| 	ErrNotConnected = errors.New("broker not connected") | 	ErrNotConnected = errors.New("broker not connected") | ||||||
| 	// ErrDisconnected returns when broker disconnected | 	// ErrDisconnected returns when broker disconnected | ||||||
| 	ErrDisconnected = errors.New("broker disconnected") | 	ErrDisconnected = errors.New("broker disconnected") | ||||||
|  | 	// ErrInvalidMessage returns when invalid Message passed | ||||||
|  | 	ErrInvalidMessage = errors.New("invalid message") | ||||||
|  | 	// ErrInvalidHandler returns when subscriber passed to Subscribe | ||||||
|  | 	ErrInvalidHandler = errors.New("invalid handler, ony func(Message) error and func([]Message) error supported") | ||||||
| 	// DefaultGracefulTimeout | 	// DefaultGracefulTimeout | ||||||
| 	DefaultGracefulTimeout = 5 * time.Second | 	DefaultGracefulTimeout = 5 * time.Second | ||||||
| ) | ) | ||||||
| @@ -36,14 +40,12 @@ type Broker interface { | |||||||
| 	Connect(ctx context.Context) error | 	Connect(ctx context.Context) error | ||||||
| 	// Disconnect disconnect from broker | 	// Disconnect disconnect from broker | ||||||
| 	Disconnect(ctx context.Context) error | 	Disconnect(ctx context.Context) error | ||||||
|  | 	// NewMessage create new broker message to publish. | ||||||
|  | 	NewMessage(ctx context.Context, hdr metadata.Metadata, body interface{}, opts ...PublishOption) (Message, error) | ||||||
| 	// Publish message to broker topic | 	// Publish message to broker topic | ||||||
| 	Publish(ctx context.Context, topic string, msg *Message, opts ...PublishOption) error | 	Publish(ctx context.Context, topic string, messages ...Message) error | ||||||
| 	// Subscribe subscribes to topic message via handler | 	// Subscribe subscribes to topic message via handler | ||||||
| 	Subscribe(ctx context.Context, topic string, h Handler, opts ...SubscribeOption) (Subscriber, error) | 	Subscribe(ctx context.Context, topic string, handler interface{}, opts ...SubscribeOption) (Subscriber, error) | ||||||
| 	// BatchPublish messages to broker with multiple topics |  | ||||||
| 	BatchPublish(ctx context.Context, msgs []*Message, opts ...PublishOption) error |  | ||||||
| 	// BatchSubscribe subscribes to topic messages via handler |  | ||||||
| 	BatchSubscribe(ctx context.Context, topic string, h BatchHandler, opts ...SubscribeOption) (Subscriber, error) |  | ||||||
| 	// String type of broker | 	// String type of broker | ||||||
| 	String() string | 	String() string | ||||||
| 	// Live returns broker liveness | 	// Live returns broker liveness | ||||||
| @@ -55,72 +57,27 @@ type Broker interface { | |||||||
| } | } | ||||||
|  |  | ||||||
| type ( | type ( | ||||||
| 	FuncPublish        func(ctx context.Context, topic string, msg *Message, opts ...PublishOption) error | 	FuncPublish   func(ctx context.Context, topic string, messages ...Message) error | ||||||
| 	HookPublish   func(next FuncPublish) FuncPublish | 	HookPublish   func(next FuncPublish) FuncPublish | ||||||
| 	FuncBatchPublish   func(ctx context.Context, msgs []*Message, opts ...PublishOption) error | 	FuncSubscribe func(ctx context.Context, topic string, handler interface{}, opts ...SubscribeOption) (Subscriber, error) | ||||||
| 	HookBatchPublish   func(next FuncBatchPublish) FuncBatchPublish |  | ||||||
| 	FuncSubscribe      func(ctx context.Context, topic string, h Handler, opts ...SubscribeOption) (Subscriber, error) |  | ||||||
| 	HookSubscribe func(next FuncSubscribe) FuncSubscribe | 	HookSubscribe func(next FuncSubscribe) FuncSubscribe | ||||||
| 	FuncBatchSubscribe func(ctx context.Context, topic string, h BatchHandler, opts ...SubscribeOption) (Subscriber, error) |  | ||||||
| 	HookBatchSubscribe func(next FuncBatchSubscribe) FuncBatchSubscribe |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Handler is used to process messages via a subscription of a topic. | // Message is given to a subscription handler for processing | ||||||
| type Handler func(Event) error | type Message interface { | ||||||
|  | 	// Context for the message. | ||||||
| // Events contains multiple events |  | ||||||
| type Events []Event |  | ||||||
|  |  | ||||||
| // Ack try to ack all events and return |  | ||||||
| func (evs Events) Ack() error { |  | ||||||
| 	var err error |  | ||||||
| 	for _, ev := range evs { |  | ||||||
| 		if err = ev.Ack(); err != nil { |  | ||||||
| 			return err |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // SetError sets error on event |  | ||||||
| func (evs Events) SetError(err error) { |  | ||||||
| 	for _, ev := range evs { |  | ||||||
| 		ev.SetError(err) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // BatchHandler is used to process messages in batches via a subscription of a topic. |  | ||||||
| type BatchHandler func(Events) error |  | ||||||
|  |  | ||||||
| // Event is given to a subscription handler for processing |  | ||||||
| type Event interface { |  | ||||||
| 	// Context return context.Context for event |  | ||||||
| 	Context() context.Context | 	Context() context.Context | ||||||
| 	// Topic returns event topic | 	// Topic returns message destination topic. | ||||||
| 	Topic() string | 	Topic() string | ||||||
| 	// Message returns broker message | 	// Header returns message headers. | ||||||
| 	Message() *Message | 	Header() metadata.Metadata | ||||||
| 	// Ack acknowledge message | 	// Body returns broker message []byte slice | ||||||
|  | 	Body() []byte | ||||||
|  | 	// Unmarshal try to decode message body to dst. | ||||||
|  | 	// This is helper method that uses codec.Unmarshal. | ||||||
|  | 	Unmarshal(dst interface{}, opts ...codec.Option) error | ||||||
|  | 	// Ack acknowledge message if supported. | ||||||
| 	Ack() error | 	Ack() error | ||||||
| 	// Error returns message error (like decoding errors or some other) |  | ||||||
| 	Error() error |  | ||||||
| 	// SetError set event processing error |  | ||||||
| 	SetError(err error) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Message is used to transfer data |  | ||||||
| type Message struct { |  | ||||||
| 	// Header contains message metadata |  | ||||||
| 	Header metadata.Metadata |  | ||||||
| 	// Body contains message body |  | ||||||
| 	Body codec.RawMessage |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewMessage create broker message with topic filled |  | ||||||
| func NewMessage(topic string) *Message { |  | ||||||
| 	m := &Message{Header: metadata.New(2)} |  | ||||||
| 	m.Header.Set(metadata.HeaderTopic, topic) |  | ||||||
| 	return m |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // Subscriber is a convenience return type for the Subscribe method | // Subscriber is a convenience return type for the Subscribe method | ||||||
|   | |||||||
| @@ -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 { | ||||||
| @@ -33,16 +42,6 @@ func SetSubscribeOption(k, v interface{}) SubscribeOption { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // 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 | // SetPublishOption returns a function to setup a context with given value | ||||||
| func SetPublishOption(k, v interface{}) PublishOption { | func SetPublishOption(k, v interface{}) PublishOption { | ||||||
| 	return func(o *PublishOptions) { | 	return func(o *PublishOptions) { | ||||||
| @@ -52,3 +51,13 @@ func SetPublishOption(k, v interface{}) PublishOption { | |||||||
| 		o.Context = context.WithValue(o.Context, k, v) | 		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) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -49,17 +49,6 @@ func TestSetSubscribeOption(t *testing.T) { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func TestSetPublishOption(t *testing.T) { |  | ||||||
| 	type key struct{} |  | ||||||
| 	o := SetPublishOption(key{}, "test") |  | ||||||
| 	opts := &PublishOptions{} |  | ||||||
| 	o(opts) |  | ||||||
|  |  | ||||||
| 	if v, ok := opts.Context.Value(key{}).(string); !ok || v == "" { |  | ||||||
| 		t.Fatal("SetPublishOption not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestSetOption(t *testing.T) { | func TestSetOption(t *testing.T) { | ||||||
| 	type key struct{} | 	type key struct{} | ||||||
| 	o := SetOption(key{}, "test") | 	o := SetOption(key{}, "test") | ||||||
|   | |||||||
| @@ -2,66 +2,104 @@ package broker | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
|  | 	"strings" | ||||||
| 	"sync" | 	"sync" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/broker" | 	"go.unistack.org/micro/v4/broker" | ||||||
| 	"go.unistack.org/micro/v3/logger" | 	"go.unistack.org/micro/v4/codec" | ||||||
| 	"go.unistack.org/micro/v3/metadata" | 	"go.unistack.org/micro/v4/logger" | ||||||
| 	"go.unistack.org/micro/v3/options" | 	"go.unistack.org/micro/v4/metadata" | ||||||
| 	maddr "go.unistack.org/micro/v3/util/addr" | 	"go.unistack.org/micro/v4/options" | ||||||
| 	"go.unistack.org/micro/v3/util/id" | 	maddr "go.unistack.org/micro/v4/util/addr" | ||||||
| 	mnet "go.unistack.org/micro/v3/util/net" | 	"go.unistack.org/micro/v4/util/id" | ||||||
| 	"go.unistack.org/micro/v3/util/rand" | 	mnet "go.unistack.org/micro/v4/util/net" | ||||||
|  | 	"go.unistack.org/micro/v4/util/rand" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type memoryBroker struct { | type Broker struct { | ||||||
| 	funcPublish   broker.FuncPublish | 	funcPublish   broker.FuncPublish | ||||||
| 	funcBatchPublish   broker.FuncBatchPublish |  | ||||||
| 	funcSubscribe broker.FuncSubscribe | 	funcSubscribe broker.FuncSubscribe | ||||||
| 	funcBatchSubscribe broker.FuncBatchSubscribe | 	subscribers   map[string][]*Subscriber | ||||||
| 	subscribers        map[string][]*memorySubscriber |  | ||||||
| 	addr          string | 	addr          string | ||||||
| 	opts          broker.Options | 	opts          broker.Options | ||||||
| 	sync.RWMutex | 	sync.RWMutex | ||||||
| 	connected bool | 	connected bool | ||||||
| } | } | ||||||
|  |  | ||||||
| type memoryEvent struct { | type memoryMessage struct { | ||||||
| 	err     error | 	c     codec.Codec | ||||||
| 	message interface{} |  | ||||||
| 	topic string | 	topic string | ||||||
| 	opts    broker.Options | 	ctx   context.Context | ||||||
|  | 	body  []byte | ||||||
|  | 	hdr   metadata.Metadata | ||||||
|  | 	opts  broker.PublishOptions | ||||||
| } | } | ||||||
|  |  | ||||||
| type memorySubscriber struct { | func (m *memoryMessage) Ack() error { | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryMessage) Body() []byte { | ||||||
|  | 	return m.body | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryMessage) Header() metadata.Metadata { | ||||||
|  | 	return m.hdr | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryMessage) Context() context.Context { | ||||||
|  | 	return m.ctx | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryMessage) Topic() string { | ||||||
|  | 	return "" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryMessage) Unmarshal(dst interface{}, opts ...codec.Option) error { | ||||||
|  | 	return m.c.Unmarshal(m.body, dst) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type Subscriber struct { | ||||||
| 	ctx     context.Context | 	ctx     context.Context | ||||||
| 	exit    chan bool | 	exit    chan bool | ||||||
| 	handler      broker.Handler | 	handler interface{} | ||||||
| 	batchhandler broker.BatchHandler |  | ||||||
| 	id      string | 	id      string | ||||||
| 	topic   string | 	topic   string | ||||||
| 	opts    broker.SubscribeOptions | 	opts    broker.SubscribeOptions | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *memoryBroker) Options() broker.Options { | func (b *Broker) newCodec(ct string) (codec.Codec, error) { | ||||||
| 	return m.opts | 	if idx := strings.IndexRune(ct, ';'); idx >= 0 { | ||||||
|  | 		ct = ct[:idx] | ||||||
|  | 	} | ||||||
|  | 	b.RLock() | ||||||
|  | 	c, ok := b.opts.Codecs[ct] | ||||||
|  | 	b.RUnlock() | ||||||
|  | 	if ok { | ||||||
|  | 		return c, nil | ||||||
|  | 	} | ||||||
|  | 	return nil, codec.ErrUnknownContentType | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *memoryBroker) Address() string { | func (b *Broker) Options() broker.Options { | ||||||
| 	return m.addr | 	return b.opts | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *memoryBroker) Connect(ctx context.Context) error { | func (b *Broker) Address() string { | ||||||
|  | 	return b.addr | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Broker) Connect(ctx context.Context) error { | ||||||
| 	select { | 	select { | ||||||
| 	case <-ctx.Done(): | 	case <-ctx.Done(): | ||||||
| 		return ctx.Err() | 		return ctx.Err() | ||||||
| 	default: | 	default: | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	m.Lock() | 	b.Lock() | ||||||
| 	defer m.Unlock() | 	defer b.Unlock() | ||||||
|  |  | ||||||
| 	if m.connected { | 	if b.connected { | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -75,156 +113,128 @@ func (m *memoryBroker) Connect(ctx context.Context) error { | |||||||
| 	// set addr with port | 	// set addr with port | ||||||
| 	addr = mnet.HostPort(addr, 10000+i) | 	addr = mnet.HostPort(addr, 10000+i) | ||||||
|  |  | ||||||
| 	m.addr = addr | 	b.addr = addr | ||||||
| 	m.connected = true | 	b.connected = true | ||||||
|  |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *memoryBroker) Disconnect(ctx context.Context) error { | func (b *Broker) Disconnect(ctx context.Context) error { | ||||||
| 	select { | 	select { | ||||||
| 	case <-ctx.Done(): | 	case <-ctx.Done(): | ||||||
| 		return ctx.Err() | 		return ctx.Err() | ||||||
| 	default: | 	default: | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	m.Lock() | 	b.Lock() | ||||||
| 	defer m.Unlock() | 	defer b.Unlock() | ||||||
|  |  | ||||||
| 	if !m.connected { | 	if !b.connected { | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	m.connected = false | 	b.connected = false | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *memoryBroker) Init(opts ...broker.Option) error { | func (b *Broker) Init(opts ...broker.Option) error { | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&m.opts) | 		o(&b.opts) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	m.funcPublish = m.fnPublish | 	b.funcPublish = b.fnPublish | ||||||
| 	m.funcBatchPublish = m.fnBatchPublish | 	b.funcSubscribe = b.fnSubscribe | ||||||
| 	m.funcSubscribe = m.fnSubscribe |  | ||||||
| 	m.funcBatchSubscribe = m.fnBatchSubscribe |  | ||||||
|  |  | ||||||
| 	m.opts.Hooks.EachNext(func(hook options.Hook) { | 	b.opts.Hooks.EachPrev(func(hook options.Hook) { | ||||||
| 		switch h := hook.(type) { | 		switch h := hook.(type) { | ||||||
| 		case broker.HookPublish: | 		case broker.HookPublish: | ||||||
| 			m.funcPublish = h(m.funcPublish) | 			b.funcPublish = h(b.funcPublish) | ||||||
| 		case broker.HookBatchPublish: |  | ||||||
| 			m.funcBatchPublish = h(m.funcBatchPublish) |  | ||||||
| 		case broker.HookSubscribe: | 		case broker.HookSubscribe: | ||||||
| 			m.funcSubscribe = h(m.funcSubscribe) | 			b.funcSubscribe = h(b.funcSubscribe) | ||||||
| 		case broker.HookBatchSubscribe: |  | ||||||
| 			m.funcBatchSubscribe = h(m.funcBatchSubscribe) |  | ||||||
| 		} | 		} | ||||||
| 	}) | 	}) | ||||||
|  |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *memoryBroker) Publish(ctx context.Context, topic string, msg *broker.Message, opts ...broker.PublishOption) error { | func (b *Broker) NewMessage(ctx context.Context, hdr metadata.Metadata, body interface{}, opts ...broker.PublishOption) (broker.Message, error) { | ||||||
| 	return m.funcPublish(ctx, topic, msg, opts...) | 	options := broker.NewPublishOptions(opts...) | ||||||
|  | 	m := &memoryMessage{ctx: ctx, hdr: hdr, opts: options} | ||||||
|  | 	c, err := b.newCodec(m.opts.ContentType) | ||||||
|  | 	if err == nil { | ||||||
|  | 		m.body, err = c.Marshal(body) | ||||||
|  | 	} | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return m, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *memoryBroker) fnPublish(ctx context.Context, topic string, msg *broker.Message, opts ...broker.PublishOption) error { | func (b *Broker) Publish(ctx context.Context, topic string, messages ...broker.Message) error { | ||||||
| 	msg.Header.Set(metadata.HeaderTopic, topic) | 	return b.funcPublish(ctx, topic, messages...) | ||||||
| 	return m.publish(ctx, []*broker.Message{msg}, opts...) |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *memoryBroker) BatchPublish(ctx context.Context, msgs []*broker.Message, opts ...broker.PublishOption) error { | func (b *Broker) fnPublish(ctx context.Context, topic string, messages ...broker.Message) error { | ||||||
| 	return m.funcBatchPublish(ctx, msgs, opts...) | 	return b.publish(ctx, topic, messages...) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *memoryBroker) fnBatchPublish(ctx context.Context, msgs []*broker.Message, opts ...broker.PublishOption) error { | func (b *Broker) publish(ctx context.Context, topic string, messages ...broker.Message) error { | ||||||
| 	return m.publish(ctx, msgs, opts...) | 	b.RLock() | ||||||
| } | 	if !b.connected { | ||||||
|  | 		b.RUnlock() | ||||||
| func (m *memoryBroker) publish(ctx context.Context, msgs []*broker.Message, opts ...broker.PublishOption) error { |  | ||||||
| 	m.RLock() |  | ||||||
| 	if !m.connected { |  | ||||||
| 		m.RUnlock() |  | ||||||
| 		return broker.ErrNotConnected | 		return broker.ErrNotConnected | ||||||
| 	} | 	} | ||||||
| 	m.RUnlock() | 	b.RUnlock() | ||||||
|  |  | ||||||
| 	var err error |  | ||||||
|  |  | ||||||
| 	select { | 	select { | ||||||
| 	case <-ctx.Done(): | 	case <-ctx.Done(): | ||||||
| 		return ctx.Err() | 		return ctx.Err() | ||||||
| 	default: | 	default: | ||||||
| 		options := broker.NewPublishOptions(opts...) |  | ||||||
|  |  | ||||||
| 		msgTopicMap := make(map[string]broker.Events) |  | ||||||
| 		for _, v := range msgs { |  | ||||||
| 			p := &memoryEvent{opts: m.opts} |  | ||||||
|  |  | ||||||
| 			if m.opts.Codec == nil || options.BodyOnly { |  | ||||||
| 				p.topic, _ = v.Header.Get(metadata.HeaderTopic) |  | ||||||
| 				p.message = v.Body |  | ||||||
| 			} else { |  | ||||||
| 				p.topic, _ = v.Header.Get(metadata.HeaderTopic) |  | ||||||
| 				p.message, err = m.opts.Codec.Marshal(v) |  | ||||||
| 				if err != nil { |  | ||||||
| 					return err |  | ||||||
| 				} |  | ||||||
| 			} |  | ||||||
| 			msgTopicMap[p.topic] = append(msgTopicMap[p.topic], p) |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 		beh := m.opts.BatchErrorHandler | 	b.RLock() | ||||||
| 		eh := m.opts.ErrorHandler | 	subs, ok := b.subscribers[topic] | ||||||
|  | 	b.RUnlock() | ||||||
| 		for t, ms := range msgTopicMap { |  | ||||||
| 			m.RLock() |  | ||||||
| 			subs, ok := m.subscribers[t] |  | ||||||
| 			m.RUnlock() |  | ||||||
| 	if !ok { | 	if !ok { | ||||||
| 				continue | 		return nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	var err error | ||||||
|  |  | ||||||
| 	for _, sub := range subs { | 	for _, sub := range subs { | ||||||
| 				if sub.opts.BatchErrorHandler != nil { | 		switch s := sub.handler.(type) { | ||||||
| 					beh = sub.opts.BatchErrorHandler | 		default: | ||||||
|  | 			if b.opts.Logger.V(logger.ErrorLevel) { | ||||||
|  | 				b.opts.Logger.Error(ctx, "broker  handler error", broker.ErrInvalidHandler) | ||||||
| 			} | 			} | ||||||
| 				if sub.opts.ErrorHandler != nil { | 		case func(broker.Message) error: | ||||||
| 					eh = sub.opts.ErrorHandler | 			for _, message := range messages { | ||||||
| 				} | 				msg, ok := message.(*memoryMessage) | ||||||
|  | 				if !ok { | ||||||
| 				switch { | 					if b.opts.Logger.V(logger.ErrorLevel) { | ||||||
| 				// batch processing | 						b.opts.Logger.Error(ctx, "broker handler error", broker.ErrInvalidMessage) | ||||||
| 				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 | 				msg.topic = topic | ||||||
| 				case sub.handler != nil: | 				if err = s(msg); err == nil && sub.opts.AutoAck { | ||||||
| 					for _, p := range ms { | 					err = msg.Ack() | ||||||
| 						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 != nil { | ||||||
| 							if err = p.Ack(); err != nil { | 					if b.opts.Logger.V(logger.ErrorLevel) { | ||||||
| 								m.opts.Logger.Error(m.opts.Context, "broker ack error", err) | 						b.opts.Logger.Error(ctx, "broker handler error", err) | ||||||
| 					} | 					} | ||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
|  | 		case func([]broker.Message) error: | ||||||
|  | 			if err = s(messages); err == nil && sub.opts.AutoAck { | ||||||
|  | 				for _, message := range messages { | ||||||
|  | 					err = message.Ack() | ||||||
|  | 					if err != nil { | ||||||
|  | 						if b.opts.Logger.V(logger.ErrorLevel) { | ||||||
|  | 							b.opts.Logger.Error(ctx, "broker handler error", err) | ||||||
|  | 						} | ||||||
|  | 					} | ||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| @@ -233,17 +243,21 @@ func (m *memoryBroker) publish(ctx context.Context, msgs []*broker.Message, opts | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *memoryBroker) BatchSubscribe(ctx context.Context, topic string, handler broker.BatchHandler, opts ...broker.SubscribeOption) (broker.Subscriber, error) { | func (b *Broker) Subscribe(ctx context.Context, topic string, handler interface{}, opts ...broker.SubscribeOption) (broker.Subscriber, error) { | ||||||
| 	return m.funcBatchSubscribe(ctx, topic, handler, opts...) | 	return b.funcSubscribe(ctx, topic, handler, opts...) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *memoryBroker) fnBatchSubscribe(ctx context.Context, topic string, handler broker.BatchHandler, opts ...broker.SubscribeOption) (broker.Subscriber, error) { | func (b *Broker) fnSubscribe(ctx context.Context, topic string, handler interface{}, opts ...broker.SubscribeOption) (broker.Subscriber, error) { | ||||||
| 	m.RLock() | 	if err := broker.IsValidHandler(handler); err != nil { | ||||||
| 	if !m.connected { | 		return nil, err | ||||||
| 		m.RUnlock() | 	} | ||||||
|  |  | ||||||
|  | 	b.RLock() | ||||||
|  | 	if !b.connected { | ||||||
|  | 		b.RUnlock() | ||||||
| 		return nil, broker.ErrNotConnected | 		return nil, broker.ErrNotConnected | ||||||
| 	} | 	} | ||||||
| 	m.RUnlock() | 	b.RUnlock() | ||||||
|  |  | ||||||
| 	sid, err := id.New() | 	sid, err := id.New() | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| @@ -252,56 +266,7 @@ func (m *memoryBroker) fnBatchSubscribe(ctx context.Context, topic string, handl | |||||||
|  |  | ||||||
| 	options := broker.NewSubscribeOptions(opts...) | 	options := broker.NewSubscribeOptions(opts...) | ||||||
|  |  | ||||||
| 	sub := &memorySubscriber{ | 	sub := &Subscriber{ | ||||||
| 		exit:         make(chan bool, 1), |  | ||||||
| 		id:           sid, |  | ||||||
| 		topic:        topic, |  | ||||||
| 		batchhandler: handler, |  | ||||||
| 		opts:         options, |  | ||||||
| 		ctx:          ctx, |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	m.Lock() |  | ||||||
| 	m.subscribers[topic] = append(m.subscribers[topic], sub) |  | ||||||
| 	m.Unlock() |  | ||||||
|  |  | ||||||
| 	go func() { |  | ||||||
| 		<-sub.exit |  | ||||||
| 		m.Lock() |  | ||||||
| 		newSubscribers := make([]*memorySubscriber, 0, len(m.subscribers)-1) |  | ||||||
| 		for _, sb := range m.subscribers[topic] { |  | ||||||
| 			if sb.id == sub.id { |  | ||||||
| 				continue |  | ||||||
| 			} |  | ||||||
| 			newSubscribers = append(newSubscribers, sb) |  | ||||||
| 		} |  | ||||||
| 		m.subscribers[topic] = newSubscribers |  | ||||||
| 		m.Unlock() |  | ||||||
| 	}() |  | ||||||
|  |  | ||||||
| 	return sub, nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryBroker) Subscribe(ctx context.Context, topic string, handler broker.Handler, opts ...broker.SubscribeOption) (broker.Subscriber, error) { |  | ||||||
| 	return m.funcSubscribe(ctx, topic, handler, opts...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryBroker) fnSubscribe(ctx context.Context, topic string, handler broker.Handler, opts ...broker.SubscribeOption) (broker.Subscriber, error) { |  | ||||||
| 	m.RLock() |  | ||||||
| 	if !m.connected { |  | ||||||
| 		m.RUnlock() |  | ||||||
| 		return nil, broker.ErrNotConnected |  | ||||||
| 	} |  | ||||||
| 	m.RUnlock() |  | ||||||
|  |  | ||||||
| 	sid, err := id.New() |  | ||||||
| 	if err != nil { |  | ||||||
| 		return nil, err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	options := broker.NewSubscribeOptions(opts...) |  | ||||||
|  |  | ||||||
| 	sub := &memorySubscriber{ |  | ||||||
| 		exit:    make(chan bool, 1), | 		exit:    make(chan bool, 1), | ||||||
| 		id:      sid, | 		id:      sid, | ||||||
| 		topic:   topic, | 		topic:   topic, | ||||||
| @@ -310,102 +275,64 @@ func (m *memoryBroker) fnSubscribe(ctx context.Context, topic string, handler br | |||||||
| 		ctx:     ctx, | 		ctx:     ctx, | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	m.Lock() | 	b.Lock() | ||||||
| 	m.subscribers[topic] = append(m.subscribers[topic], sub) | 	b.subscribers[topic] = append(b.subscribers[topic], sub) | ||||||
| 	m.Unlock() | 	b.Unlock() | ||||||
|  |  | ||||||
| 	go func() { | 	go func() { | ||||||
| 		<-sub.exit | 		<-sub.exit | ||||||
| 		m.Lock() | 		b.Lock() | ||||||
| 		newSubscribers := make([]*memorySubscriber, 0, len(m.subscribers)-1) | 		newSubscribers := make([]*Subscriber, 0, len(b.subscribers)-1) | ||||||
| 		for _, sb := range m.subscribers[topic] { | 		for _, sb := range b.subscribers[topic] { | ||||||
| 			if sb.id == sub.id { | 			if sb.id == sub.id { | ||||||
| 				continue | 				continue | ||||||
| 			} | 			} | ||||||
| 			newSubscribers = append(newSubscribers, sb) | 			newSubscribers = append(newSubscribers, sb) | ||||||
| 		} | 		} | ||||||
| 		m.subscribers[topic] = newSubscribers | 		b.subscribers[topic] = newSubscribers | ||||||
| 		m.Unlock() | 		b.Unlock() | ||||||
| 	}() | 	}() | ||||||
|  |  | ||||||
| 	return sub, nil | 	return sub, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *memoryBroker) String() string { | func (b *Broker) String() string { | ||||||
| 	return "memory" | 	return "memory" | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *memoryBroker) Name() string { | func (b *Broker) Name() string { | ||||||
| 	return m.opts.Name | 	return b.opts.Name | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *memoryBroker) Live() bool { | func (b *Broker) Live() bool { | ||||||
| 	return true | 	return true | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *memoryBroker) Ready() bool { | func (b *Broker) Ready() bool { | ||||||
| 	return true | 	return true | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *memoryBroker) Health() bool { | func (b *Broker) Health() bool { | ||||||
| 	return true | 	return true | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *memoryEvent) Topic() string { | func (m *Subscriber) Options() broker.SubscribeOptions { | ||||||
| 	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 | 	return m.opts | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *memorySubscriber) Topic() string { | func (m *Subscriber) Topic() string { | ||||||
| 	return m.topic | 	return m.topic | ||||||
| } | } | ||||||
|  |  | ||||||
| func (m *memorySubscriber) Unsubscribe(ctx context.Context) error { | func (m *Subscriber) Unsubscribe(ctx context.Context) error { | ||||||
| 	m.exit <- true | 	m.exit <- true | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewBroker return new memory broker | // NewBroker return new memory broker | ||||||
| func NewBroker(opts ...broker.Option) broker.Broker { | func NewBroker(opts ...broker.Option) broker.Broker { | ||||||
| 	return &memoryBroker{ | 	return &Broker{ | ||||||
| 		opts:        broker.NewOptions(opts...), | 		opts:        broker.NewOptions(opts...), | ||||||
| 		subscribers: make(map[string][]*memorySubscriber), | 		subscribers: make(map[string][]*Subscriber), | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -5,62 +5,23 @@ import ( | |||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"testing" | 	"testing" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/broker" | 	"go.uber.org/atomic" | ||||||
| 	"go.unistack.org/micro/v3/metadata" | 	"go.unistack.org/micro/v4/broker" | ||||||
|  | 	"go.unistack.org/micro/v4/codec" | ||||||
|  | 	"go.unistack.org/micro/v4/metadata" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| func TestMemoryBatchBroker(t *testing.T) { | type hldr struct { | ||||||
| 	b := NewBroker() | 	c atomic.Int64 | ||||||
| 	ctx := context.Background() | } | ||||||
|  |  | ||||||
| 	if err := b.Init(); err != nil { | func (h *hldr) Handler(m broker.Message) error { | ||||||
| 		t.Fatalf("Unexpected init error %v", err) | 	h.c.Add(1) | ||||||
| 	} | 	return nil | ||||||
|  |  | ||||||
| 	if err := b.Connect(ctx); err != nil { |  | ||||||
| 		t.Fatalf("Unexpected connect error %v", err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	topic := "test" |  | ||||||
| 	count := 10 |  | ||||||
|  |  | ||||||
| 	fn := func(evts broker.Events) error { |  | ||||||
| 		return evts.Ack() |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	sub, err := b.BatchSubscribe(ctx, topic, fn) |  | ||||||
| 	if err != nil { |  | ||||||
| 		t.Fatalf("Unexpected error subscribing %v", err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	msgs := make([]*broker.Message, 0, count) |  | ||||||
| 	for i := 0; i < count; i++ { |  | ||||||
| 		message := &broker.Message{ |  | ||||||
| 			Header: map[string]string{ |  | ||||||
| 				metadata.HeaderTopic: topic, |  | ||||||
| 				"foo":                "bar", |  | ||||||
| 				"id":                 fmt.Sprintf("%d", i), |  | ||||||
| 			}, |  | ||||||
| 			Body: []byte(`"hello world"`), |  | ||||||
| 		} |  | ||||||
| 		msgs = append(msgs, message) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if err := b.BatchPublish(ctx, msgs); err != nil { |  | ||||||
| 		t.Fatalf("Unexpected error publishing %v", err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if err := sub.Unsubscribe(ctx); err != nil { |  | ||||||
| 		t.Fatalf("Unexpected error unsubscribing from %s: %v", topic, err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if err := b.Disconnect(ctx); err != nil { |  | ||||||
| 		t.Fatalf("Unexpected connect error %v", err) |  | ||||||
| 	} |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func TestMemoryBroker(t *testing.T) { | func TestMemoryBroker(t *testing.T) { | ||||||
| 	b := NewBroker() | 	b := NewBroker(broker.Codec("application/octet-stream", codec.NewCodec())) | ||||||
| 	ctx := context.Background() | 	ctx := context.Background() | ||||||
|  |  | ||||||
| 	if err := b.Init(); err != nil { | 	if err := b.Init(); err != nil { | ||||||
| @@ -72,38 +33,33 @@ func TestMemoryBroker(t *testing.T) { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	topic := "test" | 	topic := "test" | ||||||
| 	count := 10 | 	count := int64(10) | ||||||
|  |  | ||||||
| 	fn := func(p broker.Event) error { | 	h := &hldr{} | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	sub, err := b.Subscribe(ctx, topic, fn) | 	sub, err := b.Subscribe(ctx, topic, h.Handler) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		t.Fatalf("Unexpected error subscribing %v", err) | 		t.Fatalf("Unexpected error subscribing %v", err) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	msgs := make([]*broker.Message, 0, count) | 	for i := int64(0); i < count; i++ { | ||||||
| 	for i := 0; i < count; i++ { | 		message, err := b.NewMessage(ctx, | ||||||
| 		message := &broker.Message{ | 			metadata.Pairs( | ||||||
| 			Header: map[string]string{ | 				"foo", "bar", | ||||||
| 				metadata.HeaderTopic: topic, | 				"id", fmt.Sprintf("%d", i), | ||||||
| 				"foo":                "bar", | 			), | ||||||
| 				"id":                 fmt.Sprintf("%d", i), | 			[]byte(`"hello world"`), | ||||||
| 			}, | 			broker.PublishContentType("application/octet-stream"), | ||||||
| 			Body: []byte(`"hello world"`), | 		) | ||||||
|  | 		if err != nil { | ||||||
|  | 			t.Fatal(err) | ||||||
| 		} | 		} | ||||||
| 		msgs = append(msgs, message) |  | ||||||
|  |  | ||||||
| 		if err := b.Publish(ctx, topic, message); err != nil { | 		if err := b.Publish(ctx, topic, message); err != nil { | ||||||
| 			t.Fatalf("Unexpected error publishing %d err: %v", i, err) | 			t.Fatalf("Unexpected error publishing %d err: %v", i, err) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if err := b.BatchPublish(ctx, msgs); err != nil { |  | ||||||
| 		t.Fatalf("Unexpected error publishing %v", err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if err := sub.Unsubscribe(ctx); err != nil { | 	if err := sub.Unsubscribe(ctx); err != nil { | ||||||
| 		t.Fatalf("Unexpected error unsubscribing from %s: %v", topic, err) | 		t.Fatalf("Unexpected error unsubscribing from %s: %v", topic, err) | ||||||
| 	} | 	} | ||||||
| @@ -111,4 +67,8 @@ func TestMemoryBroker(t *testing.T) { | |||||||
| 	if err := b.Disconnect(ctx); err != nil { | 	if err := b.Disconnect(ctx); err != nil { | ||||||
| 		t.Fatalf("Unexpected connect error %v", err) | 		t.Fatalf("Unexpected connect error %v", err) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	if h.c.Load() != count { | ||||||
|  | 		t.Fatal("invalid messages count received") | ||||||
|  | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -3,24 +3,37 @@ package broker | |||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"strings" | 	"strings" | ||||||
|  | 	"sync" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/options" | 	"go.unistack.org/micro/v4/codec" | ||||||
|  | 	"go.unistack.org/micro/v4/metadata" | ||||||
|  | 	"go.unistack.org/micro/v4/options" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type NoopBroker struct { | type NoopBroker struct { | ||||||
| 	funcPublish   FuncPublish | 	funcPublish   FuncPublish | ||||||
| 	funcBatchPublish   FuncBatchPublish |  | ||||||
| 	funcSubscribe FuncSubscribe | 	funcSubscribe FuncSubscribe | ||||||
| 	funcBatchSubscribe FuncBatchSubscribe |  | ||||||
| 	opts          Options | 	opts          Options | ||||||
|  | 	sync.RWMutex | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) newCodec(ct string) (codec.Codec, error) { | ||||||
|  | 	if idx := strings.IndexRune(ct, ';'); idx >= 0 { | ||||||
|  | 		ct = ct[:idx] | ||||||
|  | 	} | ||||||
|  | 	b.RLock() | ||||||
|  | 	c, ok := b.opts.Codecs[ct] | ||||||
|  | 	b.RUnlock() | ||||||
|  | 	if ok { | ||||||
|  | 		return c, nil | ||||||
|  | 	} | ||||||
|  | 	return nil, codec.ErrUnknownContentType | ||||||
| } | } | ||||||
|  |  | ||||||
| func NewBroker(opts ...Option) *NoopBroker { | func NewBroker(opts ...Option) *NoopBroker { | ||||||
| 	b := &NoopBroker{opts: NewOptions(opts...)} | 	b := &NoopBroker{opts: NewOptions(opts...)} | ||||||
| 	b.funcPublish = b.fnPublish | 	b.funcPublish = b.fnPublish | ||||||
| 	b.funcBatchPublish = b.fnBatchPublish |  | ||||||
| 	b.funcSubscribe = b.fnSubscribe | 	b.funcSubscribe = b.fnSubscribe | ||||||
| 	b.funcBatchSubscribe = b.fnBatchSubscribe |  | ||||||
|  |  | ||||||
| 	return b | 	return b | ||||||
| } | } | ||||||
| @@ -55,20 +68,14 @@ func (b *NoopBroker) Init(opts ...Option) error { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	b.funcPublish = b.fnPublish | 	b.funcPublish = b.fnPublish | ||||||
| 	b.funcBatchPublish = b.fnBatchPublish |  | ||||||
| 	b.funcSubscribe = b.fnSubscribe | 	b.funcSubscribe = b.fnSubscribe | ||||||
| 	b.funcBatchSubscribe = b.fnBatchSubscribe |  | ||||||
|  |  | ||||||
| 	b.opts.Hooks.EachNext(func(hook options.Hook) { | 	b.opts.Hooks.EachPrev(func(hook options.Hook) { | ||||||
| 		switch h := hook.(type) { | 		switch h := hook.(type) { | ||||||
| 		case HookPublish: | 		case HookPublish: | ||||||
| 			b.funcPublish = h(b.funcPublish) | 			b.funcPublish = h(b.funcPublish) | ||||||
| 		case HookBatchPublish: |  | ||||||
| 			b.funcBatchPublish = h(b.funcBatchPublish) |  | ||||||
| 		case HookSubscribe: | 		case HookSubscribe: | ||||||
| 			b.funcSubscribe = h(b.funcSubscribe) | 			b.funcSubscribe = h(b.funcSubscribe) | ||||||
| 		case HookBatchSubscribe: |  | ||||||
| 			b.funcBatchSubscribe = h(b.funcBatchSubscribe) |  | ||||||
| 		} | 		} | ||||||
| 	}) | 	}) | ||||||
|  |  | ||||||
| @@ -87,43 +94,72 @@ func (b *NoopBroker) Address() string { | |||||||
| 	return strings.Join(b.opts.Addrs, ",") | 	return strings.Join(b.opts.Addrs, ",") | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *NoopBroker) fnBatchPublish(_ context.Context, _ []*Message, _ ...PublishOption) error { | type noopMessage struct { | ||||||
|  | 	c    codec.Codec | ||||||
|  | 	ctx  context.Context | ||||||
|  | 	body []byte | ||||||
|  | 	hdr  metadata.Metadata | ||||||
|  | 	opts PublishOptions | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *noopMessage) Ack() error { | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *NoopBroker) BatchPublish(ctx context.Context, msgs []*Message, opts ...PublishOption) error { | func (m *noopMessage) Body() []byte { | ||||||
| 	return b.funcBatchPublish(ctx, msgs, opts...) | 	return m.body | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *NoopBroker) fnPublish(_ context.Context, _ string, _ *Message, _ ...PublishOption) error { | func (m *noopMessage) Header() metadata.Metadata { | ||||||
|  | 	return m.hdr | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *noopMessage) Context() context.Context { | ||||||
|  | 	return m.ctx | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *noopMessage) Topic() string { | ||||||
|  | 	return "" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *noopMessage) Unmarshal(dst interface{}, opts ...codec.Option) error { | ||||||
|  | 	return m.c.Unmarshal(m.body, dst) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) NewMessage(ctx context.Context, hdr metadata.Metadata, body interface{}, opts ...PublishOption) (Message, error) { | ||||||
|  | 	options := NewPublishOptions(opts...) | ||||||
|  | 	m := &noopMessage{ctx: ctx, hdr: hdr, opts: options} | ||||||
|  | 	c, err := b.newCodec(m.opts.ContentType) | ||||||
|  | 	if err == nil { | ||||||
|  | 		m.body, err = c.Marshal(body) | ||||||
|  | 	} | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return m, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) fnPublish(_ context.Context, _ string, _ ...Message) error { | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *NoopBroker) Publish(ctx context.Context, topic string, msg *Message, opts ...PublishOption) error { | func (b *NoopBroker) Publish(ctx context.Context, topic string, msg ...Message) error { | ||||||
| 	return b.funcPublish(ctx, topic, msg, opts...) | 	return b.funcPublish(ctx, topic, msg...) | ||||||
| } | } | ||||||
|  |  | ||||||
| type NoopSubscriber struct { | type NoopSubscriber struct { | ||||||
| 	ctx     context.Context | 	ctx     context.Context | ||||||
| 	topic   string | 	topic   string | ||||||
| 	handler      Handler | 	handler interface{} | ||||||
| 	batchHandler BatchHandler |  | ||||||
| 	opts    SubscribeOptions | 	opts    SubscribeOptions | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *NoopBroker) fnBatchSubscribe(ctx context.Context, topic string, handler BatchHandler, opts ...SubscribeOption) (Subscriber, error) { | func (b *NoopBroker) fnSubscribe(ctx context.Context, topic string, handler interface{}, opts ...SubscribeOption) (Subscriber, error) { | ||||||
| 	return &NoopSubscriber{ctx: ctx, topic: topic, opts: NewSubscribeOptions(opts...), batchHandler: handler}, nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (b *NoopBroker) BatchSubscribe(ctx context.Context, topic string, handler BatchHandler, opts ...SubscribeOption) (Subscriber, error) { |  | ||||||
| 	return b.funcBatchSubscribe(ctx, topic, handler, opts...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (b *NoopBroker) fnSubscribe(ctx context.Context, topic string, handler Handler, opts ...SubscribeOption) (Subscriber, error) { |  | ||||||
| 	return &NoopSubscriber{ctx: ctx, topic: topic, opts: NewSubscribeOptions(opts...), handler: handler}, nil | 	return &NoopSubscriber{ctx: ctx, topic: topic, opts: NewSubscribeOptions(opts...), handler: handler}, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (b *NoopBroker) Subscribe(ctx context.Context, topic string, handler Handler, opts ...SubscribeOption) (Subscriber, error) { | func (b *NoopBroker) Subscribe(ctx context.Context, topic string, handler interface{}, opts ...SubscribeOption) (Subscriber, error) { | ||||||
| 	return b.funcSubscribe(ctx, topic, handler, opts...) | 	return b.funcSubscribe(ctx, topic, handler, opts...) | ||||||
| } | } | ||||||
|  |  | ||||||
|   | |||||||
| @@ -10,9 +10,9 @@ type testHook struct { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (t *testHook) Publish1(fn FuncPublish) FuncPublish { | func (t *testHook) Publish1(fn FuncPublish) FuncPublish { | ||||||
| 	return func(ctx context.Context, topic string, msg *Message, opts ...PublishOption) error { | 	return func(ctx context.Context, topic string, messages ...Message) error { | ||||||
| 		t.f = true | 		t.f = true | ||||||
| 		return fn(ctx, topic, msg, opts...) | 		return fn(ctx, topic, messages...) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|   | |||||||
| @@ -5,46 +5,46 @@ import ( | |||||||
| 	"crypto/tls" | 	"crypto/tls" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/codec" | 	"go.unistack.org/micro/v4/codec" | ||||||
| 	"go.unistack.org/micro/v3/logger" | 	"go.unistack.org/micro/v4/logger" | ||||||
| 	"go.unistack.org/micro/v3/meter" | 	"go.unistack.org/micro/v4/meter" | ||||||
| 	"go.unistack.org/micro/v3/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/sync" | ||||||
| 	"go.unistack.org/micro/v3/tracer" | 	"go.unistack.org/micro/v4/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 | ||||||
| 	// Codec holds the codec for marshal/unmarshal | 	// Codecs holds the codecs for marshal/unmarshal based on content-type | ||||||
| 	Codec codec.Codec | 	Codecs map[string]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 | ||||||
| 	// TLSConfig holds tls.TLSConfig options |  | ||||||
| 	TLSConfig *tls.Config |  | ||||||
| 	// ErrorHandler used when broker can't unmarshal incoming message |  | ||||||
| 	ErrorHandler Handler |  | ||||||
| 	// BatchErrorHandler used when broker can't unmashal incoming messages |  | ||||||
| 	BatchErrorHandler BatchHandler |  | ||||||
| 	// Name holds the broker name |  | ||||||
| 	Name string |  | ||||||
| 	// Addrs holds the broker address |  | ||||||
| 	Addrs []string |  | ||||||
| 	// Wait waits for a collection of goroutines to finish | 	// Wait waits for a collection of goroutines to finish | ||||||
| 	Wait *sync.WaitGroup | 	Wait *sync.WaitGroup | ||||||
| 	// GracefulTimeout contains time to wait to finish in flight requests | 	// TLSConfig holds tls.TLSConfig options | ||||||
| 	GracefulTimeout time.Duration | 	TLSConfig *tls.Config | ||||||
|  |  | ||||||
|  | 	// Addrs holds the broker address | ||||||
|  | 	Addrs []string | ||||||
| 	// Hooks can be run before broker Publish/BatchPublish and | 	// Hooks can be run before broker Publish/BatchPublish and | ||||||
| 	// Subscribe/BatchSubscribe methods | 	// Subscribe/BatchSubscribe methods | ||||||
| 	Hooks options.Hooks | 	Hooks options.Hooks | ||||||
|  |  | ||||||
|  | 	// GracefulTimeout contains time to wait to finish in flight requests | ||||||
|  | 	GracefulTimeout time.Duration | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewOptions create new Options | // NewOptions create new Options | ||||||
| @@ -54,10 +54,11 @@ func NewOptions(opts ...Option) Options { | |||||||
| 		Logger:          logger.DefaultLogger, | 		Logger:          logger.DefaultLogger, | ||||||
| 		Context:         context.Background(), | 		Context:         context.Background(), | ||||||
| 		Meter:           meter.DefaultMeter, | 		Meter:           meter.DefaultMeter, | ||||||
| 		Codec:           codec.DefaultCodec, | 		Codecs:          make(map[string]codec.Codec), | ||||||
| 		Tracer:          tracer.DefaultTracer, | 		Tracer:          tracer.DefaultTracer, | ||||||
| 		GracefulTimeout: DefaultGracefulTimeout, | 		GracefulTimeout: DefaultGracefulTimeout, | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&options) | 		o(&options) | ||||||
| 	} | 	} | ||||||
| @@ -73,10 +74,13 @@ func Context(ctx context.Context) Option { | |||||||
|  |  | ||||||
| // PublishOptions struct | // PublishOptions struct | ||||||
| type PublishOptions struct { | type PublishOptions struct { | ||||||
| 	// Context holds external options | 	// ContentType for message body | ||||||
| 	Context context.Context | 	ContentType string | ||||||
| 	// BodyOnly flag says the message contains raw body bytes | 	// BodyOnly flag says the message contains raw body bytes and don't need | ||||||
|  | 	// codec Marshal method | ||||||
| 	BodyOnly bool | 	BodyOnly bool | ||||||
|  | 	// Context holds custom options | ||||||
|  | 	Context context.Context | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewPublishOptions creates PublishOptions struct | // NewPublishOptions creates PublishOptions struct | ||||||
| @@ -94,10 +98,6 @@ func NewPublishOptions(opts ...PublishOption) PublishOptions { | |||||||
| type SubscribeOptions struct { | type SubscribeOptions struct { | ||||||
| 	// Context holds external options | 	// Context holds external options | ||||||
| 	Context context.Context | 	Context context.Context | ||||||
| 	// ErrorHandler used when broker can't unmarshal incoming message |  | ||||||
| 	ErrorHandler Handler |  | ||||||
| 	// BatchErrorHandler used when broker can't unmashal incoming messages |  | ||||||
| 	BatchErrorHandler BatchHandler |  | ||||||
| 	// Group holds consumer group | 	// Group holds consumer group | ||||||
| 	Group string | 	Group string | ||||||
| 	// AutoAck flag specifies auto ack of incoming message when no error happens | 	// AutoAck flag specifies auto ack of incoming message when no error happens | ||||||
| @@ -116,6 +116,13 @@ type Option func(*Options) | |||||||
| // PublishOption func | // PublishOption func | ||||||
| type PublishOption func(*PublishOptions) | type PublishOption func(*PublishOptions) | ||||||
|  |  | ||||||
|  | // PublishContentType sets message content-type that used to Marshal | ||||||
|  | func PublishContentType(ct string) PublishOption { | ||||||
|  | 	return func(o *PublishOptions) { | ||||||
|  | 		o.ContentType = ct | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // PublishBodyOnly publish only body of the message | // PublishBodyOnly publish only body of the message | ||||||
| func PublishBodyOnly(b bool) PublishOption { | func PublishBodyOnly(b bool) PublishOption { | ||||||
| 	return func(o *PublishOptions) { | 	return func(o *PublishOptions) { | ||||||
| @@ -123,13 +130,6 @@ func PublishBodyOnly(b bool) PublishOption { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // PublishContext sets the context |  | ||||||
| func PublishContext(ctx context.Context) PublishOption { |  | ||||||
| 	return func(o *PublishOptions) { |  | ||||||
| 		o.Context = ctx |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Addrs sets the host addresses to be used by the broker | // Addrs sets the host addresses to be used by the broker | ||||||
| func Addrs(addrs ...string) Option { | func Addrs(addrs ...string) Option { | ||||||
| 	return func(o *Options) { | 	return func(o *Options) { | ||||||
| @@ -137,51 +137,10 @@ func Addrs(addrs ...string) Option { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // Codec sets the codec used for encoding/decoding used where | // Codec sets the codec used for encoding/decoding messages | ||||||
| // a broker does not support headers | func Codec(ct string, c codec.Codec) Option { | ||||||
| func Codec(c codec.Codec) Option { |  | ||||||
| 	return func(o *Options) { | 	return func(o *Options) { | ||||||
| 		o.Codec = c | 		o.Codecs[ct] = 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 |  | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -248,14 +207,6 @@ func SubscribeContext(ctx context.Context) SubscribeOption { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // DisableAutoAck disables auto ack |  | ||||||
| // Deprecated |  | ||||||
| func DisableAutoAck() SubscribeOption { |  | ||||||
| 	return func(o *SubscribeOptions) { |  | ||||||
| 		o.AutoAck = false |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // SubscribeAutoAck contol auto acking of messages | // SubscribeAutoAck contol auto acking of messages | ||||||
| // after they have been handled. | // after they have been handled. | ||||||
| func SubscribeAutoAck(b bool) SubscribeOption { | func SubscribeAutoAck(b bool) SubscribeOption { | ||||||
|   | |||||||
							
								
								
									
										14
									
								
								broker/subscriber.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								broker/subscriber.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,14 @@ | |||||||
|  | package broker | ||||||
|  |  | ||||||
|  | // IsValidHandler func signature | ||||||
|  | func IsValidHandler(sub interface{}) error { | ||||||
|  | 	switch sub.(type) { | ||||||
|  | 	default: | ||||||
|  | 		return ErrInvalidHandler | ||||||
|  | 	case func(Message) error: | ||||||
|  | 		break | ||||||
|  | 	case func([]Message) error: | ||||||
|  | 		break | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
| @@ -5,7 +5,7 @@ import ( | |||||||
| 	"math" | 	"math" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/util/backoff" | 	"go.unistack.org/micro/v4/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) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -5,8 +5,8 @@ import ( | |||||||
| 	"context" | 	"context" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/codec" | 	"go.unistack.org/micro/v4/codec" | ||||||
| 	"go.unistack.org/micro/v3/metadata" | 	"go.unistack.org/micro/v4/metadata" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| @@ -29,18 +29,14 @@ var ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
| // Client is the interface used to make requests to services. | // Client is the interface used to make requests to services. | ||||||
| // It supports Request/Response via Transport and Publishing via the Broker. |  | ||||||
| // It also supports bidirectional streaming of requests. | // It also supports bidirectional streaming of requests. | ||||||
| type Client interface { | type Client interface { | ||||||
| 	Name() string | 	Name() string | ||||||
| 	Init(opts ...Option) error | 	Init(opts ...Option) error | ||||||
| 	Options() Options | 	Options() Options | ||||||
| 	NewMessage(topic string, msg interface{}, opts ...MessageOption) Message |  | ||||||
| 	NewRequest(service string, endpoint string, req interface{}, opts ...RequestOption) Request | 	NewRequest(service string, endpoint string, req interface{}, opts ...RequestOption) Request | ||||||
| 	Call(ctx context.Context, req Request, rsp interface{}, opts ...CallOption) error | 	Call(ctx context.Context, req Request, rsp interface{}, opts ...CallOption) error | ||||||
| 	Stream(ctx context.Context, req Request, opts ...CallOption) (Stream, error) | 	Stream(ctx context.Context, req Request, opts ...CallOption) (Stream, error) | ||||||
| 	Publish(ctx context.Context, msg Message, opts ...PublishOption) error |  | ||||||
| 	BatchPublish(ctx context.Context, msg []Message, opts ...PublishOption) error |  | ||||||
| 	String() string | 	String() string | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -49,20 +45,8 @@ type ( | |||||||
| 	HookCall   func(next FuncCall) FuncCall | 	HookCall   func(next FuncCall) FuncCall | ||||||
| 	FuncStream func(ctx context.Context, req Request, opts ...CallOption) (Stream, error) | 	FuncStream func(ctx context.Context, req Request, opts ...CallOption) (Stream, error) | ||||||
| 	HookStream func(next FuncStream) FuncStream | 	HookStream func(next FuncStream) FuncStream | ||||||
| 	FuncPublish      func(ctx context.Context, msg Message, opts ...PublishOption) error |  | ||||||
| 	HookPublish      func(next FuncPublish) FuncPublish |  | ||||||
| 	FuncBatchPublish func(ctx context.Context, msg []Message, opts ...PublishOption) error |  | ||||||
| 	HookBatchPublish func(next FuncBatchPublish) FuncBatchPublish |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Message is the interface for publishing asynchronously |  | ||||||
| type Message interface { |  | ||||||
| 	Topic() string |  | ||||||
| 	Payload() interface{} |  | ||||||
| 	ContentType() string |  | ||||||
| 	Metadata() metadata.Metadata |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Request is the interface for a synchronous request used by Call or Stream | // Request is the interface for a synchronous request used by Call or Stream | ||||||
| type Request interface { | type Request interface { | ||||||
| 	// The service to call | 	// The service to call | ||||||
| @@ -121,11 +105,5 @@ type Option func(*Options) | |||||||
| // CallOption used by Call or Stream | // CallOption used by Call or Stream | ||||||
| type CallOption func(*CallOptions) | type CallOption func(*CallOptions) | ||||||
|  |  | ||||||
| // PublishOption used by Publish |  | ||||||
| type PublishOption func(*PublishOptions) |  | ||||||
|  |  | ||||||
| // MessageOption used by NewMessage |  | ||||||
| type MessageOption func(*MessageOptions) |  | ||||||
|  |  | ||||||
| // RequestOption used by NewRequest | // RequestOption used by NewRequest | ||||||
| type RequestOption func(*RequestOptions) | type RequestOption func(*RequestOptions) | ||||||
|   | |||||||
| @@ -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 { | ||||||
| @@ -23,16 +32,6 @@ func NewContext(ctx context.Context, c Client) context.Context { | |||||||
| 	return context.WithValue(ctx, clientKey{}, c) | 	return context.WithValue(ctx, clientKey{}, c) | ||||||
| } | } | ||||||
|  |  | ||||||
| // SetPublishOption returns a function to setup a context with given value |  | ||||||
| func SetPublishOption(k, v interface{}) PublishOption { |  | ||||||
| 	return func(o *PublishOptions) { |  | ||||||
| 		if o.Context == nil { |  | ||||||
| 			o.Context = context.Background() |  | ||||||
| 		} |  | ||||||
| 		o.Context = context.WithValue(o.Context, k, v) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // SetCallOption returns a function to setup a context with given value | // SetCallOption returns a function to setup a context with given value | ||||||
| func SetCallOption(k, v interface{}) CallOption { | func SetCallOption(k, v interface{}) CallOption { | ||||||
| 	return func(o *CallOptions) { | 	return func(o *CallOptions) { | ||||||
|   | |||||||
| @@ -39,17 +39,6 @@ func TestNewNilContext(t *testing.T) { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func TestSetPublishOption(t *testing.T) { |  | ||||||
| 	type key struct{} |  | ||||||
| 	o := SetPublishOption(key{}, "test") |  | ||||||
| 	opts := &PublishOptions{} |  | ||||||
| 	o(opts) |  | ||||||
|  |  | ||||||
| 	if v, ok := opts.Context.Value(key{}).(string); !ok || v == "" { |  | ||||||
| 		t.Fatal("SetPublishOption not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestSetCallOption(t *testing.T) { | func TestSetCallOption(t *testing.T) { | ||||||
| 	type key struct{} | 	type key struct{} | ||||||
| 	o := SetCallOption(key{}, "test") | 	o := SetCallOption(key{}, "test") | ||||||
|   | |||||||
| @@ -4,8 +4,8 @@ import ( | |||||||
| 	"context" | 	"context" | ||||||
| 	"sort" | 	"sort" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/errors" | 	"go.unistack.org/micro/v4/errors" | ||||||
| 	"go.unistack.org/micro/v3/router" | 	"go.unistack.org/micro/v4/router" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // LookupFunc is used to lookup routes for a service | // LookupFunc is used to lookup routes for a service | ||||||
|   | |||||||
							
								
								
									
										167
									
								
								client/noop.go
									
									
									
									
									
								
							
							
						
						
									
										167
									
								
								client/noop.go
									
									
									
									
									
								
							| @@ -3,18 +3,16 @@ package client | |||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"os" |  | ||||||
| 	"strconv" | 	"strconv" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/broker" | 	"go.unistack.org/micro/v4/codec" | ||||||
| 	"go.unistack.org/micro/v3/codec" | 	"go.unistack.org/micro/v4/errors" | ||||||
| 	"go.unistack.org/micro/v3/errors" | 	"go.unistack.org/micro/v4/metadata" | ||||||
| 	"go.unistack.org/micro/v3/metadata" | 	"go.unistack.org/micro/v4/options" | ||||||
| 	"go.unistack.org/micro/v3/options" | 	"go.unistack.org/micro/v4/selector" | ||||||
| 	"go.unistack.org/micro/v3/selector" | 	"go.unistack.org/micro/v4/semconv" | ||||||
| 	"go.unistack.org/micro/v3/semconv" | 	"go.unistack.org/micro/v4/tracer" | ||||||
| 	"go.unistack.org/micro/v3/tracer" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // DefaultCodecs will be used to encode/decode data | // DefaultCodecs will be used to encode/decode data | ||||||
| @@ -23,19 +21,11 @@ var DefaultCodecs = map[string]codec.Codec{ | |||||||
| } | } | ||||||
|  |  | ||||||
| type noopClient struct { | type noopClient struct { | ||||||
| 	funcPublish      FuncPublish |  | ||||||
| 	funcBatchPublish FuncBatchPublish |  | ||||||
| 	funcCall   FuncCall | 	funcCall   FuncCall | ||||||
| 	funcStream FuncStream | 	funcStream FuncStream | ||||||
| 	opts       Options | 	opts       Options | ||||||
| } | } | ||||||
|  |  | ||||||
| type noopMessage struct { |  | ||||||
| 	topic   string |  | ||||||
| 	payload interface{} |  | ||||||
| 	opts    MessageOptions |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type noopRequest struct { | type noopRequest struct { | ||||||
| 	body        interface{} | 	body        interface{} | ||||||
| 	codec       codec.Codec | 	codec       codec.Codec | ||||||
| @@ -52,8 +42,6 @@ func NewClient(opts ...Option) Client { | |||||||
|  |  | ||||||
| 	n.funcCall = n.fnCall | 	n.funcCall = n.fnCall | ||||||
| 	n.funcStream = n.fnStream | 	n.funcStream = n.fnStream | ||||||
| 	n.funcPublish = n.fnPublish |  | ||||||
| 	n.funcBatchPublish = n.fnBatchPublish |  | ||||||
|  |  | ||||||
| 	return n | 	return n | ||||||
| } | } | ||||||
| @@ -158,32 +146,6 @@ func (n *noopStream) CloseSend() error { | |||||||
| 	return n.err | 	return n.err | ||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopMessage) Topic() string { |  | ||||||
| 	return n.topic |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (n *noopMessage) Payload() interface{} { |  | ||||||
| 	return n.payload |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (n *noopMessage) ContentType() string { |  | ||||||
| 	return n.opts.ContentType |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (n *noopMessage) Metadata() metadata.Metadata { |  | ||||||
| 	return n.opts.Metadata |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (n *noopClient) newCodec(contentType string) (codec.Codec, error) { |  | ||||||
| 	if cf, ok := n.opts.Codecs[contentType]; ok { |  | ||||||
| 		return cf, nil |  | ||||||
| 	} |  | ||||||
| 	if cf, ok := DefaultCodecs[contentType]; ok { |  | ||||||
| 		return cf, nil |  | ||||||
| 	} |  | ||||||
| 	return nil, codec.ErrUnknownContentType |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (n *noopClient) Init(opts ...Option) error { | func (n *noopClient) Init(opts ...Option) error { | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&n.opts) | 		o(&n.opts) | ||||||
| @@ -191,19 +153,13 @@ func (n *noopClient) Init(opts ...Option) error { | |||||||
|  |  | ||||||
| 	n.funcCall = n.fnCall | 	n.funcCall = n.fnCall | ||||||
| 	n.funcStream = n.fnStream | 	n.funcStream = n.fnStream | ||||||
| 	n.funcPublish = n.fnPublish |  | ||||||
| 	n.funcBatchPublish = n.fnBatchPublish |  | ||||||
|  |  | ||||||
| 	n.opts.Hooks.EachNext(func(hook options.Hook) { | 	n.opts.Hooks.EachPrev(func(hook options.Hook) { | ||||||
| 		switch h := hook.(type) { | 		switch h := hook.(type) { | ||||||
| 		case HookCall: | 		case HookCall: | ||||||
| 			n.funcCall = h(n.funcCall) | 			n.funcCall = h(n.funcCall) | ||||||
| 		case HookStream: | 		case HookStream: | ||||||
| 			n.funcStream = h(n.funcStream) | 			n.funcStream = h(n.funcStream) | ||||||
| 		case HookPublish: |  | ||||||
| 			n.funcPublish = h(n.funcPublish) |  | ||||||
| 		case HookBatchPublish: |  | ||||||
| 			n.funcBatchPublish = h(n.funcBatchPublish) |  | ||||||
| 		} | 		} | ||||||
| 	}) | 	}) | ||||||
|  |  | ||||||
| @@ -222,7 +178,7 @@ func (n *noopClient) Call(ctx context.Context, req Request, rsp interface{}, opt | |||||||
| 	ts := time.Now() | 	ts := time.Now() | ||||||
| 	n.opts.Meter.Counter(semconv.ClientRequestInflight, "endpoint", req.Endpoint()).Inc() | 	n.opts.Meter.Counter(semconv.ClientRequestInflight, "endpoint", req.Endpoint()).Inc() | ||||||
| 	var sp tracer.Span | 	var sp tracer.Span | ||||||
| 	ctx, sp = n.opts.Tracer.Start(ctx, req.Endpoint()+" rpc-client", | 	ctx, sp = n.opts.Tracer.Start(ctx, "rpc-client", | ||||||
| 		tracer.WithSpanKind(tracer.SpanKindClient), | 		tracer.WithSpanKind(tracer.SpanKindClient), | ||||||
| 		tracer.WithSpanLabels("endpoint", req.Endpoint()), | 		tracer.WithSpanLabels("endpoint", req.Endpoint()), | ||||||
| 	) | 	) | ||||||
| @@ -298,7 +254,7 @@ func (n *noopClient) fnCall(ctx context.Context, req Request, rsp interface{}, o | |||||||
| 		// 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 | ||||||
| @@ -312,7 +268,7 @@ func (n *noopClient) fnCall(ctx context.Context, req Request, rsp interface{}, o | |||||||
| 			// 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 | ||||||
| @@ -372,20 +328,15 @@ func (n *noopClient) fnCall(ctx context.Context, req Request, rsp interface{}, o | |||||||
| 	return gerr | 	return gerr | ||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopClient) NewRequest(service, endpoint string, req interface{}, opts ...RequestOption) Request { | func (n *noopClient) NewRequest(service, endpoint string, _ interface{}, _ ...RequestOption) Request { | ||||||
| 	return &noopRequest{service: service, endpoint: endpoint} | 	return &noopRequest{service: service, endpoint: endpoint} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopClient) NewMessage(topic string, msg interface{}, opts ...MessageOption) Message { |  | ||||||
| 	options := NewMessageOptions(append([]MessageOption{MessageContentType(n.opts.ContentType)}, opts...)...) |  | ||||||
| 	return &noopMessage{topic: topic, payload: msg, opts: options} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (n *noopClient) Stream(ctx context.Context, req Request, opts ...CallOption) (Stream, error) { | func (n *noopClient) Stream(ctx context.Context, req Request, opts ...CallOption) (Stream, error) { | ||||||
| 	ts := time.Now() | 	ts := time.Now() | ||||||
| 	n.opts.Meter.Counter(semconv.ClientRequestInflight, "endpoint", req.Endpoint()).Inc() | 	n.opts.Meter.Counter(semconv.ClientRequestInflight, "endpoint", req.Endpoint()).Inc() | ||||||
| 	var sp tracer.Span | 	var sp tracer.Span | ||||||
| 	ctx, sp = n.opts.Tracer.Start(ctx, req.Endpoint()+" rpc-client", | 	ctx, sp = n.opts.Tracer.Start(ctx, "rpc-client", | ||||||
| 		tracer.WithSpanKind(tracer.SpanKindClient), | 		tracer.WithSpanKind(tracer.SpanKindClient), | ||||||
| 		tracer.WithSpanLabels("endpoint", req.Endpoint()), | 		tracer.WithSpanLabels("endpoint", req.Endpoint()), | ||||||
| 	) | 	) | ||||||
| @@ -466,7 +417,7 @@ func (n *noopClient) fnStream(ctx context.Context, req Request, opts ...CallOpti | |||||||
| 		// 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 | ||||||
| @@ -480,7 +431,7 @@ func (n *noopClient) fnStream(ctx context.Context, req Request, opts ...CallOpti | |||||||
| 			// 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 | ||||||
| @@ -546,92 +497,6 @@ func (n *noopClient) fnStream(ctx context.Context, req Request, opts ...CallOpti | |||||||
| 	return nil, grr | 	return nil, grr | ||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopClient) stream(ctx context.Context, addr string, req Request, opts CallOptions) (Stream, error) { | func (n *noopClient) stream(ctx context.Context, _ string, _ Request, _ CallOptions) (Stream, error) { | ||||||
| 	return &noopStream{ctx: ctx}, nil | 	return &noopStream{ctx: ctx}, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopClient) BatchPublish(ctx context.Context, ps []Message, opts ...PublishOption) error { |  | ||||||
| 	return n.funcBatchPublish(ctx, ps, opts...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (n *noopClient) fnBatchPublish(ctx context.Context, ps []Message, opts ...PublishOption) error { |  | ||||||
| 	return n.publish(ctx, ps, opts...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (n *noopClient) Publish(ctx context.Context, p Message, opts ...PublishOption) error { |  | ||||||
| 	return n.funcPublish(ctx, p, opts...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (n *noopClient) fnPublish(ctx context.Context, p Message, opts ...PublishOption) error { |  | ||||||
| 	return n.publish(ctx, []Message{p}, opts...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (n *noopClient) publish(ctx context.Context, ps []Message, opts ...PublishOption) error { |  | ||||||
| 	options := NewPublishOptions(opts...) |  | ||||||
|  |  | ||||||
| 	msgs := make([]*broker.Message, 0, len(ps)) |  | ||||||
|  |  | ||||||
| 	// get proxy |  | ||||||
| 	exchange := "" |  | ||||||
| 	if v, ok := os.LookupEnv("MICRO_PROXY"); ok { |  | ||||||
| 		exchange = v |  | ||||||
| 	} |  | ||||||
| 	// get the exchange |  | ||||||
| 	if len(options.Exchange) > 0 { |  | ||||||
| 		exchange = options.Exchange |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	omd, ok := metadata.FromOutgoingContext(ctx) |  | ||||||
| 	if !ok { |  | ||||||
| 		omd = metadata.New(0) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	for _, p := range ps { |  | ||||||
| 		md := metadata.Copy(omd) |  | ||||||
| 		md[metadata.HeaderContentType] = p.ContentType() |  | ||||||
| 		topic := p.Topic() |  | ||||||
| 		if len(exchange) > 0 { |  | ||||||
| 			topic = exchange |  | ||||||
| 		} |  | ||||||
| 		md[metadata.HeaderTopic] = topic |  | ||||||
| 		iter := p.Metadata().Iterator() |  | ||||||
| 		var k, v string |  | ||||||
| 		for iter.Next(&k, &v) { |  | ||||||
| 			md.Set(k, v) |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		var body []byte |  | ||||||
|  |  | ||||||
| 		// passed in raw data |  | ||||||
| 		if d, ok := p.Payload().(*codec.Frame); ok { |  | ||||||
| 			body = d.Data |  | ||||||
| 		} else { |  | ||||||
| 			// use codec for payload |  | ||||||
| 			cf, err := n.newCodec(p.ContentType()) |  | ||||||
| 			if err != nil { |  | ||||||
| 				return errors.InternalServerError("go.micro.client", err.Error()) |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			// set the body |  | ||||||
| 			b, err := cf.Marshal(p.Payload()) |  | ||||||
| 			if err != nil { |  | ||||||
| 				return errors.InternalServerError("go.micro.client", err.Error()) |  | ||||||
| 			} |  | ||||||
| 			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), |  | ||||||
| 	) |  | ||||||
| } |  | ||||||
|   | |||||||
| @@ -1,35 +0,0 @@ | |||||||
| 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,21 +6,31 @@ import ( | |||||||
| 	"net" | 	"net" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/broker" | 	"go.unistack.org/micro/v4/broker" | ||||||
| 	"go.unistack.org/micro/v3/codec" | 	"go.unistack.org/micro/v4/codec" | ||||||
| 	"go.unistack.org/micro/v3/logger" | 	"go.unistack.org/micro/v4/logger" | ||||||
| 	"go.unistack.org/micro/v3/metadata" | 	"go.unistack.org/micro/v4/metadata" | ||||||
| 	"go.unistack.org/micro/v3/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/router" | 	"go.unistack.org/micro/v4/router" | ||||||
| 	"go.unistack.org/micro/v3/selector" | 	"go.unistack.org/micro/v4/selector" | ||||||
| 	"go.unistack.org/micro/v3/selector/random" | 	"go.unistack.org/micro/v4/selector/random" | ||||||
| 	"go.unistack.org/micro/v3/tracer" | 	"go.unistack.org/micro/v4/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Options holds client options | // Options holds client options | ||||||
| type Options struct { | type Options struct { | ||||||
|  | 	// Codecs map | ||||||
|  | 	Codecs map[string]codec.Codec | ||||||
|  |  | ||||||
|  | 	// Proxy is used for proxy requests | ||||||
|  | 	Proxy string | ||||||
|  | 	// ContentType is used to select codec | ||||||
|  | 	ContentType string | ||||||
|  | 	// Name is the client name | ||||||
|  | 	Name string | ||||||
|  |  | ||||||
| 	// Selector used to select needed address | 	// Selector used to select needed address | ||||||
| 	Selector selector.Selector | 	Selector selector.Selector | ||||||
| 	// Logger used to log messages | 	// Logger used to log messages | ||||||
| @@ -35,31 +45,28 @@ type Options struct { | |||||||
| 	Context context.Context | 	Context context.Context | ||||||
| 	// Router used to get route | 	// Router used to get route | ||||||
| 	Router router.Router | 	Router router.Router | ||||||
|  |  | ||||||
| 	// TLSConfig specifies tls.Config for secure connection | 	// TLSConfig specifies tls.Config for secure connection | ||||||
| 	TLSConfig *tls.Config | 	TLSConfig *tls.Config | ||||||
| 	// Codecs map |  | ||||||
| 	Codecs map[string]codec.Codec |  | ||||||
| 	// Lookup func used to get destination addr | 	// Lookup func used to get destination addr | ||||||
| 	Lookup LookupFunc | 	Lookup LookupFunc | ||||||
| 	// Proxy is used for proxy requests | 	// ContextDialer used to connect | ||||||
| 	Proxy string | 	ContextDialer func(context.Context, string) (net.Conn, error) | ||||||
| 	// ContentType is used to select codec |  | ||||||
| 	ContentType string |  | ||||||
| 	// Name is the client name |  | ||||||
| 	Name string |  | ||||||
| 	// Wrappers contains wrappers | 	// Wrappers contains wrappers | ||||||
| 	Wrappers []Wrapper | 	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 can be run before broker Publish/BatchPublish and |  | ||||||
| 	// Subscribe/BatchSubscribe methods |  | ||||||
| 	Hooks options.Hooks |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewCallOptions creates new call options struct | // NewCallOptions creates new call options struct | ||||||
| @@ -73,6 +80,16 @@ func NewCallOptions(opts ...CallOption) 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 | ||||||
| @@ -80,33 +97,30 @@ 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 | ||||||
|  |  | ||||||
| 	// 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 | ||||||
| @@ -123,43 +137,6 @@ func Context(ctx context.Context) Option { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewPublishOptions create new PublishOptions struct from option |  | ||||||
| func NewPublishOptions(opts ...PublishOption) PublishOptions { |  | ||||||
| 	options := PublishOptions{} |  | ||||||
| 	for _, o := range opts { |  | ||||||
| 		o(&options) |  | ||||||
| 	} |  | ||||||
| 	return options |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // PublishOptions holds publish options |  | ||||||
| type PublishOptions struct { |  | ||||||
| 	// Context used for external options |  | ||||||
| 	Context context.Context |  | ||||||
| 	// Exchange topic exchange name |  | ||||||
| 	Exchange string |  | ||||||
| 	// BodyOnly will publish only message body |  | ||||||
| 	BodyOnly bool |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewMessageOptions creates message options struct |  | ||||||
| func NewMessageOptions(opts ...MessageOption) MessageOptions { |  | ||||||
| 	options := MessageOptions{Metadata: metadata.New(1)} |  | ||||||
| 	for _, o := range opts { |  | ||||||
| 		o(&options) |  | ||||||
| 	} |  | ||||||
| 	return options |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // MessageOptions holds client message options |  | ||||||
| type MessageOptions struct { |  | ||||||
| 	// Metadata additional metadata |  | ||||||
| 	Metadata metadata.Metadata |  | ||||||
| 	// ContentType specify content-type of message |  | ||||||
| 	// deprecated |  | ||||||
| 	ContentType string |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewRequestOptions creates new RequestOptions struct | // NewRequestOptions creates new RequestOptions struct | ||||||
| func NewRequestOptions(opts ...RequestOption) RequestOptions { | func NewRequestOptions(opts ...RequestOption) RequestOptions { | ||||||
| 	options := RequestOptions{} | 	options := RequestOptions{} | ||||||
| @@ -360,43 +337,6 @@ func DialTimeout(d time.Duration) Option { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // WithExchange sets the exchange to route a message through |  | ||||||
| // Deprecated |  | ||||||
| func WithExchange(e string) PublishOption { |  | ||||||
| 	return func(o *PublishOptions) { |  | ||||||
| 		o.Exchange = e |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // PublishExchange sets the exchange to route a message through |  | ||||||
| func PublishExchange(e string) PublishOption { |  | ||||||
| 	return func(o *PublishOptions) { |  | ||||||
| 		o.Exchange = e |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // WithBodyOnly publish only message body |  | ||||||
| // DERECATED |  | ||||||
| func WithBodyOnly(b bool) PublishOption { |  | ||||||
| 	return func(o *PublishOptions) { |  | ||||||
| 		o.BodyOnly = b |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // PublishBodyOnly publish only message body |  | ||||||
| func PublishBodyOnly(b bool) PublishOption { |  | ||||||
| 	return func(o *PublishOptions) { |  | ||||||
| 		o.BodyOnly = b |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // PublishContext sets the context in publish options |  | ||||||
| func PublishContext(ctx context.Context) PublishOption { |  | ||||||
| 	return func(o *PublishOptions) { |  | ||||||
| 		o.Context = ctx |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // WithContextDialer pass ContextDialer to client call | // WithContextDialer pass ContextDialer to client call | ||||||
| func WithContextDialer(fn func(context.Context, string) (net.Conn, error)) CallOption { | func WithContextDialer(fn func(context.Context, string) (net.Conn, error)) CallOption { | ||||||
| 	return func(o *CallOptions) { | 	return func(o *CallOptions) { | ||||||
| @@ -508,30 +448,6 @@ func WithSelectOptions(sops ...selector.SelectOption) CallOption { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // WithMessageContentType sets the message content type |  | ||||||
| // Deprecated |  | ||||||
| func WithMessageContentType(ct string) MessageOption { |  | ||||||
| 	return func(o *MessageOptions) { |  | ||||||
| 		o.Metadata.Set(metadata.HeaderContentType, ct) |  | ||||||
| 		o.ContentType = ct |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // MessageContentType sets the message content type |  | ||||||
| func MessageContentType(ct string) MessageOption { |  | ||||||
| 	return func(o *MessageOptions) { |  | ||||||
| 		o.Metadata.Set(metadata.HeaderContentType, ct) |  | ||||||
| 		o.ContentType = ct |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // MessageMetadata sets the message metadata |  | ||||||
| func MessageMetadata(k, v string) MessageOption { |  | ||||||
| 	return func(o *MessageOptions) { |  | ||||||
| 		o.Metadata.Set(k, v) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // StreamingRequest specifies that request is streaming | // StreamingRequest specifies that request is streaming | ||||||
| func StreamingRequest(b bool) RequestOption { | func StreamingRequest(b bool) RequestOption { | ||||||
| 	return func(o *RequestOptions) { | 	return func(o *RequestOptions) { | ||||||
|   | |||||||
| @@ -3,7 +3,7 @@ package client | |||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/errors" | 	"go.unistack.org/micro/v4/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/v3/errors" | 	"go.unistack.org/micro/v4/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/v3/codec" | 	"go.unistack.org/micro/v4/codec" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type testRequest struct { | type testRequest struct { | ||||||
|   | |||||||
| @@ -3,7 +3,7 @@ package cluster | |||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/metadata" | 	"go.unistack.org/micro/v4/metadata" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Message sent to member in cluster | // Message sent to member in cluster | ||||||
|   | |||||||
| @@ -3,6 +3,8 @@ package codec | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"errors" | 	"errors" | ||||||
|  |  | ||||||
|  | 	"gopkg.in/yaml.v3" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| @@ -54,3 +56,22 @@ func (m *RawMessage) UnmarshalJSON(data []byte) error { | |||||||
| 	*m = append((*m)[0:0], data...) | 	*m = append((*m)[0:0], data...) | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // MarshalYAML returns m as the JSON encoding of m. | ||||||
|  | func (m *RawMessage) MarshalYAML() ([]byte, error) { | ||||||
|  | 	if m == nil { | ||||||
|  | 		return []byte("null"), nil | ||||||
|  | 	} else if len(*m) == 0 { | ||||||
|  | 		return []byte("null"), nil | ||||||
|  | 	} | ||||||
|  | 	return *m, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // UnmarshalYAML sets *m to a copy of data. | ||||||
|  | func (m *RawMessage) UnmarshalYAML(n *yaml.Node) error { | ||||||
|  | 	if m == nil { | ||||||
|  | 		return errors.New("RawMessage UnmarshalYAML on nil pointer") | ||||||
|  | 	} | ||||||
|  | 	*m = append((*m)[0:0], []byte(n.Value)...) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|   | |||||||
| @@ -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,5 +1,7 @@ | |||||||
| package codec | package codec | ||||||
|  |  | ||||||
|  | import "gopkg.in/yaml.v3" | ||||||
|  |  | ||||||
| // Frame gives us the ability to define raw data to send over the pipes | // Frame gives us the ability to define raw data to send over the pipes | ||||||
| type Frame struct { | type Frame struct { | ||||||
| 	Data []byte | 	Data []byte | ||||||
| @@ -20,6 +22,17 @@ func (m *Frame) UnmarshalJSON(data []byte) error { | |||||||
| 	return m.Unmarshal(data) | 	return m.Unmarshal(data) | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // MarshalYAML returns frame data | ||||||
|  | func (m *Frame) MarshalYAML() ([]byte, error) { | ||||||
|  | 	return m.Marshal() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // UnmarshalYAML set frame data | ||||||
|  | func (m *Frame) UnmarshalYAML(n *yaml.Node) error { | ||||||
|  | 	m.Data = []byte(n.Value) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
| // ProtoMessage noop func | // ProtoMessage noop func | ||||||
| func (m *Frame) ProtoMessage() {} | func (m *Frame) ProtoMessage() {} | ||||||
|  |  | ||||||
|   | |||||||
| @@ -17,7 +17,7 @@ syntax = "proto3"; | |||||||
| package micro.codec; | package micro.codec; | ||||||
|  |  | ||||||
| option cc_enable_arenas = true; | option cc_enable_arenas = true; | ||||||
| option go_package = "go.unistack.org/micro/v3/codec;codec"; | option go_package = "go.unistack.org/micro/v4/codec;codec"; | ||||||
| option java_multiple_files = true; | option java_multiple_files = true; | ||||||
| option java_outer_classname = "MicroCodec"; | option java_outer_classname = "MicroCodec"; | ||||||
| option java_package = "micro.codec"; | option java_package = "micro.codec"; | ||||||
|   | |||||||
| @@ -3,7 +3,7 @@ package codec | |||||||
| import ( | import ( | ||||||
| 	"encoding/json" | 	"encoding/json" | ||||||
|  |  | ||||||
| 	codecpb "go.unistack.org/micro-proto/v3/codec" | 	codecpb "go.unistack.org/micro-proto/v4/codec" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type noopCodec struct { | type noopCodec struct { | ||||||
|   | |||||||
| @@ -3,9 +3,9 @@ package codec | |||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/logger" | 	"go.unistack.org/micro/v4/logger" | ||||||
| 	"go.unistack.org/micro/v3/meter" | 	"go.unistack.org/micro/v4/meter" | ||||||
| 	"go.unistack.org/micro/v3/tracer" | 	"go.unistack.org/micro/v4/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Option func | // Option func | ||||||
|   | |||||||
| @@ -13,7 +13,7 @@ type Validator interface { | |||||||
| } | } | ||||||
|  |  | ||||||
| // DefaultConfig default config | // DefaultConfig default config | ||||||
| var DefaultConfig Config = NewConfig() | var DefaultConfig = NewConfig() | ||||||
|  |  | ||||||
| // DefaultWatcherMinInterval default min interval for poll changes | // DefaultWatcherMinInterval default min interval for poll changes | ||||||
| var DefaultWatcherMinInterval = 5 * time.Second | var DefaultWatcherMinInterval = 5 * time.Second | ||||||
|   | |||||||
| @@ -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 { | ||||||
|   | |||||||
| @@ -9,10 +9,10 @@ import ( | |||||||
|  |  | ||||||
| 	"dario.cat/mergo" | 	"dario.cat/mergo" | ||||||
| 	"github.com/google/uuid" | 	"github.com/google/uuid" | ||||||
| 	"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 { | ||||||
| @@ -37,7 +37,7 @@ func (c *defaultConfig) Init(opts ...Option) error { | |||||||
| 	c.funcLoad = c.fnLoad | 	c.funcLoad = c.fnLoad | ||||||
| 	c.funcSave = c.fnSave | 	c.funcSave = c.fnSave | ||||||
|  |  | ||||||
| 	c.opts.Hooks.EachNext(func(hook options.Hook) { | 	c.opts.Hooks.EachPrev(func(hook options.Hook) { | ||||||
| 		switch h := hook.(type) { | 		switch h := hook.(type) { | ||||||
| 		case HookLoad: | 		case HookLoad: | ||||||
| 			c.funcLoad = h(c.funcLoad) | 			c.funcLoad = h(c.funcLoad) | ||||||
| @@ -210,7 +210,7 @@ func fillValue(value reflect.Value, val string) error { | |||||||
| 				return err | 				return err | ||||||
| 			} | 			} | ||||||
| 			value.Set(reflect.ValueOf(v)) | 			value.Set(reflect.ValueOf(v)) | ||||||
| 		case value.Type().String() == "time.Duration" && value.Type().PkgPath() == "go.unistack.org/micro/v3/util/time": | 		case value.Type().String() == "time.Duration" && value.Type().PkgPath() == "go.unistack.org/micro/v4/util/time": | ||||||
| 			v, err := mtime.ParseDuration(val) | 			v, err := mtime.ParseDuration(val) | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| 				return err | 				return err | ||||||
|   | |||||||
| @@ -3,24 +3,26 @@ package config_test | |||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"fmt" | 	"fmt" | ||||||
|  | 	"reflect" | ||||||
| 	"testing" | 	"testing" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/config" | 	"go.unistack.org/micro/v4/config" | ||||||
| 	mid "go.unistack.org/micro/v3/util/id" | 	mtime "go.unistack.org/micro/v4/util/time" | ||||||
| 	mtime "go.unistack.org/micro/v3/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 { | ||||||
| @@ -112,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) | ||||||
| @@ -134,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/v3/codec" | 	"go.unistack.org/micro/v4/codec" | ||||||
| 	"go.unistack.org/micro/v3/logger" | 	"go.unistack.org/micro/v4/logger" | ||||||
| 	"go.unistack.org/micro/v3/meter" | 	"go.unistack.org/micro/v4/meter" | ||||||
| 	"go.unistack.org/micro/v3/options" | 	"go.unistack.org/micro/v4/options" | ||||||
| 	"go.unistack.org/micro/v3/tracer" | 	"go.unistack.org/micro/v4/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Options hold the config options | // Options hold the config options | ||||||
| @@ -41,14 +41,16 @@ 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 | ||||||
| 	// AllowFail flag to allow fail in config source |  | ||||||
| 	AllowFail bool |  | ||||||
| 	// SkipLoad runs only if condition returns true | 	// SkipLoad runs only if condition returns true | ||||||
| 	SkipLoad func(context.Context, Config) bool | 	SkipLoad func(context.Context, Config) bool | ||||||
| 	// SkipSave runs only if condition returns true | 	// SkipSave runs only if condition returns true | ||||||
| 	SkipSave func(context.Context, Config) bool | 	SkipSave func(context.Context, Config) bool | ||||||
| 	// Hooks can be run before/after config Save/Load | 	// Hooks can be run before/after config Save/Load | ||||||
| 	Hooks options.Hooks | 	Hooks options.Hooks | ||||||
|  |  | ||||||
|  | 	// AllowFail flag to allow fail in config source | ||||||
|  | 	AllowFail bool | ||||||
| } | } | ||||||
|  |  | ||||||
| // Option function signature | // Option function signature | ||||||
| @@ -278,10 +280,10 @@ func WatchCoalesce(b bool) WatchOption { | |||||||
| } | } | ||||||
|  |  | ||||||
| // 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) WatchOption { | func WatchInterval(minTime, maxTime time.Duration) WatchOption { | ||||||
| 	return func(o *WatchOptions) { | 	return func(o *WatchOptions) { | ||||||
| 		o.MinInterval = min | 		o.MinInterval = minTime | ||||||
| 		o.MaxInterval = max | 		o.MaxInterval = maxTime | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|   | |||||||
| @@ -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/v3/errors;errors"; | option go_package = "go.unistack.org/micro/v4/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,7 +2,7 @@ package errors | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"encoding/json" | 	"encoding/json" | ||||||
| 	er "errors" | 	"errors" | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"net/http" | 	"net/http" | ||||||
| 	"testing" | 	"testing" | ||||||
| @@ -26,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) | ||||||
| 	} | 	} | ||||||
| @@ -42,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) | ||||||
| @@ -57,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
									
									
									
									
									
								
							
							
						
						
									
										27
									
								
								event.go
									
									
									
									
									
								
							| @@ -1,27 +0,0 @@ | |||||||
| 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...) |  | ||||||
| } |  | ||||||
| @@ -1,3 +1,5 @@ | |||||||
|  | //go:build ignore | ||||||
|  |  | ||||||
| package flow | package flow | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|   | |||||||
							
								
								
									
										342
									
								
								flow/default.go
									
									
									
									
									
								
							
							
						
						
									
										342
									
								
								flow/default.go
									
									
									
									
									
								
							| @@ -1,17 +1,19 @@ | |||||||
|  | //go:build ignore | ||||||
|  |  | ||||||
| package flow | package flow | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"fmt" | 	"fmt" | ||||||
|  | 	"path/filepath" | ||||||
| 	"sync" | 	"sync" | ||||||
|  |  | ||||||
| 	"github.com/silas/dag" | 	"github.com/heimdalr/dag" | ||||||
| 	"go.unistack.org/micro/v3/client" | 	"go.unistack.org/micro/v4/client" | ||||||
| 	"go.unistack.org/micro/v3/codec" | 	"go.unistack.org/micro/v4/codec" | ||||||
| 	"go.unistack.org/micro/v3/logger" | 	"go.unistack.org/micro/v4/metadata" | ||||||
| 	"go.unistack.org/micro/v3/metadata" | 	"go.unistack.org/micro/v4/store" | ||||||
| 	"go.unistack.org/micro/v3/store" | 	"go.unistack.org/micro/v4/util/id" | ||||||
| 	"go.unistack.org/micro/v3/util/id" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type microFlow struct { | type microFlow struct { | ||||||
| @@ -20,7 +22,7 @@ type microFlow struct { | |||||||
|  |  | ||||||
| type microWorkflow struct { | type microWorkflow struct { | ||||||
| 	opts   Options | 	opts   Options | ||||||
| 	g      *dag.AcyclicGraph | 	g      *dag.DAG | ||||||
| 	steps  map[string]Step | 	steps  map[string]Step | ||||||
| 	id     string | 	id     string | ||||||
| 	status Status | 	status Status | ||||||
| @@ -32,20 +34,20 @@ func (w *microWorkflow) ID() string { | |||||||
| 	return w.id | 	return w.id | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *microWorkflow) Steps() ([][]Step, error) { |  | ||||||
| 	return w.getSteps("", false) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *microWorkflow) Status() Status { | func (w *microWorkflow) Status() Status { | ||||||
| 	return w.status | 	return w.status | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *microWorkflow) AppendSteps(steps ...Step) error { | func (w *microWorkflow) AppendSteps(steps ...Step) error { | ||||||
|  | 	var err error | ||||||
| 	w.Lock() | 	w.Lock() | ||||||
|  | 	defer w.Unlock() | ||||||
|  |  | ||||||
| 	for _, s := range steps { | 	for _, s := range steps { | ||||||
| 		w.steps[s.String()] = s | 		w.steps[s.String()] = s | ||||||
| 		w.g.Add(s) | 		if _, err = w.g.AddVertex(s); err != nil { | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	for _, dst := range steps { | 	for _, dst := range steps { | ||||||
| @@ -54,18 +56,13 @@ func (w *microWorkflow) AppendSteps(steps ...Step) error { | |||||||
| 			if !ok { | 			if !ok { | ||||||
| 				return ErrStepNotExists | 				return ErrStepNotExists | ||||||
| 			} | 			} | ||||||
| 			w.g.Connect(dag.BasicEdge(src, dst)) | 			if err = w.g.AddEdge(src.String(), dst.String()); err != nil { | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if err := w.g.Validate(); err != nil { |  | ||||||
| 		w.Unlock() |  | ||||||
| 				return err | 				return err | ||||||
| 			} | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	w.g.TransitiveReduction() | 	w.g.ReduceTransitively() | ||||||
|  |  | ||||||
| 	w.Unlock() |  | ||||||
|  |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
| @@ -74,10 +71,11 @@ func (w *microWorkflow) RemoveSteps(steps ...Step) error { | |||||||
| 	// TODO: handle case when some step requires or required by removed step | 	// TODO: handle case when some step requires or required by removed step | ||||||
|  |  | ||||||
| 	w.Lock() | 	w.Lock() | ||||||
|  | 	defer w.Unlock() | ||||||
|  |  | ||||||
| 	for _, s := range steps { | 	for _, s := range steps { | ||||||
| 		delete(w.steps, s.String()) | 		delete(w.steps, s.String()) | ||||||
| 		w.g.Remove(s) | 		w.g.DeleteVertex(s.String()) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	for _, dst := range steps { | 	for _, dst := range steps { | ||||||
| @@ -86,91 +84,34 @@ func (w *microWorkflow) RemoveSteps(steps ...Step) error { | |||||||
| 			if !ok { | 			if !ok { | ||||||
| 				return ErrStepNotExists | 				return ErrStepNotExists | ||||||
| 			} | 			} | ||||||
| 			w.g.Connect(dag.BasicEdge(src, dst)) | 			w.g.AddEdge(src.String(), dst.String()) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if err := w.g.Validate(); err != nil { | 	w.g.ReduceTransitively() | ||||||
| 		w.Unlock() |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	w.g.TransitiveReduction() |  | ||||||
|  |  | ||||||
| 	w.Unlock() |  | ||||||
|  |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *microWorkflow) getSteps(start string, reverse bool) ([][]Step, error) { |  | ||||||
| 	var steps [][]Step |  | ||||||
| 	var root dag.Vertex |  | ||||||
| 	var err error |  | ||||||
|  |  | ||||||
| 	fn := func(n dag.Vertex, idx int) error { |  | ||||||
| 		if idx == 0 { |  | ||||||
| 			steps = make([][]Step, 1) |  | ||||||
| 			steps[0] = make([]Step, 0, 1) |  | ||||||
| 		} else if idx >= len(steps) { |  | ||||||
| 			tsteps := make([][]Step, idx+1) |  | ||||||
| 			copy(tsteps, steps) |  | ||||||
| 			steps = tsteps |  | ||||||
| 			steps[idx] = make([]Step, 0, 1) |  | ||||||
| 		} |  | ||||||
| 		steps[idx] = append(steps[idx], n.(Step)) |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if start != "" { |  | ||||||
| 		var ok bool |  | ||||||
| 		w.RLock() |  | ||||||
| 		root, ok = w.steps[start] |  | ||||||
| 		w.RUnlock() |  | ||||||
| 		if !ok { |  | ||||||
| 			return nil, ErrStepNotExists |  | ||||||
| 		} |  | ||||||
| 	} else { |  | ||||||
| 		root, err = w.g.Root() |  | ||||||
| 		if err != nil { |  | ||||||
| 			return nil, err |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if reverse { |  | ||||||
| 		err = w.g.SortedReverseDepthFirstWalk([]dag.Vertex{root}, fn) |  | ||||||
| 	} else { |  | ||||||
| 		err = w.g.SortedDepthFirstWalk([]dag.Vertex{root}, fn) |  | ||||||
| 	} |  | ||||||
| 	if err != nil { |  | ||||||
| 		return nil, err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return steps, nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *microWorkflow) Abort(ctx context.Context, id string) error { | func (w *microWorkflow) Abort(ctx context.Context, id string) error { | ||||||
| 	workflowStore := store.NewNamespaceStore(w.opts.Store, "workflows"+w.opts.Store.Options().Separator+id) | 	workflowStore := store.NewNamespaceStore(w.opts.Store, filepath.Join("workflows", id)) | ||||||
| 	return workflowStore.Write(ctx, "status", &codec.Frame{Data: []byte(StatusAborted.String())}) | 	return workflowStore.Write(ctx, "status", &codec.Frame{Data: []byte(StatusAborted.String())}) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *microWorkflow) Suspend(ctx context.Context, id string) error { | func (w *microWorkflow) Suspend(ctx context.Context, id string) error { | ||||||
| 	workflowStore := store.NewNamespaceStore(w.opts.Store, "workflows"+w.opts.Store.Options().Separator+id) | 	workflowStore := store.NewNamespaceStore(w.opts.Store, filepath.Join("workflows", id)) | ||||||
| 	return workflowStore.Write(ctx, "status", &codec.Frame{Data: []byte(StatusSuspend.String())}) | 	return workflowStore.Write(ctx, "status", &codec.Frame{Data: []byte(StatusSuspend.String())}) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *microWorkflow) Resume(ctx context.Context, id string) error { | func (w *microWorkflow) Resume(ctx context.Context, id string) error { | ||||||
| 	workflowStore := store.NewNamespaceStore(w.opts.Store, "workflows"+w.opts.Store.Options().Separator+id) | 	workflowStore := store.NewNamespaceStore(w.opts.Store, filepath.Join("workflows", id)) | ||||||
| 	return workflowStore.Write(ctx, "status", &codec.Frame{Data: []byte(StatusRunning.String())}) | 	return workflowStore.Write(ctx, "status", &codec.Frame{Data: []byte(StatusRunning.String())}) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...ExecuteOption) (string, error) { | func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...ExecuteOption) (string, error) { | ||||||
| 	w.Lock() | 	w.Lock() | ||||||
| 	if !w.init { | 	if !w.init { | ||||||
| 		if err := w.g.Validate(); err != nil { | 		w.g.ReduceTransitively() | ||||||
| 			w.Unlock() |  | ||||||
| 			return "", err |  | ||||||
| 		} |  | ||||||
| 		w.g.TransitiveReduction() |  | ||||||
| 		w.init = true | 		w.init = true | ||||||
| 	} | 	} | ||||||
| 	w.Unlock() | 	w.Unlock() | ||||||
| @@ -180,26 +121,11 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...Execu | |||||||
| 		return "", err | 		return "", err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	stepStore := store.NewNamespaceStore(w.opts.Store, "steps"+w.opts.Store.Options().Separator+eid) | 	//	stepStore := store.NewNamespaceStore(w.opts.Store, filepath.Join("steps", eid)) | ||||||
| 	workflowStore := store.NewNamespaceStore(w.opts.Store, "workflows"+w.opts.Store.Options().Separator+eid) | 	workflowStore := store.NewNamespaceStore(w.opts.Store, filepath.Join("workflows", eid)) | ||||||
|  |  | ||||||
| 	options := NewExecuteOptions(opts...) | 	options := NewExecuteOptions(opts...) | ||||||
|  |  | ||||||
| 	steps, err := w.getSteps(options.Start, options.Reverse) |  | ||||||
| 	if err != nil { |  | ||||||
| 		if werr := workflowStore.Write(w.opts.Context, "status", &codec.Frame{Data: []byte(StatusPending.String())}); werr != nil { |  | ||||||
| 			w.opts.Logger.Error(w.opts.Context, "store write error", werr) |  | ||||||
| 		} |  | ||||||
| 		return "", err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	var wg sync.WaitGroup |  | ||||||
| 	cherr := make(chan error, 1) |  | ||||||
| 	chstatus := make(chan Status, 1) |  | ||||||
|  |  | ||||||
| 	nctx, cancel := context.WithCancel(ctx) |  | ||||||
| 	defer cancel() |  | ||||||
|  |  | ||||||
| 	nopts := make([]ExecuteOption, 0, len(opts)+5) | 	nopts := make([]ExecuteOption, 0, len(opts)+5) | ||||||
|  |  | ||||||
| 	nopts = append(nopts, | 	nopts = append(nopts, | ||||||
| @@ -209,21 +135,152 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...Execu | |||||||
| 		ExecuteMeter(w.opts.Meter), | 		ExecuteMeter(w.opts.Meter), | ||||||
| 	) | 	) | ||||||
| 	nopts = append(nopts, opts...) | 	nopts = append(nopts, opts...) | ||||||
| 	done := make(chan struct{}) |  | ||||||
|  |  | ||||||
| 	if werr := workflowStore.Write(w.opts.Context, "status", &codec.Frame{Data: []byte(StatusRunning.String())}); werr != nil { | 	if werr := workflowStore.Write(ctx, "status", &codec.Frame{Data: []byte(StatusRunning.String())}); werr != nil { | ||||||
| 		w.opts.Logger.Error(w.opts.Context, "store write error", werr) | 		w.opts.Logger.Error(ctx, "store error: %v", werr) | ||||||
| 		return eid, werr | 		return eid, werr | ||||||
| 	} | 	} | ||||||
| 	for idx := range steps { |  | ||||||
| 		for nidx := range steps[idx] { | 	var startID string | ||||||
| 			cstep := steps[idx][nidx] | 	if options.Start == "" { | ||||||
| 			if werr := stepStore.Write(ctx, cstep.ID()+w.opts.Store.Options().Separator+"status", &codec.Frame{Data: []byte(StatusPending.String())}); werr != nil { | 		mp := w.g.GetRoots() | ||||||
| 				return eid, werr | 		if len(mp) != 1 { | ||||||
|  | 			return eid, ErrStepNotExists | ||||||
|  | 		} | ||||||
|  | 		for k := range mp { | ||||||
|  | 			startID = k | ||||||
|  | 		} | ||||||
|  | 	} else { | ||||||
|  | 		for k, v := range w.g.GetVertices() { | ||||||
|  | 			if v == options.Start { | ||||||
|  | 				startID = k | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	if startID == "" { | ||||||
|  | 		return eid, ErrStepNotExists | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if options.Async { | ||||||
|  | 		go w.handleWorkflow(startID, nopts...) | ||||||
|  | 		return eid, nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return eid, w.handleWorkflow(startID, nopts...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *microWorkflow) handleWorkflow(startID string, opts ...ExecuteOption) error { | ||||||
|  | 	w.RLock() | ||||||
|  | 	defer w.RUnlock() | ||||||
|  |  | ||||||
|  | 	//	stepStore := store.NewNamespaceStore(w.opts.Store, filepath.Join("steps", eid)) | ||||||
|  | 	// workflowStore := store.NewNamespaceStore(w.opts.Store, filepath.Join("workflows", eid)) | ||||||
|  |  | ||||||
|  | 	// Get IDs of all descendant vertices. | ||||||
|  | 	flowIDs, errDes := w.g.GetDescendants(startID) | ||||||
|  | 	if errDes != nil { | ||||||
|  | 		return errDes | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// inputChannels provides for input channels for each of the descendant vertices (+ the start-vertex). | ||||||
|  | 	inputChannels := make(map[string]chan FlowResult, len(flowIDs)+1) | ||||||
|  |  | ||||||
|  | 	// Iterate vertex IDs and create an input channel for each of them and a single | ||||||
|  | 	// output channel for leaves. Note, this "pre-flight" is needed to ensure we | ||||||
|  | 	// really have an input channel regardless of how we traverse the tree and spawn | ||||||
|  | 	// workers. | ||||||
|  | 	leafCount := 0 | ||||||
|  |  | ||||||
|  | 	for id := range flowIDs { | ||||||
|  |  | ||||||
|  | 		// Get all parents of this vertex. | ||||||
|  | 		parents, errPar := w.g.GetParents(id) | ||||||
|  | 		if errPar != nil { | ||||||
|  | 			return errPar | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// Create a buffered input channel that has capacity for all parent results. | ||||||
|  | 		inputChannels[id] = make(chan FlowResult, len(parents)) | ||||||
|  |  | ||||||
|  | 		if ok, err := w.g.IsLeaf(id); ok && err == nil { | ||||||
|  | 			leafCount += 1 | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// outputChannel caries the results of leaf vertices. | ||||||
|  | 	outputChannel := make(chan FlowResult, leafCount) | ||||||
|  |  | ||||||
|  | 	// To also process the start vertex and to have its results being passed to its | ||||||
|  | 	// children, add it to the vertex IDs. Also add an input channel for the start | ||||||
|  | 	// vertex and feed the inputs to this channel. | ||||||
|  | 	flowIDs[startID] = struct{}{} | ||||||
|  | 	inputChannels[startID] = make(chan FlowResult, len(inputs)) | ||||||
|  | 	for _, i := range inputs { | ||||||
|  | 		inputChannels[startID] <- i | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	wg := sync.WaitGroup{} | ||||||
|  |  | ||||||
|  | 	// Iterate all vertex IDs (now incl. start vertex) and handle each worker (incl. | ||||||
|  | 	// inputs and outputs) in a separate goroutine. | ||||||
|  | 	for id := range flowIDs { | ||||||
|  |  | ||||||
|  | 		// Get all children of this vertex that later need to be notified. Note, we | ||||||
|  | 		// collect all children before the goroutine to be able to release the read | ||||||
|  | 		// lock as early as possible. | ||||||
|  | 		children, errChildren := w.g.GetChildren(id) | ||||||
|  | 		if errChildren != nil { | ||||||
|  | 			return errChildren | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// Remember to wait for this goroutine. | ||||||
|  | 		wg.Add(1) | ||||||
|  |  | ||||||
|  | 		go func(id string) { | ||||||
|  | 			// Get this vertex's input channel. | ||||||
|  | 			// Note, only concurrent read here, which is fine. | ||||||
|  | 			c := inputChannels[id] | ||||||
|  |  | ||||||
|  | 			// Await all parent inputs and stuff them into a slice. | ||||||
|  | 			parentCount := cap(c) | ||||||
|  | 			parentResults := make([]FlowResult, parentCount) | ||||||
|  | 			for i := 0; i < parentCount; i++ { | ||||||
|  | 				parentResults[i] = <-c | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			// Execute the worker. | ||||||
|  | 			errWorker := callback(w.g, id, parentResults) | ||||||
|  | 			if errWorker != nil { | ||||||
|  | 				return errWorker | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			// Send this worker's FlowResult onto all children's input channels or, if it is | ||||||
|  | 			// a leaf (i.e. no children), send the result onto the output channel. | ||||||
|  | 			if len(children) > 0 { | ||||||
|  | 				for child := range children { | ||||||
|  | 					inputChannels[child] <- flowResult | ||||||
|  | 				} | ||||||
|  | 			} else { | ||||||
|  | 				outputChannel <- flowResult | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			// "Sign off". | ||||||
|  | 			wg.Done() | ||||||
|  | 		}(id) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// Wait for all go routines to finish. | ||||||
|  | 	wg.Wait() | ||||||
|  |  | ||||||
|  | 	// Await all leaf vertex results and stuff them into a slice. | ||||||
|  | 	resultCount := cap(outputChannel) | ||||||
|  | 	results := make([]FlowResult, resultCount) | ||||||
|  | 	for i := 0; i < resultCount; i++ { | ||||||
|  | 		results[i] = <-outputChannel | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	/* | ||||||
| 		go func() { | 		go func() { | ||||||
| 			for idx := range steps { | 			for idx := range steps { | ||||||
| 				for nidx := range steps[idx] { | 				for nidx := range steps[idx] { | ||||||
| @@ -237,7 +294,7 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...Execu | |||||||
| 						return | 						return | ||||||
| 					} | 					} | ||||||
| 					if w.opts.Logger.V(logger.TraceLevel) { | 					if w.opts.Logger.V(logger.TraceLevel) { | ||||||
| 					w.opts.Logger.Trace(nctx, fmt.Sprintf("will be executed %v", steps[idx][nidx])) | 						w.opts.Logger.Tracef(nctx, "will be executed %v", steps[idx][nidx]) | ||||||
| 					} | 					} | ||||||
| 					cstep := steps[idx][nidx] | 					cstep := steps[idx][nidx] | ||||||
| 					// nolint: nestif | 					// nolint: nestif | ||||||
| @@ -245,65 +302,65 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...Execu | |||||||
| 						wg.Add(1) | 						wg.Add(1) | ||||||
| 						go func(step Step) { | 						go func(step Step) { | ||||||
| 							defer wg.Done() | 							defer wg.Done() | ||||||
| 						if werr := stepStore.Write(ctx, step.ID()+w.opts.Store.Options().Separator+"req", req); werr != nil { | 							if werr := stepStore.Write(ctx, filepath.Join(step.ID(), "req"), req); werr != nil { | ||||||
| 								cherr <- werr | 								cherr <- werr | ||||||
| 								return | 								return | ||||||
| 							} | 							} | ||||||
| 						if werr := stepStore.Write(ctx, step.ID()+w.opts.Store.Options().Separator+"status", &codec.Frame{Data: []byte(StatusRunning.String())}); werr != nil { | 							if werr := stepStore.Write(ctx, filepath.Join(step.ID(), "status"), &codec.Frame{Data: []byte(StatusRunning.String())}); werr != nil { | ||||||
| 								cherr <- werr | 								cherr <- werr | ||||||
| 								return | 								return | ||||||
| 							} | 							} | ||||||
| 							rsp, serr := step.Execute(nctx, req, nopts...) | 							rsp, serr := step.Execute(nctx, req, nopts...) | ||||||
| 							if serr != nil { | 							if serr != nil { | ||||||
| 								step.SetStatus(StatusFailure) | 								step.SetStatus(StatusFailure) | ||||||
| 							if werr := stepStore.Write(ctx, step.ID()+w.opts.Store.Options().Separator+"rsp", serr); werr != nil && w.opts.Logger.V(logger.ErrorLevel) { | 								if werr := stepStore.Write(ctx, filepath.Join(step.ID(), "rsp"), serr); werr != nil && w.opts.Logger.V(logger.ErrorLevel) { | ||||||
| 								w.opts.Logger.Error(ctx, "store write error", werr) | 									w.opts.Logger.Errorf(ctx, "store write error: %v", 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, filepath.Join(step.ID(), "status"), &codec.Frame{Data: []byte(StatusFailure.String())}); werr != nil && w.opts.Logger.V(logger.ErrorLevel) { | ||||||
| 								w.opts.Logger.Error(ctx, "store write error", werr) | 									w.opts.Logger.Errorf(ctx, "store write error: %v", 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, filepath.Join(step.ID(), "rsp"), rsp); werr != nil { | ||||||
| 							w.opts.Logger.Error(ctx, "store write error", werr) | 								w.opts.Logger.Errorf(ctx, "store write error: %v", 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, filepath.Join(step.ID(), "status"), &codec.Frame{Data: []byte(StatusSuccess.String())}); werr != nil { | ||||||
| 							w.opts.Logger.Error(ctx, "store write error", werr) | 								w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | ||||||
| 								cherr <- werr | 								cherr <- werr | ||||||
| 								return | 								return | ||||||
| 							} | 							} | ||||||
| 						}(cstep) | 						}(cstep) | ||||||
| 						wg.Wait() | 						wg.Wait() | ||||||
| 					} else { | 					} else { | ||||||
| 					if werr := stepStore.Write(ctx, cstep.ID()+w.opts.Store.Options().Separator+"req", req); werr != nil { | 						if werr := stepStore.Write(ctx, filepath.Join(cstep.ID(), "req"), req); werr != nil { | ||||||
| 							cherr <- werr | 							cherr <- werr | ||||||
| 							return | 							return | ||||||
| 						} | 						} | ||||||
| 					if werr := stepStore.Write(ctx, cstep.ID()+w.opts.Store.Options().Separator+"status", &codec.Frame{Data: []byte(StatusRunning.String())}); werr != nil { | 						if werr := stepStore.Write(ctx, filepath.Join(cstep.ID(), "status"), &codec.Frame{Data: []byte(StatusRunning.String())}); werr != nil { | ||||||
| 							cherr <- werr | 							cherr <- werr | ||||||
| 							return | 							return | ||||||
| 						} | 						} | ||||||
| 						rsp, serr := cstep.Execute(nctx, req, nopts...) | 						rsp, serr := cstep.Execute(nctx, req, nopts...) | ||||||
| 						if serr != nil { | 						if serr != nil { | ||||||
| 							cstep.SetStatus(StatusFailure) | 							cstep.SetStatus(StatusFailure) | ||||||
| 						if werr := stepStore.Write(ctx, cstep.ID()+w.opts.Store.Options().Separator+"rsp", serr); werr != nil && w.opts.Logger.V(logger.ErrorLevel) { | 							if werr := stepStore.Write(ctx, filepath.Join(cstep.ID(), "rsp"), serr); werr != nil && w.opts.Logger.V(logger.ErrorLevel) { | ||||||
| 							w.opts.Logger.Error(ctx, "store write error", werr) | 								w.opts.Logger.Errorf(ctx, "store write error: %v", 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, filepath.Join(cstep.ID(), "status"), &codec.Frame{Data: []byte(StatusFailure.String())}); werr != nil && w.opts.Logger.V(logger.ErrorLevel) { | ||||||
| 							w.opts.Logger.Error(ctx, "store write error", werr) | 								w.opts.Logger.Errorf(ctx, "store write error: %v", 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, filepath.Join(cstep.ID(), "rsp"), rsp); werr != nil { | ||||||
| 						w.opts.Logger.Error(ctx, "store write error", werr) | 							w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | ||||||
| 							cherr <- werr | 							cherr <- werr | ||||||
| 							return | 							return | ||||||
| 						} | 						} | ||||||
| 					if werr := stepStore.Write(ctx, cstep.ID()+w.opts.Store.Options().Separator+"status", &codec.Frame{Data: []byte(StatusSuccess.String())}); werr != nil { | 						if werr := stepStore.Write(ctx, filepath.Join(cstep.ID(), "status"), &codec.Frame{Data: []byte(StatusSuccess.String())}); werr != nil { | ||||||
| 							cherr <- werr | 							cherr <- werr | ||||||
| 							return | 							return | ||||||
| 						} | 						} | ||||||
| @@ -317,7 +374,7 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...Execu | |||||||
| 			return eid, nil | 			return eid, nil | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 	logger.DefaultLogger.Trace(ctx, "wait for finish or error") | 		logger.Tracef(ctx, "wait for finish or error") | ||||||
| 		select { | 		select { | ||||||
| 		case <-nctx.Done(): | 		case <-nctx.Done(): | ||||||
| 			err = nctx.Err() | 			err = nctx.Err() | ||||||
| @@ -333,19 +390,19 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...Execu | |||||||
| 		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.Error(w.opts.Context, "store write error", werr) | 				w.opts.Logger.Errorf(w.opts.Context, "store error: %v", 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.Error(w.opts.Context, "store write error", werr) | 				w.opts.Logger.Errorf(w.opts.Context, "store error: %v", 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.Error(w.opts.Context, "store write error", werr) | 				w.opts.Logger.Errorf(w.opts.Context, "store error: %v", werr) | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
|  | 	*/ | ||||||
| 	return eid, err | 	return err | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewFlow create new flow | // NewFlow create new flow | ||||||
| @@ -385,11 +442,11 @@ func (f *microFlow) WorkflowList(ctx context.Context) ([]Workflow, error) { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (f *microFlow) WorkflowCreate(ctx context.Context, id string, steps ...Step) (Workflow, error) { | func (f *microFlow) WorkflowCreate(ctx context.Context, id string, steps ...Step) (Workflow, error) { | ||||||
| 	w := µWorkflow{opts: f.opts, id: id, g: &dag.AcyclicGraph{}, steps: make(map[string]Step, len(steps))} | 	w := µWorkflow{opts: f.opts, id: id, g: &dag.DAG{}, steps: make(map[string]Step, len(steps))} | ||||||
|  |  | ||||||
| 	for _, s := range steps { | 	for _, s := range steps { | ||||||
| 		w.steps[s.String()] = s | 		w.steps[s.String()] = s | ||||||
| 		w.g.Add(s) | 		w.g.AddVertex(s) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	for _, dst := range steps { | 	for _, dst := range steps { | ||||||
| @@ -398,14 +455,11 @@ func (f *microFlow) WorkflowCreate(ctx context.Context, id string, steps ...Step | |||||||
| 			if !ok { | 			if !ok { | ||||||
| 				return nil, ErrStepNotExists | 				return nil, ErrStepNotExists | ||||||
| 			} | 			} | ||||||
| 			w.g.Connect(dag.BasicEdge(src, dst)) | 			w.g.AddEdge(src.String(), dst.String()) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if err := w.g.Validate(); err != nil { | 	w.g.ReduceTransitively() | ||||||
| 		return nil, err |  | ||||||
| 	} |  | ||||||
| 	w.g.TransitiveReduction() |  | ||||||
|  |  | ||||||
| 	w.init = true | 	w.init = true | ||||||
|  |  | ||||||
|   | |||||||
| @@ -1,5 +1,5 @@ | |||||||
| // Package flow is an interface used for saga pattern microservice workflow | // Package flow is an interface used for saga pattern microservice workflow | ||||||
| package flow | package flow // import "go.unistack.org/micro/v4/flow" | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| @@ -7,7 +7,7 @@ import ( | |||||||
| 	"sync" | 	"sync" | ||||||
| 	"sync/atomic" | 	"sync/atomic" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/metadata" | 	"go.unistack.org/micro/v4/metadata" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| @@ -125,8 +125,6 @@ type Workflow interface { | |||||||
| 	AppendSteps(steps ...Step) error | 	AppendSteps(steps ...Step) error | ||||||
| 	// Status returns workflow status | 	// Status returns workflow status | ||||||
| 	Status() Status | 	Status() Status | ||||||
| 	// Steps returns steps slice where parallel steps returned on the same level |  | ||||||
| 	Steps() ([][]Step, error) |  | ||||||
| 	// Suspend suspends execution | 	// Suspend suspends execution | ||||||
| 	Suspend(ctx context.Context, id string) error | 	Suspend(ctx context.Context, id string) error | ||||||
| 	// Resume resumes execution | 	// Resume resumes execution | ||||||
|   | |||||||
| @@ -4,11 +4,11 @@ import ( | |||||||
| 	"context" | 	"context" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/client" | 	"go.unistack.org/micro/v4/client" | ||||||
| 	"go.unistack.org/micro/v3/logger" | 	"go.unistack.org/micro/v4/logger" | ||||||
| 	"go.unistack.org/micro/v3/meter" | 	"go.unistack.org/micro/v4/meter" | ||||||
| 	"go.unistack.org/micro/v3/store" | 	"go.unistack.org/micro/v4/store" | ||||||
| 	"go.unistack.org/micro/v3/tracer" | 	"go.unistack.org/micro/v4/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Option func | // Option func | ||||||
| @@ -123,8 +123,6 @@ type ExecuteOptions struct { | |||||||
| 	Start string | 	Start string | ||||||
| 	// Timeout for execution | 	// Timeout for execution | ||||||
| 	Timeout time.Duration | 	Timeout time.Duration | ||||||
| 	// Reverse execution |  | ||||||
| 	Reverse bool |  | ||||||
| 	// Async enables async execution | 	// Async enables async execution | ||||||
| 	Async bool | 	Async bool | ||||||
| } | } | ||||||
| @@ -167,13 +165,6 @@ func ExecuteContext(ctx context.Context) ExecuteOption { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // ExecuteReverse says that dag must be run in reverse order |  | ||||||
| func ExecuteReverse(b bool) ExecuteOption { |  | ||||||
| 	return func(o *ExecuteOptions) { |  | ||||||
| 		o.Reverse = b |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // ExecuteTimeout pass timeout time.Duration for execution | // ExecuteTimeout pass timeout time.Duration for execution | ||||||
| func ExecuteTimeout(td time.Duration) ExecuteOption { | func ExecuteTimeout(td time.Duration) ExecuteOption { | ||||||
| 	return func(o *ExecuteOptions) { | 	return func(o *ExecuteOptions) { | ||||||
|   | |||||||
| @@ -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...), | ||||||
|   | |||||||
| @@ -5,7 +5,7 @@ import ( | |||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"testing" | 	"testing" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/logger" | 	"go.unistack.org/micro/v4/logger" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| func TestFSMStart(t *testing.T) { | func TestFSMStart(t *testing.T) { | ||||||
|   | |||||||
							
								
								
									
										44
									
								
								go.mod
									
									
									
									
									
								
							
							
						
						
									
										44
									
								
								go.mod
									
									
									
									
									
								
							| @@ -1,32 +1,34 @@ | |||||||
| module go.unistack.org/micro/v3 | module go.unistack.org/micro/v4 | ||||||
|  |  | ||||||
| go 1.22 | go 1.22.0 | ||||||
|  |  | ||||||
| require ( | require ( | ||||||
| 	dario.cat/mergo v1.0.0 | 	dario.cat/mergo v1.0.1 | ||||||
| 	github.com/DATA-DOG/go-sqlmock v1.5.0 | 	github.com/DATA-DOG/go-sqlmock v1.5.2 | ||||||
| 	github.com/KimMachineGun/automemlimit v0.6.1 | 	github.com/KimMachineGun/automemlimit v0.7.0 | ||||||
| 	github.com/google/uuid v1.3.0 | 	github.com/ash3in/uuidv8 v1.2.0 | ||||||
|  | 	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 | ||||||
|  | 	github.com/spf13/cast v1.7.1 | ||||||
|  | 	go.uber.org/atomic v1.11.0 | ||||||
| 	go.uber.org/automaxprocs v1.6.0 | 	go.uber.org/automaxprocs v1.6.0 | ||||||
| 	go.unistack.org/micro-proto/v3 v3.4.1 | 	go.unistack.org/micro-proto/v4 v4.1.0 | ||||||
| 	golang.org/x/sync v0.3.0 | 	golang.org/x/sync v0.10.0 | ||||||
| 	google.golang.org/grpc v1.57.0 | 	google.golang.org/grpc v1.69.4 | ||||||
| 	google.golang.org/protobuf v1.33.0 | 	google.golang.org/protobuf v1.36.3 | ||||||
|  | 	gopkg.in/yaml.v3 v3.0.1 | ||||||
| ) | ) | ||||||
|  |  | ||||||
| require ( | require ( | ||||||
| 	github.com/cilium/ebpf v0.9.1 // indirect | 	github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect | ||||||
| 	github.com/containerd/cgroups/v3 v3.0.1 // indirect |  | ||||||
| 	github.com/coreos/go-systemd/v22 v22.3.2 // indirect |  | ||||||
| 	github.com/docker/go-units v0.4.0 // indirect |  | ||||||
| 	github.com/godbus/dbus/v5 v5.0.4 // indirect |  | ||||||
| 	github.com/golang/protobuf v1.5.3 // indirect |  | ||||||
| 	github.com/opencontainers/runtime-spec v1.0.2 // indirect |  | ||||||
| 	github.com/pbnjay/memory v0.0.0-20210728143218-7b4eea64cf58 // indirect | 	github.com/pbnjay/memory v0.0.0-20210728143218-7b4eea64cf58 // indirect | ||||||
| 	github.com/sirupsen/logrus v1.8.1 // indirect | 	github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 // indirect | ||||||
| 	golang.org/x/net v0.14.0 // indirect | 	github.com/rogpeppe/go-internal v1.13.1 // indirect | ||||||
| 	golang.org/x/sys v0.11.0 // indirect | 	github.com/stretchr/testify v1.10.0 // indirect | ||||||
| 	google.golang.org/genproto/googleapis/rpc v0.0.0-20230526203410-71b5a4ffd15e // indirect | 	golang.org/x/net v0.34.0 // indirect | ||||||
|  | 	golang.org/x/sys v0.29.0 // indirect | ||||||
|  | 	google.golang.org/genproto/googleapis/rpc v0.0.0-20241216192217-9240e9c98484 // indirect | ||||||
|  | 	gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c // indirect | ||||||
| ) | ) | ||||||
|   | |||||||
							
								
								
									
										117
									
								
								go.sum
									
									
									
									
									
								
							
							
						
						
									
										117
									
								
								go.sum
									
									
									
									
									
								
							| @@ -1,78 +1,69 @@ | |||||||
| dario.cat/mergo v1.0.0 h1:AGCNq9Evsj31mOgNPcLyXc+4PNABt905YmuqPYYpBWk= | dario.cat/mergo v1.0.1 h1:Ra4+bf83h2ztPIQYNP99R6m+Y7KfnARDfID+a+vLl4s= | ||||||
| dario.cat/mergo v1.0.0/go.mod h1:uNxQE+84aUszobStD9th8a29P2fMDhsBdgRYvZOxGmk= | 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.2 h1:OcvFkGmslmlZibjAjaHm3L//6LiuBgolP7OputlJIzU= | ||||||
| github.com/DATA-DOG/go-sqlmock v1.5.0/go.mod h1:f/Ixk793poVmq4qj/V1dPUg2JEAKC73Q5eFN3EC/SaM= | github.com/DATA-DOG/go-sqlmock v1.5.2/go.mod h1:88MAG/4G7SMwSE3CeA0ZKzrT5CiOU3OJ+JlNzwDqpNU= | ||||||
| github.com/KimMachineGun/automemlimit v0.6.1 h1:ILa9j1onAAMadBsyyUJv5cack8Y1WT26yLj/V+ulKp8= | github.com/KimMachineGun/automemlimit v0.7.0 h1:7G06p/dMSf7G8E6oq+f2uOPuVncFyIlDI/pBWK49u88= | ||||||
| github.com/KimMachineGun/automemlimit v0.6.1/go.mod h1:T7xYht7B8r6AG/AqFcUdc7fzd2bIdBKmepfP2S1svPY= | github.com/KimMachineGun/automemlimit v0.7.0/go.mod h1:QZxpHaGOQoYvFhv/r4u3U0JTC2ZcOwbSr11UZF46UBM= | ||||||
| github.com/cilium/ebpf v0.9.1 h1:64sn2K3UKw8NbP/blsixRpF3nXuyhz/VjRlRzvlBRu4= | github.com/ash3in/uuidv8 v1.2.0 h1:2oogGdtCPwaVtyvPPGin4TfZLtOGE5F+W++E880G6SI= | ||||||
| github.com/cilium/ebpf v0.9.1/go.mod h1:+OhNOIXx/Fnu1IE8bJz2dzOA+VSfyTfdNUVdlQnxUFY= | github.com/ash3in/uuidv8 v1.2.0/go.mod h1:BnU0wJBxnzdEKmVg4xckBkD+VZuecTFTUP3M0dWgyY4= | ||||||
| github.com/containerd/cgroups/v3 v3.0.1 h1:4hfGvu8rfGIwVIDd+nLzn/B9ZXx4BcCjzt5ToenJRaE= | github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1VwoXQT9A3Wy9MM3WgvqSxFWenqJduM= | ||||||
| github.com/containerd/cgroups/v3 v3.0.1/go.mod h1:/vtwk1VXrtoa5AaZLkypuOJgA/6DyPMZHJPGQNtlHnw= | github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | ||||||
| github.com/coreos/go-systemd/v22 v22.3.2 h1:D9/bQk5vlXQFZ6Kwuu6zaiXJ9oTPe68++AzAJc1DzSI= | github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8= | ||||||
| github.com/coreos/go-systemd/v22 v22.3.2/go.mod h1:Y58oyj3AT4RCenI/lSvhwexgC+NSVTIJ3seZv2GcEnc= | github.com/frankban/quicktest v1.14.6/go.mod h1:4ptaffx2x8+WTWXmUCuVU6aPUX1/Mz7zb5vbUoiM6w0= | ||||||
| github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= | github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek= | ||||||
| github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | github.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps= | ||||||
| github.com/docker/go-units v0.4.0 h1:3uh0PgVws3nIA0Q+MwDC8yjEPf9zjRfZZWXZYDct3Tw= | github.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI= | ||||||
| github.com/docker/go-units v0.4.0/go.mod h1:fgPhTUdO+D/Jk86RDLlptpiXQzgHJF7gydDDbaIK4Dk= | github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY= | ||||||
| github.com/frankban/quicktest v1.14.0 h1:+cqqvzZV87b4adx/5ayVOaYZ2CrvM4ejQvUdBzPPUss= | github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0= | ||||||
| github.com/frankban/quicktest v1.14.0/go.mod h1:NeW+ay9A/U67EYXNFA1nPE8e/tnQv/09mUdL/ijj8og= | github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= | ||||||
| github.com/godbus/dbus/v5 v5.0.4 h1:9349emZab16e7zQvpmsbtjc18ykshndd8y2PG3sgJbA= | github.com/kisielk/sqlstruct v0.0.0-20201105191214-5f3e10d3ab46/go.mod h1:yyMNCyc/Ib3bDTKd379tNMpB/7/H5TjM2Y9QJ5THLbE= | ||||||
| github.com/godbus/dbus/v5 v5.0.4/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA= | github.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI= | ||||||
| github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk= | github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE= | ||||||
| github.com/golang/protobuf v1.5.3 h1:KhyjKVUg7Usr/dYsdSqoFveMYd5ko72D+zANwlG1mmg= | github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk= | ||||||
| github.com/golang/protobuf v1.5.3/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY= | github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= | ||||||
| github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= | github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= | ||||||
| github.com/google/go-cmp v0.5.9 h1:O2Tfq5qg4qc4AmwVlvv0oLiVAGB7enBSJ2x2DqQFi38= |  | ||||||
| github.com/google/go-cmp v0.5.9/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY= |  | ||||||
| github.com/google/uuid v1.3.0 h1:t6JiXgmwXMjEs8VusXIJk2BXHsn+wx8BZdTaoZ5fu7I= |  | ||||||
| github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= |  | ||||||
| github.com/kr/pretty v0.3.0 h1:WgNl7dwNpEZ6jJ9k1snq4pZsg7DOEN8hP9Xw0Tsjwk0= |  | ||||||
| github.com/kr/pretty v0.3.0/go.mod h1:640gp4NfQd8pI5XOwp5fnNeVWj67G7CFk/SaSQn7NBk= |  | ||||||
| github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY= | github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY= | ||||||
| github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE= | github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE= | ||||||
| github.com/opencontainers/runtime-spec v1.0.2 h1:UfAcuLBJB9Coz72x1hgl8O5RVzTdNiaglX6v2DM6FI0= | github.com/matoous/go-nanoid v1.5.1 h1:aCjdvTyO9LLnTIi0fgdXhOPPvOHjpXN6Ik9DaNjIct4= | ||||||
| github.com/opencontainers/runtime-spec v1.0.2/go.mod h1:jwyrGlmzljRJv/Fgzds9SsS/C5hL+LL3ko9hs6T5lQ0= | github.com/matoous/go-nanoid v1.5.1/go.mod h1:zyD2a71IubI24efhpvkJz+ZwfwagzgSO6UNiFsZKN7U= | ||||||
| github.com/patrickmn/go-cache v2.1.0+incompatible h1:HRMgzkcYKYpi3C8ajMPV8OFXaaRUnok+kx1WdO15EQc= | github.com/patrickmn/go-cache v2.1.0+incompatible h1:HRMgzkcYKYpi3C8ajMPV8OFXaaRUnok+kx1WdO15EQc= | ||||||
| github.com/patrickmn/go-cache v2.1.0+incompatible/go.mod h1:3Qf8kWWT7OJRJbdiICTKqZju1ZixQ/KpMGzzAfe6+WQ= | github.com/patrickmn/go-cache v2.1.0+incompatible/go.mod h1:3Qf8kWWT7OJRJbdiICTKqZju1ZixQ/KpMGzzAfe6+WQ= | ||||||
| github.com/pbnjay/memory v0.0.0-20210728143218-7b4eea64cf58 h1:onHthvaw9LFnH4t2DcNVpwGmV9E1BkGknEliJkfwQj0= | github.com/pbnjay/memory v0.0.0-20210728143218-7b4eea64cf58 h1:onHthvaw9LFnH4t2DcNVpwGmV9E1BkGknEliJkfwQj0= | ||||||
| github.com/pbnjay/memory v0.0.0-20210728143218-7b4eea64cf58/go.mod h1:DXv8WO4yhMYhSNPKjeNKa5WY9YCIEBRbNzFFPJbWO6Y= | github.com/pbnjay/memory v0.0.0-20210728143218-7b4eea64cf58/go.mod h1:DXv8WO4yhMYhSNPKjeNKa5WY9YCIEBRbNzFFPJbWO6Y= | ||||||
| github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= | github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 h1:Jamvg5psRIccs7FGNTlIRMkT8wgtp5eCXdBlqhYGL6U= | ||||||
| github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= | github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= | ||||||
| github.com/prashantv/gostub v1.1.0 h1:BTyx3RfQjRHnUWaGF9oQos79AlQ5k8WNktv7VGvVH4g= | github.com/prashantv/gostub v1.1.0 h1:BTyx3RfQjRHnUWaGF9oQos79AlQ5k8WNktv7VGvVH4g= | ||||||
| github.com/prashantv/gostub v1.1.0/go.mod h1:A5zLQHz7ieHGG7is6LLXLz7I8+3LZzsrV0P1IAHhP5U= | github.com/prashantv/gostub v1.1.0/go.mod h1:A5zLQHz7ieHGG7is6LLXLz7I8+3LZzsrV0P1IAHhP5U= | ||||||
| github.com/rogpeppe/go-internal v1.6.1 h1:/FiVV8dS/e+YqF2JvO3yXRFbBLTIuSDkuC7aBOAvL+k= | github.com/rogpeppe/go-internal v1.13.1 h1:KvO1DLK/DRN07sQ1LQKScxyZJuNnedQ5/wKSR38lUII= | ||||||
| github.com/rogpeppe/go-internal v1.6.1/go.mod h1:xXDCJY+GAPziupqXw64V24skbSoqbTEfhy4qGm1nDQc= | github.com/rogpeppe/go-internal v1.13.1/go.mod h1:uMEvuHeurkdAXX61udpOXGD/AzZDWNMNyH2VO9fmH0o= | ||||||
| github.com/silas/dag v0.0.0-20220518035006-a7e85ada93c5 h1:G/FZtUu7a6NTWl3KUHMV9jkLAh/Rvtf03NWMHaEDl+E= | github.com/silas/dag v0.0.0-20220518035006-a7e85ada93c5 h1:G/FZtUu7a6NTWl3KUHMV9jkLAh/Rvtf03NWMHaEDl+E= | ||||||
| github.com/silas/dag v0.0.0-20220518035006-a7e85ada93c5/go.mod h1:7RTUFBdIRC9nZ7/3RyRNH1bdqIShrDejd1YbLwgPS+I= | github.com/silas/dag v0.0.0-20220518035006-a7e85ada93c5/go.mod h1:7RTUFBdIRC9nZ7/3RyRNH1bdqIShrDejd1YbLwgPS+I= | ||||||
| github.com/sirupsen/logrus v1.8.1 h1:dJKuHgqk1NNQlqoA6BTlM1Wf9DOH3NBjQyu0h9+AZZE= | github.com/spf13/cast v1.7.1 h1:cuNEagBQEHWN1FnbGEjCXL2szYEXqfJPbP2HNUaca9Y= | ||||||
| github.com/sirupsen/logrus v1.8.1/go.mod h1:yWOB1SBYBC5VeMP7gHvWumXLIWorT60ONWic61uBYv0= | github.com/spf13/cast v1.7.1/go.mod h1:ancEpBxwJDODSW/UG4rDrAqiKolqNNh2DX3mk86cAdo= | ||||||
| github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs= | github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA= | ||||||
| github.com/stretchr/testify v1.8.0 h1:pSgiaMZlXftHpm5L7V1+rVB+AZJydKsMxsQBIJw4PKk= | github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY= | ||||||
| github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU= | go.uber.org/atomic v1.11.0 h1:ZvwS0R+56ePWxUNi+Atn9dWONBPp/AUETXlHW0DxSjE= | ||||||
|  | go.uber.org/atomic v1.11.0/go.mod h1:LUxbIzbOniOlMKjJjyPfpl4v+PKK2cNJn91OQbhoJI0= | ||||||
| go.uber.org/automaxprocs v1.6.0 h1:O3y2/QNTOdbF+e/dpXNNW7Rx2hZ4sTIPyybbxyNqTUs= | go.uber.org/automaxprocs v1.6.0 h1:O3y2/QNTOdbF+e/dpXNNW7Rx2hZ4sTIPyybbxyNqTUs= | ||||||
| go.uber.org/automaxprocs v1.6.0/go.mod h1:ifeIMSnPZuznNm6jmdzmU3/bfk01Fe2fotchwEFJ8r8= | go.uber.org/automaxprocs v1.6.0/go.mod h1:ifeIMSnPZuznNm6jmdzmU3/bfk01Fe2fotchwEFJ8r8= | ||||||
| go.uber.org/goleak v1.1.12 h1:gZAh5/EyT/HQwlpkCy6wTpqfH9H8Lz8zbm3dZh+OyzA= | go.unistack.org/micro-proto/v4 v4.1.0 h1:qPwL2n/oqh9RE3RTTDgt28XK3QzV597VugQPaw9lKUk= | ||||||
| go.uber.org/goleak v1.1.12/go.mod h1:cwTWslyiVhfpKIDGSZEM2HlOvcqm+tG4zioyIeLoqMQ= | go.unistack.org/micro-proto/v4 v4.1.0/go.mod h1:ArmK7o+uFvxSY3dbJhKBBX4Pm1rhWdLEFf3LxBrMtec= | ||||||
| go.unistack.org/micro-proto/v3 v3.4.1 h1:UTjLSRz2YZuaHk9iSlVqqsA50JQNAEK2ZFboGqtEa9Q= | golang.org/x/net v0.34.0 h1:Mb7Mrk043xzHgnRM88suvJFwzVrRfHEHJEl5/71CKw0= | ||||||
| go.unistack.org/micro-proto/v3 v3.4.1/go.mod h1:okx/cnOhzuCX0ggl/vToatbCupi0O44diiiLLsZ93Zo= | golang.org/x/net v0.34.0/go.mod h1:di0qlW3YNM5oh6GqDGQr92MyTozJPmybPK4Ev/Gm31k= | ||||||
| golang.org/x/net v0.14.0 h1:BONx9s002vGdD9umnlX1Po8vOZmrgH34qlHcD1MfK14= | golang.org/x/sync v0.10.0 h1:3NQrjDixjgGwUOCaF8w2+VYHv0Ve/vGYSbdkTa98gmQ= | ||||||
| golang.org/x/net v0.14.0/go.mod h1:PpSgVXXLK0OxS0F31C1/tv6XNguvCrnXIDrFMspZIUI= | golang.org/x/sync v0.10.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk= | ||||||
| golang.org/x/sync v0.3.0 h1:ftCYgMx6zT/asHUrPw8BLLscYtGznsLAnjq5RH9P66E= | golang.org/x/sys v0.29.0 h1:TPYlXGxvx1MGTn2GiZDhnjPA9wZzZeGKHHmKhHYvgaU= | ||||||
| golang.org/x/sync v0.3.0/go.mod h1:FU7BRWz2tNW+3quACPkgCx/L+uEAv1htQ0V83Z9Rj+Y= | golang.org/x/sys v0.29.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= | ||||||
| golang.org/x/sys v0.0.0-20191026070338-33540a1f6037/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= | golang.org/x/text v0.21.0 h1:zyQAAkrwaneQ066sspRyJaG9VNi/YJ1NfzcGB3hZ/qo= | ||||||
| golang.org/x/sys v0.11.0 h1:eG7RXZHdqOJ1i+0lgLgCpSXAp6M3LYlAo6osgSi0xOM= | golang.org/x/text v0.21.0/go.mod h1:4IBbMaMmOPCJ8SecivzSH54+73PCFmPWxNTLm+vZkEQ= | ||||||
| golang.org/x/sys v0.11.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | google.golang.org/genproto/googleapis/rpc v0.0.0-20241216192217-9240e9c98484 h1:Z7FRVJPSMaHQxD0uXU8WdgFh8PseLM8Q8NzhnpMrBhQ= | ||||||
| golang.org/x/text v0.12.0 h1:k+n5B8goJNdU7hSvEtMUz3d1Q6D/XW4COJSJR6fN0mc= | google.golang.org/genproto/googleapis/rpc v0.0.0-20241216192217-9240e9c98484/go.mod h1:lcTa1sDdWEIHMWlITnIczmw5w60CF9ffkb8Z+DVmmjA= | ||||||
| golang.org/x/text v0.12.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE= | google.golang.org/grpc v1.69.4 h1:MF5TftSMkd8GLw/m0KM6V8CMOCY6NZ1NQDPGFgbTt4A= | ||||||
| golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= | google.golang.org/grpc v1.69.4/go.mod h1:vyjdE6jLBI76dgpDojsFGNaHlxdjXN9ghpnd2o7JGZ4= | ||||||
| google.golang.org/genproto/googleapis/rpc v0.0.0-20230526203410-71b5a4ffd15e h1:NumxXLPfHSndr3wBBdeKiVHjGVFzi9RX2HwwQke94iY= | google.golang.org/protobuf v1.36.3 h1:82DV7MYdb8anAVi3qge1wSnMDrnKK7ebr+I0hHRN1BU= | ||||||
| google.golang.org/genproto/googleapis/rpc v0.0.0-20230526203410-71b5a4ffd15e/go.mod h1:66JfowdXAEgad5O9NnYcsNPLCPZJD++2L9X0PCMODrA= | google.golang.org/protobuf v1.36.3/go.mod h1:9fA7Ob0pmnwhb644+1+CVWFRbNajQ6iRojtC/QF5bRE= | ||||||
| google.golang.org/grpc v1.57.0 h1:kfzNeI/klCGD2YPMUlaGNT3pxvYfga7smW3Vth8Zsiw= |  | ||||||
| google.golang.org/grpc v1.57.0/go.mod h1:Sd+9RMTACXwmub0zcNY2c4arhtrbBYD1AUHI/dt16Mo= |  | ||||||
| google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw= |  | ||||||
| google.golang.org/protobuf v1.26.0/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc= |  | ||||||
| google.golang.org/protobuf v1.33.0 h1:uNO2rsAINq/JlFpSdYEKIZ0uKD/R9cpdv0T+yoGwGmI= |  | ||||||
| google.golang.org/protobuf v1.33.0/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos= |  | ||||||
| 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.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= | ||||||
|   | |||||||
							
								
								
									
										117
									
								
								hooks/metadata/metadata.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										117
									
								
								hooks/metadata/metadata.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,117 @@ | |||||||
|  | package metadata | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v4/client" | ||||||
|  | 	"go.unistack.org/micro/v4/metadata" | ||||||
|  | 	"go.unistack.org/micro/v4/server" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type wrapper struct { | ||||||
|  | 	keys []string | ||||||
|  |  | ||||||
|  | 	client.Client | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func NewClientWrapper(keys ...string) client.Wrapper { | ||||||
|  | 	return func(c client.Client) client.Client { | ||||||
|  | 		handler := &wrapper{ | ||||||
|  | 			Client: c, | ||||||
|  | 			keys:   keys, | ||||||
|  | 		} | ||||||
|  | 		return handler | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func NewClientCallWrapper(keys ...string) client.CallWrapper { | ||||||
|  | 	return func(fn client.CallFunc) client.CallFunc { | ||||||
|  | 		return func(ctx context.Context, addr string, req client.Request, rsp interface{}, opts client.CallOptions) error { | ||||||
|  | 			if keys == nil { | ||||||
|  | 				return fn(ctx, addr, req, rsp, opts) | ||||||
|  | 			} | ||||||
|  | 			if imd, iok := metadata.FromIncomingContext(ctx); iok && imd != nil { | ||||||
|  | 				omd, ook := metadata.FromOutgoingContext(ctx) | ||||||
|  | 				if !ook || omd == nil { | ||||||
|  | 					omd = metadata.New(len(imd)) | ||||||
|  | 				} | ||||||
|  | 				for _, k := range keys { | ||||||
|  | 					if v, ok := imd.Get(k); ok { | ||||||
|  | 						omd.Add(k, v...) | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 				if !ook { | ||||||
|  | 					ctx = metadata.NewOutgoingContext(ctx, omd) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			return fn(ctx, addr, req, rsp, opts) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *wrapper) Call(ctx context.Context, req client.Request, rsp interface{}, opts ...client.CallOption) error { | ||||||
|  | 	if w.keys == nil { | ||||||
|  | 		return w.Client.Call(ctx, req, rsp, opts...) | ||||||
|  | 	} | ||||||
|  | 	if imd, iok := metadata.FromIncomingContext(ctx); iok && imd != nil { | ||||||
|  | 		omd, ook := metadata.FromOutgoingContext(ctx) | ||||||
|  | 		if !ook || omd == nil { | ||||||
|  | 			omd = metadata.New(len(imd)) | ||||||
|  | 		} | ||||||
|  | 		for _, k := range w.keys { | ||||||
|  | 			if v, ok := imd.Get(k); ok { | ||||||
|  | 				omd.Add(k, v...) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		if !ook { | ||||||
|  | 			ctx = metadata.NewOutgoingContext(ctx, omd) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return w.Client.Call(ctx, req, rsp, opts...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *wrapper) Stream(ctx context.Context, req client.Request, opts ...client.CallOption) (client.Stream, error) { | ||||||
|  | 	if w.keys == nil { | ||||||
|  | 		return w.Client.Stream(ctx, req, opts...) | ||||||
|  | 	} | ||||||
|  | 	if imd, iok := metadata.FromIncomingContext(ctx); iok && imd != nil { | ||||||
|  | 		omd, ook := metadata.FromOutgoingContext(ctx) | ||||||
|  | 		if !ook || omd == nil { | ||||||
|  | 			omd = metadata.New(len(imd)) | ||||||
|  | 		} | ||||||
|  | 		for _, k := range w.keys { | ||||||
|  | 			if v, ok := imd.Get(k); ok { | ||||||
|  | 				omd.Add(k, v...) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		if !ook { | ||||||
|  | 			ctx = metadata.NewOutgoingContext(ctx, omd) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return w.Client.Stream(ctx, req, opts...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func NewServerHandlerWrapper(keys ...string) server.HandlerWrapper { | ||||||
|  | 	return func(fn server.HandlerFunc) server.HandlerFunc { | ||||||
|  | 		return func(ctx context.Context, req server.Request, rsp interface{}) error { | ||||||
|  | 			if keys == nil { | ||||||
|  | 				return fn(ctx, req, rsp) | ||||||
|  | 			} | ||||||
|  | 			if imd, iok := metadata.FromIncomingContext(ctx); iok && imd != nil { | ||||||
|  | 				omd, ook := metadata.FromOutgoingContext(ctx) | ||||||
|  | 				if !ook || omd == nil { | ||||||
|  | 					omd = metadata.New(len(imd)) | ||||||
|  | 				} | ||||||
|  | 				for _, k := range keys { | ||||||
|  | 					if v, ok := imd.Get(k); ok { | ||||||
|  | 						omd.Add(k, v...) | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 				if !ook { | ||||||
|  | 					ctx = metadata.NewOutgoingContext(ctx, omd) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			return fn(ctx, req, rsp) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										63
									
								
								hooks/recovery/recovery.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										63
									
								
								hooks/recovery/recovery.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,63 @@ | |||||||
|  | package recovery | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"fmt" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v4/errors" | ||||||
|  | 	"go.unistack.org/micro/v4/server" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func NewOptions(opts ...Option) Options { | ||||||
|  | 	options := Options{ | ||||||
|  | 		ServerHandlerFn: DefaultServerHandlerFn, | ||||||
|  | 	} | ||||||
|  | 	for _, o := range opts { | ||||||
|  | 		o(&options) | ||||||
|  | 	} | ||||||
|  | 	return options | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type Options struct { | ||||||
|  | 	ServerHandlerFn func(context.Context, server.Request, interface{}, error) error | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type Option func(*Options) | ||||||
|  |  | ||||||
|  | func ServerHandlerFunc(fn func(context.Context, server.Request, interface{}, error) error) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.ServerHandlerFn = fn | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | var DefaultServerHandlerFn = func(ctx context.Context, req server.Request, rsp interface{}, err error) error { | ||||||
|  | 	return errors.BadRequest("", "%v", err) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | var Hook = NewHook() | ||||||
|  |  | ||||||
|  | type hook struct { | ||||||
|  | 	opts Options | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func NewHook(opts ...Option) *hook { | ||||||
|  | 	return &hook{opts: NewOptions(opts...)} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *hook) ServerHandler(next server.FuncHandler) server.FuncHandler { | ||||||
|  | 	return func(ctx context.Context, req server.Request, rsp interface{}) (err error) { | ||||||
|  | 		defer func() { | ||||||
|  | 			r := recover() | ||||||
|  | 			switch verr := r.(type) { | ||||||
|  | 			case nil: | ||||||
|  | 				return | ||||||
|  | 			case error: | ||||||
|  | 				err = w.opts.ServerHandlerFn(ctx, req, rsp, verr) | ||||||
|  | 			default: | ||||||
|  | 				err = w.opts.ServerHandlerFn(ctx, req, rsp, fmt.Errorf("%v", r)) | ||||||
|  | 			} | ||||||
|  | 		}() | ||||||
|  | 		err = next(ctx, req, rsp) | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										114
									
								
								hooks/requestid/requestid.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										114
									
								
								hooks/requestid/requestid.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,114 @@ | |||||||
|  | package requestid | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"net/textproto" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v4/client" | ||||||
|  | 	"go.unistack.org/micro/v4/metadata" | ||||||
|  | 	"go.unistack.org/micro/v4/server" | ||||||
|  | 	"go.unistack.org/micro/v4/util/id" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type XRequestIDKey struct{} | ||||||
|  |  | ||||||
|  | // DefaultMetadataKey contains metadata key | ||||||
|  | var DefaultMetadataKey = textproto.CanonicalMIMEHeaderKey("x-request-id") | ||||||
|  |  | ||||||
|  | // DefaultMetadataFunc wil be used if user not provide own func to fill metadata | ||||||
|  | var DefaultMetadataFunc = func(ctx context.Context) (context.Context, error) { | ||||||
|  | 	var xid string | ||||||
|  |  | ||||||
|  | 	cid, cok := ctx.Value(XRequestIDKey{}).(string) | ||||||
|  | 	if cok && cid != "" { | ||||||
|  | 		xid = cid | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	imd, iok := metadata.FromIncomingContext(ctx) | ||||||
|  | 	if !iok || imd == nil { | ||||||
|  | 		imd = metadata.New(1) | ||||||
|  | 		ctx = metadata.NewIncomingContext(ctx, imd) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	omd, ook := metadata.FromOutgoingContext(ctx) | ||||||
|  | 	if !ook || omd == nil { | ||||||
|  | 		omd = metadata.New(1) | ||||||
|  | 		ctx = metadata.NewOutgoingContext(ctx, omd) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if xid == "" { | ||||||
|  | 		var ids []string | ||||||
|  | 		if ids, iok = imd.Get(DefaultMetadataKey); iok { | ||||||
|  | 			for i := range ids { | ||||||
|  | 				if ids[i] != "" { | ||||||
|  | 					xid = ids[i] | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		if ids, ook = omd.Get(DefaultMetadataKey); ook { | ||||||
|  | 			for i := range ids { | ||||||
|  | 				if ids[i] != "" { | ||||||
|  | 					xid = ids[i] | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if xid == "" { | ||||||
|  | 		var err error | ||||||
|  | 		xid, err = id.New() | ||||||
|  | 		if err != nil { | ||||||
|  | 			return ctx, err | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !cok { | ||||||
|  | 		ctx = context.WithValue(ctx, XRequestIDKey{}, xid) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !iok { | ||||||
|  | 		imd.Set(DefaultMetadataKey, xid) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !ook { | ||||||
|  | 		omd.Set(DefaultMetadataKey, xid) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return ctx, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type hook struct{} | ||||||
|  |  | ||||||
|  | func NewHook() *hook { | ||||||
|  | 	return &hook{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *hook) ServerHandler(next server.FuncHandler) server.FuncHandler { | ||||||
|  | 	return func(ctx context.Context, req server.Request, rsp interface{}) error { | ||||||
|  | 		var err error | ||||||
|  | 		if ctx, err = DefaultMetadataFunc(ctx); err != nil { | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  | 		return next(ctx, req, rsp) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *hook) ClientCall(next client.FuncCall) client.FuncCall { | ||||||
|  | 	return func(ctx context.Context, req client.Request, rsp interface{}, opts ...client.CallOption) error { | ||||||
|  | 		var err error | ||||||
|  | 		if ctx, err = DefaultMetadataFunc(ctx); err != nil { | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  | 		return next(ctx, req, rsp, opts...) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *hook) ClientStream(next client.FuncStream) client.FuncStream { | ||||||
|  | 	return func(ctx context.Context, req client.Request, opts ...client.CallOption) (client.Stream, error) { | ||||||
|  | 		var err error | ||||||
|  | 		if ctx, err = DefaultMetadataFunc(ctx); err != nil { | ||||||
|  | 			return nil, err | ||||||
|  | 		} | ||||||
|  | 		return next(ctx, req, opts...) | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										33
									
								
								hooks/requestid/requestid_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										33
									
								
								hooks/requestid/requestid_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,33 @@ | |||||||
|  | package requestid | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"testing" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v4/metadata" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func TestDefaultMetadataFunc(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  |  | ||||||
|  | 	nctx, err := DefaultMetadataFunc(ctx) | ||||||
|  | 	if err != nil { | ||||||
|  | 		t.Fatalf("%v", err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	imd, ok := metadata.FromIncomingContext(nctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		t.Fatalf("md missing in incoming context") | ||||||
|  | 	} | ||||||
|  | 	omd, ok := metadata.FromOutgoingContext(nctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		t.Fatalf("md missing in outgoing context") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	_, iok := imd.Get(DefaultMetadataKey) | ||||||
|  | 	_, ook := omd.Get(DefaultMetadataKey) | ||||||
|  |  | ||||||
|  | 	if !iok || !ook { | ||||||
|  | 		t.Fatalf("missing metadata key value") | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										133
									
								
								hooks/validator/validator.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										133
									
								
								hooks/validator/validator.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,133 @@ | |||||||
|  | package validator | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v4/client" | ||||||
|  | 	"go.unistack.org/micro/v4/errors" | ||||||
|  | 	"go.unistack.org/micro/v4/server" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | var ( | ||||||
|  | 	DefaultClientErrorFunc = func(req client.Request, rsp interface{}, err error) error { | ||||||
|  | 		if rsp != nil { | ||||||
|  | 			return errors.BadGateway(req.Service(), "%v", err) | ||||||
|  | 		} | ||||||
|  | 		return errors.BadRequest(req.Service(), "%v", err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	DefaultServerErrorFunc = func(req server.Request, rsp interface{}, err error) error { | ||||||
|  | 		if rsp != nil { | ||||||
|  | 			return errors.BadGateway(req.Service(), "%v", err) | ||||||
|  | 		} | ||||||
|  | 		return errors.BadRequest(req.Service(), "%v", err) | ||||||
|  | 	} | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type ( | ||||||
|  | 	ClientErrorFunc func(client.Request, interface{}, error) error | ||||||
|  | 	ServerErrorFunc func(server.Request, interface{}, error) error | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Options struct holds wrapper options | ||||||
|  | type Options struct { | ||||||
|  | 	ClientErrorFn          ClientErrorFunc | ||||||
|  | 	ServerErrorFn          ServerErrorFunc | ||||||
|  | 	ClientValidateResponse bool | ||||||
|  | 	ServerValidateResponse bool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Option func signature | ||||||
|  | type Option func(*Options) | ||||||
|  |  | ||||||
|  | func ClientValidateResponse(b bool) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.ClientValidateResponse = b | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func ServerValidateResponse(b bool) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.ClientValidateResponse = b | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func ClientReqErrorFn(fn ClientErrorFunc) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.ClientErrorFn = fn | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func ServerErrorFn(fn ServerErrorFunc) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.ServerErrorFn = fn | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func NewOptions(opts ...Option) Options { | ||||||
|  | 	options := Options{ | ||||||
|  | 		ClientErrorFn: DefaultClientErrorFunc, | ||||||
|  | 		ServerErrorFn: DefaultServerErrorFunc, | ||||||
|  | 	} | ||||||
|  | 	for _, o := range opts { | ||||||
|  | 		o(&options) | ||||||
|  | 	} | ||||||
|  | 	return options | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func NewHook(opts ...Option) *hook { | ||||||
|  | 	return &hook{opts: NewOptions(opts...)} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type validator interface { | ||||||
|  | 	Validate() error | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type hook struct { | ||||||
|  | 	opts Options | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *hook) ClientCall(next client.FuncCall) client.FuncCall { | ||||||
|  | 	return func(ctx context.Context, req client.Request, rsp interface{}, opts ...client.CallOption) error { | ||||||
|  | 		if v, ok := req.Body().(validator); ok { | ||||||
|  | 			if err := v.Validate(); err != nil { | ||||||
|  | 				return w.opts.ClientErrorFn(req, nil, err) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		err := next(ctx, req, rsp, opts...) | ||||||
|  | 		if v, ok := rsp.(validator); ok && w.opts.ClientValidateResponse { | ||||||
|  | 			if verr := v.Validate(); verr != nil { | ||||||
|  | 				return w.opts.ClientErrorFn(req, rsp, verr) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *hook) ClientStream(next client.FuncStream) client.FuncStream { | ||||||
|  | 	return func(ctx context.Context, req client.Request, opts ...client.CallOption) (client.Stream, error) { | ||||||
|  | 		if v, ok := req.Body().(validator); ok { | ||||||
|  | 			if err := v.Validate(); err != nil { | ||||||
|  | 				return nil, w.opts.ClientErrorFn(req, nil, err) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		return next(ctx, req, opts...) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *hook) ServerHandler(next server.FuncHandler) server.FuncHandler { | ||||||
|  | 	return func(ctx context.Context, req server.Request, rsp interface{}) error { | ||||||
|  | 		if v, ok := req.Body().(validator); ok { | ||||||
|  | 			if err := v.Validate(); err != nil { | ||||||
|  | 				return w.opts.ServerErrorFn(req, nil, err) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		err := next(ctx, req, rsp) | ||||||
|  | 		if v, ok := rsp.(validator); ok && w.opts.ServerValidateResponse { | ||||||
|  | 			if verr := v.Validate(); verr != nil { | ||||||
|  | 				return w.opts.ServerErrorFn(req, rsp, verr) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -4,17 +4,6 @@ import "context" | |||||||
|  |  | ||||||
| type loggerKey struct{} | type loggerKey struct{} | ||||||
|  |  | ||||||
| // MustContext returns logger from passed context or DefaultLogger if empty |  | ||||||
| func MustContext(ctx context.Context) Logger { |  | ||||||
| 	if ctx == nil { |  | ||||||
| 		return DefaultLogger |  | ||||||
| 	} |  | ||||||
| 	if l, ok := ctx.Value(loggerKey{}).(Logger); ok && l != nil { |  | ||||||
| 		return l |  | ||||||
| 	} |  | ||||||
| 	return DefaultLogger |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // FromContext returns logger from passed context | // FromContext returns logger from passed context | ||||||
| func FromContext(ctx context.Context) (Logger, bool) { | func FromContext(ctx context.Context) (Logger, bool) { | ||||||
| 	if ctx == nil { | 	if ctx == nil { | ||||||
| @@ -24,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 { | ||||||
|   | |||||||
| @@ -11,7 +11,7 @@ var DefaultContextAttrFuncs []ContextAttrFunc | |||||||
|  |  | ||||||
| var ( | var ( | ||||||
| 	// DefaultLogger variable | 	// DefaultLogger variable | ||||||
| 	DefaultLogger Logger = NewLogger() | 	DefaultLogger = NewLogger() | ||||||
| 	// DefaultLevel used by logger | 	// DefaultLevel used by logger | ||||||
| 	DefaultLevel = InfoLevel | 	DefaultLevel = InfoLevel | ||||||
| ) | ) | ||||||
|   | |||||||
| @@ -5,9 +5,10 @@ import ( | |||||||
| 	"io" | 	"io" | ||||||
| 	"log/slog" | 	"log/slog" | ||||||
| 	"os" | 	"os" | ||||||
|  | 	"slices" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/meter" | 	"go.unistack.org/micro/v4/meter" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Option func signature | // Option func signature | ||||||
| @@ -15,18 +16,6 @@ type Option func(*Options) | |||||||
|  |  | ||||||
| // Options holds logger options | // Options holds logger options | ||||||
| type Options struct { | type Options struct { | ||||||
| 	// Out holds the output writer |  | ||||||
| 	Out io.Writer |  | ||||||
| 	// Context holds exernal options |  | ||||||
| 	Context context.Context |  | ||||||
| 	// Name holds the logger name |  | ||||||
| 	Name string |  | ||||||
| 	// Fields holds additional metadata |  | ||||||
| 	Fields []interface{} |  | ||||||
| 	// callerSkipCount number of frmaes to skip |  | ||||||
| 	CallerSkipCount int |  | ||||||
| 	// ContextAttrFuncs contains funcs that executed before log func on context |  | ||||||
| 	ContextAttrFuncs []ContextAttrFunc |  | ||||||
| 	// TimeKey is the key used for the time of the log call | 	// TimeKey is the key used for the time of the log call | ||||||
| 	TimeKey string | 	TimeKey string | ||||||
| 	// LevelKey is the key used for the level of the log call | 	// LevelKey is the key used for the level of the log call | ||||||
| @@ -39,16 +28,30 @@ type Options struct { | |||||||
| 	SourceKey string | 	SourceKey string | ||||||
| 	// StacktraceKey is the key used for the stacktrace | 	// StacktraceKey is the key used for the stacktrace | ||||||
| 	StacktraceKey string | 	StacktraceKey string | ||||||
| 	// AddStacktrace controls writing of stacktaces on error | 	// Name holds the logger name | ||||||
| 	AddStacktrace bool | 	Name string | ||||||
| 	// AddSource enabled writing source file and position in log | 	// Out holds the output writer | ||||||
| 	AddSource bool | 	Out io.Writer | ||||||
| 	// The logging level the logger should log | 	// Context holds exernal options | ||||||
| 	Level Level | 	Context context.Context | ||||||
| 	// TimeFunc used to obtain current time |  | ||||||
| 	TimeFunc func() time.Time |  | ||||||
| 	// Meter used to count logs for specific level | 	// Meter used to count logs for specific level | ||||||
| 	Meter meter.Meter | 	Meter meter.Meter | ||||||
|  | 	// TimeFunc used to obtain current time | ||||||
|  | 	TimeFunc func() time.Time | ||||||
|  | 	// Fields holds additional metadata | ||||||
|  | 	Fields []interface{} | ||||||
|  | 	// ContextAttrFuncs contains funcs that executed before log func on context | ||||||
|  | 	ContextAttrFuncs []ContextAttrFunc | ||||||
|  | 	// callerSkipCount number of frmaes to skip | ||||||
|  | 	CallerSkipCount int | ||||||
|  | 	// The logging level the logger should log | ||||||
|  | 	Level Level | ||||||
|  | 	// AddSource enabled writing source file and position in log | ||||||
|  | 	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 | ||||||
| @@ -80,11 +83,34 @@ func WithContextAttrFuncs(fncs ...ContextAttrFunc) Option { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // WithDedupKeys dont log duplicate keys | ||||||
|  | func WithDedupKeys(b bool) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.DedupKeys = b | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // WithAddFields add fields for the logger | // WithAddFields add fields for the logger | ||||||
| func WithAddFields(fields ...interface{}) Option { | func WithAddFields(fields ...interface{}) Option { | ||||||
| 	return func(o *Options) { | 	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 { | ||||||
|  | 						o.Fields[i+1] = fields[j+1] | ||||||
|  | 						fields = slices.Delete(fields, j, j+2) | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			if len(fields) > 0 { | ||||||
| 				o.Fields = append(o.Fields, fields...) | 				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 | ||||||
| @@ -153,8 +179,8 @@ func WithTimeFunc(fn func() time.Time) Option { | |||||||
| func WithZapKeys() Option { | func WithZapKeys() Option { | ||||||
| 	return func(o *Options) { | 	return func(o *Options) { | ||||||
| 		o.TimeKey = "@timestamp" | 		o.TimeKey = "@timestamp" | ||||||
| 		o.LevelKey = "level" | 		o.LevelKey = slog.LevelKey | ||||||
| 		o.MessageKey = "msg" | 		o.MessageKey = slog.MessageKey | ||||||
| 		o.SourceKey = "caller" | 		o.SourceKey = "caller" | ||||||
| 		o.StacktraceKey = "stacktrace" | 		o.StacktraceKey = "stacktrace" | ||||||
| 		o.ErrorKey = "error" | 		o.ErrorKey = "error" | ||||||
| @@ -163,8 +189,8 @@ func WithZapKeys() Option { | |||||||
|  |  | ||||||
| func WithZerologKeys() Option { | func WithZerologKeys() Option { | ||||||
| 	return func(o *Options) { | 	return func(o *Options) { | ||||||
| 		o.TimeKey = "time" | 		o.TimeKey = slog.TimeKey | ||||||
| 		o.LevelKey = "level" | 		o.LevelKey = slog.LevelKey | ||||||
| 		o.MessageKey = "message" | 		o.MessageKey = "message" | ||||||
| 		o.SourceKey = "caller" | 		o.SourceKey = "caller" | ||||||
| 		o.StacktraceKey = "stacktrace" | 		o.StacktraceKey = "stacktrace" | ||||||
| @@ -186,8 +212,8 @@ func WithSlogKeys() Option { | |||||||
| func WithMicroKeys() Option { | func WithMicroKeys() Option { | ||||||
| 	return func(o *Options) { | 	return func(o *Options) { | ||||||
| 		o.TimeKey = "timestamp" | 		o.TimeKey = "timestamp" | ||||||
| 		o.LevelKey = "level" | 		o.LevelKey = slog.LevelKey | ||||||
| 		o.MessageKey = "msg" | 		o.MessageKey = slog.MessageKey | ||||||
| 		o.SourceKey = "caller" | 		o.SourceKey = "caller" | ||||||
| 		o.StacktraceKey = "stacktrace" | 		o.StacktraceKey = "stacktrace" | ||||||
| 		o.ErrorKey = "error" | 		o.ErrorKey = "error" | ||||||
|   | |||||||
| @@ -2,23 +2,27 @@ package slog | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
|  | 	"io" | ||||||
| 	"log/slog" | 	"log/slog" | ||||||
| 	"os" | 	"os" | ||||||
|  | 	"reflect" | ||||||
| 	"regexp" | 	"regexp" | ||||||
| 	"runtime" | 	"runtime" | ||||||
| 	"strconv" | 	"strconv" | ||||||
| 	"sync" | 	"sync" | ||||||
| 	"sync/atomic" | 	"sync/atomic" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/logger" | 	"go.unistack.org/micro/v4/logger" | ||||||
| 	"go.unistack.org/micro/v3/semconv" | 	"go.unistack.org/micro/v4/semconv" | ||||||
| 	"go.unistack.org/micro/v3/tracer" | 	"go.unistack.org/micro/v4/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| const ( | const ( | ||||||
| 	badKey = "!BADKEY" | 	badKey = "!BADKEY" | ||||||
| 	// defaultCallerSkipCount used by logger | 	// defaultCallerSkipCount used by logger | ||||||
| 	defaultCallerSkipCount = 3 | 	defaultCallerSkipCount = 3 | ||||||
|  | 	timeFormat             = "2006-01-02T15:04:05.000000000Z07:00" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var reTrace = regexp.MustCompile(`.*/slog/logger\.go.*\n`) | var reTrace = regexp.MustCompile(`.*/slog/logger\.go.*\n`) | ||||||
| @@ -46,11 +50,11 @@ func (h *wrapper) Handle(ctx context.Context, rec slog.Record) error { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (h *wrapper) WithAttrs(attrs []slog.Attr) slog.Handler { | func (h *wrapper) WithAttrs(attrs []slog.Attr) slog.Handler { | ||||||
| 	return h.WithAttrs(attrs) | 	return h.h.WithAttrs(attrs) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (h *wrapper) WithGroup(name string) slog.Handler { | func (h *wrapper) WithGroup(name string) slog.Handler { | ||||||
| 	return h.WithGroup(name) | 	return h.h.WithGroup(name) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (s *slogLogger) renameAttr(_ []string, a slog.Attr) slog.Attr { | func (s *slogLogger) renameAttr(_ []string, a slog.Attr) slog.Attr { | ||||||
| @@ -61,6 +65,7 @@ func (s *slogLogger) renameAttr(_ []string, a slog.Attr) slog.Attr { | |||||||
| 		a.Key = s.opts.SourceKey | 		a.Key = s.opts.SourceKey | ||||||
| 	case slog.TimeKey: | 	case slog.TimeKey: | ||||||
| 		a.Key = s.opts.TimeKey | 		a.Key = s.opts.TimeKey | ||||||
|  | 		a.Value = slog.StringValue(a.Value.Time().Format(timeFormat)) | ||||||
| 	case slog.MessageKey: | 	case slog.MessageKey: | ||||||
| 		a.Key = s.opts.MessageKey | 		a.Key = s.opts.MessageKey | ||||||
| 	case slog.LevelKey: | 	case slog.LevelKey: | ||||||
| @@ -89,7 +94,6 @@ func (s *slogLogger) renameAttr(_ []string, a slog.Attr) slog.Attr { | |||||||
| } | } | ||||||
|  |  | ||||||
| type slogLogger struct { | type slogLogger struct { | ||||||
| 	leveler *slog.LevelVar |  | ||||||
| 	handler *wrapper | 	handler *wrapper | ||||||
| 	opts    logger.Options | 	opts    logger.Options | ||||||
| 	mu      sync.RWMutex | 	mu      sync.RWMutex | ||||||
| @@ -142,6 +146,7 @@ func (s *slogLogger) Fields(fields ...interface{}) logger.Logger { | |||||||
| 	s.mu.RUnlock() | 	s.mu.RUnlock() | ||||||
|  |  | ||||||
| 	l := &slogLogger{opts: options} | 	l := &slogLogger{opts: options} | ||||||
|  | 	logger.WithAddFields(fields...)(&l.opts) | ||||||
|  |  | ||||||
| 	if len(options.ContextAttrFuncs) == 0 { | 	if len(options.ContextAttrFuncs) == 0 { | ||||||
| 		options.ContextAttrFuncs = logger.DefaultContextAttrFuncs | 		options.ContextAttrFuncs = logger.DefaultContextAttrFuncs | ||||||
| @@ -172,7 +177,29 @@ func (s *slogLogger) Init(opts ...logger.Option) error { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	attrs, _ := s.argsAttrs(s.opts.Fields) | 	attrs, _ := s.argsAttrs(s.opts.Fields) | ||||||
| 	s.handler = &wrapper{h: slog.NewJSONHandler(s.opts.Out, handleOpt).WithAttrs(attrs)} |  | ||||||
|  | 	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.handler.level.Store(int64(loggerToSlogLevel(s.opts.Level))) | ||||||
| 	s.mu.Unlock() | 	s.mu.Unlock() | ||||||
|  |  | ||||||
| @@ -201,6 +228,10 @@ func (s *slogLogger) Error(ctx context.Context, msg string, attrs ...interface{} | |||||||
|  |  | ||||||
| func (s *slogLogger) Fatal(ctx context.Context, msg string, attrs ...interface{}) { | func (s *slogLogger) Fatal(ctx context.Context, msg string, attrs ...interface{}) { | ||||||
| 	s.printLog(ctx, logger.FatalLevel, msg, attrs...) | 	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) | 	os.Exit(1) | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -247,7 +278,7 @@ func (s *slogLogger) printLog(ctx context.Context, lvl logger.Level, msg string, | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if s.opts.AddStacktrace && lvl == logger.ErrorLevel { | 	if s.opts.AddStacktrace && (lvl == logger.FatalLevel || lvl == logger.ErrorLevel) { | ||||||
| 		stackInfo := make([]byte, 1024*1024) | 		stackInfo := make([]byte, 1024*1024) | ||||||
| 		if stackSize := runtime.Stack(stackInfo, false); stackSize > 0 { | 		if stackSize := runtime.Stack(stackInfo, false); stackSize > 0 { | ||||||
| 			traceLines := reTrace.Split(string(stackInfo[:stackSize]), -1) | 			traceLines := reTrace.Split(string(stackInfo[:stackSize]), -1) | ||||||
| @@ -318,7 +349,7 @@ func (s *slogLogger) argsAttrs(args []interface{}) ([]slog.Attr, error) { | |||||||
| 		case string: | 		case string: | ||||||
| 			if idx+1 < len(args) { | 			if idx+1 < len(args) { | ||||||
| 				attrs = append(attrs, slog.Any(arg, args[idx+1])) | 				attrs = append(attrs, slog.Any(arg, args[idx+1])) | ||||||
| 				idx += 1 | 				idx++ | ||||||
| 			} else { | 			} else { | ||||||
| 				attrs = append(attrs, slog.String(badKey, arg)) | 				attrs = append(attrs, slog.String(badKey, arg)) | ||||||
| 			} | 			} | ||||||
| @@ -330,3 +361,15 @@ func (s *slogLogger) argsAttrs(args []interface{}) ([]slog.Attr, error) { | |||||||
|  |  | ||||||
| 	return attrs, err | 	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) | ||||||
|  | } | ||||||
|   | |||||||
| @@ -6,15 +6,146 @@ import ( | |||||||
| 	"errors" | 	"errors" | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"log" | 	"log" | ||||||
|  | 	"log/slog" | ||||||
| 	"strings" | 	"strings" | ||||||
| 	"testing" | 	"testing" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
| 	"github.com/google/uuid" | 	"github.com/google/uuid" | ||||||
| 	"go.unistack.org/micro/v3/metadata" | 	"go.unistack.org/micro/v4/logger" | ||||||
|  | 	"go.unistack.org/micro/v4/metadata" | ||||||
| 	"go.unistack.org/micro/v3/logger" | 	"go.unistack.org/micro/v4/util/buffer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | // always first to have proper check | ||||||
|  | func TestStacktrace(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.DebugLevel), logger.WithOutput(buf), | ||||||
|  | 		WithHandlerFunc(slog.NewTextHandler), | ||||||
|  | 		logger.WithAddStacktrace(true), | ||||||
|  | 	) | ||||||
|  | 	if err := l.Init(logger.WithFields("key1", "val1")); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l.Error(ctx, "msg1", errors.New("err")) | ||||||
|  |  | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`slog_test.go:32`)) { | ||||||
|  | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestDelayedBuffer(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	dbuf := buffer.NewDelayedBuffer(100, 100*time.Millisecond, buf) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.ErrorLevel), logger.WithOutput(dbuf), | ||||||
|  | 		WithHandlerFunc(slog.NewTextHandler), | ||||||
|  | 		logger.WithAddStacktrace(true), | ||||||
|  | 	) | ||||||
|  | 	if err := l.Init(logger.WithFields("key1", "val1")); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l.Error(ctx, "msg1", errors.New("err")) | ||||||
|  | 	time.Sleep(120 * time.Millisecond) | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`key1=val1`)) { | ||||||
|  | 		t.Fatalf("logger delayed buffer not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestTime(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.ErrorLevel), logger.WithOutput(buf), | ||||||
|  | 		WithHandlerFunc(slog.NewTextHandler), | ||||||
|  | 		logger.WithAddStacktrace(true), | ||||||
|  | 		logger.WithTimeFunc(func() time.Time { | ||||||
|  | 			return time.Unix(0, 0) | ||||||
|  | 		}), | ||||||
|  | 	) | ||||||
|  | 	if err := l.Init(logger.WithFields("key1", "val1")); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l.Error(ctx, "msg1", errors.New("err")) | ||||||
|  |  | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`timestamp=1970-01-01T03:00:00.000000000+03:00`)) && | ||||||
|  | 		!bytes.Contains(buf.Bytes(), []byte(`timestamp=1970-01-01T00:00:00.000000000Z`)) { | ||||||
|  | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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=val4 key2=val3`)) { | ||||||
|  | 		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) { | func TestWithAddFields(t *testing.T) { | ||||||
| 	ctx := context.TODO() | 	ctx := context.TODO() | ||||||
| 	buf := bytes.NewBuffer(nil) | 	buf := bytes.NewBuffer(nil) | ||||||
| @@ -115,7 +246,7 @@ func TestErrorf(t *testing.T) { | |||||||
|  |  | ||||||
| 	buf := bytes.NewBuffer(nil) | 	buf := bytes.NewBuffer(nil) | ||||||
| 	l := NewLogger(logger.WithLevel(logger.ErrorLevel), logger.WithOutput(buf), logger.WithAddStacktrace(true)) | 	l := NewLogger(logger.WithLevel(logger.ErrorLevel), logger.WithOutput(buf), logger.WithAddStacktrace(true)) | ||||||
| 	if err := l.Init(logger.WithContextAttrFuncs(func(ctx context.Context) []interface{} { | 	if err := l.Init(logger.WithContextAttrFuncs(func(_ context.Context) []interface{} { | ||||||
| 		return nil | 		return nil | ||||||
| 	})); err != nil { | 	})); err != nil { | ||||||
| 		t.Fatal(err) | 		t.Fatal(err) | ||||||
| @@ -281,9 +412,10 @@ func Test_WithContextAttrFunc(t *testing.T) { | |||||||
| 			} | 			} | ||||||
| 			attrs := make([]interface{}, 0, 10) | 			attrs := make([]interface{}, 0, 10) | ||||||
| 			for k, v := range md { | 			for k, v := range md { | ||||||
| 				switch k { | 				key := strings.ToLower(k) | ||||||
| 				case "X-Request-Id", "Phone", "External-Id", "Source-Service", "X-App-Install-Id", "Client-Id", "Client-Ip": | 				switch key { | ||||||
| 					attrs = append(attrs, strings.ToLower(k), v) | 				case "x-request-id", "phone", "external-Id", "source-service", "x-app-install-id", "client-id", "client-ip": | ||||||
|  | 					attrs = append(attrs, key, v[0]) | ||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
| 			return attrs | 			return attrs | ||||||
|   | |||||||
| @@ -8,7 +8,7 @@ import ( | |||||||
| 	"strconv" | 	"strconv" | ||||||
| 	"strings" | 	"strings" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/codec" | 	"go.unistack.org/micro/v4/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/v3/codec" | 	"go.unistack.org/micro/v4/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,131 +0,0 @@ | |||||||
| // Package metadata is a way of defining message headers |  | ||||||
| package metadata |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"context" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| type ( |  | ||||||
| 	mdIncomingKey struct{} |  | ||||||
| 	mdOutgoingKey struct{} |  | ||||||
| 	mdKey         struct{} |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // FromIncomingContext returns metadata from incoming ctx |  | ||||||
| // returned metadata shoud not be modified or race condition happens |  | ||||||
| func FromIncomingContext(ctx context.Context) (Metadata, bool) { |  | ||||||
| 	if ctx == nil { |  | ||||||
| 		return nil, false |  | ||||||
| 	} |  | ||||||
| 	md, ok := ctx.Value(mdIncomingKey{}).(*rawMetadata) |  | ||||||
| 	if !ok || md.md == nil { |  | ||||||
| 		return nil, false |  | ||||||
| 	} |  | ||||||
| 	return md.md, ok |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // FromOutgoingContext returns metadata from outgoing ctx |  | ||||||
| // returned metadata shoud not be modified or race condition happens |  | ||||||
| func FromOutgoingContext(ctx context.Context) (Metadata, bool) { |  | ||||||
| 	if ctx == nil { |  | ||||||
| 		return nil, false |  | ||||||
| 	} |  | ||||||
| 	md, ok := ctx.Value(mdOutgoingKey{}).(*rawMetadata) |  | ||||||
| 	if !ok || md.md == nil { |  | ||||||
| 		return nil, false |  | ||||||
| 	} |  | ||||||
| 	return md.md, ok |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // FromContext returns metadata from the given context |  | ||||||
| // returned metadata shoud not be modified or race condition happens |  | ||||||
| func FromContext(ctx context.Context) (Metadata, bool) { |  | ||||||
| 	if ctx == nil { |  | ||||||
| 		return nil, false |  | ||||||
| 	} |  | ||||||
| 	md, ok := ctx.Value(mdKey{}).(*rawMetadata) |  | ||||||
| 	if !ok || md.md == nil { |  | ||||||
| 		return nil, false |  | ||||||
| 	} |  | ||||||
| 	return md.md, ok |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewContext creates a new context with the given metadata |  | ||||||
| func NewContext(ctx context.Context, md Metadata) context.Context { |  | ||||||
| 	if ctx == nil { |  | ||||||
| 		ctx = context.Background() |  | ||||||
| 	} |  | ||||||
| 	return context.WithValue(ctx, mdKey{}, &rawMetadata{md}) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // SetOutgoingContext modify outgoing context with given metadata |  | ||||||
| func SetOutgoingContext(ctx context.Context, md Metadata) bool { |  | ||||||
| 	if ctx == nil { |  | ||||||
| 		return false |  | ||||||
| 	} |  | ||||||
| 	if omd, ok := ctx.Value(mdOutgoingKey{}).(*rawMetadata); ok { |  | ||||||
| 		omd.md = md |  | ||||||
| 		return true |  | ||||||
| 	} |  | ||||||
| 	return false |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // SetIncomingContext modify incoming context with given metadata |  | ||||||
| func SetIncomingContext(ctx context.Context, md Metadata) bool { |  | ||||||
| 	if ctx == nil { |  | ||||||
| 		return false |  | ||||||
| 	} |  | ||||||
| 	if omd, ok := ctx.Value(mdIncomingKey{}).(*rawMetadata); ok { |  | ||||||
| 		omd.md = md |  | ||||||
| 		return true |  | ||||||
| 	} |  | ||||||
| 	return false |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewIncomingContext creates a new context with incoming metadata attached |  | ||||||
| func NewIncomingContext(ctx context.Context, md Metadata) context.Context { |  | ||||||
| 	if ctx == nil { |  | ||||||
| 		ctx = context.Background() |  | ||||||
| 	} |  | ||||||
| 	return context.WithValue(ctx, mdIncomingKey{}, &rawMetadata{md}) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewOutgoingContext creates a new context with outcoming metadata attached |  | ||||||
| func NewOutgoingContext(ctx context.Context, md Metadata) context.Context { |  | ||||||
| 	if ctx == nil { |  | ||||||
| 		ctx = context.Background() |  | ||||||
| 	} |  | ||||||
| 	return context.WithValue(ctx, mdOutgoingKey{}, &rawMetadata{md}) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // AppendOutgoingContext apends new md to context |  | ||||||
| func AppendOutgoingContext(ctx context.Context, kv ...string) context.Context { |  | ||||||
| 	md, ok := Pairs(kv...) |  | ||||||
| 	if !ok { |  | ||||||
| 		return ctx |  | ||||||
| 	} |  | ||||||
| 	omd, ok := FromOutgoingContext(ctx) |  | ||||||
| 	if !ok { |  | ||||||
| 		return NewOutgoingContext(ctx, md) |  | ||||||
| 	} |  | ||||||
| 	for k, v := range md { |  | ||||||
| 		omd.Set(k, v) |  | ||||||
| 	} |  | ||||||
| 	return NewOutgoingContext(ctx, omd) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // AppendIncomingContext apends new md to context |  | ||||||
| func AppendIncomingContext(ctx context.Context, kv ...string) context.Context { |  | ||||||
| 	md, ok := Pairs(kv...) |  | ||||||
| 	if !ok { |  | ||||||
| 		return ctx |  | ||||||
| 	} |  | ||||||
| 	omd, ok := FromIncomingContext(ctx) |  | ||||||
| 	if !ok { |  | ||||||
| 		return NewIncomingContext(ctx, md) |  | ||||||
| 	} |  | ||||||
| 	for k, v := range md { |  | ||||||
| 		omd.Set(k, v) |  | ||||||
| 	} |  | ||||||
| 	return NewIncomingContext(ctx, omd) |  | ||||||
| } |  | ||||||
| @@ -1,140 +0,0 @@ | |||||||
| package metadata |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"context" |  | ||||||
| 	"testing" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| func TestFromNilContext(t *testing.T) { |  | ||||||
| 	// nolint: staticcheck |  | ||||||
| 	c, ok := FromContext(nil) |  | ||||||
| 	if ok || c != nil { |  | ||||||
| 		t.Fatal("FromContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestNewNilContext(t *testing.T) { |  | ||||||
| 	// nolint: staticcheck |  | ||||||
| 	ctx := NewContext(nil, New(0)) |  | ||||||
|  |  | ||||||
| 	c, ok := FromContext(ctx) |  | ||||||
| 	if c == nil || !ok { |  | ||||||
| 		t.Fatal("NewContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestFromContext(t *testing.T) { |  | ||||||
| 	ctx := context.WithValue(context.TODO(), mdKey{}, &rawMetadata{New(0)}) |  | ||||||
|  |  | ||||||
| 	c, ok := FromContext(ctx) |  | ||||||
| 	if c == nil || !ok { |  | ||||||
| 		t.Fatal("FromContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestNewContext(t *testing.T) { |  | ||||||
| 	ctx := NewContext(context.TODO(), New(0)) |  | ||||||
|  |  | ||||||
| 	c, ok := FromContext(ctx) |  | ||||||
| 	if c == nil || !ok { |  | ||||||
| 		t.Fatal("NewContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestFromIncomingContext(t *testing.T) { |  | ||||||
| 	ctx := context.WithValue(context.TODO(), mdIncomingKey{}, &rawMetadata{New(0)}) |  | ||||||
|  |  | ||||||
| 	c, ok := FromIncomingContext(ctx) |  | ||||||
| 	if c == nil || !ok { |  | ||||||
| 		t.Fatal("FromIncomingContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestFromOutgoingContext(t *testing.T) { |  | ||||||
| 	ctx := context.WithValue(context.TODO(), mdOutgoingKey{}, &rawMetadata{New(0)}) |  | ||||||
|  |  | ||||||
| 	c, ok := FromOutgoingContext(ctx) |  | ||||||
| 	if c == nil || !ok { |  | ||||||
| 		t.Fatal("FromOutgoingContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestSetIncomingContext(t *testing.T) { |  | ||||||
| 	md := New(1) |  | ||||||
| 	md.Set("key", "val") |  | ||||||
| 	ctx := context.WithValue(context.TODO(), mdIncomingKey{}, &rawMetadata{}) |  | ||||||
| 	if !SetIncomingContext(ctx, md) { |  | ||||||
| 		t.Fatal("SetIncomingContext not works") |  | ||||||
| 	} |  | ||||||
| 	md, ok := FromIncomingContext(ctx) |  | ||||||
| 	if md == nil || !ok { |  | ||||||
| 		t.Fatal("SetIncomingContext not works") |  | ||||||
| 	} else if v, ok := md.Get("key"); !ok || v != "val" { |  | ||||||
| 		t.Fatal("SetIncomingContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestSetOutgoingContext(t *testing.T) { |  | ||||||
| 	md := New(1) |  | ||||||
| 	md.Set("key", "val") |  | ||||||
| 	ctx := context.WithValue(context.TODO(), mdOutgoingKey{}, &rawMetadata{}) |  | ||||||
| 	if !SetOutgoingContext(ctx, md) { |  | ||||||
| 		t.Fatal("SetOutgoingContext not works") |  | ||||||
| 	} |  | ||||||
| 	md, ok := FromOutgoingContext(ctx) |  | ||||||
| 	if md == nil || !ok { |  | ||||||
| 		t.Fatal("SetOutgoingContext not works") |  | ||||||
| 	} else if v, ok := md.Get("key"); !ok || v != "val" { |  | ||||||
| 		t.Fatal("SetOutgoingContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestNewIncomingContext(t *testing.T) { |  | ||||||
| 	md := New(1) |  | ||||||
| 	md.Set("key", "val") |  | ||||||
| 	ctx := NewIncomingContext(context.TODO(), md) |  | ||||||
|  |  | ||||||
| 	c, ok := FromIncomingContext(ctx) |  | ||||||
| 	if c == nil || !ok { |  | ||||||
| 		t.Fatal("NewIncomingContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestNewOutgoingContext(t *testing.T) { |  | ||||||
| 	md := New(1) |  | ||||||
| 	md.Set("key", "val") |  | ||||||
| 	ctx := NewOutgoingContext(context.TODO(), md) |  | ||||||
|  |  | ||||||
| 	c, ok := FromOutgoingContext(ctx) |  | ||||||
| 	if c == nil || !ok { |  | ||||||
| 		t.Fatal("NewOutgoingContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestAppendIncomingContext(t *testing.T) { |  | ||||||
| 	md := New(1) |  | ||||||
| 	md.Set("key1", "val1") |  | ||||||
| 	ctx := AppendIncomingContext(context.TODO(), "key2", "val2") |  | ||||||
|  |  | ||||||
| 	nmd, ok := FromIncomingContext(ctx) |  | ||||||
| 	if nmd == nil || !ok { |  | ||||||
| 		t.Fatal("AppendIncomingContext not works") |  | ||||||
| 	} |  | ||||||
| 	if v, ok := nmd.Get("key2"); !ok || v != "val2" { |  | ||||||
| 		t.Fatal("AppendIncomingContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestAppendOutgoingContext(t *testing.T) { |  | ||||||
| 	md := New(1) |  | ||||||
| 	md.Set("key1", "val1") |  | ||||||
| 	ctx := AppendOutgoingContext(context.TODO(), "key2", "val2") |  | ||||||
|  |  | ||||||
| 	nmd, ok := FromOutgoingContext(ctx) |  | ||||||
| 	if nmd == nil || !ok { |  | ||||||
| 		t.Fatal("AppendOutgoingContext not works") |  | ||||||
| 	} |  | ||||||
| 	if v, ok := nmd.Get("key2"); !ok || v != "val2" { |  | ||||||
| 		t.Fatal("AppendOutgoingContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
							
								
								
									
										19
									
								
								metadata/headers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										19
									
								
								metadata/headers.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,19 @@ | |||||||
|  | // Package metadata is a way of defining message headers | ||||||
|  | package metadata | ||||||
|  |  | ||||||
|  | var ( | ||||||
|  | 	// HeaderTopic is the header name that contains topic name | ||||||
|  | 	HeaderTopic = "Micro-Topic" | ||||||
|  | 	// HeaderContentType specifies content type of message | ||||||
|  | 	HeaderContentType = "Content-Type" | ||||||
|  | 	// HeaderEndpoint specifies endpoint in service | ||||||
|  | 	HeaderEndpoint = "Micro-Endpoint" | ||||||
|  | 	// HeaderService specifies service | ||||||
|  | 	HeaderService = "Micro-Service" | ||||||
|  | 	// HeaderTimeout specifies timeout of operation | ||||||
|  | 	HeaderTimeout = "Micro-Timeout" | ||||||
|  | 	// HeaderAuthorization specifies Authorization header | ||||||
|  | 	HeaderAuthorization = "Authorization" | ||||||
|  | 	// HeaderXRequestID specifies request id | ||||||
|  | 	HeaderXRequestID = "X-Request-Id" | ||||||
|  | ) | ||||||
| @@ -1,40 +1,471 @@ | |||||||
| // Package metadata is a way of defining message headers |  | ||||||
| package metadata | package metadata | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"fmt" | ||||||
| 	"net/textproto" | 	"net/textproto" | ||||||
| 	"sort" | 	"sort" | ||||||
|  | 	"strings" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( |  | ||||||
| 	// HeaderTopic is the header name that contains topic name |  | ||||||
| 	HeaderTopic = "Micro-Topic" |  | ||||||
| 	// HeaderContentType specifies content type of message |  | ||||||
| 	HeaderContentType = "Content-Type" |  | ||||||
| 	// HeaderEndpoint specifies endpoint in service |  | ||||||
| 	HeaderEndpoint = "Micro-Endpoint" |  | ||||||
| 	// HeaderService specifies service |  | ||||||
| 	HeaderService = "Micro-Service" |  | ||||||
| 	// HeaderTimeout specifies timeout of operation |  | ||||||
| 	HeaderTimeout = "Micro-Timeout" |  | ||||||
| 	// HeaderAuthorization specifies Authorization header |  | ||||||
| 	HeaderAuthorization = "Authorization" |  | ||||||
| 	// HeaderXRequestID specifies request id |  | ||||||
| 	HeaderXRequestID = "X-Request-Id" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // Metadata is our way of representing request headers internally. |  | ||||||
| // They're used at the RPC level and translate back and forth |  | ||||||
| // from Transport headers. |  | ||||||
| type Metadata map[string]string |  | ||||||
|  |  | ||||||
| type rawMetadata struct { |  | ||||||
| 	md Metadata |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // defaultMetadataSize used when need to init new Metadata | // defaultMetadataSize used when need to init new Metadata | ||||||
| var defaultMetadataSize = 2 | var defaultMetadataSize = 2 | ||||||
|  |  | ||||||
|  | // Metadata is a mapping from metadata keys to values. Users should use the following | ||||||
|  | // two convenience functions New and Pairs to generate Metadata. | ||||||
|  | type Metadata map[string][]string | ||||||
|  |  | ||||||
|  | // New creates an zero Metadata. | ||||||
|  | func New(l int) Metadata { | ||||||
|  | 	if l == 0 { | ||||||
|  | 		l = defaultMetadataSize | ||||||
|  | 	} | ||||||
|  | 	md := make(Metadata, l) | ||||||
|  | 	return md | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewWithMetadata creates an Metadata from a given key-value map. | ||||||
|  | func NewWithMetadata(m map[string]string) Metadata { | ||||||
|  | 	md := make(Metadata, len(m)) | ||||||
|  | 	for key, val := range m { | ||||||
|  | 		md[key] = append(md[key], val) | ||||||
|  | 	} | ||||||
|  | 	return md | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Pairs returns an Metadata formed by the mapping of key, value ... | ||||||
|  | // Pairs panics if len(kv) is odd. | ||||||
|  | func Pairs(kv ...string) Metadata { | ||||||
|  | 	if len(kv)%2 == 1 { | ||||||
|  | 		panic(fmt.Sprintf("metadata: Pairs got the odd number of input pairs for metadata: %d", len(kv))) | ||||||
|  | 	} | ||||||
|  | 	md := make(Metadata, len(kv)/2) | ||||||
|  | 	for i := 0; i < len(kv); i += 2 { | ||||||
|  | 		md[kv[i]] = append(md[kv[i]], kv[i+1]) | ||||||
|  | 	} | ||||||
|  | 	return md | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Len returns the number of items in Metadata. | ||||||
|  | func (md Metadata) Len() int { | ||||||
|  | 	return len(md) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Copy returns a copy of Metadata. | ||||||
|  | func Copy(src Metadata) Metadata { | ||||||
|  | 	out := make(Metadata, len(src)) | ||||||
|  | 	for k, v := range src { | ||||||
|  | 		out[k] = copyOf(v) | ||||||
|  | 	} | ||||||
|  | 	return out | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Copy returns a copy of Metadata. | ||||||
|  | func (md Metadata) Copy() Metadata { | ||||||
|  | 	out := make(Metadata, len(md)) | ||||||
|  | 	for k, v := range md { | ||||||
|  | 		out[k] = copyOf(v) | ||||||
|  | 	} | ||||||
|  | 	return out | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // AsMap returns a copy of Metadata with map[string]string. | ||||||
|  | func (md Metadata) AsMap() map[string]string { | ||||||
|  | 	out := make(map[string]string, len(md)) | ||||||
|  | 	for k, v := range md { | ||||||
|  | 		out[k] = strings.Join(v, ",") | ||||||
|  | 	} | ||||||
|  | 	return out | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // AsHTTP1 returns a copy of Metadata | ||||||
|  | // with CanonicalMIMEHeaderKey. | ||||||
|  | func (md Metadata) AsHTTP1() map[string][]string { | ||||||
|  | 	out := make(map[string][]string, len(md)) | ||||||
|  | 	for k, v := range md { | ||||||
|  | 		out[textproto.CanonicalMIMEHeaderKey(k)] = copyOf(v) | ||||||
|  | 	} | ||||||
|  | 	return out | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // AsHTTP1 returns a copy of Metadata | ||||||
|  | // with strings.ToLower. | ||||||
|  | func (md Metadata) AsHTTP2() map[string][]string { | ||||||
|  | 	out := make(map[string][]string, len(md)) | ||||||
|  | 	for k, v := range md { | ||||||
|  | 		out[strings.ToLower(k)] = copyOf(v) | ||||||
|  | 	} | ||||||
|  | 	return out | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // CopyTo copies Metadata to out. | ||||||
|  | func (md Metadata) CopyTo(out Metadata) { | ||||||
|  | 	for k, v := range md { | ||||||
|  | 		out[k] = copyOf(v) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Get obtains the values for a given key. | ||||||
|  | func (md Metadata) MustGet(k string) []string { | ||||||
|  | 	v, ok := md.Get(k) | ||||||
|  | 	if !ok { | ||||||
|  | 		panic("missing metadata key") | ||||||
|  | 	} | ||||||
|  | 	return v | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Get obtains the values for a given key. | ||||||
|  | func (md Metadata) Get(k string) ([]string, bool) { | ||||||
|  | 	v, ok := md[k] | ||||||
|  | 	if !ok { | ||||||
|  | 		v, ok = md[strings.ToLower(k)] | ||||||
|  | 	} | ||||||
|  | 	if !ok { | ||||||
|  | 		v, ok = md[textproto.CanonicalMIMEHeaderKey(k)] | ||||||
|  | 	} | ||||||
|  | 	return v, ok | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MustGetJoined obtains the values for a given key | ||||||
|  | // with joined values with "," symbol | ||||||
|  | func (md Metadata) MustGetJoined(k string) string { | ||||||
|  | 	v, ok := md.GetJoined(k) | ||||||
|  | 	if !ok { | ||||||
|  | 		panic("missing metadata key") | ||||||
|  | 	} | ||||||
|  | 	return v | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GetJoined obtains the values for a given key | ||||||
|  | // with joined values with "," symbol | ||||||
|  | func (md Metadata) GetJoined(k string) (string, bool) { | ||||||
|  | 	v, ok := md.Get(k) | ||||||
|  | 	if !ok { | ||||||
|  | 		return "", ok | ||||||
|  | 	} | ||||||
|  | 	return strings.Join(v, ","), true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Set sets the value of a given key with a slice of values. | ||||||
|  | func (md Metadata) Add(key string, vals ...string) { | ||||||
|  | 	if len(vals) == 0 { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	md[key] = vals | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Set sets the value of a given key with a slice of values. | ||||||
|  | func (md Metadata) Set(kvs ...string) { | ||||||
|  | 	if len(kvs)%2 == 1 { | ||||||
|  | 		panic(fmt.Sprintf("metadata: Set got an odd number of input pairs for metadata: %d", len(kvs))) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for i := 0; i < len(kvs); i += 2 { | ||||||
|  | 		md[kvs[i]] = append(md[kvs[i]], kvs[i+1]) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Append adds the values to key k, not overwriting what was already stored at | ||||||
|  | // that key. | ||||||
|  | func (md Metadata) Append(key string, vals ...string) { | ||||||
|  | 	if len(vals) == 0 { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	md[key] = append(md[key], vals...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Del removes the values for a given keys k. | ||||||
|  | func (md Metadata) Del(k ...string) { | ||||||
|  | 	for i := range k { | ||||||
|  | 		delete(md, k[i]) | ||||||
|  | 		delete(md, strings.ToLower(k[i])) | ||||||
|  | 		delete(md, textproto.CanonicalMIMEHeaderKey(k[i])) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Join joins any number of Metadatas into a single Metadata. | ||||||
|  | // | ||||||
|  | // The order of values for each key is determined by the order in which the Metadatas | ||||||
|  | // containing those values are presented to Join. | ||||||
|  | func Join(mds ...Metadata) Metadata { | ||||||
|  | 	out := Metadata{} | ||||||
|  | 	for _, Metadata := range mds { | ||||||
|  | 		for k, v := range Metadata { | ||||||
|  | 			out[k] = append(out[k], v...) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return out | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type ( | ||||||
|  | 	metadataIncomingKey struct{} | ||||||
|  | 	metadataOutgoingKey struct{} | ||||||
|  | 	metadataCurrentKey  struct{} | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // NewContext creates a new context with Metadata attached. Metadata must | ||||||
|  | // not be modified after calling this function. | ||||||
|  | func NewContext(ctx context.Context, md Metadata) context.Context { | ||||||
|  | 	return context.WithValue(ctx, metadataCurrentKey{}, rawMetadata{md: md}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewIncomingContext creates a new context with incoming Metadata attached. Metadata must | ||||||
|  | // not be modified after calling this function. | ||||||
|  | func NewIncomingContext(ctx context.Context, md Metadata) context.Context { | ||||||
|  | 	return context.WithValue(ctx, metadataIncomingKey{}, rawMetadata{md: md}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewOutgoingContext creates a new context with outgoing Metadata attached. If used | ||||||
|  | // in conjunction with AppendOutgoingContext, NewOutgoingContext will | ||||||
|  | // overwrite any previously-appended metadata. Metadata must not be modified after | ||||||
|  | // calling this function. | ||||||
|  | func NewOutgoingContext(ctx context.Context, md Metadata) context.Context { | ||||||
|  | 	return context.WithValue(ctx, metadataOutgoingKey{}, rawMetadata{md: md}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // AppendContext returns a new context with the provided kv merged | ||||||
|  | // with any existing metadata in the context. Please refer to the documentation | ||||||
|  | // of Pairs for a description of kv. | ||||||
|  | func AppendContext(ctx context.Context, kv ...string) context.Context { | ||||||
|  | 	if len(kv)%2 == 1 { | ||||||
|  | 		panic(fmt.Sprintf("metadata: AppendContext got an odd number of input pairs for metadata: %d", len(kv))) | ||||||
|  | 	} | ||||||
|  | 	md, _ := ctx.Value(metadataCurrentKey{}).(rawMetadata) | ||||||
|  | 	added := make([][]string, len(md.added)+1) | ||||||
|  | 	copy(added, md.added) | ||||||
|  | 	kvCopy := make([]string, 0, len(kv)) | ||||||
|  | 	for i := 0; i < len(kv); i += 2 { | ||||||
|  | 		kvCopy = append(kvCopy, strings.ToLower(kv[i]), kv[i+1]) | ||||||
|  | 	} | ||||||
|  | 	added[len(added)-1] = kvCopy | ||||||
|  | 	return context.WithValue(ctx, metadataCurrentKey{}, rawMetadata{md: md.md, added: added}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // AppendIncomingContext returns a new context with the provided kv merged | ||||||
|  | // with any existing metadata in the context. Please refer to the documentation | ||||||
|  | // of Pairs for a description of kv. | ||||||
|  | func AppendIncomingContext(ctx context.Context, kv ...string) context.Context { | ||||||
|  | 	if len(kv)%2 == 1 { | ||||||
|  | 		panic(fmt.Sprintf("metadata: AppendIncomingContext got an odd number of input pairs for metadata: %d", len(kv))) | ||||||
|  | 	} | ||||||
|  | 	md, _ := ctx.Value(metadataIncomingKey{}).(rawMetadata) | ||||||
|  | 	added := make([][]string, len(md.added)+1) | ||||||
|  | 	copy(added, md.added) | ||||||
|  | 	kvCopy := make([]string, 0, len(kv)) | ||||||
|  | 	for i := 0; i < len(kv); i += 2 { | ||||||
|  | 		kvCopy = append(kvCopy, strings.ToLower(kv[i]), kv[i+1]) | ||||||
|  | 	} | ||||||
|  | 	added[len(added)-1] = kvCopy | ||||||
|  | 	return context.WithValue(ctx, metadataIncomingKey{}, rawMetadata{md: md.md, added: added}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // AppendOutgoingContext returns a new context with the provided kv merged | ||||||
|  | // with any existing metadata in the context. Please refer to the documentation | ||||||
|  | // of Pairs for a description of kv. | ||||||
|  | func AppendOutgoingContext(ctx context.Context, kv ...string) context.Context { | ||||||
|  | 	if len(kv)%2 == 1 { | ||||||
|  | 		panic(fmt.Sprintf("metadata: AppendOutgoingContext got an odd number of input pairs for metadata: %d", len(kv))) | ||||||
|  | 	} | ||||||
|  | 	md, _ := ctx.Value(metadataOutgoingKey{}).(rawMetadata) | ||||||
|  | 	added := make([][]string, len(md.added)+1) | ||||||
|  | 	copy(added, md.added) | ||||||
|  | 	kvCopy := make([]string, 0, len(kv)) | ||||||
|  | 	for i := 0; i < len(kv); i += 2 { | ||||||
|  | 		kvCopy = append(kvCopy, strings.ToLower(kv[i]), kv[i+1]) | ||||||
|  | 	} | ||||||
|  | 	added[len(added)-1] = kvCopy | ||||||
|  | 	return context.WithValue(ctx, metadataOutgoingKey{}, rawMetadata{md: md.md, added: added}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // FromContext returns the metadata in ctx if it exists. | ||||||
|  | func FromContext(ctx context.Context) (Metadata, bool) { | ||||||
|  | 	raw, ok := ctx.Value(metadataCurrentKey{}).(rawMetadata) | ||||||
|  | 	if !ok { | ||||||
|  | 		return nil, false | ||||||
|  | 	} | ||||||
|  | 	metadataSize := len(raw.md) | ||||||
|  | 	for i := range raw.added { | ||||||
|  | 		metadataSize += len(raw.added[i]) / 2 | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	out := make(Metadata, metadataSize) | ||||||
|  | 	for k, v := range raw.md { | ||||||
|  | 		out[k] = copyOf(v) | ||||||
|  | 	} | ||||||
|  | 	for _, added := range raw.added { | ||||||
|  | 		if len(added)%2 == 1 { | ||||||
|  | 			panic(fmt.Sprintf("metadata: FromContext got an odd number of input pairs for metadata: %d", len(added))) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		for i := 0; i < len(added); i += 2 { | ||||||
|  | 			out[added[i]] = append(out[added[i]], added[i+1]) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return out, true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MustContext returns the metadata in ctx. | ||||||
|  | func MustContext(ctx context.Context) Metadata { | ||||||
|  | 	md, ok := FromContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		panic("missing metadata") | ||||||
|  | 	} | ||||||
|  | 	return md | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // FromIncomingContext returns the incoming metadata in ctx if it exists. | ||||||
|  | func FromIncomingContext(ctx context.Context) (Metadata, bool) { | ||||||
|  | 	raw, ok := ctx.Value(metadataIncomingKey{}).(rawMetadata) | ||||||
|  | 	if !ok { | ||||||
|  | 		return nil, false | ||||||
|  | 	} | ||||||
|  | 	metadataSize := len(raw.md) | ||||||
|  | 	for i := range raw.added { | ||||||
|  | 		metadataSize += len(raw.added[i]) / 2 | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	out := make(Metadata, metadataSize) | ||||||
|  | 	for k, v := range raw.md { | ||||||
|  | 		out[k] = copyOf(v) | ||||||
|  | 	} | ||||||
|  | 	for _, added := range raw.added { | ||||||
|  | 		if len(added)%2 == 1 { | ||||||
|  | 			panic(fmt.Sprintf("metadata: FromIncomingContext got an odd number of input pairs for metadata: %d", len(added))) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		for i := 0; i < len(added); i += 2 { | ||||||
|  | 			out[added[i]] = append(out[added[i]], added[i+1]) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return out, true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MustIncomingContext returns the incoming metadata in ctx. | ||||||
|  | func MustIncomingContext(ctx context.Context) Metadata { | ||||||
|  | 	md, ok := FromIncomingContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		panic("missing metadata") | ||||||
|  | 	} | ||||||
|  | 	return md | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ValueFromIncomingContext returns the metadata value corresponding to the metadata | ||||||
|  | // key from the incoming metadata if it exists. Keys are matched in a case insensitive | ||||||
|  | // manner. | ||||||
|  | func ValueFromIncomingContext(ctx context.Context, key string) []string { | ||||||
|  | 	raw, ok := ctx.Value(metadataIncomingKey{}).(rawMetadata) | ||||||
|  | 	if !ok { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if v, ok := raw.md[key]; ok { | ||||||
|  | 		return copyOf(v) | ||||||
|  | 	} | ||||||
|  | 	for k, v := range raw.md { | ||||||
|  | 		// Case insensitive comparison: Metadata is a map, and there's no guarantee | ||||||
|  | 		// that the Metadata attached to the context is created using our helper | ||||||
|  | 		// functions. | ||||||
|  | 		if strings.EqualFold(k, key) { | ||||||
|  | 			return copyOf(v) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ValueFromCurrentContext returns the metadata value corresponding to the metadata | ||||||
|  | // key from the incoming metadata if it exists. Keys are matched in a case insensitive | ||||||
|  | // manner. | ||||||
|  | func ValueFromCurrentContext(ctx context.Context, key string) []string { | ||||||
|  | 	md, ok := ctx.Value(metadataCurrentKey{}).(rawMetadata) | ||||||
|  | 	if !ok { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if v, ok := md.md[key]; ok { | ||||||
|  | 		return copyOf(v) | ||||||
|  | 	} | ||||||
|  | 	for k, v := range md.md { | ||||||
|  | 		// Case insensitive comparison: Metadata is a map, and there's no guarantee | ||||||
|  | 		// that the Metadata attached to the context is created using our helper | ||||||
|  | 		// functions. | ||||||
|  | 		if strings.EqualFold(k, key) { | ||||||
|  | 			return copyOf(v) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MustOutgoingContext returns the outgoing metadata in ctx. | ||||||
|  | func MustOutgoingContext(ctx context.Context) Metadata { | ||||||
|  | 	md, ok := FromOutgoingContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		panic("missing metadata") | ||||||
|  | 	} | ||||||
|  | 	return md | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ValueFromOutgoingContext returns the metadata value corresponding to the metadata | ||||||
|  | // key from the incoming metadata if it exists. Keys are matched in a case insensitive | ||||||
|  | // manner. | ||||||
|  | func ValueFromOutgoingContext(ctx context.Context, key string) []string { | ||||||
|  | 	md, ok := ctx.Value(metadataOutgoingKey{}).(rawMetadata) | ||||||
|  | 	if !ok { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if v, ok := md.md[key]; ok { | ||||||
|  | 		return copyOf(v) | ||||||
|  | 	} | ||||||
|  | 	for k, v := range md.md { | ||||||
|  | 		// Case insensitive comparison: Metadata is a map, and there's no guarantee | ||||||
|  | 		// that the Metadata attached to the context is created using our helper | ||||||
|  | 		// functions. | ||||||
|  | 		if strings.EqualFold(k, key) { | ||||||
|  | 			return copyOf(v) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func copyOf(v []string) []string { | ||||||
|  | 	vals := make([]string, len(v)) | ||||||
|  | 	copy(vals, v) | ||||||
|  | 	return vals | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // FromOutgoingContext returns the outgoing metadata in ctx if it exists. | ||||||
|  | func FromOutgoingContext(ctx context.Context) (Metadata, bool) { | ||||||
|  | 	raw, ok := ctx.Value(metadataOutgoingKey{}).(rawMetadata) | ||||||
|  | 	if !ok { | ||||||
|  | 		return nil, false | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	metadataSize := len(raw.md) | ||||||
|  | 	for i := range raw.added { | ||||||
|  | 		metadataSize += len(raw.added[i]) / 2 | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	out := make(Metadata, metadataSize) | ||||||
|  | 	for k, v := range raw.md { | ||||||
|  | 		out[k] = copyOf(v) | ||||||
|  | 	} | ||||||
|  | 	for _, added := range raw.added { | ||||||
|  | 		if len(added)%2 == 1 { | ||||||
|  | 			panic(fmt.Sprintf("metadata: FromOutgoingContext got an odd number of input pairs for metadata: %d", len(added))) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		for i := 0; i < len(added); i += 2 { | ||||||
|  | 			out[added[i]] = append(out[added[i]], added[i+1]) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return out, ok | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type rawMetadata struct { | ||||||
|  | 	md    Metadata | ||||||
|  | 	added [][]string | ||||||
|  | } | ||||||
|  |  | ||||||
| // Iterator used to iterate over metadata with order | // Iterator used to iterate over metadata with order | ||||||
| type Iterator struct { | type Iterator struct { | ||||||
| 	md   Metadata | 	md   Metadata | ||||||
| @@ -44,14 +475,18 @@ type Iterator struct { | |||||||
| } | } | ||||||
|  |  | ||||||
| // Next advance iterator to next element | // Next advance iterator to next element | ||||||
| func (iter *Iterator) Next(k, v *string) bool { | func (iter *Iterator) Next(k *string, v *[]string) bool { | ||||||
| 	if iter.cur+1 > iter.cnt { | 	if iter.cur+1 > iter.cnt { | ||||||
| 		return false | 		return false | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	if k != nil && v != nil { | ||||||
| 		*k = iter.keys[iter.cur] | 		*k = iter.keys[iter.cur] | ||||||
| 	*v = iter.md[*k] | 		vv := iter.md[*k] | ||||||
|  | 		*v = make([]string, len(vv)) | ||||||
|  | 		copy(*v, vv) | ||||||
| 		iter.cur++ | 		iter.cur++ | ||||||
|  | 	} | ||||||
| 	return true | 	return true | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -65,80 +500,3 @@ func (md Metadata) Iterator() *Iterator { | |||||||
| 	sort.Strings(iter.keys) | 	sort.Strings(iter.keys) | ||||||
| 	return iter | 	return iter | ||||||
| } | } | ||||||
|  |  | ||||||
| // Get returns value from metadata by key |  | ||||||
| func (md Metadata) Get(key string) (string, bool) { |  | ||||||
| 	// fast path |  | ||||||
| 	val, ok := md[key] |  | ||||||
| 	if !ok { |  | ||||||
| 		// slow path |  | ||||||
| 		val, ok = md[textproto.CanonicalMIMEHeaderKey(key)] |  | ||||||
| 	} |  | ||||||
| 	return val, ok |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Set is used to store value in metadata |  | ||||||
| func (md Metadata) Set(kv ...string) { |  | ||||||
| 	if len(kv)%2 == 1 { |  | ||||||
| 		kv = kv[:len(kv)-1] |  | ||||||
| 	} |  | ||||||
| 	for idx := 0; idx < len(kv); idx += 2 { |  | ||||||
| 		md[textproto.CanonicalMIMEHeaderKey(kv[idx])] = kv[idx+1] |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Del is used to remove value from metadata |  | ||||||
| func (md Metadata) Del(keys ...string) { |  | ||||||
| 	for _, key := range keys { |  | ||||||
| 		// fast path |  | ||||||
| 		delete(md, key) |  | ||||||
| 		// slow path |  | ||||||
| 		delete(md, textproto.CanonicalMIMEHeaderKey(key)) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Copy makes a copy of the metadata |  | ||||||
| func Copy(md Metadata, exclude ...string) Metadata { |  | ||||||
| 	nmd := New(len(md)) |  | ||||||
| 	for key, val := range md { |  | ||||||
| 		nmd.Set(key, val) |  | ||||||
| 	} |  | ||||||
| 	nmd.Del(exclude...) |  | ||||||
| 	return nmd |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // New return new sized metadata |  | ||||||
| func New(size int) Metadata { |  | ||||||
| 	if size == 0 { |  | ||||||
| 		size = defaultMetadataSize |  | ||||||
| 	} |  | ||||||
| 	return make(Metadata, size) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Merge merges metadata to existing metadata, overwriting if specified |  | ||||||
| func Merge(omd Metadata, mmd Metadata, overwrite bool) Metadata { |  | ||||||
| 	var ok bool |  | ||||||
| 	nmd := Copy(omd) |  | ||||||
| 	for key, val := range mmd { |  | ||||||
| 		_, ok = nmd[key] |  | ||||||
| 		switch { |  | ||||||
| 		case ok && !overwrite: |  | ||||||
| 			continue |  | ||||||
| 		case val != "": |  | ||||||
| 			nmd.Set(key, val) |  | ||||||
| 		case ok && val == "": |  | ||||||
| 			nmd.Del(key) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return nmd |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Pairs from which metadata created |  | ||||||
| func Pairs(kv ...string) (Metadata, bool) { |  | ||||||
| 	if len(kv)%2 == 1 { |  | ||||||
| 		return nil, false |  | ||||||
| 	} |  | ||||||
| 	md := New(len(kv) / 2) |  | ||||||
| 	md.Set(kv...) |  | ||||||
| 	return md, true |  | ||||||
| } |  | ||||||
|   | |||||||
| @@ -5,6 +5,26 @@ import ( | |||||||
| 	"testing" | 	"testing" | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | /* | ||||||
|  | func TestAppendOutgoingContextModify(t *testing.T) { | ||||||
|  | 	md := Pairs("key1", "val1") | ||||||
|  | 	ctx := NewOutgoingContext(context.TODO(), md) | ||||||
|  | 	nctx := AppendOutgoingContext(ctx, "key1", "val3", "key2", "val2") | ||||||
|  | 	_ = nctx | ||||||
|  | 	omd := MustOutgoingContext(nctx) | ||||||
|  | 	fmt.Printf("%#+v\n", omd) | ||||||
|  | } | ||||||
|  | */ | ||||||
|  |  | ||||||
|  | func TestLowercase(t *testing.T) { | ||||||
|  | 	md := New(1) | ||||||
|  | 	md["x-request-id"] = []string{"12345"} | ||||||
|  | 	v, ok := md.GetJoined("X-Request-Id") | ||||||
|  | 	if !ok || v == "" { | ||||||
|  | 		t.Fatalf("metadata invalid %#+v", md) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| func TestMultipleUsage(t *testing.T) { | func TestMultipleUsage(t *testing.T) { | ||||||
| 	ctx := context.TODO() | 	ctx := context.TODO() | ||||||
| 	md := New(0) | 	md := New(0) | ||||||
| @@ -23,21 +43,18 @@ func TestMultipleUsage(t *testing.T) { | |||||||
| 		m.Del("key1_3") | 		m.Del("key1_3") | ||||||
| 		return ctx | 		return ctx | ||||||
| 	}(ctx) | 	}(ctx) | ||||||
| 	t.Logf("imd %#+v", imd) | 	_ = imd | ||||||
| 	t.Logf("omd %#+v", omd) | 	_ = 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") | ||||||
|  |  | ||||||
| 	if v, ok := md.Get("key1"); !ok || v != "val1" { | 	if v, ok := md.GetJoined("key1"); !ok || v != "val1" { | ||||||
| 		t.Fatalf("invalid kv %#+v", md) | 		t.Fatalf("invalid kv %#+v", md) | ||||||
| 	} | 	} | ||||||
| 	if v, ok := md.Get("key2"); !ok || v != "val2" { | 	if v, ok := md.GetJoined("key2"); !ok || v != "val2" { | ||||||
| 		t.Fatalf("invalid kv %#+v", md) |  | ||||||
| 	} |  | ||||||
| 	if _, ok := md.Get("key3"); ok { |  | ||||||
| 		t.Fatalf("invalid kv %#+v", md) | 		t.Fatalf("invalid kv %#+v", md) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| @@ -55,22 +72,12 @@ func TestAppend(t *testing.T) { | |||||||
| } | } | ||||||
|  |  | ||||||
| func TestPairs(t *testing.T) { | func TestPairs(t *testing.T) { | ||||||
| 	md, ok := Pairs("key1", "val1", "key2", "val2") | 	md := Pairs("key1", "val1", "key2", "val2") | ||||||
| 	if !ok { | 	if _, ok := md.Get("key1"); !ok { | ||||||
| 		t.Fatal("odd number of kv") |  | ||||||
| 	} |  | ||||||
| 	if _, ok = md.Get("key1"); !ok { |  | ||||||
| 		t.Fatal("key1 not found") | 		t.Fatal("key1 not found") | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func testCtx(ctx context.Context) { |  | ||||||
| 	md := New(2) |  | ||||||
| 	md.Set("Key1", "Val1_new") |  | ||||||
| 	md.Set("Key3", "Val3") |  | ||||||
| 	SetOutgoingContext(ctx, md) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestPassing(t *testing.T) { | func TestPassing(t *testing.T) { | ||||||
| 	ctx := context.TODO() | 	ctx := context.TODO() | ||||||
| 	md1 := New(2) | 	md1 := New(2) | ||||||
| @@ -78,69 +85,62 @@ func TestPassing(t *testing.T) { | |||||||
| 	md1.Set("Key2", "Val2") | 	md1.Set("Key2", "Val2") | ||||||
|  |  | ||||||
| 	ctx = NewIncomingContext(ctx, md1) | 	ctx = NewIncomingContext(ctx, md1) | ||||||
| 	testCtx(ctx) |  | ||||||
| 	md, ok := FromOutgoingContext(ctx) | 	_, ok := FromOutgoingContext(ctx) | ||||||
| 	if ok { | 	if ok { | ||||||
| 		t.Fatalf("create outgoing context") | 		t.Fatalf("create outgoing context") | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	ctx = NewOutgoingContext(ctx, New(1)) | 	ctx = NewOutgoingContext(ctx, md1) | ||||||
| 	testCtx(ctx) |  | ||||||
| 	md, ok = FromOutgoingContext(ctx) | 	md, ok := FromOutgoingContext(ctx) | ||||||
| 	if !ok { | 	if !ok { | ||||||
| 		t.Fatalf("missing metadata from outgoing context") | 		t.Fatalf("missing metadata from outgoing context") | ||||||
| 	} | 	} | ||||||
| 	if v, ok := md.Get("Key1"); !ok || v != "Val1_new" { | 	if v, ok := md.Get("Key1"); !ok || v[0] != "Val1" { | ||||||
| 		t.Fatalf("invalid metadata value %#+v", md) | 		t.Fatalf("invalid metadata value %#+v", md) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func TestMerge(t *testing.T) { |  | ||||||
| 	omd := Metadata{ |  | ||||||
| 		"key1": "val1", |  | ||||||
| 	} |  | ||||||
| 	mmd := Metadata{ |  | ||||||
| 		"key2": "val2", |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	nmd := Merge(omd, mmd, true) |  | ||||||
| 	if len(nmd) != 2 { |  | ||||||
| 		t.Fatalf("merge failed: %v", nmd) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestIterator(t *testing.T) { | func TestIterator(t *testing.T) { | ||||||
| 	md := Metadata{ | 	md := Pairs( | ||||||
| 		"1Last":   "last", | 		"1Last", "last", | ||||||
| 		"2First":  "first", | 		"2First", "first", | ||||||
| 		"3Second": "second", | 		"3Second", "second", | ||||||
| 	} | 	) | ||||||
|  |  | ||||||
| 	iter := md.Iterator() | 	iter := md.Iterator() | ||||||
| 	var k, v string | 	var k string | ||||||
|  | 	var v []string | ||||||
|  | 	chk := New(3) | ||||||
| 	for iter.Next(&k, &v) { | 	for iter.Next(&k, &v) { | ||||||
| 		// fmt.Printf("k: %s, v: %s\n", k, v) | 		chk[k] = v | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for k, v := range chk { | ||||||
|  | 		if cv, ok := md[k]; !ok || len(cv) != len(v) || cv[0] != v[0] { | ||||||
|  | 			t.Fatalf("XXXX %#+v %#+v", chk, md) | ||||||
|  | 		} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func TestMedataCanonicalKey(t *testing.T) { | func TestMedataCanonicalKey(t *testing.T) { | ||||||
| 	md := New(1) | 	md := New(1) | ||||||
| 	md.Set("x-request-id", "12345") | 	md.Set("x-request-id", "12345") | ||||||
| 	v, ok := md.Get("x-request-id") | 	v, ok := md.GetJoined("x-request-id") | ||||||
| 	if !ok { | 	if !ok { | ||||||
| 		t.Fatalf("failed to get x-request-id") | 		t.Fatalf("failed to get x-request-id") | ||||||
| 	} else if v != "12345" { | 	} else if v != "12345" { | ||||||
| 		t.Fatalf("invalid metadata value: %s != %s", "12345", v) | 		t.Fatalf("invalid metadata value: %s != %s", "12345", v) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	v, ok = md.Get("X-Request-Id") | 	v, ok = md.GetJoined("X-Request-Id") | ||||||
| 	if !ok { | 	if !ok { | ||||||
| 		t.Fatalf("failed to get x-request-id") | 		t.Fatalf("failed to get x-request-id") | ||||||
| 	} else if v != "12345" { | 	} else if v != "12345" { | ||||||
| 		t.Fatalf("invalid metadata value: %s != %s", "12345", v) | 		t.Fatalf("invalid metadata value: %s != %s", "12345", v) | ||||||
| 	} | 	} | ||||||
| 	v, ok = md.Get("X-Request-ID") | 	v, ok = md.GetJoined("X-Request-ID") | ||||||
| 	if !ok { | 	if !ok { | ||||||
| 		t.Fatalf("failed to get x-request-id") | 		t.Fatalf("failed to get x-request-id") | ||||||
| 	} else if v != "12345" { | 	} else if v != "12345" { | ||||||
| @@ -153,7 +153,7 @@ func TestMetadataSet(t *testing.T) { | |||||||
|  |  | ||||||
| 	md.Set("Key", "val") | 	md.Set("Key", "val") | ||||||
|  |  | ||||||
| 	val, ok := md.Get("Key") | 	val, ok := md.GetJoined("Key") | ||||||
| 	if !ok { | 	if !ok { | ||||||
| 		t.Fatal("key Key not found") | 		t.Fatal("key Key not found") | ||||||
| 	} | 	} | ||||||
| @@ -164,8 +164,8 @@ func TestMetadataSet(t *testing.T) { | |||||||
|  |  | ||||||
| func TestMetadataDelete(t *testing.T) { | func TestMetadataDelete(t *testing.T) { | ||||||
| 	md := Metadata{ | 	md := Metadata{ | ||||||
| 		"Foo": "bar", | 		"Foo": []string{"bar"}, | ||||||
| 		"Baz": "empty", | 		"Baz": []string{"empty"}, | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	md.Del("Baz") | 	md.Del("Baz") | ||||||
| @@ -175,25 +175,16 @@ func TestMetadataDelete(t *testing.T) { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func TestNilContext(t *testing.T) { |  | ||||||
| 	var ctx context.Context |  | ||||||
|  |  | ||||||
| 	_, ok := FromContext(ctx) |  | ||||||
| 	if ok { |  | ||||||
| 		t.Fatal("nil context") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestMetadataCopy(t *testing.T) { | func TestMetadataCopy(t *testing.T) { | ||||||
| 	md := Metadata{ | 	md := Metadata{ | ||||||
| 		"Foo": "bar", | 		"Foo": []string{"bar"}, | ||||||
| 		"Bar": "baz", | 		"Bar": []string{"baz"}, | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	cp := Copy(md) | 	cp := Copy(md) | ||||||
|  |  | ||||||
| 	for k, v := range md { | 	for k, v := range md { | ||||||
| 		if cv := cp[k]; cv != v { | 		if cv := cp[k]; cv[0] != v[0] { | ||||||
| 			t.Fatalf("Got %s:%s for %s:%s", k, cv, k, v) | 			t.Fatalf("Got %s:%s for %s:%s", k, cv, k, v) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| @@ -201,7 +192,7 @@ func TestMetadataCopy(t *testing.T) { | |||||||
|  |  | ||||||
| func TestMetadataContext(t *testing.T) { | func TestMetadataContext(t *testing.T) { | ||||||
| 	md := Metadata{ | 	md := Metadata{ | ||||||
| 		"Foo": "bar", | 		"Foo": []string{"bar"}, | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	ctx := NewContext(context.TODO(), md) | 	ctx := NewContext(context.TODO(), md) | ||||||
| @@ -211,7 +202,7 @@ func TestMetadataContext(t *testing.T) { | |||||||
| 		t.Errorf("Unexpected error retrieving metadata, got %t", ok) | 		t.Errorf("Unexpected error retrieving metadata, got %t", ok) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if emd["Foo"] != md["Foo"] { | 	if emd["Foo"][0] != md["Foo"][0] { | ||||||
| 		t.Errorf("Expected key: %s val: %s, got key: %s val: %s", "Foo", md["Foo"], "Foo", emd["Foo"]) | 		t.Errorf("Expected key: %s val: %s, got key: %s val: %s", "Foo", md["Foo"], "Foo", emd["Foo"]) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -220,13 +211,88 @@ func TestMetadataContext(t *testing.T) { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func TestCopy(t *testing.T) { | func TestFromContext(t *testing.T) { | ||||||
| 	md := New(2) | 	ctx := context.WithValue(context.TODO(), metadataCurrentKey{}, rawMetadata{md: New(0)}) | ||||||
| 	md.Set("key1", "val1", "key2", "val2") |  | ||||||
| 	nmd := Copy(md, "key2") | 	c, ok := FromContext(ctx) | ||||||
| 	if len(nmd) != 1 { | 	if c == nil || !ok { | ||||||
| 		t.Fatal("Copy exclude not works") | 		t.Fatal("FromContext not works") | ||||||
| 	} else if nmd["Key1"] != "val1" { | 	} | ||||||
| 		t.Fatal("Copy exclude not works") | } | ||||||
|  |  | ||||||
|  | func TestNewContext(t *testing.T) { | ||||||
|  | 	ctx := NewContext(context.TODO(), New(0)) | ||||||
|  |  | ||||||
|  | 	c, ok := FromContext(ctx) | ||||||
|  | 	if c == nil || !ok { | ||||||
|  | 		t.Fatal("NewContext not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestFromIncomingContext(t *testing.T) { | ||||||
|  | 	ctx := context.WithValue(context.TODO(), metadataIncomingKey{}, rawMetadata{md: New(0)}) | ||||||
|  |  | ||||||
|  | 	c, ok := FromIncomingContext(ctx) | ||||||
|  | 	if c == nil || !ok { | ||||||
|  | 		t.Fatal("FromIncomingContext not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestFromOutgoingContext(t *testing.T) { | ||||||
|  | 	ctx := context.WithValue(context.TODO(), metadataOutgoingKey{}, rawMetadata{md: New(0)}) | ||||||
|  |  | ||||||
|  | 	c, ok := FromOutgoingContext(ctx) | ||||||
|  | 	if c == nil || !ok { | ||||||
|  | 		t.Fatal("FromOutgoingContext not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestNewIncomingContext(t *testing.T) { | ||||||
|  | 	md := New(1) | ||||||
|  | 	md.Set("key", "val") | ||||||
|  | 	ctx := NewIncomingContext(context.TODO(), md) | ||||||
|  |  | ||||||
|  | 	c, ok := FromIncomingContext(ctx) | ||||||
|  | 	if c == nil || !ok { | ||||||
|  | 		t.Fatal("NewIncomingContext not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestNewOutgoingContext(t *testing.T) { | ||||||
|  | 	md := New(1) | ||||||
|  | 	md.Set("key", "val") | ||||||
|  | 	ctx := NewOutgoingContext(context.TODO(), md) | ||||||
|  |  | ||||||
|  | 	c, ok := FromOutgoingContext(ctx) | ||||||
|  | 	if c == nil || !ok { | ||||||
|  | 		t.Fatal("NewOutgoingContext not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestAppendIncomingContext(t *testing.T) { | ||||||
|  | 	md := New(1) | ||||||
|  | 	md.Set("key1", "val1") | ||||||
|  | 	ctx := AppendIncomingContext(context.TODO(), "key2", "val2") | ||||||
|  |  | ||||||
|  | 	nmd, ok := FromIncomingContext(ctx) | ||||||
|  | 	if nmd == nil || !ok { | ||||||
|  | 		t.Fatal("AppendIncomingContext not works") | ||||||
|  | 	} | ||||||
|  | 	if v, ok := nmd.GetJoined("key2"); !ok || v != "val2" { | ||||||
|  | 		t.Fatal("AppendIncomingContext not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestAppendOutgoingContext(t *testing.T) { | ||||||
|  | 	md := New(1) | ||||||
|  | 	md.Set("key1", "val1") | ||||||
|  | 	ctx := AppendOutgoingContext(context.TODO(), "key2", "val2") | ||||||
|  |  | ||||||
|  | 	nmd, ok := FromOutgoingContext(ctx) | ||||||
|  | 	if nmd == nil || !ok { | ||||||
|  | 		t.Fatal("AppendOutgoingContext not works") | ||||||
|  | 	} | ||||||
|  | 	if v, ok := nmd.GetJoined("key2"); !ok || v != "val2" { | ||||||
|  | 		t.Fatal("AppendOutgoingContext not works") | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -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 { | ||||||
|   | |||||||
| @@ -11,7 +11,7 @@ import ( | |||||||
|  |  | ||||||
| var ( | var ( | ||||||
| 	// DefaultMeter is the default meter | 	// DefaultMeter is the default meter | ||||||
| 	DefaultMeter Meter = NewMeter() | 	DefaultMeter = NewMeter() | ||||||
| 	// DefaultAddress data will be made available on this host:port | 	// DefaultAddress data will be made available on this host:port | ||||||
| 	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 | ||||||
|   | |||||||
| @@ -37,32 +37,32 @@ func (r *noopMeter) Init(opts ...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} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -77,7 +77,7 @@ func (r *noopMeter) Set(opts ...Option) Meter { | |||||||
| 	return m | 	return m | ||||||
| } | } | ||||||
|  |  | ||||||
| func (r *noopMeter) Write(w io.Writer, opts ...Option) error { | func (r *noopMeter) Write(_ io.Writer, _ ...Option) error { | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
|   | |||||||
							
								
								
									
										66
									
								
								micro.go
									
									
									
									
									
								
							
							
						
						
									
										66
									
								
								micro.go
									
									
									
									
									
								
							| @@ -3,21 +3,21 @@ package micro | |||||||
| import ( | import ( | ||||||
| 	"reflect" | 	"reflect" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/broker" | 	"go.unistack.org/micro/v4/broker" | ||||||
| 	"go.unistack.org/micro/v3/client" | 	"go.unistack.org/micro/v4/client" | ||||||
| 	"go.unistack.org/micro/v3/codec" | 	"go.unistack.org/micro/v4/codec" | ||||||
| 	"go.unistack.org/micro/v3/flow" | 	"go.unistack.org/micro/v4/flow" | ||||||
| 	"go.unistack.org/micro/v3/fsm" | 	"go.unistack.org/micro/v4/fsm" | ||||||
| 	"go.unistack.org/micro/v3/logger" | 	"go.unistack.org/micro/v4/logger" | ||||||
| 	"go.unistack.org/micro/v3/meter" | 	"go.unistack.org/micro/v4/meter" | ||||||
| 	"go.unistack.org/micro/v3/register" | 	"go.unistack.org/micro/v4/register" | ||||||
| 	"go.unistack.org/micro/v3/resolver" | 	"go.unistack.org/micro/v4/resolver" | ||||||
| 	"go.unistack.org/micro/v3/router" | 	"go.unistack.org/micro/v4/router" | ||||||
| 	"go.unistack.org/micro/v3/selector" | 	"go.unistack.org/micro/v4/selector" | ||||||
| 	"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/sync" | 	"go.unistack.org/micro/v4/sync" | ||||||
| 	"go.unistack.org/micro/v3/tracer" | 	"go.unistack.org/micro/v4/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| func As(b any, target any) bool { | func As(b any, target any) bool { | ||||||
| @@ -65,6 +65,8 @@ func As(b any, target any) bool { | |||||||
| 			break | 			break | ||||||
| 		case targetType.Implements(routerType): | 		case targetType.Implements(routerType): | ||||||
| 			break | 			break | ||||||
|  | 		case targetType.Implements(tracerType): | ||||||
|  | 			break | ||||||
| 		default: | 		default: | ||||||
| 			return false | 			return false | ||||||
| 		} | 		} | ||||||
| @@ -76,19 +78,21 @@ func As(b any, target any) bool { | |||||||
| 	return false | 	return false | ||||||
| } | } | ||||||
|  |  | ||||||
| var brokerType = reflect.TypeOf((*broker.Broker)(nil)).Elem() | var ( | ||||||
| var loggerType = reflect.TypeOf((*logger.Logger)(nil)).Elem() | 	brokerType   = reflect.TypeOf((*broker.Broker)(nil)).Elem() | ||||||
| var clientType = reflect.TypeOf((*client.Client)(nil)).Elem() | 	loggerType   = reflect.TypeOf((*logger.Logger)(nil)).Elem() | ||||||
| var serverType = reflect.TypeOf((*server.Server)(nil)).Elem() | 	clientType   = reflect.TypeOf((*client.Client)(nil)).Elem() | ||||||
| var codecType = reflect.TypeOf((*codec.Codec)(nil)).Elem() | 	serverType   = reflect.TypeOf((*server.Server)(nil)).Elem() | ||||||
| var flowType = reflect.TypeOf((*flow.Flow)(nil)).Elem() | 	codecType    = reflect.TypeOf((*codec.Codec)(nil)).Elem() | ||||||
| var fsmType = reflect.TypeOf((*fsm.FSM)(nil)).Elem() | 	flowType     = reflect.TypeOf((*flow.Flow)(nil)).Elem() | ||||||
| var meterType = reflect.TypeOf((*meter.Meter)(nil)).Elem() | 	fsmType      = reflect.TypeOf((*fsm.FSM)(nil)).Elem() | ||||||
| var registerType = reflect.TypeOf((*register.Register)(nil)).Elem() | 	meterType    = reflect.TypeOf((*meter.Meter)(nil)).Elem() | ||||||
| var resolverType = reflect.TypeOf((*resolver.Resolver)(nil)).Elem() | 	registerType = reflect.TypeOf((*register.Register)(nil)).Elem() | ||||||
| var routerType = reflect.TypeOf((*router.Router)(nil)).Elem() | 	resolverType = reflect.TypeOf((*resolver.Resolver)(nil)).Elem() | ||||||
| var selectorType = reflect.TypeOf((*selector.Selector)(nil)).Elem() | 	routerType   = reflect.TypeOf((*router.Router)(nil)).Elem() | ||||||
| var storeType = reflect.TypeOf((*store.Store)(nil)).Elem() | 	selectorType = reflect.TypeOf((*selector.Selector)(nil)).Elem() | ||||||
| var syncType = reflect.TypeOf((*sync.Sync)(nil)).Elem() | 	storeType    = reflect.TypeOf((*store.Store)(nil)).Elem() | ||||||
| var tracerType = reflect.TypeOf((*tracer.Tracer)(nil)).Elem() | 	syncType     = reflect.TypeOf((*sync.Sync)(nil)).Elem() | ||||||
| var serviceType = reflect.TypeOf((*Service)(nil)).Elem() | 	tracerType   = reflect.TypeOf((*tracer.Tracer)(nil)).Elem() | ||||||
|  | 	serviceType  = reflect.TypeOf((*Service)(nil)).Elem() | ||||||
|  | ) | ||||||
|   | |||||||
| @@ -6,8 +6,9 @@ import ( | |||||||
| 	"reflect" | 	"reflect" | ||||||
| 	"testing" | 	"testing" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/broker" | 	"go.unistack.org/micro/v4/broker" | ||||||
| 	"go.unistack.org/micro/v3/fsm" | 	"go.unistack.org/micro/v4/fsm" | ||||||
|  | 	"go.unistack.org/micro/v4/metadata" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| func TestAs(t *testing.T) { | func TestAs(t *testing.T) { | ||||||
| @@ -18,26 +19,27 @@ func TestAs(t *testing.T) { | |||||||
| 	testCases := []struct { | 	testCases := []struct { | ||||||
| 		b      any | 		b      any | ||||||
| 		target any | 		target any | ||||||
| 		match  bool |  | ||||||
| 		want   any | 		want   any | ||||||
|  |  | ||||||
|  | 		match bool | ||||||
| 	}{ | 	}{ | ||||||
| 		{ | 		{ | ||||||
| 			broTarget, | 			b:      broTarget, | ||||||
| 			&b, | 			target: &b, | ||||||
| 			true, | 			match:  true, | ||||||
| 			broTarget, | 			want:   broTarget, | ||||||
| 		}, | 		}, | ||||||
| 		{ | 		{ | ||||||
| 			nil, | 			b:      nil, | ||||||
| 			&b, | 			target: &b, | ||||||
| 			false, | 			match:  false, | ||||||
| 			nil, | 			want:   nil, | ||||||
| 		}, | 		}, | ||||||
| 		{ | 		{ | ||||||
| 			fsmTarget, | 			b:      fsmTarget, | ||||||
| 			&b, | 			target: &b, | ||||||
| 			false, | 			match:  false, | ||||||
| 			nil, | 			want:   nil, | ||||||
| 		}, | 		}, | ||||||
| 	} | 	} | ||||||
| 	for i, tc := range testCases { | 	for i, tc := range testCases { | ||||||
| @@ -60,6 +62,8 @@ func TestAs(t *testing.T) { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | var _ broker.Broker = (*bro)(nil) | ||||||
|  |  | ||||||
| type bro struct { | type bro struct { | ||||||
| 	name string | 	name string | ||||||
| } | } | ||||||
| @@ -72,7 +76,7 @@ func (p *bro) Ready() bool { return true } | |||||||
|  |  | ||||||
| func (p *bro) Health() bool { return true } | func (p *bro) Health() bool { return true } | ||||||
|  |  | ||||||
| func (p *bro) Init(opts ...broker.Option) error { return nil } | func (p *bro) Init(_ ...broker.Option) error { return nil } | ||||||
|  |  | ||||||
| // Options returns broker options | // Options returns broker options | ||||||
| func (p *bro) Options() broker.Options { return broker.Options{} } | func (p *bro) Options() broker.Options { return broker.Options{} } | ||||||
| @@ -81,28 +85,23 @@ func (p *bro) Options() broker.Options { return broker.Options{} } | |||||||
| func (p *bro) Address() string { return "" } | func (p *bro) Address() string { return "" } | ||||||
|  |  | ||||||
| // Connect connects to broker | // Connect connects to broker | ||||||
| func (p *bro) Connect(ctx context.Context) error { return nil } | func (p *bro) Connect(_ context.Context) error { return nil } | ||||||
|  |  | ||||||
| // Disconnect disconnect from broker | // Disconnect disconnect from broker | ||||||
| func (p *bro) Disconnect(ctx context.Context) error { return nil } | func (p *bro) Disconnect(_ context.Context) error { return nil } | ||||||
|  |  | ||||||
| // Publish message, msg can be single broker.Message or []broker.Message | // NewMessage creates new message | ||||||
| func (p *bro) Publish(ctx context.Context, topic string, msg *broker.Message, opts ...broker.PublishOption) error { | func (p *bro) NewMessage(_ context.Context, _ metadata.Metadata, _ interface{}, _ ...broker.PublishOption) (broker.Message, error) { | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // BatchPublish messages to broker with multiple topics |  | ||||||
| func (p *bro) BatchPublish(ctx context.Context, msgs []*broker.Message, opts ...broker.PublishOption) error { |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // BatchSubscribe subscribes to topic messages via handler |  | ||||||
| func (p *bro) BatchSubscribe(ctx context.Context, topic string, h broker.BatchHandler, opts ...broker.SubscribeOption) (broker.Subscriber, error) { |  | ||||||
| 	return nil, nil | 	return nil, nil | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // Publish message, msg can be single broker.Message or []broker.Message | ||||||
|  | func (p *bro) Publish(_ context.Context, _ string, _ ...broker.Message) error { | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
| // Subscribe subscribes to topic message via handler | // Subscribe subscribes to topic message via handler | ||||||
| func (p *bro) Subscribe(ctx context.Context, topic string, handler broker.Handler, opts ...broker.SubscribeOption) (broker.Subscriber, error) { | func (p *bro) Subscribe(_ context.Context, _ string, _ interface{}, _ ...broker.SubscribeOption) (broker.Subscriber, error) { | ||||||
| 	return nil, nil | 	return nil, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -113,9 +112,9 @@ type fsmT struct { | |||||||
| 	name string | 	name string | ||||||
| } | } | ||||||
|  |  | ||||||
| func (f *fsmT) Start(ctx context.Context, a interface{}, o ...Option) (interface{}, error) { | func (f *fsmT) Start(_ context.Context, _ interface{}, _ ...Option) (interface{}, error) { | ||||||
| 	return nil, nil | 	return nil, nil | ||||||
| } | } | ||||||
| func (f *fsmT) Current() string                 { return f.name } | func (f *fsmT) Current() string                 { return f.name } | ||||||
| func (f *fsmT) Reset()                          {} | func (f *fsmT) Reset()                          {} | ||||||
| func (f *fsmT) State(s string, sf fsm.StateFunc) {} | func (f *fsmT) State(_ string, _ fsm.StateFunc) {} | ||||||
|   | |||||||
| @@ -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 | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,55 +0,0 @@ | |||||||
| // 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 |  | ||||||
| } |  | ||||||
| @@ -1,135 +0,0 @@ | |||||||
| 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.Must(), |  | ||||||
| 		Name:    "go.micro", |  | ||||||
| 		Address: ":0", |  | ||||||
| 		Logger:  logger.DefaultLogger, |  | ||||||
| 		Meter:   meter.DefaultMeter, |  | ||||||
| 		Tracer:  tracer.DefaultTracer, |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	for _, o := range opts { |  | ||||||
| 		o(&options) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return options |  | ||||||
| } |  | ||||||
| @@ -1,34 +0,0 @@ | |||||||
| 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) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -1,258 +0,0 @@ | |||||||
| 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), |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -1,100 +0,0 @@ | |||||||
| 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") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -1,175 +0,0 @@ | |||||||
| 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 |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -1,63 +0,0 @@ | |||||||
| // 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 Transport = 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) |  | ||||||
| @@ -1,372 +0,0 @@ | |||||||
| // 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, opts ...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, opts ...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) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -1,192 +0,0 @@ | |||||||
| 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.Must(), |  | ||||||
| 		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 |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -1,30 +0,0 @@ | |||||||
| 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) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -1,113 +0,0 @@ | |||||||
| // 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) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -1,106 +0,0 @@ | |||||||
| // 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 |  | ||||||
| } |  | ||||||
							
								
								
									
										24
									
								
								options.go
									
									
									
									
									
								
							
							
						
						
									
										24
									
								
								options.go
									
									
									
									
									
								
							| @@ -5,17 +5,17 @@ import ( | |||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/broker" | 	"go.unistack.org/micro/v4/broker" | ||||||
| 	"go.unistack.org/micro/v3/client" | 	"go.unistack.org/micro/v4/client" | ||||||
| 	"go.unistack.org/micro/v3/config" | 	"go.unistack.org/micro/v4/config" | ||||||
| 	"go.unistack.org/micro/v3/logger" | 	"go.unistack.org/micro/v4/logger" | ||||||
| 	"go.unistack.org/micro/v3/metadata" | 	"go.unistack.org/micro/v4/metadata" | ||||||
| 	"go.unistack.org/micro/v3/meter" | 	"go.unistack.org/micro/v4/meter" | ||||||
| 	"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 | ||||||
| @@ -294,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 | ||||||
| } | } | ||||||
|  |  | ||||||
|   | |||||||
							
								
								
									
										222
									
								
								options/options.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										222
									
								
								options/options.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,222 @@ | |||||||
|  | package options | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"reflect" | ||||||
|  | 	"strings" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
|  | 	"github.com/spf13/cast" | ||||||
|  | 	mreflect "go.unistack.org/micro/v4/util/reflect" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Options interface must be used by all options | ||||||
|  | type Validator interface { | ||||||
|  | 	// Validate returns nil, if all options are correct, | ||||||
|  | 	// otherwise returns an error explaining the mistake | ||||||
|  | 	Validate() error | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Option func signature | ||||||
|  | type Option func(interface{}) error | ||||||
|  |  | ||||||
|  | // Apply assign options to struct src | ||||||
|  | func Apply(src interface{}, opts ...Option) error { | ||||||
|  | 	for _, opt := range opts { | ||||||
|  | 		if err := opt(src); err != nil { | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // SetValueByPath set src struct field to val dst via path | ||||||
|  | func SetValueByPath(src interface{}, dst interface{}, path string) error { | ||||||
|  | 	var err error | ||||||
|  |  | ||||||
|  | 	switch v := dst.(type) { | ||||||
|  | 	case []interface{}: | ||||||
|  | 		if len(v) == 1 { | ||||||
|  | 			dst = v[0] | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	var sv reflect.Value | ||||||
|  | 	switch t := src.(type) { | ||||||
|  | 	case reflect.Value: | ||||||
|  | 		sv = t | ||||||
|  | 	default: | ||||||
|  | 		sv = reflect.ValueOf(src) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	parts := strings.Split(path, ".") | ||||||
|  |  | ||||||
|  | 	for _, p := range parts { | ||||||
|  |  | ||||||
|  | 		if sv.Kind() == reflect.Ptr { | ||||||
|  | 			sv = sv.Elem() | ||||||
|  | 		} | ||||||
|  | 		if sv.Kind() != reflect.Struct { | ||||||
|  | 			return mreflect.ErrInvalidStruct | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		typ := sv.Type() | ||||||
|  | 		for idx := 0; idx < typ.NumField(); idx++ { | ||||||
|  | 			fld := typ.Field(idx) | ||||||
|  | 			val := sv.Field(idx) | ||||||
|  |  | ||||||
|  | 			/* | ||||||
|  | 				if len(fld.PkgPath) != 0 { | ||||||
|  | 					continue | ||||||
|  | 				} | ||||||
|  | 			*/ | ||||||
|  |  | ||||||
|  | 			if fld.Anonymous { | ||||||
|  | 				if len(parts) == 1 && val.Kind() == reflect.Struct { | ||||||
|  | 					if err = SetValueByPath(val, dst, p); err != nil { | ||||||
|  | 						return err | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			if fld.Name != p && !strings.EqualFold(strings.ToLower(fld.Name), strings.ToLower(p)) { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			switch val.Interface().(type) { | ||||||
|  | 			case []time.Duration: | ||||||
|  | 				dst, err = cast.ToDurationSliceE(dst) | ||||||
|  | 				if err != nil { | ||||||
|  | 					return err | ||||||
|  | 				} | ||||||
|  | 				reflect.Copy(val, reflect.ValueOf(dst)) | ||||||
|  | 				return nil | ||||||
|  | 			case time.Duration: | ||||||
|  | 				dst, err = cast.ToDurationE(dst) | ||||||
|  | 				if err != nil { | ||||||
|  | 					return err | ||||||
|  | 				} | ||||||
|  | 				val.Set(reflect.ValueOf(dst)) | ||||||
|  | 				return nil | ||||||
|  | 			case time.Time: | ||||||
|  | 				dst, err = cast.ToTimeE(dst) | ||||||
|  | 				if err != nil { | ||||||
|  | 					return err | ||||||
|  | 				} | ||||||
|  | 				val.Set(reflect.ValueOf(dst)) | ||||||
|  | 				return nil | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			switch val.Kind() { | ||||||
|  | 			case reflect.Map: | ||||||
|  | 				if val.IsZero() { | ||||||
|  | 					val.Set(reflect.MakeMap(val.Type())) | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 				return setMap(val.Interface(), dst) | ||||||
|  | 			case reflect.Array, reflect.Slice: | ||||||
|  | 				switch val.Type().Elem().Kind() { | ||||||
|  | 				case reflect.Bool: | ||||||
|  | 					dst, err = cast.ToBoolSliceE(dst) | ||||||
|  | 				case reflect.String: | ||||||
|  | 					dst, err = cast.ToStringSliceE(dst) | ||||||
|  | 				case reflect.Float32: | ||||||
|  | 					dst, err = toFloat32SliceE(dst) | ||||||
|  | 				case reflect.Float64: | ||||||
|  | 					dst, err = toFloat64SliceE(dst) | ||||||
|  | 				case reflect.Int8: | ||||||
|  | 					dst, err = toInt8SliceE(dst) | ||||||
|  | 				case reflect.Int: | ||||||
|  | 					dst, err = cast.ToIntSliceE(dst) | ||||||
|  | 				case reflect.Int16: | ||||||
|  | 					dst, err = toInt16SliceE(dst) | ||||||
|  | 				case reflect.Int32: | ||||||
|  | 					dst, err = toInt32SliceE(dst) | ||||||
|  | 				case reflect.Int64: | ||||||
|  | 					dst, err = toInt64SliceE(dst) | ||||||
|  | 				case reflect.Uint8: | ||||||
|  | 					dst, err = toUint8SliceE(dst) | ||||||
|  | 				case reflect.Uint: | ||||||
|  | 					dst, err = toUintSliceE(dst) | ||||||
|  | 				case reflect.Uint16: | ||||||
|  | 					dst, err = toUint16SliceE(dst) | ||||||
|  | 				case reflect.Uint32: | ||||||
|  | 					dst, err = toUint32SliceE(dst) | ||||||
|  | 				case reflect.Uint64: | ||||||
|  | 					dst, err = toUint64SliceE(dst) | ||||||
|  | 				} | ||||||
|  | 				if err != nil { | ||||||
|  | 					return err | ||||||
|  | 				} | ||||||
|  | 				if val.Kind() == reflect.Slice { | ||||||
|  | 					val.Set(reflect.ValueOf(dst)) | ||||||
|  | 				} else { | ||||||
|  | 					reflect.Copy(val, reflect.ValueOf(dst)) | ||||||
|  | 				} | ||||||
|  | 				return nil | ||||||
|  | 			case reflect.Float32: | ||||||
|  | 				dst, err = toFloat32SliceE(dst) | ||||||
|  | 			case reflect.Float64: | ||||||
|  | 				dst, err = toFloat64SliceE(dst) | ||||||
|  | 			case reflect.Bool: | ||||||
|  | 				dst, err = cast.ToBoolE(dst) | ||||||
|  | 			case reflect.String: | ||||||
|  | 				dst, err = cast.ToStringE(dst) | ||||||
|  | 			case reflect.Int8: | ||||||
|  | 				dst, err = cast.ToInt8E(dst) | ||||||
|  | 			case reflect.Int: | ||||||
|  | 				dst, err = cast.ToIntE(dst) | ||||||
|  | 			case reflect.Int16: | ||||||
|  | 				dst, err = cast.ToInt16E(dst) | ||||||
|  | 			case reflect.Int32: | ||||||
|  | 				dst, err = cast.ToInt32E(dst) | ||||||
|  | 			case reflect.Int64: | ||||||
|  | 				dst, err = cast.ToInt64E(dst) | ||||||
|  | 			case reflect.Uint8: | ||||||
|  | 				dst, err = cast.ToUint8E(dst) | ||||||
|  | 			case reflect.Uint: | ||||||
|  | 				dst, err = cast.ToUintE(dst) | ||||||
|  | 			case reflect.Uint16: | ||||||
|  | 				dst, err = cast.ToUint16E(dst) | ||||||
|  | 			case reflect.Uint32: | ||||||
|  | 				dst, err = cast.ToUint32E(dst) | ||||||
|  | 			case reflect.Uint64: | ||||||
|  | 				dst, err = cast.ToUint64E(dst) | ||||||
|  | 			default: | ||||||
|  | 			} | ||||||
|  | 			if err != nil { | ||||||
|  | 				return err | ||||||
|  | 			} | ||||||
|  | 			val.Set(reflect.ValueOf(dst)) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewOption create new option with name | ||||||
|  | func NewOption(name string) func(...interface{}) Option { | ||||||
|  | 	return func(dst ...interface{}) Option { | ||||||
|  | 		return func(src interface{}) error { | ||||||
|  | 			return SetValueByPath(src, dst, name) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | var ( | ||||||
|  | 	Address   = NewOption("Address") | ||||||
|  | 	Name      = NewOption("Name") | ||||||
|  | 	Broker    = NewOption("Broker") | ||||||
|  | 	Logger    = NewOption("Logger") | ||||||
|  | 	Meter     = NewOption("Meter") | ||||||
|  | 	Tracer    = NewOption("Tracer") | ||||||
|  | 	Store     = NewOption("Store") | ||||||
|  | 	Register  = NewOption("Register") | ||||||
|  | 	Router    = NewOption("Router") | ||||||
|  | 	Codec     = NewOption("Codec") | ||||||
|  | 	Codecs    = NewOption("Codecs") | ||||||
|  | 	Client    = NewOption("Client") | ||||||
|  | 	Context   = NewOption("Context") | ||||||
|  | 	TLSConfig = NewOption("TLSConfig") | ||||||
|  | 	Metadata  = NewOption("Metadata") | ||||||
|  | 	Timeout   = NewOption("Timeout") | ||||||
|  | ) | ||||||
							
								
								
									
										181
									
								
								options/options_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										181
									
								
								options/options_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,181 @@ | |||||||
|  | package options_test | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"crypto/tls" | ||||||
|  | 	"sync" | ||||||
|  | 	"testing" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v4/options" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type codec interface { | ||||||
|  | 	Marshal(v interface{}, opts ...options.Option) ([]byte, error) | ||||||
|  | 	Unmarshal(b []byte, v interface{}, opts ...options.Option) error | ||||||
|  | 	String() string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestCodecs(t *testing.T) { | ||||||
|  | 	type s struct { | ||||||
|  | 		Codecs map[string]codec | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	wg := &sync.WaitGroup{} | ||||||
|  | 	tc := &tls.Config{InsecureSkipVerify: true} | ||||||
|  | 	opts := []options.Option{ | ||||||
|  | 		options.NewOption("Codecs")(wg), | ||||||
|  | 		options.NewOption("TLSConfig")(tc), | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	src := &s{} | ||||||
|  |  | ||||||
|  | 	if err := options.Apply(src, opts...); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestSpecial(t *testing.T) { | ||||||
|  | 	type s struct { | ||||||
|  | 		Wait      *sync.WaitGroup | ||||||
|  | 		TLSConfig *tls.Config | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	wg := &sync.WaitGroup{} | ||||||
|  | 	tc := &tls.Config{InsecureSkipVerify: true} | ||||||
|  | 	opts := []options.Option{ | ||||||
|  | 		options.NewOption("Wait")(wg), | ||||||
|  | 		options.NewOption("TLSConfig")(tc), | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	src := &s{} | ||||||
|  |  | ||||||
|  | 	if err := options.Apply(src, opts...); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if src.Wait == nil { | ||||||
|  | 		t.Fatalf("failed to set Wait %#+v", src) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if src.TLSConfig == nil { | ||||||
|  | 		t.Fatalf("failed to set TLSConfig %#+v", src) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if src.TLSConfig.InsecureSkipVerify != true { | ||||||
|  | 		t.Fatalf("failed to set TLSConfig %#+v", src) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestNested(t *testing.T) { | ||||||
|  | 	type server struct { | ||||||
|  | 		Address []string | ||||||
|  | 	} | ||||||
|  | 	type ownserver struct { | ||||||
|  | 		server | ||||||
|  | 		OwnField string | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	opts := []options.Option{ | ||||||
|  | 		options.Address("host:port"), | ||||||
|  | 		options.NewOption("OwnField")("fieldval"), | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	src := &ownserver{} | ||||||
|  |  | ||||||
|  | 	if err := options.Apply(src, opts...); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if src.Address[0] != "host:port" { | ||||||
|  | 		t.Fatalf("failed to set Address %#+v", src) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if src.OwnField != "fieldval" { | ||||||
|  | 		t.Fatalf("failed to set OwnField %#+v", src) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestAddress(t *testing.T) { | ||||||
|  | 	type s struct { | ||||||
|  | 		Address []string | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	opts := []options.Option{options.Address("host:port")} | ||||||
|  |  | ||||||
|  | 	src := &s{} | ||||||
|  |  | ||||||
|  | 	if err := options.Apply(src, opts...); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if src.Address[0] != "host:port" { | ||||||
|  | 		t.Fatalf("failed to set Address %#+v", src) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestNewOption(t *testing.T) { | ||||||
|  | 	type s struct { | ||||||
|  | 		Address []string | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	opts := []options.Option{options.NewOption("Address")("host1:port1", "host2:port2")} | ||||||
|  |  | ||||||
|  | 	src := &s{} | ||||||
|  |  | ||||||
|  | 	if err := options.Apply(src, opts...); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if src.Address[0] != "host1:port1" { | ||||||
|  | 		t.Fatalf("failed to set Address %#+v", src) | ||||||
|  | 	} | ||||||
|  | 	if src.Address[1] != "host2:port2" { | ||||||
|  | 		t.Fatalf("failed to set Address %#+v", src) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestArray(t *testing.T) { | ||||||
|  | 	type s struct { | ||||||
|  | 		Address [1]string | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	opts := []options.Option{options.NewOption("Address")("host:port", "host1:port1")} | ||||||
|  |  | ||||||
|  | 	src := &s{} | ||||||
|  |  | ||||||
|  | 	if err := options.Apply(src, opts...); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if src.Address[0] != "host:port" { | ||||||
|  | 		t.Fatalf("failed to set Address %#+v", src) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestMap(t *testing.T) { | ||||||
|  | 	type s struct { | ||||||
|  | 		Metadata map[string]string | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	opts := []options.Option{ | ||||||
|  | 		options.NewOption("Metadata")("key1", "val1"), | ||||||
|  | 		options.NewOption("Metadata")(map[string]string{"key2": "val2"}), | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	src := &s{} | ||||||
|  |  | ||||||
|  | 	if err := options.Apply(src, opts...); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if len(src.Metadata) != 2 { | ||||||
|  | 		t.Fatalf("failed to set Metadata %#+v", src) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if src.Metadata["key1"] != "val1" { | ||||||
|  | 		t.Fatalf("failed to set Metadata %#+v", src) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if src.Metadata["key2"] != "val2" { | ||||||
|  | 		t.Fatalf("failed to set Metadata %#+v", src) | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										577
									
								
								options/util.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										577
									
								
								options/util.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,577 @@ | |||||||
|  | package options | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"reflect" | ||||||
|  |  | ||||||
|  | 	"github.com/spf13/cast" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func toInt8SliceE(i interface{}) ([]int8, error) { | ||||||
|  | 	if i == nil { | ||||||
|  | 		return []int8{}, fmt.Errorf("unable to cast %#v of type %T to []int8", i, i) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	switch v := i.(type) { | ||||||
|  | 	case []int8: | ||||||
|  | 		return v, nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	kind := reflect.TypeOf(i).Kind() | ||||||
|  | 	switch kind { | ||||||
|  | 	case reflect.Slice, reflect.Array: | ||||||
|  | 		s := reflect.ValueOf(i) | ||||||
|  | 		a := make([]int8, s.Len()) | ||||||
|  | 		for j := 0; j < s.Len(); j++ { | ||||||
|  | 			val, err := cast.ToInt8E(s.Index(j).Interface()) | ||||||
|  | 			if err != nil { | ||||||
|  | 				return []int8{}, fmt.Errorf("unable to cast %#v of type %T to []int8", i, i) | ||||||
|  | 			} | ||||||
|  | 			a[j] = val | ||||||
|  | 		} | ||||||
|  | 		return a, nil | ||||||
|  | 	default: | ||||||
|  | 		return []int8{}, fmt.Errorf("unable to cast %#v of type %T to []int8", i, i) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func toInt16SliceE(i interface{}) ([]int16, error) { | ||||||
|  | 	if i == nil { | ||||||
|  | 		return []int16{}, fmt.Errorf("unable to cast %#v of type %T to []int16", i, i) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	switch v := i.(type) { | ||||||
|  | 	case []int16: | ||||||
|  | 		return v, nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	kind := reflect.TypeOf(i).Kind() | ||||||
|  | 	switch kind { | ||||||
|  | 	case reflect.Slice, reflect.Array: | ||||||
|  | 		s := reflect.ValueOf(i) | ||||||
|  | 		a := make([]int16, s.Len()) | ||||||
|  | 		for j := 0; j < s.Len(); j++ { | ||||||
|  | 			val, err := cast.ToInt16E(s.Index(j).Interface()) | ||||||
|  | 			if err != nil { | ||||||
|  | 				return []int16{}, fmt.Errorf("unable to cast %#v of type %T to []int16", i, i) | ||||||
|  | 			} | ||||||
|  | 			a[j] = val | ||||||
|  | 		} | ||||||
|  | 		return a, nil | ||||||
|  | 	default: | ||||||
|  | 		return []int16{}, fmt.Errorf("unable to cast %#v of type %T to []int16", i, i) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func toInt32SliceE(i interface{}) ([]int32, error) { | ||||||
|  | 	if i == nil { | ||||||
|  | 		return []int32{}, fmt.Errorf("unable to cast %#v of type %T to []int32", i, i) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	switch v := i.(type) { | ||||||
|  | 	case []int32: | ||||||
|  | 		return v, nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	kind := reflect.TypeOf(i).Kind() | ||||||
|  | 	switch kind { | ||||||
|  | 	case reflect.Slice, reflect.Array: | ||||||
|  | 		s := reflect.ValueOf(i) | ||||||
|  | 		a := make([]int32, s.Len()) | ||||||
|  | 		for j := 0; j < s.Len(); j++ { | ||||||
|  | 			val, err := cast.ToInt32E(s.Index(j).Interface()) | ||||||
|  | 			if err != nil { | ||||||
|  | 				return []int32{}, fmt.Errorf("unable to cast %#v of type %T to []int32", i, i) | ||||||
|  | 			} | ||||||
|  | 			a[j] = val | ||||||
|  | 		} | ||||||
|  | 		return a, nil | ||||||
|  | 	default: | ||||||
|  | 		return []int32{}, fmt.Errorf("unable to cast %#v of type %T to []int32", i, i) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func toInt64SliceE(i interface{}) ([]int64, error) { | ||||||
|  | 	if i == nil { | ||||||
|  | 		return []int64{}, fmt.Errorf("unable to cast %#v of type %T to []int64", i, i) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	switch v := i.(type) { | ||||||
|  | 	case []int64: | ||||||
|  | 		return v, nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	kind := reflect.TypeOf(i).Kind() | ||||||
|  | 	switch kind { | ||||||
|  | 	case reflect.Slice, reflect.Array: | ||||||
|  | 		s := reflect.ValueOf(i) | ||||||
|  | 		a := make([]int64, s.Len()) | ||||||
|  | 		for j := 0; j < s.Len(); j++ { | ||||||
|  | 			val, err := cast.ToInt64E(s.Index(j).Interface()) | ||||||
|  | 			if err != nil { | ||||||
|  | 				return []int64{}, fmt.Errorf("unable to cast %#v of type %T to []int64", i, i) | ||||||
|  | 			} | ||||||
|  | 			a[j] = val | ||||||
|  | 		} | ||||||
|  | 		return a, nil | ||||||
|  | 	default: | ||||||
|  | 		return []int64{}, fmt.Errorf("unable to cast %#v of type %T to []int64", i, i) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func toUintSliceE(i interface{}) ([]uint, error) { | ||||||
|  | 	if i == nil { | ||||||
|  | 		return []uint{}, fmt.Errorf("unable to cast %#v of type %T to []uint", i, i) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	switch v := i.(type) { | ||||||
|  | 	case []uint: | ||||||
|  | 		return v, nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	kind := reflect.TypeOf(i).Kind() | ||||||
|  | 	switch kind { | ||||||
|  | 	case reflect.Slice, reflect.Array: | ||||||
|  | 		s := reflect.ValueOf(i) | ||||||
|  | 		a := make([]uint, s.Len()) | ||||||
|  | 		for j := 0; j < s.Len(); j++ { | ||||||
|  | 			val, err := cast.ToUintE(s.Index(j).Interface()) | ||||||
|  | 			if err != nil { | ||||||
|  | 				return []uint{}, fmt.Errorf("unable to cast %#v of type %T to []uint", i, i) | ||||||
|  | 			} | ||||||
|  | 			a[j] = val | ||||||
|  | 		} | ||||||
|  | 		return a, nil | ||||||
|  | 	default: | ||||||
|  | 		return []uint{}, fmt.Errorf("unable to cast %#v of type %T to []uint", i, i) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func toUint8SliceE(i interface{}) ([]uint8, error) { | ||||||
|  | 	if i == nil { | ||||||
|  | 		return []uint8{}, fmt.Errorf("unable to cast %#v of type %T to []uint8", i, i) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	switch v := i.(type) { | ||||||
|  | 	case []uint8: | ||||||
|  | 		return v, nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	kind := reflect.TypeOf(i).Kind() | ||||||
|  | 	switch kind { | ||||||
|  | 	case reflect.Slice, reflect.Array: | ||||||
|  | 		s := reflect.ValueOf(i) | ||||||
|  | 		a := make([]uint8, s.Len()) | ||||||
|  | 		for j := 0; j < s.Len(); j++ { | ||||||
|  | 			val, err := cast.ToUint8E(s.Index(j).Interface()) | ||||||
|  | 			if err != nil { | ||||||
|  | 				return []uint8{}, fmt.Errorf("unable to cast %#v of type %T to []uint8", i, i) | ||||||
|  | 			} | ||||||
|  | 			a[j] = val | ||||||
|  | 		} | ||||||
|  | 		return a, nil | ||||||
|  | 	default: | ||||||
|  | 		return []uint8{}, fmt.Errorf("unable to cast %#v of type %T to []uint8", i, i) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func toUint16SliceE(i interface{}) ([]uint16, error) { | ||||||
|  | 	if i == nil { | ||||||
|  | 		return []uint16{}, fmt.Errorf("unable to cast %#v of type %T to []uint16", i, i) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	switch v := i.(type) { | ||||||
|  | 	case []uint16: | ||||||
|  | 		return v, nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	kind := reflect.TypeOf(i).Kind() | ||||||
|  | 	switch kind { | ||||||
|  | 	case reflect.Slice, reflect.Array: | ||||||
|  | 		s := reflect.ValueOf(i) | ||||||
|  | 		a := make([]uint16, s.Len()) | ||||||
|  | 		for j := 0; j < s.Len(); j++ { | ||||||
|  | 			val, err := cast.ToUint16E(s.Index(j).Interface()) | ||||||
|  | 			if err != nil { | ||||||
|  | 				return []uint16{}, fmt.Errorf("unable to cast %#v of type %T to []uint16", i, i) | ||||||
|  | 			} | ||||||
|  | 			a[j] = val | ||||||
|  | 		} | ||||||
|  | 		return a, nil | ||||||
|  | 	default: | ||||||
|  | 		return []uint16{}, fmt.Errorf("unable to cast %#v of type %T to []uint16", i, i) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func toUint32SliceE(i interface{}) ([]uint32, error) { | ||||||
|  | 	if i == nil { | ||||||
|  | 		return []uint32{}, fmt.Errorf("unable to cast %#v of type %T to []uint32", i, i) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	switch v := i.(type) { | ||||||
|  | 	case []uint32: | ||||||
|  | 		return v, nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	kind := reflect.TypeOf(i).Kind() | ||||||
|  | 	switch kind { | ||||||
|  | 	case reflect.Slice, reflect.Array: | ||||||
|  | 		s := reflect.ValueOf(i) | ||||||
|  | 		a := make([]uint32, s.Len()) | ||||||
|  | 		for j := 0; j < s.Len(); j++ { | ||||||
|  | 			val, err := cast.ToUint32E(s.Index(j).Interface()) | ||||||
|  | 			if err != nil { | ||||||
|  | 				return []uint32{}, fmt.Errorf("unable to cast %#v of type %T to []uint32", i, i) | ||||||
|  | 			} | ||||||
|  | 			a[j] = val | ||||||
|  | 		} | ||||||
|  | 		return a, nil | ||||||
|  | 	default: | ||||||
|  | 		return []uint32{}, fmt.Errorf("unable to cast %#v of type %T to []uint32", i, i) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func toUint64SliceE(i interface{}) ([]uint64, error) { | ||||||
|  | 	if i == nil { | ||||||
|  | 		return []uint64{}, fmt.Errorf("unable to cast %#v of type %T to []uint64", i, i) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	switch v := i.(type) { | ||||||
|  | 	case []uint64: | ||||||
|  | 		return v, nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	kind := reflect.TypeOf(i).Kind() | ||||||
|  | 	switch kind { | ||||||
|  | 	case reflect.Slice, reflect.Array: | ||||||
|  | 		s := reflect.ValueOf(i) | ||||||
|  | 		a := make([]uint64, s.Len()) | ||||||
|  | 		for j := 0; j < s.Len(); j++ { | ||||||
|  | 			val, err := cast.ToUint64E(s.Index(j).Interface()) | ||||||
|  | 			if err != nil { | ||||||
|  | 				return []uint64{}, fmt.Errorf("unable to cast %#v of type %T to []uint64", i, i) | ||||||
|  | 			} | ||||||
|  | 			a[j] = val | ||||||
|  | 		} | ||||||
|  | 		return a, nil | ||||||
|  | 	default: | ||||||
|  | 		return []uint64{}, fmt.Errorf("unable to cast %#v of type %T to []uint64", i, i) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func toFloat32SliceE(i interface{}) ([]float32, error) { | ||||||
|  | 	if i == nil { | ||||||
|  | 		return []float32{}, fmt.Errorf("unable to cast %#v of type %T to []float32", i, i) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	switch v := i.(type) { | ||||||
|  | 	case []float32: | ||||||
|  | 		return v, nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	kind := reflect.TypeOf(i).Kind() | ||||||
|  | 	switch kind { | ||||||
|  | 	case reflect.Slice, reflect.Array: | ||||||
|  | 		s := reflect.ValueOf(i) | ||||||
|  | 		a := make([]float32, s.Len()) | ||||||
|  | 		for j := 0; j < s.Len(); j++ { | ||||||
|  | 			val, err := cast.ToFloat32E(s.Index(j).Interface()) | ||||||
|  | 			if err != nil { | ||||||
|  | 				return []float32{}, fmt.Errorf("unable to cast %#v of type %T to []float32", i, i) | ||||||
|  | 			} | ||||||
|  | 			a[j] = val | ||||||
|  | 		} | ||||||
|  | 		return a, nil | ||||||
|  | 	default: | ||||||
|  | 		return []float32{}, fmt.Errorf("unable to cast %#v of type %T to []float32", i, i) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func toFloat64SliceE(i interface{}) ([]float64, error) { | ||||||
|  | 	if i == nil { | ||||||
|  | 		return []float64{}, fmt.Errorf("unable to cast %#v of type %T to []float64", i, i) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	switch v := i.(type) { | ||||||
|  | 	case []float64: | ||||||
|  | 		return v, nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	kind := reflect.TypeOf(i).Kind() | ||||||
|  | 	switch kind { | ||||||
|  | 	case reflect.Slice, reflect.Array: | ||||||
|  | 		s := reflect.ValueOf(i) | ||||||
|  | 		a := make([]float64, s.Len()) | ||||||
|  | 		for j := 0; j < s.Len(); j++ { | ||||||
|  | 			val, err := cast.ToFloat64E(s.Index(j).Interface()) | ||||||
|  | 			if err != nil { | ||||||
|  | 				return []float64{}, fmt.Errorf("unable to cast %#v of type %T to []float64", i, i) | ||||||
|  | 			} | ||||||
|  | 			a[j] = val | ||||||
|  | 		} | ||||||
|  | 		return a, nil | ||||||
|  | 	default: | ||||||
|  | 		return []float64{}, fmt.Errorf("unable to cast %#v of type %T to []float32", i, i) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func setMap(src interface{}, dst interface{}) error { | ||||||
|  | 	var err error | ||||||
|  |  | ||||||
|  | 	if src == nil { | ||||||
|  | 		return fmt.Errorf("unable to cast %#v of type %T", src, src) | ||||||
|  | 	} | ||||||
|  | 	if dst == nil { | ||||||
|  | 		return fmt.Errorf("unable to cast %#v of type %T", dst, dst) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	val := reflect.ValueOf(src) | ||||||
|  |  | ||||||
|  | 	keyKind := val.Type().Key().Kind() | ||||||
|  | 	valKind := val.Type().Elem().Kind() | ||||||
|  |  | ||||||
|  | 	switch v := dst.(type) { | ||||||
|  | 	case []interface{}: | ||||||
|  | 		if len(v) == 1 { | ||||||
|  | 			dstVal := reflect.ValueOf(v[0]) | ||||||
|  | 			if dstVal.Kind() != reflect.Map { | ||||||
|  | 				return nil | ||||||
|  | 			} | ||||||
|  | 			mapIter := dstVal.MapRange() | ||||||
|  | 			for mapIter.Next() { | ||||||
|  | 				var ( | ||||||
|  | 					keyVal interface{} | ||||||
|  | 					valVal interface{} | ||||||
|  | 				) | ||||||
|  | 				switch keyKind { | ||||||
|  | 				case reflect.Bool: | ||||||
|  | 					keyVal, err = cast.ToBoolE(mapIter.Key()) | ||||||
|  | 				case reflect.String: | ||||||
|  | 					keyVal, err = cast.ToStringE(mapIter.Key()) | ||||||
|  | 				case reflect.Float32: | ||||||
|  | 					keyVal, err = cast.ToFloat32E(mapIter.Key()) | ||||||
|  | 				case reflect.Float64: | ||||||
|  | 					keyVal, err = cast.ToFloat64E(mapIter.Key()) | ||||||
|  | 				case reflect.Int8: | ||||||
|  | 					keyVal, err = cast.ToInt8E(mapIter.Key()) | ||||||
|  | 				case reflect.Int: | ||||||
|  | 					keyVal, err = cast.ToIntE(mapIter.Key()) | ||||||
|  | 				case reflect.Int16: | ||||||
|  | 					keyVal, err = cast.ToInt16E(mapIter.Key()) | ||||||
|  | 				case reflect.Int32: | ||||||
|  | 					keyVal, err = cast.ToInt32E(mapIter.Key()) | ||||||
|  | 				case reflect.Int64: | ||||||
|  | 					keyVal, err = cast.ToInt64E(mapIter.Key()) | ||||||
|  | 				case reflect.Uint8: | ||||||
|  | 					keyVal, err = cast.ToUint8E(mapIter.Key()) | ||||||
|  | 				case reflect.Uint: | ||||||
|  | 					keyVal, err = cast.ToUintE(mapIter.Key()) | ||||||
|  | 				case reflect.Uint16: | ||||||
|  | 					keyVal, err = cast.ToUint16E(mapIter.Key()) | ||||||
|  | 				case reflect.Uint32: | ||||||
|  | 					keyVal, err = cast.ToUint32E(mapIter.Key()) | ||||||
|  | 				case reflect.Uint64: | ||||||
|  | 					keyVal, err = cast.ToUint64E(mapIter.Key()) | ||||||
|  | 				} | ||||||
|  | 				if err != nil { | ||||||
|  | 					return err | ||||||
|  | 				} | ||||||
|  | 				switch valKind { | ||||||
|  | 				case reflect.Bool: | ||||||
|  | 					valVal, err = cast.ToBoolE(mapIter.Value()) | ||||||
|  | 				case reflect.String: | ||||||
|  | 					valVal, err = cast.ToStringE(mapIter.Value()) | ||||||
|  | 				case reflect.Float32: | ||||||
|  | 					valVal, err = cast.ToFloat32E(mapIter.Value()) | ||||||
|  | 				case reflect.Float64: | ||||||
|  | 					valVal, err = cast.ToFloat64E(mapIter.Value()) | ||||||
|  | 				case reflect.Int8: | ||||||
|  | 					valVal, err = cast.ToInt8E(mapIter.Value()) | ||||||
|  | 				case reflect.Int: | ||||||
|  | 					valVal, err = cast.ToIntE(mapIter.Value()) | ||||||
|  | 				case reflect.Int16: | ||||||
|  | 					valVal, err = cast.ToInt16E(mapIter.Value()) | ||||||
|  | 				case reflect.Int32: | ||||||
|  | 					valVal, err = cast.ToInt32E(mapIter.Value()) | ||||||
|  | 				case reflect.Int64: | ||||||
|  | 					valVal, err = cast.ToInt64E(mapIter.Value()) | ||||||
|  | 				case reflect.Uint8: | ||||||
|  | 					valVal, err = cast.ToUint8E(mapIter.Value()) | ||||||
|  | 				case reflect.Uint: | ||||||
|  | 					valVal, err = cast.ToUintE(mapIter.Value()) | ||||||
|  | 				case reflect.Uint16: | ||||||
|  | 					valVal, err = cast.ToUint16E(mapIter.Value()) | ||||||
|  | 				case reflect.Uint32: | ||||||
|  | 					valVal, err = cast.ToUint32E(mapIter.Value()) | ||||||
|  | 				case reflect.Uint64: | ||||||
|  | 					valVal, err = cast.ToUint64E(mapIter.Value()) | ||||||
|  | 				} | ||||||
|  | 				if err != nil { | ||||||
|  | 					return err | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 				val.SetMapIndex(reflect.ValueOf(keyVal), reflect.ValueOf(valVal)) | ||||||
|  | 			} | ||||||
|  | 			return nil | ||||||
|  | 		} | ||||||
|  | 		if l := len(v) % 2; l == 1 { | ||||||
|  | 			v = v[:len(v)-1] | ||||||
|  | 		} | ||||||
|  | 		var ( | ||||||
|  | 			keyVal interface{} | ||||||
|  | 			valVal interface{} | ||||||
|  | 		) | ||||||
|  | 		for i := 0; i < len(v); i += 2 { | ||||||
|  | 			switch keyKind { | ||||||
|  | 			case reflect.Bool: | ||||||
|  | 				keyVal, err = cast.ToBoolE(v[i]) | ||||||
|  | 			case reflect.String: | ||||||
|  | 				keyVal, err = cast.ToStringE(v[i]) | ||||||
|  | 			case reflect.Float32: | ||||||
|  | 				keyVal, err = cast.ToFloat32E(v[i]) | ||||||
|  | 			case reflect.Float64: | ||||||
|  | 				keyVal, err = cast.ToFloat64E(v[i]) | ||||||
|  | 			case reflect.Int8: | ||||||
|  | 				keyVal, err = cast.ToInt8E(v[i]) | ||||||
|  | 			case reflect.Int: | ||||||
|  | 				keyVal, err = cast.ToIntE(v[i]) | ||||||
|  | 			case reflect.Int16: | ||||||
|  | 				keyVal, err = cast.ToInt16E(v[i]) | ||||||
|  | 			case reflect.Int32: | ||||||
|  | 				keyVal, err = cast.ToInt32E(v[i]) | ||||||
|  | 			case reflect.Int64: | ||||||
|  | 				keyVal, err = cast.ToInt64E(v[i]) | ||||||
|  | 			case reflect.Uint8: | ||||||
|  | 				keyVal, err = cast.ToUint8E(v[i]) | ||||||
|  | 			case reflect.Uint: | ||||||
|  | 				keyVal, err = cast.ToUintE(v[i]) | ||||||
|  | 			case reflect.Uint16: | ||||||
|  | 				keyVal, err = cast.ToUint16E(v[i]) | ||||||
|  | 			case reflect.Uint32: | ||||||
|  | 				keyVal, err = cast.ToUint32E(v[i]) | ||||||
|  | 			case reflect.Uint64: | ||||||
|  | 				keyVal, err = cast.ToUint64E(v[i]) | ||||||
|  | 			} | ||||||
|  | 			if err != nil { | ||||||
|  | 				return err | ||||||
|  | 			} | ||||||
|  | 			switch valKind { | ||||||
|  | 			case reflect.Bool: | ||||||
|  | 				valVal, err = cast.ToBoolE(v[i+1]) | ||||||
|  | 			case reflect.String: | ||||||
|  | 				valVal, err = cast.ToStringE(v[i+1]) | ||||||
|  | 			case reflect.Float32: | ||||||
|  | 				valVal, err = cast.ToFloat32E(v[i+1]) | ||||||
|  | 			case reflect.Float64: | ||||||
|  | 				valVal, err = cast.ToFloat64E(v[i+1]) | ||||||
|  | 			case reflect.Int8: | ||||||
|  | 				valVal, err = cast.ToInt8E(v[i+1]) | ||||||
|  | 			case reflect.Int: | ||||||
|  | 				valVal, err = cast.ToIntE(v[i+1]) | ||||||
|  | 			case reflect.Int16: | ||||||
|  | 				valVal, err = cast.ToInt16E(v[i+1]) | ||||||
|  | 			case reflect.Int32: | ||||||
|  | 				valVal, err = cast.ToInt32E(v[i+1]) | ||||||
|  | 			case reflect.Int64: | ||||||
|  | 				valVal, err = cast.ToInt64E(v[i+1]) | ||||||
|  | 			case reflect.Uint8: | ||||||
|  | 				valVal, err = cast.ToUint8E(v[i+1]) | ||||||
|  | 			case reflect.Uint: | ||||||
|  | 				valVal, err = cast.ToUintE(v[i+1]) | ||||||
|  | 			case reflect.Uint16: | ||||||
|  | 				valVal, err = cast.ToUint16E(v[i+1]) | ||||||
|  | 			case reflect.Uint32: | ||||||
|  | 				valVal, err = cast.ToUint32E(v[i+1]) | ||||||
|  | 			case reflect.Uint64: | ||||||
|  | 				valVal, err = cast.ToUint64E(v[i+1]) | ||||||
|  | 			} | ||||||
|  | 			if err != nil { | ||||||
|  | 				return err | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			val.SetMapIndex(reflect.ValueOf(keyVal), reflect.ValueOf(valVal)) | ||||||
|  | 		} | ||||||
|  | 	default: | ||||||
|  | 		dstVal := reflect.ValueOf(dst) | ||||||
|  | 		if dstVal.Kind() != reflect.Map { | ||||||
|  | 			return nil | ||||||
|  | 		} | ||||||
|  | 		mapIter := dstVal.MapRange() | ||||||
|  | 		for mapIter.Next() { | ||||||
|  | 			var ( | ||||||
|  | 				keyVal interface{} | ||||||
|  | 				valVal interface{} | ||||||
|  | 			) | ||||||
|  | 			switch keyKind { | ||||||
|  | 			case reflect.Bool: | ||||||
|  | 				keyVal, err = cast.ToBoolE(mapIter.Key()) | ||||||
|  | 			case reflect.String: | ||||||
|  | 				keyVal, err = cast.ToStringE(mapIter.Key()) | ||||||
|  | 			case reflect.Float32: | ||||||
|  | 				keyVal, err = cast.ToFloat32E(mapIter.Key()) | ||||||
|  | 			case reflect.Float64: | ||||||
|  | 				keyVal, err = cast.ToFloat64E(mapIter.Key()) | ||||||
|  | 			case reflect.Int8: | ||||||
|  | 				keyVal, err = cast.ToInt8E(mapIter.Key()) | ||||||
|  | 			case reflect.Int: | ||||||
|  | 				keyVal, err = cast.ToIntE(mapIter.Key()) | ||||||
|  | 			case reflect.Int16: | ||||||
|  | 				keyVal, err = cast.ToInt16E(mapIter.Key()) | ||||||
|  | 			case reflect.Int32: | ||||||
|  | 				keyVal, err = cast.ToInt32E(mapIter.Key()) | ||||||
|  | 			case reflect.Int64: | ||||||
|  | 				keyVal, err = cast.ToInt64E(mapIter.Key()) | ||||||
|  | 			case reflect.Uint8: | ||||||
|  | 				keyVal, err = cast.ToUint8E(mapIter.Key()) | ||||||
|  | 			case reflect.Uint: | ||||||
|  | 				keyVal, err = cast.ToUintE(mapIter.Key()) | ||||||
|  | 			case reflect.Uint16: | ||||||
|  | 				keyVal, err = cast.ToUint16E(mapIter.Key()) | ||||||
|  | 			case reflect.Uint32: | ||||||
|  | 				keyVal, err = cast.ToUint32E(mapIter.Key()) | ||||||
|  | 			case reflect.Uint64: | ||||||
|  | 				keyVal, err = cast.ToUint64E(mapIter.Key()) | ||||||
|  | 			} | ||||||
|  | 			if err != nil { | ||||||
|  | 				return err | ||||||
|  | 			} | ||||||
|  | 			switch valKind { | ||||||
|  | 			case reflect.Bool: | ||||||
|  | 				valVal, err = cast.ToBoolE(mapIter.Value()) | ||||||
|  | 			case reflect.String: | ||||||
|  | 				valVal, err = cast.ToStringE(mapIter.Value()) | ||||||
|  | 			case reflect.Float32: | ||||||
|  | 				valVal, err = cast.ToFloat32E(mapIter.Value()) | ||||||
|  | 			case reflect.Float64: | ||||||
|  | 				valVal, err = cast.ToFloat64E(mapIter.Value()) | ||||||
|  | 			case reflect.Int8: | ||||||
|  | 				valVal, err = cast.ToInt8E(mapIter.Value()) | ||||||
|  | 			case reflect.Int: | ||||||
|  | 				valVal, err = cast.ToIntE(mapIter.Value()) | ||||||
|  | 			case reflect.Int16: | ||||||
|  | 				valVal, err = cast.ToInt16E(mapIter.Value()) | ||||||
|  | 			case reflect.Int32: | ||||||
|  | 				valVal, err = cast.ToInt32E(mapIter.Value()) | ||||||
|  | 			case reflect.Int64: | ||||||
|  | 				valVal, err = cast.ToInt64E(mapIter.Value()) | ||||||
|  | 			case reflect.Uint8: | ||||||
|  | 				valVal, err = cast.ToUint8E(mapIter.Value()) | ||||||
|  | 			case reflect.Uint: | ||||||
|  | 				valVal, err = cast.ToUintE(mapIter.Value()) | ||||||
|  | 			case reflect.Uint16: | ||||||
|  | 				valVal, err = cast.ToUint16E(mapIter.Value()) | ||||||
|  | 			case reflect.Uint32: | ||||||
|  | 				valVal, err = cast.ToUint32E(mapIter.Value()) | ||||||
|  | 			case reflect.Uint64: | ||||||
|  | 				valVal, err = cast.ToUint64E(mapIter.Value()) | ||||||
|  | 			} | ||||||
|  | 			if err != nil { | ||||||
|  | 				return err | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			val.SetMapIndex(reflect.ValueOf(keyVal), reflect.ValueOf(valVal)) | ||||||
|  | 		} | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
| @@ -7,7 +7,7 @@ import ( | |||||||
| 	"net/http/pprof" | 	"net/http/pprof" | ||||||
| 	"sync" | 	"sync" | ||||||
|  |  | ||||||
| 	profile "go.unistack.org/micro/v3/profiler" | 	profile "go.unistack.org/micro/v4/profiler" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type httpProfile struct { | type httpProfile struct { | ||||||
|   | |||||||
| @@ -15,6 +15,6 @@ func (p *noopProfiler) String() string { | |||||||
| } | } | ||||||
|  |  | ||||||
| // NewProfiler returns new noop profiler | // NewProfiler returns new noop profiler | ||||||
| func NewProfiler(opts ...Option) Profiler { | func NewProfiler(_ ...Option) Profiler { | ||||||
| 	return &noopProfiler{} | 	return &noopProfiler{} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -9,7 +9,7 @@ import ( | |||||||
| 	"sync" | 	"sync" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	profile "go.unistack.org/micro/v3/profiler" | 	profile "go.unistack.org/micro/v4/profiler" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type profiler struct { | type profiler struct { | ||||||
|   | |||||||
| @@ -12,7 +12,7 @@ type Profiler interface { | |||||||
| } | } | ||||||
|  |  | ||||||
| // DefaultProfiler holds the default profiler | // DefaultProfiler holds the default profiler | ||||||
| var DefaultProfiler Profiler = NewProfiler() | var DefaultProfiler = NewProfiler() | ||||||
|  |  | ||||||
| // Options holds the options for profiler | // Options holds the options for profiler | ||||||
| type Options struct { | type Options struct { | ||||||
|   | |||||||
| @@ -1,98 +0,0 @@ | |||||||
| // Package proxy is a transparent proxy built on the micro/server |  | ||||||
| package proxy |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"go.unistack.org/micro/v3/client" |  | ||||||
| 	"go.unistack.org/micro/v3/logger" |  | ||||||
| 	"go.unistack.org/micro/v3/meter" |  | ||||||
| 	"go.unistack.org/micro/v3/router" |  | ||||||
| 	"go.unistack.org/micro/v3/tracer" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // Options for proxy |  | ||||||
| type Options struct { |  | ||||||
| 	// Tracer used for tracing |  | ||||||
| 	Tracer tracer.Tracer |  | ||||||
| 	// Client for communication |  | ||||||
| 	Client client.Client |  | ||||||
| 	// Router for routing |  | ||||||
| 	Router router.Router |  | ||||||
| 	// Logger used for logging |  | ||||||
| 	Logger logger.Logger |  | ||||||
| 	// Meter used for metrics |  | ||||||
| 	Meter meter.Meter |  | ||||||
| 	// Links holds the communication links |  | ||||||
| 	Links map[string]client.Client |  | ||||||
| 	// Endpoint holds the destination address |  | ||||||
| 	Endpoint string |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Option func signature |  | ||||||
| type Option func(o *Options) |  | ||||||
|  |  | ||||||
| // NewOptions returns new options struct that filled by opts |  | ||||||
| func NewOptions(opts ...Option) Options { |  | ||||||
| 	options := Options{ |  | ||||||
| 		Logger: logger.DefaultLogger, |  | ||||||
| 		Meter:  meter.DefaultMeter, |  | ||||||
| 		Tracer: tracer.DefaultTracer, |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	for _, o := range opts { |  | ||||||
| 		o(&options) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return options |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // WithEndpoint sets a proxy endpoint |  | ||||||
| func WithEndpoint(e string) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.Endpoint = e |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // WithClient sets the client |  | ||||||
| func WithClient(c client.Client) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.Client = c |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // WithRouter specifies the router to use |  | ||||||
| func WithRouter(r router.Router) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.Router = r |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // WithLogger specifies the logger to use |  | ||||||
| func WithLogger(l logger.Logger) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.Logger = l |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // WithMeter specifies the meter to use |  | ||||||
| func WithMeter(m meter.Meter) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.Meter = m |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // WithLink sets a link for outbound requests |  | ||||||
| func WithLink(name string, c client.Client) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		if o.Links == nil { |  | ||||||
| 			o.Links = make(map[string]client.Client) |  | ||||||
| 		} |  | ||||||
| 		o.Links[name] = c |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Tracer to be used for tracing |  | ||||||
| func Tracer(t tracer.Tracer) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.Tracer = t |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -1,21 +0,0 @@ | |||||||
| // Package proxy is a transparent proxy built on the micro/server |  | ||||||
| package proxy |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"context" |  | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/server" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // DefaultEndpoint holds default proxy address |  | ||||||
| var DefaultEndpoint = "localhost:9090" |  | ||||||
|  |  | ||||||
| // Proxy can be used as a proxy server for micro services |  | ||||||
| type Proxy interface { |  | ||||||
| 	// ProcessMessage handles inbound messages |  | ||||||
| 	ProcessMessage(context.Context, server.Message) error |  | ||||||
| 	// ServeRequest handles inbound requests |  | ||||||
| 	ServeRequest(context.Context, server.Request, server.Response) error |  | ||||||
| 	// Name of the proxy protocol |  | ||||||
| 	String() string |  | ||||||
| } |  | ||||||
| @@ -15,6 +15,15 @@ func FromContext(ctx context.Context) (Register, bool) { | |||||||
| 	return c, ok | 	return c, ok | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // MustContext get register from context | ||||||
|  | func MustContext(ctx context.Context) Register { | ||||||
|  | 	r, ok := FromContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		panic("missing register") | ||||||
|  | 	} | ||||||
|  | 	return r | ||||||
|  | } | ||||||
|  |  | ||||||
| // NewContext put register in context | // NewContext put register in context | ||||||
| func NewContext(ctx context.Context, c Register) context.Context { | func NewContext(ctx context.Context, c Register) context.Context { | ||||||
| 	if ctx == nil { | 	if ctx == nil { | ||||||
|   | |||||||
Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user