Compare commits
	
		
			271 Commits
		
	
	
		
			v3.10.8
			...
			vtolstov-p
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
|  | 629087ec7e | ||
| c576abf26a | |||
| 70d6a79274 | |||
| 664b1586af | |||
| 8d747c64a8 | |||
| 94beb5ed3b | |||
| 98981ba86c | |||
| 1013f50d0e | |||
| 0b190997b1 | |||
| 69a44eb190 | |||
| 0476028f69 | |||
| 330d8b149a | |||
| 19b04fe070 | |||
| 4cd55875c6 | |||
| a7896cc728 | |||
| ff991bf49c | |||
| 5a6551b703 | |||
| 9406a33d60 | |||
| 8f185abd9d | |||
| 86492e0644 | |||
| b21972964a | |||
| f5ee065d09 | |||
| 8cb02f2b08 | |||
| bc926cd6bd | |||
| 356abfd818 | |||
| 18444d3f98 | |||
| d5f07922e8 | |||
| 675e717410 | |||
| 7b6aea235a | |||
| 2cb7200467 | |||
| f430f97a97 | |||
| 0060c4377a | |||
| e46579fe9a | |||
| ca52973194 | |||
| 5bb33c7e1d | |||
| b71fc25328 | |||
| 9345dd075a | |||
| 1c1b9c0a28 | |||
| 2969494c5a | |||
| cbd3fa38ba | |||
| 569a36383d | |||
| 90bed77526 | |||
| ba4478a5e0 | |||
| 6dc76cdfea | |||
| e266683d96 | |||
| 2b62ad04f2 | |||
| 275b0a64e5 | |||
| 38c5fe8b5a | |||
| b6a0e4d983 | |||
| d9b822deff | |||
| 0e66688f8f | |||
| 9213fd212f | |||
| aa360dcf51 | |||
| 2df259b5b8 | |||
| 15e9310368 | |||
|  | 16d8cf3434 | ||
| 9704ef2e5e | |||
| 94e8f90f00 | |||
| 34d1587881 | |||
| bf4143cde5 | |||
| 36b7b9f5fb | |||
| ae97023092 | |||
| 115ca6a018 | |||
| 89cf4ef8af | |||
| 2a6ce6d4da | |||
| ad19fe2b90 | |||
| 49055a28ea | |||
| d1c6e121c1 | |||
| 7cd7fb0c0a | |||
| 77eb5b5264 | |||
| 929e46c087 | |||
| 1fb5673d27 | |||
| 3bbb0cbc72 | |||
| 71fe0df73f | |||
| f1b8ecbdb3 | |||
| fd2b2762e9 | |||
| 82d269cfb4 | |||
| 6641463eed | |||
| faf2454f0a | |||
| de9e4d73f5 | |||
| 4ae7277140 | |||
| a98618ed5b | |||
| 3aaf1182cb | |||
| eb1482d789 | |||
| a305f7553f | |||
|  | d9b2f2a45d | ||
| 3ace7657dc | |||
| 53b40617e2 | |||
| 1a9236caad | |||
| 6c68d39081 | |||
| 35e62fbeb0 | |||
| 00b3ceb468 | |||
| 7dc8f088c9 | |||
| c65afcea1b | |||
| 3eebfb5b11 | |||
| fa1427014c | |||
| 62074965ee | |||
| 9c8fbb2202 | |||
| 7c0a5f5e2a | |||
| b08f5321b0 | |||
| cc0f24e012 | |||
| 307a08f50c | |||
| edc93e8c37 | |||
| 391813c260 | |||
| 1a1459dd0e | |||
| 4e99680c30 | |||
| 92a3a547b8 | |||
| 849c462037 | |||
| 54a55c83e2 | |||
| 781dee03db | |||
| 26dd2eb405 | |||
| 3a21069b86 | |||
| add3ce478c | |||
|  | c3de003e4a | ||
| 7b7cf18a65 | |||
| 1bcf71c189 | |||
| c320d8e518 | |||
| b5f8316b57 | |||
| d7ddd912a8 | |||
| c020d90cb4 | |||
| db47b62159 | |||
| 8254456c8b | |||
| c2808679c3 | |||
| f418235c16 | |||
| 67ba7b3753 | |||
| e48d7cadf9 | |||
| c906186011 | |||
| dc0ff91b83 | |||
| e739c2d438 | |||
| bf4a036652 | |||
| f83a29eb67 | |||
| aef7f53d88 | |||
| 02c8e4fb7f | |||
| f5693bd940 | |||
| 701afb7bea | |||
| 019b407e74 | |||
| f29a346434 | |||
| 27db1876c0 | |||
| f66ac9736b | |||
| ed7972a1fa | |||
| 2cc004b01c | |||
| df951e5daf | |||
| 5bec0cef03 | |||
| 34940b68d7 | |||
| 1c57127128 | |||
| a4dd1a494c | |||
| 60e5e42167 | |||
| b519b61fff | |||
| f62b26eda3 | |||
| 13eda451da | |||
| 89cad06121 | |||
| 0bebf3d59f | |||
| 01e05e8df6 | |||
| 2b69a4f51c | |||
| 4af2b077dd | |||
| de4418189d | |||
| 2c44550897 | |||
| 99b8a3c950 | |||
| 4c7e1607d4 | |||
| 897be419b4 | |||
| 81b9a4341f | |||
| d3bb2f7236 | |||
| 97fd62cb21 | |||
| 3cd8bc33d6 | |||
| f6f67af8d0 | |||
| 2d5acaca2f | |||
| 0674df3d9f | |||
| 2c282825ce | |||
| e87ff942bb | |||
| 0459ea0613 | |||
| d44a75d074 | |||
|  | ccf92eb84d | ||
| 6baf1f2744 | |||
| 8e2eafde9c | |||
| c2b97b0f20 | |||
| 1db017d966 | |||
| debf8cb03d | |||
| 1dc9c1891f | |||
| 930859a537 | |||
| 3141f1ed8b | |||
| 47943cfb05 | |||
| ed4e9d54b1 | |||
| b4b8583594 | |||
| fb43e8c58c | |||
| 8863c10ef4 | |||
| 8058095bcc | |||
| 092f5d96b1 | |||
| 84552513f7 | |||
| 80a2db264e | |||
| 0be09c8b3e | |||
| 047f479e1b | |||
| 8f757c953e | |||
| 5f1c673a24 | |||
| 6794ea9871 | |||
| 089e7b6812 | |||
| 1c703f0f0c | |||
| d167c8c67c | |||
| df4f96a2d8 | |||
| fac3b20bd4 | |||
| 7c6bd98498 | |||
| 23e1174f25 | |||
| 52bed214cf | |||
| 64c4f5f47e | |||
| 036c612137 | |||
| ca80e3ecf2 | |||
| 18e7bb41ca | |||
| 8e72fb1c35 | |||
| 17f21a03f4 | |||
| a076d43a26 | |||
| de6efaee0b | |||
| 9e0e657003 | |||
| be5f9ab77f | |||
| 144dca0cae | |||
| 75173560e3 | |||
| 9b3bccd1f1 | |||
| ce125b77c1 | |||
| 2ee8d4ed46 | |||
| f58781d076 | |||
| e1af4aa3a4 | |||
| 1d5e795443 | |||
| a3a434d923 | |||
| bcc06054f1 | |||
| 270d26f1ae | |||
| 646212cc08 | |||
| 00c2c749db | |||
| 2dbada0e94 | |||
| 7b8f4410fb | |||
| 45ebef5544 | |||
| cf4cac0733 | |||
| 50d60b5825 | |||
| 46ef491764 | |||
| a51b8b8102 | |||
| 15aac48f1e | |||
| 078069b2d7 | |||
| 258812304a | |||
| da5d50db5b | |||
| 384e4d113d | |||
| dfd1da7f0d | |||
| 8e5015e580 | |||
| bd0c309b71 | |||
| b4f0c3e29a | |||
| 8fddaa0455 | |||
| 2710c269a8 | |||
| 70ea93e466 | |||
| a87d0ab1c1 | |||
| 2e5e102719 | |||
| 36e492314d | |||
| 0c78873277 | |||
| 7f57dc09d3 | |||
| 447206d256 | |||
| 33a7feb970 | |||
| 3950f2ed86 | |||
| 68c1048a7d | |||
| bff40bd317 | |||
| 2878d0a4ea | |||
| 3138a9fded | |||
| 742b99636a | |||
| 34387f129d | |||
| 47075acb06 | |||
| 09cb998ba4 | |||
| b9dbfb1cfc | |||
| 56efccc4cf | |||
| ce9f896287 | |||
| 83d87a40e4 | |||
| 75fd1e43b9 | |||
| 395a3eed3d | |||
| 3ba8cb7f9e | |||
| b07806b9a1 | |||
| 0f583218d4 | |||
| f4d0237785 | |||
| 0f343dad0b | 
							
								
								
									
										18
									
								
								.gitea/ISSUE_TEMPLATE/bug_report.md
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										18
									
								
								.gitea/ISSUE_TEMPLATE/bug_report.md
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,18 @@ | ||||
| --- | ||||
| name: Bug report | ||||
| about: For reporting bugs in micro | ||||
| title: "[BUG]" | ||||
| labels: '' | ||||
| assignees: '' | ||||
|  | ||||
| --- | ||||
|  | ||||
| **Describe the bug** | ||||
|  | ||||
| 1. What are you trying to do? | ||||
| 2. What did you expect to happen? | ||||
| 3. What happens instead? | ||||
|  | ||||
| **How to reproduce the bug:** | ||||
|  | ||||
| If possible, please include a minimal code snippet here. | ||||
							
								
								
									
										17
									
								
								.gitea/ISSUE_TEMPLATE/feature-request---enhancement.md
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										17
									
								
								.gitea/ISSUE_TEMPLATE/feature-request---enhancement.md
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,17 @@ | ||||
| --- | ||||
| name: Feature request / Enhancement | ||||
| about: If you have a need not served by micro | ||||
| title: "[FEATURE]" | ||||
| labels: '' | ||||
| assignees: '' | ||||
|  | ||||
| --- | ||||
|  | ||||
| **Is your feature request related to a problem? Please describe.** | ||||
| A clear and concise description of what the problem is. Ex. I'm always frustrated when [...] | ||||
|  | ||||
| **Describe the solution you'd like** | ||||
| A clear and concise description of what you want to happen. | ||||
|  | ||||
| **Additional context** | ||||
| Add any other context or screenshots about the feature request here. | ||||
							
								
								
									
										8
									
								
								.gitea/ISSUE_TEMPLATE/question.md
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										8
									
								
								.gitea/ISSUE_TEMPLATE/question.md
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,8 @@ | ||||
| --- | ||||
| name: Question | ||||
| about: Ask a question about micro | ||||
| title: '' | ||||
| labels: '' | ||||
| assignees: '' | ||||
|  | ||||
| --- | ||||
							
								
								
									
										28
									
								
								.gitea/autoapprove.yml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										28
									
								
								.gitea/autoapprove.yml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,28 @@ | ||||
| name: "autoapprove" | ||||
|  | ||||
| on: | ||||
|   pull_request_target: | ||||
|     types: [assigned, opened, synchronize, reopened] | ||||
|   workflow_run: | ||||
|     workflows: ["prbuild"] | ||||
|     types: | ||||
|       - completed | ||||
|  | ||||
| permissions: | ||||
|   pull-requests: write | ||||
|   contents: write | ||||
|  | ||||
| jobs: | ||||
|   autoapprove: | ||||
|     runs-on: ubuntu-latest | ||||
|     steps: | ||||
|       - name: approve | ||||
|         run: [ "curl -o tea https://dl.gitea.com/tea/main/tea-main-linux-amd64", | ||||
|                "chmod +x ./tea", | ||||
|                "./tea login add --name unistack --token ${{ secrets.GITHUB_TOKEN }} --url https://git.unistack.org", | ||||
|                "./tea pr --repo ${{ github.event.repository.name }}" | ||||
|         ] | ||||
|         if: github.actor == 'vtolstov' | ||||
|         id: approve | ||||
|         with: | ||||
|           github-token: ${{ secrets.GITHUB_TOKEN }} | ||||
							
								
								
									
										51
									
								
								.gitea/workflows/job_coverage.yml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										51
									
								
								.gitea/workflows/job_coverage.yml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,51 @@ | ||||
| name: coverage | ||||
|  | ||||
| on: | ||||
|   push: | ||||
|     branches: [ main, v3, v4 ] | ||||
|   pull_request: | ||||
|     branches: [ main, v3, v4 ] | ||||
|   # Allows you to run this workflow manually from the Actions tab | ||||
|   workflow_dispatch: | ||||
|  | ||||
| jobs: | ||||
|  | ||||
|   build: | ||||
|     runs-on: ubuntu-latest | ||||
|     steps: | ||||
|     - name: checkout code | ||||
|       uses: actions/checkout@v4 | ||||
|       with: | ||||
|         filter: 'blob:none' | ||||
|  | ||||
|     - name: setup go | ||||
|       uses: actions/setup-go@v5 | ||||
|       with: | ||||
|         cache-dependency-path: "**/*.sum" | ||||
|         go-version: 'stable'  | ||||
|  | ||||
|     - name: test coverage | ||||
|       run: | | ||||
|         go test -v -cover ./... -coverprofile coverage.out -coverpkg ./... | ||||
|         go tool cover -func coverage.out -o coverage.out | ||||
|  | ||||
|     - name: coverage badge | ||||
|       uses: tj-actions/coverage-badge-go@v1 | ||||
|       with: | ||||
|         green: 80 | ||||
|         filename: coverage.out | ||||
|  | ||||
|     - uses: stefanzweifel/git-auto-commit-action@v4 | ||||
|       id: auto-commit-action | ||||
|       with: | ||||
|         commit_message: Apply Code Coverage Badge | ||||
|         skip_fetch: true | ||||
|         skip_checkout: true | ||||
|         file_pattern: ./README.md | ||||
|  | ||||
|     - name: Push Changes | ||||
|       if: steps.auto-commit-action.outputs.changes_detected == 'true' | ||||
|       uses: ad-m/github-push-action@master | ||||
|       with: | ||||
|         github_token: ${{ github.token }} | ||||
|         branch: ${{ github.ref }} | ||||
							
								
								
									
										29
									
								
								.gitea/workflows/job_lint.yml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										29
									
								
								.gitea/workflows/job_lint.yml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,29 @@ | ||||
| name: lint | ||||
|  | ||||
| on: | ||||
|   pull_request: | ||||
|     types: [opened, reopened, synchronize] | ||||
|     branches: | ||||
|     - master | ||||
|     - v3 | ||||
|     - v4 | ||||
|  | ||||
| jobs: | ||||
|   lint: | ||||
|     runs-on: ubuntu-latest | ||||
|     steps: | ||||
|     - name: checkout code | ||||
|       uses: actions/checkout@v4 | ||||
|       with: | ||||
|         filter: 'blob:none' | ||||
|     - name: setup go | ||||
|       uses: actions/setup-go@v5 | ||||
|       with: | ||||
|         cache-dependency-path: "**/*.sum" | ||||
|         go-version: 'stable'  | ||||
|     - name: setup deps | ||||
|       run: go get -v ./... | ||||
|     - name: run lint | ||||
|       uses: https://github.com/golangci/golangci-lint-action@v6 | ||||
|       with: | ||||
|         version: 'latest' | ||||
							
								
								
									
										34
									
								
								.gitea/workflows/job_test.yml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										34
									
								
								.gitea/workflows/job_test.yml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,34 @@ | ||||
| name: test | ||||
|  | ||||
| on: | ||||
|   pull_request: | ||||
|     types: [opened, reopened, synchronize] | ||||
|     branches: | ||||
|     - master | ||||
|     - v3 | ||||
|     - v4 | ||||
|   push: | ||||
|     branches: | ||||
|     - master | ||||
|     - v3 | ||||
|     - v4 | ||||
|  | ||||
| jobs: | ||||
|   test: | ||||
|     runs-on: ubuntu-latest | ||||
|     steps: | ||||
|     - name: checkout code | ||||
|       uses: actions/checkout@v4 | ||||
|       with: | ||||
|         filter: 'blob:none' | ||||
|     - name: setup go | ||||
|       uses: actions/setup-go@v5 | ||||
|       with: | ||||
|         cache-dependency-path: "**/*.sum" | ||||
|         go-version: 'stable' | ||||
|     - name: setup deps | ||||
|       run: go get -v ./... | ||||
|     - name: run test | ||||
|       env: | ||||
|         INTEGRATION_TESTS: yes | ||||
|       run: go test -mod readonly -v ./... | ||||
							
								
								
									
										53
									
								
								.gitea/workflows/job_tests.yml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										53
									
								
								.gitea/workflows/job_tests.yml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,53 @@ | ||||
| name: test | ||||
|  | ||||
| on: | ||||
|   pull_request: | ||||
|     types: [opened, reopened, synchronize] | ||||
|     branches: | ||||
|     - master | ||||
|     - v3 | ||||
|     - v4 | ||||
|   push: | ||||
|     branches: | ||||
|     - master | ||||
|     - v3 | ||||
|     - v4 | ||||
|  | ||||
| jobs: | ||||
|   test: | ||||
|     runs-on: ubuntu-latest | ||||
|     steps: | ||||
|     - name: checkout code | ||||
|       uses: actions/checkout@v4 | ||||
|       with: | ||||
|         filter: 'blob:none' | ||||
|     - name: checkout tests | ||||
|       uses: actions/checkout@v4 | ||||
|       with: | ||||
|         ref: master | ||||
|         filter: 'blob:none' | ||||
|         repository: unistack-org/micro-tests | ||||
|         path: micro-tests | ||||
|     - name: setup go | ||||
|       uses: actions/setup-go@v5 | ||||
|       with: | ||||
|         cache-dependency-path: "**/*.sum" | ||||
|         go-version: 'stable' | ||||
|     - name: setup go work | ||||
|       env: | ||||
|         GOWORK: /workspace/${{ github.repository_owner }}/go.work | ||||
|       run: | | ||||
|         go work init | ||||
|         go work use . | ||||
|         go work use micro-tests | ||||
|     - name: setup deps | ||||
|       env: | ||||
|         GOWORK: /workspace/${{ github.repository_owner }}/go.work | ||||
|       run: go get -v ./... | ||||
|     - name: run tests | ||||
|       env: | ||||
|         INTEGRATION_TESTS: yes | ||||
|         GOWORK: /workspace/${{ github.repository_owner }}/go.work | ||||
|       run: | | ||||
|         cd micro-tests | ||||
|         go test -mod readonly -v ./... || true | ||||
							
								
								
									
										9
									
								
								.github.old/PULL_REQUEST_TEMPLATE.md
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										9
									
								
								.github.old/PULL_REQUEST_TEMPLATE.md
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,9 @@ | ||||
| ## Pull Request template | ||||
| Please, go through these steps before clicking submit on this PR. | ||||
|  | ||||
| 1. Give a descriptive title to your PR. | ||||
| 2. Provide a description of your changes. | ||||
| 3. Make sure you have some relevant tests. | ||||
| 4. Put `closes #XXXX` in your comment to auto-close the issue that your PR fixes (if applicable). | ||||
|  | ||||
| **PLEASE REMOVE THIS TEMPLATE BEFORE SUBMITTING** | ||||
| @@ -37,11 +37,4 @@ jobs: | ||||
|         uses: golangci/golangci-lint-action@v3.4.0 | ||||
|         continue-on-error: true | ||||
|         with: | ||||
|           # Required: the version of golangci-lint is required and must be specified without patch version: we always use the latest patch version. | ||||
|           version: v1.30 | ||||
|           # Optional: working directory, useful for monorepos | ||||
|           # working-directory: somedir | ||||
|           # Optional: golangci-lint command line arguments. | ||||
|           # args: --issues-exit-code=0 | ||||
|           # Optional: show only new issues if it's a pull request. The default value is `false`. | ||||
|           # only-new-issues: true | ||||
							
								
								
									
										3
									
								
								.gitignore
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										3
									
								
								.gitignore
									
									
									
									
										vendored
									
									
								
							| @@ -1,6 +1,8 @@ | ||||
| # Develop tools | ||||
| /.vscode/ | ||||
| /.idea/ | ||||
| .idea | ||||
| .vscode | ||||
|  | ||||
| # Binaries for programs and plugins | ||||
| *.exe | ||||
| @@ -13,6 +15,7 @@ | ||||
| _obj | ||||
| _test | ||||
| _build | ||||
| .DS_Store | ||||
|  | ||||
| # Architecture specific extensions/prefixes | ||||
| *.[568vq] | ||||
|   | ||||
| @@ -1,44 +1,5 @@ | ||||
| run: | ||||
|   concurrency: 4 | ||||
|   concurrency: 8 | ||||
|   deadline: 5m | ||||
|   issues-exit-code: 1 | ||||
|   tests: true | ||||
|  | ||||
| linters-settings: | ||||
|   govet: | ||||
|     check-shadowing: true | ||||
|     enable: | ||||
|       - fieldalignment | ||||
|  | ||||
| linters: | ||||
|   enable: | ||||
|     - govet | ||||
|     - deadcode | ||||
|     - errcheck | ||||
|     - govet | ||||
|     - ineffassign | ||||
|     - staticcheck | ||||
|     - structcheck | ||||
|     - typecheck | ||||
|     - unused | ||||
|     - varcheck | ||||
|     - bodyclose | ||||
|     - gci | ||||
|     - goconst | ||||
|     - gocritic | ||||
|     - gosimple | ||||
|     - gofmt | ||||
|     - gofumpt | ||||
|     - goimports | ||||
|     - revive | ||||
|     - gosec | ||||
|     - makezero | ||||
|     - misspell | ||||
|     - nakedret | ||||
|     - nestif | ||||
|     - nilerr | ||||
|     - noctx | ||||
|     - prealloc | ||||
|     - unconvert | ||||
|     - unparam | ||||
|   disable-all: false | ||||
|   | ||||
| @@ -1,4 +1,5 @@ | ||||
| # Micro [](https://opensource.org/licenses/Apache-2.0) [](https://pkg.go.dev/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  | ||||
|  | ||||
|  | ||||
| Micro is a standard library for microservices. | ||||
|  | ||||
|   | ||||
							
								
								
									
										182
									
								
								api/api.go
									
									
									
									
									
								
							
							
						
						
									
										182
									
								
								api/api.go
									
									
									
									
									
								
							| @@ -1,182 +0,0 @@ | ||||
| package api // import "go.unistack.org/micro/v3/api" | ||||
|  | ||||
| import ( | ||||
| 	"errors" | ||||
| 	"regexp" | ||||
| 	"strings" | ||||
|  | ||||
| 	"go.unistack.org/micro/v3/metadata" | ||||
| 	"go.unistack.org/micro/v3/register" | ||||
| 	"go.unistack.org/micro/v3/server" | ||||
| ) | ||||
|  | ||||
| // nolint: revive | ||||
| // Api interface | ||||
| type Api interface { | ||||
| 	// Initialise options | ||||
| 	Init(...Option) error | ||||
| 	// Get the options | ||||
| 	Options() Options | ||||
| 	// Register a http handler | ||||
| 	Register(*Endpoint) error | ||||
| 	// Register a route | ||||
| 	Deregister(*Endpoint) error | ||||
| 	// Implementation of api | ||||
| 	String() string | ||||
| } | ||||
|  | ||||
| // Options holds the options | ||||
| type Options struct{} | ||||
|  | ||||
| // Option func signature | ||||
| type Option func(*Options) error | ||||
|  | ||||
| // Endpoint is a mapping between an RPC method and HTTP endpoint | ||||
| type Endpoint struct { | ||||
| 	// Name Greeter.Hello | ||||
| 	Name string | ||||
| 	// Desciption for endpoint | ||||
| 	Description string | ||||
| 	// Handler e.g rpc, proxy | ||||
| 	Handler string | ||||
| 	// Body destination | ||||
| 	// "*" or "" - top level message value | ||||
| 	// "string" - inner message value | ||||
| 	Body string | ||||
| 	// Host e.g example.com | ||||
| 	Host []string | ||||
| 	// Method e.g GET, POST | ||||
| 	Method []string | ||||
| 	// Path e.g /greeter. Expect POSIX regex | ||||
| 	Path []string | ||||
| 	// Stream flag | ||||
| 	Stream bool | ||||
| } | ||||
|  | ||||
| // Service represents an API service | ||||
| type Service struct { | ||||
| 	// Name of service | ||||
| 	Name string | ||||
| 	// Endpoint for this service | ||||
| 	Endpoint *Endpoint | ||||
| 	// Services that provides service | ||||
| 	Services []*register.Service | ||||
| } | ||||
|  | ||||
| // Encode encodes an endpoint to endpoint metadata | ||||
| func Encode(e *Endpoint) map[string]string { | ||||
| 	if e == nil { | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	// endpoint map | ||||
| 	ep := make(map[string]string) | ||||
|  | ||||
| 	// set vals only if they exist | ||||
| 	set := func(k, v string) { | ||||
| 		if len(v) == 0 { | ||||
| 			return | ||||
| 		} | ||||
| 		ep[k] = v | ||||
| 	} | ||||
|  | ||||
| 	set("endpoint", e.Name) | ||||
| 	set("description", e.Description) | ||||
| 	set("handler", e.Handler) | ||||
| 	set("method", strings.Join(e.Method, ",")) | ||||
| 	set("path", strings.Join(e.Path, ",")) | ||||
| 	set("host", strings.Join(e.Host, ",")) | ||||
| 	set("body", e.Body) | ||||
|  | ||||
| 	return ep | ||||
| } | ||||
|  | ||||
| // Decode decodes endpoint metadata into an endpoint | ||||
| func Decode(e metadata.Metadata) *Endpoint { | ||||
| 	if e == nil { | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	ep := &Endpoint{} | ||||
| 	ep.Name, _ = e.Get("endpoint") | ||||
| 	ep.Description, _ = e.Get("description") | ||||
| 	epmethod, _ := e.Get("method") | ||||
| 	ep.Method = []string{epmethod} | ||||
| 	eppath, _ := e.Get("path") | ||||
| 	ep.Path = []string{eppath} | ||||
| 	ephost, _ := e.Get("host") | ||||
| 	ep.Host = []string{ephost} | ||||
| 	ep.Handler, _ = e.Get("handler") | ||||
| 	ep.Body, _ = e.Get("body") | ||||
|  | ||||
| 	return ep | ||||
| } | ||||
|  | ||||
| // Validate validates an endpoint to guarantee it won't blow up when being served | ||||
| func Validate(e *Endpoint) error { | ||||
| 	if e == nil { | ||||
| 		return errors.New("endpoint is nil") | ||||
| 	} | ||||
|  | ||||
| 	if len(e.Name) == 0 { | ||||
| 		return errors.New("name required") | ||||
| 	} | ||||
|  | ||||
| 	for _, p := range e.Path { | ||||
| 		ps := p[0] | ||||
| 		pe := p[len(p)-1] | ||||
|  | ||||
| 		switch { | ||||
| 		case ps == '^' && pe == '$': | ||||
| 			if _, err := regexp.CompilePOSIX(p); err != nil { | ||||
| 				return err | ||||
| 			} | ||||
| 		case ps == '^' && pe != '$': | ||||
| 			return errors.New("invalid path") | ||||
| 		case ps != '^' && pe == '$': | ||||
| 			return errors.New("invalid path") | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	if len(e.Handler) == 0 { | ||||
| 		return errors.New("invalid handler") | ||||
| 	} | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| /* | ||||
| Design ideas | ||||
|  | ||||
| // Gateway is an api gateway interface | ||||
| type Gateway interface { | ||||
| 	// Register a http handler | ||||
| 	Handle(pattern string, http.Handler) | ||||
| 	// Register a route | ||||
| 	RegisterRoute(r Route) | ||||
| 	// Init initialises the command line. | ||||
| 	// It also parses further options. | ||||
| 	Init(...Option) error | ||||
| 	// Run the gateway | ||||
| 	Run() error | ||||
| } | ||||
|  | ||||
| // NewGateway returns a new api gateway | ||||
| func NewGateway() Gateway { | ||||
| 	return newGateway() | ||||
| } | ||||
| */ | ||||
|  | ||||
| // WithEndpoint returns a server.HandlerOption with endpoint metadata set | ||||
| // | ||||
| // Usage: | ||||
| // | ||||
| // 	proto.RegisterHandler(service.Server(), new(Handler), api.WithEndpoint( | ||||
| //		&api.Endpoint{ | ||||
| //			Name: "Greeter.Hello", | ||||
| //			Path: []string{"/greeter"}, | ||||
| //		}, | ||||
| //	)) | ||||
| func WithEndpoint(e *Endpoint) server.HandlerOption { | ||||
| 	return server.EndpointMetadata(e.Name, Encode(e)) | ||||
| } | ||||
							
								
								
									
										245
									
								
								api/api_test.go
									
									
									
									
									
								
							
							
						
						
									
										245
									
								
								api/api_test.go
									
									
									
									
									
								
							| @@ -1,245 +0,0 @@ | ||||
| package api | ||||
|  | ||||
| import ( | ||||
| 	"strings" | ||||
| 	"testing" | ||||
|  | ||||
| 	"go.unistack.org/micro/v3/metadata" | ||||
| 	"go.unistack.org/micro/v3/server" | ||||
| ) | ||||
|  | ||||
| func TestDecode(t *testing.T) { | ||||
| 	md := metadata.New(0) | ||||
| 	md.Set("host", "localhost", "method", "GET", "path", "/") | ||||
| 	ep := Decode(md) | ||||
| 	if md == nil { | ||||
| 		t.Fatalf("failed to decode md %#+v", md) | ||||
| 	} else if len(ep.Host) != 1 || len(ep.Method) != 1 || len(ep.Path) != 1 { | ||||
| 		t.Fatalf("ep invalid after decode %#+v", ep) | ||||
| 	} | ||||
| 	if ep.Host[0] != "localhost" || ep.Method[0] != "GET" || ep.Path[0] != "/" { | ||||
| 		t.Fatalf("ep invalid after decode %#+v", ep) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| //nolint:gocyclo | ||||
| func TestEncode(t *testing.T) { | ||||
| 	testData := []*Endpoint{ | ||||
| 		nil, | ||||
| 		{ | ||||
| 			Name:        "Foo.Bar", | ||||
| 			Description: "A test endpoint", | ||||
| 			Handler:     "meta", | ||||
| 			Host:        []string{"foo.com"}, | ||||
| 			Method:      []string{"GET"}, | ||||
| 			Path:        []string{"/test"}, | ||||
| 		}, | ||||
| 	} | ||||
|  | ||||
| 	compare := func(expect, got []string) bool { | ||||
| 		// no data to compare, return true | ||||
| 		if len(expect) == 0 && len(got) == 0 { | ||||
| 			return true | ||||
| 		} | ||||
| 		// no data expected but got some return false | ||||
| 		if len(expect) == 0 && len(got) > 0 { | ||||
| 			return false | ||||
| 		} | ||||
|  | ||||
| 		// compare expected with what we got | ||||
| 		for _, e := range expect { | ||||
| 			var seen bool | ||||
| 			for _, g := range got { | ||||
| 				if e == g { | ||||
| 					seen = true | ||||
| 					break | ||||
| 				} | ||||
| 			} | ||||
| 			if !seen { | ||||
| 				return false | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		// we're done, return true | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	for _, d := range testData { | ||||
| 		// encode | ||||
| 		e := Encode(d) | ||||
| 		// decode | ||||
| 		de := Decode(e) | ||||
|  | ||||
| 		// nil endpoint returns nil | ||||
| 		if d == nil { | ||||
| 			if e != nil { | ||||
| 				t.Fatalf("expected nil got %v", e) | ||||
| 			} | ||||
| 			if de != nil { | ||||
| 				t.Fatalf("expected nil got %v", de) | ||||
| 			} | ||||
|  | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		// check encoded map | ||||
| 		name := e["endpoint"] | ||||
| 		desc := e["description"] | ||||
| 		method := strings.Split(e["method"], ",") | ||||
| 		path := strings.Split(e["path"], ",") | ||||
| 		host := strings.Split(e["host"], ",") | ||||
| 		handler := e["handler"] | ||||
|  | ||||
| 		if name != d.Name { | ||||
| 			t.Fatalf("expected %v got %v", d.Name, name) | ||||
| 		} | ||||
| 		if desc != d.Description { | ||||
| 			t.Fatalf("expected %v got %v", d.Description, desc) | ||||
| 		} | ||||
| 		if handler != d.Handler { | ||||
| 			t.Fatalf("expected %v got %v", d.Handler, handler) | ||||
| 		} | ||||
| 		if ok := compare(d.Method, method); !ok { | ||||
| 			t.Fatalf("expected %v got %v", d.Method, method) | ||||
| 		} | ||||
| 		if ok := compare(d.Path, path); !ok { | ||||
| 			t.Fatalf("expected %v got %v", d.Path, path) | ||||
| 		} | ||||
| 		if ok := compare(d.Host, host); !ok { | ||||
| 			t.Fatalf("expected %v got %v", d.Host, host) | ||||
| 		} | ||||
|  | ||||
| 		if de.Name != d.Name { | ||||
| 			t.Fatalf("expected %v got %v", d.Name, de.Name) | ||||
| 		} | ||||
| 		if de.Description != d.Description { | ||||
| 			t.Fatalf("expected %v got %v", d.Description, de.Description) | ||||
| 		} | ||||
| 		if de.Handler != d.Handler { | ||||
| 			t.Fatalf("expected %v got %v", d.Handler, de.Handler) | ||||
| 		} | ||||
| 		if ok := compare(d.Method, de.Method); !ok { | ||||
| 			t.Fatalf("expected %v got %v", d.Method, de.Method) | ||||
| 		} | ||||
| 		if ok := compare(d.Path, de.Path); !ok { | ||||
| 			t.Fatalf("expected %v got %v", d.Path, de.Path) | ||||
| 		} | ||||
| 		if ok := compare(d.Host, de.Host); !ok { | ||||
| 			t.Fatalf("expected %v got %v", d.Host, de.Host) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestValidate(t *testing.T) { | ||||
| 	epPcre := &Endpoint{ | ||||
| 		Name:        "Foo.Bar", | ||||
| 		Description: "A test endpoint", | ||||
| 		Handler:     "meta", | ||||
| 		Host:        []string{"foo.com"}, | ||||
| 		Method:      []string{"GET"}, | ||||
| 		Path:        []string{"^/test/?$"}, | ||||
| 	} | ||||
| 	if err := Validate(epPcre); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	epGpath := &Endpoint{ | ||||
| 		Name:        "Foo.Bar", | ||||
| 		Description: "A test endpoint", | ||||
| 		Handler:     "meta", | ||||
| 		Host:        []string{"foo.com"}, | ||||
| 		Method:      []string{"GET"}, | ||||
| 		Path:        []string{"/test/{id}"}, | ||||
| 	} | ||||
| 	if err := Validate(epGpath); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	epPcreInvalid := &Endpoint{ | ||||
| 		Name:        "Foo.Bar", | ||||
| 		Description: "A test endpoint", | ||||
| 		Handler:     "meta", | ||||
| 		Host:        []string{"foo.com"}, | ||||
| 		Method:      []string{"GET"}, | ||||
| 		Path:        []string{"/test/?$"}, | ||||
| 	} | ||||
| 	if err := Validate(epPcreInvalid); err == nil { | ||||
| 		t.Fatalf("invalid pcre %v", epPcreInvalid.Path[0]) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestWithEndpoint(t *testing.T) { | ||||
| 	ep := &Endpoint{ | ||||
| 		Name:        "Foo.Bar", | ||||
| 		Description: "A test endpoint", | ||||
| 		Handler:     "meta", | ||||
| 		Host:        []string{"foo.com"}, | ||||
| 		Method:      []string{"GET"}, | ||||
| 		Path:        []string{"/test/{id}"}, | ||||
| 	} | ||||
| 	o := WithEndpoint(ep) | ||||
| 	opts := server.NewHandlerOptions(o) | ||||
| 	if opts.Metadata == nil { | ||||
| 		t.Fatalf("WithEndpoint not works %#+v", opts) | ||||
| 	} | ||||
| 	md, ok := opts.Metadata[ep.Name] | ||||
| 	if !ok { | ||||
| 		t.Fatalf("WithEndpoint not works %#+v", opts) | ||||
| 	} | ||||
| 	if v, ok := md.Get("Endpoint"); !ok || v != "Foo.Bar" { | ||||
| 		t.Fatalf("WithEndpoint not works %#+v", md) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestValidateNilErr(t *testing.T) { | ||||
| 	var ep *Endpoint | ||||
| 	if err := Validate(ep); err == nil { | ||||
| 		t.Fatalf("Validate not works") | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestValidateMissingNameErr(t *testing.T) { | ||||
| 	ep := &Endpoint{} | ||||
| 	if err := Validate(ep); err == nil { | ||||
| 		t.Fatalf("Validate not works") | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestValidateMissingHandlerErr(t *testing.T) { | ||||
| 	ep := &Endpoint{Name: "test"} | ||||
| 	if err := Validate(ep); err == nil { | ||||
| 		t.Fatalf("Validate not works") | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestValidateRegexpStartErr(t *testing.T) { | ||||
| 	ep := &Endpoint{Name: "test", Handler: "test"} | ||||
| 	ep.Path = []string{"^/"} | ||||
| 	if err := Validate(ep); err == nil { | ||||
| 		t.Fatalf("Validate not works") | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestValidateRegexpEndErr(t *testing.T) { | ||||
| 	ep := &Endpoint{Name: "test", Handler: "test", Path: []string{""}} | ||||
| 	ep.Path[0] = "/$" | ||||
| 	if err := Validate(ep); err == nil { | ||||
| 		t.Fatalf("Validate not works") | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestValidateRegexpNonErr(t *testing.T) { | ||||
| 	ep := &Endpoint{Name: "test", Handler: "test", Path: []string{""}} | ||||
| 	ep.Path[0] = "^/(.*)$" | ||||
| 	if err := Validate(ep); err != nil { | ||||
| 		t.Fatalf("Validate not works") | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestValidateRegexpErr(t *testing.T) { | ||||
| 	ep := &Endpoint{Name: "test", Handler: "test", Path: []string{""}} | ||||
| 	ep.Path[0] = "^/(.$" | ||||
| 	if err := Validate(ep); err == nil { | ||||
| 		t.Fatalf("Validate not works") | ||||
| 	} | ||||
| } | ||||
| @@ -1,21 +1,25 @@ | ||||
| // Package broker is an interface used for asynchronous messaging | ||||
| package broker // import "go.unistack.org/micro/v3/broker" | ||||
| package broker | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"errors" | ||||
| 	"time" | ||||
|  | ||||
| 	"go.unistack.org/micro/v3/codec" | ||||
| 	"go.unistack.org/micro/v3/metadata" | ||||
| ) | ||||
|  | ||||
| // DefaultBroker default memory broker | ||||
| var DefaultBroker = NewBroker() | ||||
| var DefaultBroker Broker = NewBroker() | ||||
|  | ||||
| var ( | ||||
| 	// ErrNotConnected returns when broker used but not connected yet | ||||
| 	ErrNotConnected = errors.New("broker not connected") | ||||
| 	// ErrDisconnected returns when broker disconnected | ||||
| 	ErrDisconnected = errors.New("broker disconnected") | ||||
| 	// DefaultGracefulTimeout | ||||
| 	DefaultGracefulTimeout = 5 * time.Second | ||||
| ) | ||||
|  | ||||
| // Broker is an interface used for asynchronous messaging. | ||||
| @@ -42,8 +46,25 @@ type Broker interface { | ||||
| 	BatchSubscribe(ctx context.Context, topic string, h BatchHandler, opts ...SubscribeOption) (Subscriber, error) | ||||
| 	// String type of broker | ||||
| 	String() string | ||||
| 	// Live returns broker liveness | ||||
| 	Live() bool | ||||
| 	// Ready returns broker readiness | ||||
| 	Ready() bool | ||||
| 	// Health returns broker health | ||||
| 	Health() bool | ||||
| } | ||||
|  | ||||
| type ( | ||||
| 	FuncPublish        func(ctx context.Context, topic string, msg *Message, opts ...PublishOption) error | ||||
| 	HookPublish        func(next FuncPublish) FuncPublish | ||||
| 	FuncBatchPublish   func(ctx context.Context, msgs []*Message, opts ...PublishOption) error | ||||
| 	HookBatchPublish   func(next FuncBatchPublish) FuncBatchPublish | ||||
| 	FuncSubscribe      func(ctx context.Context, topic string, h Handler, opts ...SubscribeOption) (Subscriber, error) | ||||
| 	HookSubscribe      func(next FuncSubscribe) FuncSubscribe | ||||
| 	FuncBatchSubscribe func(ctx context.Context, topic string, h BatchHandler, opts ...SubscribeOption) (Subscriber, error) | ||||
| 	HookBatchSubscribe func(next FuncBatchSubscribe) FuncBatchSubscribe | ||||
| ) | ||||
|  | ||||
| // Handler is used to process messages via a subscription of a topic. | ||||
| type Handler func(Event) error | ||||
|  | ||||
| @@ -73,6 +94,8 @@ 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 | ||||
| 	// Topic returns event topic | ||||
| 	Topic() string | ||||
| 	// Message returns broker message | ||||
| @@ -85,33 +108,12 @@ type Event interface { | ||||
| 	SetError(err error) | ||||
| } | ||||
|  | ||||
| // RawMessage is a raw encoded JSON value. | ||||
| // It implements Marshaler and Unmarshaler and can be used to delay decoding or precompute a encoding. | ||||
| type RawMessage []byte | ||||
|  | ||||
| // MarshalJSON returns m as the JSON encoding of m. | ||||
| func (m *RawMessage) MarshalJSON() ([]byte, error) { | ||||
| 	if m == nil { | ||||
| 		return []byte("null"), nil | ||||
| 	} | ||||
| 	return *m, nil | ||||
| } | ||||
|  | ||||
| // UnmarshalJSON sets *m to a copy of data. | ||||
| func (m *RawMessage) UnmarshalJSON(data []byte) error { | ||||
| 	if m == nil { | ||||
| 		return errors.New("RawMessage UnmarshalJSON on nil pointer") | ||||
| 	} | ||||
| 	*m = append((*m)[0:0], data...) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // Message is used to transfer data | ||||
| type Message struct { | ||||
| 	// Header contains message metadata | ||||
| 	Header metadata.Metadata | ||||
| 	// Body contains message body | ||||
| 	Body RawMessage | ||||
| 	Body codec.RawMessage | ||||
| } | ||||
|  | ||||
| // NewMessage create broker message with topic filled | ||||
|   | ||||
| @@ -15,6 +15,15 @@ func FromContext(ctx context.Context) (Broker, bool) { | ||||
| 	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 | ||||
| func NewContext(ctx context.Context, s Broker) context.Context { | ||||
| 	if ctx == nil { | ||||
|   | ||||
| @@ -4,8 +4,10 @@ import ( | ||||
| 	"context" | ||||
| 	"sync" | ||||
| 
 | ||||
| 	"go.unistack.org/micro/v3/broker" | ||||
| 	"go.unistack.org/micro/v3/logger" | ||||
| 	"go.unistack.org/micro/v3/metadata" | ||||
| 	"go.unistack.org/micro/v3/options" | ||||
| 	maddr "go.unistack.org/micro/v3/util/addr" | ||||
| 	"go.unistack.org/micro/v3/util/id" | ||||
| 	mnet "go.unistack.org/micro/v3/util/net" | ||||
| @@ -13,9 +15,13 @@ import ( | ||||
| ) | ||||
| 
 | ||||
| type memoryBroker struct { | ||||
| 	funcPublish        broker.FuncPublish | ||||
| 	funcBatchPublish   broker.FuncBatchPublish | ||||
| 	funcSubscribe      broker.FuncSubscribe | ||||
| 	funcBatchSubscribe broker.FuncBatchSubscribe | ||||
| 	subscribers        map[string][]*memorySubscriber | ||||
| 	addr               string | ||||
| 	opts        Options | ||||
| 	opts               broker.Options | ||||
| 	sync.RWMutex | ||||
| 	connected bool | ||||
| } | ||||
| @@ -24,20 +30,20 @@ type memoryEvent struct { | ||||
| 	err     error | ||||
| 	message interface{} | ||||
| 	topic   string | ||||
| 	opts    Options | ||||
| 	opts    broker.Options | ||||
| } | ||||
| 
 | ||||
| type memorySubscriber struct { | ||||
| 	ctx          context.Context | ||||
| 	exit         chan bool | ||||
| 	handler      Handler | ||||
| 	batchhandler BatchHandler | ||||
| 	handler      broker.Handler | ||||
| 	batchhandler broker.BatchHandler | ||||
| 	id           string | ||||
| 	topic        string | ||||
| 	opts         SubscribeOptions | ||||
| 	opts         broker.SubscribeOptions | ||||
| } | ||||
| 
 | ||||
| func (m *memoryBroker) Options() Options { | ||||
| func (m *memoryBroker) Options() broker.Options { | ||||
| 	return m.opts | ||||
| } | ||||
| 
 | ||||
| @@ -46,6 +52,12 @@ func (m *memoryBroker) Address() string { | ||||
| } | ||||
| 
 | ||||
| func (m *memoryBroker) Connect(ctx context.Context) error { | ||||
| 	select { | ||||
| 	case <-ctx.Done(): | ||||
| 		return ctx.Err() | ||||
| 	default: | ||||
| 	} | ||||
| 
 | ||||
| 	m.Lock() | ||||
| 	defer m.Unlock() | ||||
| 
 | ||||
| @@ -70,6 +82,12 @@ func (m *memoryBroker) Connect(ctx context.Context) error { | ||||
| } | ||||
| 
 | ||||
| func (m *memoryBroker) Disconnect(ctx context.Context) error { | ||||
| 	select { | ||||
| 	case <-ctx.Done(): | ||||
| 		return ctx.Err() | ||||
| 	default: | ||||
| 	} | ||||
| 
 | ||||
| 	m.Lock() | ||||
| 	defer m.Unlock() | ||||
| 
 | ||||
| @@ -81,27 +99,54 @@ func (m *memoryBroker) Disconnect(ctx context.Context) error { | ||||
| 	return nil | ||||
| } | ||||
| 
 | ||||
| func (m *memoryBroker) Init(opts ...Option) error { | ||||
| func (m *memoryBroker) Init(opts ...broker.Option) error { | ||||
| 	for _, o := range opts { | ||||
| 		o(&m.opts) | ||||
| 	} | ||||
| 
 | ||||
| 	m.funcPublish = m.fnPublish | ||||
| 	m.funcBatchPublish = m.fnBatchPublish | ||||
| 	m.funcSubscribe = m.fnSubscribe | ||||
| 	m.funcBatchSubscribe = m.fnBatchSubscribe | ||||
| 
 | ||||
| 	m.opts.Hooks.EachPrev(func(hook options.Hook) { | ||||
| 		switch h := hook.(type) { | ||||
| 		case broker.HookPublish: | ||||
| 			m.funcPublish = h(m.funcPublish) | ||||
| 		case broker.HookBatchPublish: | ||||
| 			m.funcBatchPublish = h(m.funcBatchPublish) | ||||
| 		case broker.HookSubscribe: | ||||
| 			m.funcSubscribe = h(m.funcSubscribe) | ||||
| 		case broker.HookBatchSubscribe: | ||||
| 			m.funcBatchSubscribe = h(m.funcBatchSubscribe) | ||||
| 		} | ||||
| 	}) | ||||
| 
 | ||||
| 	return nil | ||||
| } | ||||
| 
 | ||||
| func (m *memoryBroker) Publish(ctx context.Context, topic string, msg *Message, opts ...PublishOption) error { | ||||
| 	msg.Header.Set(metadata.HeaderTopic, topic) | ||||
| 	return m.publish(ctx, []*Message{msg}, opts...) | ||||
| func (m *memoryBroker) Publish(ctx context.Context, topic string, msg *broker.Message, opts ...broker.PublishOption) error { | ||||
| 	return m.funcPublish(ctx, topic, msg, opts...) | ||||
| } | ||||
| 
 | ||||
| func (m *memoryBroker) BatchPublish(ctx context.Context, msgs []*Message, opts ...PublishOption) error { | ||||
| func (m *memoryBroker) fnPublish(ctx context.Context, topic string, msg *broker.Message, opts ...broker.PublishOption) error { | ||||
| 	msg.Header.Set(metadata.HeaderTopic, topic) | ||||
| 	return m.publish(ctx, []*broker.Message{msg}, opts...) | ||||
| } | ||||
| 
 | ||||
| func (m *memoryBroker) BatchPublish(ctx context.Context, msgs []*broker.Message, opts ...broker.PublishOption) error { | ||||
| 	return m.funcBatchPublish(ctx, msgs, opts...) | ||||
| } | ||||
| 
 | ||||
| func (m *memoryBroker) fnBatchPublish(ctx context.Context, msgs []*broker.Message, opts ...broker.PublishOption) error { | ||||
| 	return m.publish(ctx, msgs, opts...) | ||||
| } | ||||
| 
 | ||||
| func (m *memoryBroker) publish(ctx context.Context, msgs []*Message, opts ...PublishOption) error { | ||||
| func (m *memoryBroker) publish(ctx context.Context, msgs []*broker.Message, opts ...broker.PublishOption) error { | ||||
| 	m.RLock() | ||||
| 	if !m.connected { | ||||
| 		m.RUnlock() | ||||
| 		return ErrNotConnected | ||||
| 		return broker.ErrNotConnected | ||||
| 	} | ||||
| 	m.RUnlock() | ||||
| 
 | ||||
| @@ -111,9 +156,9 @@ func (m *memoryBroker) publish(ctx context.Context, msgs []*Message, opts ...Pub | ||||
| 	case <-ctx.Done(): | ||||
| 		return ctx.Err() | ||||
| 	default: | ||||
| 		options := NewPublishOptions(opts...) | ||||
| 		options := broker.NewPublishOptions(opts...) | ||||
| 
 | ||||
| 		msgTopicMap := make(map[string]Events) | ||||
| 		msgTopicMap := make(map[string]broker.Events) | ||||
| 		for _, v := range msgs { | ||||
| 			p := &memoryEvent{opts: m.opts} | ||||
| 
 | ||||
| @@ -161,7 +206,7 @@ func (m *memoryBroker) publish(ctx context.Context, msgs []*Message, opts ...Pub | ||||
| 						} | ||||
| 					} else if sub.opts.AutoAck { | ||||
| 						if err = ms.Ack(); err != nil { | ||||
| 							m.opts.Logger.Errorf(m.opts.Context, "ack failed: %v", err) | ||||
| 							m.opts.Logger.Error(m.opts.Context, "broker ack error", err) | ||||
| 						} | ||||
| 					} | ||||
| 					// single processing | ||||
| @@ -172,11 +217,11 @@ func (m *memoryBroker) publish(ctx context.Context, msgs []*Message, opts ...Pub | ||||
| 							if eh != nil { | ||||
| 								_ = eh(p) | ||||
| 							} else if m.opts.Logger.V(logger.ErrorLevel) { | ||||
| 								m.opts.Logger.Error(m.opts.Context, err.Error()) | ||||
| 								m.opts.Logger.Error(m.opts.Context, "broker handler error", err) | ||||
| 							} | ||||
| 						} else if sub.opts.AutoAck { | ||||
| 							if err = p.Ack(); err != nil { | ||||
| 								m.opts.Logger.Errorf(m.opts.Context, "ack failed: %v", err) | ||||
| 								m.opts.Logger.Error(m.opts.Context, "broker ack error", err) | ||||
| 							} | ||||
| 						} | ||||
| 					} | ||||
| @@ -188,11 +233,15 @@ func (m *memoryBroker) publish(ctx context.Context, msgs []*Message, opts ...Pub | ||||
| 	return nil | ||||
| } | ||||
| 
 | ||||
| func (m *memoryBroker) BatchSubscribe(ctx context.Context, topic string, handler BatchHandler, opts ...SubscribeOption) (Subscriber, error) { | ||||
| func (m *memoryBroker) BatchSubscribe(ctx context.Context, topic string, handler broker.BatchHandler, opts ...broker.SubscribeOption) (broker.Subscriber, error) { | ||||
| 	return m.funcBatchSubscribe(ctx, topic, handler, opts...) | ||||
| } | ||||
| 
 | ||||
| func (m *memoryBroker) fnBatchSubscribe(ctx context.Context, topic string, handler broker.BatchHandler, opts ...broker.SubscribeOption) (broker.Subscriber, error) { | ||||
| 	m.RLock() | ||||
| 	if !m.connected { | ||||
| 		m.RUnlock() | ||||
| 		return nil, ErrNotConnected | ||||
| 		return nil, broker.ErrNotConnected | ||||
| 	} | ||||
| 	m.RUnlock() | ||||
| 
 | ||||
| @@ -201,7 +250,7 @@ func (m *memoryBroker) BatchSubscribe(ctx context.Context, topic string, handler | ||||
| 		return nil, err | ||||
| 	} | ||||
| 
 | ||||
| 	options := NewSubscribeOptions(opts...) | ||||
| 	options := broker.NewSubscribeOptions(opts...) | ||||
| 
 | ||||
| 	sub := &memorySubscriber{ | ||||
| 		exit:         make(chan bool, 1), | ||||
| @@ -233,11 +282,15 @@ func (m *memoryBroker) BatchSubscribe(ctx context.Context, topic string, handler | ||||
| 	return sub, nil | ||||
| } | ||||
| 
 | ||||
| func (m *memoryBroker) Subscribe(ctx context.Context, topic string, handler Handler, opts ...SubscribeOption) (Subscriber, error) { | ||||
| 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, ErrNotConnected | ||||
| 		return nil, broker.ErrNotConnected | ||||
| 	} | ||||
| 	m.RUnlock() | ||||
| 
 | ||||
| @@ -246,7 +299,7 @@ func (m *memoryBroker) Subscribe(ctx context.Context, topic string, handler Hand | ||||
| 		return nil, err | ||||
| 	} | ||||
| 
 | ||||
| 	options := NewSubscribeOptions(opts...) | ||||
| 	options := broker.NewSubscribeOptions(opts...) | ||||
| 
 | ||||
| 	sub := &memorySubscriber{ | ||||
| 		exit:    make(chan bool, 1), | ||||
| @@ -286,16 +339,28 @@ func (m *memoryBroker) Name() string { | ||||
| 	return m.opts.Name | ||||
| } | ||||
| 
 | ||||
| func (m *memoryBroker) Live() bool { | ||||
| 	return true | ||||
| } | ||||
| 
 | ||||
| func (m *memoryBroker) Ready() bool { | ||||
| 	return true | ||||
| } | ||||
| 
 | ||||
| func (m *memoryBroker) Health() bool { | ||||
| 	return true | ||||
| } | ||||
| 
 | ||||
| func (m *memoryEvent) Topic() string { | ||||
| 	return m.topic | ||||
| } | ||||
| 
 | ||||
| func (m *memoryEvent) Message() *Message { | ||||
| func (m *memoryEvent) Message() *broker.Message { | ||||
| 	switch v := m.message.(type) { | ||||
| 	case *Message: | ||||
| 	case *broker.Message: | ||||
| 		return v | ||||
| 	case []byte: | ||||
| 		msg := &Message{} | ||||
| 		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) | ||||
| @@ -320,7 +385,11 @@ func (m *memoryEvent) SetError(err error) { | ||||
| 	m.err = err | ||||
| } | ||||
| 
 | ||||
| func (m *memorySubscriber) Options() SubscribeOptions { | ||||
| func (m *memoryEvent) Context() context.Context { | ||||
| 	return m.opts.Context | ||||
| } | ||||
| 
 | ||||
| func (m *memorySubscriber) Options() broker.SubscribeOptions { | ||||
| 	return m.opts | ||||
| } | ||||
| 
 | ||||
| @@ -334,9 +403,9 @@ func (m *memorySubscriber) Unsubscribe(ctx context.Context) error { | ||||
| } | ||||
| 
 | ||||
| // NewBroker return new memory broker | ||||
| func NewBroker(opts ...Option) Broker { | ||||
| func NewBroker(opts ...broker.Option) broker.Broker { | ||||
| 	return &memoryBroker{ | ||||
| 		opts:        NewOptions(opts...), | ||||
| 		opts:        broker.NewOptions(opts...), | ||||
| 		subscribers: make(map[string][]*memorySubscriber), | ||||
| 	} | ||||
| } | ||||
| @@ -5,6 +5,7 @@ import ( | ||||
| 	"fmt" | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"go.unistack.org/micro/v3/broker" | ||||
| 	"go.unistack.org/micro/v3/metadata" | ||||
| ) | ||||
| 
 | ||||
| @@ -12,6 +13,10 @@ func TestMemoryBatchBroker(t *testing.T) { | ||||
| 	b := NewBroker() | ||||
| 	ctx := context.Background() | ||||
| 
 | ||||
| 	if err := b.Init(); err != nil { | ||||
| 		t.Fatalf("Unexpected init error %v", err) | ||||
| 	} | ||||
| 
 | ||||
| 	if err := b.Connect(ctx); err != nil { | ||||
| 		t.Fatalf("Unexpected connect error %v", err) | ||||
| 	} | ||||
| @@ -19,7 +24,7 @@ func TestMemoryBatchBroker(t *testing.T) { | ||||
| 	topic := "test" | ||||
| 	count := 10 | ||||
| 
 | ||||
| 	fn := func(evts Events) error { | ||||
| 	fn := func(evts broker.Events) error { | ||||
| 		return evts.Ack() | ||||
| 	} | ||||
| 
 | ||||
| @@ -28,9 +33,9 @@ func TestMemoryBatchBroker(t *testing.T) { | ||||
| 		t.Fatalf("Unexpected error subscribing %v", err) | ||||
| 	} | ||||
| 
 | ||||
| 	msgs := make([]*Message, 0, count) | ||||
| 	msgs := make([]*broker.Message, 0, count) | ||||
| 	for i := 0; i < count; i++ { | ||||
| 		message := &Message{ | ||||
| 		message := &broker.Message{ | ||||
| 			Header: map[string]string{ | ||||
| 				metadata.HeaderTopic: topic, | ||||
| 				"foo":                "bar", | ||||
| @@ -58,6 +63,10 @@ func TestMemoryBroker(t *testing.T) { | ||||
| 	b := NewBroker() | ||||
| 	ctx := context.Background() | ||||
| 
 | ||||
| 	if err := b.Init(); err != nil { | ||||
| 		t.Fatalf("Unexpected init error %v", err) | ||||
| 	} | ||||
| 
 | ||||
| 	if err := b.Connect(ctx); err != nil { | ||||
| 		t.Fatalf("Unexpected connect error %v", err) | ||||
| 	} | ||||
| @@ -65,7 +74,7 @@ func TestMemoryBroker(t *testing.T) { | ||||
| 	topic := "test" | ||||
| 	count := 10 | ||||
| 
 | ||||
| 	fn := func(p Event) error { | ||||
| 	fn := func(_ broker.Event) error { | ||||
| 		return nil | ||||
| 	} | ||||
| 
 | ||||
| @@ -74,9 +83,9 @@ func TestMemoryBroker(t *testing.T) { | ||||
| 		t.Fatalf("Unexpected error subscribing %v", err) | ||||
| 	} | ||||
| 
 | ||||
| 	msgs := make([]*Message, 0, count) | ||||
| 	msgs := make([]*broker.Message, 0, count) | ||||
| 	for i := 0; i < count; i++ { | ||||
| 		message := &Message{ | ||||
| 		message := &broker.Message{ | ||||
| 			Header: map[string]string{ | ||||
| 				metadata.HeaderTopic: topic, | ||||
| 				"foo":                "bar", | ||||
							
								
								
									
										140
									
								
								broker/noop.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										140
									
								
								broker/noop.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,140 @@ | ||||
| package broker | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"strings" | ||||
|  | ||||
| 	"go.unistack.org/micro/v3/options" | ||||
| ) | ||||
|  | ||||
| type NoopBroker struct { | ||||
| 	funcPublish        FuncPublish | ||||
| 	funcBatchPublish   FuncBatchPublish | ||||
| 	funcSubscribe      FuncSubscribe | ||||
| 	funcBatchSubscribe FuncBatchSubscribe | ||||
| 	opts               Options | ||||
| } | ||||
|  | ||||
| func NewBroker(opts ...Option) *NoopBroker { | ||||
| 	b := &NoopBroker{opts: NewOptions(opts...)} | ||||
| 	b.funcPublish = b.fnPublish | ||||
| 	b.funcBatchPublish = b.fnBatchPublish | ||||
| 	b.funcSubscribe = b.fnSubscribe | ||||
| 	b.funcBatchSubscribe = b.fnBatchSubscribe | ||||
|  | ||||
| 	return b | ||||
| } | ||||
|  | ||||
| func (b *NoopBroker) Health() bool { | ||||
| 	return true | ||||
| } | ||||
|  | ||||
| func (b *NoopBroker) Live() bool { | ||||
| 	return true | ||||
| } | ||||
|  | ||||
| func (b *NoopBroker) Ready() bool { | ||||
| 	return true | ||||
| } | ||||
|  | ||||
| func (b *NoopBroker) Name() string { | ||||
| 	return b.opts.Name | ||||
| } | ||||
|  | ||||
| func (b *NoopBroker) String() string { | ||||
| 	return "noop" | ||||
| } | ||||
|  | ||||
| func (b *NoopBroker) Options() Options { | ||||
| 	return b.opts | ||||
| } | ||||
|  | ||||
| func (b *NoopBroker) Init(opts ...Option) error { | ||||
| 	for _, opt := range opts { | ||||
| 		opt(&b.opts) | ||||
| 	} | ||||
|  | ||||
| 	b.funcPublish = b.fnPublish | ||||
| 	b.funcBatchPublish = b.fnBatchPublish | ||||
| 	b.funcSubscribe = b.fnSubscribe | ||||
| 	b.funcBatchSubscribe = b.fnBatchSubscribe | ||||
|  | ||||
| 	b.opts.Hooks.EachPrev(func(hook options.Hook) { | ||||
| 		switch h := hook.(type) { | ||||
| 		case HookPublish: | ||||
| 			b.funcPublish = h(b.funcPublish) | ||||
| 		case HookBatchPublish: | ||||
| 			b.funcBatchPublish = h(b.funcBatchPublish) | ||||
| 		case HookSubscribe: | ||||
| 			b.funcSubscribe = h(b.funcSubscribe) | ||||
| 		case HookBatchSubscribe: | ||||
| 			b.funcBatchSubscribe = h(b.funcBatchSubscribe) | ||||
| 		} | ||||
| 	}) | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *NoopBroker) Connect(_ context.Context) error { | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *NoopBroker) Disconnect(_ context.Context) error { | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *NoopBroker) Address() string { | ||||
| 	return strings.Join(b.opts.Addrs, ",") | ||||
| } | ||||
|  | ||||
| func (b *NoopBroker) fnBatchPublish(_ context.Context, _ []*Message, _ ...PublishOption) error { | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *NoopBroker) BatchPublish(ctx context.Context, msgs []*Message, opts ...PublishOption) error { | ||||
| 	return b.funcBatchPublish(ctx, msgs, opts...) | ||||
| } | ||||
|  | ||||
| func (b *NoopBroker) fnPublish(_ context.Context, _ string, _ *Message, _ ...PublishOption) error { | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *NoopBroker) Publish(ctx context.Context, topic string, msg *Message, opts ...PublishOption) error { | ||||
| 	return b.funcPublish(ctx, topic, msg, opts...) | ||||
| } | ||||
|  | ||||
| type NoopSubscriber struct { | ||||
| 	ctx          context.Context | ||||
| 	topic        string | ||||
| 	handler      Handler | ||||
| 	batchHandler BatchHandler | ||||
| 	opts         SubscribeOptions | ||||
| } | ||||
|  | ||||
| func (b *NoopBroker) fnBatchSubscribe(ctx context.Context, topic string, handler BatchHandler, opts ...SubscribeOption) (Subscriber, error) { | ||||
| 	return &NoopSubscriber{ctx: ctx, topic: topic, opts: NewSubscribeOptions(opts...), batchHandler: handler}, nil | ||||
| } | ||||
|  | ||||
| func (b *NoopBroker) BatchSubscribe(ctx context.Context, topic string, handler BatchHandler, opts ...SubscribeOption) (Subscriber, error) { | ||||
| 	return b.funcBatchSubscribe(ctx, topic, handler, opts...) | ||||
| } | ||||
|  | ||||
| func (b *NoopBroker) fnSubscribe(ctx context.Context, topic string, handler Handler, opts ...SubscribeOption) (Subscriber, error) { | ||||
| 	return &NoopSubscriber{ctx: ctx, topic: topic, opts: NewSubscribeOptions(opts...), handler: handler}, nil | ||||
| } | ||||
|  | ||||
| func (b *NoopBroker) Subscribe(ctx context.Context, topic string, handler Handler, opts ...SubscribeOption) (Subscriber, error) { | ||||
| 	return b.funcSubscribe(ctx, topic, handler, opts...) | ||||
| } | ||||
|  | ||||
| func (s *NoopSubscriber) Options() SubscribeOptions { | ||||
| 	return s.opts | ||||
| } | ||||
|  | ||||
| func (s *NoopSubscriber) Topic() string { | ||||
| 	return s.topic | ||||
| } | ||||
|  | ||||
| func (s *NoopSubscriber) Unsubscribe(_ context.Context) error { | ||||
| 	return nil | ||||
| } | ||||
							
								
								
									
										35
									
								
								broker/noop_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										35
									
								
								broker/noop_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,35 @@ | ||||
| package broker | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"testing" | ||||
| ) | ||||
|  | ||||
| type testHook struct { | ||||
| 	f bool | ||||
| } | ||||
|  | ||||
| func (t *testHook) Publish1(fn FuncPublish) FuncPublish { | ||||
| 	return func(ctx context.Context, topic string, msg *Message, opts ...PublishOption) error { | ||||
| 		t.f = true | ||||
| 		return fn(ctx, topic, msg, opts...) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestNoopHook(t *testing.T) { | ||||
| 	h := &testHook{} | ||||
|  | ||||
| 	b := NewBroker(Hooks(HookPublish(h.Publish1))) | ||||
|  | ||||
| 	if err := b.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	if err := b.Publish(context.TODO(), "", nil); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	if !h.f { | ||||
| 		t.Fatal("hook not works") | ||||
| 	} | ||||
| } | ||||
| @@ -8,12 +8,17 @@ import ( | ||||
| 	"go.unistack.org/micro/v3/codec" | ||||
| 	"go.unistack.org/micro/v3/logger" | ||||
| 	"go.unistack.org/micro/v3/meter" | ||||
| 	"go.unistack.org/micro/v3/options" | ||||
| 	"go.unistack.org/micro/v3/register" | ||||
| 	"go.unistack.org/micro/v3/sync" | ||||
| 	"go.unistack.org/micro/v3/tracer" | ||||
| ) | ||||
|  | ||||
| // Options struct | ||||
| type Options struct { | ||||
| 	// Name holds the broker name | ||||
| 	Name string | ||||
|  | ||||
| 	// Tracer used for tracing | ||||
| 	Tracer tracer.Tracer | ||||
| 	// Register can be used for clustering | ||||
| @@ -26,16 +31,25 @@ type Options struct { | ||||
| 	Meter meter.Meter | ||||
| 	// Context holds external options | ||||
| 	Context context.Context | ||||
|  | ||||
| 	// Wait waits for a collection of goroutines to finish | ||||
| 	Wait *sync.WaitGroup | ||||
| 	// 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 | ||||
| 	// Hooks can be run before broker Publish/BatchPublish and | ||||
| 	// Subscribe/BatchSubscribe methods | ||||
| 	Hooks options.Hooks | ||||
|  | ||||
| 	// GracefulTimeout contains time to wait to finish in flight requests | ||||
| 	GracefulTimeout time.Duration | ||||
| } | ||||
|  | ||||
| // NewOptions create new Options | ||||
| @@ -47,6 +61,7 @@ func NewOptions(opts ...Option) Options { | ||||
| 		Meter:           meter.DefaultMeter, | ||||
| 		Codec:           codec.DefaultCodec, | ||||
| 		Tracer:          tracer.DefaultTracer, | ||||
| 		GracefulTimeout: DefaultGracefulTimeout, | ||||
| 	} | ||||
| 	for _, o := range opts { | ||||
| 		o(&options) | ||||
| @@ -224,6 +239,13 @@ func Name(n string) Option { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Hooks sets hook runs before action | ||||
| func Hooks(h ...options.Hook) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.Hooks = append(o.Hooks, h...) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // SubscribeContext set context | ||||
| func SubscribeContext(ctx context.Context) SubscribeOption { | ||||
| 	return func(o *SubscribeOptions) { | ||||
|   | ||||
| @@ -17,13 +17,13 @@ func BackoffExp(_ context.Context, _ Request, attempts int) (time.Duration, erro | ||||
| } | ||||
|  | ||||
| // 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) { | ||||
| 		td := time.Duration(math.Pow(float64(attempts), math.E)) * time.Millisecond * 100 | ||||
| 		if td < min { | ||||
| 			return min, nil | ||||
| 		} else if td > max { | ||||
| 			return max, nil | ||||
| 		if td < minTime { | ||||
| 			return minTime, nil | ||||
| 		} else if td > maxTime { | ||||
| 			return maxTime, nil | ||||
| 		} | ||||
| 		return td, nil | ||||
| 	} | ||||
|   | ||||
| @@ -34,23 +34,23 @@ func TestBackoffExp(t *testing.T) { | ||||
| } | ||||
|  | ||||
| func TestBackoffInterval(t *testing.T) { | ||||
| 	min := 100 * time.Millisecond | ||||
| 	max := 300 * time.Millisecond | ||||
| 	minTime := 100 * time.Millisecond | ||||
| 	maxTime := 300 * time.Millisecond | ||||
|  | ||||
| 	r := &testRequest{ | ||||
| 		service: "test", | ||||
| 		method:  "test", | ||||
| 	} | ||||
|  | ||||
| 	fn := BackoffInterval(min, max) | ||||
| 	fn := BackoffInterval(minTime, maxTime) | ||||
| 	for i := 0; i < 5; i++ { | ||||
| 		d, err := fn(context.TODO(), r, i) | ||||
| 		if err != nil { | ||||
| 			t.Fatal(err) | ||||
| 		} | ||||
|  | ||||
| 		if d < min || d > max { | ||||
| 			t.Fatalf("Expected %v < %v < %v", min, d, max) | ||||
| 		if d < minTime || d > maxTime { | ||||
| 			t.Fatalf("Expected %v < %v < %v", minTime, d, maxTime) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -1,5 +1,5 @@ | ||||
| // Package client is an interface for an RPC client | ||||
| package client // import "go.unistack.org/micro/v3/client" | ||||
| package client | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| @@ -44,6 +44,17 @@ type Client interface { | ||||
| 	String() string | ||||
| } | ||||
|  | ||||
| type ( | ||||
| 	FuncCall         func(ctx context.Context, req Request, rsp interface{}, opts ...CallOption) error | ||||
| 	HookCall         func(next FuncCall) FuncCall | ||||
| 	FuncStream       func(ctx context.Context, req Request, opts ...CallOption) (Stream, error) | ||||
| 	HookStream       func(next FuncStream) FuncStream | ||||
| 	FuncPublish      func(ctx context.Context, msg Message, opts ...PublishOption) error | ||||
| 	HookPublish      func(next FuncPublish) FuncPublish | ||||
| 	FuncBatchPublish func(ctx context.Context, msg []Message, opts ...PublishOption) error | ||||
| 	HookBatchPublish func(next FuncBatchPublish) FuncBatchPublish | ||||
| ) | ||||
|  | ||||
| // Message is the interface for publishing asynchronously | ||||
| type Message interface { | ||||
| 	Topic() string | ||||
|   | ||||
| @@ -1,26 +0,0 @@ | ||||
| package client | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"testing" | ||||
| 	"time" | ||||
| ) | ||||
|  | ||||
| func TestNewClientCallOptions(t *testing.T) { | ||||
| 	var flag bool | ||||
| 	w := func(fn CallFunc) CallFunc { | ||||
| 		flag = true | ||||
| 		return fn | ||||
| 	} | ||||
| 	c := NewClientCallOptions(NewClient(), | ||||
| 		WithAddress("127.0.0.1"), | ||||
| 		WithCallWrapper(w), | ||||
| 		WithRequestTimeout(1*time.Millisecond), | ||||
| 		WithRetries(0), | ||||
| 		WithBackoff(BackoffInterval(10*time.Millisecond, 100*time.Millisecond)), | ||||
| 	) | ||||
| 	_ = c.Call(context.TODO(), c.NewRequest("service", "endpoint", nil), nil) | ||||
| 	if !flag { | ||||
| 		t.Fatalf("NewClientCallOptions not works") | ||||
| 	} | ||||
| } | ||||
| @@ -15,6 +15,15 @@ func FromContext(ctx context.Context) (Client, bool) { | ||||
| 	return c, ok | ||||
| } | ||||
|  | ||||
| // 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 | ||||
| func NewContext(ctx context.Context, c Client) context.Context { | ||||
| 	if ctx == nil { | ||||
|   | ||||
							
								
								
									
										186
									
								
								client/noop.go
									
									
									
									
									
								
							
							
						
						
									
										186
									
								
								client/noop.go
									
									
									
									
									
								
							| @@ -3,13 +3,18 @@ package client | ||||
| import ( | ||||
| 	"context" | ||||
| 	"fmt" | ||||
| 	"os" | ||||
| 	"strconv" | ||||
| 	"time" | ||||
|  | ||||
| 	"go.unistack.org/micro/v3/broker" | ||||
| 	"go.unistack.org/micro/v3/codec" | ||||
| 	"go.unistack.org/micro/v3/errors" | ||||
| 	"go.unistack.org/micro/v3/metadata" | ||||
| 	"go.unistack.org/micro/v3/options" | ||||
| 	"go.unistack.org/micro/v3/selector" | ||||
| 	"go.unistack.org/micro/v3/semconv" | ||||
| 	"go.unistack.org/micro/v3/tracer" | ||||
| ) | ||||
|  | ||||
| // DefaultCodecs will be used to encode/decode data | ||||
| @@ -18,6 +23,10 @@ var DefaultCodecs = map[string]codec.Codec{ | ||||
| } | ||||
|  | ||||
| type noopClient struct { | ||||
| 	funcPublish      FuncPublish | ||||
| 	funcBatchPublish FuncBatchPublish | ||||
| 	funcCall         FuncCall | ||||
| 	funcStream       FuncStream | ||||
| 	opts             Options | ||||
| } | ||||
|  | ||||
| @@ -39,16 +48,14 @@ type noopRequest struct { | ||||
|  | ||||
| // NewClient returns new noop client | ||||
| func NewClient(opts ...Option) Client { | ||||
| 	nc := &noopClient{opts: NewOptions(opts...)} | ||||
| 	// wrap in reverse | ||||
| 	n := &noopClient{opts: NewOptions(opts...)} | ||||
|  | ||||
| 	c := Client(nc) | ||||
| 	n.funcCall = n.fnCall | ||||
| 	n.funcStream = n.fnStream | ||||
| 	n.funcPublish = n.fnPublish | ||||
| 	n.funcBatchPublish = n.fnBatchPublish | ||||
|  | ||||
| 	for i := len(nc.opts.Wrappers); i > 0; i-- { | ||||
| 		c = nc.opts.Wrappers[i-1](c) | ||||
| 	} | ||||
|  | ||||
| 	return c | ||||
| 	return n | ||||
| } | ||||
|  | ||||
| func (n *noopClient) Name() string { | ||||
| @@ -100,10 +107,13 @@ func (n *noopResponse) Read() ([]byte, error) { | ||||
| 	return nil, nil | ||||
| } | ||||
|  | ||||
| type noopStream struct{} | ||||
| type noopStream struct { | ||||
| 	err error | ||||
| 	ctx context.Context | ||||
| } | ||||
|  | ||||
| func (n *noopStream) Context() context.Context { | ||||
| 	return context.Background() | ||||
| 	return n.ctx | ||||
| } | ||||
|  | ||||
| func (n *noopStream) Request() Request { | ||||
| @@ -131,15 +141,21 @@ func (n *noopStream) RecvMsg(interface{}) error { | ||||
| } | ||||
|  | ||||
| func (n *noopStream) Error() error { | ||||
| 	return nil | ||||
| 	return n.err | ||||
| } | ||||
|  | ||||
| func (n *noopStream) Close() error { | ||||
| 	return nil | ||||
| 	if sp, ok := tracer.SpanFromContext(n.ctx); ok && sp != nil { | ||||
| 		if n.err != nil { | ||||
| 			sp.SetStatus(tracer.SpanStatusError, n.err.Error()) | ||||
| 		} | ||||
| 		sp.Finish() | ||||
| 	} | ||||
| 	return n.err | ||||
| } | ||||
|  | ||||
| func (n *noopStream) CloseSend() error { | ||||
| 	return nil | ||||
| 	return n.err | ||||
| } | ||||
|  | ||||
| func (n *noopMessage) Topic() string { | ||||
| @@ -172,6 +188,25 @@ func (n *noopClient) Init(opts ...Option) error { | ||||
| 	for _, o := range opts { | ||||
| 		o(&n.opts) | ||||
| 	} | ||||
|  | ||||
| 	n.funcCall = n.fnCall | ||||
| 	n.funcStream = n.fnStream | ||||
| 	n.funcPublish = n.fnPublish | ||||
| 	n.funcBatchPublish = n.fnBatchPublish | ||||
|  | ||||
| 	n.opts.Hooks.EachPrev(func(hook options.Hook) { | ||||
| 		switch h := hook.(type) { | ||||
| 		case HookCall: | ||||
| 			n.funcCall = h(n.funcCall) | ||||
| 		case HookStream: | ||||
| 			n.funcStream = h(n.funcStream) | ||||
| 		case HookPublish: | ||||
| 			n.funcPublish = h(n.funcPublish) | ||||
| 		case HookBatchPublish: | ||||
| 			n.funcBatchPublish = h(n.funcBatchPublish) | ||||
| 		} | ||||
| 	}) | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| @@ -184,6 +219,31 @@ func (n *noopClient) String() string { | ||||
| } | ||||
|  | ||||
| func (n *noopClient) Call(ctx context.Context, req Request, rsp interface{}, opts ...CallOption) error { | ||||
| 	ts := time.Now() | ||||
| 	n.opts.Meter.Counter(semconv.ClientRequestInflight, "endpoint", req.Endpoint()).Inc() | ||||
| 	var sp tracer.Span | ||||
| 	ctx, sp = n.opts.Tracer.Start(ctx, "rpc-client", | ||||
| 		tracer.WithSpanKind(tracer.SpanKindClient), | ||||
| 		tracer.WithSpanLabels("endpoint", req.Endpoint()), | ||||
| 	) | ||||
| 	err := n.funcCall(ctx, req, rsp, opts...) | ||||
| 	n.opts.Meter.Counter(semconv.ClientRequestInflight, "endpoint", req.Endpoint()).Dec() | ||||
| 	te := time.Since(ts) | ||||
| 	n.opts.Meter.Summary(semconv.ClientRequestLatencyMicroseconds, "endpoint", req.Endpoint()).Update(te.Seconds()) | ||||
| 	n.opts.Meter.Histogram(semconv.ClientRequestDurationSeconds, "endpoint", req.Endpoint()).Update(te.Seconds()) | ||||
|  | ||||
| 	if me := errors.FromError(err); me == nil { | ||||
| 		sp.Finish() | ||||
| 		n.opts.Meter.Counter(semconv.ClientRequestTotal, "endpoint", req.Endpoint(), "status", "success", "code", strconv.Itoa(int(200))).Inc() | ||||
| 	} else { | ||||
| 		sp.SetStatus(tracer.SpanStatusError, err.Error()) | ||||
| 		n.opts.Meter.Counter(semconv.ClientRequestTotal, "endpoint", req.Endpoint(), "status", "failure", "code", strconv.Itoa(int(me.Code))).Inc() | ||||
| 	} | ||||
|  | ||||
| 	return err | ||||
| } | ||||
|  | ||||
| func (n *noopClient) fnCall(ctx context.Context, req Request, rsp interface{}, opts ...CallOption) error { | ||||
| 	// make a copy of call opts | ||||
| 	callOpts := n.opts.CallOptions | ||||
| 	for _, opt := range opts { | ||||
| @@ -212,11 +272,8 @@ func (n *noopClient) Call(ctx context.Context, req Request, rsp interface{}, opt | ||||
| 	} | ||||
|  | ||||
| 	// make copy of call method | ||||
| 	hcall := n.call | ||||
|  | ||||
| 	// wrap the call in reverse | ||||
| 	for i := len(callOpts.CallWrappers); i > 0; i-- { | ||||
| 		hcall = callOpts.CallWrappers[i-1](hcall) | ||||
| 	hcall := func(ctx context.Context, addr string, req Request, rsp interface{}, opts CallOptions) error { | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	// use the router passed as a call option, or fallback to the rpc clients router | ||||
| @@ -241,7 +298,7 @@ func (n *noopClient) Call(ctx context.Context, req Request, rsp interface{}, opt | ||||
| 		// call backoff first. Someone may want an initial start delay | ||||
| 		t, err := callOpts.Backoff(ctx, req, i) | ||||
| 		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 | ||||
| @@ -255,7 +312,7 @@ func (n *noopClient) Call(ctx context.Context, req Request, rsp interface{}, opt | ||||
| 			// TODO apply any filtering here | ||||
| 			routes, err = n.opts.Lookup(ctx, req, callOpts) | ||||
| 			if err != nil { | ||||
| 				return errors.InternalServerError("go.micro.client", err.Error()) | ||||
| 				return errors.InternalServerError("go.micro.client", "%s", err) | ||||
| 			} | ||||
|  | ||||
| 			// balance the list of nodes | ||||
| @@ -315,11 +372,7 @@ func (n *noopClient) Call(ctx context.Context, req Request, rsp interface{}, opt | ||||
| 	return gerr | ||||
| } | ||||
|  | ||||
| func (n *noopClient) call(ctx context.Context, addr string, req Request, rsp interface{}, opts CallOptions) error { | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| 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} | ||||
| } | ||||
|  | ||||
| @@ -329,6 +382,31 @@ func (n *noopClient) NewMessage(topic string, msg interface{}, opts ...MessageOp | ||||
| } | ||||
|  | ||||
| func (n *noopClient) Stream(ctx context.Context, req Request, opts ...CallOption) (Stream, error) { | ||||
| 	ts := time.Now() | ||||
| 	n.opts.Meter.Counter(semconv.ClientRequestInflight, "endpoint", req.Endpoint()).Inc() | ||||
| 	var sp tracer.Span | ||||
| 	ctx, sp = n.opts.Tracer.Start(ctx, "rpc-client", | ||||
| 		tracer.WithSpanKind(tracer.SpanKindClient), | ||||
| 		tracer.WithSpanLabels("endpoint", req.Endpoint()), | ||||
| 	) | ||||
| 	stream, err := n.funcStream(ctx, req, opts...) | ||||
| 	n.opts.Meter.Counter(semconv.ClientRequestInflight, "endpoint", req.Endpoint()).Dec() | ||||
| 	te := time.Since(ts) | ||||
| 	n.opts.Meter.Summary(semconv.ClientRequestLatencyMicroseconds, "endpoint", req.Endpoint()).Update(te.Seconds()) | ||||
| 	n.opts.Meter.Histogram(semconv.ClientRequestDurationSeconds, "endpoint", req.Endpoint()).Update(te.Seconds()) | ||||
|  | ||||
| 	if me := errors.FromError(err); me == nil { | ||||
| 		sp.Finish() | ||||
| 		n.opts.Meter.Counter(semconv.ClientRequestTotal, "endpoint", req.Endpoint(), "status", "success", "code", strconv.Itoa(int(200))).Inc() | ||||
| 	} else { | ||||
| 		sp.SetStatus(tracer.SpanStatusError, err.Error()) | ||||
| 		n.opts.Meter.Counter(semconv.ClientRequestTotal, "endpoint", req.Endpoint(), "status", "failure", "code", strconv.Itoa(int(me.Code))).Inc() | ||||
| 	} | ||||
|  | ||||
| 	return stream, err | ||||
| } | ||||
|  | ||||
| func (n *noopClient) fnStream(ctx context.Context, req Request, opts ...CallOption) (Stream, error) { | ||||
| 	var err error | ||||
|  | ||||
| 	// make a copy of call opts | ||||
| @@ -388,7 +466,7 @@ func (n *noopClient) Stream(ctx context.Context, req Request, opts ...CallOption | ||||
| 		// call backoff first. Someone may want an initial start delay | ||||
| 		t, cerr := callOpts.Backoff(ctx, req, i) | ||||
| 		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 | ||||
| @@ -402,7 +480,7 @@ func (n *noopClient) Stream(ctx context.Context, req Request, opts ...CallOption | ||||
| 			// TODO apply any filtering here | ||||
| 			routes, err = n.opts.Lookup(ctx, req, callOpts) | ||||
| 			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 | ||||
| @@ -468,15 +546,23 @@ func (n *noopClient) Stream(ctx context.Context, req Request, opts ...CallOption | ||||
| 	return nil, grr | ||||
| } | ||||
|  | ||||
| func (n *noopClient) stream(ctx context.Context, addr string, req Request, opts CallOptions) (Stream, error) { | ||||
| 	return &noopStream{}, nil | ||||
| func (n *noopClient) stream(ctx context.Context, _ string, _ Request, _ CallOptions) (Stream, error) { | ||||
| 	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...) | ||||
| } | ||||
|  | ||||
| @@ -485,21 +571,34 @@ func (n *noopClient) publish(ctx context.Context, ps []Message, opts ...PublishO | ||||
|  | ||||
| 	msgs := make([]*broker.Message, 0, len(ps)) | ||||
|  | ||||
| 	for _, p := range ps { | ||||
| 		md, ok := metadata.FromOutgoingContext(ctx) | ||||
| 		if !ok { | ||||
| 			md = metadata.New(0) | ||||
| 	// get proxy | ||||
| 	exchange := "" | ||||
| 	if v, ok := os.LookupEnv("MICRO_PROXY"); ok { | ||||
| 		exchange = v | ||||
| 	} | ||||
| 		md[metadata.HeaderContentType] = p.ContentType() | ||||
|  | ||||
| 		topic := p.Topic() | ||||
|  | ||||
| 	// get the exchange | ||||
| 	if len(options.Exchange) > 0 { | ||||
| 			topic = options.Exchange | ||||
| 		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 | ||||
|  | ||||
| @@ -510,13 +609,13 @@ func (n *noopClient) publish(ctx context.Context, ps []Message, opts ...PublishO | ||||
| 			// use codec for payload | ||||
| 			cf, err := n.newCodec(p.ContentType()) | ||||
| 			if err != nil { | ||||
| 				return errors.InternalServerError("go.micro.client", err.Error()) | ||||
| 				return errors.InternalServerError("go.micro.client", "%s", err) | ||||
| 			} | ||||
|  | ||||
| 			// set the body | ||||
| 			b, err := cf.Marshal(p.Payload()) | ||||
| 			if err != nil { | ||||
| 				return errors.InternalServerError("go.micro.client", err.Error()) | ||||
| 				return errors.InternalServerError("go.micro.client", "%s", err) | ||||
| 			} | ||||
| 			body = b | ||||
| 		} | ||||
| @@ -524,6 +623,13 @@ func (n *noopClient) publish(ctx context.Context, ps []Message, opts ...PublishO | ||||
| 		msgs = append(msgs, &broker.Message{Header: md, Body: body}) | ||||
| 	} | ||||
|  | ||||
| 	if len(msgs) == 1 { | ||||
| 		return n.opts.Broker.Publish(ctx, msgs[0].Header[metadata.HeaderTopic], msgs[0], | ||||
| 			broker.PublishContext(options.Context), | ||||
| 			broker.PublishBodyOnly(options.BodyOnly), | ||||
| 		) | ||||
| 	} | ||||
|  | ||||
| 	return n.opts.Broker.BatchPublish(ctx, msgs, | ||||
| 		broker.PublishContext(options.Context), | ||||
| 		broker.PublishBodyOnly(options.BodyOnly), | ||||
|   | ||||
							
								
								
									
										35
									
								
								client/noop_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										35
									
								
								client/noop_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,35 @@ | ||||
| package client | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"testing" | ||||
| ) | ||||
|  | ||||
| type testHook struct { | ||||
| 	f bool | ||||
| } | ||||
|  | ||||
| func (t *testHook) Publish(fn FuncPublish) FuncPublish { | ||||
| 	return func(ctx context.Context, msg Message, opts ...PublishOption) error { | ||||
| 		t.f = true | ||||
| 		return fn(ctx, msg, opts...) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestNoopHook(t *testing.T) { | ||||
| 	h := &testHook{} | ||||
|  | ||||
| 	c := NewClient(Hooks(HookPublish(h.Publish))) | ||||
|  | ||||
| 	if err := c.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	if err := c.Publish(context.TODO(), c.NewMessage("", nil, MessageContentType("application/octet-stream"))); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	if !h.f { | ||||
| 		t.Fatal("hook not works") | ||||
| 	} | ||||
| } | ||||
| @@ -11,7 +11,7 @@ import ( | ||||
| 	"go.unistack.org/micro/v3/logger" | ||||
| 	"go.unistack.org/micro/v3/metadata" | ||||
| 	"go.unistack.org/micro/v3/meter" | ||||
| 	"go.unistack.org/micro/v3/network/transport" | ||||
| 	"go.unistack.org/micro/v3/options" | ||||
| 	"go.unistack.org/micro/v3/register" | ||||
| 	"go.unistack.org/micro/v3/router" | ||||
| 	"go.unistack.org/micro/v3/selector" | ||||
| @@ -21,8 +21,16 @@ import ( | ||||
|  | ||||
| // Options holds client options | ||||
| type Options struct { | ||||
| 	// Transport used for transfer messages | ||||
| 	Transport transport.Transport | ||||
| 	// 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 selector.Selector | ||||
| 	// Logger used to log messages | ||||
| @@ -37,28 +45,28 @@ type Options struct { | ||||
| 	Context context.Context | ||||
| 	// Router used to get route | ||||
| 	Router router.Router | ||||
|  | ||||
| 	// TLSConfig specifies tls.Config for secure connection | ||||
| 	TLSConfig *tls.Config | ||||
| 	// Codecs map | ||||
| 	Codecs map[string]codec.Codec | ||||
|  | ||||
| 	// Lookup func used to get destination addr | ||||
| 	Lookup LookupFunc | ||||
| 	// Proxy is used for proxy requests | ||||
| 	Proxy string | ||||
| 	// ContentType is used to select codec | ||||
| 	ContentType string | ||||
| 	// Name is the client name | ||||
| 	Name string | ||||
| 	// ContextDialer used to connect | ||||
| 	ContextDialer func(context.Context, string) (net.Conn, error) | ||||
|  | ||||
| 	// Wrappers contains wrappers | ||||
| 	Wrappers []Wrapper | ||||
| 	// Hooks can be run before broker Publish/BatchPublish and | ||||
| 	// Subscribe/BatchSubscribe methods | ||||
| 	Hooks options.Hooks | ||||
|  | ||||
| 	// CallOptions contains default CallOptions | ||||
| 	CallOptions CallOptions | ||||
|  | ||||
| 	// PoolSize connection pool size | ||||
| 	PoolSize int | ||||
| 	// PoolTTL connection pool ttl | ||||
| 	PoolTTL time.Duration | ||||
| 	// ContextDialer used to connect | ||||
| 	ContextDialer func(context.Context, string) (net.Conn, error) | ||||
| } | ||||
|  | ||||
| // NewCallOptions creates new call options struct | ||||
| @@ -72,6 +80,16 @@ func NewCallOptions(opts ...CallOption) CallOptions { | ||||
|  | ||||
| // CallOptions holds client call options | ||||
| 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 selector.Selector | ||||
| 	// Context used for deadline | ||||
| @@ -79,35 +97,30 @@ type CallOptions struct { | ||||
| 	// Router used for route | ||||
| 	Router router.Router | ||||
| 	// Retry func used for retries | ||||
|  | ||||
| 	// ResponseMetadata holds additional metadata from call | ||||
| 	ResponseMetadata *metadata.Metadata | ||||
|  | ||||
| 	Retry RetryFunc | ||||
| 	// Backoff func used for backoff when retry | ||||
| 	Backoff BackoffFunc | ||||
| 	// Network name | ||||
| 	Network string | ||||
| 	// Content-Type | ||||
| 	ContentType string | ||||
| 	// AuthToken string | ||||
| 	AuthToken string | ||||
| 	// ContextDialer used to connect | ||||
| 	ContextDialer func(context.Context, string) (net.Conn, error) | ||||
|  | ||||
| 	// Address specifies static addr list | ||||
| 	Address []string | ||||
| 	// SelectOptions selector options | ||||
| 	SelectOptions []selector.SelectOption | ||||
| 	// CallWrappers call wrappers | ||||
| 	CallWrappers []CallWrapper | ||||
|  | ||||
| 	// StreamTimeout stream timeout | ||||
| 	StreamTimeout time.Duration | ||||
| 	// RequestTimeout request timeout | ||||
| 	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 time.Duration | ||||
| 	// Retries specifies retries num | ||||
| 	Retries int | ||||
| 	// ContextDialer used to connect | ||||
| 	ContextDialer func(context.Context, string) (net.Conn, error) | ||||
| } | ||||
|  | ||||
| // ContextDialer pass ContextDialer to client | ||||
| @@ -185,14 +198,13 @@ func NewOptions(opts ...Option) Options { | ||||
| 	options := Options{ | ||||
| 		Context:     context.Background(), | ||||
| 		ContentType: DefaultContentType, | ||||
| 		Codecs:      make(map[string]codec.Codec), | ||||
| 		Codecs:      DefaultCodecs, | ||||
| 		CallOptions: CallOptions{ | ||||
| 			Context:        context.Background(), | ||||
| 			Backoff:        DefaultBackoff, | ||||
| 			Retry:          DefaultRetry, | ||||
| 			Retries:        DefaultRetries, | ||||
| 			RequestTimeout: DefaultRequestTimeout, | ||||
| 			DialTimeout:    transport.DefaultDialTimeout, | ||||
| 		}, | ||||
| 		Lookup:   LookupRoute, | ||||
| 		PoolSize: DefaultPoolSize, | ||||
| @@ -203,7 +215,6 @@ func NewOptions(opts ...Option) Options { | ||||
| 		Meter:    meter.DefaultMeter, | ||||
| 		Tracer:   tracer.DefaultTracer, | ||||
| 		Router:   router.DefaultRouter, | ||||
| 		Transport: transport.DefaultTransport, | ||||
| 	} | ||||
|  | ||||
| 	for _, o := range opts { | ||||
| @@ -276,13 +287,6 @@ func PoolTTL(d time.Duration) Option { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Transport to use for communication e.g http, rabbitmq, etc | ||||
| func Transport(t transport.Transport) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.Transport = t | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Register sets the routers register | ||||
| func Register(r register.Register) Option { | ||||
| 	return func(o *Options) { | ||||
| @@ -306,20 +310,6 @@ func Selector(s selector.Selector) Option { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Wrap adds a wrapper to the list of options passed into the client | ||||
| func Wrap(w Wrapper) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.Wrappers = append(o.Wrappers, w) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // WrapCall adds a wrapper to the list of CallFunc wrappers | ||||
| func WrapCall(cw ...CallWrapper) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.CallOptions.CallWrappers = append(o.CallOptions.CallWrappers, cw...) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Backoff is used to set the backoff function used when retrying Calls | ||||
| func Backoff(fn BackoffFunc) Option { | ||||
| 	return func(o *Options) { | ||||
| @@ -346,14 +336,6 @@ func TLSConfig(t *tls.Config) Option { | ||||
| 	return func(o *Options) { | ||||
| 		// set the internal tls | ||||
| 		o.TLSConfig = t | ||||
|  | ||||
| 		// set the default transport if one is not | ||||
| 		// already set. Required for Init call below. | ||||
|  | ||||
| 		// set the transport tls | ||||
| 		_ = o.Transport.Init( | ||||
| 			transport.TLSConfig(t), | ||||
| 		) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| @@ -450,13 +432,6 @@ func WithAddress(a ...string) CallOption { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // WithCallWrapper is a CallOption which adds to the existing CallFunc wrappers | ||||
| func WithCallWrapper(cw ...CallWrapper) CallOption { | ||||
| 	return func(o *CallOptions) { | ||||
| 		o.CallWrappers = append(o.CallWrappers, cw...) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // WithBackoff is a CallOption which overrides that which | ||||
| // set in Options.CallOptions | ||||
| func WithBackoff(fn BackoffFunc) CallOption { | ||||
| @@ -526,13 +501,6 @@ func WithAuthToken(t string) CallOption { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // WithNetwork is a CallOption which sets the network attribute | ||||
| func WithNetwork(n string) CallOption { | ||||
| 	return func(o *CallOptions) { | ||||
| 		o.Network = n | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // WithRouter sets the router to use for this call | ||||
| func WithRouter(r router.Router) CallOption { | ||||
| 	return func(o *CallOptions) { | ||||
| @@ -591,3 +559,10 @@ func RequestContentType(ct string) RequestOption { | ||||
| 		o.ContentType = ct | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Hooks sets hook runs before action | ||||
| func Hooks(h ...options.Hook) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.Hooks = append(o.Hooks, h...) | ||||
| 	} | ||||
| } | ||||
|   | ||||
							
								
								
									
										47
									
								
								cluster/cluster.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										47
									
								
								cluster/cluster.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,47 @@ | ||||
| package cluster | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
|  | ||||
| 	"go.unistack.org/micro/v3/metadata" | ||||
| ) | ||||
|  | ||||
| // Message sent to member in cluster | ||||
| type Message interface { | ||||
| 	// Header returns message headers | ||||
| 	Header() metadata.Metadata | ||||
| 	// Body returns broker message may be []byte slice or some go struct or interface | ||||
| 	Body() interface{} | ||||
| } | ||||
|  | ||||
| type Node interface { | ||||
| 	// Name returns node name | ||||
| 	Name() string | ||||
| 	// Address returns node address | ||||
| 	Address() string | ||||
| 	// Metadata returns node metadata | ||||
| 	Metadata() metadata.Metadata | ||||
| } | ||||
|  | ||||
| // Cluster interface used for cluster communication across nodes | ||||
| type Cluster interface { | ||||
| 	// Join is used to take an existing members and performing state sync | ||||
| 	Join(ctx context.Context, addr ...string) error | ||||
| 	// Leave broadcast a leave message and stop listeners | ||||
| 	Leave(ctx context.Context) error | ||||
| 	// Ping is used to probe live status of the node | ||||
| 	Ping(ctx context.Context, node Node, payload []byte) error | ||||
| 	// Members returns the cluster members | ||||
| 	Members() ([]Node, error) | ||||
| 	// Broadcast send message for all members in cluster, if filter is not nil, nodes may be filtered | ||||
| 	// by key/value pairs | ||||
| 	Broadcast(ctx context.Context, msg Message, filter ...string) error | ||||
| 	// Unicast send message to single member in cluster | ||||
| 	Unicast(ctx context.Context, node Node, msg Message) error | ||||
| 	// Live returns cluster liveness | ||||
| 	Live() bool | ||||
| 	// Ready returns cluster readiness | ||||
| 	Ready() bool | ||||
| 	// Health returns cluster health | ||||
| 	Health() bool | ||||
| } | ||||
| @@ -1,19 +1,8 @@ | ||||
| // Package codec is an interface for encoding messages | ||||
| package codec // import "go.unistack.org/micro/v3/codec" | ||||
| package codec | ||||
|  | ||||
| import ( | ||||
| 	"errors" | ||||
| 	"io" | ||||
|  | ||||
| 	"go.unistack.org/micro/v3/metadata" | ||||
| ) | ||||
|  | ||||
| // Message types | ||||
| const ( | ||||
| 	Error MessageType = iota | ||||
| 	Request | ||||
| 	Response | ||||
| 	Event | ||||
| ) | ||||
|  | ||||
| var ( | ||||
| @@ -24,63 +13,44 @@ var ( | ||||
| ) | ||||
|  | ||||
| var ( | ||||
| 	// DefaultMaxMsgSize specifies how much data codec can handle | ||||
| 	DefaultMaxMsgSize = 1024 * 1024 * 4 // 4Mb | ||||
| 	// DefaultCodec is the global default codec | ||||
| 	DefaultCodec = NewCodec() | ||||
| 	// DefaultTagName specifies struct tag name to control codec Marshal/Unmarshal | ||||
| 	DefaultTagName = "codec" | ||||
| ) | ||||
|  | ||||
| // MessageType specifies message type for codec | ||||
| type MessageType int | ||||
|  | ||||
| // Codec encodes/decodes various types of messages used within micro. | ||||
| // ReadHeader and ReadBody are called in pairs to read requests/responses | ||||
| // from the connection. Close is called when finished with the | ||||
| // connection. ReadBody may be called with a nil argument to force the | ||||
| // body to be read and discarded. | ||||
| // Codec encodes/decodes various types of messages. | ||||
| type Codec interface { | ||||
| 	ReadHeader(r io.Reader, m *Message, mt MessageType) error | ||||
| 	ReadBody(r io.Reader, v interface{}) error | ||||
| 	Write(w io.Writer, m *Message, v interface{}) error | ||||
| 	Marshal(v interface{}, opts ...Option) ([]byte, error) | ||||
| 	Unmarshal(b []byte, v interface{}, opts ...Option) error | ||||
| 	String() string | ||||
| } | ||||
|  | ||||
| // Message represents detailed information about | ||||
| // the communication, likely followed by the body. | ||||
| // In the case of an error, body may be nil. | ||||
| type Message struct { | ||||
| 	Header   metadata.Metadata | ||||
| 	Target   string | ||||
| 	Method   string | ||||
| 	Endpoint string | ||||
| 	Error    string | ||||
| 	ID       string | ||||
| 	Body     []byte | ||||
| 	Type     MessageType | ||||
| type CodecV2 interface { | ||||
| 	Marshal(buf []byte, v interface{}, opts ...Option) ([]byte, error) | ||||
| 	Unmarshal(buf []byte, v interface{}, opts ...Option) error | ||||
| 	String() string | ||||
| } | ||||
|  | ||||
| // NewMessage creates new codec message | ||||
| func NewMessage(t MessageType) *Message { | ||||
| 	return &Message{Type: t, Header: metadata.New(0)} | ||||
| } | ||||
| // RawMessage is a raw encoded JSON value. | ||||
| // It implements Marshaler and Unmarshaler and can be used to delay decoding or precompute a encoding. | ||||
| type RawMessage []byte | ||||
|  | ||||
| // MarshalAppend calls codec.Marshal(v) and returns the data appended to buf. | ||||
| // If codec implements MarshalAppend, that is called instead. | ||||
| func MarshalAppend(buf []byte, c Codec, v interface{}, opts ...Option) ([]byte, error) { | ||||
| 	if nc, ok := c.(interface { | ||||
| 		MarshalAppend([]byte, interface{}, ...Option) ([]byte, error) | ||||
| 	}); ok { | ||||
| 		return nc.MarshalAppend(buf, v, opts...) | ||||
| // MarshalJSON returns m as the JSON encoding of m. | ||||
| func (m *RawMessage) MarshalJSON() ([]byte, error) { | ||||
| 	if m == nil { | ||||
| 		return []byte("null"), nil | ||||
| 	} else if len(*m) == 0 { | ||||
| 		return []byte("null"), nil | ||||
| 	} | ||||
|  | ||||
| 	mbuf, err := c.Marshal(v, opts...) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
|  | ||||
| 	return append(buf, mbuf...), nil | ||||
| 	return *m, nil | ||||
| } | ||||
|  | ||||
| // UnmarshalJSON sets *m to a copy of data. | ||||
| func (m *RawMessage) UnmarshalJSON(data []byte) error { | ||||
| 	if m == nil { | ||||
| 		return errors.New("RawMessage UnmarshalJSON on nil pointer") | ||||
| 	} | ||||
| 	*m = append((*m)[0:0], data...) | ||||
| 	return nil | ||||
| } | ||||
|   | ||||
| @@ -15,6 +15,15 @@ func FromContext(ctx context.Context) (Codec, bool) { | ||||
| 	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 | ||||
| func NewContext(ctx context.Context, c Codec) context.Context { | ||||
| 	if ctx == nil { | ||||
|   | ||||
| @@ -2,70 +2,14 @@ package codec | ||||
|  | ||||
| import ( | ||||
| 	"encoding/json" | ||||
| 	"io" | ||||
|  | ||||
| 	codecpb "go.unistack.org/micro-proto/v3/codec" | ||||
| ) | ||||
|  | ||||
| type noopCodec struct { | ||||
| 	opts Options | ||||
| } | ||||
|  | ||||
| func (c *noopCodec) ReadHeader(conn io.Reader, m *Message, t MessageType) error { | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (c *noopCodec) ReadBody(conn io.Reader, b interface{}) error { | ||||
| 	// read bytes | ||||
| 	buf, err := io.ReadAll(conn) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	if b == nil { | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	switch v := b.(type) { | ||||
| 	case *string: | ||||
| 		*v = string(buf) | ||||
| 	case *[]byte: | ||||
| 		*v = buf | ||||
| 	case *Frame: | ||||
| 		v.Data = buf | ||||
| 	default: | ||||
| 		return json.Unmarshal(buf, v) | ||||
| 	} | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (c *noopCodec) Write(conn io.Writer, m *Message, b interface{}) error { | ||||
| 	if b == nil { | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	var v []byte | ||||
| 	switch vb := b.(type) { | ||||
| 	case *Frame: | ||||
| 		v = vb.Data | ||||
| 	case string: | ||||
| 		v = []byte(vb) | ||||
| 	case *string: | ||||
| 		v = []byte(*vb) | ||||
| 	case *[]byte: | ||||
| 		v = *vb | ||||
| 	case []byte: | ||||
| 		v = vb | ||||
| 	default: | ||||
| 		var err error | ||||
| 		v, err = json.Marshal(vb) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 	} | ||||
| 	_, err := conn.Write(v) | ||||
| 	return err | ||||
| } | ||||
|  | ||||
| func (c *noopCodec) String() string { | ||||
| 	return "noop" | ||||
| } | ||||
| @@ -91,8 +35,8 @@ func (c *noopCodec) Marshal(v interface{}, opts ...Option) ([]byte, error) { | ||||
| 		return ve, nil | ||||
| 	case *Frame: | ||||
| 		return ve.Data, nil | ||||
| 	case *Message: | ||||
| 		return ve.Body, nil | ||||
| 	case *codecpb.Frame: | ||||
| 		return ve.Data, nil | ||||
| 	} | ||||
|  | ||||
| 	return json.Marshal(v) | ||||
| @@ -115,8 +59,8 @@ func (c *noopCodec) Unmarshal(d []byte, v interface{}, opts ...Option) error { | ||||
| 	case *Frame: | ||||
| 		ve.Data = d | ||||
| 		return nil | ||||
| 	case *Message: | ||||
| 		ve.Body = d | ||||
| 	case *codecpb.Frame: | ||||
| 		ve.Data = d | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
|   | ||||
| @@ -23,15 +23,8 @@ type Options struct { | ||||
| 	Context context.Context | ||||
| 	// TagName specifies tag name in struct to control codec | ||||
| 	TagName string | ||||
| 	// MaxMsgSize specifies max messages size that reads by codec | ||||
| 	MaxMsgSize int | ||||
| } | ||||
|  | ||||
| // MaxMsgSize sets the max message size | ||||
| func MaxMsgSize(n int) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.MaxMsgSize = n | ||||
| 	} | ||||
| 	// Flatten specifies that struct must be analyzed for flatten tag | ||||
| 	Flatten bool | ||||
| } | ||||
|  | ||||
| // TagName sets the codec tag name in struct | ||||
| @@ -41,6 +34,13 @@ func TagName(n string) Option { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Flatten enables checking for flatten tag name | ||||
| func Flatten(b bool) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.Flatten = b | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Logger sets the logger | ||||
| func Logger(l logger.Logger) Option { | ||||
| 	return func(o *Options) { | ||||
| @@ -69,8 +69,8 @@ func NewOptions(opts ...Option) Options { | ||||
| 		Logger:  logger.DefaultLogger, | ||||
| 		Meter:   meter.DefaultMeter, | ||||
| 		Tracer:  tracer.DefaultTracer, | ||||
| 		MaxMsgSize: DefaultMaxMsgSize, | ||||
| 		TagName: DefaultTagName, | ||||
| 		Flatten: false, | ||||
| 	} | ||||
|  | ||||
| 	for _, o := range opts { | ||||
|   | ||||
							
								
								
									
										107
									
								
								config/config.go
									
									
									
									
									
								
							
							
						
						
									
										107
									
								
								config/config.go
									
									
									
									
									
								
							| @@ -1,5 +1,5 @@ | ||||
| // Package config is an interface for dynamic configuration. | ||||
| package config // import "go.unistack.org/micro/v3/config" | ||||
| package config | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| @@ -50,6 +50,13 @@ type Config interface { | ||||
| 	String() string | ||||
| } | ||||
|  | ||||
| type ( | ||||
| 	FuncLoad func(ctx context.Context, opts ...LoadOption) error | ||||
| 	HookLoad func(next FuncLoad) FuncLoad | ||||
| 	FuncSave func(ctx context.Context, opts ...SaveOption) error | ||||
| 	HookSave func(next FuncSave) FuncSave | ||||
| ) | ||||
|  | ||||
| // Watcher is the config watcher | ||||
| type Watcher interface { | ||||
| 	// Next blocks until update happens or error returned | ||||
| @@ -124,35 +131,14 @@ func Validate(ctx context.Context, cfg interface{}) error { | ||||
| } | ||||
|  | ||||
| var ( | ||||
| 	// DefaultAfterLoad default func that runs after config load | ||||
| 	DefaultAfterLoad = func(ctx context.Context, c Config) error { | ||||
| 		for _, fn := range c.Options().AfterLoad { | ||||
| 			if err := fn(ctx, c); err != nil { | ||||
| 				c.Options().Logger.Errorf(ctx, "%s AfterLoad err: %v", c.String(), err) | ||||
| 				if !c.Options().AllowFail { | ||||
| 					return err | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 		return nil | ||||
| 	} | ||||
| 	// DefaultAfterSave default func that runs after config save | ||||
| 	DefaultAfterSave = func(ctx context.Context, c Config) error { | ||||
| 		for _, fn := range c.Options().AfterSave { | ||||
| 			if err := fn(ctx, c); err != nil { | ||||
| 				c.Options().Logger.Errorf(ctx, "%s AfterSave err: %v", c.String(), err) | ||||
| 				if !c.Options().AllowFail { | ||||
| 					return err | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 		return nil | ||||
| 	} | ||||
| 	// DefaultBeforeLoad default func that runs before config load | ||||
| 	// DefaultBeforeLoad default func that runs before config Load | ||||
| 	DefaultBeforeLoad = func(ctx context.Context, c Config) error { | ||||
| 		for _, fn := range c.Options().BeforeLoad { | ||||
| 			if fn == nil { | ||||
| 				return nil | ||||
| 			} | ||||
| 			if err := fn(ctx, c); err != nil { | ||||
| 				c.Options().Logger.Errorf(ctx, "%s BeforeLoad err: %v", c.String(), err) | ||||
| 				c.Options().Logger.Error(ctx, c.String()+" BeforeLoad error", err) | ||||
| 				if !c.Options().AllowFail { | ||||
| 					return err | ||||
| 				} | ||||
| @@ -160,11 +146,74 @@ var ( | ||||
| 		} | ||||
| 		return nil | ||||
| 	} | ||||
| 	// DefaultBeforeSave default func that runs befora config save | ||||
| 	// DefaultAfterLoad default func that runs after config Load | ||||
| 	DefaultAfterLoad = func(ctx context.Context, c Config) error { | ||||
| 		for _, fn := range c.Options().AfterLoad { | ||||
| 			if fn == nil { | ||||
| 				return nil | ||||
| 			} | ||||
| 			if err := fn(ctx, c); err != nil { | ||||
| 				c.Options().Logger.Error(ctx, c.String()+" AfterLoad error", err) | ||||
| 				if !c.Options().AllowFail { | ||||
| 					return err | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 		return nil | ||||
| 	} | ||||
| 	// DefaultBeforeSave default func that runs befora config Save | ||||
| 	DefaultBeforeSave = func(ctx context.Context, c Config) error { | ||||
| 		for _, fn := range c.Options().BeforeSave { | ||||
| 			if fn == nil { | ||||
| 				return nil | ||||
| 			} | ||||
| 			if err := fn(ctx, c); err != nil { | ||||
| 				c.Options().Logger.Errorf(ctx, "%s BeforeSave err: %v", c.String(), err) | ||||
| 				c.Options().Logger.Error(ctx, c.String()+" BeforeSave error", err) | ||||
| 				if !c.Options().AllowFail { | ||||
| 					return err | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 		return nil | ||||
| 	} | ||||
| 	// DefaultAfterSave default func that runs after config Save | ||||
| 	DefaultAfterSave = func(ctx context.Context, c Config) error { | ||||
| 		for _, fn := range c.Options().AfterSave { | ||||
| 			if fn == nil { | ||||
| 				return nil | ||||
| 			} | ||||
| 			if err := fn(ctx, c); err != nil { | ||||
| 				c.Options().Logger.Error(ctx, c.String()+" AfterSave error", err) | ||||
| 				if !c.Options().AllowFail { | ||||
| 					return err | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 		return nil | ||||
| 	} | ||||
| 	// DefaultBeforeInit default func that runs befora config Init | ||||
| 	DefaultBeforeInit = func(ctx context.Context, c Config) error { | ||||
| 		for _, fn := range c.Options().BeforeInit { | ||||
| 			if fn == nil { | ||||
| 				return nil | ||||
| 			} | ||||
| 			if err := fn(ctx, c); err != nil { | ||||
| 				c.Options().Logger.Error(ctx, c.String()+" BeforeInit error", err) | ||||
| 				if !c.Options().AllowFail { | ||||
| 					return err | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 		return nil | ||||
| 	} | ||||
| 	// DefaultAfterInit default func that runs after config Init | ||||
| 	DefaultAfterInit = func(ctx context.Context, c Config) error { | ||||
| 		for _, fn := range c.Options().AfterSave { | ||||
| 			if fn == nil { | ||||
| 				return nil | ||||
| 			} | ||||
| 			if err := fn(ctx, c); err != nil { | ||||
| 				c.Options().Logger.Error(ctx, c.String()+" AfterInit error", err) | ||||
| 				if !c.Options().AllowFail { | ||||
| 					return err | ||||
| 				} | ||||
|   | ||||
| @@ -15,6 +15,15 @@ func FromContext(ctx context.Context) (Config, bool) { | ||||
| 	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 | ||||
| func NewContext(ctx context.Context, c Config) context.Context { | ||||
| 	if ctx == nil { | ||||
|   | ||||
| @@ -7,12 +7,17 @@ import ( | ||||
| 	"strings" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/imdario/mergo" | ||||
| 	"dario.cat/mergo" | ||||
| 	"github.com/google/uuid" | ||||
| 	"go.unistack.org/micro/v3/options" | ||||
| 	mid "go.unistack.org/micro/v3/util/id" | ||||
| 	rutil "go.unistack.org/micro/v3/util/reflect" | ||||
| 	mtime "go.unistack.org/micro/v3/util/time" | ||||
| ) | ||||
|  | ||||
| type defaultConfig struct { | ||||
| 	funcLoad FuncLoad | ||||
| 	funcSave FuncSave | ||||
| 	opts     Options | ||||
| } | ||||
|  | ||||
| @@ -24,11 +29,42 @@ func (c *defaultConfig) Init(opts ...Option) error { | ||||
| 	for _, o := range opts { | ||||
| 		o(&c.opts) | ||||
| 	} | ||||
|  | ||||
| 	if err := DefaultBeforeInit(c.opts.Context, c); err != nil && !c.opts.AllowFail { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	c.funcLoad = c.fnLoad | ||||
| 	c.funcSave = c.fnSave | ||||
|  | ||||
| 	c.opts.Hooks.EachPrev(func(hook options.Hook) { | ||||
| 		switch h := hook.(type) { | ||||
| 		case HookLoad: | ||||
| 			c.funcLoad = h(c.funcLoad) | ||||
| 		case HookSave: | ||||
| 			c.funcSave = h(c.funcSave) | ||||
| 		} | ||||
| 	}) | ||||
|  | ||||
| 	if err := DefaultAfterInit(c.opts.Context, c); err != nil && !c.opts.AllowFail { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (c *defaultConfig) Load(ctx context.Context, opts ...LoadOption) error { | ||||
| 	if err := DefaultBeforeLoad(ctx, c); err != nil { | ||||
| 	return c.funcLoad(ctx, opts...) | ||||
| } | ||||
|  | ||||
| func (c *defaultConfig) fnLoad(ctx context.Context, opts ...LoadOption) error { | ||||
| 	var err error | ||||
|  | ||||
| 	if c.opts.SkipLoad != nil && c.opts.SkipLoad(ctx, c) { | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	if err = DefaultBeforeLoad(ctx, c); err != nil && !c.opts.AllowFail { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| @@ -51,21 +87,20 @@ func (c *defaultConfig) Load(ctx context.Context, opts ...LoadOption) error { | ||||
| 		if !c.opts.AllowFail { | ||||
| 			return err | ||||
| 		} | ||||
| 		return DefaultAfterLoad(ctx, c) | ||||
| 		if err = DefaultAfterLoad(ctx, c); err != nil && !c.opts.AllowFail { | ||||
| 			return err | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	if err = fillValues(reflect.ValueOf(src), c.opts.StructTag); err == nil { | ||||
| 		err = mergo.Merge(dst, src, mopts...) | ||||
| 	} | ||||
|  | ||||
| 	if err != nil { | ||||
| 		c.opts.Logger.Errorf(ctx, "default load error: %v", err) | ||||
| 		if !c.opts.AllowFail { | ||||
| 	if err != nil && !c.opts.AllowFail { | ||||
| 		return err | ||||
| 	} | ||||
| 	} | ||||
|  | ||||
| 	if err := DefaultAfterLoad(ctx, c); err != nil { | ||||
| 	if err := DefaultAfterLoad(ctx, c); err != nil && !c.opts.AllowFail { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| @@ -116,6 +151,20 @@ func fillValue(value reflect.Value, val string) error { | ||||
| 		} | ||||
| 		value.Set(reflect.ValueOf(v)) | ||||
| 	case reflect.String: | ||||
| 		switch val { | ||||
| 		case "micro:generate uuid": | ||||
| 			uid, err := uuid.NewRandom() | ||||
| 			if err != nil { | ||||
| 				return err | ||||
| 			} | ||||
| 			val = uid.String() | ||||
| 		case "micro:generate id": | ||||
| 			uid, err := mid.New() | ||||
| 			if err != nil { | ||||
| 				return err | ||||
| 			} | ||||
| 			val = uid | ||||
| 		} | ||||
| 		value.Set(reflect.ValueOf(val)) | ||||
| 	case reflect.Float32: | ||||
| 		v, err := strconv.ParseFloat(val, 32) | ||||
| @@ -205,6 +254,7 @@ func fillValue(value reflect.Value, val string) error { | ||||
| 		} | ||||
| 		value.Set(reflect.ValueOf(v)) | ||||
| 	} | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| @@ -268,6 +318,14 @@ func fillValues(valueOf reflect.Value, tname string) error { | ||||
| } | ||||
|  | ||||
| func (c *defaultConfig) Save(ctx context.Context, opts ...SaveOption) error { | ||||
| 	return c.funcSave(ctx, opts...) | ||||
| } | ||||
|  | ||||
| func (c *defaultConfig) fnSave(ctx context.Context, opts ...SaveOption) error { | ||||
| 	if c.opts.SkipSave != nil && c.opts.SkipSave(ctx, c) { | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	if err := DefaultBeforeSave(ctx, c); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| @@ -287,7 +345,7 @@ func (c *defaultConfig) Name() string { | ||||
| 	return c.opts.Name | ||||
| } | ||||
|  | ||||
| func (c *defaultConfig) Watch(ctx context.Context, opts ...WatchOption) (Watcher, error) { | ||||
| func (c *defaultConfig) Watch(_ context.Context, _ ...WatchOption) (Watcher, error) { | ||||
| 	return nil, ErrWatcherNotImplemented | ||||
| } | ||||
|  | ||||
| @@ -297,5 +355,9 @@ func NewConfig(opts ...Option) Config { | ||||
| 	if len(options.StructTag) == 0 { | ||||
| 		options.StructTag = "default" | ||||
| 	} | ||||
| 	return &defaultConfig{opts: options} | ||||
| 	c := &defaultConfig{opts: options} | ||||
| 	c.funcLoad = c.fnLoad | ||||
| 	c.funcSave = c.fnSave | ||||
|  | ||||
| 	return c | ||||
| } | ||||
|   | ||||
| @@ -3,6 +3,7 @@ package config_test | ||||
| import ( | ||||
| 	"context" | ||||
| 	"fmt" | ||||
| 	"reflect" | ||||
| 	"testing" | ||||
| 	"time" | ||||
|  | ||||
| @@ -11,12 +12,17 @@ import ( | ||||
| ) | ||||
|  | ||||
| type cfg struct { | ||||
| 	MapValue    map[string]bool `default:"key1=true,key2=false"` | ||||
| 	StructValue *cfgStructValue | ||||
|  | ||||
| 	StringValue string `default:"string_value"` | ||||
| 	IgnoreValue string `json:"-"` | ||||
| 	StructValue    *cfgStructValue | ||||
| 	IntValue       int            `default:"99"` | ||||
| 	UUIDValue   string `default:"micro:generate uuid"` | ||||
| 	IDValue     string `default:"micro:generate id"` | ||||
|  | ||||
| 	DurationValue  time.Duration  `default:"10s"` | ||||
| 	MDurationValue mtime.Duration `default:"10s"` | ||||
| 	IntValue       int            `default:"99"` | ||||
| } | ||||
|  | ||||
| type cfgStructValue struct { | ||||
| @@ -37,6 +43,35 @@ func (c *cfgStructValue) Validate() error { | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| type testHook struct { | ||||
| 	f bool | ||||
| } | ||||
|  | ||||
| func (t *testHook) Load(fn config.FuncLoad) config.FuncLoad { | ||||
| 	return func(ctx context.Context, opts ...config.LoadOption) error { | ||||
| 		t.f = true | ||||
| 		return fn(ctx, opts...) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestHook(t *testing.T) { | ||||
| 	h := &testHook{} | ||||
|  | ||||
| 	c := config.NewConfig(config.Struct(h), config.Hooks(config.HookLoad(h.Load))) | ||||
|  | ||||
| 	if err := c.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	if err := c.Load(context.TODO()); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	if !h.f { | ||||
| 		t.Fatal("hook not works") | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestDefault(t *testing.T) { | ||||
| 	ctx := context.Background() | ||||
| 	conf := &cfg{IntValue: 10} | ||||
| @@ -67,6 +102,19 @@ func TestDefault(t *testing.T) { | ||||
| 	if conf.StringValue != "after_load" { | ||||
| 		t.Fatal("AfterLoad option not working") | ||||
| 	} | ||||
| 	if len(conf.MapValue) != 2 { | ||||
| 		t.Fatalf("map value invalid: %#+v\n", conf.MapValue) | ||||
| 	} | ||||
|  | ||||
| 	if conf.UUIDValue == "" { | ||||
| 		t.Fatalf("uuid value empty") | ||||
| 	} else if len(conf.UUIDValue) != 36 { | ||||
| 		t.Fatalf("uuid value invalid: %s", conf.UUIDValue) | ||||
| 	} | ||||
|  | ||||
| 	if conf.IDValue == "" { | ||||
| 		t.Fatalf("id value empty") | ||||
| 	} | ||||
| 	_ = conf | ||||
| 	// t.Logf("%#+v\n", conf) | ||||
| } | ||||
| @@ -86,3 +134,13 @@ func TestValidate(t *testing.T) { | ||||
| 		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()) | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -7,6 +7,7 @@ import ( | ||||
| 	"go.unistack.org/micro/v3/codec" | ||||
| 	"go.unistack.org/micro/v3/logger" | ||||
| 	"go.unistack.org/micro/v3/meter" | ||||
| 	"go.unistack.org/micro/v3/options" | ||||
| 	"go.unistack.org/micro/v3/tracer" | ||||
| ) | ||||
|  | ||||
| @@ -28,14 +29,26 @@ type Options struct { | ||||
| 	Name string | ||||
| 	// StructTag name | ||||
| 	StructTag string | ||||
| 	// BeforeSave contains slice of funcs that runs before save | ||||
| 	// BeforeSave contains slice of funcs that runs before Save | ||||
| 	BeforeSave []func(context.Context, Config) error | ||||
| 	// AfterLoad contains slice of funcs that runs after load | ||||
| 	AfterLoad []func(context.Context, Config) error | ||||
| 	// BeforeLoad contains slice of funcs that runs before load | ||||
| 	BeforeLoad []func(context.Context, Config) error | ||||
| 	// AfterSave contains slice of funcs that runs after save | ||||
| 	// AfterSave contains slice of funcs that runs after Save | ||||
| 	AfterSave []func(context.Context, Config) error | ||||
| 	// BeforeLoad contains slice of funcs that runs before Load | ||||
| 	BeforeLoad []func(context.Context, Config) error | ||||
| 	// AfterLoad contains slice of funcs that runs after Load | ||||
| 	AfterLoad []func(context.Context, Config) error | ||||
| 	// BeforeInit contains slice of funcs that runs before Init | ||||
| 	BeforeInit []func(context.Context, Config) error | ||||
| 	// AfterInit contains slice of funcs that runs after Init | ||||
| 	AfterInit []func(context.Context, Config) error | ||||
|  | ||||
| 	// SkipLoad runs only if condition returns true | ||||
| 	SkipLoad func(context.Context, Config) bool | ||||
| 	// SkipSave runs only if condition returns true | ||||
| 	SkipSave func(context.Context, Config) bool | ||||
| 	// Hooks can be run before/after config Save/Load | ||||
| 	Hooks options.Hooks | ||||
|  | ||||
| 	// AllowFail flag to allow fail in config source | ||||
| 	AllowFail bool | ||||
| } | ||||
| @@ -64,9 +77,9 @@ type LoadOption func(o *LoadOptions) | ||||
| // LoadOptions struct | ||||
| type LoadOptions struct { | ||||
| 	Struct   interface{} | ||||
| 	Context  context.Context | ||||
| 	Override bool | ||||
| 	Append   bool | ||||
| 	Context  context.Context | ||||
| } | ||||
|  | ||||
| // NewLoadOptions create LoadOptions struct with provided opts | ||||
| @@ -131,6 +144,20 @@ func AllowFail(b bool) Option { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // BeforeInit run funcs before config Init | ||||
| func BeforeInit(fn ...func(context.Context, Config) error) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.BeforeInit = fn | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // AfterInit run funcs after config Init | ||||
| func AfterInit(fn ...func(context.Context, Config) error) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.AfterInit = fn | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // BeforeLoad run funcs before config load | ||||
| func BeforeLoad(fn ...func(context.Context, Config) error) Option { | ||||
| 	return func(o *Options) { | ||||
| @@ -253,10 +280,10 @@ func WatchCoalesce(b bool) WatchOption { | ||||
| } | ||||
|  | ||||
| // 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) { | ||||
| 		o.MinInterval = min | ||||
| 		o.MaxInterval = max | ||||
| 		o.MinInterval = minTime | ||||
| 		o.MaxInterval = maxTime | ||||
| 	} | ||||
| } | ||||
|  | ||||
| @@ -266,3 +293,10 @@ func WatchStruct(src interface{}) WatchOption { | ||||
| 		o.Struct = src | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Hooks sets hook runs before action | ||||
| func Hooks(h ...options.Hook) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.Hooks = append(o.Hooks, h...) | ||||
| 	} | ||||
| } | ||||
|   | ||||
							
								
								
									
										157
									
								
								database/dsn.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										157
									
								
								database/dsn.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,157 @@ | ||||
| package database | ||||
|  | ||||
| import ( | ||||
| 	"crypto/tls" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"net/url" | ||||
| 	"strings" | ||||
| ) | ||||
|  | ||||
| var ( | ||||
| 	ErrInvalidDSNAddr      = errors.New("invalid dsn addr") | ||||
| 	ErrInvalidDSNUnescaped = errors.New("dsn must be escaped") | ||||
| 	ErrInvalidDSNNoSlash   = errors.New("dsn must contains slash") | ||||
| ) | ||||
|  | ||||
| type Config struct { | ||||
| 	TLSConfig *tls.Config | ||||
| 	Username  string | ||||
| 	Password  string | ||||
| 	Scheme    string | ||||
| 	Host      string | ||||
| 	Port      string | ||||
| 	Database  string | ||||
| 	Params    []string | ||||
| } | ||||
|  | ||||
| func (cfg *Config) FormatDSN() string { | ||||
| 	var s strings.Builder | ||||
|  | ||||
| 	if len(cfg.Scheme) > 0 { | ||||
| 		s.WriteString(cfg.Scheme + "://") | ||||
| 	} | ||||
| 	// [username[:password]@] | ||||
| 	if len(cfg.Username) > 0 { | ||||
| 		s.WriteString(cfg.Username) | ||||
| 		if len(cfg.Password) > 0 { | ||||
| 			s.WriteByte(':') | ||||
| 			s.WriteString(url.PathEscape(cfg.Password)) | ||||
| 		} | ||||
| 		s.WriteByte('@') | ||||
| 	} | ||||
|  | ||||
| 	// [host:port] | ||||
| 	if len(cfg.Host) > 0 { | ||||
| 		s.WriteString(cfg.Host) | ||||
| 		if len(cfg.Port) > 0 { | ||||
| 			s.WriteByte(':') | ||||
| 			s.WriteString(cfg.Port) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// /dbname | ||||
| 	s.WriteByte('/') | ||||
| 	s.WriteString(url.PathEscape(cfg.Database)) | ||||
|  | ||||
| 	for i := 0; i < len(cfg.Params); i += 2 { | ||||
| 		if i == 0 { | ||||
| 			s.WriteString("?") | ||||
| 		} else { | ||||
| 			s.WriteString("&") | ||||
| 		} | ||||
| 		s.WriteString(cfg.Params[i]) | ||||
| 		s.WriteString("=") | ||||
| 		s.WriteString(cfg.Params[i+1]) | ||||
| 	} | ||||
|  | ||||
| 	return s.String() | ||||
| } | ||||
|  | ||||
| func ParseDSN(dsn string) (*Config, error) { | ||||
| 	cfg := &Config{} | ||||
|  | ||||
| 	// [user[:password]@][net[(addr)]]/dbname[?param1=value1¶mN=valueN] | ||||
| 	// Find last '/' that goes before dbname | ||||
| 	foundSlash := false | ||||
| 	for i := len(dsn) - 1; i >= 0; i-- { | ||||
| 		if dsn[i] == '/' { | ||||
| 			foundSlash = true | ||||
| 			var j, k int | ||||
|  | ||||
| 			// left part is empty if i <= 0 | ||||
| 			if i > 0 { | ||||
| 				// Find the first ':' in dsn | ||||
| 				for j = i; j >= 0; j-- { | ||||
| 					if dsn[j] == ':' { | ||||
| 						cfg.Scheme = dsn[0:j] | ||||
| 					} | ||||
| 				} | ||||
|  | ||||
| 				// [username[:password]@][host] | ||||
| 				// Find the last '@' in dsn[:i] | ||||
| 				for j = i; j >= 0; j-- { | ||||
| 					if dsn[j] == '@' { | ||||
| 						// username[:password] | ||||
| 						// Find the second ':' in dsn[:j] | ||||
| 						for k = 0; k < j; k++ { | ||||
| 							if dsn[k] == ':' { | ||||
| 								if cfg.Scheme == dsn[:k] { | ||||
| 									continue | ||||
| 								} | ||||
| 								var err error | ||||
| 								cfg.Password, err = url.PathUnescape(dsn[k+1 : j]) | ||||
| 								if err != nil { | ||||
| 									return nil, err | ||||
| 								} | ||||
| 								break | ||||
| 							} | ||||
| 						} | ||||
| 						cfg.Username = dsn[len(cfg.Scheme)+3 : k] | ||||
| 						break | ||||
| 					} | ||||
| 				} | ||||
|  | ||||
| 				for k = j + 1; k < i; k++ { | ||||
| 					if dsn[k] == ':' { | ||||
| 						cfg.Host = dsn[j+1 : k] | ||||
| 						cfg.Port = dsn[k+1 : i] | ||||
| 						break | ||||
| 					} | ||||
| 				} | ||||
|  | ||||
| 			} | ||||
|  | ||||
| 			// dbname[?param1=value1&...¶mN=valueN] | ||||
| 			// Find the first '?' in dsn[i+1:] | ||||
| 			for j = i + 1; j < len(dsn); j++ { | ||||
| 				if dsn[j] == '?' { | ||||
| 					parts := strings.Split(dsn[j+1:], "&") | ||||
| 					cfg.Params = make([]string, 0, len(parts)*2) | ||||
| 					for _, p := range parts { | ||||
| 						k, v, found := strings.Cut(p, "=") | ||||
| 						if !found { | ||||
| 							continue | ||||
| 						} | ||||
| 						cfg.Params = append(cfg.Params, k, v) | ||||
| 					} | ||||
|  | ||||
| 					break | ||||
| 				} | ||||
| 			} | ||||
| 			var err error | ||||
| 			dbname := dsn[i+1 : j] | ||||
| 			if cfg.Database, err = url.PathUnescape(dbname); err != nil { | ||||
| 				return nil, fmt.Errorf("invalid dbname %q: %w", dbname, err) | ||||
| 			} | ||||
|  | ||||
| 			break | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	if !foundSlash && len(dsn) > 0 { | ||||
| 		return nil, ErrInvalidDSNNoSlash | ||||
| 	} | ||||
|  | ||||
| 	return cfg, nil | ||||
| } | ||||
							
								
								
									
										31
									
								
								database/dsn_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										31
									
								
								database/dsn_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,31 @@ | ||||
| package database | ||||
|  | ||||
| import ( | ||||
| 	"net/url" | ||||
| 	"testing" | ||||
| ) | ||||
|  | ||||
| func TestParseDSN(t *testing.T) { | ||||
| 	cfg, err := ParseDSN("postgres://username:p@ssword#@host:12345/dbname?key1=val2&key2=val2") | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	if cfg.Password != "p@ssword#" { | ||||
| 		t.Fatalf("parsing error") | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestFormatDSN(t *testing.T) { | ||||
| 	src := "postgres://username:p@ssword#@host:12345/dbname?key1=val2&key2=val2" | ||||
| 	cfg, err := ParseDSN(src) | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	dst, err := url.PathUnescape(cfg.FormatDSN()) | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	if src != dst { | ||||
| 		t.Fatalf("\n%s\n%s", src, dst) | ||||
| 	} | ||||
| } | ||||
							
								
								
									
										158
									
								
								errors/errors.go
									
									
									
									
									
								
							
							
						
						
									
										158
									
								
								errors/errors.go
									
									
									
									
									
								
							| @@ -1,14 +1,20 @@ | ||||
| // Package errors provides a way to return detailed information | ||||
| // for an RPC request error. The error is normally JSON encoded. | ||||
| package errors // import "go.unistack.org/micro/v3/errors" | ||||
| package errors | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"context" | ||||
| 	"encoding/json" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"io" | ||||
| 	"net/http" | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
|  | ||||
| 	"google.golang.org/grpc/codes" | ||||
| 	"google.golang.org/grpc/status" | ||||
| ) | ||||
|  | ||||
| var ( | ||||
| @@ -38,6 +44,20 @@ var ( | ||||
| 	ErrGatewayTimeout = &Error{Code: 504} | ||||
| ) | ||||
|  | ||||
| const ProblemContentType = "application/problem+json" | ||||
|  | ||||
| type Problem struct { | ||||
| 	Type     string `json:"type,omitempty"` | ||||
| 	Title    string `json:"title,omitempty"` | ||||
| 	Detail   string `json:"detail,omitempty"` | ||||
| 	Instance string `json:"instance,omitempty"` | ||||
| 	Errors   []struct { | ||||
| 		Title  string `json:"title,omitempty"` | ||||
| 		Detail string `json:"detail,omitempty"` | ||||
| 	} `json:"errors,omitempty"` | ||||
| 	Status int `json:"status,omitempty"` | ||||
| } | ||||
|  | ||||
| // Error type | ||||
| type Error struct { | ||||
| 	// ID holds error id or service, usually someting like my_service or id | ||||
| @@ -256,6 +276,10 @@ func CodeIn(err interface{}, codes ...int32) bool { | ||||
|  | ||||
| // FromError try to convert go error to *Error | ||||
| func FromError(err error) *Error { | ||||
| 	if err == nil { | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	if verr, ok := err.(*Error); ok && verr != nil { | ||||
| 		return verr | ||||
| 	} | ||||
| @@ -340,3 +364,135 @@ func addslashes(str string) string { | ||||
| 	} | ||||
| 	return buf.String() | ||||
| } | ||||
|  | ||||
| type retryableError struct { | ||||
| 	err error | ||||
| } | ||||
|  | ||||
| // Retryable returns error that can be retried later | ||||
| func Retryable(err error) error { | ||||
| 	return &retryableError{err: err} | ||||
| } | ||||
|  | ||||
| type IsRetryableFunc func(error) bool | ||||
|  | ||||
| var ( | ||||
| 	RetrayableOracleErrors = []IsRetryableFunc{ | ||||
| 		func(err error) bool { | ||||
| 			errmsg := err.Error() | ||||
| 			switch { | ||||
| 			case strings.Contains(errmsg, `ORA-`): | ||||
| 				return true | ||||
| 			case strings.Contains(errmsg, `can not assign`): | ||||
| 				return true | ||||
| 			case strings.Contains(errmsg, `can't assign`): | ||||
| 				return true | ||||
| 			} | ||||
| 			return false | ||||
| 		}, | ||||
| 	} | ||||
| 	RetrayablePostgresErrors = []IsRetryableFunc{ | ||||
| 		func(err error) bool { | ||||
| 			errmsg := err.Error() | ||||
| 			switch { | ||||
| 			case strings.Contains(errmsg, `number of field descriptions must equal number of`): | ||||
| 				return true | ||||
| 			case strings.Contains(errmsg, `not a pointer`): | ||||
| 				return true | ||||
| 			case strings.Contains(errmsg, `values, but dst struct has only`): | ||||
| 				return true | ||||
| 			case strings.Contains(errmsg, `struct doesn't have corresponding row field`): | ||||
| 				return true | ||||
| 			case strings.Contains(errmsg, `cannot find field`): | ||||
| 				return true | ||||
| 			case strings.Contains(errmsg, `cannot scan`) || strings.Contains(errmsg, `cannot convert`): | ||||
| 				return true | ||||
| 			case strings.Contains(errmsg, `failed to connect to`): | ||||
| 				return true | ||||
| 			} | ||||
| 			return false | ||||
| 		}, | ||||
| 	} | ||||
| 	RetryableMicroErrors = []IsRetryableFunc{ | ||||
| 		func(err error) bool { | ||||
| 			switch verr := err.(type) { | ||||
| 			case *Error: | ||||
| 				switch verr.Code { | ||||
| 				case 401, 403, 408, 500, 501, 502, 503, 504: | ||||
| 					return true | ||||
| 				default: | ||||
| 					return false | ||||
| 				} | ||||
| 			case *retryableError: | ||||
| 				return true | ||||
| 			} | ||||
| 			return false | ||||
| 		}, | ||||
| 	} | ||||
| 	RetryableGoErrors = []IsRetryableFunc{ | ||||
| 		func(err error) bool { | ||||
| 			switch verr := err.(type) { | ||||
| 			case interface{ SafeToRetry() bool }: | ||||
| 				return verr.SafeToRetry() | ||||
| 			case interface{ Timeout() bool }: | ||||
| 				return verr.Timeout() | ||||
| 			} | ||||
| 			switch { | ||||
| 			case errors.Is(err, io.EOF), errors.Is(err, io.ErrUnexpectedEOF): | ||||
| 				return true | ||||
| 			case errors.Is(err, context.DeadlineExceeded): | ||||
| 				return true | ||||
| 			case errors.Is(err, io.ErrClosedPipe), errors.Is(err, io.ErrShortBuffer), errors.Is(err, io.ErrShortWrite): | ||||
| 				return true | ||||
| 			} | ||||
| 			return false | ||||
| 		}, | ||||
| 	} | ||||
| 	RetryableGrpcErrors = []IsRetryableFunc{ | ||||
| 		func(err error) bool { | ||||
| 			st, ok := status.FromError(err) | ||||
| 			if !ok { | ||||
| 				return false | ||||
| 			} | ||||
| 			switch st.Code() { | ||||
| 			case codes.Unavailable, codes.ResourceExhausted: | ||||
| 				return true | ||||
| 			case codes.DeadlineExceeded: | ||||
| 				return true | ||||
| 			case codes.Internal: | ||||
| 				switch { | ||||
| 				case strings.Contains(st.Message(), `transport: received the unexpected content-type "text/html; charset=UTF-8"`): | ||||
| 					return true | ||||
| 				case strings.Contains(st.Message(), io.ErrUnexpectedEOF.Error()): | ||||
| 					return true | ||||
| 				case strings.Contains(st.Message(), `stream terminated by RST_STREAM with error code: INTERNAL_ERROR`): | ||||
| 					return true | ||||
| 				} | ||||
| 			} | ||||
| 			return false | ||||
| 		}, | ||||
| 	} | ||||
| ) | ||||
|  | ||||
| // Unwrap provides error wrapping | ||||
| func (e *retryableError) Unwrap() error { | ||||
| 	return e.err | ||||
| } | ||||
|  | ||||
| // Error returns the error string | ||||
| func (e *retryableError) Error() string { | ||||
| 	if e.err == nil { | ||||
| 		return "" | ||||
| 	} | ||||
| 	return e.err.Error() | ||||
| } | ||||
|  | ||||
| // IsRetryable checks error for ability to retry later | ||||
| func IsRetryable(err error, fns ...IsRetryableFunc) bool { | ||||
| 	for _, fn := range fns { | ||||
| 		if ok := fn(err); ok { | ||||
| 			return true | ||||
| 		} | ||||
| 	} | ||||
| 	return false | ||||
| } | ||||
|   | ||||
| @@ -2,12 +2,19 @@ package errors | ||||
|  | ||||
| import ( | ||||
| 	"encoding/json" | ||||
| 	er "errors" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"net/http" | ||||
| 	"testing" | ||||
| ) | ||||
|  | ||||
| func TestIsRetrayable(t *testing.T) { | ||||
| 	err := fmt.Errorf("ORA-") | ||||
| 	if !IsRetryable(err, RetrayableOracleErrors...) { | ||||
| 		t.Fatalf("IsRetrayable not works") | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestMarshalJSON(t *testing.T) { | ||||
| 	e := InternalServerError("id", "err: %v", fmt.Errorf("err: %v", `xxx: "UNIX_TIMESTAMP": invalid identifier`)) | ||||
| 	_, err := json.Marshal(e) | ||||
| @@ -19,7 +26,7 @@ func TestMarshalJSON(t *testing.T) { | ||||
| func TestEmpty(t *testing.T) { | ||||
| 	msg := "test" | ||||
| 	var err *Error | ||||
| 	err = FromError(fmt.Errorf(msg)) | ||||
| 	err = FromError(errors.New(msg)) | ||||
| 	if err.Detail != msg { | ||||
| 		t.Fatalf("invalid error %v", err) | ||||
| 	} | ||||
| @@ -35,7 +42,7 @@ func TestFromError(t *testing.T) { | ||||
| 	if merr.ID != "go.micro.test" || merr.Code != 404 { | ||||
| 		t.Fatalf("invalid conversation %v != %v", err, merr) | ||||
| 	} | ||||
| 	err = er.New(err.Error()) | ||||
| 	err = errors.New(err.Error()) | ||||
| 	merr = FromError(err) | ||||
| 	if merr.ID != "go.micro.test" || merr.Code != 404 { | ||||
| 		t.Fatalf("invalid conversation %v != %v", err, merr) | ||||
| @@ -50,7 +57,7 @@ func TestEqual(t *testing.T) { | ||||
| 		t.Fatal("errors must be equal") | ||||
| 	} | ||||
|  | ||||
| 	err3 := er.New("my test err") | ||||
| 	err3 := errors.New("my test err") | ||||
| 	if Equal(err1, err3) { | ||||
| 		t.Fatal("errors must be not equal") | ||||
| 	} | ||||
|   | ||||
| @@ -15,6 +15,15 @@ func FromContext(ctx context.Context) (Flow, bool) { | ||||
| 	return c, ok | ||||
| } | ||||
|  | ||||
| // MustContext returns Flow from context | ||||
| func MustContext(ctx context.Context) Flow { | ||||
| 	f, ok := FromContext(ctx) | ||||
| 	if !ok { | ||||
| 		panic("missing flow") | ||||
| 	} | ||||
| 	return f | ||||
| } | ||||
|  | ||||
| // NewContext stores Flow to context | ||||
| func NewContext(ctx context.Context, f Flow) context.Context { | ||||
| 	if ctx == nil { | ||||
|   | ||||
| @@ -8,6 +8,7 @@ import ( | ||||
| ) | ||||
|  | ||||
| func TestDeps(t *testing.T) { | ||||
| 	t.Skip() | ||||
| 	d := &dag.AcyclicGraph{} | ||||
|  | ||||
| 	v0 := d.Add(&node{"v0"}) | ||||
|   | ||||
| @@ -188,7 +188,7 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...Execu | ||||
| 	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.Errorf(w.opts.Context, "store error: %v", werr) | ||||
| 			w.opts.Logger.Error(w.opts.Context, "store write error", werr) | ||||
| 		} | ||||
| 		return "", err | ||||
| 	} | ||||
| @@ -212,7 +212,7 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...Execu | ||||
| 	done := make(chan struct{}) | ||||
|  | ||||
| 	if werr := workflowStore.Write(w.opts.Context, "status", &codec.Frame{Data: []byte(StatusRunning.String())}); werr != nil { | ||||
| 		w.opts.Logger.Errorf(w.opts.Context, "store error: %v", werr) | ||||
| 		w.opts.Logger.Error(w.opts.Context, "store write error", werr) | ||||
| 		return eid, werr | ||||
| 	} | ||||
| 	for idx := range steps { | ||||
| @@ -237,7 +237,7 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...Execu | ||||
| 					return | ||||
| 				} | ||||
| 				if w.opts.Logger.V(logger.TraceLevel) { | ||||
| 					w.opts.Logger.Tracef(nctx, "will be executed %v", steps[idx][nidx]) | ||||
| 					w.opts.Logger.Trace(nctx, fmt.Sprintf("will be executed %v", steps[idx][nidx])) | ||||
| 				} | ||||
| 				cstep := steps[idx][nidx] | ||||
| 				// nolint: nestif | ||||
| @@ -257,21 +257,21 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...Execu | ||||
| 						if serr != nil { | ||||
| 							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) { | ||||
| 								w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | ||||
| 								w.opts.Logger.Error(ctx, "store write error", werr) | ||||
| 							} | ||||
| 							if werr := stepStore.Write(ctx, step.ID()+w.opts.Store.Options().Separator+"status", &codec.Frame{Data: []byte(StatusFailure.String())}); werr != nil && w.opts.Logger.V(logger.ErrorLevel) { | ||||
| 								w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | ||||
| 								w.opts.Logger.Error(ctx, "store write error", werr) | ||||
| 							} | ||||
| 							cherr <- serr | ||||
| 							return | ||||
| 						} | ||||
| 						if werr := stepStore.Write(ctx, step.ID()+w.opts.Store.Options().Separator+"rsp", rsp); werr != nil { | ||||
| 							w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | ||||
| 							w.opts.Logger.Error(ctx, "store write error", werr) | ||||
| 							cherr <- werr | ||||
| 							return | ||||
| 						} | ||||
| 						if werr := stepStore.Write(ctx, step.ID()+w.opts.Store.Options().Separator+"status", &codec.Frame{Data: []byte(StatusSuccess.String())}); werr != nil { | ||||
| 							w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | ||||
| 							w.opts.Logger.Error(ctx, "store write error", werr) | ||||
| 							cherr <- werr | ||||
| 							return | ||||
| 						} | ||||
| @@ -290,16 +290,16 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...Execu | ||||
| 					if serr != nil { | ||||
| 						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) { | ||||
| 							w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | ||||
| 							w.opts.Logger.Error(ctx, "store write error", werr) | ||||
| 						} | ||||
| 						if werr := stepStore.Write(ctx, cstep.ID()+w.opts.Store.Options().Separator+"status", &codec.Frame{Data: []byte(StatusFailure.String())}); werr != nil && w.opts.Logger.V(logger.ErrorLevel) { | ||||
| 							w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | ||||
| 							w.opts.Logger.Error(ctx, "store write error", werr) | ||||
| 						} | ||||
| 						cherr <- serr | ||||
| 						return | ||||
| 					} | ||||
| 					if werr := stepStore.Write(ctx, cstep.ID()+w.opts.Store.Options().Separator+"rsp", rsp); werr != nil { | ||||
| 						w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | ||||
| 						w.opts.Logger.Error(ctx, "store write error", werr) | ||||
| 						cherr <- werr | ||||
| 						return | ||||
| 					} | ||||
| @@ -317,7 +317,7 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...Execu | ||||
| 		return eid, nil | ||||
| 	} | ||||
|  | ||||
| 	logger.Tracef(ctx, "wait for finish or error") | ||||
| 	logger.DefaultLogger.Trace(ctx, "wait for finish or error") | ||||
| 	select { | ||||
| 	case <-nctx.Done(): | ||||
| 		err = nctx.Err() | ||||
| @@ -333,15 +333,15 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...Execu | ||||
| 	switch { | ||||
| 	case nctx.Err() != nil: | ||||
| 		if werr := workflowStore.Write(w.opts.Context, "status", &codec.Frame{Data: []byte(StatusAborted.String())}); werr != nil { | ||||
| 			w.opts.Logger.Errorf(w.opts.Context, "store error: %v", werr) | ||||
| 			w.opts.Logger.Error(w.opts.Context, "store write error", werr) | ||||
| 		} | ||||
| 	case err == nil: | ||||
| 		if werr := workflowStore.Write(w.opts.Context, "status", &codec.Frame{Data: []byte(StatusSuccess.String())}); werr != nil { | ||||
| 			w.opts.Logger.Errorf(w.opts.Context, "store error: %v", werr) | ||||
| 			w.opts.Logger.Error(w.opts.Context, "store write error", werr) | ||||
| 		} | ||||
| 	case err != nil: | ||||
| 		if werr := workflowStore.Write(w.opts.Context, "status", &codec.Frame{Data: []byte(StatusFailure.String())}); werr != nil { | ||||
| 			w.opts.Logger.Errorf(w.opts.Context, "store error: %v", werr) | ||||
| 			w.opts.Logger.Error(w.opts.Context, "store write error", werr) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
|   | ||||
| @@ -1,5 +1,5 @@ | ||||
| // Package flow is an interface used for saga pattern microservice workflow | ||||
| package flow // import "go.unistack.org/micro/v3/flow" | ||||
| package flow | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
|   | ||||
| @@ -32,7 +32,7 @@ type fsm struct { | ||||
|  | ||||
| // NewFSM creates a new finite state machine having the specified initial state | ||||
| // with specified options | ||||
| func NewFSM(opts ...Option) *fsm { | ||||
| func NewFSM(opts ...Option) FSM { | ||||
| 	return &fsm{ | ||||
| 		statesMap: map[string]StateFunc{}, | ||||
| 		opts:      NewOptions(opts...), | ||||
|   | ||||
| @@ -1,4 +1,4 @@ | ||||
| package fsm // import "go.unistack.org/micro/v3/fsm" | ||||
| package fsm | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
|   | ||||
| @@ -17,7 +17,7 @@ func TestFSMStart(t *testing.T) { | ||||
|  | ||||
| 	wrapper := func(next StateFunc) StateFunc { | ||||
| 		return func(sctx context.Context, s State, opts ...StateOption) (State, error) { | ||||
| 			sctx = logger.NewContext(sctx, logger.Fields("state", s.Name())) | ||||
| 			sctx = logger.NewContext(sctx, logger.DefaultLogger.Fields("state", s.Name())) | ||||
| 			return next(sctx, s, opts...) | ||||
| 		} | ||||
| 	} | ||||
|   | ||||
							
								
								
									
										44
									
								
								go.mod
									
									
									
									
									
								
							
							
						
						
									
										44
									
								
								go.mod
									
									
									
									
									
								
							| @@ -1,13 +1,43 @@ | ||||
| module go.unistack.org/micro/v3 | ||||
|  | ||||
| go 1.16 | ||||
| go 1.23.4 | ||||
|  | ||||
| require ( | ||||
| 	github.com/google/go-cmp v0.5.7 // indirect | ||||
| 	github.com/imdario/mergo v0.3.13 | ||||
| 	github.com/kr/pretty v0.2.1 // indirect | ||||
| 	github.com/kr/text v0.2.0 // indirect | ||||
| 	dario.cat/mergo v1.0.1 | ||||
| 	github.com/DATA-DOG/go-sqlmock v1.5.0 | ||||
| 	github.com/KimMachineGun/automemlimit v0.6.1 | ||||
| 	github.com/ash3in/uuidv8 v1.0.1 | ||||
| 	github.com/google/uuid v1.6.0 | ||||
| 	github.com/matoous/go-nanoid v1.5.1 | ||||
| 	github.com/patrickmn/go-cache v2.1.0+incompatible | ||||
| 	github.com/silas/dag v0.0.0-20211117232152-9d50aa809f35 | ||||
| 	go.unistack.org/micro-proto/v3 v3.3.1 | ||||
| 	github.com/silas/dag v0.0.0-20220518035006-a7e85ada93c5 | ||||
| 	go.uber.org/automaxprocs v1.6.0 | ||||
| 	go.unistack.org/micro-proto/v3 v3.4.1 | ||||
| 	golang.org/x/sync v0.10.0 | ||||
| 	google.golang.org/grpc v1.68.1 | ||||
| 	google.golang.org/protobuf v1.35.2 | ||||
| 	gopkg.in/yaml.v3 v3.0.1 | ||||
| ) | ||||
|  | ||||
| require ( | ||||
| 	github.com/cilium/ebpf v0.16.0 // indirect | ||||
| 	github.com/containerd/cgroups/v3 v3.0.4 // indirect | ||||
| 	github.com/containerd/log v0.1.0 // indirect | ||||
| 	github.com/coreos/go-systemd/v22 v22.5.0 // indirect | ||||
| 	github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect | ||||
| 	github.com/docker/go-units v0.5.0 // indirect | ||||
| 	github.com/godbus/dbus/v5 v5.1.0 // indirect | ||||
| 	github.com/moby/sys/userns v0.1.0 // indirect | ||||
| 	github.com/opencontainers/runtime-spec v1.2.0 // indirect | ||||
| 	github.com/pbnjay/memory v0.0.0-20210728143218-7b4eea64cf58 // indirect | ||||
| 	github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 // indirect | ||||
| 	github.com/rogpeppe/go-internal v1.13.1 // indirect | ||||
| 	github.com/sirupsen/logrus v1.9.3 // indirect | ||||
| 	github.com/stretchr/testify v1.10.0 // indirect | ||||
| 	go.uber.org/goleak v1.3.0 // indirect | ||||
| 	golang.org/x/exp v0.0.0-20241210194714-1829a127f884 // indirect | ||||
| 	golang.org/x/net v0.32.0 // indirect | ||||
| 	golang.org/x/sys v0.28.0 // indirect | ||||
| 	google.golang.org/genproto/googleapis/rpc v0.0.0-20241209162323-e6fa225c2576 // indirect | ||||
| 	gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c // indirect | ||||
| ) | ||||
|   | ||||
							
								
								
									
										238
									
								
								go.sum
									
									
									
									
									
								
							
							
						
						
									
										238
									
								
								go.sum
									
									
									
									
									
								
							| @@ -1,160 +1,102 @@ | ||||
| cloud.google.com/go v0.26.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw= | ||||
| cloud.google.com/go v0.34.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw= | ||||
| github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU= | ||||
| github.com/OneOfOne/xxhash v1.2.2/go.mod h1:HSdplMjZKSmBqAxg5vPj2TmRDmfkzw+cTzAElWljhcU= | ||||
| github.com/antihax/optional v1.0.0/go.mod h1:uupD/76wgC+ih3iEmQUL+0Ugr19nfwCT1kdvxnR2qWY= | ||||
| github.com/buger/jsonparser v1.1.1/go.mod h1:6RYKKt7H4d4+iWqouImQ9R2FZql3VbhNgx27UK13J/0= | ||||
| github.com/census-instrumentation/opencensus-proto v0.2.1/go.mod h1:f6KPmirojxKA12rnyqOA5BBL4O983OfeGPqjHWSTneU= | ||||
| github.com/cespare/xxhash v1.1.0/go.mod h1:XrSqR1VqqWfGrhpAt58auRo0WTKS1nRRg3ghfAqPWnc= | ||||
| github.com/client9/misspell v0.3.4/go.mod h1:qj6jICC3Q7zFZvVWo7KLAzC3yx5G7kyvSDkc90ppPyw= | ||||
| github.com/cncf/udpa/go v0.0.0-20191209042840-269d4d468f6f/go.mod h1:M8M6+tZqaGXZJjfX53e64911xZQV5JYwmTeXPW+k8Sc= | ||||
| github.com/cncf/udpa/go v0.0.0-20201120205902-5459f2c99403/go.mod h1:WmhPx2Nbnhtbo57+VJT5O0JRkEi1Wbu0z5j0R8u5Hbk= | ||||
| github.com/cncf/xds/go v0.0.0-20210312221358-fbca930ec8ed/go.mod h1:eXthEFrGJvWHgFFCl3hGmgk+/aYT6PnTQLykKQRLhEs= | ||||
| github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E= | ||||
| dario.cat/mergo v1.0.1 h1:Ra4+bf83h2ztPIQYNP99R6m+Y7KfnARDfID+a+vLl4s= | ||||
| dario.cat/mergo v1.0.1/go.mod h1:uNxQE+84aUszobStD9th8a29P2fMDhsBdgRYvZOxGmk= | ||||
| github.com/DATA-DOG/go-sqlmock v1.5.0 h1:Shsta01QNfFxHCfpW6YH2STWB0MudeXXEWMr20OEh60= | ||||
| github.com/DATA-DOG/go-sqlmock v1.5.0/go.mod h1:f/Ixk793poVmq4qj/V1dPUg2JEAKC73Q5eFN3EC/SaM= | ||||
| github.com/KimMachineGun/automemlimit v0.6.1 h1:ILa9j1onAAMadBsyyUJv5cack8Y1WT26yLj/V+ulKp8= | ||||
| github.com/KimMachineGun/automemlimit v0.6.1/go.mod h1:T7xYht7B8r6AG/AqFcUdc7fzd2bIdBKmepfP2S1svPY= | ||||
| github.com/ash3in/uuidv8 v1.0.1 h1:dIq1XRkWT8lGA7N5s7WRTB4V3k49WTBLvILz7aCLp80= | ||||
| github.com/ash3in/uuidv8 v1.0.1/go.mod h1:EoyUgCtxNBnrnpc9efw5rVN1cQ+LFGCoJiFuD6maOMw= | ||||
| github.com/cilium/ebpf v0.16.0 h1:+BiEnHL6Z7lXnlGUsXQPPAE7+kenAd4ES8MQ5min0Ok= | ||||
| github.com/cilium/ebpf v0.16.0/go.mod h1:L7u2Blt2jMM/vLAVgjxluxtBKlz3/GWjB0dMOEngfwE= | ||||
| github.com/containerd/cgroups/v3 v3.0.4 h1:2fs7l3P0Qxb1nKWuJNFiwhp2CqiKzho71DQkDrHJIo4= | ||||
| github.com/containerd/cgroups/v3 v3.0.4/go.mod h1:SA5DLYnXO8pTGYiAHXz94qvLQTKfVM5GEVisn4jpins= | ||||
| github.com/containerd/log v0.1.0 h1:TCJt7ioM2cr/tfR8GPbGf9/VRAX8D2B4PjzCpfX540I= | ||||
| github.com/containerd/log v0.1.0/go.mod h1:VRRf09a7mHDIRezVKTRCrOq78v577GXq3bSa3EhrzVo= | ||||
| github.com/coreos/go-systemd/v22 v22.5.0 h1:RrqgGjYQKalulkV8NGVIfkXQf6YYmOyiJKk8iXXhfZs= | ||||
| github.com/coreos/go-systemd/v22 v22.5.0/go.mod h1:Y58oyj3AT4RCenI/lSvhwexgC+NSVTIJ3seZv2GcEnc= | ||||
| github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | ||||
| github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | ||||
| github.com/docopt/docopt-go v0.0.0-20180111231733-ee0de3bc6815/go.mod h1:WwZ+bS3ebgob9U8Nd0kOddGdZWjyMGR8Wziv+TBNwSE= | ||||
| github.com/envoyproxy/go-control-plane v0.9.0/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4= | ||||
| github.com/envoyproxy/go-control-plane v0.9.1-0.20191026205805-5f8ba28d4473/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4= | ||||
| github.com/envoyproxy/go-control-plane v0.9.4/go.mod h1:6rpuAdCZL397s3pYoYcLgu1mIlRU8Am5FuJP05cCM98= | ||||
| github.com/envoyproxy/go-control-plane v0.9.9-0.20201210154907-fd9021fe5dad/go.mod h1:cXg6YxExXjJnVBQHBLXeUAgxn2UodCpnH306RInaBQk= | ||||
| github.com/envoyproxy/go-control-plane v0.9.9-0.20210512163311-63b5d3c536b0/go.mod h1:hliV/p42l8fGbc6Y9bQ70uLwIvmJyVE5k4iMKlh8wCQ= | ||||
| github.com/envoyproxy/protoc-gen-validate v0.1.0/go.mod h1:iSmxcyjqTsJpI2R4NaDN7+kN2VEUnK/pcBlmesArF7c= | ||||
| github.com/flowstack/go-jsonschema v0.1.1/go.mod h1:yL7fNggx1o8rm9RlgXv7hTBWxdBM0rVwpMwimd3F3N0= | ||||
| github.com/ghodss/yaml v1.0.0/go.mod h1:4dBDuWmgqj2HViK6kFavaiC9ZROes6MMH2rRYeMEF04= | ||||
| github.com/golang/glog v0.0.0-20160126235308-23def4e6c14b/go.mod h1:SBH7ygxi8pfUlaOkMMuAQtPIUF8ecWP5IEl/CR7VP2Q= | ||||
| github.com/golang/mock v1.1.1/go.mod h1:oTYuIxOrZwtPieC+H1uAHpcLFnEyAGVDL/k47Jfbm0A= | ||||
| github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U= | ||||
| github.com/golang/protobuf v1.3.2/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U= | ||||
| github.com/golang/protobuf v1.3.3/go.mod h1:vzj43D7+SQXF/4pzW/hwtAqwc6iTitCiVSaWz5lYuqw= | ||||
| github.com/golang/protobuf v1.4.0-rc.1/go.mod h1:ceaxUfeHdC40wWswd/P6IGgMaK3YpKi5j83Wpe3EHw8= | ||||
| github.com/golang/protobuf v1.4.0-rc.1.0.20200221234624-67d41d38c208/go.mod h1:xKAWHe0F5eneWXFV3EuXVDTCmh+JuBKY0li0aMyXATA= | ||||
| github.com/golang/protobuf v1.4.0-rc.2/go.mod h1:LlEzMj4AhA7rCAGe4KMBDvJI+AwstrUpVNzEA03Pprs= | ||||
| github.com/golang/protobuf v1.4.0-rc.4.0.20200313231945-b860323f09d0/go.mod h1:WU3c8KckQ9AFe+yFwt9sWVRKCVIyN9cPHBJSNnbL67w= | ||||
| github.com/golang/protobuf v1.4.0/go.mod h1:jodUvKwWbYaEsadDk5Fwe5c77LiNKVO9IDvqG2KuDX0= | ||||
| github.com/golang/protobuf v1.4.1/go.mod h1:U8fpvMrcmy5pZrNK1lt4xCsGvpyWQ/VVv6QDs8UjoX8= | ||||
| github.com/golang/protobuf v1.4.2/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI= | ||||
| github.com/golang/protobuf v1.4.3/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI= | ||||
| github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk= | ||||
| github.com/golang/protobuf v1.5.2 h1:ROPKBNFfQgOUMifHyP+KYbvpjbdoFNs+aK7DXlji0Tw= | ||||
| github.com/golang/protobuf v1.5.2/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY= | ||||
| github.com/google/gnostic v0.6.9 h1:ZK/5VhkoX835RikCHpSUJV9a+S3e1zLh59YnyWeBW+0= | ||||
| github.com/google/gnostic v0.6.9/go.mod h1:Nm8234We1lq6iB9OmlgNv3nH91XLLVZHCDayfA3xq+E= | ||||
| github.com/google/go-cmp v0.2.0/go.mod h1:oXzfMopK8JAjlY9xF4vHSVASa0yLyX7SntLO5aqRK0M= | ||||
| github.com/google/go-cmp v0.3.0/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU= | ||||
| github.com/google/go-cmp v0.3.1/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU= | ||||
| github.com/google/go-cmp v0.4.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= | ||||
| github.com/google/go-cmp v0.5.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= | ||||
| github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= | ||||
| github.com/google/go-cmp v0.5.7 h1:81/ik6ipDQS2aGcBfIN5dHDB36BwrStyeAQquSYCV4o= | ||||
| github.com/google/go-cmp v0.5.7/go.mod h1:n+brtR0CgQNWTVd5ZUFpTBC8YFBDLK/h/bpaJ8/DtOE= | ||||
| github.com/google/uuid v1.1.2/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= | ||||
| github.com/grpc-ecosystem/grpc-gateway v1.16.0/go.mod h1:BDjrQk3hbvj6Nolgz8mAMFbcEtjT1g+wF4CSlocrBnw= | ||||
| github.com/imdario/mergo v0.3.13 h1:lFzP57bqS/wsqKssCGmtLAb8A0wKjLGrve2q3PPVcBk= | ||||
| github.com/imdario/mergo v0.3.13/go.mod h1:4lJ1jqUDcsbIECGy0RUJAXNIhg+6ocWgb1ALK2O4oXg= | ||||
| github.com/kr/pretty v0.2.0/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI= | ||||
| github.com/kr/pretty v0.2.1 h1:Fmg33tUaq4/8ym9TJN1x7sLJnHVwhP33CNkpYV/7rwI= | ||||
| github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1VwoXQT9A3Wy9MM3WgvqSxFWenqJduM= | ||||
| github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | ||||
| github.com/docker/go-units v0.5.0 h1:69rxXcBk27SvSaaxTtLh/8llcHD8vYHT7WSdRZ/jvr4= | ||||
| github.com/docker/go-units v0.5.0/go.mod h1:fgPhTUdO+D/Jk86RDLlptpiXQzgHJF7gydDDbaIK4Dk= | ||||
| github.com/go-quicktest/qt v1.101.0 h1:O1K29Txy5P2OK0dGo59b7b0LR6wKfIhttaAhHUyn7eI= | ||||
| github.com/go-quicktest/qt v1.101.0/go.mod h1:14Bz/f7NwaXPtdYEgzsx46kqSxVwTbzVZsDC26tQJow= | ||||
| github.com/godbus/dbus/v5 v5.0.4/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA= | ||||
| github.com/godbus/dbus/v5 v5.1.0 h1:4KLkAxT3aOY8Li4FRJe/KvhoNFFxo0m6fNuFUO8QJUk= | ||||
| github.com/godbus/dbus/v5 v5.1.0/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA= | ||||
| github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek= | ||||
| github.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps= | ||||
| github.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI= | ||||
| github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY= | ||||
| github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0= | ||||
| github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= | ||||
| github.com/josharian/native v1.1.0 h1:uuaP0hAbW7Y4l0ZRQ6C9zfb7Mg1mbFKry/xzDAfmtLA= | ||||
| github.com/josharian/native v1.1.0/go.mod h1:7X/raswPFr05uY3HiLlYeyQntB6OO7E/d2Cu7qoaN2w= | ||||
| github.com/jsimonetti/rtnetlink/v2 v2.0.1 h1:xda7qaHDSVOsADNouv7ukSuicKZO7GgVUCXxpaIEIlM= | ||||
| github.com/jsimonetti/rtnetlink/v2 v2.0.1/go.mod h1:7MoNYNbb3UaDHtF8udiJo/RH6VsTKP1pqKLUTVCvToE= | ||||
| github.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI= | ||||
| github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE= | ||||
| github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk= | ||||
| github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= | ||||
| github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= | ||||
| github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY= | ||||
| github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE= | ||||
| github.com/matoous/go-nanoid v1.5.1 h1:aCjdvTyO9LLnTIi0fgdXhOPPvOHjpXN6Ik9DaNjIct4= | ||||
| github.com/matoous/go-nanoid v1.5.1/go.mod h1:zyD2a71IubI24efhpvkJz+ZwfwagzgSO6UNiFsZKN7U= | ||||
| github.com/mdlayher/netlink v1.7.2 h1:/UtM3ofJap7Vl4QWCPDGXY8d3GIY2UGSDbK+QWmY8/g= | ||||
| github.com/mdlayher/netlink v1.7.2/go.mod h1:xraEF7uJbxLhc5fpHL4cPe221LI2bdttWlU+ZGLfQSw= | ||||
| github.com/mdlayher/socket v0.4.1 h1:eM9y2/jlbs1M615oshPQOHZzj6R6wMT7bX5NPiQvn2U= | ||||
| github.com/mdlayher/socket v0.4.1/go.mod h1:cAqeGjoufqdxWkD7DkpyS+wcefOtmu5OQ8KuoJGIReA= | ||||
| github.com/moby/sys/userns v0.1.0 h1:tVLXkFOxVu9A64/yh59slHVv9ahO9UIev4JZusOLG/g= | ||||
| github.com/moby/sys/userns v0.1.0/go.mod h1:IHUYgu/kao6N8YZlp9Cf444ySSvCmDlmzUcYfDHOl28= | ||||
| github.com/opencontainers/runtime-spec v1.2.0 h1:z97+pHb3uELt/yiAWD691HNHQIF07bE7dzrbT927iTk= | ||||
| github.com/opencontainers/runtime-spec v1.2.0/go.mod h1:jwyrGlmzljRJv/Fgzds9SsS/C5hL+LL3ko9hs6T5lQ0= | ||||
| github.com/patrickmn/go-cache v2.1.0+incompatible h1:HRMgzkcYKYpi3C8ajMPV8OFXaaRUnok+kx1WdO15EQc= | ||||
| github.com/patrickmn/go-cache v2.1.0+incompatible/go.mod h1:3Qf8kWWT7OJRJbdiICTKqZju1ZixQ/KpMGzzAfe6+WQ= | ||||
| github.com/pbnjay/memory v0.0.0-20210728143218-7b4eea64cf58 h1:onHthvaw9LFnH4t2DcNVpwGmV9E1BkGknEliJkfwQj0= | ||||
| github.com/pbnjay/memory v0.0.0-20210728143218-7b4eea64cf58/go.mod h1:DXv8WO4yhMYhSNPKjeNKa5WY9YCIEBRbNzFFPJbWO6Y= | ||||
| github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= | ||||
| github.com/prometheus/client_model v0.0.0-20190812154241-14fe0d1b01d4/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA= | ||||
| github.com/rogpeppe/fastuuid v1.2.0/go.mod h1:jVj6XXZzXRy/MSR5jhDC/2q6DgLz+nrA6LYCDYWNEvQ= | ||||
| github.com/silas/dag v0.0.0-20211117232152-9d50aa809f35 h1:4mohWoM/UGg1BvFFiqSPRl5uwJY3rVV0HQX0ETqauqQ= | ||||
| github.com/silas/dag v0.0.0-20211117232152-9d50aa809f35/go.mod h1:7RTUFBdIRC9nZ7/3RyRNH1bdqIShrDejd1YbLwgPS+I= | ||||
| github.com/spaolacci/murmur3 v0.0.0-20180118202830-f09979ecbc72/go.mod h1:JwIasOWyU6f++ZhiEuf87xNszmSA2myDM2Kzu9HwQUA= | ||||
| github.com/stoewer/go-strcase v1.2.0/go.mod h1:IBiWB2sKIp3wVVQ3Y035++gc+knqhUQag1KpM8ahLw8= | ||||
| github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 h1:Jamvg5psRIccs7FGNTlIRMkT8wgtp5eCXdBlqhYGL6U= | ||||
| github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= | ||||
| github.com/prashantv/gostub v1.1.0 h1:BTyx3RfQjRHnUWaGF9oQos79AlQ5k8WNktv7VGvVH4g= | ||||
| github.com/prashantv/gostub v1.1.0/go.mod h1:A5zLQHz7ieHGG7is6LLXLz7I8+3LZzsrV0P1IAHhP5U= | ||||
| github.com/rogpeppe/go-internal v1.13.1 h1:KvO1DLK/DRN07sQ1LQKScxyZJuNnedQ5/wKSR38lUII= | ||||
| github.com/rogpeppe/go-internal v1.13.1/go.mod h1:uMEvuHeurkdAXX61udpOXGD/AzZDWNMNyH2VO9fmH0o= | ||||
| github.com/silas/dag v0.0.0-20220518035006-a7e85ada93c5 h1:G/FZtUu7a6NTWl3KUHMV9jkLAh/Rvtf03NWMHaEDl+E= | ||||
| github.com/silas/dag v0.0.0-20220518035006-a7e85ada93c5/go.mod h1:7RTUFBdIRC9nZ7/3RyRNH1bdqIShrDejd1YbLwgPS+I= | ||||
| github.com/sirupsen/logrus v1.9.3 h1:dueUQJ1C2q9oE3F7wvmSGAaVtTmUizReu6fjN8uqzbQ= | ||||
| github.com/sirupsen/logrus v1.9.3/go.mod h1:naHLuLoDiP4jHNo9R0sCBMtWGeIprob74mVsIT4qYEQ= | ||||
| github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= | ||||
| github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= | ||||
| github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA= | ||||
| github.com/xeipuuv/gojsonpointer v0.0.0-20180127040702-4e3ac2762d5f/go.mod h1:N2zxlSyiKSe5eX1tZViRH5QA0qijqEDrYZiPEAiq3wU= | ||||
| github.com/xeipuuv/gojsonreference v0.0.0-20180127040603-bd5ef7bd5415/go.mod h1:GwrjFmJcFw6At/Gs6z4yjiIwzuJ1/+UwLxMQDVQXShQ= | ||||
| github.com/xeipuuv/gojsonschema v1.2.0/go.mod h1:anYRn/JVcOK2ZgGU+IjEV4nwlhoK5sQluxsYJ78Id3Y= | ||||
| go.opentelemetry.io/proto/otlp v0.7.0/go.mod h1:PqfVotwruBrMGOCsRd/89rSnXhoiJIqeYNgFYFoEGnI= | ||||
| go.unistack.org/micro-proto/v3 v3.3.1 h1:nQ0MtWvP2G3QrpOgawVOPhpZZYkq6umTGDqs8FxJYIo= | ||||
| go.unistack.org/micro-proto/v3 v3.3.1/go.mod h1:cwRyv8uInM2I7EbU7O8Fx2Ls3N90Uw9UCCcq4olOdfE= | ||||
| golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= | ||||
| golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto= | ||||
| golang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA= | ||||
| golang.org/x/lint v0.0.0-20181026193005-c67002cb31c3/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE= | ||||
| golang.org/x/lint v0.0.0-20190227174305-5b3e6a55c961/go.mod h1:wehouNa3lNwaWXcvxsM5YxQ5yQlVC4a0KAMCusXpPoU= | ||||
| golang.org/x/lint v0.0.0-20190313153728-d0100b6bd8b3/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc= | ||||
| golang.org/x/net v0.0.0-20180724234803-3673e40ba225/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= | ||||
| golang.org/x/net v0.0.0-20180826012351-8a410e7b638d/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= | ||||
| golang.org/x/net v0.0.0-20190108225652-1e06a53dbb7e/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= | ||||
| golang.org/x/net v0.0.0-20190213061140-3a22650c66bd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= | ||||
| golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= | ||||
| golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= | ||||
| golang.org/x/net v0.0.0-20200822124328-c89045814202/go.mod h1:/O7V0waA8r7cgGh81Ro3o1hOxt32SMVPicZroKQ2sZA= | ||||
| golang.org/x/net v0.0.0-20210405180319-a5a99cb37ef4/go.mod h1:p54w0d4576C0XHj96bSt6lcn1PtDYWL6XObtHCRCNQM= | ||||
| golang.org/x/net v0.0.0-20210805182204-aaa1db679c0d/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y= | ||||
| golang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U= | ||||
| golang.org/x/oauth2 v0.0.0-20200107190931-bf48bf16ab8d/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw= | ||||
| golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= | ||||
| golang.org/x/sync v0.0.0-20181108010431-42b317875d0f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= | ||||
| golang.org/x/sync v0.0.0-20181221193216-37e7f081c4d4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= | ||||
| golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= | ||||
| golang.org/x/sys v0.0.0-20180830151530-49385e6e1522/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= | ||||
| golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= | ||||
| golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= | ||||
| golang.org/x/sys v0.0.0-20200323222414-85ca7c5b95cd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= | ||||
| golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= | ||||
| golang.org/x/sys v0.0.0-20210330210617-4fbd30eecc44/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= | ||||
| golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= | ||||
| golang.org/x/sys v0.0.0-20210510120138-977fb7262007/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||||
| golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= | ||||
| golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= | ||||
| golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= | ||||
| golang.org/x/text v0.3.5/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= | ||||
| golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= | ||||
| golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= | ||||
| golang.org/x/tools v0.0.0-20190114222345-bf090417da8b/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= | ||||
| golang.org/x/tools v0.0.0-20190226205152-f727befe758c/go.mod h1:9Yl7xja0Znq3iFh3HoIrodX9oNMXvdceNzlUR8zjMvY= | ||||
| golang.org/x/tools v0.0.0-20190311212946-11955173bddd/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs= | ||||
| golang.org/x/tools v0.0.0-20190524140312-2c0ae7006135/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q= | ||||
| golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= | ||||
| golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1 h1:go1bK/D/BFZV2I8cIQd1NKEZ+0owSTG1fDTci4IqFcE= | ||||
| golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= | ||||
| google.golang.org/appengine v1.1.0/go.mod h1:EbEs0AVv82hx2wNQdGPgUI5lhzA/G0D9YwlJXL52JkM= | ||||
| google.golang.org/appengine v1.4.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4= | ||||
| google.golang.org/genproto v0.0.0-20180817151627-c66870c02cf8/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc= | ||||
| google.golang.org/genproto v0.0.0-20190819201941-24fa4b261c55/go.mod h1:DMBHOl98Agz4BDEuKkezgsaosCRResVns1a3J2ZsMNc= | ||||
| google.golang.org/genproto v0.0.0-20200513103714-09dca8ec2884/go.mod h1:55QSHmfGQM9UVYDPBsyGGes0y52j32PQ3BqQfXhyH3c= | ||||
| google.golang.org/genproto v0.0.0-20200526211855-cb27e3aa2013/go.mod h1:NbSheEEYHJ7i3ixzK3sjbqSGDJWnxyFXZblF3eUsNvo= | ||||
| google.golang.org/genproto v0.0.0-20220107163113-42d7afdf6368/go.mod h1:5CzLGKJ67TSI2B9POpiiyGha0AjJvZIUgRMt1dSmuhc= | ||||
| google.golang.org/grpc v1.19.0/go.mod h1:mqu4LbDTu4XGKhr4mRzUsmM4RtVoemTSY81AxZiDr8c= | ||||
| google.golang.org/grpc v1.23.0/go.mod h1:Y5yQAOtifL1yxbo5wqy6BxZv8vAUGQwXBOALyacEbxg= | ||||
| google.golang.org/grpc v1.25.1/go.mod h1:c3i+UQWmh7LiEpx4sFZnkU36qjEYZ0imhYfXVyQciAY= | ||||
| google.golang.org/grpc v1.27.0/go.mod h1:qbnxyOmOxrQa7FizSgH+ReBfzJrCY1pSN7KXBS8abTk= | ||||
| google.golang.org/grpc v1.33.1/go.mod h1:fr5YgcSWrqhRRxogOsw7RzIpsmvOZ6IcH4kBYTpR3n0= | ||||
| google.golang.org/grpc v1.36.0/go.mod h1:qjiiYl8FncCW8feJPdyg3v6XW24KsRHe+dy9BAGRRjU= | ||||
| google.golang.org/grpc v1.40.0/go.mod h1:ogyxbiOoUXAkP+4+xa6PZSE9DZgIHtSpzjDTB9KAK34= | ||||
| google.golang.org/protobuf v0.0.0-20200109180630-ec00e32a8dfd/go.mod h1:DFci5gLYBciE7Vtevhsrf46CRTquxDuWsQurQQe4oz8= | ||||
| google.golang.org/protobuf v0.0.0-20200221191635-4d8936d0db64/go.mod h1:kwYJMbMJ01Woi6D6+Kah6886xMZcty6N08ah7+eCXa0= | ||||
| google.golang.org/protobuf v0.0.0-20200228230310-ab0ca4ff8a60/go.mod h1:cfTl7dwQJ+fmap5saPgwCLgHXTUD7jkjRqWcaiX5VyM= | ||||
| google.golang.org/protobuf v1.20.1-0.20200309200217-e05f789c0967/go.mod h1:A+miEFZTKqfCUM6K7xSMQL9OKL/b6hQv+e19PK+JZNE= | ||||
| google.golang.org/protobuf v1.21.0/go.mod h1:47Nbq4nVaFHyn7ilMalzfO3qCViNmqZ2kzikPIcrTAo= | ||||
| google.golang.org/protobuf v1.22.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU= | ||||
| google.golang.org/protobuf v1.23.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU= | ||||
| google.golang.org/protobuf v1.23.1-0.20200526195155-81db48ad09cc/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU= | ||||
| google.golang.org/protobuf v1.25.0/go.mod h1:9JNX74DMeImyA3h4bdi1ymwjUzf21/xIlbajtzgsN7c= | ||||
| 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.27.1/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc= | ||||
| google.golang.org/protobuf v1.28.1 h1:d0NfwRgPtno5B1Wa6L2DAG+KivqkdutMf1UhdNx175w= | ||||
| google.golang.org/protobuf v1.28.1/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I= | ||||
| github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= | ||||
| github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA= | ||||
| github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY= | ||||
| 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/goleak v1.3.0 h1:2K3zAYmnTNqV73imy9J1T3WC+gmCePx2hEGkimedGto= | ||||
| go.uber.org/goleak v1.3.0/go.mod h1:CoHD4mav9JJNrW/WLlf7HGZPjdw8EucARQHekz1X6bE= | ||||
| go.unistack.org/micro-proto/v3 v3.4.1 h1:UTjLSRz2YZuaHk9iSlVqqsA50JQNAEK2ZFboGqtEa9Q= | ||||
| go.unistack.org/micro-proto/v3 v3.4.1/go.mod h1:okx/cnOhzuCX0ggl/vToatbCupi0O44diiiLLsZ93Zo= | ||||
| golang.org/x/exp v0.0.0-20241210194714-1829a127f884 h1:Y/Mj/94zIQQGHVSv1tTtQBDaQaJe62U9bkDZKKyhPCU= | ||||
| golang.org/x/exp v0.0.0-20241210194714-1829a127f884/go.mod h1:qj5a5QZpwLU2NLQudwIN5koi3beDhSAlJwa67PuM98c= | ||||
| golang.org/x/net v0.32.0 h1:ZqPmj8Kzc+Y6e0+skZsuACbx+wzMgo5MQsJh9Qd6aYI= | ||||
| golang.org/x/net v0.32.0/go.mod h1:CwU0IoeOlnQQWJ6ioyFrfRuomB8GKF6KbYXZVyeXNfs= | ||||
| golang.org/x/sync v0.10.0 h1:3NQrjDixjgGwUOCaF8w2+VYHv0Ve/vGYSbdkTa98gmQ= | ||||
| golang.org/x/sync v0.10.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk= | ||||
| golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||||
| golang.org/x/sys v0.28.0 h1:Fksou7UEQUWlKvIdsqzJmUmCX3cZuD2+P3XyyzwMhlA= | ||||
| golang.org/x/sys v0.28.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= | ||||
| golang.org/x/text v0.21.0 h1:zyQAAkrwaneQ066sspRyJaG9VNi/YJ1NfzcGB3hZ/qo= | ||||
| golang.org/x/text v0.21.0/go.mod h1:4IBbMaMmOPCJ8SecivzSH54+73PCFmPWxNTLm+vZkEQ= | ||||
| google.golang.org/genproto/googleapis/rpc v0.0.0-20241209162323-e6fa225c2576 h1:8ZmaLZE4XWrtU3MyClkYqqtl6Oegr3235h7jxsDyqCY= | ||||
| google.golang.org/genproto/googleapis/rpc v0.0.0-20241209162323-e6fa225c2576/go.mod h1:5uTbfoYQed2U9p3KIj2/Zzm02PYhndfdmML0qC3q3FU= | ||||
| google.golang.org/grpc v1.68.1 h1:oI5oTa11+ng8r8XMMN7jAOmWfPZWbYpCFaMUTACxkM0= | ||||
| google.golang.org/grpc v1.68.1/go.mod h1:+q1XYFJjShcqn0QZHvCyeR4CXPA+llXIeUIfIe00waw= | ||||
| google.golang.org/protobuf v1.35.2 h1:8Ar7bF+apOIoThw1EdZl0p1oWvMqTHmpA2fRTyZO8io= | ||||
| google.golang.org/protobuf v1.35.2/go.mod h1:9fA7Ob0pmnwhb644+1+CVWFRbNajQ6iRojtC/QF5bRE= | ||||
| gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= | ||||
| gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15 h1:YR8cESwS4TdDjEe65xsg0ogRM/Nc3DYOhEAlW+xobZo= | ||||
| gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= | ||||
| gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= | ||||
| gopkg.in/yaml.v2 v2.2.3/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= | ||||
| gopkg.in/yaml.v3 v3.0.0-20200615113413-eeeca48fe776/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= | ||||
| gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= | ||||
| gopkg.in/yaml.v3 v3.0.0 h1:hjy8E9ON/egN1tAYqKb61G10WtihqetD4sz2H+8nIeA= | ||||
| gopkg.in/yaml.v3 v3.0.0/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= | ||||
| honnef.co/go/tools v0.0.0-20190102054323-c2f93a96b099/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4= | ||||
| honnef.co/go/tools v0.0.0-20190523083050-ea95bdfd59fc/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4= | ||||
| gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk= | ||||
| gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q= | ||||
| gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= | ||||
| gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= | ||||
| gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= | ||||
|   | ||||
| @@ -13,6 +13,15 @@ func FromContext(ctx context.Context) (Logger, bool) { | ||||
| 	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 | ||||
| func NewContext(ctx context.Context, l Logger) context.Context { | ||||
| 	if ctx == nil { | ||||
|   | ||||
| @@ -1,230 +0,0 @@ | ||||
| package logger | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"encoding/json" | ||||
| 	"fmt" | ||||
| 	"os" | ||||
| 	"runtime" | ||||
| 	"strings" | ||||
| 	"sync" | ||||
| 	"time" | ||||
| ) | ||||
|  | ||||
| type defaultLogger struct { | ||||
| 	enc  *json.Encoder | ||||
| 	opts Options | ||||
| 	sync.RWMutex | ||||
| } | ||||
|  | ||||
| // Init(opts...) should only overwrite provided options | ||||
| func (l *defaultLogger) Init(opts ...Option) error { | ||||
| 	l.Lock() | ||||
| 	for _, o := range opts { | ||||
| 		o(&l.opts) | ||||
| 	} | ||||
| 	l.enc = json.NewEncoder(l.opts.Out) | ||||
| 	// wrap the Log func | ||||
| 	l.Unlock() | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (l *defaultLogger) String() string { | ||||
| 	return "micro" | ||||
| } | ||||
|  | ||||
| func (l *defaultLogger) Clone(opts ...Option) Logger { | ||||
| 	newopts := NewOptions(opts...) | ||||
| 	oldopts := l.opts | ||||
| 	for _, o := range opts { | ||||
| 		o(&newopts) | ||||
| 		o(&oldopts) | ||||
| 	} | ||||
|  | ||||
| 	l.Lock() | ||||
| 	cl := &defaultLogger{opts: oldopts, enc: json.NewEncoder(l.opts.Out)} | ||||
| 	l.Unlock() | ||||
|  | ||||
| 	return cl | ||||
| } | ||||
|  | ||||
| func (l *defaultLogger) V(level Level) bool { | ||||
| 	l.RLock() | ||||
| 	ok := l.opts.Level.Enabled(level) | ||||
| 	l.RUnlock() | ||||
| 	return ok | ||||
| } | ||||
|  | ||||
| func (l *defaultLogger) Level(level Level) { | ||||
| 	l.Lock() | ||||
| 	l.opts.Level = level | ||||
| 	l.Unlock() | ||||
| } | ||||
|  | ||||
| func (l *defaultLogger) Fields(fields ...interface{}) Logger { | ||||
| 	l.RLock() | ||||
| 	nl := &defaultLogger{opts: l.opts, enc: l.enc} | ||||
| 	if len(fields) == 0 { | ||||
| 		l.RUnlock() | ||||
| 		return nl | ||||
| 	} else if len(fields)%2 != 0 { | ||||
| 		fields = fields[:len(fields)-1] | ||||
| 	} | ||||
| 	nl.opts.Fields = copyFields(l.opts.Fields) | ||||
| 	nl.opts.Fields = append(nl.opts.Fields, fields...) | ||||
| 	l.RUnlock() | ||||
| 	return nl | ||||
| } | ||||
|  | ||||
| func copyFields(src []interface{}) []interface{} { | ||||
| 	dst := make([]interface{}, len(src)) | ||||
| 	copy(dst, src) | ||||
| 	return dst | ||||
| } | ||||
|  | ||||
| // logCallerfilePath returns a package/file:line description of the caller, | ||||
| // preserving only the leaf directory name and file name. | ||||
| func logCallerfilePath(loggingFilePath string) string { | ||||
| 	// To make sure we trim the path correctly on Windows too, we | ||||
| 	// counter-intuitively need to use '/' and *not* os.PathSeparator here, | ||||
| 	// because the path given originates from Go stdlib, specifically | ||||
| 	// runtime.Caller() which (as of Mar/17) returns forward slashes even on | ||||
| 	// Windows. | ||||
| 	// | ||||
| 	// See https://github.com/golang/go/issues/3335 | ||||
| 	// and https://github.com/golang/go/issues/18151 | ||||
| 	// | ||||
| 	// for discussion on the issue on Go side. | ||||
| 	idx := strings.LastIndexByte(loggingFilePath, '/') | ||||
| 	if idx == -1 { | ||||
| 		return loggingFilePath | ||||
| 	} | ||||
| 	idx = strings.LastIndexByte(loggingFilePath[:idx], '/') | ||||
| 	if idx == -1 { | ||||
| 		return loggingFilePath | ||||
| 	} | ||||
| 	return loggingFilePath[idx+1:] | ||||
| } | ||||
|  | ||||
| func (l *defaultLogger) Info(ctx context.Context, args ...interface{}) { | ||||
| 	l.Log(ctx, InfoLevel, args...) | ||||
| } | ||||
|  | ||||
| func (l *defaultLogger) Error(ctx context.Context, args ...interface{}) { | ||||
| 	l.Log(ctx, ErrorLevel, args...) | ||||
| } | ||||
|  | ||||
| func (l *defaultLogger) Debug(ctx context.Context, args ...interface{}) { | ||||
| 	l.Log(ctx, DebugLevel, args...) | ||||
| } | ||||
|  | ||||
| func (l *defaultLogger) Warn(ctx context.Context, args ...interface{}) { | ||||
| 	l.Log(ctx, WarnLevel, args...) | ||||
| } | ||||
|  | ||||
| func (l *defaultLogger) Trace(ctx context.Context, args ...interface{}) { | ||||
| 	l.Log(ctx, TraceLevel, args...) | ||||
| } | ||||
|  | ||||
| func (l *defaultLogger) Fatal(ctx context.Context, args ...interface{}) { | ||||
| 	l.Log(ctx, FatalLevel, args...) | ||||
| 	os.Exit(1) | ||||
| } | ||||
|  | ||||
| func (l *defaultLogger) Infof(ctx context.Context, msg string, args ...interface{}) { | ||||
| 	l.Logf(ctx, InfoLevel, msg, args...) | ||||
| } | ||||
|  | ||||
| func (l *defaultLogger) Errorf(ctx context.Context, msg string, args ...interface{}) { | ||||
| 	l.Logf(ctx, ErrorLevel, msg, args...) | ||||
| } | ||||
|  | ||||
| func (l *defaultLogger) Debugf(ctx context.Context, msg string, args ...interface{}) { | ||||
| 	l.Logf(ctx, DebugLevel, msg, args...) | ||||
| } | ||||
|  | ||||
| func (l *defaultLogger) Warnf(ctx context.Context, msg string, args ...interface{}) { | ||||
| 	l.Logf(ctx, WarnLevel, msg, args...) | ||||
| } | ||||
|  | ||||
| func (l *defaultLogger) Tracef(ctx context.Context, msg string, args ...interface{}) { | ||||
| 	l.Logf(ctx, TraceLevel, msg, args...) | ||||
| } | ||||
|  | ||||
| func (l *defaultLogger) Fatalf(ctx context.Context, msg string, args ...interface{}) { | ||||
| 	l.Logf(ctx, FatalLevel, msg, args...) | ||||
| 	os.Exit(1) | ||||
| } | ||||
|  | ||||
| func (l *defaultLogger) Log(ctx context.Context, level Level, args ...interface{}) { | ||||
| 	if !l.V(level) { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	l.RLock() | ||||
| 	fields := copyFields(l.opts.Fields) | ||||
| 	l.RUnlock() | ||||
|  | ||||
| 	fields = append(fields, "level", level.String()) | ||||
|  | ||||
| 	if _, file, line, ok := runtime.Caller(l.opts.CallerSkipCount); ok { | ||||
| 		fields = append(fields, "caller", fmt.Sprintf("%s:%d", logCallerfilePath(file), line)) | ||||
| 	} | ||||
| 	fields = append(fields, "timestamp", time.Now().Format("2006-01-02 15:04:05")) | ||||
|  | ||||
| 	if len(args) > 0 { | ||||
| 		fields = append(fields, "msg", fmt.Sprint(args...)) | ||||
| 	} | ||||
|  | ||||
| 	out := make(map[string]interface{}, len(fields)/2) | ||||
| 	for i := 0; i < len(fields); i += 2 { | ||||
| 		out[fields[i].(string)] = fields[i+1] | ||||
| 	} | ||||
| 	l.RLock() | ||||
| 	_ = l.enc.Encode(out) | ||||
| 	l.RUnlock() | ||||
| } | ||||
|  | ||||
| func (l *defaultLogger) Logf(ctx context.Context, level Level, msg string, args ...interface{}) { | ||||
| 	if !l.V(level) { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	l.RLock() | ||||
| 	fields := copyFields(l.opts.Fields) | ||||
| 	l.RUnlock() | ||||
|  | ||||
| 	fields = append(fields, "level", level.String()) | ||||
|  | ||||
| 	if _, file, line, ok := runtime.Caller(l.opts.CallerSkipCount); ok { | ||||
| 		fields = append(fields, "caller", fmt.Sprintf("%s:%d", logCallerfilePath(file), line)) | ||||
| 	} | ||||
|  | ||||
| 	fields = append(fields, "timestamp", time.Now().Format("2006-01-02 15:04:05")) | ||||
| 	if len(args) > 0 { | ||||
| 		fields = append(fields, "msg", fmt.Sprintf(msg, args...)) | ||||
| 	} else if msg != "" { | ||||
| 		fields = append(fields, "msg", msg) | ||||
| 	} | ||||
|  | ||||
| 	out := make(map[string]interface{}, len(fields)/2) | ||||
| 	for i := 0; i < len(fields); i += 2 { | ||||
| 		out[fields[i].(string)] = fields[i+1] | ||||
| 	} | ||||
| 	l.RLock() | ||||
| 	_ = l.enc.Encode(out) | ||||
| 	l.RUnlock() | ||||
| } | ||||
|  | ||||
| func (l *defaultLogger) Options() Options { | ||||
| 	return l.opts | ||||
| } | ||||
|  | ||||
| // NewLogger builds a new logger based on options | ||||
| func NewLogger(opts ...Option) Logger { | ||||
| 	l := &defaultLogger{ | ||||
| 		opts: NewOptions(opts...), | ||||
| 	} | ||||
| 	l.enc = json.NewEncoder(l.opts.Out) | ||||
| 	return l | ||||
| } | ||||
							
								
								
									
										118
									
								
								logger/logger.go
									
									
									
									
									
								
							
							
						
						
									
										118
									
								
								logger/logger.go
									
									
									
									
									
								
							| @@ -1,18 +1,19 @@ | ||||
| // Package logger provides a log interface | ||||
| package logger // import "go.unistack.org/micro/v3/logger" | ||||
| package logger | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"os" | ||||
| ) | ||||
|  | ||||
| type ContextAttrFunc func(ctx context.Context) []interface{} | ||||
|  | ||||
| var DefaultContextAttrFuncs []ContextAttrFunc | ||||
|  | ||||
| var ( | ||||
| 	// DefaultLogger variable | ||||
| 	DefaultLogger = NewLogger(WithLevel(ParseLevel(os.Getenv("MICRO_LOG_LEVEL")))) | ||||
| 	DefaultLogger = NewLogger() | ||||
| 	// DefaultLevel used by logger | ||||
| 	DefaultLevel = InfoLevel | ||||
| 	// DefaultCallerSkipCount used by logger | ||||
| 	DefaultCallerSkipCount = 2 | ||||
| ) | ||||
|  | ||||
| // Logger is a generic logging interface | ||||
| @@ -30,111 +31,24 @@ type Logger interface { | ||||
| 	// Fields set fields to always be logged with keyval pairs | ||||
| 	Fields(fields ...interface{}) Logger | ||||
| 	// Info level message | ||||
| 	Info(ctx context.Context, args ...interface{}) | ||||
| 	Info(ctx context.Context, msg string, args ...interface{}) | ||||
| 	// Trace level message | ||||
| 	Trace(ctx context.Context, args ...interface{}) | ||||
| 	Trace(ctx context.Context, msg string, args ...interface{}) | ||||
| 	// Debug level message | ||||
| 	Debug(ctx context.Context, args ...interface{}) | ||||
| 	Debug(ctx context.Context, msg string, args ...interface{}) | ||||
| 	// Warn level message | ||||
| 	Warn(ctx context.Context, args ...interface{}) | ||||
| 	Warn(ctx context.Context, msg string, args ...interface{}) | ||||
| 	// Error level message | ||||
| 	Error(ctx context.Context, args ...interface{}) | ||||
| 	Error(ctx context.Context, msg string, args ...interface{}) | ||||
| 	// Fatal level message | ||||
| 	Fatal(ctx context.Context, args ...interface{}) | ||||
| 	// Infof level message | ||||
| 	Infof(ctx context.Context, msg string, args ...interface{}) | ||||
| 	// Tracef level message | ||||
| 	Tracef(ctx context.Context, msg string, args ...interface{}) | ||||
| 	// Debug level message | ||||
| 	Debugf(ctx context.Context, msg string, args ...interface{}) | ||||
| 	// Warn level message | ||||
| 	Warnf(ctx context.Context, msg string, args ...interface{}) | ||||
| 	// Error level message | ||||
| 	Errorf(ctx context.Context, msg string, args ...interface{}) | ||||
| 	// Fatal level message | ||||
| 	Fatalf(ctx context.Context, msg string, args ...interface{}) | ||||
| 	Fatal(ctx context.Context, msg string, args ...interface{}) | ||||
| 	// Log logs message with needed level | ||||
| 	Log(ctx context.Context, level Level, args ...interface{}) | ||||
| 	// Logf logs message with needed level | ||||
| 	Logf(ctx context.Context, level Level, msg string, args ...interface{}) | ||||
| 	// String returns the name of logger | ||||
| 	Log(ctx context.Context, level Level, msg string, args ...interface{}) | ||||
| 	// Name returns broker instance name | ||||
| 	Name() string | ||||
| 	// String returns the type of logger | ||||
| 	String() string | ||||
| } | ||||
|  | ||||
| // Field contains keyval pair | ||||
| type Field interface{} | ||||
|  | ||||
| // Info writes msg to default logger on info level | ||||
| func Info(ctx context.Context, args ...interface{}) { | ||||
| 	DefaultLogger.Info(ctx, args...) | ||||
| } | ||||
|  | ||||
| // Error writes msg to default logger on error level | ||||
| func Error(ctx context.Context, args ...interface{}) { | ||||
| 	DefaultLogger.Error(ctx, args...) | ||||
| } | ||||
|  | ||||
| // Debug writes msg to default logger on debug level | ||||
| func Debug(ctx context.Context, args ...interface{}) { | ||||
| 	DefaultLogger.Debug(ctx, args...) | ||||
| } | ||||
|  | ||||
| // Warn writes msg to default logger on warn level | ||||
| func Warn(ctx context.Context, args ...interface{}) { | ||||
| 	DefaultLogger.Warn(ctx, args...) | ||||
| } | ||||
|  | ||||
| // Trace writes msg to default logger on trace level | ||||
| func Trace(ctx context.Context, args ...interface{}) { | ||||
| 	DefaultLogger.Trace(ctx, args...) | ||||
| } | ||||
|  | ||||
| // Fatal writes msg to default logger on fatal level | ||||
| func Fatal(ctx context.Context, args ...interface{}) { | ||||
| 	DefaultLogger.Fatal(ctx, args...) | ||||
| } | ||||
|  | ||||
| // Infof writes formatted msg to default logger on info level | ||||
| func Infof(ctx context.Context, msg string, args ...interface{}) { | ||||
| 	DefaultLogger.Infof(ctx, msg, args...) | ||||
| } | ||||
|  | ||||
| // Errorf writes formatted msg to default logger on error level | ||||
| func Errorf(ctx context.Context, msg string, args ...interface{}) { | ||||
| 	DefaultLogger.Errorf(ctx, msg, args...) | ||||
| } | ||||
|  | ||||
| // Debugf writes formatted msg to default logger on debug level | ||||
| func Debugf(ctx context.Context, msg string, args ...interface{}) { | ||||
| 	DefaultLogger.Debugf(ctx, msg, args...) | ||||
| } | ||||
|  | ||||
| // Warnf writes formatted msg to default logger on warn level | ||||
| func Warnf(ctx context.Context, msg string, args ...interface{}) { | ||||
| 	DefaultLogger.Warnf(ctx, msg, args...) | ||||
| } | ||||
|  | ||||
| // Tracef writes formatted msg to default logger on trace level | ||||
| func Tracef(ctx context.Context, msg string, args ...interface{}) { | ||||
| 	DefaultLogger.Tracef(ctx, msg, args...) | ||||
| } | ||||
|  | ||||
| // Fatalf writes formatted msg to default logger on fatal level | ||||
| func Fatalf(ctx context.Context, msg string, args ...interface{}) { | ||||
| 	DefaultLogger.Fatalf(ctx, msg, args...) | ||||
| } | ||||
|  | ||||
| // V returns true if passed level enabled in default logger | ||||
| func V(level Level) bool { | ||||
| 	return DefaultLogger.V(level) | ||||
| } | ||||
|  | ||||
| // Init initialize logger | ||||
| func Init(opts ...Option) error { | ||||
| 	return DefaultLogger.Init(opts...) | ||||
| } | ||||
|  | ||||
| // Fields create logger with specific fields | ||||
| func Fields(fields ...interface{}) Logger { | ||||
| 	return DefaultLogger.Fields(fields...) | ||||
| } | ||||
|   | ||||
| @@ -1,138 +0,0 @@ | ||||
| package logger | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"context" | ||||
| 	"log" | ||||
| 	"testing" | ||||
| ) | ||||
|  | ||||
| func TestContext(t *testing.T) { | ||||
| 	ctx := context.TODO() | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	l := NewLogger(WithLevel(TraceLevel), WithOutput(buf)) | ||||
| 	if err := l.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	nl, ok := FromContext(NewContext(ctx, l.Fields("key", "val"))) | ||||
| 	if !ok { | ||||
| 		t.Fatal("context without logger") | ||||
| 	} | ||||
| 	nl.Info(ctx, "message") | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"key":"val"`)) { | ||||
| 		t.Fatalf("logger fields not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestFields(t *testing.T) { | ||||
| 	ctx := context.TODO() | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	l := NewLogger(WithLevel(TraceLevel), WithOutput(buf)) | ||||
| 	if err := l.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	nl := l.Fields("key", "val") | ||||
|  | ||||
| 	nl.Info(ctx, "message") | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"key":"val"`)) { | ||||
| 		t.Fatalf("logger fields not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestFromContextWithFields(t *testing.T) { | ||||
| 	ctx := context.TODO() | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	var ok bool | ||||
| 	l := NewLogger(WithLevel(TraceLevel), WithOutput(buf)) | ||||
| 	if err := l.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	nl := l.Fields("key", "val") | ||||
|  | ||||
| 	ctx = NewContext(ctx, nl) | ||||
|  | ||||
| 	l, ok = FromContext(ctx) | ||||
| 	if !ok { | ||||
| 		t.Fatalf("context does not have logger") | ||||
| 	} | ||||
|  | ||||
| 	l.Info(ctx, "message") | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"key":"val"`)) { | ||||
| 		t.Fatalf("logger fields not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestClone(t *testing.T) { | ||||
| 	ctx := context.TODO() | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	l := NewLogger(WithLevel(TraceLevel), WithOutput(buf)) | ||||
| 	if err := l.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	nl := l.Clone(WithLevel(ErrorLevel)) | ||||
| 	if err := nl.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	nl.Info(ctx, "info message") | ||||
| 	if len(buf.Bytes()) != 0 { | ||||
| 		t.Fatal("message must not be logged") | ||||
| 	} | ||||
| 	l.Info(ctx, "info message") | ||||
| 	if len(buf.Bytes()) == 0 { | ||||
| 		t.Fatal("message must be logged") | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestRedirectStdLogger(t *testing.T) { | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	l := NewLogger(WithLevel(TraceLevel), WithOutput(buf)) | ||||
| 	if err := l.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	fn := RedirectStdLogger(l, ErrorLevel) | ||||
| 	defer fn() | ||||
| 	log.Print("test") | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"level":"error","msg":"test","timestamp"`)) { | ||||
| 		t.Fatalf("logger error, buf %s", buf.Bytes()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestStdLogger(t *testing.T) { | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	l := NewLogger(WithLevel(TraceLevel), WithOutput(buf)) | ||||
| 	if err := l.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	lg := NewStdLogger(l, ErrorLevel) | ||||
| 	lg.Print("test") | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"level":"error","msg":"test","timestamp"`)) { | ||||
| 		t.Fatalf("logger error, buf %s", buf.Bytes()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestLogger(t *testing.T) { | ||||
| 	ctx := context.TODO() | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	l := NewLogger(WithLevel(TraceLevel), WithOutput(buf)) | ||||
| 	if err := l.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	l.Trace(ctx, "trace_msg1") | ||||
| 	l.Warn(ctx, "warn_msg1") | ||||
| 	l.Fields("error", "test").Info(ctx, "error message") | ||||
| 	l.Warn(ctx, "first", " ", "second") | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"level":"trace","msg":"trace_msg1"`)) { | ||||
| 		t.Fatalf("logger error, buf %s", buf.Bytes()) | ||||
| 	} | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"warn","msg":"warn_msg1"`)) { | ||||
| 		t.Fatalf("logger error, buf %s", buf.Bytes()) | ||||
| 	} | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"error":"test","level":"info","msg":"error message"`)) { | ||||
| 		t.Fatalf("logger error, buf %s", buf.Bytes()) | ||||
| 	} | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"level":"warn","msg":"first second"`)) { | ||||
| 		t.Fatalf("logger error, buf %s", buf.Bytes()) | ||||
| 	} | ||||
| } | ||||
							
								
								
									
										78
									
								
								logger/noop.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										78
									
								
								logger/noop.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,78 @@ | ||||
| package logger | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| ) | ||||
|  | ||||
| const ( | ||||
| 	defaultCallerSkipCount = 2 | ||||
| ) | ||||
|  | ||||
| type noopLogger struct { | ||||
| 	opts Options | ||||
| } | ||||
|  | ||||
| func NewLogger(opts ...Option) Logger { | ||||
| 	options := NewOptions(opts...) | ||||
| 	options.CallerSkipCount = defaultCallerSkipCount | ||||
| 	return &noopLogger{opts: options} | ||||
| } | ||||
|  | ||||
| func (l *noopLogger) V(_ Level) bool { | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| func (l *noopLogger) Level(_ Level) { | ||||
| } | ||||
|  | ||||
| func (l *noopLogger) Name() string { | ||||
| 	return l.opts.Name | ||||
| } | ||||
|  | ||||
| func (l *noopLogger) Init(opts ...Option) error { | ||||
| 	for _, o := range opts { | ||||
| 		o(&l.opts) | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (l *noopLogger) Clone(opts ...Option) Logger { | ||||
| 	nl := &noopLogger{opts: l.opts} | ||||
| 	for _, o := range opts { | ||||
| 		o(&nl.opts) | ||||
| 	} | ||||
| 	return nl | ||||
| } | ||||
|  | ||||
| func (l *noopLogger) Fields(_ ...interface{}) Logger { | ||||
| 	return l | ||||
| } | ||||
|  | ||||
| func (l *noopLogger) Options() Options { | ||||
| 	return l.opts | ||||
| } | ||||
|  | ||||
| func (l *noopLogger) String() string { | ||||
| 	return "noop" | ||||
| } | ||||
|  | ||||
| func (l *noopLogger) Log(ctx context.Context, lvl Level, msg string, attrs ...interface{}) { | ||||
| } | ||||
|  | ||||
| func (l *noopLogger) Info(ctx context.Context, msg string, attrs ...interface{}) { | ||||
| } | ||||
|  | ||||
| func (l *noopLogger) Debug(ctx context.Context, msg string, attrs ...interface{}) { | ||||
| } | ||||
|  | ||||
| func (l *noopLogger) Error(ctx context.Context, msg string, attrs ...interface{}) { | ||||
| } | ||||
|  | ||||
| func (l *noopLogger) Trace(ctx context.Context, msg string, attrs ...interface{}) { | ||||
| } | ||||
|  | ||||
| func (l *noopLogger) Warn(ctx context.Context, msg string, attrs ...interface{}) { | ||||
| } | ||||
|  | ||||
| func (l *noopLogger) Fatal(ctx context.Context, msg string, attrs ...interface{}) { | ||||
| } | ||||
| @@ -3,26 +3,55 @@ package logger | ||||
| import ( | ||||
| 	"context" | ||||
| 	"io" | ||||
| 	"log/slog" | ||||
| 	"os" | ||||
| 	"slices" | ||||
| 	"time" | ||||
|  | ||||
| 	"go.unistack.org/micro/v3/meter" | ||||
| ) | ||||
|  | ||||
| // Option func | ||||
| // Option func signature | ||||
| type Option func(*Options) | ||||
|  | ||||
| // Options holds logger options | ||||
| type Options struct { | ||||
| 	// TimeKey is the key used for the time of the log call | ||||
| 	TimeKey string | ||||
| 	// LevelKey is the key used for the level of the log call | ||||
| 	LevelKey string | ||||
| 	// ErroreKey is the key used for the error of the log call | ||||
| 	ErrorKey string | ||||
| 	// MessageKey is the key used for the message of the log call | ||||
| 	MessageKey string | ||||
| 	// SourceKey is the key used for the source file and line of the log call | ||||
| 	SourceKey string | ||||
| 	// StacktraceKey is the key used for the stacktrace | ||||
| 	StacktraceKey string | ||||
| 	// Name holds the logger name | ||||
| 	Name string | ||||
| 	// Out holds the output writer | ||||
| 	Out io.Writer | ||||
| 	// Context holds exernal options | ||||
| 	Context context.Context | ||||
| 	// Meter used to count logs for specific level | ||||
| 	Meter meter.Meter | ||||
| 	// TimeFunc used to obtain current time | ||||
| 	TimeFunc func() time.Time | ||||
| 	// Fields holds additional metadata | ||||
| 	Fields []interface{} | ||||
| 	// Name holds the logger name | ||||
| 	Name string | ||||
| 	// 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 | ||||
| 	// CallerSkipCount number of frmaes to skip | ||||
| 	CallerSkipCount int | ||||
| 	// 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 | ||||
| @@ -31,15 +60,58 @@ func NewOptions(opts ...Option) Options { | ||||
| 		Level:            DefaultLevel, | ||||
| 		Fields:           make([]interface{}, 0, 6), | ||||
| 		Out:              os.Stderr, | ||||
| 		CallerSkipCount: DefaultCallerSkipCount, | ||||
| 		Context:          context.Background(), | ||||
| 		ContextAttrFuncs: DefaultContextAttrFuncs, | ||||
| 		AddSource:        true, | ||||
| 		TimeFunc:         time.Now, | ||||
| 		Meter:            meter.DefaultMeter, | ||||
| 	} | ||||
|  | ||||
| 	WithMicroKeys()(&options) | ||||
|  | ||||
| 	for _, o := range opts { | ||||
| 		o(&options) | ||||
| 	} | ||||
|  | ||||
| 	return options | ||||
| } | ||||
|  | ||||
| // WithContextAttrFuncs appends default funcs for the context attrs filler | ||||
| func WithContextAttrFuncs(fncs ...ContextAttrFunc) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.ContextAttrFuncs = append(o.ContextAttrFuncs, fncs...) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // WithDedupKeys dont log duplicate keys | ||||
| func WithDedupKeys(b bool) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.DedupKeys = b | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // WithAddFields add fields for the logger | ||||
| func WithAddFields(fields ...interface{}) Option { | ||||
| 	return func(o *Options) { | ||||
| 		if o.DedupKeys { | ||||
| 			for i := 0; i < len(o.Fields); i += 2 { | ||||
| 				for j := 0; j < len(fields); j += 2 { | ||||
| 					iv, iok := o.Fields[i].(string) | ||||
| 					jv, jok := fields[j].(string) | ||||
| 					if iok && jok && iv == jv { | ||||
| 						fields = slices.Delete(fields, j, j+2) | ||||
| 					} | ||||
| 				} | ||||
| 			} | ||||
| 			if len(fields) > 0 { | ||||
| 				o.Fields = append(o.Fields, fields...) | ||||
| 			} | ||||
| 		} else { | ||||
| 			o.Fields = append(o.Fields, fields...) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // WithFields set default fields for the logger | ||||
| func WithFields(fields ...interface{}) Option { | ||||
| 	return func(o *Options) { | ||||
| @@ -61,10 +133,17 @@ func WithOutput(out io.Writer) Option { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // WithCallerSkipCount set frame count to skip | ||||
| func WithCallerSkipCount(c int) Option { | ||||
| // WithAddStacktrace controls writing stacktrace on error | ||||
| func WithAddStacktrace(v bool) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.CallerSkipCount = c | ||||
| 		o.AddStacktrace = v | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // WithAddSource controls writing source file and pos in log | ||||
| func WithAddSource(v bool) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.AddSource = v | ||||
| 	} | ||||
| } | ||||
|  | ||||
| @@ -81,3 +160,70 @@ func WithName(n string) Option { | ||||
| 		o.Name = n | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // WithMeter sets the meter | ||||
| func WithMeter(m meter.Meter) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.Meter = m | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // WithTimeFunc sets the func to obtain current time | ||||
| func WithTimeFunc(fn func() time.Time) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.TimeFunc = fn | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func WithZapKeys() Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.TimeKey = "@timestamp" | ||||
| 		o.LevelKey = slog.LevelKey | ||||
| 		o.MessageKey = slog.MessageKey | ||||
| 		o.SourceKey = "caller" | ||||
| 		o.StacktraceKey = "stacktrace" | ||||
| 		o.ErrorKey = "error" | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func WithZerologKeys() Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.TimeKey = slog.TimeKey | ||||
| 		o.LevelKey = slog.LevelKey | ||||
| 		o.MessageKey = "message" | ||||
| 		o.SourceKey = "caller" | ||||
| 		o.StacktraceKey = "stacktrace" | ||||
| 		o.ErrorKey = "error" | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func WithSlogKeys() Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.TimeKey = slog.TimeKey | ||||
| 		o.LevelKey = slog.LevelKey | ||||
| 		o.MessageKey = slog.MessageKey | ||||
| 		o.SourceKey = slog.SourceKey | ||||
| 		o.StacktraceKey = "stacktrace" | ||||
| 		o.ErrorKey = "error" | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func WithMicroKeys() Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.TimeKey = "timestamp" | ||||
| 		o.LevelKey = slog.LevelKey | ||||
| 		o.MessageKey = slog.MessageKey | ||||
| 		o.SourceKey = "caller" | ||||
| 		o.StacktraceKey = "stacktrace" | ||||
| 		o.ErrorKey = "error" | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // WithAddCallerSkipCount add skip count for copy logger | ||||
| func WithAddCallerSkipCount(n int) Option { | ||||
| 	return func(o *Options) { | ||||
| 		if n > 0 { | ||||
| 			o.CallerSkipCount += n | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|   | ||||
							
								
								
									
										373
									
								
								logger/slog/slog.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										373
									
								
								logger/slog/slog.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,373 @@ | ||||
| package slog | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"io" | ||||
| 	"log/slog" | ||||
| 	"os" | ||||
| 	"reflect" | ||||
| 	"regexp" | ||||
| 	"runtime" | ||||
| 	"strconv" | ||||
| 	"sync" | ||||
| 	"sync/atomic" | ||||
| 	"time" | ||||
|  | ||||
| 	"go.unistack.org/micro/v3/logger" | ||||
| 	"go.unistack.org/micro/v3/semconv" | ||||
| 	"go.unistack.org/micro/v3/tracer" | ||||
| ) | ||||
|  | ||||
| const ( | ||||
| 	badKey = "!BADKEY" | ||||
| 	// defaultCallerSkipCount used by logger | ||||
| 	defaultCallerSkipCount = 3 | ||||
| ) | ||||
|  | ||||
| var reTrace = regexp.MustCompile(`.*/slog/logger\.go.*\n`) | ||||
|  | ||||
| var ( | ||||
| 	traceValue = slog.StringValue("trace") | ||||
| 	debugValue = slog.StringValue("debug") | ||||
| 	infoValue  = slog.StringValue("info") | ||||
| 	warnValue  = slog.StringValue("warn") | ||||
| 	errorValue = slog.StringValue("error") | ||||
| 	fatalValue = slog.StringValue("fatal") | ||||
| ) | ||||
|  | ||||
| type wrapper struct { | ||||
| 	h     slog.Handler | ||||
| 	level atomic.Int64 | ||||
| } | ||||
|  | ||||
| func (h *wrapper) Enabled(ctx context.Context, level slog.Level) bool { | ||||
| 	return level >= slog.Level(int(h.level.Load())) | ||||
| } | ||||
|  | ||||
| func (h *wrapper) Handle(ctx context.Context, rec slog.Record) error { | ||||
| 	return h.h.Handle(ctx, rec) | ||||
| } | ||||
|  | ||||
| func (h *wrapper) WithAttrs(attrs []slog.Attr) slog.Handler { | ||||
| 	return h.h.WithAttrs(attrs) | ||||
| } | ||||
|  | ||||
| func (h *wrapper) WithGroup(name string) slog.Handler { | ||||
| 	return h.h.WithGroup(name) | ||||
| } | ||||
|  | ||||
| func (s *slogLogger) renameAttr(_ []string, a slog.Attr) slog.Attr { | ||||
| 	switch a.Key { | ||||
| 	case slog.SourceKey: | ||||
| 		source := a.Value.Any().(*slog.Source) | ||||
| 		a.Value = slog.StringValue(source.File + ":" + strconv.Itoa(source.Line)) | ||||
| 		a.Key = s.opts.SourceKey | ||||
| 	case slog.TimeKey: | ||||
| 		a.Key = s.opts.TimeKey | ||||
| 	case slog.MessageKey: | ||||
| 		a.Key = s.opts.MessageKey | ||||
| 	case slog.LevelKey: | ||||
| 		level := a.Value.Any().(slog.Level) | ||||
| 		lvl := slogToLoggerLevel(level) | ||||
| 		a.Key = s.opts.LevelKey | ||||
| 		switch { | ||||
| 		case lvl < logger.DebugLevel: | ||||
| 			a.Value = traceValue | ||||
| 		case lvl < logger.InfoLevel: | ||||
| 			a.Value = debugValue | ||||
| 		case lvl < logger.WarnLevel: | ||||
| 			a.Value = infoValue | ||||
| 		case lvl < logger.ErrorLevel: | ||||
| 			a.Value = warnValue | ||||
| 		case lvl < logger.FatalLevel: | ||||
| 			a.Value = errorValue | ||||
| 		case lvl >= logger.FatalLevel: | ||||
| 			a.Value = fatalValue | ||||
| 		default: | ||||
| 			a.Value = infoValue | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return a | ||||
| } | ||||
|  | ||||
| type slogLogger struct { | ||||
| 	handler *wrapper | ||||
| 	opts    logger.Options | ||||
| 	mu      sync.RWMutex | ||||
| } | ||||
|  | ||||
| func (s *slogLogger) Clone(opts ...logger.Option) logger.Logger { | ||||
| 	s.mu.RLock() | ||||
| 	options := s.opts | ||||
| 	s.mu.RUnlock() | ||||
|  | ||||
| 	for _, o := range opts { | ||||
| 		o(&options) | ||||
| 	} | ||||
|  | ||||
| 	if len(options.ContextAttrFuncs) == 0 { | ||||
| 		options.ContextAttrFuncs = logger.DefaultContextAttrFuncs | ||||
| 	} | ||||
|  | ||||
| 	attrs, _ := s.argsAttrs(options.Fields) | ||||
| 	l := &slogLogger{ | ||||
| 		handler: &wrapper{h: s.handler.h.WithAttrs(attrs)}, | ||||
| 		opts:    options, | ||||
| 	} | ||||
| 	l.handler.level.Store(int64(loggerToSlogLevel(options.Level))) | ||||
|  | ||||
| 	return l | ||||
| } | ||||
|  | ||||
| func (s *slogLogger) V(level logger.Level) bool { | ||||
| 	s.mu.Lock() | ||||
| 	v := s.opts.Level.Enabled(level) | ||||
| 	s.mu.Unlock() | ||||
| 	return v | ||||
| } | ||||
|  | ||||
| func (s *slogLogger) Level(level logger.Level) { | ||||
| 	s.mu.Lock() | ||||
| 	s.opts.Level = level | ||||
| 	s.handler.level.Store(int64(loggerToSlogLevel(level))) | ||||
| 	s.mu.Unlock() | ||||
| } | ||||
|  | ||||
| func (s *slogLogger) Options() logger.Options { | ||||
| 	return s.opts | ||||
| } | ||||
|  | ||||
| func (s *slogLogger) Fields(fields ...interface{}) logger.Logger { | ||||
| 	s.mu.RLock() | ||||
| 	options := s.opts | ||||
| 	s.mu.RUnlock() | ||||
|  | ||||
| 	l := &slogLogger{opts: options} | ||||
| 	logger.WithAddFields(fields...)(&l.opts) | ||||
|  | ||||
| 	if len(options.ContextAttrFuncs) == 0 { | ||||
| 		options.ContextAttrFuncs = logger.DefaultContextAttrFuncs | ||||
| 	} | ||||
|  | ||||
| 	attrs, _ := s.argsAttrs(fields) | ||||
| 	l.handler = &wrapper{h: s.handler.h.WithAttrs(attrs)} | ||||
| 	l.handler.level.Store(int64(loggerToSlogLevel(l.opts.Level))) | ||||
|  | ||||
| 	return l | ||||
| } | ||||
|  | ||||
| func (s *slogLogger) Init(opts ...logger.Option) error { | ||||
| 	s.mu.Lock() | ||||
|  | ||||
| 	for _, o := range opts { | ||||
| 		o(&s.opts) | ||||
| 	} | ||||
|  | ||||
| 	if len(s.opts.ContextAttrFuncs) == 0 { | ||||
| 		s.opts.ContextAttrFuncs = logger.DefaultContextAttrFuncs | ||||
| 	} | ||||
|  | ||||
| 	handleOpt := &slog.HandlerOptions{ | ||||
| 		ReplaceAttr: s.renameAttr, | ||||
| 		Level:       loggerToSlogLevel(logger.TraceLevel), | ||||
| 		AddSource:   s.opts.AddSource, | ||||
| 	} | ||||
|  | ||||
| 	attrs, _ := s.argsAttrs(s.opts.Fields) | ||||
|  | ||||
| 	var h slog.Handler | ||||
| 	if s.opts.Context != nil { | ||||
| 		if v, ok := s.opts.Context.Value(handlerKey{}).(slog.Handler); ok && v != nil { | ||||
| 			h = v | ||||
| 		} | ||||
|  | ||||
| 		if fn := s.opts.Context.Value(handlerFnKey{}); fn != nil { | ||||
| 			if rfn := reflect.ValueOf(fn); rfn.Kind() == reflect.Func { | ||||
| 				if ret := rfn.Call([]reflect.Value{reflect.ValueOf(s.opts.Out), reflect.ValueOf(handleOpt)}); len(ret) == 1 { | ||||
| 					if iface, ok := ret[0].Interface().(slog.Handler); ok && iface != nil { | ||||
| 						h = iface | ||||
| 					} | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	if h == nil { | ||||
| 		h = slog.NewJSONHandler(s.opts.Out, handleOpt) | ||||
| 	} | ||||
|  | ||||
| 	s.handler = &wrapper{h: h.WithAttrs(attrs)} | ||||
| 	s.handler.level.Store(int64(loggerToSlogLevel(s.opts.Level))) | ||||
| 	s.mu.Unlock() | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (s *slogLogger) Log(ctx context.Context, lvl logger.Level, msg string, attrs ...interface{}) { | ||||
| 	s.printLog(ctx, lvl, msg, attrs...) | ||||
| } | ||||
|  | ||||
| func (s *slogLogger) Info(ctx context.Context, msg string, attrs ...interface{}) { | ||||
| 	s.printLog(ctx, logger.InfoLevel, msg, attrs...) | ||||
| } | ||||
|  | ||||
| func (s *slogLogger) Debug(ctx context.Context, msg string, attrs ...interface{}) { | ||||
| 	s.printLog(ctx, logger.DebugLevel, msg, attrs...) | ||||
| } | ||||
|  | ||||
| func (s *slogLogger) Trace(ctx context.Context, msg string, attrs ...interface{}) { | ||||
| 	s.printLog(ctx, logger.TraceLevel, msg, attrs...) | ||||
| } | ||||
|  | ||||
| func (s *slogLogger) Error(ctx context.Context, msg string, attrs ...interface{}) { | ||||
| 	s.printLog(ctx, logger.ErrorLevel, msg, attrs...) | ||||
| } | ||||
|  | ||||
| func (s *slogLogger) Fatal(ctx context.Context, msg string, attrs ...interface{}) { | ||||
| 	s.printLog(ctx, logger.FatalLevel, msg, attrs...) | ||||
| 	if closer, ok := s.opts.Out.(io.Closer); ok { | ||||
| 		closer.Close() | ||||
| 	} | ||||
| 	time.Sleep(1 * time.Second) | ||||
| 	os.Exit(1) | ||||
| } | ||||
|  | ||||
| func (s *slogLogger) Warn(ctx context.Context, msg string, attrs ...interface{}) { | ||||
| 	s.printLog(ctx, logger.WarnLevel, msg, attrs...) | ||||
| } | ||||
|  | ||||
| func (s *slogLogger) Name() string { | ||||
| 	return s.opts.Name | ||||
| } | ||||
|  | ||||
| func (s *slogLogger) String() string { | ||||
| 	return "slog" | ||||
| } | ||||
|  | ||||
| func (s *slogLogger) printLog(ctx context.Context, lvl logger.Level, msg string, args ...interface{}) { | ||||
| 	if !s.V(lvl) { | ||||
| 		return | ||||
| 	} | ||||
| 	var argError error | ||||
|  | ||||
| 	s.opts.Meter.Counter(semconv.LoggerMessageTotal, "level", lvl.String()).Inc() | ||||
|  | ||||
| 	attrs, err := s.argsAttrs(args) | ||||
| 	if err != nil { | ||||
| 		argError = err | ||||
| 	} | ||||
| 	if argError != nil { | ||||
| 		if span, ok := tracer.SpanFromContext(ctx); ok { | ||||
| 			span.SetStatus(tracer.SpanStatusError, argError.Error()) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	for _, fn := range s.opts.ContextAttrFuncs { | ||||
| 		ctxAttrs, err := s.argsAttrs(fn(ctx)) | ||||
| 		if err != nil { | ||||
| 			argError = err | ||||
| 		} | ||||
| 		attrs = append(attrs, ctxAttrs...) | ||||
| 	} | ||||
| 	if argError != nil { | ||||
| 		if span, ok := tracer.SpanFromContext(ctx); ok { | ||||
| 			span.SetStatus(tracer.SpanStatusError, argError.Error()) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	if (s.opts.AddStacktrace || lvl == logger.FatalLevel) || (s.opts.AddStacktrace && lvl == logger.ErrorLevel) { | ||||
| 		stackInfo := make([]byte, 1024*1024) | ||||
| 		if stackSize := runtime.Stack(stackInfo, false); stackSize > 0 { | ||||
| 			traceLines := reTrace.Split(string(stackInfo[:stackSize]), -1) | ||||
| 			if len(traceLines) != 0 { | ||||
| 				attrs = append(attrs, slog.String(s.opts.StacktraceKey, traceLines[len(traceLines)-1])) | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	var pcs [1]uintptr | ||||
| 	runtime.Callers(s.opts.CallerSkipCount, pcs[:]) // skip [Callers, printLog, LogLvlMethod] | ||||
| 	r := slog.NewRecord(s.opts.TimeFunc(), loggerToSlogLevel(lvl), msg, pcs[0]) | ||||
| 	r.AddAttrs(attrs...) | ||||
| 	_ = s.handler.Handle(ctx, r) | ||||
| } | ||||
|  | ||||
| func NewLogger(opts ...logger.Option) logger.Logger { | ||||
| 	s := &slogLogger{ | ||||
| 		opts: logger.NewOptions(opts...), | ||||
| 	} | ||||
| 	s.opts.CallerSkipCount = defaultCallerSkipCount | ||||
|  | ||||
| 	return s | ||||
| } | ||||
|  | ||||
| func loggerToSlogLevel(level logger.Level) slog.Level { | ||||
| 	switch level { | ||||
| 	case logger.DebugLevel: | ||||
| 		return slog.LevelDebug | ||||
| 	case logger.WarnLevel: | ||||
| 		return slog.LevelWarn | ||||
| 	case logger.ErrorLevel: | ||||
| 		return slog.LevelError | ||||
| 	case logger.TraceLevel: | ||||
| 		return slog.LevelDebug - 1 | ||||
| 	case logger.FatalLevel: | ||||
| 		return slog.LevelError + 1 | ||||
| 	default: | ||||
| 		return slog.LevelInfo | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func slogToLoggerLevel(level slog.Level) logger.Level { | ||||
| 	switch level { | ||||
| 	case slog.LevelDebug: | ||||
| 		return logger.DebugLevel | ||||
| 	case slog.LevelWarn: | ||||
| 		return logger.WarnLevel | ||||
| 	case slog.LevelError: | ||||
| 		return logger.ErrorLevel | ||||
| 	case slog.LevelDebug - 1: | ||||
| 		return logger.TraceLevel | ||||
| 	case slog.LevelError + 1: | ||||
| 		return logger.FatalLevel | ||||
| 	default: | ||||
| 		return logger.InfoLevel | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (s *slogLogger) argsAttrs(args []interface{}) ([]slog.Attr, error) { | ||||
| 	attrs := make([]slog.Attr, 0, len(args)) | ||||
| 	var err error | ||||
|  | ||||
| 	for idx := 0; idx < len(args); idx++ { | ||||
| 		switch arg := args[idx].(type) { | ||||
| 		case slog.Attr: | ||||
| 			attrs = append(attrs, arg) | ||||
| 		case string: | ||||
| 			if idx+1 < len(args) { | ||||
| 				attrs = append(attrs, slog.Any(arg, args[idx+1])) | ||||
| 				idx++ | ||||
| 			} else { | ||||
| 				attrs = append(attrs, slog.String(badKey, arg)) | ||||
| 			} | ||||
| 		case error: | ||||
| 			attrs = append(attrs, slog.String(s.opts.ErrorKey, arg.Error())) | ||||
| 			err = arg | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return attrs, err | ||||
| } | ||||
|  | ||||
| type handlerKey struct{} | ||||
|  | ||||
| func WithHandler(h slog.Handler) logger.Option { | ||||
| 	return logger.SetOption(handlerKey{}, h) | ||||
| } | ||||
|  | ||||
| type handlerFnKey struct{} | ||||
|  | ||||
| func WithHandlerFunc(fn any) logger.Option { | ||||
| 	return logger.SetOption(handlerFnKey{}, fn) | ||||
| } | ||||
							
								
								
									
										409
									
								
								logger/slog/slog_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										409
									
								
								logger/slog/slog_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,409 @@ | ||||
| package slog | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"context" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"log" | ||||
| 	"log/slog" | ||||
| 	"strings" | ||||
| 	"testing" | ||||
|  | ||||
| 	"github.com/google/uuid" | ||||
| 	"go.unistack.org/micro/v3/logger" | ||||
| 	"go.unistack.org/micro/v3/metadata" | ||||
| ) | ||||
|  | ||||
| func TestStacktrace(t *testing.T) { | ||||
| 	ctx := context.TODO() | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	l := NewLogger(logger.WithLevel(logger.ErrorLevel), logger.WithOutput(buf), | ||||
| 		WithHandlerFunc(slog.NewTextHandler), | ||||
| 		logger.WithAddStacktrace(true), | ||||
| 	) | ||||
| 	if err := l.Init(logger.WithFields("key1", "val1")); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	l.Error(ctx, "msg1", errors.New("err")) | ||||
|  | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`slog_test.go:29`)) { | ||||
| 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestWithFields(t *testing.T) { | ||||
| 	ctx := context.TODO() | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	l := NewLogger(logger.WithLevel(logger.InfoLevel), logger.WithOutput(buf), | ||||
| 		WithHandlerFunc(slog.NewTextHandler), | ||||
| 		logger.WithDedupKeys(true), | ||||
| 	) | ||||
| 	if err := l.Init(logger.WithFields("key1", "val1")); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	l.Info(ctx, "msg1") | ||||
|  | ||||
| 	l = l.Fields("key1", "val2") | ||||
|  | ||||
| 	l.Info(ctx, "msg2") | ||||
|  | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`msg=msg2 key1=val1`)) { | ||||
| 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestWithDedupKeysWithAddFields(t *testing.T) { | ||||
| 	ctx := context.TODO() | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	l := NewLogger(logger.WithLevel(logger.InfoLevel), logger.WithOutput(buf), | ||||
| 		WithHandlerFunc(slog.NewTextHandler), | ||||
| 		logger.WithDedupKeys(true), | ||||
| 	) | ||||
| 	if err := l.Init(logger.WithFields("key1", "val1")); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	l.Info(ctx, "msg1") | ||||
|  | ||||
| 	if err := l.Init(logger.WithAddFields("key2", "val2")); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	l.Info(ctx, "msg2") | ||||
|  | ||||
| 	if err := l.Init(logger.WithAddFields("key2", "val3", "key1", "val4")); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	l.Info(ctx, "msg3") | ||||
|  | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`msg=msg3 key1=val1 key2=val2`)) { | ||||
| 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestWithHandlerFunc(t *testing.T) { | ||||
| 	ctx := context.TODO() | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	l := NewLogger(logger.WithLevel(logger.InfoLevel), logger.WithOutput(buf), | ||||
| 		WithHandlerFunc(slog.NewTextHandler), | ||||
| 	) | ||||
| 	if err := l.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	l.Info(ctx, "msg1") | ||||
|  | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`msg=msg1`)) { | ||||
| 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestWithAddFields(t *testing.T) { | ||||
| 	ctx := context.TODO() | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	l := NewLogger(logger.WithLevel(logger.InfoLevel), logger.WithOutput(buf)) | ||||
| 	if err := l.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	l.Info(ctx, "msg1") | ||||
|  | ||||
| 	if err := l.Init(logger.WithAddFields("key1", "val1")); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	l.Info(ctx, "msg2") | ||||
|  | ||||
| 	if err := l.Init(logger.WithAddFields("key2", "val2")); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	l.Info(ctx, "msg3") | ||||
|  | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"key1"`)) { | ||||
| 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"key2"`)) { | ||||
| 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestMultipleFieldsWithLevel(t *testing.T) { | ||||
| 	ctx := context.TODO() | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	l := NewLogger(logger.WithLevel(logger.InfoLevel), logger.WithOutput(buf)) | ||||
| 	if err := l.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	l = l.Fields("key", "val") | ||||
|  | ||||
| 	l.Info(ctx, "msg1") | ||||
| 	nl := l.Clone(logger.WithLevel(logger.DebugLevel)) | ||||
| 	nl.Debug(ctx, "msg2") | ||||
| 	l.Debug(ctx, "msg3") | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"key":"val"`)) { | ||||
| 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"msg1"`)) { | ||||
| 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"msg2"`)) { | ||||
| 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
| 	if bytes.Contains(buf.Bytes(), []byte(`"msg3"`)) { | ||||
| 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestMultipleFields(t *testing.T) { | ||||
| 	ctx := context.TODO() | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	l := NewLogger(logger.WithLevel(logger.InfoLevel), logger.WithOutput(buf)) | ||||
| 	if err := l.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	l = l.Fields("key", "val") | ||||
|  | ||||
| 	l = l.Fields("key1", "val1") | ||||
|  | ||||
| 	l.Info(ctx, "msg") | ||||
|  | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"key":"val"`)) { | ||||
| 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"key1":"val1"`)) { | ||||
| 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestError(t *testing.T) { | ||||
| 	ctx := context.TODO() | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	l := NewLogger(logger.WithLevel(logger.ErrorLevel), logger.WithOutput(buf), logger.WithAddStacktrace(true)) | ||||
| 	if err := l.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	l.Error(ctx, "message", fmt.Errorf("error message")) | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"stacktrace":"`)) { | ||||
| 		t.Fatalf("logger stacktrace not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"error":"`)) { | ||||
| 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestErrorf(t *testing.T) { | ||||
| 	ctx := context.TODO() | ||||
|  | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	l := NewLogger(logger.WithLevel(logger.ErrorLevel), logger.WithOutput(buf), logger.WithAddStacktrace(true)) | ||||
| 	if err := l.Init(logger.WithContextAttrFuncs(func(_ context.Context) []interface{} { | ||||
| 		return nil | ||||
| 	})); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	l.Log(ctx, logger.ErrorLevel, "message", errors.New("error msg")) | ||||
|  | ||||
| 	l.Log(ctx, logger.ErrorLevel, "", errors.New("error msg")) | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"error":"error msg"`)) { | ||||
| 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
|  | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"stacktrace":"`)) { | ||||
| 		t.Fatalf("logger stacktrace not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"error":"`)) { | ||||
| 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestContext(t *testing.T) { | ||||
| 	ctx := context.TODO() | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	l := NewLogger(logger.WithLevel(logger.TraceLevel), logger.WithOutput(buf)) | ||||
| 	if err := l.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	nl, ok := logger.FromContext(logger.NewContext(ctx, l.Fields("key", "val"))) | ||||
| 	if !ok { | ||||
| 		t.Fatal("context without logger") | ||||
| 	} | ||||
| 	nl.Info(ctx, "message") | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"key":"val"`)) { | ||||
| 		t.Fatalf("logger fields not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestFields(t *testing.T) { | ||||
| 	ctx := context.TODO() | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	l := NewLogger(logger.WithLevel(logger.TraceLevel), logger.WithOutput(buf)) | ||||
| 	if err := l.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	nl := l.Fields("key", "val") | ||||
|  | ||||
| 	nl.Info(ctx, "message") | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"key":"val"`)) { | ||||
| 		t.Fatalf("logger fields not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestFromContextWithFields(t *testing.T) { | ||||
| 	ctx := context.TODO() | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	var ok bool | ||||
| 	l := NewLogger(logger.WithLevel(logger.TraceLevel), logger.WithOutput(buf)) | ||||
| 	if err := l.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	nl := l.Fields("key", "val") | ||||
|  | ||||
| 	ctx = logger.NewContext(ctx, nl) | ||||
|  | ||||
| 	l, ok = logger.FromContext(ctx) | ||||
| 	if !ok { | ||||
| 		t.Fatalf("context does not have logger") | ||||
| 	} | ||||
|  | ||||
| 	l.Info(ctx, "message") | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"key":"val"`)) { | ||||
| 		t.Fatalf("logger fields not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
|  | ||||
| 	l.Info(ctx, "test", "uncorrected number attributes") | ||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`"!BADKEY":"uncorrected number attributes"`)) { | ||||
| 		t.Fatalf("logger fields not works, buf contains: %s", buf.Bytes()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestClone(t *testing.T) { | ||||
| 	ctx := context.TODO() | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	l := NewLogger(logger.WithLevel(logger.TraceLevel), logger.WithOutput(buf)) | ||||
| 	if err := l.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	nl := l.Clone(logger.WithLevel(logger.ErrorLevel)) | ||||
| 	if err := nl.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	nl.Info(ctx, "info message") | ||||
| 	if len(buf.Bytes()) != 0 { | ||||
| 		t.Fatal("message must not be logged") | ||||
| 	} | ||||
| 	l.Info(ctx, "info message") | ||||
| 	if len(buf.Bytes()) == 0 { | ||||
| 		t.Fatal("message must be logged") | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestRedirectStdLogger(t *testing.T) { | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	l := NewLogger(logger.WithLevel(logger.ErrorLevel), logger.WithOutput(buf)) | ||||
| 	if err := l.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	fn := logger.RedirectStdLogger(l, logger.ErrorLevel) | ||||
| 	defer fn() | ||||
| 	log.Print("test") | ||||
| 	if !(bytes.Contains(buf.Bytes(), []byte(`"level":"error"`)) && bytes.Contains(buf.Bytes(), []byte(`"msg":"test"`))) { | ||||
| 		t.Fatalf("logger error, buf %s", buf.Bytes()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestStdLogger(t *testing.T) { | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	l := NewLogger(logger.WithLevel(logger.TraceLevel), logger.WithOutput(buf)) | ||||
| 	if err := l.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	lg := logger.NewStdLogger(l, logger.ErrorLevel) | ||||
| 	lg.Print("test") | ||||
| 	if !(bytes.Contains(buf.Bytes(), []byte(`"level":"error"`)) && bytes.Contains(buf.Bytes(), []byte(`"msg":"test"`))) { | ||||
| 		t.Fatalf("logger error, buf %s", buf.Bytes()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestLogger(t *testing.T) { | ||||
| 	ctx := context.TODO() | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	l := NewLogger(logger.WithLevel(logger.TraceLevel), logger.WithOutput(buf)) | ||||
| 	if err := l.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	l.Trace(ctx, "trace_msg1") | ||||
| 	l.Warn(ctx, "warn_msg1") | ||||
| 	l.Fields("error", "test").Info(ctx, "error message") | ||||
| 	l.Warn(ctx, "first second") | ||||
|  | ||||
| 	if !(bytes.Contains(buf.Bytes(), []byte(`"level":"trace"`)) && bytes.Contains(buf.Bytes(), []byte(`"msg":"trace_msg1"`))) { | ||||
| 		t.Fatalf("logger tracer, buf %s", buf.Bytes()) | ||||
| 	} | ||||
| 	if !(bytes.Contains(buf.Bytes(), []byte(`"level":"warn"`)) && bytes.Contains(buf.Bytes(), []byte(`"msg":"warn_msg1"`))) { | ||||
| 		t.Fatalf("logger warn, buf %s", buf.Bytes()) | ||||
| 	} | ||||
| 	if !(bytes.Contains(buf.Bytes(), []byte(`"level":"info"`)) && bytes.Contains(buf.Bytes(), []byte(`"msg":"error message","error":"test"`))) { | ||||
| 		t.Fatalf("logger info, buf %s", buf.Bytes()) | ||||
| 	} | ||||
| 	if !(bytes.Contains(buf.Bytes(), []byte(`"level":"warn"`)) && bytes.Contains(buf.Bytes(), []byte(`"msg":"first second"`))) { | ||||
| 		t.Fatalf("logger warn, buf %s", buf.Bytes()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func Test_WithContextAttrFunc(t *testing.T) { | ||||
| 	loggerContextAttrFuncs := []logger.ContextAttrFunc{ | ||||
| 		func(ctx context.Context) []interface{} { | ||||
| 			md, ok := metadata.FromIncomingContext(ctx) | ||||
| 			if !ok { | ||||
| 				return nil | ||||
| 			} | ||||
| 			attrs := make([]interface{}, 0, 10) | ||||
| 			for k, v := range md { | ||||
| 				switch k { | ||||
| 				case "X-Request-Id", "Phone", "External-Id", "Source-Service", "X-App-Install-Id", "Client-Id", "Client-Ip": | ||||
| 					attrs = append(attrs, strings.ToLower(k), v) | ||||
| 				} | ||||
| 			} | ||||
| 			return attrs | ||||
| 		}, | ||||
| 	} | ||||
|  | ||||
| 	logger.DefaultContextAttrFuncs = append(logger.DefaultContextAttrFuncs, loggerContextAttrFuncs...) | ||||
|  | ||||
| 	ctx := context.TODO() | ||||
| 	ctx = metadata.AppendIncomingContext(ctx, "X-Request-Id", uuid.New().String(), | ||||
| 		"Source-Service", "Test-System") | ||||
|  | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	l := NewLogger(logger.WithLevel(logger.TraceLevel), logger.WithOutput(buf)) | ||||
| 	if err := l.Init(); err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	l.Info(ctx, "test message") | ||||
| 	if !(bytes.Contains(buf.Bytes(), []byte(`"level":"info"`)) && bytes.Contains(buf.Bytes(), []byte(`"msg":"test message"`))) { | ||||
| 		t.Fatalf("logger info, buf %s", buf.Bytes()) | ||||
| 	} | ||||
| 	if !(bytes.Contains(buf.Bytes(), []byte(`"x-request-id":"`))) { | ||||
| 		t.Fatalf("logger info, buf %s", buf.Bytes()) | ||||
| 	} | ||||
| 	if !(bytes.Contains(buf.Bytes(), []byte(`"source-service":"Test-System"`))) { | ||||
| 		t.Fatalf("logger info, buf %s", buf.Bytes()) | ||||
| 	} | ||||
| 	buf.Reset() | ||||
| 	imd, _ := metadata.FromIncomingContext(ctx) | ||||
| 	l.Info(ctx, "test message1") | ||||
| 	imd.Set("Source-Service", "Test-System2") | ||||
| 	l.Info(ctx, "test message2") | ||||
|  | ||||
| 	// t.Logf("xxx %s", buf.Bytes()) | ||||
| } | ||||
| @@ -36,8 +36,8 @@ var ( | ||||
| 	circularShortBytes = []byte("<shown>") | ||||
| 	invalidAngleBytes  = []byte("<invalid>") | ||||
| 	filteredBytes      = []byte("<filtered>") | ||||
| 	openBracketBytes   = []byte("[") | ||||
| 	closeBracketBytes  = []byte("]") | ||||
| 	// openBracketBytes   = []byte("[") | ||||
| 	// closeBracketBytes  = []byte("]") | ||||
| 	percentBytes    = []byte("%") | ||||
| 	precisionBytes  = []byte(".") | ||||
| 	openAngleBytes  = []byte("<") | ||||
| @@ -46,14 +46,21 @@ var ( | ||||
| 	closeMapBytes   = []byte("}") | ||||
| ) | ||||
|  | ||||
| type protoMessage interface { | ||||
| 	Reset() | ||||
| 	ProtoMessage() | ||||
| } | ||||
|  | ||||
| type Wrapper struct { | ||||
| 	pointers map[uintptr]int | ||||
| 	takeMap  map[int]bool | ||||
|  | ||||
| 	val  interface{} | ||||
| 	s    fmt.State | ||||
| 	pointers         map[uintptr]int | ||||
| 	opts *Options | ||||
|  | ||||
| 	depth            int | ||||
| 	ignoreNextType   bool | ||||
| 	takeAll          map[int]bool | ||||
| 	protoWrapperType bool | ||||
| 	sqlWrapperType   bool | ||||
| } | ||||
| @@ -111,7 +118,7 @@ func Tagged(b bool) Option { | ||||
|  | ||||
| func Unwrap(val interface{}, opts ...Option) *Wrapper { | ||||
| 	options := NewOptions(opts...) | ||||
| 	return &Wrapper{val: val, opts: &options, pointers: make(map[uintptr]int), takeAll: make(map[int]bool)} | ||||
| 	return &Wrapper{val: val, opts: &options, pointers: make(map[uintptr]int), takeMap: make(map[int]bool)} | ||||
| } | ||||
|  | ||||
| func (w *Wrapper) unpackValue(v reflect.Value) reflect.Value { | ||||
| @@ -237,9 +244,6 @@ func (w *Wrapper) format(v reflect.Value) { | ||||
| 		_, _ = w.s.Write(buf) | ||||
| 		return | ||||
| 	} | ||||
| 	if w.opts.Tagged { | ||||
| 		w.checkTakeAll(v, 1) | ||||
| 	} | ||||
|  | ||||
| 	// Handle invalid reflect values immediately. | ||||
| 	kind := v.Kind() | ||||
| @@ -256,6 +260,10 @@ func (w *Wrapper) format(v reflect.Value) { | ||||
| 				w.protoWrapperType = true | ||||
| 			} else if strings.HasPrefix(reflect.Indirect(v).Type().String(), "sql.Null") { | ||||
| 				w.sqlWrapperType = true | ||||
| 			} else if v.CanInterface() { | ||||
| 				if _, ok := v.Interface().(protoMessage); ok { | ||||
| 					w.protoWrapperType = true | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 		w.formatPtr(v) | ||||
| @@ -378,6 +386,12 @@ func (w *Wrapper) format(v reflect.Value) { | ||||
| 		prevSkip := false | ||||
|  | ||||
| 		for i := 0; i < numFields; i++ { | ||||
| 			switch vt.Field(i).Type.PkgPath() { | ||||
| 			case "google.golang.org/protobuf/internal/impl", "google.golang.org/protobuf/internal/pragma": | ||||
| 				w.protoWrapperType = true | ||||
| 				prevSkip = true | ||||
| 				continue | ||||
| 			} | ||||
| 			if w.protoWrapperType && !vt.Field(i).IsExported() { | ||||
| 				prevSkip = true | ||||
| 				continue | ||||
| @@ -385,6 +399,9 @@ func (w *Wrapper) format(v reflect.Value) { | ||||
| 				prevSkip = true | ||||
| 				continue | ||||
| 			} | ||||
| 			if _, ok := vt.Field(i).Tag.Lookup("protobuf"); ok && !w.protoWrapperType { | ||||
| 				w.protoWrapperType = true | ||||
| 			} | ||||
| 			sv, ok := vt.Field(i).Tag.Lookup("logger") | ||||
| 			switch { | ||||
| 			case ok: | ||||
| @@ -395,12 +412,17 @@ func (w *Wrapper) format(v reflect.Value) { | ||||
| 				case "take": | ||||
| 					break | ||||
| 				} | ||||
| 			case w.takeAll[w.depth]: | ||||
| 				break | ||||
| 			case !ok && w.opts.Tagged: | ||||
| 				// skip top level untagged | ||||
| 				if w.depth == 1 { | ||||
| 					prevSkip = true | ||||
| 					continue | ||||
| 				} | ||||
| 				if tv, ok := w.takeMap[w.depth]; ok && !tv { | ||||
| 					prevSkip = true | ||||
| 					continue | ||||
| 				} | ||||
| 			} | ||||
|  | ||||
| 			if prevSkip { | ||||
| 				prevSkip = false | ||||
| @@ -416,9 +438,7 @@ func (w *Wrapper) format(v reflect.Value) { | ||||
| 				_, _ = w.s.Write([]byte(vt.Name)) | ||||
| 				_, _ = w.s.Write(colonBytes) | ||||
| 			} | ||||
| 			unpackValue := w.unpackValue(v.Field(i)) | ||||
| 			w.checkTakeAll(unpackValue, w.depth) | ||||
| 			w.format(unpackValue) | ||||
| 			w.format(w.unpackValue(v.Field(i))) | ||||
| 			numWritten++ | ||||
| 		} | ||||
| 		w.depth-- | ||||
| @@ -461,6 +481,10 @@ func (w *Wrapper) Format(s fmt.State, verb rune) { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	if w.opts.Tagged { | ||||
| 		w.buildTakeMap(reflect.ValueOf(w.val), 1) | ||||
| 	} | ||||
|  | ||||
| 	w.format(reflect.ValueOf(w.val)) | ||||
| } | ||||
|  | ||||
| @@ -615,24 +639,28 @@ func (w *Wrapper) constructOrigFormat(verb rune) string { | ||||
| 	return buf.String() | ||||
| } | ||||
|  | ||||
| func (w *Wrapper) checkTakeAll(v reflect.Value, depth int) { | ||||
| 	if _, ok := w.takeAll[depth]; ok { | ||||
| 		return | ||||
| 	} | ||||
| func (w *Wrapper) buildTakeMap(v reflect.Value, depth int) { | ||||
| 	if !v.IsValid() || v.IsZero() { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	switch v.Kind() { | ||||
| 	case reflect.Slice, reflect.Array: | ||||
| 		for i := 0; i < v.Len(); i++ { | ||||
| 			w.buildTakeMap(v.Index(i), depth+1) | ||||
| 		} | ||||
| 		w.takeMap[depth] = true | ||||
| 		return | ||||
| 	case reflect.Struct: | ||||
| 		break | ||||
| 	case reflect.Ptr: | ||||
| 		v = v.Elem() | ||||
| 		if v.Kind() != reflect.Struct { | ||||
| 			w.takeAll[depth] = true | ||||
| 			w.takeMap[depth] = true | ||||
| 			return | ||||
| 		} | ||||
| 	default: | ||||
| 		w.takeAll[depth] = true | ||||
| 		w.takeMap[depth] = true | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| @@ -641,8 +669,15 @@ func (w *Wrapper) checkTakeAll(v reflect.Value, depth int) { | ||||
| 	for i := 0; i < v.NumField(); i++ { | ||||
| 		sv, ok := vt.Field(i).Tag.Lookup("logger") | ||||
| 		if ok && sv == "take" { | ||||
| 			w.takeAll[depth] = false | ||||
| 			w.takeMap[depth] = false | ||||
| 		} | ||||
| 		w.checkTakeAll(v.Field(i), depth+1) | ||||
| 		if v.Kind() == reflect.Struct || | ||||
| 			(v.Kind() == reflect.Ptr && v.Elem().Kind() == reflect.Struct) { | ||||
| 			w.buildTakeMap(v.Field(i), depth+1) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	if _, ok := w.takeMap[depth]; !ok { | ||||
| 		w.takeMap[depth] = true | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -82,11 +82,11 @@ func TestTagged(t *testing.T) { | ||||
| func TestTaggedNested(t *testing.T) { | ||||
| 	type val struct { | ||||
| 		key string `logger:"take"` | ||||
| 		val string `logger:"omit"` | ||||
| 		// val string `logger:"omit"` | ||||
| 		unk string | ||||
| 	} | ||||
| 	type str struct { | ||||
| 		key string `logger:"omit"` | ||||
| 		// key string `logger:"omit"` | ||||
| 		val *val `logger:"take"` | ||||
| 	} | ||||
|  | ||||
|   | ||||
| @@ -1,399 +0,0 @@ | ||||
| // Package wrapper provides wrapper for Logger | ||||
| package wrapper | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"fmt" | ||||
|  | ||||
| 	"go.unistack.org/micro/v3/client" | ||||
| 	"go.unistack.org/micro/v3/logger" | ||||
| 	"go.unistack.org/micro/v3/server" | ||||
| ) | ||||
|  | ||||
| var ( | ||||
| 	// DefaultClientCallObserver called by wrapper in client Call | ||||
| 	DefaultClientCallObserver = func(ctx context.Context, req client.Request, rsp interface{}, opts []client.CallOption, err error) []string { | ||||
| 		labels := []string{"service", req.Service(), "endpoint", req.Endpoint()} | ||||
| 		if err != nil { | ||||
| 			labels = append(labels, "error", err.Error()) | ||||
| 		} | ||||
| 		return labels | ||||
| 	} | ||||
|  | ||||
| 	// DefaultClientStreamObserver called by wrapper in client Stream | ||||
| 	DefaultClientStreamObserver = func(ctx context.Context, req client.Request, opts []client.CallOption, stream client.Stream, err error) []string { | ||||
| 		labels := []string{"service", req.Service(), "endpoint", req.Endpoint()} | ||||
| 		if err != nil { | ||||
| 			labels = append(labels, "error", err.Error()) | ||||
| 		} | ||||
| 		return labels | ||||
| 	} | ||||
|  | ||||
| 	// DefaultClientPublishObserver called by wrapper in client Publish | ||||
| 	DefaultClientPublishObserver = func(ctx context.Context, msg client.Message, opts []client.PublishOption, err error) []string { | ||||
| 		labels := []string{"endpoint", msg.Topic()} | ||||
| 		if err != nil { | ||||
| 			labels = append(labels, "error", err.Error()) | ||||
| 		} | ||||
| 		return labels | ||||
| 	} | ||||
|  | ||||
| 	// DefaultServerHandlerObserver called by wrapper in server Handler | ||||
| 	DefaultServerHandlerObserver = func(ctx context.Context, req server.Request, rsp interface{}, err error) []string { | ||||
| 		labels := []string{"service", req.Service(), "endpoint", req.Endpoint()} | ||||
| 		if err != nil { | ||||
| 			labels = append(labels, "error", err.Error()) | ||||
| 		} | ||||
| 		return labels | ||||
| 	} | ||||
|  | ||||
| 	// DefaultServerSubscriberObserver called by wrapper in server Subscriber | ||||
| 	DefaultServerSubscriberObserver = func(ctx context.Context, msg server.Message, err error) []string { | ||||
| 		labels := []string{"endpoint", msg.Topic()} | ||||
| 		if err != nil { | ||||
| 			labels = append(labels, "error", err.Error()) | ||||
| 		} | ||||
| 		return labels | ||||
| 	} | ||||
|  | ||||
| 	// DefaultClientCallFuncObserver called by wrapper in client CallFunc | ||||
| 	DefaultClientCallFuncObserver = func(ctx context.Context, addr string, req client.Request, rsp interface{}, opts client.CallOptions, err error) []string { | ||||
| 		labels := []string{"service", req.Service(), "endpoint", req.Endpoint()} | ||||
| 		if err != nil { | ||||
| 			labels = append(labels, "error", err.Error()) | ||||
| 		} | ||||
| 		return labels | ||||
| 	} | ||||
|  | ||||
| 	// DefaultSkipEndpoints wrapper not called for this endpoints | ||||
| 	DefaultSkipEndpoints = []string{"Meter.Metrics", "Health.Live", "Health.Ready", "Health.Version"} | ||||
| ) | ||||
|  | ||||
| type lWrapper struct { | ||||
| 	client.Client | ||||
| 	serverHandler    server.HandlerFunc | ||||
| 	serverSubscriber server.SubscriberFunc | ||||
| 	clientCallFunc   client.CallFunc | ||||
| 	opts             Options | ||||
| } | ||||
|  | ||||
| type ( | ||||
| 	// ClientCallObserver func signature | ||||
| 	ClientCallObserver func(context.Context, client.Request, interface{}, []client.CallOption, error) []string | ||||
| 	// ClientStreamObserver func signature | ||||
| 	ClientStreamObserver func(context.Context, client.Request, []client.CallOption, client.Stream, error) []string | ||||
| 	// ClientPublishObserver func signature | ||||
| 	ClientPublishObserver func(context.Context, client.Message, []client.PublishOption, error) []string | ||||
| 	// ClientCallFuncObserver func signature | ||||
| 	ClientCallFuncObserver func(context.Context, string, client.Request, interface{}, client.CallOptions, error) []string | ||||
| 	// ServerHandlerObserver func signature | ||||
| 	ServerHandlerObserver func(context.Context, server.Request, interface{}, error) []string | ||||
| 	// ServerSubscriberObserver func signature | ||||
| 	ServerSubscriberObserver func(context.Context, server.Message, error) []string | ||||
| ) | ||||
|  | ||||
| // Options struct for wrapper | ||||
| type Options struct { | ||||
| 	// Logger that used for log | ||||
| 	Logger logger.Logger | ||||
| 	// ServerHandlerObservers funcs | ||||
| 	ServerHandlerObservers []ServerHandlerObserver | ||||
| 	// ServerSubscriberObservers funcs | ||||
| 	ServerSubscriberObservers []ServerSubscriberObserver | ||||
| 	// ClientCallObservers funcs | ||||
| 	ClientCallObservers []ClientCallObserver | ||||
| 	// ClientStreamObservers funcs | ||||
| 	ClientStreamObservers []ClientStreamObserver | ||||
| 	// ClientPublishObservers funcs | ||||
| 	ClientPublishObservers []ClientPublishObserver | ||||
| 	// ClientCallFuncObservers funcs | ||||
| 	ClientCallFuncObservers []ClientCallFuncObserver | ||||
| 	// SkipEndpoints | ||||
| 	SkipEndpoints []string | ||||
| 	// Level for logger | ||||
| 	Level logger.Level | ||||
| 	// Enabled flag | ||||
| 	Enabled bool | ||||
| } | ||||
|  | ||||
| // Option func signature | ||||
| type Option func(*Options) | ||||
|  | ||||
| // NewOptions creates Options from Option slice | ||||
| func NewOptions(opts ...Option) Options { | ||||
| 	options := Options{ | ||||
| 		Logger:                    logger.DefaultLogger, | ||||
| 		Level:                     logger.TraceLevel, | ||||
| 		ClientCallObservers:       []ClientCallObserver{DefaultClientCallObserver}, | ||||
| 		ClientStreamObservers:     []ClientStreamObserver{DefaultClientStreamObserver}, | ||||
| 		ClientPublishObservers:    []ClientPublishObserver{DefaultClientPublishObserver}, | ||||
| 		ClientCallFuncObservers:   []ClientCallFuncObserver{DefaultClientCallFuncObserver}, | ||||
| 		ServerHandlerObservers:    []ServerHandlerObserver{DefaultServerHandlerObserver}, | ||||
| 		ServerSubscriberObservers: []ServerSubscriberObserver{DefaultServerSubscriberObserver}, | ||||
| 		SkipEndpoints:             DefaultSkipEndpoints, | ||||
| 	} | ||||
|  | ||||
| 	for _, o := range opts { | ||||
| 		o(&options) | ||||
| 	} | ||||
|  | ||||
| 	return options | ||||
| } | ||||
|  | ||||
| // WithEnabled enable/diable flag | ||||
| func WithEnabled(b bool) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.Enabled = b | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // WithLevel log level | ||||
| func WithLevel(l logger.Level) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.Level = l | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // WithLogger logger | ||||
| func WithLogger(l logger.Logger) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.Logger = l | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // WithClientCallObservers funcs | ||||
| func WithClientCallObservers(ob ...ClientCallObserver) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.ClientCallObservers = ob | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // WithClientStreamObservers funcs | ||||
| func WithClientStreamObservers(ob ...ClientStreamObserver) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.ClientStreamObservers = ob | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // WithClientPublishObservers funcs | ||||
| func WithClientPublishObservers(ob ...ClientPublishObserver) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.ClientPublishObservers = ob | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // WithClientCallFuncObservers funcs | ||||
| func WithClientCallFuncObservers(ob ...ClientCallFuncObserver) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.ClientCallFuncObservers = ob | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // WithServerHandlerObservers funcs | ||||
| func WithServerHandlerObservers(ob ...ServerHandlerObserver) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.ServerHandlerObservers = ob | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // WithServerSubscriberObservers funcs | ||||
| func WithServerSubscriberObservers(ob ...ServerSubscriberObserver) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.ServerSubscriberObservers = ob | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // SkipEndpoins | ||||
| func SkipEndpoints(eps ...string) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.SkipEndpoints = append(o.SkipEndpoints, eps...) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (l *lWrapper) Call(ctx context.Context, req client.Request, rsp interface{}, opts ...client.CallOption) error { | ||||
| 	err := l.Client.Call(ctx, req, rsp, opts...) | ||||
|  | ||||
| 	endpoint := fmt.Sprintf("%s.%s", req.Service(), req.Endpoint()) | ||||
| 	for _, ep := range l.opts.SkipEndpoints { | ||||
| 		if ep == endpoint { | ||||
| 			return err | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	if !l.opts.Enabled { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	var labels []string | ||||
| 	for _, o := range l.opts.ClientCallObservers { | ||||
| 		labels = append(labels, o(ctx, req, rsp, opts, err)...) | ||||
| 	} | ||||
| 	l.opts.Logger.Fields(labels).Log(ctx, l.opts.Level) | ||||
|  | ||||
| 	return err | ||||
| } | ||||
|  | ||||
| func (l *lWrapper) Stream(ctx context.Context, req client.Request, opts ...client.CallOption) (client.Stream, error) { | ||||
| 	stream, err := l.Client.Stream(ctx, req, opts...) | ||||
|  | ||||
| 	endpoint := fmt.Sprintf("%s.%s", req.Service(), req.Endpoint()) | ||||
| 	for _, ep := range l.opts.SkipEndpoints { | ||||
| 		if ep == endpoint { | ||||
| 			return stream, err | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	if !l.opts.Enabled { | ||||
| 		return stream, err | ||||
| 	} | ||||
|  | ||||
| 	var labels []string | ||||
| 	for _, o := range l.opts.ClientStreamObservers { | ||||
| 		labels = append(labels, o(ctx, req, opts, stream, err)...) | ||||
| 	} | ||||
| 	l.opts.Logger.Fields(labels).Log(ctx, l.opts.Level) | ||||
|  | ||||
| 	return stream, err | ||||
| } | ||||
|  | ||||
| func (l *lWrapper) Publish(ctx context.Context, msg client.Message, opts ...client.PublishOption) error { | ||||
| 	err := l.Client.Publish(ctx, msg, opts...) | ||||
|  | ||||
| 	endpoint := msg.Topic() | ||||
| 	for _, ep := range l.opts.SkipEndpoints { | ||||
| 		if ep == endpoint { | ||||
| 			return err | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	if !l.opts.Enabled { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	var labels []string | ||||
| 	for _, o := range l.opts.ClientPublishObservers { | ||||
| 		labels = append(labels, o(ctx, msg, opts, err)...) | ||||
| 	} | ||||
| 	l.opts.Logger.Fields(labels).Log(ctx, l.opts.Level) | ||||
|  | ||||
| 	return err | ||||
| } | ||||
|  | ||||
| func (l *lWrapper) ServerHandler(ctx context.Context, req server.Request, rsp interface{}) error { | ||||
| 	err := l.serverHandler(ctx, req, rsp) | ||||
|  | ||||
| 	endpoint := req.Endpoint() | ||||
| 	for _, ep := range l.opts.SkipEndpoints { | ||||
| 		if ep == endpoint { | ||||
| 			return err | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	if !l.opts.Enabled { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	var labels []string | ||||
| 	for _, o := range l.opts.ServerHandlerObservers { | ||||
| 		labels = append(labels, o(ctx, req, rsp, err)...) | ||||
| 	} | ||||
| 	l.opts.Logger.Fields(labels).Log(ctx, l.opts.Level) | ||||
|  | ||||
| 	return err | ||||
| } | ||||
|  | ||||
| func (l *lWrapper) ServerSubscriber(ctx context.Context, msg server.Message) error { | ||||
| 	err := l.serverSubscriber(ctx, msg) | ||||
|  | ||||
| 	endpoint := msg.Topic() | ||||
| 	for _, ep := range l.opts.SkipEndpoints { | ||||
| 		if ep == endpoint { | ||||
| 			return err | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	if !l.opts.Enabled { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	var labels []string | ||||
| 	for _, o := range l.opts.ServerSubscriberObservers { | ||||
| 		labels = append(labels, o(ctx, msg, err)...) | ||||
| 	} | ||||
| 	l.opts.Logger.Fields(labels).Log(ctx, l.opts.Level) | ||||
|  | ||||
| 	return err | ||||
| } | ||||
|  | ||||
| // NewClientWrapper accepts an open options and returns a Client Wrapper | ||||
| func NewClientWrapper(opts ...Option) client.Wrapper { | ||||
| 	return func(c client.Client) client.Client { | ||||
| 		options := NewOptions() | ||||
| 		for _, o := range opts { | ||||
| 			o(&options) | ||||
| 		} | ||||
| 		return &lWrapper{opts: options, Client: c} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewClientCallWrapper accepts an options and returns a Call Wrapper | ||||
| func NewClientCallWrapper(opts ...Option) client.CallWrapper { | ||||
| 	return func(h client.CallFunc) client.CallFunc { | ||||
| 		options := NewOptions() | ||||
| 		for _, o := range opts { | ||||
| 			o(&options) | ||||
| 		} | ||||
|  | ||||
| 		l := &lWrapper{opts: options, clientCallFunc: h} | ||||
| 		return l.ClientCallFunc | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (l *lWrapper) ClientCallFunc(ctx context.Context, addr string, req client.Request, rsp interface{}, opts client.CallOptions) error { | ||||
| 	err := l.clientCallFunc(ctx, addr, req, rsp, opts) | ||||
|  | ||||
| 	endpoint := fmt.Sprintf("%s.%s", req.Service(), req.Endpoint()) | ||||
| 	for _, ep := range l.opts.SkipEndpoints { | ||||
| 		if ep == endpoint { | ||||
| 			return err | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	if !l.opts.Enabled { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	var labels []string | ||||
| 	for _, o := range l.opts.ClientCallFuncObservers { | ||||
| 		labels = append(labels, o(ctx, addr, req, rsp, opts, err)...) | ||||
| 	} | ||||
| 	l.opts.Logger.Fields(labels).Log(ctx, l.opts.Level) | ||||
|  | ||||
| 	return err | ||||
| } | ||||
|  | ||||
| // NewServerHandlerWrapper accepts an options and returns a Handler Wrapper | ||||
| func NewServerHandlerWrapper(opts ...Option) server.HandlerWrapper { | ||||
| 	return func(h server.HandlerFunc) server.HandlerFunc { | ||||
| 		options := NewOptions() | ||||
| 		for _, o := range opts { | ||||
| 			o(&options) | ||||
| 		} | ||||
|  | ||||
| 		l := &lWrapper{opts: options, serverHandler: h} | ||||
| 		return l.ServerHandler | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewServerSubscriberWrapper accepts an options and returns a Subscriber Wrapper | ||||
| func NewServerSubscriberWrapper(opts ...Option) server.SubscriberWrapper { | ||||
| 	return func(h server.SubscriberFunc) server.SubscriberFunc { | ||||
| 		options := NewOptions() | ||||
| 		for _, o := range opts { | ||||
| 			o(&options) | ||||
| 		} | ||||
|  | ||||
| 		l := &lWrapper{opts: options, serverSubscriber: h} | ||||
| 		return l.ServerSubscriber | ||||
| 	} | ||||
| } | ||||
| @@ -24,6 +24,17 @@ func FromIncomingContext(ctx context.Context) (Metadata, bool) { | ||||
| 	return md.md, ok | ||||
| } | ||||
|  | ||||
| // MustIncomingContext returns metadata from incoming ctx | ||||
| // returned metadata shoud not be modified or race condition happens. | ||||
| // If metadata not exists panics. | ||||
| func MustIncomingContext(ctx context.Context) Metadata { | ||||
| 	md, ok := FromIncomingContext(ctx) | ||||
| 	if !ok { | ||||
| 		panic("missing metadata") | ||||
| 	} | ||||
| 	return md | ||||
| } | ||||
|  | ||||
| // FromOutgoingContext returns metadata from outgoing ctx | ||||
| // returned metadata shoud not be modified or race condition happens | ||||
| func FromOutgoingContext(ctx context.Context) (Metadata, bool) { | ||||
| @@ -37,10 +48,19 @@ func FromOutgoingContext(ctx context.Context) (Metadata, bool) { | ||||
| 	return md.md, ok | ||||
| } | ||||
|  | ||||
| // MustOutgoingContext returns metadata from outgoing ctx | ||||
| // returned metadata shoud not be modified or race condition happens. | ||||
| // If metadata not exists panics. | ||||
| func MustOutgoingContext(ctx context.Context) Metadata { | ||||
| 	md, ok := FromOutgoingContext(ctx) | ||||
| 	if !ok { | ||||
| 		panic("missing metadata") | ||||
| 	} | ||||
| 	return md | ||||
| } | ||||
|  | ||||
| // FromContext returns metadata from the given context | ||||
| // returned metadata shoud not be modified or race condition happens | ||||
| // | ||||
| // Deprecated: use FromIncomingContext or FromOutgoingContext | ||||
| func FromContext(ctx context.Context) (Metadata, bool) { | ||||
| 	if ctx == nil { | ||||
| 		return nil, false | ||||
| @@ -52,17 +72,22 @@ func FromContext(ctx context.Context) (Metadata, bool) { | ||||
| 	return md.md, ok | ||||
| } | ||||
|  | ||||
| // MustContext returns metadata from the given context | ||||
| // returned metadata shoud not be modified or race condition happens | ||||
| func MustContext(ctx context.Context) Metadata { | ||||
| 	md, ok := FromContext(ctx) | ||||
| 	if !ok { | ||||
| 		panic("missing metadata") | ||||
| 	} | ||||
| 	return md | ||||
| } | ||||
|  | ||||
| // NewContext creates a new context with the given metadata | ||||
| // | ||||
| // Deprecated: use NewIncomingContext or NewOutgoingContext | ||||
| func NewContext(ctx context.Context, md Metadata) context.Context { | ||||
| 	if ctx == nil { | ||||
| 		ctx = context.Background() | ||||
| 	} | ||||
| 	ctx = context.WithValue(ctx, mdKey{}, &rawMetadata{md}) | ||||
| 	ctx = context.WithValue(ctx, mdIncomingKey{}, &rawMetadata{}) | ||||
| 	ctx = context.WithValue(ctx, mdOutgoingKey{}, &rawMetadata{}) | ||||
| 	return ctx | ||||
| 	return context.WithValue(ctx, mdKey{}, &rawMetadata{md}) | ||||
| } | ||||
|  | ||||
| // SetOutgoingContext modify outgoing context with given metadata | ||||
| @@ -94,11 +119,7 @@ func NewIncomingContext(ctx context.Context, md Metadata) context.Context { | ||||
| 	if ctx == nil { | ||||
| 		ctx = context.Background() | ||||
| 	} | ||||
| 	ctx = context.WithValue(ctx, mdIncomingKey{}, &rawMetadata{md}) | ||||
| 	if v, ok := ctx.Value(mdOutgoingKey{}).(*rawMetadata); !ok || v == nil { | ||||
| 		ctx = context.WithValue(ctx, mdOutgoingKey{}, &rawMetadata{}) | ||||
| 	} | ||||
| 	return ctx | ||||
| 	return context.WithValue(ctx, mdIncomingKey{}, &rawMetadata{md}) | ||||
| } | ||||
|  | ||||
| // NewOutgoingContext creates a new context with outcoming metadata attached | ||||
| @@ -106,11 +127,7 @@ func NewOutgoingContext(ctx context.Context, md Metadata) context.Context { | ||||
| 	if ctx == nil { | ||||
| 		ctx = context.Background() | ||||
| 	} | ||||
| 	ctx = context.WithValue(ctx, mdOutgoingKey{}, &rawMetadata{md}) | ||||
| 	if v, ok := ctx.Value(mdIncomingKey{}).(*rawMetadata); !ok || v == nil { | ||||
| 		ctx = context.WithValue(ctx, mdIncomingKey{}, &rawMetadata{}) | ||||
| 	} | ||||
| 	return ctx | ||||
| 	return context.WithValue(ctx, mdOutgoingKey{}, &rawMetadata{md}) | ||||
| } | ||||
|  | ||||
| // AppendOutgoingContext apends new md to context | ||||
| @@ -126,7 +143,7 @@ func AppendOutgoingContext(ctx context.Context, kv ...string) context.Context { | ||||
| 	for k, v := range md { | ||||
| 		omd.Set(k, v) | ||||
| 	} | ||||
| 	return NewOutgoingContext(ctx, omd) | ||||
| 	return ctx | ||||
| } | ||||
|  | ||||
| // AppendIncomingContext apends new md to context | ||||
| @@ -142,5 +159,21 @@ func AppendIncomingContext(ctx context.Context, kv ...string) context.Context { | ||||
| 	for k, v := range md { | ||||
| 		omd.Set(k, v) | ||||
| 	} | ||||
| 	return NewIncomingContext(ctx, omd) | ||||
| 	return ctx | ||||
| } | ||||
|  | ||||
| // AppendContext apends new md to context | ||||
| func AppendContext(ctx context.Context, kv ...string) context.Context { | ||||
| 	md, ok := Pairs(kv...) | ||||
| 	if !ok { | ||||
| 		return ctx | ||||
| 	} | ||||
| 	omd, ok := FromContext(ctx) | ||||
| 	if !ok { | ||||
| 		return NewContext(ctx, md) | ||||
| 	} | ||||
| 	for k, v := range md { | ||||
| 		omd.Set(k, v) | ||||
| 	} | ||||
| 	return ctx | ||||
| } | ||||
|   | ||||
| @@ -1,9 +1,10 @@ | ||||
| // Package metadata is a way of defining message headers | ||||
| package metadata // import "go.unistack.org/micro/v3/metadata" | ||||
| package metadata | ||||
|  | ||||
| import ( | ||||
| 	"net/textproto" | ||||
| 	"sort" | ||||
| 	"strings" | ||||
| ) | ||||
|  | ||||
| var ( | ||||
| @@ -19,6 +20,8 @@ var ( | ||||
| 	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. | ||||
| @@ -64,6 +67,14 @@ func (md Metadata) Iterator() *Iterator { | ||||
| 	return iter | ||||
| } | ||||
|  | ||||
| func (md Metadata) MustGet(key string) string { | ||||
| 	val, ok := md.Get(key) | ||||
| 	if !ok { | ||||
| 		panic("missing metadata key") | ||||
| 	} | ||||
| 	return val | ||||
| } | ||||
|  | ||||
| // Get returns value from metadata by key | ||||
| func (md Metadata) Get(key string) (string, bool) { | ||||
| 	// fast path | ||||
| @@ -71,6 +82,9 @@ func (md Metadata) Get(key string) (string, bool) { | ||||
| 	if !ok { | ||||
| 		// slow path | ||||
| 		val, ok = md[textproto.CanonicalMIMEHeaderKey(key)] | ||||
| 		if !ok { | ||||
| 			val, ok = md[strings.ToLower(key)] | ||||
| 		} | ||||
| 	} | ||||
| 	return val, ok | ||||
| } | ||||
| @@ -92,15 +106,25 @@ func (md Metadata) Del(keys ...string) { | ||||
| 		delete(md, key) | ||||
| 		// slow path | ||||
| 		delete(md, textproto.CanonicalMIMEHeaderKey(key)) | ||||
| 		// very slow path | ||||
| 		delete(md, strings.ToLower(key)) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Copy makes a copy of the metadata | ||||
| func Copy(md Metadata) Metadata { | ||||
| 	nmd := New(len(md)) | ||||
| 	for key, val := range md { | ||||
| 		nmd.Set(key, val) | ||||
| func (md Metadata) CopyTo(dst Metadata) { | ||||
| 	for k, v := range md { | ||||
| 		dst[k] = v | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Copy makes a copy of the metadata | ||||
| func Copy(md Metadata, exclude ...string) Metadata { | ||||
| 	nmd := New(len(md)) | ||||
| 	for k, v := range md { | ||||
| 		nmd[k] = v | ||||
| 	} | ||||
| 	nmd.Del(exclude...) | ||||
| 	return nmd | ||||
| } | ||||
|  | ||||
| @@ -122,7 +146,7 @@ func Merge(omd Metadata, mmd Metadata, overwrite bool) Metadata { | ||||
| 		case ok && !overwrite: | ||||
| 			continue | ||||
| 		case val != "": | ||||
| 			nmd.Set(key, val) | ||||
| 			nmd[key] = val | ||||
| 		case ok && val == "": | ||||
| 			nmd.Del(key) | ||||
| 		} | ||||
| @@ -136,6 +160,8 @@ func Pairs(kv ...string) (Metadata, bool) { | ||||
| 		return nil, false | ||||
| 	} | ||||
| 	md := New(len(kv) / 2) | ||||
| 	md.Set(kv...) | ||||
| 	for idx := 0; idx < len(kv); idx += 2 { | ||||
| 		md[kv[idx]] = kv[idx+1] | ||||
| 	} | ||||
| 	return md, true | ||||
| } | ||||
|   | ||||
| @@ -5,6 +5,37 @@ import ( | ||||
| 	"testing" | ||||
| ) | ||||
|  | ||||
| func TestLowercase(t *testing.T) { | ||||
| 	md := New(1) | ||||
| 	md["x-request-id"] = "12345" | ||||
| 	v, ok := md.Get("X-Request-Id") | ||||
| 	if !ok || v == "" { | ||||
| 		t.Fatalf("metadata invalid %#+v", md) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestMultipleUsage(t *testing.T) { | ||||
| 	ctx := context.TODO() | ||||
| 	md := New(0) | ||||
| 	md.Set("key1_1", "val1_1", "key1_2", "val1_2", "key1_3", "val1_3") | ||||
| 	ctx = NewIncomingContext(ctx, Copy(md)) | ||||
| 	ctx = NewOutgoingContext(ctx, Copy(md)) | ||||
| 	imd, _ := FromIncomingContext(ctx) | ||||
| 	omd, _ := FromOutgoingContext(ctx) | ||||
| 	_ = func(x context.Context) context.Context { | ||||
| 		m, _ := FromIncomingContext(x) | ||||
| 		m.Del("key1_2") | ||||
| 		return ctx | ||||
| 	}(ctx) | ||||
| 	_ = func(x context.Context) context.Context { | ||||
| 		m, _ := FromIncomingContext(x) | ||||
| 		m.Del("key1_3") | ||||
| 		return ctx | ||||
| 	}(ctx) | ||||
| 	_ = imd | ||||
| 	_ = omd | ||||
| } | ||||
|  | ||||
| func TestMetadataSetMultiple(t *testing.T) { | ||||
| 	md := New(4) | ||||
| 	md.Set("key1", "val1", "key2", "val2", "key3") | ||||
| @@ -57,6 +88,13 @@ func TestPassing(t *testing.T) { | ||||
|  | ||||
| 	ctx = NewIncomingContext(ctx, md1) | ||||
| 	testCtx(ctx) | ||||
| 	_, ok := FromOutgoingContext(ctx) | ||||
| 	if ok { | ||||
| 		t.Fatalf("create outgoing context") | ||||
| 	} | ||||
|  | ||||
| 	ctx = NewOutgoingContext(ctx, New(1)) | ||||
| 	testCtx(ctx) | ||||
| 	md, ok := FromOutgoingContext(ctx) | ||||
| 	if !ok { | ||||
| 		t.Fatalf("missing metadata from outgoing context") | ||||
| @@ -80,7 +118,7 @@ func TestMerge(t *testing.T) { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestIterator(t *testing.T) { | ||||
| func TestIterator(_ *testing.T) { | ||||
| 	md := Metadata{ | ||||
| 		"1Last":   "last", | ||||
| 		"2First":  "first", | ||||
| @@ -190,3 +228,14 @@ func TestMetadataContext(t *testing.T) { | ||||
| 		t.Errorf("Expected metadata length 1 got %d", i) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestCopy(t *testing.T) { | ||||
| 	md := New(2) | ||||
| 	md.Set("key1", "val1", "key2", "val2") | ||||
| 	nmd := Copy(md, "key2") | ||||
| 	if len(nmd) != 1 { | ||||
| 		t.Fatal("Copy exclude not works") | ||||
| 	} else if nmd["Key1"] != "val1" { | ||||
| 		t.Fatal("Copy exclude not works") | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -15,6 +15,15 @@ func FromContext(ctx context.Context) (Meter, bool) { | ||||
| 	return c, ok | ||||
| } | ||||
|  | ||||
| // 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 | ||||
| func NewContext(ctx context.Context, c Meter) context.Context { | ||||
| 	if ctx == nil { | ||||
|   | ||||
| @@ -1,12 +0,0 @@ | ||||
| package meter | ||||
|  | ||||
| //go:generate sh -c "protoc -I./handler -I../ -I$(go list -f '{{ .Dir }}' -m go.unistack.org/micro-proto/v3) --go-micro_out='components=micro|http|server',standalone=false,debug=true,paths=source_relative:./handler handler/handler.proto" | ||||
|  | ||||
| import ( | ||||
|  | ||||
| 	// import required packages | ||||
| 	_ "go.unistack.org/micro-proto/v3/api" | ||||
|  | ||||
| 	// import required packages | ||||
| 	_ "go.unistack.org/micro-proto/v3/openapiv3" | ||||
| ) | ||||
| @@ -1,67 +0,0 @@ | ||||
| package handler // import "go.unistack.org/micro/v3/meter/handler" | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"context" | ||||
|  | ||||
| 	"go.unistack.org/micro/v3/codec" | ||||
| 	"go.unistack.org/micro/v3/errors" | ||||
| 	"go.unistack.org/micro/v3/meter" | ||||
| ) | ||||
|  | ||||
| // guard to fail early | ||||
| var _ MeterServer = &Handler{} | ||||
|  | ||||
| type Handler struct { | ||||
| 	opts Options | ||||
| } | ||||
|  | ||||
| type Option func(*Options) | ||||
|  | ||||
| type Options struct { | ||||
| 	Meter        meter.Meter | ||||
| 	Name         string | ||||
| 	MeterOptions []meter.Option | ||||
| } | ||||
|  | ||||
| func Meter(m meter.Meter) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.Meter = m | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func Name(name string) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.Name = name | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func MeterOptions(opts ...meter.Option) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.MeterOptions = append(o.MeterOptions, opts...) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func NewOptions(opts ...Option) Options { | ||||
| 	options := Options{Meter: meter.DefaultMeter} | ||||
| 	for _, o := range opts { | ||||
| 		o(&options) | ||||
| 	} | ||||
| 	return options | ||||
| } | ||||
|  | ||||
| func NewHandler(opts ...Option) *Handler { | ||||
| 	options := NewOptions(opts...) | ||||
| 	return &Handler{opts: options} | ||||
| } | ||||
|  | ||||
| func (h *Handler) Metrics(ctx context.Context, req *codec.Frame, rsp *codec.Frame) error { | ||||
| 	buf := bytes.NewBuffer(nil) | ||||
| 	if err := h.opts.Meter.Write(buf, h.opts.MeterOptions...); err != nil { | ||||
| 		return errors.InternalServerError(h.opts.Name, "%v", err) | ||||
| 	} | ||||
|  | ||||
| 	rsp.Data = buf.Bytes() | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
| @@ -1,24 +0,0 @@ | ||||
| syntax = "proto3"; | ||||
|  | ||||
| package micro.meter.handler; | ||||
| option go_package = "go.unistack.org/micro/v3/meter/handler;handler"; | ||||
|  | ||||
| import "api/annotations.proto"; | ||||
| import "openapiv3/annotations.proto"; | ||||
| import "codec/frame.proto"; | ||||
|  | ||||
| service Meter { | ||||
| 	rpc Metrics(micro.codec.Frame) returns (micro.codec.Frame) { | ||||
| 		option (micro.openapiv3.openapiv3_operation) = { | ||||
|       operation_id: "Metrics"; | ||||
|       responses: { | ||||
|         default: { | ||||
|           reference: { | ||||
|             _ref: "micro.codec.Frame"; | ||||
|           }; | ||||
|         }; | ||||
|       }; | ||||
|     }; | ||||
| 		option (micro.api.http) = { get: "/metrics"; }; | ||||
|   }; | ||||
| }; | ||||
| @@ -1,32 +0,0 @@ | ||||
| // Code generated by protoc-gen-go-micro. DO NOT EDIT. | ||||
| // protoc-gen-go-micro version: v3.5.3 | ||||
| // source: handler.proto | ||||
|  | ||||
| package handler | ||||
|  | ||||
| import ( | ||||
| 	context "context" | ||||
| 	api "go.unistack.org/micro/v3/api" | ||||
| 	codec "go.unistack.org/micro/v3/codec" | ||||
| ) | ||||
|  | ||||
| var ( | ||||
| 	MeterName = "Meter" | ||||
|  | ||||
| 	MeterEndpoints = []api.Endpoint{ | ||||
| 		{ | ||||
| 			Name:    "Meter.Metrics", | ||||
| 			Path:    []string{"/metrics"}, | ||||
| 			Method:  []string{"GET"}, | ||||
| 			Handler: "rpc", | ||||
| 		}, | ||||
| 	} | ||||
| ) | ||||
|  | ||||
| func NewMeterEndpoints() []api.Endpoint { | ||||
| 	return MeterEndpoints | ||||
| } | ||||
|  | ||||
| type MeterServer interface { | ||||
| 	Metrics(ctx context.Context, req *codec.Frame, rsp *codec.Frame) error | ||||
| } | ||||
| @@ -1,35 +0,0 @@ | ||||
| // Code generated by protoc-gen-go-micro. DO NOT EDIT. | ||||
| // protoc-gen-go-micro version: v3.5.3 | ||||
| // source: handler.proto | ||||
|  | ||||
| package handler | ||||
|  | ||||
| import ( | ||||
| 	context "context" | ||||
| 	api "go.unistack.org/micro/v3/api" | ||||
| 	codec "go.unistack.org/micro/v3/codec" | ||||
| 	server "go.unistack.org/micro/v3/server" | ||||
| ) | ||||
|  | ||||
| type meterServer struct { | ||||
| 	MeterServer | ||||
| } | ||||
|  | ||||
| func (h *meterServer) Metrics(ctx context.Context, req *codec.Frame, rsp *codec.Frame) error { | ||||
| 	return h.MeterServer.Metrics(ctx, req, rsp) | ||||
| } | ||||
|  | ||||
| func RegisterMeterServer(s server.Server, sh MeterServer, opts ...server.HandlerOption) error { | ||||
| 	type meter interface { | ||||
| 		Metrics(ctx context.Context, req *codec.Frame, rsp *codec.Frame) error | ||||
| 	} | ||||
| 	type Meter struct { | ||||
| 		meter | ||||
| 	} | ||||
| 	h := &meterServer{sh} | ||||
| 	var nopts []server.HandlerOption | ||||
| 	for _, endpoint := range MeterEndpoints { | ||||
| 		nopts = append(nopts, api.WithEndpoint(&endpoint)) | ||||
| 	} | ||||
| 	return s.Handle(s.NewHandler(&Meter{h}, append(nopts, opts...)...)) | ||||
| } | ||||
| @@ -1,5 +1,5 @@ | ||||
| // Package meter is for instrumentation | ||||
| package meter // import "go.unistack.org/micro/v3/meter" | ||||
| package meter | ||||
|  | ||||
| import ( | ||||
| 	"io" | ||||
| @@ -16,14 +16,19 @@ var ( | ||||
| 	DefaultAddress = ":9090" | ||||
| 	// DefaultPath the meter endpoint where the Meter data will be made available | ||||
| 	DefaultPath = "/metrics" | ||||
| 	// DefaultMetricPrefix holds the string that prepends to all metrics | ||||
| 	DefaultMetricPrefix = "micro_" | ||||
| 	// DefaultLabelPrefix holds the string that prepends to all labels | ||||
| 	DefaultLabelPrefix = "micro_" | ||||
| 	// DefaultMeterStatsInterval specifies interval for meter updating | ||||
| 	DefaultMeterStatsInterval = 5 * time.Second | ||||
| 	// DefaultSummaryQuantiles is the default spread of stats for summary | ||||
| 	DefaultSummaryQuantiles = []float64{0.5, 0.9, 0.97, 0.99, 1} | ||||
| 	// DefaultSummaryWindow is the default window for summary | ||||
| 	DefaultSummaryWindow = 5 * time.Minute | ||||
| 	// DefaultSkipEndpoints is the slice of endpoint that must not be metered | ||||
| 	DefaultSkipEndpoints = []string{ | ||||
| 		"MeterService.Metrics", | ||||
| 		"HealthService.Live", | ||||
| 		"HealthService.Ready", | ||||
| 		"HealthService.Version", | ||||
| 	} | ||||
| ) | ||||
|  | ||||
| // Meter is an interface for collecting and instrumenting metrics | ||||
|   | ||||
| @@ -37,32 +37,32 @@ func (r *noopMeter) Init(opts ...Option) error { | ||||
| } | ||||
|  | ||||
| // 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} | ||||
| } | ||||
|  | ||||
| // 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} | ||||
| } | ||||
|  | ||||
| // 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} | ||||
| } | ||||
|  | ||||
| // 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} | ||||
| } | ||||
|  | ||||
| // 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} | ||||
| } | ||||
|  | ||||
| // 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} | ||||
| } | ||||
|  | ||||
| @@ -77,7 +77,7 @@ func (r *noopMeter) Set(opts ...Option) Meter { | ||||
| 	return m | ||||
| } | ||||
|  | ||||
| func (r *noopMeter) Write(w io.Writer, opts ...Option) error { | ||||
| func (r *noopMeter) Write(_ io.Writer, _ ...Option) error { | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
|   | ||||
| @@ -2,8 +2,6 @@ package meter | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
|  | ||||
| 	"go.unistack.org/micro/v3/logger" | ||||
| ) | ||||
|  | ||||
| // Option powers the configuration for metrics implementations: | ||||
| @@ -11,8 +9,6 @@ type Option func(*Options) | ||||
|  | ||||
| // Options for metrics implementations | ||||
| type Options struct { | ||||
| 	// Logger used for logging | ||||
| 	Logger logger.Logger | ||||
| 	// Context holds external options | ||||
| 	Context context.Context | ||||
| 	// Name holds the meter name | ||||
| @@ -21,10 +17,6 @@ type Options struct { | ||||
| 	Address string | ||||
| 	// Path holds the path for metrics | ||||
| 	Path string | ||||
| 	// MetricPrefix holds the prefix for all metrics | ||||
| 	MetricPrefix string | ||||
| 	// LabelPrefix holds the prefix for all labels | ||||
| 	LabelPrefix string | ||||
| 	// Labels holds the default labels | ||||
| 	Labels []string | ||||
| 	// WriteProcessMetrics flag to write process metrics | ||||
| @@ -39,9 +31,6 @@ func NewOptions(opt ...Option) Options { | ||||
| 		Address: DefaultAddress, | ||||
| 		Path:    DefaultPath, | ||||
| 		Context: context.Background(), | ||||
| 		Logger:       logger.DefaultLogger, | ||||
| 		MetricPrefix: DefaultMetricPrefix, | ||||
| 		LabelPrefix:  DefaultLabelPrefix, | ||||
| 	} | ||||
|  | ||||
| 	for _, o := range opt { | ||||
| @@ -51,20 +40,6 @@ func NewOptions(opt ...Option) Options { | ||||
| 	return opts | ||||
| } | ||||
|  | ||||
| // LabelPrefix sets the labels prefix | ||||
| func LabelPrefix(pref string) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.LabelPrefix = pref | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // MetricPrefix sets the metric prefix | ||||
| func MetricPrefix(pref string) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.MetricPrefix = pref | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Context sets the metrics context | ||||
| func Context(ctx context.Context) Option { | ||||
| 	return func(o *Options) { | ||||
| @@ -95,14 +70,7 @@ func TimingObjectives(value map[float64]float64) Option { | ||||
| } | ||||
| */ | ||||
|  | ||||
| // Logger sets the logger | ||||
| func Logger(l logger.Logger) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.Logger = l | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Labels sets the meter labels | ||||
| // Labels add the meter labels | ||||
| func Labels(ls ...string) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.Labels = append(o.Labels, ls...) | ||||
|   | ||||
| @@ -1,347 +0,0 @@ | ||||
| package wrapper // import "go.unistack.org/micro/v3/meter/wrapper" | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"fmt" | ||||
| 	"time" | ||||
|  | ||||
| 	"go.unistack.org/micro/v3/client" | ||||
| 	"go.unistack.org/micro/v3/meter" | ||||
| 	"go.unistack.org/micro/v3/server" | ||||
| ) | ||||
|  | ||||
| var ( | ||||
| 	// ClientRequestDurationSeconds specifies meter metric name | ||||
| 	ClientRequestDurationSeconds = "client_request_duration_seconds" | ||||
| 	// ClientRequestLatencyMicroseconds specifies meter metric name | ||||
| 	ClientRequestLatencyMicroseconds = "client_request_latency_microseconds" | ||||
| 	// ClientRequestTotal specifies meter metric name | ||||
| 	ClientRequestTotal = "client_request_total" | ||||
| 	// ClientRequestInflight specifies meter metric name | ||||
| 	ClientRequestInflight = "client_request_inflight" | ||||
| 	// ServerRequestDurationSeconds specifies meter metric name | ||||
| 	ServerRequestDurationSeconds = "server_request_duration_seconds" | ||||
| 	// ServerRequestLatencyMicroseconds specifies meter metric name | ||||
| 	ServerRequestLatencyMicroseconds = "server_request_latency_microseconds" | ||||
| 	// ServerRequestTotal specifies meter metric name | ||||
| 	ServerRequestTotal = "server_request_total" | ||||
| 	// ServerRequestInflight specifies meter metric name | ||||
| 	ServerRequestInflight = "server_request_inflight" | ||||
| 	// PublishMessageDurationSeconds specifies meter metric name | ||||
| 	PublishMessageDurationSeconds = "publish_message_duration_seconds" | ||||
| 	// PublishMessageLatencyMicroseconds specifies meter metric name | ||||
| 	PublishMessageLatencyMicroseconds = "publish_message_latency_microseconds" | ||||
| 	// PublishMessageTotal specifies meter metric name | ||||
| 	PublishMessageTotal = "publish_message_total" | ||||
| 	// PublishMessageInflight specifies meter metric name | ||||
| 	PublishMessageInflight = "publish_message_inflight" | ||||
| 	// SubscribeMessageDurationSeconds specifies meter metric name | ||||
| 	SubscribeMessageDurationSeconds = "subscribe_message_duration_seconds" | ||||
| 	// SubscribeMessageLatencyMicroseconds specifies meter metric name | ||||
| 	SubscribeMessageLatencyMicroseconds = "subscribe_message_latency_microseconds" | ||||
| 	// SubscribeMessageTotal specifies meter metric name | ||||
| 	SubscribeMessageTotal = "subscribe_message_total" | ||||
| 	// SubscribeMessageInflight specifies meter metric name | ||||
| 	SubscribeMessageInflight = "subscribe_message_inflight" | ||||
|  | ||||
| 	labelSuccess  = "success" | ||||
| 	labelFailure  = "failure" | ||||
| 	labelStatus   = "status" | ||||
| 	labelEndpoint = "endpoint" | ||||
|  | ||||
| 	// DefaultSkipEndpoints contains list of endpoints that not evaluted by wrapper | ||||
| 	DefaultSkipEndpoints = []string{"Meter.Metrics", "Health.Live", "Health.Ready", "Health.Version"} | ||||
| ) | ||||
|  | ||||
| // Options struct | ||||
| type Options struct { | ||||
| 	Meter         meter.Meter | ||||
| 	lopts         []meter.Option | ||||
| 	SkipEndpoints []string | ||||
| } | ||||
|  | ||||
| // Option func signature | ||||
| type Option func(*Options) | ||||
|  | ||||
| // NewOptions creates new Options struct | ||||
| func NewOptions(opts ...Option) Options { | ||||
| 	options := Options{ | ||||
| 		Meter:         meter.DefaultMeter, | ||||
| 		lopts:         make([]meter.Option, 0, 5), | ||||
| 		SkipEndpoints: DefaultSkipEndpoints, | ||||
| 	} | ||||
| 	for _, o := range opts { | ||||
| 		o(&options) | ||||
| 	} | ||||
| 	return options | ||||
| } | ||||
|  | ||||
| // ServiceName passes service name to meter label | ||||
| func ServiceName(name string) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.lopts = append(o.lopts, meter.Labels("name", name)) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // ServiceVersion passes service version to meter label | ||||
| func ServiceVersion(version string) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.lopts = append(o.lopts, meter.Labels("version", version)) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // ServiceID passes service id to meter label | ||||
| func ServiceID(id string) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.lopts = append(o.lopts, meter.Labels("id", id)) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Meter passes meter | ||||
| func Meter(m meter.Meter) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.Meter = m | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // SkipEndoints add endpoint to skip | ||||
| func SkipEndoints(eps ...string) Option { | ||||
| 	return func(o *Options) { | ||||
| 		o.SkipEndpoints = append(o.SkipEndpoints, eps...) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| type wrapper struct { | ||||
| 	client.Client | ||||
| 	callFunc client.CallFunc | ||||
| 	opts     Options | ||||
| } | ||||
|  | ||||
| // NewClientWrapper create new client wrapper | ||||
| func NewClientWrapper(opts ...Option) client.Wrapper { | ||||
| 	return func(c client.Client) client.Client { | ||||
| 		handler := &wrapper{ | ||||
| 			opts:   NewOptions(opts...), | ||||
| 			Client: c, | ||||
| 		} | ||||
| 		return handler | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewCallWrapper create new call wrapper | ||||
| func NewCallWrapper(opts ...Option) client.CallWrapper { | ||||
| 	return func(fn client.CallFunc) client.CallFunc { | ||||
| 		handler := &wrapper{ | ||||
| 			opts:     NewOptions(opts...), | ||||
| 			callFunc: fn, | ||||
| 		} | ||||
| 		return handler.CallFunc | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (w *wrapper) CallFunc(ctx context.Context, addr string, req client.Request, rsp interface{}, opts client.CallOptions) error { | ||||
| 	endpoint := fmt.Sprintf("%s.%s", req.Service(), req.Endpoint()) | ||||
| 	for _, ep := range w.opts.SkipEndpoints { | ||||
| 		if ep == endpoint { | ||||
| 			return w.callFunc(ctx, addr, req, rsp, opts) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	labels := make([]string, 0, 4) | ||||
| 	labels = append(labels, labelEndpoint, endpoint) | ||||
|  | ||||
| 	w.opts.Meter.Counter(ClientRequestInflight, labels...).Inc() | ||||
| 	ts := time.Now() | ||||
| 	err := w.callFunc(ctx, addr, req, rsp, opts) | ||||
| 	te := time.Since(ts) | ||||
| 	w.opts.Meter.Counter(ClientRequestInflight, labels...).Dec() | ||||
|  | ||||
| 	w.opts.Meter.Summary(ClientRequestLatencyMicroseconds, labels...).Update(te.Seconds()) | ||||
| 	w.opts.Meter.Histogram(ClientRequestDurationSeconds, labels...).Update(te.Seconds()) | ||||
|  | ||||
| 	if err == nil { | ||||
| 		labels = append(labels, labelStatus, labelSuccess) | ||||
| 	} else { | ||||
| 		labels = append(labels, labelStatus, labelFailure) | ||||
| 	} | ||||
| 	w.opts.Meter.Counter(ClientRequestTotal, labels...).Inc() | ||||
|  | ||||
| 	return err | ||||
| } | ||||
|  | ||||
| func (w *wrapper) Call(ctx context.Context, req client.Request, rsp interface{}, opts ...client.CallOption) error { | ||||
| 	endpoint := fmt.Sprintf("%s.%s", req.Service(), req.Endpoint()) | ||||
| 	for _, ep := range w.opts.SkipEndpoints { | ||||
| 		if ep == endpoint { | ||||
| 			return w.Client.Call(ctx, req, rsp, opts...) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	labels := make([]string, 0, 4) | ||||
| 	labels = append(labels, labelEndpoint, endpoint) | ||||
|  | ||||
| 	w.opts.Meter.Counter(ClientRequestInflight, labels...).Inc() | ||||
| 	ts := time.Now() | ||||
| 	err := w.Client.Call(ctx, req, rsp, opts...) | ||||
| 	te := time.Since(ts) | ||||
| 	w.opts.Meter.Counter(ClientRequestInflight, labels...).Dec() | ||||
|  | ||||
| 	w.opts.Meter.Summary(ClientRequestLatencyMicroseconds, labels...).Update(te.Seconds()) | ||||
| 	w.opts.Meter.Histogram(ClientRequestDurationSeconds, labels...).Update(te.Seconds()) | ||||
|  | ||||
| 	if err == nil { | ||||
| 		labels = append(labels, labelStatus, labelSuccess) | ||||
| 	} else { | ||||
| 		labels = append(labels, labelStatus, labelFailure) | ||||
| 	} | ||||
| 	w.opts.Meter.Counter(ClientRequestTotal, labels...).Inc() | ||||
|  | ||||
| 	return err | ||||
| } | ||||
|  | ||||
| func (w *wrapper) Stream(ctx context.Context, req client.Request, opts ...client.CallOption) (client.Stream, error) { | ||||
| 	endpoint := fmt.Sprintf("%s.%s", req.Service(), req.Endpoint()) | ||||
| 	for _, ep := range w.opts.SkipEndpoints { | ||||
| 		if ep == endpoint { | ||||
| 			return w.Client.Stream(ctx, req, opts...) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	labels := make([]string, 0, 4) | ||||
| 	labels = append(labels, labelEndpoint, endpoint) | ||||
|  | ||||
| 	w.opts.Meter.Counter(ClientRequestInflight, labels...).Inc() | ||||
| 	ts := time.Now() | ||||
| 	stream, err := w.Client.Stream(ctx, req, opts...) | ||||
| 	te := time.Since(ts) | ||||
| 	w.opts.Meter.Counter(ClientRequestInflight, labels...).Dec() | ||||
|  | ||||
| 	w.opts.Meter.Summary(ClientRequestLatencyMicroseconds, labels...).Update(te.Seconds()) | ||||
| 	w.opts.Meter.Histogram(ClientRequestDurationSeconds, labels...).Update(te.Seconds()) | ||||
|  | ||||
| 	if err == nil { | ||||
| 		labels = append(labels, labelStatus, labelSuccess) | ||||
| 	} else { | ||||
| 		labels = append(labels, labelStatus, labelFailure) | ||||
| 	} | ||||
| 	w.opts.Meter.Counter(ClientRequestTotal, labels...).Inc() | ||||
|  | ||||
| 	return stream, err | ||||
| } | ||||
|  | ||||
| func (w *wrapper) Publish(ctx context.Context, p client.Message, opts ...client.PublishOption) error { | ||||
| 	endpoint := p.Topic() | ||||
|  | ||||
| 	labels := make([]string, 0, 4) | ||||
| 	labels = append(labels, labelEndpoint, endpoint) | ||||
|  | ||||
| 	w.opts.Meter.Counter(PublishMessageInflight, labels...).Inc() | ||||
| 	ts := time.Now() | ||||
| 	err := w.Client.Publish(ctx, p, opts...) | ||||
| 	te := time.Since(ts) | ||||
| 	w.opts.Meter.Counter(PublishMessageInflight, labels...).Dec() | ||||
|  | ||||
| 	w.opts.Meter.Summary(PublishMessageLatencyMicroseconds, labels...).Update(te.Seconds()) | ||||
| 	w.opts.Meter.Histogram(PublishMessageDurationSeconds, labels...).Update(te.Seconds()) | ||||
|  | ||||
| 	if err == nil { | ||||
| 		labels = append(labels, labelStatus, labelSuccess) | ||||
| 	} else { | ||||
| 		labels = append(labels, labelStatus, labelFailure) | ||||
| 	} | ||||
| 	w.opts.Meter.Counter(PublishMessageTotal, labels...).Inc() | ||||
|  | ||||
| 	return err | ||||
| } | ||||
|  | ||||
| // NewHandlerWrapper create new server handler wrapper | ||||
| // deprecated | ||||
| func NewHandlerWrapper(opts ...Option) server.HandlerWrapper { | ||||
| 	handler := &wrapper{ | ||||
| 		opts: NewOptions(opts...), | ||||
| 	} | ||||
| 	return handler.HandlerFunc | ||||
| } | ||||
|  | ||||
| // NewServerHandlerWrapper create new server handler wrapper | ||||
| func NewServerHandlerWrapper(opts ...Option) server.HandlerWrapper { | ||||
| 	handler := &wrapper{ | ||||
| 		opts: NewOptions(opts...), | ||||
| 	} | ||||
| 	return handler.HandlerFunc | ||||
| } | ||||
|  | ||||
| func (w *wrapper) HandlerFunc(fn server.HandlerFunc) server.HandlerFunc { | ||||
| 	return func(ctx context.Context, req server.Request, rsp interface{}) error { | ||||
| 		endpoint := req.Service() + "." + req.Endpoint() | ||||
| 		for _, ep := range w.opts.SkipEndpoints { | ||||
| 			if ep == endpoint { | ||||
| 				return fn(ctx, req, rsp) | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		labels := make([]string, 0, 4) | ||||
| 		labels = append(labels, labelEndpoint, endpoint) | ||||
|  | ||||
| 		w.opts.Meter.Counter(ServerRequestInflight, labels...).Inc() | ||||
| 		ts := time.Now() | ||||
| 		err := fn(ctx, req, rsp) | ||||
| 		te := time.Since(ts) | ||||
| 		w.opts.Meter.Counter(ServerRequestInflight, labels...).Dec() | ||||
|  | ||||
| 		w.opts.Meter.Summary(ServerRequestLatencyMicroseconds, labels...).Update(te.Seconds()) | ||||
| 		w.opts.Meter.Histogram(ServerRequestDurationSeconds, labels...).Update(te.Seconds()) | ||||
|  | ||||
| 		if err == nil { | ||||
| 			labels = append(labels, labelStatus, labelSuccess) | ||||
| 		} else { | ||||
| 			labels = append(labels, labelStatus, labelFailure) | ||||
| 		} | ||||
| 		w.opts.Meter.Counter(ServerRequestTotal, labels...).Inc() | ||||
|  | ||||
| 		return err | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewSubscriberWrapper create server subscribe wrapper | ||||
| // deprecated | ||||
| func NewSubscriberWrapper(opts ...Option) server.SubscriberWrapper { | ||||
| 	handler := &wrapper{ | ||||
| 		opts: NewOptions(opts...), | ||||
| 	} | ||||
| 	return handler.SubscriberFunc | ||||
| } | ||||
|  | ||||
| func NewServerSubscriberWrapper(opts ...Option) server.SubscriberWrapper { | ||||
| 	handler := &wrapper{ | ||||
| 		opts: NewOptions(opts...), | ||||
| 	} | ||||
| 	return handler.SubscriberFunc | ||||
| } | ||||
|  | ||||
| func (w *wrapper) SubscriberFunc(fn server.SubscriberFunc) server.SubscriberFunc { | ||||
| 	return func(ctx context.Context, msg server.Message) error { | ||||
| 		endpoint := msg.Topic() | ||||
|  | ||||
| 		labels := make([]string, 0, 4) | ||||
| 		labels = append(labels, labelEndpoint, endpoint) | ||||
|  | ||||
| 		w.opts.Meter.Counter(SubscribeMessageInflight, labels...).Inc() | ||||
| 		ts := time.Now() | ||||
| 		err := fn(ctx, msg) | ||||
| 		te := time.Since(ts) | ||||
| 		w.opts.Meter.Counter(SubscribeMessageInflight, labels...).Dec() | ||||
|  | ||||
| 		w.opts.Meter.Summary(SubscribeMessageLatencyMicroseconds, labels...).Update(te.Seconds()) | ||||
| 		w.opts.Meter.Histogram(SubscribeMessageDurationSeconds, labels...).Update(te.Seconds()) | ||||
|  | ||||
| 		if err == nil { | ||||
| 			labels = append(labels, labelStatus, labelSuccess) | ||||
| 		} else { | ||||
| 			labels = append(labels, labelStatus, labelFailure) | ||||
| 		} | ||||
| 		w.opts.Meter.Counter(SubscribeMessageTotal, labels...).Inc() | ||||
|  | ||||
| 		return err | ||||
| 	} | ||||
| } | ||||
							
								
								
									
										98
									
								
								micro.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										98
									
								
								micro.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,98 @@ | ||||
| package micro | ||||
|  | ||||
| import ( | ||||
| 	"reflect" | ||||
|  | ||||
| 	"go.unistack.org/micro/v3/broker" | ||||
| 	"go.unistack.org/micro/v3/client" | ||||
| 	"go.unistack.org/micro/v3/codec" | ||||
| 	"go.unistack.org/micro/v3/flow" | ||||
| 	"go.unistack.org/micro/v3/fsm" | ||||
| 	"go.unistack.org/micro/v3/logger" | ||||
| 	"go.unistack.org/micro/v3/meter" | ||||
| 	"go.unistack.org/micro/v3/register" | ||||
| 	"go.unistack.org/micro/v3/resolver" | ||||
| 	"go.unistack.org/micro/v3/router" | ||||
| 	"go.unistack.org/micro/v3/selector" | ||||
| 	"go.unistack.org/micro/v3/server" | ||||
| 	"go.unistack.org/micro/v3/store" | ||||
| 	"go.unistack.org/micro/v3/sync" | ||||
| 	"go.unistack.org/micro/v3/tracer" | ||||
| ) | ||||
|  | ||||
| func As(b any, target any) bool { | ||||
| 	if b == nil { | ||||
| 		return false | ||||
| 	} | ||||
| 	if target == nil { | ||||
| 		return false | ||||
| 	} | ||||
| 	val := reflect.ValueOf(target) | ||||
| 	typ := val.Type() | ||||
| 	if typ.Kind() != reflect.Ptr || val.IsNil() { | ||||
| 		return false | ||||
| 	} | ||||
| 	targetType := typ.Elem() | ||||
| 	if targetType.Kind() != reflect.Interface { | ||||
| 		switch { | ||||
| 		case targetType.Implements(brokerType): | ||||
| 			break | ||||
| 		case targetType.Implements(loggerType): | ||||
| 			break | ||||
| 		case targetType.Implements(clientType): | ||||
| 			break | ||||
| 		case targetType.Implements(serverType): | ||||
| 			break | ||||
| 		case targetType.Implements(codecType): | ||||
| 			break | ||||
| 		case targetType.Implements(flowType): | ||||
| 			break | ||||
| 		case targetType.Implements(fsmType): | ||||
| 			break | ||||
| 		case targetType.Implements(meterType): | ||||
| 			break | ||||
| 		case targetType.Implements(registerType): | ||||
| 			break | ||||
| 		case targetType.Implements(resolverType): | ||||
| 			break | ||||
| 		case targetType.Implements(selectorType): | ||||
| 			break | ||||
| 		case targetType.Implements(storeType): | ||||
| 			break | ||||
| 		case targetType.Implements(syncType): | ||||
| 			break | ||||
| 		case targetType.Implements(serviceType): | ||||
| 			break | ||||
| 		case targetType.Implements(routerType): | ||||
| 			break | ||||
| 		case targetType.Implements(tracerType): | ||||
| 			break | ||||
| 		default: | ||||
| 			return false | ||||
| 		} | ||||
| 	} | ||||
| 	if reflect.TypeOf(b).AssignableTo(targetType) { | ||||
| 		val.Elem().Set(reflect.ValueOf(b)) | ||||
| 		return true | ||||
| 	} | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| var ( | ||||
| 	brokerType   = reflect.TypeOf((*broker.Broker)(nil)).Elem() | ||||
| 	loggerType   = reflect.TypeOf((*logger.Logger)(nil)).Elem() | ||||
| 	clientType   = reflect.TypeOf((*client.Client)(nil)).Elem() | ||||
| 	serverType   = reflect.TypeOf((*server.Server)(nil)).Elem() | ||||
| 	codecType    = reflect.TypeOf((*codec.Codec)(nil)).Elem() | ||||
| 	flowType     = reflect.TypeOf((*flow.Flow)(nil)).Elem() | ||||
| 	fsmType      = reflect.TypeOf((*fsm.FSM)(nil)).Elem() | ||||
| 	meterType    = reflect.TypeOf((*meter.Meter)(nil)).Elem() | ||||
| 	registerType = reflect.TypeOf((*register.Register)(nil)).Elem() | ||||
| 	resolverType = reflect.TypeOf((*resolver.Resolver)(nil)).Elem() | ||||
| 	routerType   = reflect.TypeOf((*router.Router)(nil)).Elem() | ||||
| 	selectorType = reflect.TypeOf((*selector.Selector)(nil)).Elem() | ||||
| 	storeType    = reflect.TypeOf((*store.Store)(nil)).Elem() | ||||
| 	syncType     = reflect.TypeOf((*sync.Sync)(nil)).Elem() | ||||
| 	tracerType   = reflect.TypeOf((*tracer.Tracer)(nil)).Elem() | ||||
| 	serviceType  = reflect.TypeOf((*Service)(nil)).Elem() | ||||
| ) | ||||
							
								
								
									
										122
									
								
								micro_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										122
									
								
								micro_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,122 @@ | ||||
| package micro | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"fmt" | ||||
| 	"reflect" | ||||
| 	"testing" | ||||
|  | ||||
| 	"go.unistack.org/micro/v3/broker" | ||||
| 	"go.unistack.org/micro/v3/fsm" | ||||
| ) | ||||
|  | ||||
| func TestAs(t *testing.T) { | ||||
| 	var b *bro | ||||
| 	broTarget := &bro{name: "kafka"} | ||||
| 	fsmTarget := &fsmT{name: "fsm"} | ||||
|  | ||||
| 	testCases := []struct { | ||||
| 		b      any | ||||
| 		target any | ||||
| 		want   any | ||||
|  | ||||
| 		match bool | ||||
| 	}{ | ||||
| 		{ | ||||
| 			b:      broTarget, | ||||
| 			target: &b, | ||||
| 			match:  true, | ||||
| 			want:   broTarget, | ||||
| 		}, | ||||
| 		{ | ||||
| 			b:      nil, | ||||
| 			target: &b, | ||||
| 			match:  false, | ||||
| 			want:   nil, | ||||
| 		}, | ||||
| 		{ | ||||
| 			b:      fsmTarget, | ||||
| 			target: &b, | ||||
| 			match:  false, | ||||
| 			want:   nil, | ||||
| 		}, | ||||
| 	} | ||||
| 	for i, tc := range testCases { | ||||
| 		name := fmt.Sprintf("%d:As(Errorf(..., %v), %v)", i, tc.b, tc.target) | ||||
| 		// Clear the target pointer, in case it was set in a previous test. | ||||
| 		rtarget := reflect.ValueOf(tc.target) | ||||
| 		rtarget.Elem().Set(reflect.Zero(reflect.TypeOf(tc.target).Elem())) | ||||
| 		t.Run(name, func(t *testing.T) { | ||||
| 			match := As(tc.b, tc.target) | ||||
| 			if match != tc.match { | ||||
| 				t.Fatalf("match: got %v; want %v", match, tc.match) | ||||
| 			} | ||||
| 			if !match { | ||||
| 				return | ||||
| 			} | ||||
| 			if got := rtarget.Elem().Interface(); got != tc.want { | ||||
| 				t.Fatalf("got %#v, want %#v", got, tc.want) | ||||
| 			} | ||||
| 		}) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| type bro struct { | ||||
| 	name string | ||||
| } | ||||
|  | ||||
| func (p *bro) Name() string { return p.name } | ||||
|  | ||||
| func (p *bro) Live() bool { return true } | ||||
|  | ||||
| func (p *bro) Ready() bool { return true } | ||||
|  | ||||
| func (p *bro) Health() bool { return true } | ||||
|  | ||||
| func (p *bro) Init(_ ...broker.Option) error { return nil } | ||||
|  | ||||
| // Options returns broker options | ||||
| func (p *bro) Options() broker.Options { return broker.Options{} } | ||||
|  | ||||
| // Address return configured address | ||||
| func (p *bro) Address() string { return "" } | ||||
|  | ||||
| // Connect connects to broker | ||||
| func (p *bro) Connect(_ context.Context) error { return nil } | ||||
|  | ||||
| // Disconnect disconnect from broker | ||||
| func (p *bro) Disconnect(_ context.Context) error { return nil } | ||||
|  | ||||
| // Publish message, msg can be single broker.Message or []broker.Message | ||||
| func (p *bro) Publish(_ context.Context, _ string, _ *broker.Message, _ ...broker.PublishOption) error { | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // BatchPublish messages to broker with multiple topics | ||||
| func (p *bro) BatchPublish(_ context.Context, _ []*broker.Message, _ ...broker.PublishOption) error { | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // BatchSubscribe subscribes to topic messages via handler | ||||
| func (p *bro) BatchSubscribe(_ context.Context, _ string, _ broker.BatchHandler, _ ...broker.SubscribeOption) (broker.Subscriber, error) { | ||||
| 	return nil, nil | ||||
| } | ||||
|  | ||||
| // Subscribe subscribes to topic message via handler | ||||
| func (p *bro) Subscribe(_ context.Context, _ string, _ broker.Handler, _ ...broker.SubscribeOption) (broker.Subscriber, error) { | ||||
| 	return nil, nil | ||||
| } | ||||
|  | ||||
| // String type of broker | ||||
| func (p *bro) String() string { return p.name } | ||||
|  | ||||
| type fsmT struct { | ||||
| 	name string | ||||
| } | ||||
|  | ||||
| func (f *fsmT) Start(_ context.Context, _ interface{}, _ ...Option) (interface{}, error) { | ||||
| 	return nil, nil | ||||
| } | ||||
| func (f *fsmT) Current() string                 { return f.name } | ||||
| func (f *fsmT) Reset()                          {} | ||||
| func (f *fsmT) State(_ string, _ fsm.StateFunc) {} | ||||
| @@ -1,4 +1,4 @@ | ||||
| package mtls // import "go.unistack.org/micro/v3/mtls" | ||||
| package mtls | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
|   | ||||
| @@ -8,17 +8,18 @@ import ( | ||||
|  | ||||
| // CertificateOptions holds options for x509.CreateCertificate | ||||
| type CertificateOptions struct { | ||||
| 	Organization          []string | ||||
| 	OrganizationalUnit    []string | ||||
| 	CommonName            string | ||||
| 	OCSPServer            []string | ||||
| 	IssuingCertificateURL []string | ||||
| 	SerialNumber          *big.Int | ||||
| 	NotAfter              time.Time | ||||
| 	NotBefore             time.Time | ||||
| 	CommonName            string | ||||
| 	Organization          []string | ||||
| 	OrganizationalUnit    []string | ||||
| 	OCSPServer            []string | ||||
| 	IssuingCertificateURL []string | ||||
| 	ExtKeyUsage           []x509.ExtKeyUsage | ||||
|  | ||||
| 	SignatureAlgorithm x509.SignatureAlgorithm | ||||
| 	PublicKeyAlgorithm x509.PublicKeyAlgorithm | ||||
| 	ExtKeyUsage           []x509.ExtKeyUsage | ||||
| 	KeyUsage           x509.KeyUsage | ||||
| 	IsCA               bool | ||||
| } | ||||
|   | ||||
| @@ -1,5 +1,5 @@ | ||||
| // Package network is for creating internetworks | ||||
| package network // import "go.unistack.org/micro/v3/network" | ||||
| package network | ||||
|  | ||||
| import ( | ||||
| 	"go.unistack.org/micro/v3/client" | ||||
|   | ||||
| @@ -119,7 +119,7 @@ func Tracer(t tracer.Tracer) Option { | ||||
| // NewOptions returns network default options | ||||
| func NewOptions(opts ...Option) Options { | ||||
| 	options := Options{ | ||||
| 		ID:      id.Must(), | ||||
| 		ID:      id.MustNew(), | ||||
| 		Name:    "go.micro", | ||||
| 		Address: ":0", | ||||
| 		Logger:  logger.DefaultLogger, | ||||
|   | ||||
| @@ -1,5 +1,5 @@ | ||||
| // Package transport is an interface for synchronous connection based communication | ||||
| package transport // import "go.unistack.org/micro/v3/network/transport" | ||||
| package transport | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| @@ -10,7 +10,7 @@ import ( | ||||
|  | ||||
| var ( | ||||
| 	// DefaultTransport is the global default transport | ||||
| 	DefaultTransport Transport = NewTransport() | ||||
| 	DefaultTransport = NewTransport() | ||||
| 	// DefaultDialTimeout the default dial timeout | ||||
| 	DefaultDialTimeout = time.Second * 5 | ||||
| ) | ||||
|   | ||||
| @@ -1,5 +1,5 @@ | ||||
| // Package broker is a tunnel broker | ||||
| package broker // import "go.unistack.org/micro/v3/network/tunnel/broker" | ||||
| package broker | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| @@ -45,6 +45,18 @@ type ( | ||||
| 	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) | ||||
| @@ -72,7 +84,7 @@ 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 { | ||||
| func (t *tunBroker) BatchPublish(ctx context.Context, msgs []*broker.Message, _ ...broker.PublishOption) error { | ||||
| 	// TODO: this is probably inefficient, we might want to just maintain an open connection | ||||
| 	// it may be easier to add broadcast to the tunnel | ||||
| 	topicMap := make(map[string]tunnel.Session) | ||||
| @@ -102,7 +114,7 @@ func (t *tunBroker) BatchPublish(ctx context.Context, msgs []*broker.Message, op | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (t *tunBroker) Publish(ctx context.Context, topic string, m *broker.Message, opts ...broker.PublishOption) error { | ||||
| func (t *tunBroker) Publish(ctx context.Context, topic string, m *broker.Message, _ ...broker.PublishOption) error { | ||||
| 	// TODO: this is probably inefficient, we might want to just maintain an open connection | ||||
| 	// it may be easier to add broadcast to the tunnel | ||||
| 	c, err := t.tunnel.Dial(ctx, topic, tunnel.DialMode(tunnel.Multicast)) | ||||
| @@ -177,12 +189,12 @@ func (t *tunBatchSubscriber) run() { | ||||
| 		// receive message | ||||
| 		m := new(transport.Message) | ||||
| 		if err := c.Recv(m); err != nil { | ||||
| 			if logger.V(logger.ErrorLevel) { | ||||
| 				logger.Error(t.opts.Context, err.Error()) | ||||
| 			if logger.DefaultLogger.V(logger.ErrorLevel) { | ||||
| 				logger.DefaultLogger.Error(t.opts.Context, err.Error(), err) | ||||
| 			} | ||||
| 			if err = c.Close(); err != nil { | ||||
| 				if logger.V(logger.ErrorLevel) { | ||||
| 					logger.Error(t.opts.Context, err.Error()) | ||||
| 				if logger.DefaultLogger.V(logger.ErrorLevel) { | ||||
| 					logger.DefaultLogger.Error(t.opts.Context, err.Error(), err) | ||||
| 				} | ||||
| 			} | ||||
| 			continue | ||||
| @@ -222,12 +234,12 @@ func (t *tunSubscriber) run() { | ||||
| 		// receive message | ||||
| 		m := new(transport.Message) | ||||
| 		if err := c.Recv(m); err != nil { | ||||
| 			if logger.V(logger.ErrorLevel) { | ||||
| 				logger.Error(t.opts.Context, err.Error()) | ||||
| 			if logger.DefaultLogger.V(logger.ErrorLevel) { | ||||
| 				logger.DefaultLogger.Error(t.opts.Context, err.Error(), err) | ||||
| 			} | ||||
| 			if err = c.Close(); err != nil { | ||||
| 				if logger.V(logger.ErrorLevel) { | ||||
| 					logger.Error(t.opts.Context, err.Error()) | ||||
| 				if logger.DefaultLogger.V(logger.ErrorLevel) { | ||||
| 					logger.DefaultLogger.Error(t.opts.Context, err.Error(), err) | ||||
| 				} | ||||
| 			} | ||||
| 			continue | ||||
| @@ -305,6 +317,10 @@ 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...) | ||||
|   | ||||
| @@ -164,7 +164,7 @@ func DialWait(b bool) DialOption { | ||||
| // NewOptions returns router default options with filled values | ||||
| func NewOptions(opts ...Option) Options { | ||||
| 	options := Options{ | ||||
| 		ID:      id.Must(), | ||||
| 		ID:      id.MustNew(), | ||||
| 		Address: DefaultAddress, | ||||
| 		Token:   DefaultToken, | ||||
| 		Logger:  logger.DefaultLogger, | ||||
|   | ||||
| @@ -1,5 +1,5 @@ | ||||
| // Package transport provides a tunnel transport | ||||
| package transport // import "go.unistack.org/micro/v3/network/tunnel/transport" | ||||
| package transport | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
|   | ||||
| @@ -1,5 +1,5 @@ | ||||
| // Package tunnel provides gre network tunnelling | ||||
| package tunnel // import "go.unistack.org/micro/v3/network/transport/tunnel" | ||||
| package tunnel | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
|   | ||||
							
								
								
									
										15
									
								
								options.go
									
									
									
									
									
								
							
							
						
						
									
										15
									
								
								options.go
									
									
									
									
									
								
							| @@ -211,6 +211,7 @@ func Stores(s ...store.Store) Option { | ||||
| } | ||||
|  | ||||
| // Logger set the logger to use | ||||
| // | ||||
| //nolint:gocyclo | ||||
| func Logger(l logger.Logger, opts ...LoggerOption) Option { | ||||
| 	return func(o *Options) error { | ||||
| @@ -268,15 +269,7 @@ func Logger(l logger.Logger, opts ...LoggerOption) Option { | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 		for _, mtr := range o.Meters { | ||||
| 			for _, or := range lopts.meters { | ||||
| 				if mtr.Name() == or || all { | ||||
| 					if err = mtr.Init(meter.Logger(l)); err != nil { | ||||
| 						return err | ||||
| 					} | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		for _, trc := range o.Tracers { | ||||
| 			for _, ot := range lopts.tracers { | ||||
| 				if trc.Name() == ot || all { | ||||
| @@ -301,7 +294,7 @@ type loggerOptions struct { | ||||
| 	brokers   []string | ||||
| 	registers []string | ||||
| 	stores    []string | ||||
| 	meters    []string | ||||
| 	// meters    []string | ||||
| 	tracers []string | ||||
| } | ||||
|  | ||||
| @@ -329,6 +322,7 @@ func Meters(m ...meter.Meter) Option { | ||||
|  | ||||
| // Register sets the register for the service | ||||
| // and the underlying components | ||||
| // | ||||
| //nolint:gocyclo | ||||
| func Register(r register.Register, opts ...RegisterOption) Option { | ||||
| 	return func(o *Options) error { | ||||
| @@ -403,6 +397,7 @@ func RegisterBroker(n string) RegisterOption { | ||||
| } | ||||
|  | ||||
| // Tracer sets the tracer | ||||
| // | ||||
| //nolint:gocyclo | ||||
| func Tracer(t tracer.Tracer, opts ...TracerOption) Option { | ||||
| 	return func(o *Options) error { | ||||
|   | ||||
Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user