Compare commits
	
		
			392 Commits
		
	
	
		
			v3.10.1
			...
			6c9dbc77dd
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 6c9dbc77dd | |||
| 70f0ace92e | |||
| 3f21bafc2f | |||
| a9ed8b16c1 | |||
|  | 740cd5931d | ||
| 85a78063d0 | |||
| 604ad9cd9d | |||
| 91137537a2 | |||
| 950e2352fd | |||
| 0bb29b29cf | |||
| 17bcd0b0ab | |||
| 20f9f4da3b | |||
| 66fa04b8dc | |||
| 1ef3ad6531 | |||
| c95a91349d | |||
| fdcf8e6ca4 | |||
| 8cb2d9db4a | |||
| 04da4388ac | |||
| 79fb23e644 | |||
| f8fe923ab1 | |||
| 105f56dbfe | |||
| 9fed5a368b | |||
| 7374d41cf8 | |||
| a4a8935c1f | |||
| 5f498c8232 | |||
| a00fdf679b | |||
| dc9ebe4155 | |||
| 87ced484b7 | |||
| af99b11a59 | |||
| 2724b51f7c | |||
| 5b5d0e02b9 | |||
| afc2de6819 | |||
| 32a8ab9c05 | |||
|  | 7e5401bded | ||
| 64b91cea06 | |||
|  | 0f59fdcbde | ||
| 50979e6708 | |||
| 46f3108870 | |||
|  | 5fed91a65f | ||
| 1c5bba908d | |||
|  | bc8ebdcad5 | ||
| fc24f3af92 | |||
| 1058177d1c | |||
|  | fa53fac085 | ||
| 8c060df5e3 | |||
| e1f8c62685 | |||
| 562b1ab9b7 | |||
|  | f3c877a37b | ||
| 0999b2ad78 | |||
| a365513177 | |||
|  | d1e3f3cab2 | ||
| ec94a09417 | |||
| 1728b88e06 | |||
| d3c31da9db | |||
| 59095681be | |||
| f11ceba225 | |||
| ffa01de78f | |||
| 816abc2bbc | |||
| f3f2a9b737 | |||
| 3f82cb3ba4 | |||
|  | 306b7a3962 | ||
| a8eda9d58d | |||
| 7e4477dcb4 | |||
|  | d846044fc6 | ||
| 29d956e74e | |||
| fcc4faff8a | |||
| 5df8f83f45 | |||
|  | 27fa6e9173 | ||
| bd55a35dc3 | |||
| 653bd386cc | |||
|  | 558c6f4d7c | ||
| d7dd6fbeb2 | |||
| a00cf2c8d9 | |||
|  | a3e8ab2492 | ||
| 06da500ef4 | |||
| 277f04ba19 | |||
|  | 470263ff5f | ||
| b8232e02be | |||
|  | f8c5e10c1d | ||
| 397e71f815 | |||
| 74e31d99f6 | |||
| f39de15d93 | |||
| d291102877 | |||
| 37ffbb18d8 | |||
| 9a85dead86 | |||
| a489aab1c3 | |||
| d160664ef1 | |||
| fa868edcaa | |||
|  | 6ed0b0e090 | ||
| 533b265d19 | |||
| 1ace2631a4 | |||
| 3dd5ca68d1 | |||
| 66ccd6021f | |||
| e5346f7e4f | |||
|  | daf19f031a | ||
| 5989fd54ca | |||
| ed30c26324 | |||
| 0f8f93d09a | |||
|  | f460e2f8dd | ||
| 70d6a79274 | |||
| 664b1586af | |||
| 8d747c64a8 | |||
| 94beb5ed3b | |||
| 98981ba86c | |||
| 1013f50d0e | |||
| 0b190997b1 | |||
| 69a44eb190 | |||
| 0476028f69 | |||
| 330d8b149a | |||
| 19b04fe070 | |||
| 4cd55875c6 | |||
| a7896cc728 | |||
| ff991bf49c | |||
| 5a6551b703 | |||
| 9406a33d60 | |||
| 8f185abd9d | |||
| 86492e0644 | |||
| b21972964a | |||
| f5ee065d09 | |||
| 8cb02f2b08 | |||
| bc926cd6bd | |||
| 356abfd818 | |||
| 18444d3f98 | |||
| d5f07922e8 | |||
| 675e717410 | |||
| 7b6aea235a | |||
| 2cb7200467 | |||
| f430f97a97 | |||
| 0060c4377a | |||
| e46579fe9a | |||
| ca52973194 | |||
| 5bb33c7e1d | |||
| b71fc25328 | |||
| 9345dd075a | |||
| 1c1b9c0a28 | |||
| 2969494c5a | |||
| cbd3fa38ba | |||
| 569a36383d | |||
| 90bed77526 | |||
| ba4478a5e0 | |||
| 6dc76cdfea | |||
| e266683d96 | |||
| 2b62ad04f2 | |||
| 275b0a64e5 | |||
| 38c5fe8b5a | |||
| b6a0e4d983 | |||
| d9b822deff | |||
| 0e66688f8f | |||
| 9213fd212f | |||
| aa360dcf51 | |||
| 2df259b5b8 | |||
| 15e9310368 | |||
|  | 16d8cf3434 | ||
| 9704ef2e5e | |||
| 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 | |||
| 7c29afba0b | |||
| 8159b9d233 | |||
| 45cdac5c29 | |||
| 98db0dc8bc | |||
| 453d2232bd | |||
| 9b387312da | |||
| 84024f7713 | |||
| 5a554f9f0c | |||
| 9c33cbc8e2 | |||
| 848fe1c0d4 | |||
| 6cbf23fec5 | |||
| 7462b0b53c | |||
| cb743cee3f | |||
| d18952951c | |||
|  | f6b7f1b4bc | ||
|  | 33fa702104 | ||
| 4debc392d1 | |||
| 7137d99102 | |||
| f76b3171d9 | |||
| db3fc42204 | |||
| f59023d741 | |||
| 6f17fd891a | |||
| fd93308e8e | |||
| 2aaa0a74d8 | 
							
								
								
									
										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 }} | ||||||
							
								
								
									
										54
									
								
								.gitea/workflows/job_coverage.yml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										54
									
								
								.gitea/workflows/job_coverage.yml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,54 @@ | |||||||
|  | name: coverage | ||||||
|  |  | ||||||
|  | on: | ||||||
|  |   push: | ||||||
|  |     branches: [ main, v3, v4 ] | ||||||
|  |     paths-ignore: | ||||||
|  |       - '.github/**' | ||||||
|  |       - '.gitea/**' | ||||||
|  |   pull_request: | ||||||
|  |     branches: [ main, v3, v4 ] | ||||||
|  |   # Allows you to run this workflow manually from the Actions tab | ||||||
|  |   workflow_dispatch: | ||||||
|  |  | ||||||
|  | jobs: | ||||||
|  |  | ||||||
|  |   build: | ||||||
|  |     runs-on: ubuntu-latest | ||||||
|  |     steps: | ||||||
|  |     - name: checkout code | ||||||
|  |       uses: actions/checkout@v4 | ||||||
|  |       with: | ||||||
|  |         filter: 'blob:none' | ||||||
|  |  | ||||||
|  |     - name: setup go | ||||||
|  |       uses: actions/setup-go@v5 | ||||||
|  |       with: | ||||||
|  |         cache-dependency-path: "**/*.sum" | ||||||
|  |         go-version: 'stable'  | ||||||
|  |  | ||||||
|  |     - name: test coverage | ||||||
|  |       run: | | ||||||
|  |         go test -v -cover ./... -covermode=count -coverprofile coverage.out -coverpkg ./... | ||||||
|  |         go tool cover -func coverage.out -o coverage.out | ||||||
|  |  | ||||||
|  |     - name: coverage badge | ||||||
|  |       uses: tj-actions/coverage-badge-go@v2 | ||||||
|  |       with: | ||||||
|  |         green: 80 | ||||||
|  |         filename: coverage.out | ||||||
|  |  | ||||||
|  |     - uses: stefanzweifel/git-auto-commit-action@v4 | ||||||
|  |       name: autocommit | ||||||
|  |       with: | ||||||
|  |         commit_message: Apply Code Coverage Badge | ||||||
|  |         skip_fetch: true | ||||||
|  |         skip_checkout: true | ||||||
|  |         file_pattern: ./README.md | ||||||
|  |  | ||||||
|  |     - name: push | ||||||
|  |       if: steps.auto-commit-action.outputs.changes_detected == 'true' | ||||||
|  |       uses: ad-m/github-push-action@master | ||||||
|  |       with: | ||||||
|  |         github_token: ${{ github.token }} | ||||||
|  |         branch: ${{ github.ref }} | ||||||
							
								
								
									
										29
									
								
								.gitea/workflows/job_lint.yml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										29
									
								
								.gitea/workflows/job_lint.yml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,29 @@ | |||||||
|  | name: lint | ||||||
|  |  | ||||||
|  | on: | ||||||
|  |   pull_request: | ||||||
|  |     types: [opened, reopened, synchronize] | ||||||
|  |     branches: [ master, v3, v4 ] | ||||||
|  |     paths-ignore: | ||||||
|  |       - '.github/**' | ||||||
|  |       - '.gitea/**' | ||||||
|  |  | ||||||
|  | jobs: | ||||||
|  |   lint: | ||||||
|  |     runs-on: ubuntu-latest | ||||||
|  |     steps: | ||||||
|  |     - name: checkout code | ||||||
|  |       uses: actions/checkout@v4 | ||||||
|  |       with: | ||||||
|  |         filter: 'blob:none' | ||||||
|  |     - name: setup go | ||||||
|  |       uses: actions/setup-go@v5 | ||||||
|  |       with: | ||||||
|  |         cache-dependency-path: "**/*.sum" | ||||||
|  |         go-version: 'stable'  | ||||||
|  |     - name: setup deps | ||||||
|  |       run: go get -v ./... | ||||||
|  |     - name: run lint | ||||||
|  |       uses: https://github.com/golangci/golangci-lint-action@v6 | ||||||
|  |       with: | ||||||
|  |         version: 'latest' | ||||||
							
								
								
									
										54
									
								
								.gitea/workflows/job_syncpull.yml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										54
									
								
								.gitea/workflows/job_syncpull.yml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,54 @@ | |||||||
|  | name: syncpull | ||||||
|  |  | ||||||
|  | on: | ||||||
|  |   schedule: | ||||||
|  |     - cron: '* * * * *' | ||||||
|  |   # Allows you to run this workflow manually from the Actions tab | ||||||
|  |   workflow_dispatch: | ||||||
|  |  | ||||||
|  | jobs: | ||||||
|  |   pull: | ||||||
|  |     runs-on: ubuntu-latest | ||||||
|  |     steps: | ||||||
|  |     - name: init | ||||||
|  |       run: | | ||||||
|  |         git config --global user.email "vtolstov <vtolstov@users.noreply.github.com>" | ||||||
|  |         git config --global user.name "github-actions[bot]" | ||||||
|  |         echo "machine git.unistack.org login vtolstov password ${{ secrets.TOKEN_GITEA }}" | tee -a /root/.netrc | ||||||
|  |         echo "machine github.com login vtolstov password ${{ secrets.TOKEN_GITHUB }}" | tee -a /root/.netrc | ||||||
|  |  | ||||||
|  |     - name: track master | ||||||
|  |       run: | | ||||||
|  |         git clone --depth=10 --branch master --single-branch ${GITHUB_SERVER_URL}/${GITHUB_REPOSITORY} repo | ||||||
|  |         cd repo | ||||||
|  |         git remote add --no-tags --fetch --track master upstream https://github.com/${GITHUB_REPOSITORY} | ||||||
|  |         git pull --rebase upstream master | ||||||
|  |         git push upstream master --progress | ||||||
|  |         git merge --allow-unrelated-histories "upstream/master" | ||||||
|  |         git push origin master --progress | ||||||
|  |         cd ../ | ||||||
|  |         rm -rf repo | ||||||
|  |  | ||||||
|  |     - name: track v3 | ||||||
|  |       run: | | ||||||
|  |         git clone --depth=10 --branch v3 --single-branch ${GITHUB_SERVER_URL}/${GITHUB_REPOSITORY} repo | ||||||
|  |         cd repo | ||||||
|  |         git remote add --no-tags --fetch --track v3 upstream https://github.com/${GITHUB_REPOSITORY} | ||||||
|  |         git pull --rebase upstream v3 | ||||||
|  |         git push upstream v3 | ||||||
|  |         git merge --allow-unrelated-histories "upstream/v3" | ||||||
|  |         git push origin v3 --progress | ||||||
|  |         cd ../ | ||||||
|  |         rm -rf repo | ||||||
|  |  | ||||||
|  |     - name: track v4 | ||||||
|  |       run: | | ||||||
|  |         git clone --depth=10 --branch v4 --single-branch ${GITHUB_SERVER_URL}/${GITHUB_REPOSITORY} repo | ||||||
|  |         cd repo | ||||||
|  |         git remote add --no-tags --fetch --track v4 upstream https://github.com/${GITHUB_REPOSITORY} | ||||||
|  |         git pull --rebase upstream v4 | ||||||
|  |         git push upstream v4 | ||||||
|  |         git merge --allow-unrelated-histories "upstream/v4" | ||||||
|  |         git push origin v4 --progress | ||||||
|  |         cd ../ | ||||||
|  |         rm -rf repo | ||||||
							
								
								
									
										31
									
								
								.gitea/workflows/job_test.yml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										31
									
								
								.gitea/workflows/job_test.yml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,31 @@ | |||||||
|  | name: test | ||||||
|  |  | ||||||
|  | on: | ||||||
|  |   pull_request: | ||||||
|  |     types: [opened, reopened, synchronize] | ||||||
|  |     branches: [ master, v3, v4 ] | ||||||
|  |   push: | ||||||
|  |     branches: [ master, v3, v4 ] | ||||||
|  |     paths-ignore: | ||||||
|  |       - '.github/**' | ||||||
|  |       - '.gitea/**' | ||||||
|  |  | ||||||
|  | jobs: | ||||||
|  |   test: | ||||||
|  |     runs-on: ubuntu-latest | ||||||
|  |     steps: | ||||||
|  |     - name: checkout code | ||||||
|  |       uses: actions/checkout@v4 | ||||||
|  |       with: | ||||||
|  |         filter: 'blob:none' | ||||||
|  |     - name: setup go | ||||||
|  |       uses: actions/setup-go@v5 | ||||||
|  |       with: | ||||||
|  |         cache-dependency-path: "**/*.sum" | ||||||
|  |         go-version: 'stable' | ||||||
|  |     - name: setup deps | ||||||
|  |       run: go get -v ./... | ||||||
|  |     - name: run test | ||||||
|  |       env: | ||||||
|  |         INTEGRATION_TESTS: yes | ||||||
|  |       run: go test -mod readonly -v ./... | ||||||
							
								
								
									
										50
									
								
								.gitea/workflows/job_tests.yml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										50
									
								
								.gitea/workflows/job_tests.yml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,50 @@ | |||||||
|  | name: test | ||||||
|  |  | ||||||
|  | on: | ||||||
|  |   pull_request: | ||||||
|  |     types: [opened, reopened, synchronize] | ||||||
|  |     branches: [ master, v3, v4 ] | ||||||
|  |   push: | ||||||
|  |     branches: [ master, v3, v4 ] | ||||||
|  |     paths-ignore: | ||||||
|  |       - '.github/**' | ||||||
|  |       - '.gitea/**' | ||||||
|  |  | ||||||
|  | jobs: | ||||||
|  |   test: | ||||||
|  |     runs-on: ubuntu-latest | ||||||
|  |     steps: | ||||||
|  |     - name: checkout code | ||||||
|  |       uses: actions/checkout@v4 | ||||||
|  |       with: | ||||||
|  |         filter: 'blob:none' | ||||||
|  |     - name: checkout tests | ||||||
|  |       uses: actions/checkout@v4 | ||||||
|  |       with: | ||||||
|  |         ref: master | ||||||
|  |         filter: 'blob:none' | ||||||
|  |         repository: unistack-org/micro-tests | ||||||
|  |         path: micro-tests | ||||||
|  |     - name: setup go | ||||||
|  |       uses: actions/setup-go@v5 | ||||||
|  |       with: | ||||||
|  |         cache-dependency-path: "**/*.sum" | ||||||
|  |         go-version: 'stable' | ||||||
|  |     - name: setup go work | ||||||
|  |       env: | ||||||
|  |         GOWORK: /workspace/${{ github.repository_owner }}/go.work | ||||||
|  |       run: | | ||||||
|  |         go work init | ||||||
|  |         go work use . | ||||||
|  |         go work use micro-tests | ||||||
|  |     - name: setup deps | ||||||
|  |       env: | ||||||
|  |         GOWORK: /workspace/${{ github.repository_owner }}/go.work | ||||||
|  |       run: go get -v ./... | ||||||
|  |     - name: run tests | ||||||
|  |       env: | ||||||
|  |         INTEGRATION_TESTS: yes | ||||||
|  |         GOWORK: /workspace/${{ github.repository_owner }}/go.work | ||||||
|  |       run: | | ||||||
|  |         cd micro-tests | ||||||
|  |         go test -mod readonly -v ./... || true | ||||||
							
								
								
									
										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** | ||||||
| @@ -34,7 +34,7 @@ jobs: | |||||||
|       - name: checkout |       - name: checkout | ||||||
|         uses: actions/checkout@v3 |         uses: actions/checkout@v3 | ||||||
|       - name: lint |       - name: lint | ||||||
|         uses: golangci/golangci-lint-action@v3.3.1 |         uses: golangci/golangci-lint-action@v3.4.0 | ||||||
|         continue-on-error: true |         continue-on-error: true | ||||||
|         with: |         with: | ||||||
|           # Required: the version of golangci-lint is required and must be specified without patch version: we always use the latest patch version. |           # Required: the version of golangci-lint is required and must be specified without patch version: we always use the latest patch version. | ||||||
| @@ -15,7 +15,7 @@ jobs: | |||||||
|     steps: |     steps: | ||||||
|       - name: metadata |       - name: metadata | ||||||
|         id: metadata |         id: metadata | ||||||
|         uses: dependabot/fetch-metadata@v1.3.5 |         uses: dependabot/fetch-metadata@v1.3.6 | ||||||
|         with: |         with: | ||||||
|           github-token: "${{ secrets.TOKEN }}" |           github-token: "${{ secrets.TOKEN }}" | ||||||
|       - name: merge |       - name: merge | ||||||
| @@ -34,14 +34,7 @@ jobs: | |||||||
|       - name: checkout |       - name: checkout | ||||||
|         uses: actions/checkout@v3 |         uses: actions/checkout@v3 | ||||||
|       - name: lint |       - name: lint | ||||||
|         uses: golangci/golangci-lint-action@v3.3.1 |         uses: golangci/golangci-lint-action@v3.4.0 | ||||||
|         continue-on-error: true |         continue-on-error: true | ||||||
|         with: |         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 | ||||||
|           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 | # Develop tools | ||||||
| /.vscode/ | /.vscode/ | ||||||
| /.idea/ | /.idea/ | ||||||
|  | .idea | ||||||
|  | .vscode | ||||||
|  |  | ||||||
| # Binaries for programs and plugins | # Binaries for programs and plugins | ||||||
| *.exe | *.exe | ||||||
| @@ -13,6 +15,7 @@ | |||||||
| _obj | _obj | ||||||
| _test | _test | ||||||
| _build | _build | ||||||
|  | .DS_Store | ||||||
|  |  | ||||||
| # Architecture specific extensions/prefixes | # Architecture specific extensions/prefixes | ||||||
| *.[568vq] | *.[568vq] | ||||||
|   | |||||||
| @@ -1,44 +1,5 @@ | |||||||
| run: | run: | ||||||
|   concurrency: 4 |   concurrency: 8 | ||||||
|   deadline: 5m |   timeout: 5m | ||||||
|   issues-exit-code: 1 |   issues-exit-code: 1 | ||||||
|   tests: true |   tests: true | ||||||
|  |  | ||||||
| linters-settings: |  | ||||||
|   govet: |  | ||||||
|     check-shadowing: true |  | ||||||
|     enable: |  | ||||||
|       - fieldalignment |  | ||||||
|  |  | ||||||
| linters: |  | ||||||
|   enable: |  | ||||||
|     - govet |  | ||||||
|     - deadcode |  | ||||||
|     - errcheck |  | ||||||
|     - govet |  | ||||||
|     - ineffassign |  | ||||||
|     - staticcheck |  | ||||||
|     - structcheck |  | ||||||
|     - typecheck |  | ||||||
|     - unused |  | ||||||
|     - varcheck |  | ||||||
|     - bodyclose |  | ||||||
|     - gci |  | ||||||
|     - goconst |  | ||||||
|     - gocritic |  | ||||||
|     - gosimple |  | ||||||
|     - gofmt |  | ||||||
|     - gofumpt |  | ||||||
|     - goimports |  | ||||||
|     - revive |  | ||||||
|     - gosec |  | ||||||
|     - makezero |  | ||||||
|     - misspell |  | ||||||
|     - nakedret |  | ||||||
|     - nestif |  | ||||||
|     - nilerr |  | ||||||
|     - noctx |  | ||||||
|     - prealloc |  | ||||||
|     - unconvert |  | ||||||
|     - unparam |  | ||||||
|   disable-all: false |  | ||||||
|   | |||||||
							
								
								
									
										27
									
								
								README.md
									
									
									
									
									
								
							
							
						
						
									
										27
									
								
								README.md
									
									
									
									
									
								
							| @@ -1,4 +1,9 @@ | |||||||
| # Micro [](https://opensource.org/licenses/Apache-2.0) [](https://pkg.go.dev/github.com/unistack-org/micro/v3?tab=overview) [](https://github.com/unistack-org/micro/actions?query=workflow%3Abuild+branch%3Amaster+event%3Apush) [](https://goreportcard.com/report/go.unistack.org/micro/v3) [](https://codecov.io/gh/unistack-org/micro) | # Micro | ||||||
|  |  | ||||||
|  | [](https://opensource.org/licenses/Apache-2.0) | ||||||
|  | [](https://pkg.go.dev/go.unistack.org/micro/v4?tab=overview) | ||||||
|  | [](https://git.unistack.org/unistack-org/micro/actions?query=workflow%3Abuild+branch%3Av4+event%3Apush) | ||||||
|  | [](https://goreportcard.com/report/go.unistack.org/micro/v4) | ||||||
|  |  | ||||||
| Micro is a standard library for microservices. | Micro is a standard library for microservices. | ||||||
|  |  | ||||||
| @@ -10,30 +15,20 @@ Micro provides the core requirements for distributed systems development includi | |||||||
|  |  | ||||||
| Micro abstracts away the details of distributed systems. Here are the main features. | Micro abstracts away the details of distributed systems. Here are the main features. | ||||||
|  |  | ||||||
| - **Authentication** - Auth is built in as a first class citizen. Authentication and authorization enable secure  |  | ||||||
| zero trust networking by providing every service an identity and certificates. This additionally includes rule  |  | ||||||
| based access control. |  | ||||||
|  |  | ||||||
| - **Dynamic Config** - Load and hot reload dynamic config from anywhere. The config interface provides a way to load application  | - **Dynamic Config** - Load and hot reload dynamic config from anywhere. The config interface provides a way to load application  | ||||||
| level config from any source such as env vars, file, etcd. You can merge the sources and even define fallbacks. | level config from any source such as env vars, cmdline, file, consul, vault... You can merge the sources and even define fallbacks. | ||||||
|  |  | ||||||
| - **Data Storage** - A simple data store interface to read, write and delete records. It includes support for memory, file and  | - **Data Storage** - A simple data store interface to read, write and delete records. It includes support for memory, file and  | ||||||
| CockroachDB by default. State and persistence becomes a core requirement beyond prototyping and Micro looks to build that into the framework. | s3. State and persistence becomes a core requirement beyond prototyping and Micro looks to build that into the framework. | ||||||
|  |  | ||||||
| - **Service Discovery** - Automatic service registration and name resolution. Service discovery is at the core of micro service  | - **Service Discovery** - Automatic service registration and name resolution. Service discovery is at the core of micro service  | ||||||
| development. When service A needs to speak to service B it needs the location of that service. | development. When service A needs to speak to service B it needs the location of that service. | ||||||
|  |  | ||||||
| - **Load Balancing** - Client side load balancing built on service discovery. Once we have the addresses of any number of instances  |  | ||||||
| of a service we now need a way to decide which node to route to. We use random hashed load balancing to provide even distribution  |  | ||||||
| across the services and retry a different node if there's a problem.  |  | ||||||
|  |  | ||||||
| - **Message Encoding** - Dynamic message encoding based on content-type. The client and server will use codecs along with content-type  | - **Message Encoding** - Dynamic message encoding based on content-type. The client and server will use codecs along with content-type  | ||||||
| to seamlessly encode and decode Go types for you. Any variety of messages could be encoded and sent from different clients. The client  | to seamlessly encode and decode Go types for you. Any variety of messages could be encoded and sent from different clients. The client  | ||||||
| and server handle this by default. | and server handle this by default. | ||||||
|  |  | ||||||
| - **Transport** - gRPC or http based request/response with support for bidirectional streaming. We provide an abstraction for synchronous communication. A request made to a service will be automatically resolved, load balanced, dialled and streamed. | - **Async Messaging** - Pub/Sub is built in as a first class citizen for asynchronous communication and event driven architectures. | ||||||
|  |  | ||||||
| - **Async Messaging** - PubSub is built in as a first class citizen for asynchronous communication and event driven architectures. |  | ||||||
| Event notifications are a core pattern in micro service development. | Event notifications are a core pattern in micro service development. | ||||||
|  |  | ||||||
| - **Synchronization** - Distributed systems are often built in an eventually consistent manner. Support for distributed locking and  | - **Synchronization** - Distributed systems are often built in an eventually consistent manner. Support for distributed locking and  | ||||||
| @@ -42,10 +37,6 @@ leadership are built in as a Sync interface. When using an eventually consistent | |||||||
| - **Pluggable Interfaces** - Micro makes use of Go interfaces for each system abstraction. Because of this these interfaces  | - **Pluggable Interfaces** - Micro makes use of Go interfaces for each system abstraction. Because of this these interfaces  | ||||||
| are pluggable and allows Micro to be runtime agnostic. | are pluggable and allows Micro to be runtime agnostic. | ||||||
|  |  | ||||||
| ## Getting Started |  | ||||||
|  |  | ||||||
| To be created. |  | ||||||
|  |  | ||||||
| ## License | ## License | ||||||
|  |  | ||||||
| Micro is Apache 2.0 licensed. | Micro is Apache 2.0 licensed. | ||||||
|   | |||||||
							
								
								
									
										15
									
								
								SECURITY.md
									
									
									
									
									
								
							
							
						
						
									
										15
									
								
								SECURITY.md
									
									
									
									
									
								
							| @@ -1,15 +0,0 @@ | |||||||
| # Security Policy |  | ||||||
|  |  | ||||||
| ## Supported Versions |  | ||||||
|  |  | ||||||
| Use this section to tell people about which versions of your project are |  | ||||||
| currently being supported with security updates. |  | ||||||
|  |  | ||||||
| | Version | Supported          | |  | ||||||
| | ------- | ------------------ | |  | ||||||
| | 3.7.x   | :white_check_mark: | |  | ||||||
| | < 3.7.0 | :x:                | |  | ||||||
|  |  | ||||||
| ## Reporting a Vulnerability |  | ||||||
|  |  | ||||||
| If you find any issue, please create github issue in this repo |  | ||||||
							
								
								
									
										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") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
							
								
								
									
										121
									
								
								broker/broker.go
									
									
									
									
									
								
							
							
						
						
									
										121
									
								
								broker/broker.go
									
									
									
									
									
								
							| @@ -1,21 +1,29 @@ | |||||||
| // Package broker is an interface used for asynchronous messaging | // Package broker is an interface used for asynchronous messaging | ||||||
| package broker // import "go.unistack.org/micro/v3/broker" | package broker | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"errors" | 	"errors" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/metadata" | 	"go.unistack.org/micro/v4/codec" | ||||||
|  | 	"go.unistack.org/micro/v4/metadata" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // DefaultBroker default memory broker | // DefaultBroker default memory broker | ||||||
| var DefaultBroker = NewBroker() | var DefaultBroker Broker = NewBroker() | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| 	// ErrNotConnected returns when broker used but not connected yet | 	// ErrNotConnected returns when broker used but not connected yet | ||||||
| 	ErrNotConnected = errors.New("broker not connected") | 	ErrNotConnected = errors.New("broker not connected") | ||||||
| 	// ErrDisconnected returns when broker disconnected | 	// ErrDisconnected returns when broker disconnected | ||||||
| 	ErrDisconnected = errors.New("broker disconnected") | 	ErrDisconnected = errors.New("broker disconnected") | ||||||
|  | 	// ErrInvalidMessage returns when invalid Message passed | ||||||
|  | 	ErrInvalidMessage = errors.New("invalid message") | ||||||
|  | 	// ErrInvalidHandler returns when subscriber passed to Subscribe | ||||||
|  | 	ErrInvalidHandler = errors.New("invalid handler, ony func(Message) error and func([]Message) error supported") | ||||||
|  | 	// DefaultGracefulTimeout | ||||||
|  | 	DefaultGracefulTimeout = 5 * time.Second | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Broker is an interface used for asynchronous messaging. | // Broker is an interface used for asynchronous messaging. | ||||||
| @@ -32,93 +40,44 @@ type Broker interface { | |||||||
| 	Connect(ctx context.Context) error | 	Connect(ctx context.Context) error | ||||||
| 	// Disconnect disconnect from broker | 	// Disconnect disconnect from broker | ||||||
| 	Disconnect(ctx context.Context) error | 	Disconnect(ctx context.Context) error | ||||||
|  | 	// NewMessage create new broker message to publish. | ||||||
|  | 	NewMessage(ctx context.Context, hdr metadata.Metadata, body interface{}, opts ...PublishOption) (Message, error) | ||||||
| 	// Publish message to broker topic | 	// Publish message to broker topic | ||||||
| 	Publish(ctx context.Context, topic string, msg *Message, opts ...PublishOption) error | 	Publish(ctx context.Context, topic string, messages ...Message) error | ||||||
| 	// Subscribe subscribes to topic message via handler | 	// Subscribe subscribes to topic message via handler | ||||||
| 	Subscribe(ctx context.Context, topic string, h Handler, opts ...SubscribeOption) (Subscriber, error) | 	Subscribe(ctx context.Context, topic string, handler interface{}, opts ...SubscribeOption) (Subscriber, error) | ||||||
| 	// BatchPublish messages to broker with multiple topics |  | ||||||
| 	BatchPublish(ctx context.Context, msgs []*Message, opts ...PublishOption) error |  | ||||||
| 	// BatchSubscribe subscribes to topic messages via handler |  | ||||||
| 	BatchSubscribe(ctx context.Context, topic string, h BatchHandler, opts ...SubscribeOption) (Subscriber, error) |  | ||||||
| 	// String type of broker | 	// String type of broker | ||||||
| 	String() string | 	String() string | ||||||
|  | 	// Live returns broker liveness | ||||||
|  | 	Live() bool | ||||||
|  | 	// Ready returns broker readiness | ||||||
|  | 	Ready() bool | ||||||
|  | 	// Health returns broker health | ||||||
|  | 	Health() bool | ||||||
| } | } | ||||||
|  |  | ||||||
| // Handler is used to process messages via a subscription of a topic. | type ( | ||||||
| type Handler func(Event) error | 	FuncPublish   func(ctx context.Context, topic string, messages ...Message) error | ||||||
|  | 	HookPublish   func(next FuncPublish) FuncPublish | ||||||
|  | 	FuncSubscribe func(ctx context.Context, topic string, handler interface{}, opts ...SubscribeOption) (Subscriber, error) | ||||||
|  | 	HookSubscribe func(next FuncSubscribe) FuncSubscribe | ||||||
|  | ) | ||||||
|  |  | ||||||
| // Events contains multiple events | // Message is given to a subscription handler for processing | ||||||
| type Events []Event | type Message interface { | ||||||
|  | 	// Context for the message. | ||||||
| // Ack try to ack all events and return | 	Context() context.Context | ||||||
| func (evs Events) Ack() error { | 	// Topic returns message destination topic. | ||||||
| 	var err error |  | ||||||
| 	for _, ev := range evs { |  | ||||||
| 		if err = ev.Ack(); err != nil { |  | ||||||
| 			return err |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // SetError sets error on event |  | ||||||
| func (evs Events) SetError(err error) { |  | ||||||
| 	for _, ev := range evs { |  | ||||||
| 		ev.SetError(err) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // BatchHandler is used to process messages in batches via a subscription of a topic. |  | ||||||
| type BatchHandler func(Events) error |  | ||||||
|  |  | ||||||
| // Event is given to a subscription handler for processing |  | ||||||
| type Event interface { |  | ||||||
| 	// Topic returns event topic |  | ||||||
| 	Topic() string | 	Topic() string | ||||||
| 	// Message returns broker message | 	// Header returns message headers. | ||||||
| 	Message() *Message | 	Header() metadata.Metadata | ||||||
| 	// Ack acknowledge message | 	// Body returns broker message []byte slice | ||||||
|  | 	Body() []byte | ||||||
|  | 	// Unmarshal try to decode message body to dst. | ||||||
|  | 	// This is helper method that uses codec.Unmarshal. | ||||||
|  | 	Unmarshal(dst interface{}, opts ...codec.Option) error | ||||||
|  | 	// Ack acknowledge message if supported. | ||||||
| 	Ack() error | 	Ack() error | ||||||
| 	// Error returns message error (like decoding errors or some other) |  | ||||||
| 	Error() error |  | ||||||
| 	// SetError set event processing error |  | ||||||
| 	SetError(err error) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // 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 |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewMessage create broker message with topic filled |  | ||||||
| func NewMessage(topic string) *Message { |  | ||||||
| 	m := &Message{Header: metadata.New(2)} |  | ||||||
| 	m.Header.Set(metadata.HeaderTopic, topic) |  | ||||||
| 	return m |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // Subscriber is a convenience return type for the Subscribe method | // Subscriber is a convenience return type for the Subscribe method | ||||||
|   | |||||||
| @@ -15,6 +15,15 @@ func FromContext(ctx context.Context) (Broker, bool) { | |||||||
| 	return c, ok | 	return c, ok | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // MustContext returns broker from passed context | ||||||
|  | func MustContext(ctx context.Context) Broker { | ||||||
|  | 	b, ok := FromContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		panic("missing broker") | ||||||
|  | 	} | ||||||
|  | 	return b | ||||||
|  | } | ||||||
|  |  | ||||||
| // NewContext savess broker in context | // NewContext savess broker in context | ||||||
| func NewContext(ctx context.Context, s Broker) context.Context { | func NewContext(ctx context.Context, s Broker) context.Context { | ||||||
| 	if ctx == nil { | 	if ctx == nil { | ||||||
| @@ -33,16 +42,6 @@ func SetSubscribeOption(k, v interface{}) SubscribeOption { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // SetOption returns a function to setup a context with given value |  | ||||||
| func SetOption(k, v interface{}) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		if o.Context == nil { |  | ||||||
| 			o.Context = context.Background() |  | ||||||
| 		} |  | ||||||
| 		o.Context = context.WithValue(o.Context, k, v) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // SetPublishOption returns a function to setup a context with given value | // SetPublishOption returns a function to setup a context with given value | ||||||
| func SetPublishOption(k, v interface{}) PublishOption { | func SetPublishOption(k, v interface{}) PublishOption { | ||||||
| 	return func(o *PublishOptions) { | 	return func(o *PublishOptions) { | ||||||
| @@ -52,3 +51,13 @@ func SetPublishOption(k, v interface{}) PublishOption { | |||||||
| 		o.Context = context.WithValue(o.Context, k, v) | 		o.Context = context.WithValue(o.Context, k, v) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // SetOption returns a function to setup a context with given value | ||||||
|  | func SetOption(k, v interface{}) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		if o.Context == nil { | ||||||
|  | 			o.Context = context.Background() | ||||||
|  | 		} | ||||||
|  | 		o.Context = context.WithValue(o.Context, k, v) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -49,17 +49,6 @@ func TestSetSubscribeOption(t *testing.T) { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func TestSetPublishOption(t *testing.T) { |  | ||||||
| 	type key struct{} |  | ||||||
| 	o := SetPublishOption(key{}, "test") |  | ||||||
| 	opts := &PublishOptions{} |  | ||||||
| 	o(opts) |  | ||||||
|  |  | ||||||
| 	if v, ok := opts.Context.Value(key{}).(string); !ok || v == "" { |  | ||||||
| 		t.Fatal("SetPublishOption not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestSetOption(t *testing.T) { | func TestSetOption(t *testing.T) { | ||||||
| 	type key struct{} | 	type key struct{} | ||||||
| 	o := SetOption(key{}, "test") | 	o := SetOption(key{}, "test") | ||||||
|   | |||||||
							
								
								
									
										342
									
								
								broker/memory.go
									
									
									
									
									
								
							
							
						
						
									
										342
									
								
								broker/memory.go
									
									
									
									
									
								
							| @@ -1,342 +0,0 @@ | |||||||
| package broker |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"context" |  | ||||||
| 	"sync" |  | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/logger" |  | ||||||
| 	"go.unistack.org/micro/v3/metadata" |  | ||||||
| 	maddr "go.unistack.org/micro/v3/util/addr" |  | ||||||
| 	"go.unistack.org/micro/v3/util/id" |  | ||||||
| 	mnet "go.unistack.org/micro/v3/util/net" |  | ||||||
| 	"go.unistack.org/micro/v3/util/rand" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| type memoryBroker struct { |  | ||||||
| 	subscribers map[string][]*memorySubscriber |  | ||||||
| 	addr        string |  | ||||||
| 	opts        Options |  | ||||||
| 	sync.RWMutex |  | ||||||
| 	connected bool |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type memoryEvent struct { |  | ||||||
| 	err     error |  | ||||||
| 	message interface{} |  | ||||||
| 	topic   string |  | ||||||
| 	opts    Options |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type memorySubscriber struct { |  | ||||||
| 	ctx          context.Context |  | ||||||
| 	exit         chan bool |  | ||||||
| 	handler      Handler |  | ||||||
| 	batchhandler BatchHandler |  | ||||||
| 	id           string |  | ||||||
| 	topic        string |  | ||||||
| 	opts         SubscribeOptions |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryBroker) Options() Options { |  | ||||||
| 	return m.opts |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryBroker) Address() string { |  | ||||||
| 	return m.addr |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryBroker) Connect(ctx context.Context) error { |  | ||||||
| 	m.Lock() |  | ||||||
| 	defer m.Unlock() |  | ||||||
|  |  | ||||||
| 	if m.connected { |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	// use 127.0.0.1 to avoid scan of all network interfaces |  | ||||||
| 	addr, err := maddr.Extract("127.0.0.1") |  | ||||||
| 	if err != nil { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
| 	var rng rand.Rand |  | ||||||
| 	i := rng.Intn(20000) |  | ||||||
| 	// set addr with port |  | ||||||
| 	addr = mnet.HostPort(addr, 10000+i) |  | ||||||
|  |  | ||||||
| 	m.addr = addr |  | ||||||
| 	m.connected = true |  | ||||||
|  |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryBroker) Disconnect(ctx context.Context) error { |  | ||||||
| 	m.Lock() |  | ||||||
| 	defer m.Unlock() |  | ||||||
|  |  | ||||||
| 	if !m.connected { |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	m.connected = false |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryBroker) Init(opts ...Option) error { |  | ||||||
| 	for _, o := range opts { |  | ||||||
| 		o(&m.opts) |  | ||||||
| 	} |  | ||||||
| 	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) BatchPublish(ctx context.Context, msgs []*Message, opts ...PublishOption) error { |  | ||||||
| 	return m.publish(ctx, msgs, opts...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryBroker) publish(ctx context.Context, msgs []*Message, opts ...PublishOption) error { |  | ||||||
| 	m.RLock() |  | ||||||
| 	if !m.connected { |  | ||||||
| 		m.RUnlock() |  | ||||||
| 		return ErrNotConnected |  | ||||||
| 	} |  | ||||||
| 	m.RUnlock() |  | ||||||
|  |  | ||||||
| 	var err error |  | ||||||
|  |  | ||||||
| 	select { |  | ||||||
| 	case <-ctx.Done(): |  | ||||||
| 		return ctx.Err() |  | ||||||
| 	default: |  | ||||||
| 		options := NewPublishOptions(opts...) |  | ||||||
|  |  | ||||||
| 		msgTopicMap := make(map[string]Events) |  | ||||||
| 		for _, v := range msgs { |  | ||||||
| 			p := &memoryEvent{opts: m.opts} |  | ||||||
|  |  | ||||||
| 			if m.opts.Codec == nil || options.BodyOnly { |  | ||||||
| 				p.topic, _ = v.Header.Get(metadata.HeaderTopic) |  | ||||||
| 				p.message = v.Body |  | ||||||
| 			} else { |  | ||||||
| 				p.topic, _ = v.Header.Get(metadata.HeaderTopic) |  | ||||||
| 				p.message, err = m.opts.Codec.Marshal(v) |  | ||||||
| 				if err != nil { |  | ||||||
| 					return err |  | ||||||
| 				} |  | ||||||
| 			} |  | ||||||
| 			msgTopicMap[p.topic] = append(msgTopicMap[p.topic], p) |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		beh := m.opts.BatchErrorHandler |  | ||||||
| 		eh := m.opts.ErrorHandler |  | ||||||
|  |  | ||||||
| 		for t, ms := range msgTopicMap { |  | ||||||
| 			m.RLock() |  | ||||||
| 			subs, ok := m.subscribers[t] |  | ||||||
| 			m.RUnlock() |  | ||||||
| 			if !ok { |  | ||||||
| 				continue |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			for _, sub := range subs { |  | ||||||
| 				if sub.opts.BatchErrorHandler != nil { |  | ||||||
| 					beh = sub.opts.BatchErrorHandler |  | ||||||
| 				} |  | ||||||
| 				if sub.opts.ErrorHandler != nil { |  | ||||||
| 					eh = sub.opts.ErrorHandler |  | ||||||
| 				} |  | ||||||
|  |  | ||||||
| 				switch { |  | ||||||
| 				// batch processing |  | ||||||
| 				case sub.batchhandler != nil: |  | ||||||
| 					if err = sub.batchhandler(ms); err != nil { |  | ||||||
| 						ms.SetError(err) |  | ||||||
| 						if beh != nil { |  | ||||||
| 							_ = beh(ms) |  | ||||||
| 						} else if m.opts.Logger.V(logger.ErrorLevel) { |  | ||||||
| 							m.opts.Logger.Error(m.opts.Context, err.Error()) |  | ||||||
| 						} |  | ||||||
| 					} else if sub.opts.AutoAck { |  | ||||||
| 						if err = ms.Ack(); err != nil { |  | ||||||
| 							m.opts.Logger.Errorf(m.opts.Context, "ack failed: %v", err) |  | ||||||
| 						} |  | ||||||
| 					} |  | ||||||
| 					// single processing |  | ||||||
| 				case sub.handler != nil: |  | ||||||
| 					for _, p := range ms { |  | ||||||
| 						if err = sub.handler(p); err != nil { |  | ||||||
| 							p.SetError(err) |  | ||||||
| 							if eh != nil { |  | ||||||
| 								_ = eh(p) |  | ||||||
| 							} else if m.opts.Logger.V(logger.ErrorLevel) { |  | ||||||
| 								m.opts.Logger.Error(m.opts.Context, err.Error()) |  | ||||||
| 							} |  | ||||||
| 						} else if sub.opts.AutoAck { |  | ||||||
| 							if err = p.Ack(); err != nil { |  | ||||||
| 								m.opts.Logger.Errorf(m.opts.Context, "ack failed: %v", err) |  | ||||||
| 							} |  | ||||||
| 						} |  | ||||||
| 					} |  | ||||||
| 				} |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryBroker) BatchSubscribe(ctx context.Context, topic string, handler BatchHandler, opts ...SubscribeOption) (Subscriber, error) { |  | ||||||
| 	m.RLock() |  | ||||||
| 	if !m.connected { |  | ||||||
| 		m.RUnlock() |  | ||||||
| 		return nil, ErrNotConnected |  | ||||||
| 	} |  | ||||||
| 	m.RUnlock() |  | ||||||
|  |  | ||||||
| 	sid, err := id.New() |  | ||||||
| 	if err != nil { |  | ||||||
| 		return nil, err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	options := NewSubscribeOptions(opts...) |  | ||||||
|  |  | ||||||
| 	sub := &memorySubscriber{ |  | ||||||
| 		exit:         make(chan bool, 1), |  | ||||||
| 		id:           sid, |  | ||||||
| 		topic:        topic, |  | ||||||
| 		batchhandler: handler, |  | ||||||
| 		opts:         options, |  | ||||||
| 		ctx:          ctx, |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	m.Lock() |  | ||||||
| 	m.subscribers[topic] = append(m.subscribers[topic], sub) |  | ||||||
| 	m.Unlock() |  | ||||||
|  |  | ||||||
| 	go func() { |  | ||||||
| 		<-sub.exit |  | ||||||
| 		m.Lock() |  | ||||||
| 		newSubscribers := make([]*memorySubscriber, 0, len(m.subscribers)-1) |  | ||||||
| 		for _, sb := range m.subscribers[topic] { |  | ||||||
| 			if sb.id == sub.id { |  | ||||||
| 				continue |  | ||||||
| 			} |  | ||||||
| 			newSubscribers = append(newSubscribers, sb) |  | ||||||
| 		} |  | ||||||
| 		m.subscribers[topic] = newSubscribers |  | ||||||
| 		m.Unlock() |  | ||||||
| 	}() |  | ||||||
|  |  | ||||||
| 	return sub, nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryBroker) Subscribe(ctx context.Context, topic string, handler Handler, opts ...SubscribeOption) (Subscriber, error) { |  | ||||||
| 	m.RLock() |  | ||||||
| 	if !m.connected { |  | ||||||
| 		m.RUnlock() |  | ||||||
| 		return nil, ErrNotConnected |  | ||||||
| 	} |  | ||||||
| 	m.RUnlock() |  | ||||||
|  |  | ||||||
| 	sid, err := id.New() |  | ||||||
| 	if err != nil { |  | ||||||
| 		return nil, err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	options := NewSubscribeOptions(opts...) |  | ||||||
|  |  | ||||||
| 	sub := &memorySubscriber{ |  | ||||||
| 		exit:    make(chan bool, 1), |  | ||||||
| 		id:      sid, |  | ||||||
| 		topic:   topic, |  | ||||||
| 		handler: handler, |  | ||||||
| 		opts:    options, |  | ||||||
| 		ctx:     ctx, |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	m.Lock() |  | ||||||
| 	m.subscribers[topic] = append(m.subscribers[topic], sub) |  | ||||||
| 	m.Unlock() |  | ||||||
|  |  | ||||||
| 	go func() { |  | ||||||
| 		<-sub.exit |  | ||||||
| 		m.Lock() |  | ||||||
| 		newSubscribers := make([]*memorySubscriber, 0, len(m.subscribers)-1) |  | ||||||
| 		for _, sb := range m.subscribers[topic] { |  | ||||||
| 			if sb.id == sub.id { |  | ||||||
| 				continue |  | ||||||
| 			} |  | ||||||
| 			newSubscribers = append(newSubscribers, sb) |  | ||||||
| 		} |  | ||||||
| 		m.subscribers[topic] = newSubscribers |  | ||||||
| 		m.Unlock() |  | ||||||
| 	}() |  | ||||||
|  |  | ||||||
| 	return sub, nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryBroker) String() string { |  | ||||||
| 	return "memory" |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryBroker) Name() string { |  | ||||||
| 	return m.opts.Name |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryEvent) Topic() string { |  | ||||||
| 	return m.topic |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryEvent) Message() *Message { |  | ||||||
| 	switch v := m.message.(type) { |  | ||||||
| 	case *Message: |  | ||||||
| 		return v |  | ||||||
| 	case []byte: |  | ||||||
| 		msg := &Message{} |  | ||||||
| 		if err := m.opts.Codec.Unmarshal(v, msg); err != nil { |  | ||||||
| 			if m.opts.Logger.V(logger.ErrorLevel) { |  | ||||||
| 				m.opts.Logger.Error(m.opts.Context, "[memory]: failed to unmarshal: %v", err) |  | ||||||
| 			} |  | ||||||
| 			return nil |  | ||||||
| 		} |  | ||||||
| 		return msg |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryEvent) Ack() error { |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryEvent) Error() error { |  | ||||||
| 	return m.err |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memoryEvent) SetError(err error) { |  | ||||||
| 	m.err = err |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memorySubscriber) Options() SubscribeOptions { |  | ||||||
| 	return m.opts |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memorySubscriber) Topic() string { |  | ||||||
| 	return m.topic |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (m *memorySubscriber) Unsubscribe(ctx context.Context) error { |  | ||||||
| 	m.exit <- true |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewBroker return new memory broker |  | ||||||
| func NewBroker(opts ...Option) Broker { |  | ||||||
| 	return &memoryBroker{ |  | ||||||
| 		opts:        NewOptions(opts...), |  | ||||||
| 		subscribers: make(map[string][]*memorySubscriber), |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
							
								
								
									
										338
									
								
								broker/memory/memory.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										338
									
								
								broker/memory/memory.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,338 @@ | |||||||
|  | package broker | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"strings" | ||||||
|  | 	"sync" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v4/broker" | ||||||
|  | 	"go.unistack.org/micro/v4/codec" | ||||||
|  | 	"go.unistack.org/micro/v4/logger" | ||||||
|  | 	"go.unistack.org/micro/v4/metadata" | ||||||
|  | 	"go.unistack.org/micro/v4/options" | ||||||
|  | 	maddr "go.unistack.org/micro/v4/util/addr" | ||||||
|  | 	"go.unistack.org/micro/v4/util/id" | ||||||
|  | 	mnet "go.unistack.org/micro/v4/util/net" | ||||||
|  | 	"go.unistack.org/micro/v4/util/rand" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type Broker struct { | ||||||
|  | 	funcPublish   broker.FuncPublish | ||||||
|  | 	funcSubscribe broker.FuncSubscribe | ||||||
|  | 	subscribers   map[string][]*Subscriber | ||||||
|  | 	addr          string | ||||||
|  | 	opts          broker.Options | ||||||
|  | 	sync.RWMutex | ||||||
|  | 	connected bool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type memoryMessage struct { | ||||||
|  | 	c     codec.Codec | ||||||
|  | 	topic string | ||||||
|  | 	ctx   context.Context | ||||||
|  | 	body  []byte | ||||||
|  | 	hdr   metadata.Metadata | ||||||
|  | 	opts  broker.PublishOptions | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryMessage) Ack() error { | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryMessage) Body() []byte { | ||||||
|  | 	return m.body | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryMessage) Header() metadata.Metadata { | ||||||
|  | 	return m.hdr | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryMessage) Context() context.Context { | ||||||
|  | 	return m.ctx | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryMessage) Topic() string { | ||||||
|  | 	return "" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *memoryMessage) Unmarshal(dst interface{}, opts ...codec.Option) error { | ||||||
|  | 	return m.c.Unmarshal(m.body, dst) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type Subscriber struct { | ||||||
|  | 	ctx     context.Context | ||||||
|  | 	exit    chan bool | ||||||
|  | 	handler interface{} | ||||||
|  | 	id      string | ||||||
|  | 	topic   string | ||||||
|  | 	opts    broker.SubscribeOptions | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Broker) newCodec(ct string) (codec.Codec, error) { | ||||||
|  | 	if idx := strings.IndexRune(ct, ';'); idx >= 0 { | ||||||
|  | 		ct = ct[:idx] | ||||||
|  | 	} | ||||||
|  | 	b.RLock() | ||||||
|  | 	c, ok := b.opts.Codecs[ct] | ||||||
|  | 	b.RUnlock() | ||||||
|  | 	if ok { | ||||||
|  | 		return c, nil | ||||||
|  | 	} | ||||||
|  | 	return nil, codec.ErrUnknownContentType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Broker) Options() broker.Options { | ||||||
|  | 	return b.opts | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Broker) Address() string { | ||||||
|  | 	return b.addr | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Broker) Connect(ctx context.Context) error { | ||||||
|  | 	select { | ||||||
|  | 	case <-ctx.Done(): | ||||||
|  | 		return ctx.Err() | ||||||
|  | 	default: | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	b.Lock() | ||||||
|  | 	defer b.Unlock() | ||||||
|  |  | ||||||
|  | 	if b.connected { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// use 127.0.0.1 to avoid scan of all network interfaces | ||||||
|  | 	addr, err := maddr.Extract("127.0.0.1") | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	var rng rand.Rand | ||||||
|  | 	i := rng.Intn(20000) | ||||||
|  | 	// set addr with port | ||||||
|  | 	addr = mnet.HostPort(addr, 10000+i) | ||||||
|  |  | ||||||
|  | 	b.addr = addr | ||||||
|  | 	b.connected = true | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Broker) Disconnect(ctx context.Context) error { | ||||||
|  | 	select { | ||||||
|  | 	case <-ctx.Done(): | ||||||
|  | 		return ctx.Err() | ||||||
|  | 	default: | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	b.Lock() | ||||||
|  | 	defer b.Unlock() | ||||||
|  |  | ||||||
|  | 	if !b.connected { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	b.connected = false | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Broker) Init(opts ...broker.Option) error { | ||||||
|  | 	for _, o := range opts { | ||||||
|  | 		o(&b.opts) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	b.funcPublish = b.fnPublish | ||||||
|  | 	b.funcSubscribe = b.fnSubscribe | ||||||
|  |  | ||||||
|  | 	b.opts.Hooks.EachPrev(func(hook options.Hook) { | ||||||
|  | 		switch h := hook.(type) { | ||||||
|  | 		case broker.HookPublish: | ||||||
|  | 			b.funcPublish = h(b.funcPublish) | ||||||
|  | 		case broker.HookSubscribe: | ||||||
|  | 			b.funcSubscribe = h(b.funcSubscribe) | ||||||
|  | 		} | ||||||
|  | 	}) | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Broker) NewMessage(ctx context.Context, hdr metadata.Metadata, body interface{}, opts ...broker.PublishOption) (broker.Message, error) { | ||||||
|  | 	options := broker.NewPublishOptions(opts...) | ||||||
|  | 	m := &memoryMessage{ctx: ctx, hdr: hdr, opts: options} | ||||||
|  | 	c, err := b.newCodec(m.opts.ContentType) | ||||||
|  | 	if err == nil { | ||||||
|  | 		m.body, err = c.Marshal(body) | ||||||
|  | 	} | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return m, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Broker) Publish(ctx context.Context, topic string, messages ...broker.Message) error { | ||||||
|  | 	return b.funcPublish(ctx, topic, messages...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Broker) fnPublish(ctx context.Context, topic string, messages ...broker.Message) error { | ||||||
|  | 	return b.publish(ctx, topic, messages...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Broker) publish(ctx context.Context, topic string, messages ...broker.Message) error { | ||||||
|  | 	b.RLock() | ||||||
|  | 	if !b.connected { | ||||||
|  | 		b.RUnlock() | ||||||
|  | 		return broker.ErrNotConnected | ||||||
|  | 	} | ||||||
|  | 	b.RUnlock() | ||||||
|  |  | ||||||
|  | 	select { | ||||||
|  | 	case <-ctx.Done(): | ||||||
|  | 		return ctx.Err() | ||||||
|  | 	default: | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	b.RLock() | ||||||
|  | 	subs, ok := b.subscribers[topic] | ||||||
|  | 	b.RUnlock() | ||||||
|  | 	if !ok { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	var err error | ||||||
|  |  | ||||||
|  | 	for _, sub := range subs { | ||||||
|  | 		switch s := sub.handler.(type) { | ||||||
|  | 		default: | ||||||
|  | 			if b.opts.Logger.V(logger.ErrorLevel) { | ||||||
|  | 				b.opts.Logger.Error(ctx, "broker  handler error", broker.ErrInvalidHandler) | ||||||
|  | 			} | ||||||
|  | 		case func(broker.Message) error: | ||||||
|  | 			for _, message := range messages { | ||||||
|  | 				msg, ok := message.(*memoryMessage) | ||||||
|  | 				if !ok { | ||||||
|  | 					if b.opts.Logger.V(logger.ErrorLevel) { | ||||||
|  | 						b.opts.Logger.Error(ctx, "broker handler error", broker.ErrInvalidMessage) | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 				msg.topic = topic | ||||||
|  | 				if err = s(msg); err == nil && sub.opts.AutoAck { | ||||||
|  | 					err = msg.Ack() | ||||||
|  | 				} | ||||||
|  | 				if err != nil { | ||||||
|  | 					if b.opts.Logger.V(logger.ErrorLevel) { | ||||||
|  | 						b.opts.Logger.Error(ctx, "broker handler error", err) | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		case func([]broker.Message) error: | ||||||
|  | 			if err = s(messages); err == nil && sub.opts.AutoAck { | ||||||
|  | 				for _, message := range messages { | ||||||
|  | 					err = message.Ack() | ||||||
|  | 					if err != nil { | ||||||
|  | 						if b.opts.Logger.V(logger.ErrorLevel) { | ||||||
|  | 							b.opts.Logger.Error(ctx, "broker handler error", err) | ||||||
|  | 						} | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Broker) Subscribe(ctx context.Context, topic string, handler interface{}, opts ...broker.SubscribeOption) (broker.Subscriber, error) { | ||||||
|  | 	return b.funcSubscribe(ctx, topic, handler, opts...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Broker) fnSubscribe(ctx context.Context, topic string, handler interface{}, opts ...broker.SubscribeOption) (broker.Subscriber, error) { | ||||||
|  | 	if err := broker.IsValidHandler(handler); err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	b.RLock() | ||||||
|  | 	if !b.connected { | ||||||
|  | 		b.RUnlock() | ||||||
|  | 		return nil, broker.ErrNotConnected | ||||||
|  | 	} | ||||||
|  | 	b.RUnlock() | ||||||
|  |  | ||||||
|  | 	sid, err := id.New() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	options := broker.NewSubscribeOptions(opts...) | ||||||
|  |  | ||||||
|  | 	sub := &Subscriber{ | ||||||
|  | 		exit:    make(chan bool, 1), | ||||||
|  | 		id:      sid, | ||||||
|  | 		topic:   topic, | ||||||
|  | 		handler: handler, | ||||||
|  | 		opts:    options, | ||||||
|  | 		ctx:     ctx, | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	b.Lock() | ||||||
|  | 	b.subscribers[topic] = append(b.subscribers[topic], sub) | ||||||
|  | 	b.Unlock() | ||||||
|  |  | ||||||
|  | 	go func() { | ||||||
|  | 		<-sub.exit | ||||||
|  | 		b.Lock() | ||||||
|  | 		newSubscribers := make([]*Subscriber, 0, len(b.subscribers)-1) | ||||||
|  | 		for _, sb := range b.subscribers[topic] { | ||||||
|  | 			if sb.id == sub.id { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			newSubscribers = append(newSubscribers, sb) | ||||||
|  | 		} | ||||||
|  | 		b.subscribers[topic] = newSubscribers | ||||||
|  | 		b.Unlock() | ||||||
|  | 	}() | ||||||
|  |  | ||||||
|  | 	return sub, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Broker) String() string { | ||||||
|  | 	return "memory" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Broker) Name() string { | ||||||
|  | 	return b.opts.Name | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Broker) Live() bool { | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Broker) Ready() bool { | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *Broker) Health() bool { | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *Subscriber) Options() broker.SubscribeOptions { | ||||||
|  | 	return m.opts | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *Subscriber) Topic() string { | ||||||
|  | 	return m.topic | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *Subscriber) Unsubscribe(ctx context.Context) error { | ||||||
|  | 	m.exit <- true | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewBroker return new memory broker | ||||||
|  | func NewBroker(opts ...broker.Option) broker.Broker { | ||||||
|  | 	return &Broker{ | ||||||
|  | 		opts:        broker.NewOptions(opts...), | ||||||
|  | 		subscribers: make(map[string][]*Subscriber), | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										74
									
								
								broker/memory/memory_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										74
									
								
								broker/memory/memory_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,74 @@ | |||||||
|  | package broker | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"fmt" | ||||||
|  | 	"testing" | ||||||
|  |  | ||||||
|  | 	"go.uber.org/atomic" | ||||||
|  | 	"go.unistack.org/micro/v4/broker" | ||||||
|  | 	"go.unistack.org/micro/v4/codec" | ||||||
|  | 	"go.unistack.org/micro/v4/metadata" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type hldr struct { | ||||||
|  | 	c atomic.Int64 | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (h *hldr) Handler(m broker.Message) error { | ||||||
|  | 	h.c.Add(1) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestMemoryBroker(t *testing.T) { | ||||||
|  | 	b := NewBroker(broker.Codec("application/octet-stream", codec.NewCodec())) | ||||||
|  | 	ctx := context.Background() | ||||||
|  |  | ||||||
|  | 	if err := b.Init(); err != nil { | ||||||
|  | 		t.Fatalf("Unexpected init error %v", err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if err := b.Connect(ctx); err != nil { | ||||||
|  | 		t.Fatalf("Unexpected connect error %v", err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	topic := "test" | ||||||
|  | 	count := int64(10) | ||||||
|  |  | ||||||
|  | 	h := &hldr{} | ||||||
|  |  | ||||||
|  | 	sub, err := b.Subscribe(ctx, topic, h.Handler) | ||||||
|  | 	if err != nil { | ||||||
|  | 		t.Fatalf("Unexpected error subscribing %v", err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for i := int64(0); i < count; i++ { | ||||||
|  | 		message, err := b.NewMessage(ctx, | ||||||
|  | 			metadata.Pairs( | ||||||
|  | 				"foo", "bar", | ||||||
|  | 				"id", fmt.Sprintf("%d", i), | ||||||
|  | 			), | ||||||
|  | 			[]byte(`"hello world"`), | ||||||
|  | 			broker.PublishContentType("application/octet-stream"), | ||||||
|  | 		) | ||||||
|  | 		if err != nil { | ||||||
|  | 			t.Fatal(err) | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		if err := b.Publish(ctx, topic, message); err != nil { | ||||||
|  | 			t.Fatalf("Unexpected error publishing %d err: %v", i, err) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if err := sub.Unsubscribe(ctx); err != nil { | ||||||
|  | 		t.Fatalf("Unexpected error unsubscribing from %s: %v", topic, err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if err := b.Disconnect(ctx); err != nil { | ||||||
|  | 		t.Fatalf("Unexpected connect error %v", err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if h.c.Load() != count { | ||||||
|  | 		t.Fatal("invalid messages count received") | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -1,105 +0,0 @@ | |||||||
| package broker |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"context" |  | ||||||
| 	"fmt" |  | ||||||
| 	"testing" |  | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/metadata" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| func TestMemoryBatchBroker(t *testing.T) { |  | ||||||
| 	b := NewBroker() |  | ||||||
| 	ctx := context.Background() |  | ||||||
|  |  | ||||||
| 	if err := b.Connect(ctx); err != nil { |  | ||||||
| 		t.Fatalf("Unexpected connect error %v", err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	topic := "test" |  | ||||||
| 	count := 10 |  | ||||||
|  |  | ||||||
| 	fn := func(evts Events) error { |  | ||||||
| 		return evts.Ack() |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	sub, err := b.BatchSubscribe(ctx, topic, fn) |  | ||||||
| 	if err != nil { |  | ||||||
| 		t.Fatalf("Unexpected error subscribing %v", err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	msgs := make([]*Message, 0, count) |  | ||||||
| 	for i := 0; i < count; i++ { |  | ||||||
| 		message := &Message{ |  | ||||||
| 			Header: map[string]string{ |  | ||||||
| 				metadata.HeaderTopic: topic, |  | ||||||
| 				"foo":                "bar", |  | ||||||
| 				"id":                 fmt.Sprintf("%d", i), |  | ||||||
| 			}, |  | ||||||
| 			Body: []byte(`"hello world"`), |  | ||||||
| 		} |  | ||||||
| 		msgs = append(msgs, message) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if err := b.BatchPublish(ctx, msgs); err != nil { |  | ||||||
| 		t.Fatalf("Unexpected error publishing %v", err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if err := sub.Unsubscribe(ctx); err != nil { |  | ||||||
| 		t.Fatalf("Unexpected error unsubscribing from %s: %v", topic, err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if err := b.Disconnect(ctx); err != nil { |  | ||||||
| 		t.Fatalf("Unexpected connect error %v", err) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestMemoryBroker(t *testing.T) { |  | ||||||
| 	b := NewBroker() |  | ||||||
| 	ctx := context.Background() |  | ||||||
|  |  | ||||||
| 	if err := b.Connect(ctx); err != nil { |  | ||||||
| 		t.Fatalf("Unexpected connect error %v", err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	topic := "test" |  | ||||||
| 	count := 10 |  | ||||||
|  |  | ||||||
| 	fn := func(p Event) error { |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	sub, err := b.Subscribe(ctx, topic, fn) |  | ||||||
| 	if err != nil { |  | ||||||
| 		t.Fatalf("Unexpected error subscribing %v", err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	msgs := make([]*Message, 0, count) |  | ||||||
| 	for i := 0; i < count; i++ { |  | ||||||
| 		message := &Message{ |  | ||||||
| 			Header: map[string]string{ |  | ||||||
| 				metadata.HeaderTopic: topic, |  | ||||||
| 				"foo":                "bar", |  | ||||||
| 				"id":                 fmt.Sprintf("%d", i), |  | ||||||
| 			}, |  | ||||||
| 			Body: []byte(`"hello world"`), |  | ||||||
| 		} |  | ||||||
| 		msgs = append(msgs, message) |  | ||||||
|  |  | ||||||
| 		if err := b.Publish(ctx, topic, message); err != nil { |  | ||||||
| 			t.Fatalf("Unexpected error publishing %d err: %v", i, err) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if err := b.BatchPublish(ctx, msgs); err != nil { |  | ||||||
| 		t.Fatalf("Unexpected error publishing %v", err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if err := sub.Unsubscribe(ctx); err != nil { |  | ||||||
| 		t.Fatalf("Unexpected error unsubscribing from %s: %v", topic, err) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if err := b.Disconnect(ctx); err != nil { |  | ||||||
| 		t.Fatalf("Unexpected connect error %v", err) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
							
								
								
									
										176
									
								
								broker/noop.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										176
									
								
								broker/noop.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,176 @@ | |||||||
|  | package broker | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"strings" | ||||||
|  | 	"sync" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v4/codec" | ||||||
|  | 	"go.unistack.org/micro/v4/metadata" | ||||||
|  | 	"go.unistack.org/micro/v4/options" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type NoopBroker struct { | ||||||
|  | 	funcPublish   FuncPublish | ||||||
|  | 	funcSubscribe FuncSubscribe | ||||||
|  | 	opts          Options | ||||||
|  | 	sync.RWMutex | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) newCodec(ct string) (codec.Codec, error) { | ||||||
|  | 	if idx := strings.IndexRune(ct, ';'); idx >= 0 { | ||||||
|  | 		ct = ct[:idx] | ||||||
|  | 	} | ||||||
|  | 	b.RLock() | ||||||
|  | 	c, ok := b.opts.Codecs[ct] | ||||||
|  | 	b.RUnlock() | ||||||
|  | 	if ok { | ||||||
|  | 		return c, nil | ||||||
|  | 	} | ||||||
|  | 	return nil, codec.ErrUnknownContentType | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func NewBroker(opts ...Option) *NoopBroker { | ||||||
|  | 	b := &NoopBroker{opts: NewOptions(opts...)} | ||||||
|  | 	b.funcPublish = b.fnPublish | ||||||
|  | 	b.funcSubscribe = b.fnSubscribe | ||||||
|  |  | ||||||
|  | 	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.funcSubscribe = b.fnSubscribe | ||||||
|  |  | ||||||
|  | 	b.opts.Hooks.EachPrev(func(hook options.Hook) { | ||||||
|  | 		switch h := hook.(type) { | ||||||
|  | 		case HookPublish: | ||||||
|  | 			b.funcPublish = h(b.funcPublish) | ||||||
|  | 		case HookSubscribe: | ||||||
|  | 			b.funcSubscribe = h(b.funcSubscribe) | ||||||
|  | 		} | ||||||
|  | 	}) | ||||||
|  |  | ||||||
|  | 	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, ",") | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type noopMessage struct { | ||||||
|  | 	c    codec.Codec | ||||||
|  | 	ctx  context.Context | ||||||
|  | 	body []byte | ||||||
|  | 	hdr  metadata.Metadata | ||||||
|  | 	opts PublishOptions | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *noopMessage) Ack() error { | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *noopMessage) Body() []byte { | ||||||
|  | 	return m.body | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *noopMessage) Header() metadata.Metadata { | ||||||
|  | 	return m.hdr | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *noopMessage) Context() context.Context { | ||||||
|  | 	return m.ctx | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *noopMessage) Topic() string { | ||||||
|  | 	return "" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *noopMessage) Unmarshal(dst interface{}, opts ...codec.Option) error { | ||||||
|  | 	return m.c.Unmarshal(m.body, dst) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) NewMessage(ctx context.Context, hdr metadata.Metadata, body interface{}, opts ...PublishOption) (Message, error) { | ||||||
|  | 	options := NewPublishOptions(opts...) | ||||||
|  | 	m := &noopMessage{ctx: ctx, hdr: hdr, opts: options} | ||||||
|  | 	c, err := b.newCodec(m.opts.ContentType) | ||||||
|  | 	if err == nil { | ||||||
|  | 		m.body, err = c.Marshal(body) | ||||||
|  | 	} | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return m, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) fnPublish(_ context.Context, _ string, _ ...Message) error { | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) Publish(ctx context.Context, topic string, msg ...Message) error { | ||||||
|  | 	return b.funcPublish(ctx, topic, msg...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type NoopSubscriber struct { | ||||||
|  | 	ctx     context.Context | ||||||
|  | 	topic   string | ||||||
|  | 	handler interface{} | ||||||
|  | 	opts    SubscribeOptions | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (b *NoopBroker) fnSubscribe(ctx context.Context, topic string, handler interface{}, 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 interface{}, 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, messages ...Message) error { | ||||||
|  | 		t.f = true | ||||||
|  | 		return fn(ctx, topic, messages...) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestNoopHook(t *testing.T) { | ||||||
|  | 	h := &testHook{} | ||||||
|  |  | ||||||
|  | 	b := NewBroker(Hooks(HookPublish(h.Publish1))) | ||||||
|  |  | ||||||
|  | 	if err := b.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if err := b.Publish(context.TODO(), "", nil); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !h.f { | ||||||
|  | 		t.Fatal("hook not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -5,49 +5,60 @@ import ( | |||||||
| 	"crypto/tls" | 	"crypto/tls" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/codec" | 	"go.unistack.org/micro/v4/codec" | ||||||
| 	"go.unistack.org/micro/v3/logger" | 	"go.unistack.org/micro/v4/logger" | ||||||
| 	"go.unistack.org/micro/v3/meter" | 	"go.unistack.org/micro/v4/meter" | ||||||
| 	"go.unistack.org/micro/v3/register" | 	"go.unistack.org/micro/v4/options" | ||||||
| 	"go.unistack.org/micro/v3/tracer" | 	"go.unistack.org/micro/v4/register" | ||||||
|  | 	"go.unistack.org/micro/v4/sync" | ||||||
|  | 	"go.unistack.org/micro/v4/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Options struct | // Options struct | ||||||
| type Options struct { | type Options struct { | ||||||
|  | 	// Name holds the broker name | ||||||
|  | 	Name string | ||||||
|  |  | ||||||
| 	// Tracer used for tracing | 	// Tracer used for tracing | ||||||
| 	Tracer tracer.Tracer | 	Tracer tracer.Tracer | ||||||
| 	// Register can be used for clustering | 	// Register can be used for clustering | ||||||
| 	Register register.Register | 	Register register.Register | ||||||
| 	// Codec holds the codec for marshal/unmarshal | 	// Codecs holds the codecs for marshal/unmarshal based on content-type | ||||||
| 	Codec codec.Codec | 	Codecs map[string]codec.Codec | ||||||
| 	// Logger used for logging | 	// Logger used for logging | ||||||
| 	Logger logger.Logger | 	Logger logger.Logger | ||||||
| 	// Meter used for metrics | 	// Meter used for metrics | ||||||
| 	Meter meter.Meter | 	Meter meter.Meter | ||||||
| 	// Context holds external options | 	// Context holds external options | ||||||
| 	Context context.Context | 	Context context.Context | ||||||
|  |  | ||||||
|  | 	// Wait waits for a collection of goroutines to finish | ||||||
|  | 	Wait *sync.WaitGroup | ||||||
| 	// TLSConfig holds tls.TLSConfig options | 	// TLSConfig holds tls.TLSConfig options | ||||||
| 	TLSConfig *tls.Config | 	TLSConfig *tls.Config | ||||||
| 	// ErrorHandler used when broker 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 holds the broker address | ||||||
| 	Addrs []string | 	Addrs []string | ||||||
|  | 	// Hooks can be run before broker Publish/BatchPublish and | ||||||
|  | 	// Subscribe/BatchSubscribe methods | ||||||
|  | 	Hooks options.Hooks | ||||||
|  |  | ||||||
|  | 	// GracefulTimeout contains time to wait to finish in flight requests | ||||||
|  | 	GracefulTimeout time.Duration | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewOptions create new Options | // NewOptions create new Options | ||||||
| func NewOptions(opts ...Option) Options { | func NewOptions(opts ...Option) Options { | ||||||
| 	options := Options{ | 	options := Options{ | ||||||
| 		Register: register.DefaultRegister, | 		Register:        register.DefaultRegister, | ||||||
| 		Logger:   logger.DefaultLogger, | 		Logger:          logger.DefaultLogger, | ||||||
| 		Context:  context.Background(), | 		Context:         context.Background(), | ||||||
| 		Meter:    meter.DefaultMeter, | 		Meter:           meter.DefaultMeter, | ||||||
| 		Codec:    codec.DefaultCodec, | 		Codecs:          make(map[string]codec.Codec), | ||||||
| 		Tracer:   tracer.DefaultTracer, | 		Tracer:          tracer.DefaultTracer, | ||||||
|  | 		GracefulTimeout: DefaultGracefulTimeout, | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&options) | 		o(&options) | ||||||
| 	} | 	} | ||||||
| @@ -63,10 +74,13 @@ func Context(ctx context.Context) Option { | |||||||
|  |  | ||||||
| // PublishOptions struct | // PublishOptions struct | ||||||
| type PublishOptions struct { | type PublishOptions struct { | ||||||
| 	// Context holds external options | 	// ContentType for message body | ||||||
| 	Context context.Context | 	ContentType string | ||||||
| 	// BodyOnly flag says the message contains raw body bytes | 	// BodyOnly flag says the message contains raw body bytes and don't need | ||||||
|  | 	// codec Marshal method | ||||||
| 	BodyOnly bool | 	BodyOnly bool | ||||||
|  | 	// Context holds custom options | ||||||
|  | 	Context context.Context | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewPublishOptions creates PublishOptions struct | // NewPublishOptions creates PublishOptions struct | ||||||
| @@ -84,10 +98,6 @@ func NewPublishOptions(opts ...PublishOption) PublishOptions { | |||||||
| type SubscribeOptions struct { | type SubscribeOptions struct { | ||||||
| 	// Context holds external options | 	// Context holds external options | ||||||
| 	Context context.Context | 	Context context.Context | ||||||
| 	// ErrorHandler used when broker can't unmarshal incoming message |  | ||||||
| 	ErrorHandler Handler |  | ||||||
| 	// BatchErrorHandler used when broker can't unmashal incoming messages |  | ||||||
| 	BatchErrorHandler BatchHandler |  | ||||||
| 	// Group holds consumer group | 	// Group holds consumer group | ||||||
| 	Group string | 	Group string | ||||||
| 	// AutoAck flag specifies auto ack of incoming message when no error happens | 	// AutoAck flag specifies auto ack of incoming message when no error happens | ||||||
| @@ -106,6 +116,13 @@ type Option func(*Options) | |||||||
| // PublishOption func | // PublishOption func | ||||||
| type PublishOption func(*PublishOptions) | type PublishOption func(*PublishOptions) | ||||||
|  |  | ||||||
|  | // PublishContentType sets message content-type that used to Marshal | ||||||
|  | func PublishContentType(ct string) PublishOption { | ||||||
|  | 	return func(o *PublishOptions) { | ||||||
|  | 		o.ContentType = ct | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // PublishBodyOnly publish only body of the message | // PublishBodyOnly publish only body of the message | ||||||
| func PublishBodyOnly(b bool) PublishOption { | func PublishBodyOnly(b bool) PublishOption { | ||||||
| 	return func(o *PublishOptions) { | 	return func(o *PublishOptions) { | ||||||
| @@ -113,13 +130,6 @@ func PublishBodyOnly(b bool) PublishOption { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // PublishContext sets the context |  | ||||||
| func PublishContext(ctx context.Context) PublishOption { |  | ||||||
| 	return func(o *PublishOptions) { |  | ||||||
| 		o.Context = ctx |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Addrs sets the host addresses to be used by the broker | // Addrs sets the host addresses to be used by the broker | ||||||
| func Addrs(addrs ...string) Option { | func Addrs(addrs ...string) Option { | ||||||
| 	return func(o *Options) { | 	return func(o *Options) { | ||||||
| @@ -127,51 +137,10 @@ func Addrs(addrs ...string) Option { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // Codec sets the codec used for encoding/decoding used where | // Codec sets the codec used for encoding/decoding messages | ||||||
| // a broker does not support headers | func Codec(ct string, c codec.Codec) Option { | ||||||
| func Codec(c codec.Codec) Option { |  | ||||||
| 	return func(o *Options) { | 	return func(o *Options) { | ||||||
| 		o.Codec = c | 		o.Codecs[ct] = c | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // ErrorHandler will catch all broker errors that cant be handled |  | ||||||
| // in normal way, for example Codec errors |  | ||||||
| func ErrorHandler(h Handler) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.ErrorHandler = h |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // BatchErrorHandler will catch all broker errors that cant be handled |  | ||||||
| // in normal way, for example Codec errors |  | ||||||
| func BatchErrorHandler(h BatchHandler) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.BatchErrorHandler = h |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // SubscribeErrorHandler will catch all broker errors that cant be handled |  | ||||||
| // in normal way, for example Codec errors |  | ||||||
| func SubscribeErrorHandler(h Handler) SubscribeOption { |  | ||||||
| 	return func(o *SubscribeOptions) { |  | ||||||
| 		o.ErrorHandler = h |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // SubscribeBatchErrorHandler will catch all broker errors that cant be handled |  | ||||||
| // in normal way, for example Codec errors |  | ||||||
| func SubscribeBatchErrorHandler(h BatchHandler) SubscribeOption { |  | ||||||
| 	return func(o *SubscribeOptions) { |  | ||||||
| 		o.BatchErrorHandler = h |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Queue sets the subscribers queue |  | ||||||
| // Deprecated |  | ||||||
| func Queue(name string) SubscribeOption { |  | ||||||
| 	return func(o *SubscribeOptions) { |  | ||||||
| 		o.Group = name |  | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -224,6 +193,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 | // SubscribeContext set context | ||||||
| func SubscribeContext(ctx context.Context) SubscribeOption { | func SubscribeContext(ctx context.Context) SubscribeOption { | ||||||
| 	return func(o *SubscribeOptions) { | 	return func(o *SubscribeOptions) { | ||||||
| @@ -231,14 +207,6 @@ func SubscribeContext(ctx context.Context) SubscribeOption { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // DisableAutoAck disables auto ack |  | ||||||
| // Deprecated |  | ||||||
| func DisableAutoAck() SubscribeOption { |  | ||||||
| 	return func(o *SubscribeOptions) { |  | ||||||
| 		o.AutoAck = false |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // SubscribeAutoAck contol auto acking of messages | // SubscribeAutoAck contol auto acking of messages | ||||||
| // after they have been handled. | // after they have been handled. | ||||||
| func SubscribeAutoAck(b bool) SubscribeOption { | func SubscribeAutoAck(b bool) SubscribeOption { | ||||||
|   | |||||||
							
								
								
									
										14
									
								
								broker/subscriber.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								broker/subscriber.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,14 @@ | |||||||
|  | package broker | ||||||
|  |  | ||||||
|  | // IsValidHandler func signature | ||||||
|  | func IsValidHandler(sub interface{}) error { | ||||||
|  | 	switch sub.(type) { | ||||||
|  | 	default: | ||||||
|  | 		return ErrInvalidHandler | ||||||
|  | 	case func(Message) error: | ||||||
|  | 		break | ||||||
|  | 	case func([]Message) error: | ||||||
|  | 		break | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
| @@ -5,7 +5,7 @@ import ( | |||||||
| 	"math" | 	"math" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/util/backoff" | 	"go.unistack.org/micro/v4/util/backoff" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // BackoffFunc is the backoff call func | // BackoffFunc is the backoff call func | ||||||
| @@ -17,13 +17,13 @@ func BackoffExp(_ context.Context, _ Request, attempts int) (time.Duration, erro | |||||||
| } | } | ||||||
|  |  | ||||||
| // BackoffInterval specifies randomization interval for backoff func | // BackoffInterval specifies randomization interval for backoff func | ||||||
| func BackoffInterval(min time.Duration, max time.Duration) BackoffFunc { | func BackoffInterval(minTime time.Duration, maxTime time.Duration) BackoffFunc { | ||||||
| 	return func(_ context.Context, _ Request, attempts int) (time.Duration, error) { | 	return func(_ context.Context, _ Request, attempts int) (time.Duration, error) { | ||||||
| 		td := time.Duration(math.Pow(float64(attempts), math.E)) * time.Millisecond * 100 | 		td := time.Duration(math.Pow(float64(attempts), math.E)) * time.Millisecond * 100 | ||||||
| 		if td < min { | 		if td < minTime { | ||||||
| 			return min, nil | 			return minTime, nil | ||||||
| 		} else if td > max { | 		} else if td > maxTime { | ||||||
| 			return max, nil | 			return maxTime, nil | ||||||
| 		} | 		} | ||||||
| 		return td, nil | 		return td, nil | ||||||
| 	} | 	} | ||||||
|   | |||||||
| @@ -34,23 +34,23 @@ func TestBackoffExp(t *testing.T) { | |||||||
| } | } | ||||||
|  |  | ||||||
| func TestBackoffInterval(t *testing.T) { | func TestBackoffInterval(t *testing.T) { | ||||||
| 	min := 100 * time.Millisecond | 	minTime := 100 * time.Millisecond | ||||||
| 	max := 300 * time.Millisecond | 	maxTime := 300 * time.Millisecond | ||||||
|  |  | ||||||
| 	r := &testRequest{ | 	r := &testRequest{ | ||||||
| 		service: "test", | 		service: "test", | ||||||
| 		method:  "test", | 		method:  "test", | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	fn := BackoffInterval(min, max) | 	fn := BackoffInterval(minTime, maxTime) | ||||||
| 	for i := 0; i < 5; i++ { | 	for i := 0; i < 5; i++ { | ||||||
| 		d, err := fn(context.TODO(), r, i) | 		d, err := fn(context.TODO(), r, i) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			t.Fatal(err) | 			t.Fatal(err) | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		if d < min || d > max { | 		if d < minTime || d > maxTime { | ||||||
| 			t.Fatalf("Expected %v < %v < %v", min, d, max) | 			t.Fatalf("Expected %v < %v < %v", minTime, d, maxTime) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,12 +1,12 @@ | |||||||
| // Package client is an interface for an RPC client | // Package client is an interface for an RPC client | ||||||
| package client // import "go.unistack.org/micro/v3/client" | package client | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/codec" | 	"go.unistack.org/micro/v4/codec" | ||||||
| 	"go.unistack.org/micro/v3/metadata" | 	"go.unistack.org/micro/v4/metadata" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| @@ -29,28 +29,23 @@ var ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
| // Client is the interface used to make requests to services. | // Client is the interface used to make requests to services. | ||||||
| // It supports Request/Response via Transport and Publishing via the Broker. |  | ||||||
| // It also supports bidirectional streaming of requests. | // It also supports bidirectional streaming of requests. | ||||||
| type Client interface { | type Client interface { | ||||||
| 	Name() string | 	Name() string | ||||||
| 	Init(opts ...Option) error | 	Init(opts ...Option) error | ||||||
| 	Options() Options | 	Options() Options | ||||||
| 	NewMessage(topic string, msg interface{}, opts ...MessageOption) Message |  | ||||||
| 	NewRequest(service string, endpoint string, req interface{}, opts ...RequestOption) Request | 	NewRequest(service string, endpoint string, req interface{}, opts ...RequestOption) Request | ||||||
| 	Call(ctx context.Context, req Request, rsp interface{}, opts ...CallOption) error | 	Call(ctx context.Context, req Request, rsp interface{}, opts ...CallOption) error | ||||||
| 	Stream(ctx context.Context, req Request, opts ...CallOption) (Stream, error) | 	Stream(ctx context.Context, req Request, opts ...CallOption) (Stream, error) | ||||||
| 	Publish(ctx context.Context, msg Message, opts ...PublishOption) error |  | ||||||
| 	BatchPublish(ctx context.Context, msg []Message, opts ...PublishOption) error |  | ||||||
| 	String() string | 	String() string | ||||||
| } | } | ||||||
|  |  | ||||||
| // Message is the interface for publishing asynchronously | type ( | ||||||
| type Message interface { | 	FuncCall   func(ctx context.Context, req Request, rsp interface{}, opts ...CallOption) error | ||||||
| 	Topic() string | 	HookCall   func(next FuncCall) FuncCall | ||||||
| 	Payload() interface{} | 	FuncStream func(ctx context.Context, req Request, opts ...CallOption) (Stream, error) | ||||||
| 	ContentType() string | 	HookStream func(next FuncStream) FuncStream | ||||||
| 	Metadata() metadata.Metadata | ) | ||||||
| } |  | ||||||
|  |  | ||||||
| // Request is the interface for a synchronous request used by Call or Stream | // Request is the interface for a synchronous request used by Call or Stream | ||||||
| type Request interface { | type Request interface { | ||||||
| @@ -110,11 +105,5 @@ type Option func(*Options) | |||||||
| // CallOption used by Call or Stream | // CallOption used by Call or Stream | ||||||
| type CallOption func(*CallOptions) | type CallOption func(*CallOptions) | ||||||
|  |  | ||||||
| // PublishOption used by Publish |  | ||||||
| type PublishOption func(*PublishOptions) |  | ||||||
|  |  | ||||||
| // MessageOption used by NewMessage |  | ||||||
| type MessageOption func(*MessageOptions) |  | ||||||
|  |  | ||||||
| // RequestOption used by NewRequest | // RequestOption used by NewRequest | ||||||
| type RequestOption func(*RequestOptions) | type RequestOption func(*RequestOptions) | ||||||
|   | |||||||
| @@ -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 | 	return c, ok | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // MustContext get client from context | ||||||
|  | func MustContext(ctx context.Context) Client { | ||||||
|  | 	c, ok := FromContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		panic("missing client") | ||||||
|  | 	} | ||||||
|  | 	return c | ||||||
|  | } | ||||||
|  |  | ||||||
| // NewContext put client in context | // NewContext put client in context | ||||||
| func NewContext(ctx context.Context, c Client) context.Context { | func NewContext(ctx context.Context, c Client) context.Context { | ||||||
| 	if ctx == nil { | 	if ctx == nil { | ||||||
| @@ -23,16 +32,6 @@ func NewContext(ctx context.Context, c Client) context.Context { | |||||||
| 	return context.WithValue(ctx, clientKey{}, c) | 	return context.WithValue(ctx, clientKey{}, c) | ||||||
| } | } | ||||||
|  |  | ||||||
| // SetPublishOption returns a function to setup a context with given value |  | ||||||
| func SetPublishOption(k, v interface{}) PublishOption { |  | ||||||
| 	return func(o *PublishOptions) { |  | ||||||
| 		if o.Context == nil { |  | ||||||
| 			o.Context = context.Background() |  | ||||||
| 		} |  | ||||||
| 		o.Context = context.WithValue(o.Context, k, v) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // SetCallOption returns a function to setup a context with given value | // SetCallOption returns a function to setup a context with given value | ||||||
| func SetCallOption(k, v interface{}) CallOption { | func SetCallOption(k, v interface{}) CallOption { | ||||||
| 	return func(o *CallOptions) { | 	return func(o *CallOptions) { | ||||||
|   | |||||||
| @@ -39,17 +39,6 @@ func TestNewNilContext(t *testing.T) { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func TestSetPublishOption(t *testing.T) { |  | ||||||
| 	type key struct{} |  | ||||||
| 	o := SetPublishOption(key{}, "test") |  | ||||||
| 	opts := &PublishOptions{} |  | ||||||
| 	o(opts) |  | ||||||
|  |  | ||||||
| 	if v, ok := opts.Context.Value(key{}).(string); !ok || v == "" { |  | ||||||
| 		t.Fatal("SetPublishOption not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestSetCallOption(t *testing.T) { | func TestSetCallOption(t *testing.T) { | ||||||
| 	type key struct{} | 	type key struct{} | ||||||
| 	o := SetCallOption(key{}, "test") | 	o := SetCallOption(key{}, "test") | ||||||
|   | |||||||
| @@ -4,8 +4,8 @@ import ( | |||||||
| 	"context" | 	"context" | ||||||
| 	"sort" | 	"sort" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/errors" | 	"go.unistack.org/micro/v4/errors" | ||||||
| 	"go.unistack.org/micro/v3/router" | 	"go.unistack.org/micro/v4/router" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // LookupFunc is used to lookup routes for a service | // LookupFunc is used to lookup routes for a service | ||||||
|   | |||||||
							
								
								
									
										229
									
								
								client/noop.go
									
									
									
									
									
								
							
							
						
						
									
										229
									
								
								client/noop.go
									
									
									
									
									
								
							| @@ -3,13 +3,16 @@ package client | |||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"fmt" | 	"fmt" | ||||||
|  | 	"strconv" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/broker" | 	"go.unistack.org/micro/v4/codec" | ||||||
| 	"go.unistack.org/micro/v3/codec" | 	"go.unistack.org/micro/v4/errors" | ||||||
| 	"go.unistack.org/micro/v3/errors" | 	"go.unistack.org/micro/v4/metadata" | ||||||
| 	"go.unistack.org/micro/v3/metadata" | 	"go.unistack.org/micro/v4/options" | ||||||
| 	"go.unistack.org/micro/v3/selector" | 	"go.unistack.org/micro/v4/selector" | ||||||
|  | 	"go.unistack.org/micro/v4/semconv" | ||||||
|  | 	"go.unistack.org/micro/v4/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // DefaultCodecs will be used to encode/decode data | // DefaultCodecs will be used to encode/decode data | ||||||
| @@ -18,13 +21,9 @@ var DefaultCodecs = map[string]codec.Codec{ | |||||||
| } | } | ||||||
|  |  | ||||||
| type noopClient struct { | type noopClient struct { | ||||||
| 	opts Options | 	funcCall   FuncCall | ||||||
| } | 	funcStream FuncStream | ||||||
|  | 	opts       Options | ||||||
| type noopMessage struct { |  | ||||||
| 	topic   string |  | ||||||
| 	payload interface{} |  | ||||||
| 	opts    MessageOptions |  | ||||||
| } | } | ||||||
|  |  | ||||||
| type noopRequest struct { | type noopRequest struct { | ||||||
| @@ -39,16 +38,12 @@ type noopRequest struct { | |||||||
|  |  | ||||||
| // NewClient returns new noop client | // NewClient returns new noop client | ||||||
| func NewClient(opts ...Option) Client { | func NewClient(opts ...Option) Client { | ||||||
| 	nc := &noopClient{opts: NewOptions(opts...)} | 	n := &noopClient{opts: NewOptions(opts...)} | ||||||
| 	// wrap in reverse |  | ||||||
|  |  | ||||||
| 	c := Client(nc) | 	n.funcCall = n.fnCall | ||||||
|  | 	n.funcStream = n.fnStream | ||||||
|  |  | ||||||
| 	for i := len(nc.opts.Wrappers); i > 0; i-- { | 	return n | ||||||
| 		c = nc.opts.Wrappers[i-1](c) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return c |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopClient) Name() string { | func (n *noopClient) Name() string { | ||||||
| @@ -100,10 +95,13 @@ func (n *noopResponse) Read() ([]byte, error) { | |||||||
| 	return nil, nil | 	return nil, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| type noopStream struct{} | type noopStream struct { | ||||||
|  | 	err error | ||||||
|  | 	ctx context.Context | ||||||
|  | } | ||||||
|  |  | ||||||
| func (n *noopStream) Context() context.Context { | func (n *noopStream) Context() context.Context { | ||||||
| 	return context.Background() | 	return n.ctx | ||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopStream) Request() Request { | func (n *noopStream) Request() Request { | ||||||
| @@ -131,47 +129,40 @@ func (n *noopStream) RecvMsg(interface{}) error { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopStream) Error() error { | func (n *noopStream) Error() error { | ||||||
| 	return nil | 	return n.err | ||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopStream) Close() error { | func (n *noopStream) Close() error { | ||||||
| 	return nil | 	if sp, ok := tracer.SpanFromContext(n.ctx); ok && sp != nil { | ||||||
|  | 		if n.err != nil { | ||||||
|  | 			sp.SetStatus(tracer.SpanStatusError, n.err.Error()) | ||||||
|  | 		} | ||||||
|  | 		sp.Finish() | ||||||
|  | 	} | ||||||
|  | 	return n.err | ||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopStream) CloseSend() error { | func (n *noopStream) CloseSend() error { | ||||||
| 	return nil | 	return n.err | ||||||
| } |  | ||||||
|  |  | ||||||
| func (n *noopMessage) Topic() string { |  | ||||||
| 	return n.topic |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (n *noopMessage) Payload() interface{} { |  | ||||||
| 	return n.payload |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (n *noopMessage) ContentType() string { |  | ||||||
| 	return n.opts.ContentType |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (n *noopMessage) Metadata() metadata.Metadata { |  | ||||||
| 	return n.opts.Metadata |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (n *noopClient) newCodec(contentType string) (codec.Codec, error) { |  | ||||||
| 	if cf, ok := n.opts.Codecs[contentType]; ok { |  | ||||||
| 		return cf, nil |  | ||||||
| 	} |  | ||||||
| 	if cf, ok := DefaultCodecs[contentType]; ok { |  | ||||||
| 		return cf, nil |  | ||||||
| 	} |  | ||||||
| 	return nil, codec.ErrUnknownContentType |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopClient) Init(opts ...Option) error { | func (n *noopClient) Init(opts ...Option) error { | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&n.opts) | 		o(&n.opts) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	n.funcCall = n.fnCall | ||||||
|  | 	n.funcStream = n.fnStream | ||||||
|  |  | ||||||
|  | 	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) | ||||||
|  | 		} | ||||||
|  | 	}) | ||||||
|  |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -184,6 +175,31 @@ func (n *noopClient) String() string { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopClient) Call(ctx context.Context, req Request, rsp interface{}, opts ...CallOption) error { | func (n *noopClient) Call(ctx context.Context, req Request, rsp interface{}, opts ...CallOption) error { | ||||||
|  | 	ts := time.Now() | ||||||
|  | 	n.opts.Meter.Counter(semconv.ClientRequestInflight, "endpoint", req.Endpoint()).Inc() | ||||||
|  | 	var sp tracer.Span | ||||||
|  | 	ctx, sp = n.opts.Tracer.Start(ctx, "rpc-client", | ||||||
|  | 		tracer.WithSpanKind(tracer.SpanKindClient), | ||||||
|  | 		tracer.WithSpanLabels("endpoint", req.Endpoint()), | ||||||
|  | 	) | ||||||
|  | 	err := n.funcCall(ctx, req, rsp, opts...) | ||||||
|  | 	n.opts.Meter.Counter(semconv.ClientRequestInflight, "endpoint", req.Endpoint()).Dec() | ||||||
|  | 	te := time.Since(ts) | ||||||
|  | 	n.opts.Meter.Summary(semconv.ClientRequestLatencyMicroseconds, "endpoint", req.Endpoint()).Update(te.Seconds()) | ||||||
|  | 	n.opts.Meter.Histogram(semconv.ClientRequestDurationSeconds, "endpoint", req.Endpoint()).Update(te.Seconds()) | ||||||
|  |  | ||||||
|  | 	if me := errors.FromError(err); me == nil { | ||||||
|  | 		sp.Finish() | ||||||
|  | 		n.opts.Meter.Counter(semconv.ClientRequestTotal, "endpoint", req.Endpoint(), "status", "success", "code", strconv.Itoa(int(200))).Inc() | ||||||
|  | 	} else { | ||||||
|  | 		sp.SetStatus(tracer.SpanStatusError, err.Error()) | ||||||
|  | 		n.opts.Meter.Counter(semconv.ClientRequestTotal, "endpoint", req.Endpoint(), "status", "failure", "code", strconv.Itoa(int(me.Code))).Inc() | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return err | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (n *noopClient) fnCall(ctx context.Context, req Request, rsp interface{}, opts ...CallOption) error { | ||||||
| 	// make a copy of call opts | 	// make a copy of call opts | ||||||
| 	callOpts := n.opts.CallOptions | 	callOpts := n.opts.CallOptions | ||||||
| 	for _, opt := range opts { | 	for _, opt := range opts { | ||||||
| @@ -212,11 +228,8 @@ func (n *noopClient) Call(ctx context.Context, req Request, rsp interface{}, opt | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// make copy of call method | 	// make copy of call method | ||||||
| 	hcall := n.call | 	hcall := func(ctx context.Context, addr string, req Request, rsp interface{}, opts CallOptions) error { | ||||||
|  | 		return nil | ||||||
| 	// wrap the call in reverse |  | ||||||
| 	for i := len(callOpts.CallWrappers); i > 0; i-- { |  | ||||||
| 		hcall = callOpts.CallWrappers[i-1](hcall) |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// use the router passed as a call option, or fallback to the rpc clients router | 	// use the router passed as a call option, or fallback to the rpc clients router | ||||||
| @@ -241,7 +254,7 @@ func (n *noopClient) Call(ctx context.Context, req Request, rsp interface{}, opt | |||||||
| 		// call backoff first. Someone may want an initial start delay | 		// call backoff first. Someone may want an initial start delay | ||||||
| 		t, err := callOpts.Backoff(ctx, req, i) | 		t, err := callOpts.Backoff(ctx, req, i) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			return errors.InternalServerError("go.micro.client", err.Error()) | 			return errors.InternalServerError("go.micro.client", "%s", err) | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		// only sleep if greater than 0 | 		// only sleep if greater than 0 | ||||||
| @@ -255,7 +268,7 @@ func (n *noopClient) Call(ctx context.Context, req Request, rsp interface{}, opt | |||||||
| 			// TODO apply any filtering here | 			// TODO apply any filtering here | ||||||
| 			routes, err = n.opts.Lookup(ctx, req, callOpts) | 			routes, err = n.opts.Lookup(ctx, req, callOpts) | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| 				return errors.InternalServerError("go.micro.client", err.Error()) | 				return errors.InternalServerError("go.micro.client", "%s", err) | ||||||
| 			} | 			} | ||||||
|  |  | ||||||
| 			// balance the list of nodes | 			// balance the list of nodes | ||||||
| @@ -315,20 +328,36 @@ func (n *noopClient) Call(ctx context.Context, req Request, rsp interface{}, opt | |||||||
| 	return gerr | 	return gerr | ||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopClient) call(ctx context.Context, addr string, req Request, rsp interface{}, opts CallOptions) error { | func (n *noopClient) NewRequest(service, endpoint string, _ interface{}, _ ...RequestOption) Request { | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (n *noopClient) NewRequest(service, endpoint string, req interface{}, opts ...RequestOption) Request { |  | ||||||
| 	return &noopRequest{service: service, endpoint: endpoint} | 	return &noopRequest{service: service, endpoint: endpoint} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopClient) NewMessage(topic string, msg interface{}, opts ...MessageOption) Message { | func (n *noopClient) Stream(ctx context.Context, req Request, opts ...CallOption) (Stream, error) { | ||||||
| 	options := NewMessageOptions(append([]MessageOption{MessageContentType(n.opts.ContentType)}, opts...)...) | 	ts := time.Now() | ||||||
| 	return &noopMessage{topic: topic, payload: msg, opts: options} | 	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) Stream(ctx context.Context, req Request, opts ...CallOption) (Stream, error) { | func (n *noopClient) fnStream(ctx context.Context, req Request, opts ...CallOption) (Stream, error) { | ||||||
| 	var err error | 	var err error | ||||||
|  |  | ||||||
| 	// make a copy of call opts | 	// make a copy of call opts | ||||||
| @@ -388,7 +417,7 @@ func (n *noopClient) Stream(ctx context.Context, req Request, opts ...CallOption | |||||||
| 		// call backoff first. Someone may want an initial start delay | 		// call backoff first. Someone may want an initial start delay | ||||||
| 		t, cerr := callOpts.Backoff(ctx, req, i) | 		t, cerr := callOpts.Backoff(ctx, req, i) | ||||||
| 		if cerr != nil { | 		if cerr != nil { | ||||||
| 			return nil, errors.InternalServerError("go.micro.client", cerr.Error()) | 			return nil, errors.InternalServerError("go.micro.client", "%s", cerr) | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		// only sleep if greater than 0 | 		// only sleep if greater than 0 | ||||||
| @@ -402,7 +431,7 @@ func (n *noopClient) Stream(ctx context.Context, req Request, opts ...CallOption | |||||||
| 			// TODO apply any filtering here | 			// TODO apply any filtering here | ||||||
| 			routes, err = n.opts.Lookup(ctx, req, callOpts) | 			routes, err = n.opts.Lookup(ctx, req, callOpts) | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| 				return nil, errors.InternalServerError("go.micro.client", err.Error()) | 				return nil, errors.InternalServerError("go.micro.client", "%s", err) | ||||||
| 			} | 			} | ||||||
|  |  | ||||||
| 			// balance the list of nodes | 			// balance the list of nodes | ||||||
| @@ -468,64 +497,6 @@ func (n *noopClient) Stream(ctx context.Context, req Request, opts ...CallOption | |||||||
| 	return nil, grr | 	return nil, grr | ||||||
| } | } | ||||||
|  |  | ||||||
| func (n *noopClient) stream(ctx context.Context, addr string, req Request, opts CallOptions) (Stream, error) { | func (n *noopClient) stream(ctx context.Context, _ string, _ Request, _ CallOptions) (Stream, error) { | ||||||
| 	return &noopStream{}, nil | 	return &noopStream{ctx: ctx}, nil | ||||||
| } |  | ||||||
|  |  | ||||||
| func (n *noopClient) BatchPublish(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.publish(ctx, []Message{p}, opts...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (n *noopClient) publish(ctx context.Context, ps []Message, opts ...PublishOption) error { |  | ||||||
| 	options := NewPublishOptions(opts...) |  | ||||||
|  |  | ||||||
| 	msgs := make([]*broker.Message, 0, len(ps)) |  | ||||||
|  |  | ||||||
| 	for _, p := range ps { |  | ||||||
| 		md, ok := metadata.FromOutgoingContext(ctx) |  | ||||||
| 		if !ok { |  | ||||||
| 			md = metadata.New(0) |  | ||||||
| 		} |  | ||||||
| 		md[metadata.HeaderContentType] = p.ContentType() |  | ||||||
|  |  | ||||||
| 		topic := p.Topic() |  | ||||||
|  |  | ||||||
| 		// get the exchange |  | ||||||
| 		if len(options.Exchange) > 0 { |  | ||||||
| 			topic = options.Exchange |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		md[metadata.HeaderTopic] = topic |  | ||||||
|  |  | ||||||
| 		var body []byte |  | ||||||
|  |  | ||||||
| 		// passed in raw data |  | ||||||
| 		if d, ok := p.Payload().(*codec.Frame); ok { |  | ||||||
| 			body = d.Data |  | ||||||
| 		} else { |  | ||||||
| 			// use codec for payload |  | ||||||
| 			cf, err := n.newCodec(p.ContentType()) |  | ||||||
| 			if err != nil { |  | ||||||
| 				return errors.InternalServerError("go.micro.client", err.Error()) |  | ||||||
| 			} |  | ||||||
|  |  | ||||||
| 			// set the body |  | ||||||
| 			b, err := cf.Marshal(p.Payload()) |  | ||||||
| 			if err != nil { |  | ||||||
| 				return errors.InternalServerError("go.micro.client", err.Error()) |  | ||||||
| 			} |  | ||||||
| 			body = b |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		msgs = append(msgs, &broker.Message{Header: md, Body: body}) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return n.opts.Broker.BatchPublish(ctx, msgs, |  | ||||||
| 		broker.PublishContext(options.Context), |  | ||||||
| 		broker.PublishBodyOnly(options.BodyOnly), |  | ||||||
| 	) |  | ||||||
| } | } | ||||||
|   | |||||||
| @@ -6,23 +6,31 @@ import ( | |||||||
| 	"net" | 	"net" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/broker" | 	"go.unistack.org/micro/v4/broker" | ||||||
| 	"go.unistack.org/micro/v3/codec" | 	"go.unistack.org/micro/v4/codec" | ||||||
| 	"go.unistack.org/micro/v3/logger" | 	"go.unistack.org/micro/v4/logger" | ||||||
| 	"go.unistack.org/micro/v3/metadata" | 	"go.unistack.org/micro/v4/metadata" | ||||||
| 	"go.unistack.org/micro/v3/meter" | 	"go.unistack.org/micro/v4/meter" | ||||||
| 	"go.unistack.org/micro/v3/network/transport" | 	"go.unistack.org/micro/v4/options" | ||||||
| 	"go.unistack.org/micro/v3/register" | 	"go.unistack.org/micro/v4/register" | ||||||
| 	"go.unistack.org/micro/v3/router" | 	"go.unistack.org/micro/v4/router" | ||||||
| 	"go.unistack.org/micro/v3/selector" | 	"go.unistack.org/micro/v4/selector" | ||||||
| 	"go.unistack.org/micro/v3/selector/random" | 	"go.unistack.org/micro/v4/selector/random" | ||||||
| 	"go.unistack.org/micro/v3/tracer" | 	"go.unistack.org/micro/v4/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Options holds client options | // Options holds client options | ||||||
| type Options struct { | type Options struct { | ||||||
| 	// Transport used for transfer messages | 	// Codecs map | ||||||
| 	Transport transport.Transport | 	Codecs map[string]codec.Codec | ||||||
|  |  | ||||||
|  | 	// Proxy is used for proxy requests | ||||||
|  | 	Proxy string | ||||||
|  | 	// ContentType is used to select codec | ||||||
|  | 	ContentType string | ||||||
|  | 	// Name is the client name | ||||||
|  | 	Name string | ||||||
|  |  | ||||||
| 	// Selector used to select needed address | 	// Selector used to select needed address | ||||||
| 	Selector selector.Selector | 	Selector selector.Selector | ||||||
| 	// Logger used to log messages | 	// Logger used to log messages | ||||||
| @@ -37,28 +45,28 @@ type Options struct { | |||||||
| 	Context context.Context | 	Context context.Context | ||||||
| 	// Router used to get route | 	// Router used to get route | ||||||
| 	Router router.Router | 	Router router.Router | ||||||
|  |  | ||||||
| 	// TLSConfig specifies tls.Config for secure connection | 	// TLSConfig specifies tls.Config for secure connection | ||||||
| 	TLSConfig *tls.Config | 	TLSConfig *tls.Config | ||||||
| 	// Codecs map |  | ||||||
| 	Codecs map[string]codec.Codec |  | ||||||
| 	// Lookup func used to get destination addr | 	// Lookup func used to get destination addr | ||||||
| 	Lookup LookupFunc | 	Lookup LookupFunc | ||||||
| 	// Proxy is used for proxy requests | 	// ContextDialer used to connect | ||||||
| 	Proxy string | 	ContextDialer func(context.Context, string) (net.Conn, error) | ||||||
| 	// ContentType is used to select codec |  | ||||||
| 	ContentType string |  | ||||||
| 	// Name is the client name |  | ||||||
| 	Name string |  | ||||||
| 	// Wrappers contains wrappers | 	// Wrappers contains wrappers | ||||||
| 	Wrappers []Wrapper | 	Wrappers []Wrapper | ||||||
|  | 	// Hooks can be run before broker Publish/BatchPublish and | ||||||
|  | 	// Subscribe/BatchSubscribe methods | ||||||
|  | 	Hooks options.Hooks | ||||||
|  |  | ||||||
| 	// CallOptions contains default CallOptions | 	// CallOptions contains default CallOptions | ||||||
| 	CallOptions CallOptions | 	CallOptions CallOptions | ||||||
|  |  | ||||||
| 	// PoolSize connection pool size | 	// PoolSize connection pool size | ||||||
| 	PoolSize int | 	PoolSize int | ||||||
| 	// PoolTTL connection pool ttl | 	// PoolTTL connection pool ttl | ||||||
| 	PoolTTL time.Duration | 	PoolTTL time.Duration | ||||||
| 	// ContextDialer used to connect |  | ||||||
| 	ContextDialer func(context.Context, string) (net.Conn, error) |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewCallOptions creates new call options struct | // NewCallOptions creates new call options struct | ||||||
| @@ -72,6 +80,16 @@ func NewCallOptions(opts ...CallOption) CallOptions { | |||||||
|  |  | ||||||
| // CallOptions holds client call options | // CallOptions holds client call options | ||||||
| type CallOptions struct { | type CallOptions struct { | ||||||
|  | 	// RequestMetadata holds additional metadata for call | ||||||
|  | 	RequestMetadata metadata.Metadata | ||||||
|  |  | ||||||
|  | 	// Network name | ||||||
|  | 	Network string | ||||||
|  | 	// Content-Type | ||||||
|  | 	ContentType string | ||||||
|  | 	// AuthToken string | ||||||
|  | 	AuthToken string | ||||||
|  |  | ||||||
| 	// Selector selects addr | 	// Selector selects addr | ||||||
| 	Selector selector.Selector | 	Selector selector.Selector | ||||||
| 	// Context used for deadline | 	// Context used for deadline | ||||||
| @@ -79,35 +97,30 @@ type CallOptions struct { | |||||||
| 	// Router used for route | 	// Router used for route | ||||||
| 	Router router.Router | 	Router router.Router | ||||||
| 	// Retry func used for retries | 	// Retry func used for retries | ||||||
|  |  | ||||||
|  | 	// ResponseMetadata holds additional metadata from call | ||||||
|  | 	ResponseMetadata *metadata.Metadata | ||||||
|  |  | ||||||
| 	Retry RetryFunc | 	Retry RetryFunc | ||||||
| 	// Backoff func used for backoff when retry | 	// Backoff func used for backoff when retry | ||||||
| 	Backoff BackoffFunc | 	Backoff BackoffFunc | ||||||
| 	// Network name | 	// ContextDialer used to connect | ||||||
| 	Network string | 	ContextDialer func(context.Context, string) (net.Conn, error) | ||||||
| 	// Content-Type |  | ||||||
| 	ContentType string |  | ||||||
| 	// AuthToken string |  | ||||||
| 	AuthToken string |  | ||||||
| 	// Address specifies static addr list | 	// Address specifies static addr list | ||||||
| 	Address []string | 	Address []string | ||||||
| 	// SelectOptions selector options | 	// SelectOptions selector options | ||||||
| 	SelectOptions []selector.SelectOption | 	SelectOptions []selector.SelectOption | ||||||
| 	// CallWrappers call wrappers |  | ||||||
| 	CallWrappers []CallWrapper |  | ||||||
| 	// StreamTimeout stream timeout | 	// StreamTimeout stream timeout | ||||||
| 	StreamTimeout time.Duration | 	StreamTimeout time.Duration | ||||||
| 	// RequestTimeout request timeout | 	// RequestTimeout request timeout | ||||||
| 	RequestTimeout time.Duration | 	RequestTimeout time.Duration | ||||||
| 	// RequestMetadata holds additional metadata for call |  | ||||||
| 	RequestMetadata metadata.Metadata |  | ||||||
| 	// ResponseMetadata holds additional metadata from call |  | ||||||
| 	ResponseMetadata *metadata.Metadata |  | ||||||
| 	// DialTimeout dial timeout | 	// DialTimeout dial timeout | ||||||
| 	DialTimeout time.Duration | 	DialTimeout time.Duration | ||||||
| 	// Retries specifies retries num | 	// Retries specifies retries num | ||||||
| 	Retries int | 	Retries int | ||||||
| 	// ContextDialer used to connect |  | ||||||
| 	ContextDialer func(context.Context, string) (net.Conn, error) |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // ContextDialer pass ContextDialer to client | // ContextDialer pass ContextDialer to client | ||||||
| @@ -124,43 +137,6 @@ func Context(ctx context.Context) Option { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewPublishOptions create new PublishOptions struct from option |  | ||||||
| func NewPublishOptions(opts ...PublishOption) PublishOptions { |  | ||||||
| 	options := PublishOptions{} |  | ||||||
| 	for _, o := range opts { |  | ||||||
| 		o(&options) |  | ||||||
| 	} |  | ||||||
| 	return options |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // PublishOptions holds publish options |  | ||||||
| type PublishOptions struct { |  | ||||||
| 	// Context used for external options |  | ||||||
| 	Context context.Context |  | ||||||
| 	// Exchange topic exchange name |  | ||||||
| 	Exchange string |  | ||||||
| 	// BodyOnly will publish only message body |  | ||||||
| 	BodyOnly bool |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewMessageOptions creates message options struct |  | ||||||
| func NewMessageOptions(opts ...MessageOption) MessageOptions { |  | ||||||
| 	options := MessageOptions{Metadata: metadata.New(1)} |  | ||||||
| 	for _, o := range opts { |  | ||||||
| 		o(&options) |  | ||||||
| 	} |  | ||||||
| 	return options |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // MessageOptions holds client message options |  | ||||||
| type MessageOptions struct { |  | ||||||
| 	// Metadata additional metadata |  | ||||||
| 	Metadata metadata.Metadata |  | ||||||
| 	// ContentType specify content-type of message |  | ||||||
| 	// deprecated |  | ||||||
| 	ContentType string |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewRequestOptions creates new RequestOptions struct | // NewRequestOptions creates new RequestOptions struct | ||||||
| func NewRequestOptions(opts ...RequestOption) RequestOptions { | func NewRequestOptions(opts ...RequestOption) RequestOptions { | ||||||
| 	options := RequestOptions{} | 	options := RequestOptions{} | ||||||
| @@ -185,25 +161,23 @@ func NewOptions(opts ...Option) Options { | |||||||
| 	options := Options{ | 	options := Options{ | ||||||
| 		Context:     context.Background(), | 		Context:     context.Background(), | ||||||
| 		ContentType: DefaultContentType, | 		ContentType: DefaultContentType, | ||||||
| 		Codecs:      make(map[string]codec.Codec), | 		Codecs:      DefaultCodecs, | ||||||
| 		CallOptions: CallOptions{ | 		CallOptions: CallOptions{ | ||||||
| 			Context:        context.Background(), | 			Context:        context.Background(), | ||||||
| 			Backoff:        DefaultBackoff, | 			Backoff:        DefaultBackoff, | ||||||
| 			Retry:          DefaultRetry, | 			Retry:          DefaultRetry, | ||||||
| 			Retries:        DefaultRetries, | 			Retries:        DefaultRetries, | ||||||
| 			RequestTimeout: DefaultRequestTimeout, | 			RequestTimeout: DefaultRequestTimeout, | ||||||
| 			DialTimeout:    transport.DefaultDialTimeout, |  | ||||||
| 		}, | 		}, | ||||||
| 		Lookup:    LookupRoute, | 		Lookup:   LookupRoute, | ||||||
| 		PoolSize:  DefaultPoolSize, | 		PoolSize: DefaultPoolSize, | ||||||
| 		PoolTTL:   DefaultPoolTTL, | 		PoolTTL:  DefaultPoolTTL, | ||||||
| 		Selector:  random.NewSelector(), | 		Selector: random.NewSelector(), | ||||||
| 		Logger:    logger.DefaultLogger, | 		Logger:   logger.DefaultLogger, | ||||||
| 		Broker:    broker.DefaultBroker, | 		Broker:   broker.DefaultBroker, | ||||||
| 		Meter:     meter.DefaultMeter, | 		Meter:    meter.DefaultMeter, | ||||||
| 		Tracer:    tracer.DefaultTracer, | 		Tracer:   tracer.DefaultTracer, | ||||||
| 		Router:    router.DefaultRouter, | 		Router:   router.DefaultRouter, | ||||||
| 		Transport: transport.DefaultTransport, |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| @@ -276,13 +250,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 | // Register sets the routers register | ||||||
| func Register(r register.Register) Option { | func Register(r register.Register) Option { | ||||||
| 	return func(o *Options) { | 	return func(o *Options) { | ||||||
| @@ -306,20 +273,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 | // Backoff is used to set the backoff function used when retrying Calls | ||||||
| func Backoff(fn BackoffFunc) Option { | func Backoff(fn BackoffFunc) Option { | ||||||
| 	return func(o *Options) { | 	return func(o *Options) { | ||||||
| @@ -346,14 +299,6 @@ func TLSConfig(t *tls.Config) Option { | |||||||
| 	return func(o *Options) { | 	return func(o *Options) { | ||||||
| 		// set the internal tls | 		// set the internal tls | ||||||
| 		o.TLSConfig = t | 		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), |  | ||||||
| 		) |  | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -392,43 +337,6 @@ func DialTimeout(d time.Duration) Option { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // WithExchange sets the exchange to route a message through |  | ||||||
| // Deprecated |  | ||||||
| func WithExchange(e string) PublishOption { |  | ||||||
| 	return func(o *PublishOptions) { |  | ||||||
| 		o.Exchange = e |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // PublishExchange sets the exchange to route a message through |  | ||||||
| func PublishExchange(e string) PublishOption { |  | ||||||
| 	return func(o *PublishOptions) { |  | ||||||
| 		o.Exchange = e |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // WithBodyOnly publish only message body |  | ||||||
| // DERECATED |  | ||||||
| func WithBodyOnly(b bool) PublishOption { |  | ||||||
| 	return func(o *PublishOptions) { |  | ||||||
| 		o.BodyOnly = b |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // PublishBodyOnly publish only message body |  | ||||||
| func PublishBodyOnly(b bool) PublishOption { |  | ||||||
| 	return func(o *PublishOptions) { |  | ||||||
| 		o.BodyOnly = b |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // PublishContext sets the context in publish options |  | ||||||
| func PublishContext(ctx context.Context) PublishOption { |  | ||||||
| 	return func(o *PublishOptions) { |  | ||||||
| 		o.Context = ctx |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // WithContextDialer pass ContextDialer to client call | // WithContextDialer pass ContextDialer to client call | ||||||
| func WithContextDialer(fn func(context.Context, string) (net.Conn, error)) CallOption { | func WithContextDialer(fn func(context.Context, string) (net.Conn, error)) CallOption { | ||||||
| 	return func(o *CallOptions) { | 	return func(o *CallOptions) { | ||||||
| @@ -450,13 +358,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 | // WithBackoff is a CallOption which overrides that which | ||||||
| // set in Options.CallOptions | // set in Options.CallOptions | ||||||
| func WithBackoff(fn BackoffFunc) CallOption { | func WithBackoff(fn BackoffFunc) CallOption { | ||||||
| @@ -526,13 +427,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 | // WithRouter sets the router to use for this call | ||||||
| func WithRouter(r router.Router) CallOption { | func WithRouter(r router.Router) CallOption { | ||||||
| 	return func(o *CallOptions) { | 	return func(o *CallOptions) { | ||||||
| @@ -554,30 +448,6 @@ func WithSelectOptions(sops ...selector.SelectOption) CallOption { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // WithMessageContentType sets the message content type |  | ||||||
| // Deprecated |  | ||||||
| func WithMessageContentType(ct string) MessageOption { |  | ||||||
| 	return func(o *MessageOptions) { |  | ||||||
| 		o.Metadata.Set(metadata.HeaderContentType, ct) |  | ||||||
| 		o.ContentType = ct |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // MessageContentType sets the message content type |  | ||||||
| func MessageContentType(ct string) MessageOption { |  | ||||||
| 	return func(o *MessageOptions) { |  | ||||||
| 		o.Metadata.Set(metadata.HeaderContentType, ct) |  | ||||||
| 		o.ContentType = ct |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // MessageMetadata sets the message metadata |  | ||||||
| func MessageMetadata(k, v string) MessageOption { |  | ||||||
| 	return func(o *MessageOptions) { |  | ||||||
| 		o.Metadata.Set(k, v) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // StreamingRequest specifies that request is streaming | // StreamingRequest specifies that request is streaming | ||||||
| func StreamingRequest(b bool) RequestOption { | func StreamingRequest(b bool) RequestOption { | ||||||
| 	return func(o *RequestOptions) { | 	return func(o *RequestOptions) { | ||||||
| @@ -591,3 +461,10 @@ func RequestContentType(ct string) RequestOption { | |||||||
| 		o.ContentType = ct | 		o.ContentType = ct | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // Hooks sets hook runs before action | ||||||
|  | func Hooks(h ...options.Hook) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Hooks = append(o.Hooks, h...) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -3,7 +3,7 @@ package client | |||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/errors" | 	"go.unistack.org/micro/v4/errors" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // RetryFunc that returning either false or a non-nil error will result in the call not being retried | // RetryFunc that returning either false or a non-nil error will result in the call not being retried | ||||||
|   | |||||||
| @@ -5,7 +5,7 @@ import ( | |||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"testing" | 	"testing" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/errors" | 	"go.unistack.org/micro/v4/errors" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| func TestRetryAlways(t *testing.T) { | func TestRetryAlways(t *testing.T) { | ||||||
|   | |||||||
| @@ -1,7 +1,7 @@ | |||||||
| package client | package client | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"go.unistack.org/micro/v3/codec" | 	"go.unistack.org/micro/v4/codec" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type testRequest struct { | type testRequest struct { | ||||||
|   | |||||||
							
								
								
									
										47
									
								
								cluster/cluster.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										47
									
								
								cluster/cluster.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,47 @@ | |||||||
|  | package cluster | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v4/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,10 @@ | |||||||
| // Package codec is an interface for encoding messages | // Package codec is an interface for encoding messages | ||||||
| package codec // import "go.unistack.org/micro/v3/codec" | package codec | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"errors" | 	"errors" | ||||||
| 	"io" |  | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/metadata" | 	"gopkg.in/yaml.v3" | ||||||
| ) |  | ||||||
|  |  | ||||||
| // Message types |  | ||||||
| const ( |  | ||||||
| 	Error MessageType = iota |  | ||||||
| 	Request |  | ||||||
| 	Response |  | ||||||
| 	Event |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| @@ -24,63 +15,63 @@ var ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| 	// DefaultMaxMsgSize specifies how much data codec can handle |  | ||||||
| 	DefaultMaxMsgSize = 1024 * 1024 * 4 // 4Mb |  | ||||||
| 	// DefaultCodec is the global default codec | 	// DefaultCodec is the global default codec | ||||||
| 	DefaultCodec = NewCodec() | 	DefaultCodec = NewCodec() | ||||||
| 	// DefaultTagName specifies struct tag name to control codec Marshal/Unmarshal | 	// DefaultTagName specifies struct tag name to control codec Marshal/Unmarshal | ||||||
| 	DefaultTagName = "codec" | 	DefaultTagName = "codec" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // MessageType specifies message type for codec | // Codec encodes/decodes various types of messages. | ||||||
| type MessageType int |  | ||||||
|  |  | ||||||
| // Codec encodes/decodes various types of messages used within micro. |  | ||||||
| // ReadHeader and ReadBody are called in pairs to read requests/responses |  | ||||||
| // from the connection. Close is called when finished with the |  | ||||||
| // connection. ReadBody may be called with a nil argument to force the |  | ||||||
| // body to be read and discarded. |  | ||||||
| type Codec interface { | type Codec interface { | ||||||
| 	ReadHeader(r io.Reader, m *Message, mt MessageType) error |  | ||||||
| 	ReadBody(r io.Reader, v interface{}) error |  | ||||||
| 	Write(w io.Writer, m *Message, v interface{}) error |  | ||||||
| 	Marshal(v interface{}, opts ...Option) ([]byte, error) | 	Marshal(v interface{}, opts ...Option) ([]byte, error) | ||||||
| 	Unmarshal(b []byte, v interface{}, opts ...Option) error | 	Unmarshal(b []byte, v interface{}, opts ...Option) error | ||||||
| 	String() string | 	String() string | ||||||
| } | } | ||||||
|  |  | ||||||
| // Message represents detailed information about | type CodecV2 interface { | ||||||
| // the communication, likely followed by the body. | 	Marshal(buf []byte, v interface{}, opts ...Option) ([]byte, error) | ||||||
| // In the case of an error, body may be nil. | 	Unmarshal(buf []byte, v interface{}, opts ...Option) error | ||||||
| type Message struct { | 	String() string | ||||||
| 	Header   metadata.Metadata |  | ||||||
| 	Target   string |  | ||||||
| 	Method   string |  | ||||||
| 	Endpoint string |  | ||||||
| 	Error    string |  | ||||||
| 	ID       string |  | ||||||
| 	Body     []byte |  | ||||||
| 	Type     MessageType |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewMessage creates new codec message | // RawMessage is a raw encoded JSON value. | ||||||
| func NewMessage(t MessageType) *Message { | // It implements Marshaler and Unmarshaler and can be used to delay decoding or precompute a encoding. | ||||||
| 	return &Message{Type: t, Header: metadata.New(0)} | type RawMessage []byte | ||||||
| } |  | ||||||
|  |  | ||||||
| // MarshalAppend calls codec.Marshal(v) and returns the data appended to buf. | // MarshalJSON returns m as the JSON encoding of m. | ||||||
| // If codec implements MarshalAppend, that is called instead. | func (m *RawMessage) MarshalJSON() ([]byte, error) { | ||||||
| func MarshalAppend(buf []byte, c Codec, v interface{}, opts ...Option) ([]byte, error) { | 	if m == nil { | ||||||
| 	if nc, ok := c.(interface { | 		return []byte("null"), nil | ||||||
| 		MarshalAppend([]byte, interface{}, ...Option) ([]byte, error) | 	} else if len(*m) == 0 { | ||||||
| 	}); ok { | 		return []byte("null"), nil | ||||||
| 		return nc.MarshalAppend(buf, v, opts...) |  | ||||||
| 	} | 	} | ||||||
|  | 	return *m, nil | ||||||
| 	mbuf, err := c.Marshal(v, opts...) | } | ||||||
| 	if err != nil { |  | ||||||
| 		return nil, err | // UnmarshalJSON sets *m to a copy of data. | ||||||
| 	} | func (m *RawMessage) UnmarshalJSON(data []byte) error { | ||||||
|  | 	if m == nil { | ||||||
| 	return append(buf, mbuf...), nil | 		return errors.New("RawMessage UnmarshalJSON on nil pointer") | ||||||
|  | 	} | ||||||
|  | 	*m = append((*m)[0:0], data...) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // MarshalYAML returns m as the JSON encoding of m. | ||||||
|  | func (m *RawMessage) MarshalYAML() ([]byte, error) { | ||||||
|  | 	if m == nil { | ||||||
|  | 		return []byte("null"), nil | ||||||
|  | 	} else if len(*m) == 0 { | ||||||
|  | 		return []byte("null"), nil | ||||||
|  | 	} | ||||||
|  | 	return *m, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // UnmarshalYAML sets *m to a copy of data. | ||||||
|  | func (m *RawMessage) UnmarshalYAML(n *yaml.Node) error { | ||||||
|  | 	if m == nil { | ||||||
|  | 		return errors.New("RawMessage UnmarshalYAML on nil pointer") | ||||||
|  | 	} | ||||||
|  | 	*m = append((*m)[0:0], []byte(n.Value)...) | ||||||
|  | 	return nil | ||||||
| } | } | ||||||
|   | |||||||
| @@ -15,6 +15,15 @@ func FromContext(ctx context.Context) (Codec, bool) { | |||||||
| 	return c, ok | 	return c, ok | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // MustContext returns codec from context | ||||||
|  | func MustContext(ctx context.Context) Codec { | ||||||
|  | 	c, ok := FromContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		panic("missing codec") | ||||||
|  | 	} | ||||||
|  | 	return c | ||||||
|  | } | ||||||
|  |  | ||||||
| // NewContext put codec in context | // NewContext put codec in context | ||||||
| func NewContext(ctx context.Context, c Codec) context.Context { | func NewContext(ctx context.Context, c Codec) context.Context { | ||||||
| 	if ctx == nil { | 	if ctx == nil { | ||||||
|   | |||||||
| @@ -1,5 +1,7 @@ | |||||||
| package codec | package codec | ||||||
|  |  | ||||||
|  | import "gopkg.in/yaml.v3" | ||||||
|  |  | ||||||
| // Frame gives us the ability to define raw data to send over the pipes | // Frame gives us the ability to define raw data to send over the pipes | ||||||
| type Frame struct { | type Frame struct { | ||||||
| 	Data []byte | 	Data []byte | ||||||
| @@ -20,6 +22,17 @@ func (m *Frame) UnmarshalJSON(data []byte) error { | |||||||
| 	return m.Unmarshal(data) | 	return m.Unmarshal(data) | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // MarshalYAML returns frame data | ||||||
|  | func (m *Frame) MarshalYAML() ([]byte, error) { | ||||||
|  | 	return m.Marshal() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // UnmarshalYAML set frame data | ||||||
|  | func (m *Frame) UnmarshalYAML(n *yaml.Node) error { | ||||||
|  | 	m.Data = []byte(n.Value) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
| // ProtoMessage noop func | // ProtoMessage noop func | ||||||
| func (m *Frame) ProtoMessage() {} | func (m *Frame) ProtoMessage() {} | ||||||
|  |  | ||||||
|   | |||||||
| @@ -17,7 +17,7 @@ syntax = "proto3"; | |||||||
| package micro.codec; | package micro.codec; | ||||||
|  |  | ||||||
| option cc_enable_arenas = true; | option cc_enable_arenas = true; | ||||||
| option go_package = "go.unistack.org/micro/v3/codec;codec"; | option go_package = "go.unistack.org/micro/v4/codec;codec"; | ||||||
| option java_multiple_files = true; | option java_multiple_files = true; | ||||||
| option java_outer_classname = "MicroCodec"; | option java_outer_classname = "MicroCodec"; | ||||||
| option java_package = "micro.codec"; | option java_package = "micro.codec"; | ||||||
|   | |||||||
| @@ -2,70 +2,14 @@ package codec | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"encoding/json" | 	"encoding/json" | ||||||
| 	"io" |  | ||||||
|  | 	codecpb "go.unistack.org/micro-proto/v4/codec" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type noopCodec struct { | type noopCodec struct { | ||||||
| 	opts Options | 	opts Options | ||||||
| } | } | ||||||
|  |  | ||||||
| func (c *noopCodec) ReadHeader(conn io.Reader, m *Message, t MessageType) error { |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (c *noopCodec) ReadBody(conn io.Reader, b interface{}) error { |  | ||||||
| 	// read bytes |  | ||||||
| 	buf, err := io.ReadAll(conn) |  | ||||||
| 	if err != nil { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if b == nil { |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	switch v := b.(type) { |  | ||||||
| 	case *string: |  | ||||||
| 		*v = string(buf) |  | ||||||
| 	case *[]byte: |  | ||||||
| 		*v = buf |  | ||||||
| 	case *Frame: |  | ||||||
| 		v.Data = buf |  | ||||||
| 	default: |  | ||||||
| 		return json.Unmarshal(buf, v) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (c *noopCodec) Write(conn io.Writer, m *Message, b interface{}) error { |  | ||||||
| 	if b == nil { |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	var v []byte |  | ||||||
| 	switch vb := b.(type) { |  | ||||||
| 	case *Frame: |  | ||||||
| 		v = vb.Data |  | ||||||
| 	case string: |  | ||||||
| 		v = []byte(vb) |  | ||||||
| 	case *string: |  | ||||||
| 		v = []byte(*vb) |  | ||||||
| 	case *[]byte: |  | ||||||
| 		v = *vb |  | ||||||
| 	case []byte: |  | ||||||
| 		v = vb |  | ||||||
| 	default: |  | ||||||
| 		var err error |  | ||||||
| 		v, err = json.Marshal(vb) |  | ||||||
| 		if err != nil { |  | ||||||
| 			return err |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 	_, err := conn.Write(v) |  | ||||||
| 	return err |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (c *noopCodec) String() string { | func (c *noopCodec) String() string { | ||||||
| 	return "noop" | 	return "noop" | ||||||
| } | } | ||||||
| @@ -91,8 +35,8 @@ func (c *noopCodec) Marshal(v interface{}, opts ...Option) ([]byte, error) { | |||||||
| 		return ve, nil | 		return ve, nil | ||||||
| 	case *Frame: | 	case *Frame: | ||||||
| 		return ve.Data, nil | 		return ve.Data, nil | ||||||
| 	case *Message: | 	case *codecpb.Frame: | ||||||
| 		return ve.Body, nil | 		return ve.Data, nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return json.Marshal(v) | 	return json.Marshal(v) | ||||||
| @@ -115,8 +59,8 @@ func (c *noopCodec) Unmarshal(d []byte, v interface{}, opts ...Option) error { | |||||||
| 	case *Frame: | 	case *Frame: | ||||||
| 		ve.Data = d | 		ve.Data = d | ||||||
| 		return nil | 		return nil | ||||||
| 	case *Message: | 	case *codecpb.Frame: | ||||||
| 		ve.Body = d | 		ve.Data = d | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|   | |||||||
| @@ -3,9 +3,9 @@ package codec | |||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/logger" | 	"go.unistack.org/micro/v4/logger" | ||||||
| 	"go.unistack.org/micro/v3/meter" | 	"go.unistack.org/micro/v4/meter" | ||||||
| 	"go.unistack.org/micro/v3/tracer" | 	"go.unistack.org/micro/v4/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Option func | // Option func | ||||||
| @@ -23,15 +23,8 @@ type Options struct { | |||||||
| 	Context context.Context | 	Context context.Context | ||||||
| 	// TagName specifies tag name in struct to control codec | 	// TagName specifies tag name in struct to control codec | ||||||
| 	TagName string | 	TagName string | ||||||
| 	// MaxMsgSize specifies max messages size that reads by codec | 	// Flatten specifies that struct must be analyzed for flatten tag | ||||||
| 	MaxMsgSize int | 	Flatten bool | ||||||
| } |  | ||||||
|  |  | ||||||
| // MaxMsgSize sets the max message size |  | ||||||
| func MaxMsgSize(n int) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.MaxMsgSize = n |  | ||||||
| 	} |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // TagName sets the codec tag name in struct | // TagName sets the codec tag name in struct | ||||||
| @@ -41,6 +34,13 @@ func TagName(n string) Option { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // Flatten enables checking for flatten tag name | ||||||
|  | func Flatten(b bool) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Flatten = b | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // Logger sets the logger | // Logger sets the logger | ||||||
| func Logger(l logger.Logger) Option { | func Logger(l logger.Logger) Option { | ||||||
| 	return func(o *Options) { | 	return func(o *Options) { | ||||||
| @@ -65,12 +65,12 @@ func Meter(m meter.Meter) Option { | |||||||
| // NewOptions returns new options | // NewOptions returns new options | ||||||
| func NewOptions(opts ...Option) Options { | func NewOptions(opts ...Option) Options { | ||||||
| 	options := Options{ | 	options := Options{ | ||||||
| 		Context:    context.Background(), | 		Context: context.Background(), | ||||||
| 		Logger:     logger.DefaultLogger, | 		Logger:  logger.DefaultLogger, | ||||||
| 		Meter:      meter.DefaultMeter, | 		Meter:   meter.DefaultMeter, | ||||||
| 		Tracer:     tracer.DefaultTracer, | 		Tracer:  tracer.DefaultTracer, | ||||||
| 		MaxMsgSize: DefaultMaxMsgSize, | 		TagName: DefaultTagName, | ||||||
| 		TagName:    DefaultTagName, | 		Flatten: false, | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
|   | |||||||
							
								
								
									
										107
									
								
								config/config.go
									
									
									
									
									
								
							
							
						
						
									
										107
									
								
								config/config.go
									
									
									
									
									
								
							| @@ -1,5 +1,5 @@ | |||||||
| // Package config is an interface for dynamic configuration. | // Package config is an interface for dynamic configuration. | ||||||
| package config // import "go.unistack.org/micro/v3/config" | package config | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| @@ -50,6 +50,13 @@ type Config interface { | |||||||
| 	String() string | 	String() string | ||||||
| } | } | ||||||
|  |  | ||||||
|  | type ( | ||||||
|  | 	FuncLoad func(ctx context.Context, opts ...LoadOption) error | ||||||
|  | 	HookLoad func(next FuncLoad) FuncLoad | ||||||
|  | 	FuncSave func(ctx context.Context, opts ...SaveOption) error | ||||||
|  | 	HookSave func(next FuncSave) FuncSave | ||||||
|  | ) | ||||||
|  |  | ||||||
| // Watcher is the config watcher | // Watcher is the config watcher | ||||||
| type Watcher interface { | type Watcher interface { | ||||||
| 	// Next blocks until update happens or error returned | 	// Next blocks until update happens or error returned | ||||||
| @@ -124,35 +131,14 @@ func Validate(ctx context.Context, cfg interface{}) error { | |||||||
| } | } | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| 	// DefaultAfterLoad default func that runs after config load | 	// DefaultBeforeLoad default func that runs before 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 = func(ctx context.Context, c Config) error { | 	DefaultBeforeLoad = func(ctx context.Context, c Config) error { | ||||||
| 		for _, fn := range c.Options().BeforeLoad { | 		for _, fn := range c.Options().BeforeLoad { | ||||||
|  | 			if fn == nil { | ||||||
|  | 				return nil | ||||||
|  | 			} | ||||||
| 			if err := fn(ctx, c); err != nil { | 			if err := fn(ctx, c); err != nil { | ||||||
| 				c.Options().Logger.Errorf(ctx, "%s BeforeLoad err: %v", c.String(), err) | 				c.Options().Logger.Error(ctx, c.String()+" BeforeLoad error", err) | ||||||
| 				if !c.Options().AllowFail { | 				if !c.Options().AllowFail { | ||||||
| 					return err | 					return err | ||||||
| 				} | 				} | ||||||
| @@ -160,11 +146,74 @@ var ( | |||||||
| 		} | 		} | ||||||
| 		return nil | 		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 { | 	DefaultBeforeSave = func(ctx context.Context, c Config) error { | ||||||
| 		for _, fn := range c.Options().BeforeSave { | 		for _, fn := range c.Options().BeforeSave { | ||||||
|  | 			if fn == nil { | ||||||
|  | 				return nil | ||||||
|  | 			} | ||||||
| 			if err := fn(ctx, c); err != nil { | 			if err := fn(ctx, c); err != nil { | ||||||
| 				c.Options().Logger.Errorf(ctx, "%s BeforeSave err: %v", c.String(), err) | 				c.Options().Logger.Error(ctx, c.String()+" BeforeSave error", err) | ||||||
|  | 				if !c.Options().AllowFail { | ||||||
|  | 					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 { | 				if !c.Options().AllowFail { | ||||||
| 					return err | 					return err | ||||||
| 				} | 				} | ||||||
|   | |||||||
| @@ -15,6 +15,15 @@ func FromContext(ctx context.Context) (Config, bool) { | |||||||
| 	return c, ok | 	return c, ok | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // MustContext returns store from context | ||||||
|  | func MustContext(ctx context.Context) Config { | ||||||
|  | 	c, ok := FromContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		panic("missing config") | ||||||
|  | 	} | ||||||
|  | 	return c | ||||||
|  | } | ||||||
|  |  | ||||||
| // NewContext put store in context | // NewContext put store in context | ||||||
| func NewContext(ctx context.Context, c Config) context.Context { | func NewContext(ctx context.Context, c Config) context.Context { | ||||||
| 	if ctx == nil { | 	if ctx == nil { | ||||||
|   | |||||||
| @@ -5,13 +5,20 @@ import ( | |||||||
| 	"reflect" | 	"reflect" | ||||||
| 	"strconv" | 	"strconv" | ||||||
| 	"strings" | 	"strings" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
| 	"github.com/imdario/mergo" | 	"dario.cat/mergo" | ||||||
| 	rutil "go.unistack.org/micro/v3/util/reflect" | 	"github.com/google/uuid" | ||||||
|  | 	"go.unistack.org/micro/v4/options" | ||||||
|  | 	mid "go.unistack.org/micro/v4/util/id" | ||||||
|  | 	rutil "go.unistack.org/micro/v4/util/reflect" | ||||||
|  | 	mtime "go.unistack.org/micro/v4/util/time" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type defaultConfig struct { | type defaultConfig struct { | ||||||
| 	opts Options | 	funcLoad FuncLoad | ||||||
|  | 	funcSave FuncSave | ||||||
|  | 	opts     Options | ||||||
| } | } | ||||||
|  |  | ||||||
| func (c *defaultConfig) Options() Options { | func (c *defaultConfig) Options() Options { | ||||||
| @@ -22,11 +29,42 @@ func (c *defaultConfig) Init(opts ...Option) error { | |||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&c.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 | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (c *defaultConfig) Load(ctx context.Context, opts ...LoadOption) error { | 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 | 		return err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -49,21 +87,20 @@ func (c *defaultConfig) Load(ctx context.Context, opts ...LoadOption) error { | |||||||
| 		if !c.opts.AllowFail { | 		if !c.opts.AllowFail { | ||||||
| 			return err | 			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 { | 	if err = fillValues(reflect.ValueOf(src), c.opts.StructTag); err == nil { | ||||||
| 		err = mergo.Merge(dst, src, mopts...) | 		err = mergo.Merge(dst, src, mopts...) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if err != nil { | 	if err != nil && !c.opts.AllowFail { | ||||||
| 		c.opts.Logger.Errorf(ctx, "default load error: %v", err) | 		return err | ||||||
| 		if !c.opts.AllowFail { |  | ||||||
| 			return err |  | ||||||
| 		} |  | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if err := DefaultAfterLoad(ctx, c); err != nil { | 	if err := DefaultAfterLoad(ctx, c); err != nil && !c.opts.AllowFail { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -75,6 +112,7 @@ func fillValue(value reflect.Value, val string) error { | |||||||
| 	if !rutil.IsEmpty(value) { | 	if !rutil.IsEmpty(value) { | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	switch value.Kind() { | 	switch value.Kind() { | ||||||
| 	case reflect.Map: | 	case reflect.Map: | ||||||
| 		t := value.Type() | 		t := value.Type() | ||||||
| @@ -113,6 +151,20 @@ func fillValue(value reflect.Value, val string) error { | |||||||
| 		} | 		} | ||||||
| 		value.Set(reflect.ValueOf(v)) | 		value.Set(reflect.ValueOf(v)) | ||||||
| 	case reflect.String: | 	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)) | 		value.Set(reflect.ValueOf(val)) | ||||||
| 	case reflect.Float32: | 	case reflect.Float32: | ||||||
| 		v, err := strconv.ParseFloat(val, 32) | 		v, err := strconv.ParseFloat(val, 32) | ||||||
| @@ -151,11 +203,26 @@ func fillValue(value reflect.Value, val string) error { | |||||||
| 		} | 		} | ||||||
| 		value.Set(reflect.ValueOf(int32(v))) | 		value.Set(reflect.ValueOf(int32(v))) | ||||||
| 	case reflect.Int64: | 	case reflect.Int64: | ||||||
| 		v, err := strconv.ParseInt(val, 10, 64) | 		switch { | ||||||
| 		if err != nil { | 		case value.Type().String() == "time.Duration" && value.Type().PkgPath() == "time": | ||||||
| 			return err | 			v, err := time.ParseDuration(val) | ||||||
|  | 			if err != nil { | ||||||
|  | 				return err | ||||||
|  | 			} | ||||||
|  | 			value.Set(reflect.ValueOf(v)) | ||||||
|  | 		case value.Type().String() == "time.Duration" && value.Type().PkgPath() == "go.unistack.org/micro/v4/util/time": | ||||||
|  | 			v, err := mtime.ParseDuration(val) | ||||||
|  | 			if err != nil { | ||||||
|  | 				return err | ||||||
|  | 			} | ||||||
|  | 			value.SetInt(int64(v)) | ||||||
|  | 		default: | ||||||
|  | 			v, err := strconv.ParseInt(val, 10, 64) | ||||||
|  | 			if err != nil { | ||||||
|  | 				return err | ||||||
|  | 			} | ||||||
|  | 			value.Set(reflect.ValueOf(v)) | ||||||
| 		} | 		} | ||||||
| 		value.Set(reflect.ValueOf(v)) |  | ||||||
| 	case reflect.Uint: | 	case reflect.Uint: | ||||||
| 		v, err := strconv.ParseUint(val, 10, 0) | 		v, err := strconv.ParseUint(val, 10, 0) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| @@ -187,6 +254,7 @@ func fillValue(value reflect.Value, val string) error { | |||||||
| 		} | 		} | ||||||
| 		value.Set(reflect.ValueOf(v)) | 		value.Set(reflect.ValueOf(v)) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -250,6 +318,14 @@ func fillValues(valueOf reflect.Value, tname string) error { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (c *defaultConfig) Save(ctx context.Context, opts ...SaveOption) error { | func (c *defaultConfig) Save(ctx context.Context, opts ...SaveOption) error { | ||||||
|  | 	return c.funcSave(ctx, opts...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (c *defaultConfig) fnSave(ctx context.Context, opts ...SaveOption) error { | ||||||
|  | 	if c.opts.SkipSave != nil && c.opts.SkipSave(ctx, c) { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	if err := DefaultBeforeSave(ctx, c); err != nil { | 	if err := DefaultBeforeSave(ctx, c); err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| @@ -269,7 +345,7 @@ func (c *defaultConfig) Name() string { | |||||||
| 	return c.opts.Name | 	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 | 	return nil, ErrWatcherNotImplemented | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -279,5 +355,9 @@ func NewConfig(opts ...Option) Config { | |||||||
| 	if len(options.StructTag) == 0 { | 	if len(options.StructTag) == 0 { | ||||||
| 		options.StructTag = "default" | 		options.StructTag = "default" | ||||||
| 	} | 	} | ||||||
| 	return &defaultConfig{opts: options} | 	c := &defaultConfig{opts: options} | ||||||
|  | 	c.funcLoad = c.fnLoad | ||||||
|  | 	c.funcSave = c.fnSave | ||||||
|  |  | ||||||
|  | 	return c | ||||||
| } | } | ||||||
|   | |||||||
| @@ -3,16 +3,26 @@ package config_test | |||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"fmt" | 	"fmt" | ||||||
|  | 	"reflect" | ||||||
| 	"testing" | 	"testing" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/config" | 	"go.unistack.org/micro/v4/config" | ||||||
|  | 	mtime "go.unistack.org/micro/v4/util/time" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type cfg struct { | type cfg struct { | ||||||
|  | 	MapValue    map[string]bool `default:"key1=true,key2=false"` | ||||||
|  | 	StructValue *cfgStructValue | ||||||
|  |  | ||||||
| 	StringValue string `default:"string_value"` | 	StringValue string `default:"string_value"` | ||||||
| 	IgnoreValue string `json:"-"` | 	IgnoreValue string `json:"-"` | ||||||
| 	StructValue *cfgStructValue | 	UUIDValue   string `default:"micro:generate uuid"` | ||||||
| 	IntValue    int `default:"99"` | 	IDValue     string `default:"micro:generate id"` | ||||||
|  |  | ||||||
|  | 	DurationValue  time.Duration  `default:"10s"` | ||||||
|  | 	MDurationValue mtime.Duration `default:"10s"` | ||||||
|  | 	IntValue       int            `default:"99"` | ||||||
| } | } | ||||||
|  |  | ||||||
| type cfgStructValue struct { | type cfgStructValue struct { | ||||||
| @@ -33,6 +43,35 @@ func (c *cfgStructValue) Validate() error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
|  | type testHook struct { | ||||||
|  | 	f bool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (t *testHook) Load(fn config.FuncLoad) config.FuncLoad { | ||||||
|  | 	return func(ctx context.Context, opts ...config.LoadOption) error { | ||||||
|  | 		t.f = true | ||||||
|  | 		return fn(ctx, opts...) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestHook(t *testing.T) { | ||||||
|  | 	h := &testHook{} | ||||||
|  |  | ||||||
|  | 	c := config.NewConfig(config.Struct(h), config.Hooks(config.HookLoad(h.Load))) | ||||||
|  |  | ||||||
|  | 	if err := c.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if err := c.Load(context.TODO()); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !h.f { | ||||||
|  | 		t.Fatal("hook not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| func TestDefault(t *testing.T) { | func TestDefault(t *testing.T) { | ||||||
| 	ctx := context.Background() | 	ctx := context.Background() | ||||||
| 	conf := &cfg{IntValue: 10} | 	conf := &cfg{IntValue: 10} | ||||||
| @@ -63,6 +102,19 @@ func TestDefault(t *testing.T) { | |||||||
| 	if conf.StringValue != "after_load" { | 	if conf.StringValue != "after_load" { | ||||||
| 		t.Fatal("AfterLoad option not working") | 		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 | 	_ = conf | ||||||
| 	// t.Logf("%#+v\n", conf) | 	// t.Logf("%#+v\n", conf) | ||||||
| } | } | ||||||
| @@ -82,3 +134,13 @@ func TestValidate(t *testing.T) { | |||||||
| 		t.Fatal(err) | 		t.Fatal(err) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func Test_SizeOf(t *testing.T) { | ||||||
|  | 	st := cfg{} | ||||||
|  |  | ||||||
|  | 	tVal := reflect.TypeOf(st) | ||||||
|  | 	for i := 0; i < tVal.NumField(); i++ { | ||||||
|  | 		field := tVal.Field(i) | ||||||
|  | 		fmt.Printf("Field: %s, Offset: %d, Size: %d\n", field.Name, field.Offset, field.Type.Size()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
| @@ -4,10 +4,11 @@ import ( | |||||||
| 	"context" | 	"context" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/codec" | 	"go.unistack.org/micro/v4/codec" | ||||||
| 	"go.unistack.org/micro/v3/logger" | 	"go.unistack.org/micro/v4/logger" | ||||||
| 	"go.unistack.org/micro/v3/meter" | 	"go.unistack.org/micro/v4/meter" | ||||||
| 	"go.unistack.org/micro/v3/tracer" | 	"go.unistack.org/micro/v4/options" | ||||||
|  | 	"go.unistack.org/micro/v4/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Options hold the config options | // Options hold the config options | ||||||
| @@ -28,14 +29,26 @@ type Options struct { | |||||||
| 	Name string | 	Name string | ||||||
| 	// StructTag name | 	// StructTag name | ||||||
| 	StructTag string | 	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 | 	BeforeSave []func(context.Context, Config) error | ||||||
| 	// AfterLoad contains slice of funcs that runs after load | 	// AfterSave contains slice of funcs that runs after Save | ||||||
| 	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 []func(context.Context, Config) error | 	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 flag to allow fail in config source | ||||||
| 	AllowFail bool | 	AllowFail bool | ||||||
| } | } | ||||||
| @@ -64,9 +77,9 @@ type LoadOption func(o *LoadOptions) | |||||||
| // LoadOptions struct | // LoadOptions struct | ||||||
| type LoadOptions struct { | type LoadOptions struct { | ||||||
| 	Struct   interface{} | 	Struct   interface{} | ||||||
|  | 	Context  context.Context | ||||||
| 	Override bool | 	Override bool | ||||||
| 	Append   bool | 	Append   bool | ||||||
| 	Context  context.Context |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewLoadOptions create LoadOptions struct with provided opts | // NewLoadOptions create LoadOptions struct with provided opts | ||||||
| @@ -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 | // BeforeLoad run funcs before config load | ||||||
| func BeforeLoad(fn ...func(context.Context, Config) error) Option { | func BeforeLoad(fn ...func(context.Context, Config) error) Option { | ||||||
| 	return func(o *Options) { | 	return func(o *Options) { | ||||||
| @@ -253,10 +280,10 @@ func WatchCoalesce(b bool) WatchOption { | |||||||
| } | } | ||||||
|  |  | ||||||
| // WatchInterval specifies min and max time.Duration for pulling changes | // WatchInterval specifies min and max time.Duration for pulling changes | ||||||
| func WatchInterval(min, max time.Duration) WatchOption { | func WatchInterval(minTime, maxTime time.Duration) WatchOption { | ||||||
| 	return func(o *WatchOptions) { | 	return func(o *WatchOptions) { | ||||||
| 		o.MinInterval = min | 		o.MinInterval = minTime | ||||||
| 		o.MaxInterval = max | 		o.MaxInterval = maxTime | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -266,3 +293,10 @@ func WatchStruct(src interface{}) WatchOption { | |||||||
| 		o.Struct = src | 		o.Struct = src | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // Hooks sets hook runs before action | ||||||
|  | func Hooks(h ...options.Hook) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Hooks = append(o.Hooks, h...) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
							
								
								
									
										157
									
								
								database/dsn.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										157
									
								
								database/dsn.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,157 @@ | |||||||
|  | package database | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"crypto/tls" | ||||||
|  | 	"errors" | ||||||
|  | 	"fmt" | ||||||
|  | 	"net/url" | ||||||
|  | 	"strings" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | var ( | ||||||
|  | 	ErrInvalidDSNAddr      = errors.New("invalid dsn addr") | ||||||
|  | 	ErrInvalidDSNUnescaped = errors.New("dsn must be escaped") | ||||||
|  | 	ErrInvalidDSNNoSlash   = errors.New("dsn must contains slash") | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type Config struct { | ||||||
|  | 	TLSConfig *tls.Config | ||||||
|  | 	Username  string | ||||||
|  | 	Password  string | ||||||
|  | 	Scheme    string | ||||||
|  | 	Host      string | ||||||
|  | 	Port      string | ||||||
|  | 	Database  string | ||||||
|  | 	Params    []string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (cfg *Config) FormatDSN() string { | ||||||
|  | 	var s strings.Builder | ||||||
|  |  | ||||||
|  | 	if len(cfg.Scheme) > 0 { | ||||||
|  | 		s.WriteString(cfg.Scheme + "://") | ||||||
|  | 	} | ||||||
|  | 	// [username[:password]@] | ||||||
|  | 	if len(cfg.Username) > 0 { | ||||||
|  | 		s.WriteString(cfg.Username) | ||||||
|  | 		if len(cfg.Password) > 0 { | ||||||
|  | 			s.WriteByte(':') | ||||||
|  | 			s.WriteString(url.PathEscape(cfg.Password)) | ||||||
|  | 		} | ||||||
|  | 		s.WriteByte('@') | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// [host:port] | ||||||
|  | 	if len(cfg.Host) > 0 { | ||||||
|  | 		s.WriteString(cfg.Host) | ||||||
|  | 		if len(cfg.Port) > 0 { | ||||||
|  | 			s.WriteByte(':') | ||||||
|  | 			s.WriteString(cfg.Port) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// /dbname | ||||||
|  | 	s.WriteByte('/') | ||||||
|  | 	s.WriteString(url.PathEscape(cfg.Database)) | ||||||
|  |  | ||||||
|  | 	for i := 0; i < len(cfg.Params); i += 2 { | ||||||
|  | 		if i == 0 { | ||||||
|  | 			s.WriteString("?") | ||||||
|  | 		} else { | ||||||
|  | 			s.WriteString("&") | ||||||
|  | 		} | ||||||
|  | 		s.WriteString(cfg.Params[i]) | ||||||
|  | 		s.WriteString("=") | ||||||
|  | 		s.WriteString(cfg.Params[i+1]) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return s.String() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func ParseDSN(dsn string) (*Config, error) { | ||||||
|  | 	cfg := &Config{} | ||||||
|  |  | ||||||
|  | 	// [user[:password]@][net[(addr)]]/dbname[?param1=value1¶mN=valueN] | ||||||
|  | 	// Find last '/' that goes before dbname | ||||||
|  | 	foundSlash := false | ||||||
|  | 	for i := len(dsn) - 1; i >= 0; i-- { | ||||||
|  | 		if dsn[i] == '/' { | ||||||
|  | 			foundSlash = true | ||||||
|  | 			var j, k int | ||||||
|  |  | ||||||
|  | 			// left part is empty if i <= 0 | ||||||
|  | 			if i > 0 { | ||||||
|  | 				// Find the first ':' in dsn | ||||||
|  | 				for j = i; j >= 0; j-- { | ||||||
|  | 					if dsn[j] == ':' { | ||||||
|  | 						cfg.Scheme = dsn[0:j] | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 				// [username[:password]@][host] | ||||||
|  | 				// Find the last '@' in dsn[:i] | ||||||
|  | 				for j = i; j >= 0; j-- { | ||||||
|  | 					if dsn[j] == '@' { | ||||||
|  | 						// username[:password] | ||||||
|  | 						// Find the second ':' in dsn[:j] | ||||||
|  | 						for k = 0; k < j; k++ { | ||||||
|  | 							if dsn[k] == ':' { | ||||||
|  | 								if cfg.Scheme == dsn[:k] { | ||||||
|  | 									continue | ||||||
|  | 								} | ||||||
|  | 								var err error | ||||||
|  | 								cfg.Password, err = url.PathUnescape(dsn[k+1 : j]) | ||||||
|  | 								if err != nil { | ||||||
|  | 									return nil, err | ||||||
|  | 								} | ||||||
|  | 								break | ||||||
|  | 							} | ||||||
|  | 						} | ||||||
|  | 						cfg.Username = dsn[len(cfg.Scheme)+3 : k] | ||||||
|  | 						break | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 				for k = j + 1; k < i; k++ { | ||||||
|  | 					if dsn[k] == ':' { | ||||||
|  | 						cfg.Host = dsn[j+1 : k] | ||||||
|  | 						cfg.Port = dsn[k+1 : i] | ||||||
|  | 						break | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			// dbname[?param1=value1&...¶mN=valueN] | ||||||
|  | 			// Find the first '?' in dsn[i+1:] | ||||||
|  | 			for j = i + 1; j < len(dsn); j++ { | ||||||
|  | 				if dsn[j] == '?' { | ||||||
|  | 					parts := strings.Split(dsn[j+1:], "&") | ||||||
|  | 					cfg.Params = make([]string, 0, len(parts)*2) | ||||||
|  | 					for _, p := range parts { | ||||||
|  | 						k, v, found := strings.Cut(p, "=") | ||||||
|  | 						if !found { | ||||||
|  | 							continue | ||||||
|  | 						} | ||||||
|  | 						cfg.Params = append(cfg.Params, k, v) | ||||||
|  | 					} | ||||||
|  |  | ||||||
|  | 					break | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			var err error | ||||||
|  | 			dbname := dsn[i+1 : j] | ||||||
|  | 			if cfg.Database, err = url.PathUnescape(dbname); err != nil { | ||||||
|  | 				return nil, fmt.Errorf("invalid dbname %q: %w", dbname, err) | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			break | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !foundSlash && len(dsn) > 0 { | ||||||
|  | 		return nil, ErrInvalidDSNNoSlash | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return cfg, nil | ||||||
|  | } | ||||||
							
								
								
									
										31
									
								
								database/dsn_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										31
									
								
								database/dsn_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,31 @@ | |||||||
|  | package database | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"net/url" | ||||||
|  | 	"testing" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func TestParseDSN(t *testing.T) { | ||||||
|  | 	cfg, err := ParseDSN("postgres://username:p@ssword#@host:12345/dbname?key1=val2&key2=val2") | ||||||
|  | 	if err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  | 	if cfg.Password != "p@ssword#" { | ||||||
|  | 		t.Fatalf("parsing error") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestFormatDSN(t *testing.T) { | ||||||
|  | 	src := "postgres://username:p@ssword#@host:12345/dbname?key1=val2&key2=val2" | ||||||
|  | 	cfg, err := ParseDSN(src) | ||||||
|  | 	if err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  | 	dst, err := url.PathUnescape(cfg.FormatDSN()) | ||||||
|  | 	if err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  | 	if src != dst { | ||||||
|  | 		t.Fatalf("\n%s\n%s", src, dst) | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										158
									
								
								errors/errors.go
									
									
									
									
									
								
							
							
						
						
									
										158
									
								
								errors/errors.go
									
									
									
									
									
								
							| @@ -1,14 +1,20 @@ | |||||||
| // Package errors provides a way to return detailed information | // Package errors provides a way to return detailed information | ||||||
| // for an RPC request error. The error is normally JSON encoded. | // for an RPC request error. The error is normally JSON encoded. | ||||||
| package errors // import "go.unistack.org/micro/v3/errors" | package errors | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"bytes" | 	"bytes" | ||||||
|  | 	"context" | ||||||
| 	"encoding/json" | 	"encoding/json" | ||||||
|  | 	"errors" | ||||||
| 	"fmt" | 	"fmt" | ||||||
|  | 	"io" | ||||||
| 	"net/http" | 	"net/http" | ||||||
| 	"strconv" | 	"strconv" | ||||||
| 	"strings" | 	"strings" | ||||||
|  |  | ||||||
|  | 	"google.golang.org/grpc/codes" | ||||||
|  | 	"google.golang.org/grpc/status" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| @@ -38,6 +44,20 @@ var ( | |||||||
| 	ErrGatewayTimeout = &Error{Code: 504} | 	ErrGatewayTimeout = &Error{Code: 504} | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | const ProblemContentType = "application/problem+json" | ||||||
|  |  | ||||||
|  | type Problem struct { | ||||||
|  | 	Type     string `json:"type,omitempty"` | ||||||
|  | 	Title    string `json:"title,omitempty"` | ||||||
|  | 	Detail   string `json:"detail,omitempty"` | ||||||
|  | 	Instance string `json:"instance,omitempty"` | ||||||
|  | 	Errors   []struct { | ||||||
|  | 		Title  string `json:"title,omitempty"` | ||||||
|  | 		Detail string `json:"detail,omitempty"` | ||||||
|  | 	} `json:"errors,omitempty"` | ||||||
|  | 	Status int `json:"status,omitempty"` | ||||||
|  | } | ||||||
|  |  | ||||||
| // Error type | // Error type | ||||||
| type Error struct { | type Error struct { | ||||||
| 	// ID holds error id or service, usually someting like my_service or id | 	// ID holds error id or service, usually someting like my_service or id | ||||||
| @@ -256,6 +276,10 @@ func CodeIn(err interface{}, codes ...int32) bool { | |||||||
|  |  | ||||||
| // FromError try to convert go error to *Error | // FromError try to convert go error to *Error | ||||||
| func FromError(err error) *Error { | func FromError(err error) *Error { | ||||||
|  | 	if err == nil { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	if verr, ok := err.(*Error); ok && verr != nil { | 	if verr, ok := err.(*Error); ok && verr != nil { | ||||||
| 		return verr | 		return verr | ||||||
| 	} | 	} | ||||||
| @@ -340,3 +364,135 @@ func addslashes(str string) string { | |||||||
| 	} | 	} | ||||||
| 	return buf.String() | 	return buf.String() | ||||||
| } | } | ||||||
|  |  | ||||||
|  | type retryableError struct { | ||||||
|  | 	err error | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Retryable returns error that can be retried later | ||||||
|  | func Retryable(err error) error { | ||||||
|  | 	return &retryableError{err: err} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type IsRetryableFunc func(error) bool | ||||||
|  |  | ||||||
|  | var ( | ||||||
|  | 	RetrayableOracleErrors = []IsRetryableFunc{ | ||||||
|  | 		func(err error) bool { | ||||||
|  | 			errmsg := err.Error() | ||||||
|  | 			switch { | ||||||
|  | 			case strings.Contains(errmsg, `ORA-`): | ||||||
|  | 				return true | ||||||
|  | 			case strings.Contains(errmsg, `can not assign`): | ||||||
|  | 				return true | ||||||
|  | 			case strings.Contains(errmsg, `can't assign`): | ||||||
|  | 				return true | ||||||
|  | 			} | ||||||
|  | 			return false | ||||||
|  | 		}, | ||||||
|  | 	} | ||||||
|  | 	RetrayablePostgresErrors = []IsRetryableFunc{ | ||||||
|  | 		func(err error) bool { | ||||||
|  | 			errmsg := err.Error() | ||||||
|  | 			switch { | ||||||
|  | 			case strings.Contains(errmsg, `number of field descriptions must equal number of`): | ||||||
|  | 				return true | ||||||
|  | 			case strings.Contains(errmsg, `not a pointer`): | ||||||
|  | 				return true | ||||||
|  | 			case strings.Contains(errmsg, `values, but dst struct has only`): | ||||||
|  | 				return true | ||||||
|  | 			case strings.Contains(errmsg, `struct doesn't have corresponding row field`): | ||||||
|  | 				return true | ||||||
|  | 			case strings.Contains(errmsg, `cannot find field`): | ||||||
|  | 				return true | ||||||
|  | 			case strings.Contains(errmsg, `cannot scan`) || strings.Contains(errmsg, `cannot convert`): | ||||||
|  | 				return true | ||||||
|  | 			case strings.Contains(errmsg, `failed to connect to`): | ||||||
|  | 				return true | ||||||
|  | 			} | ||||||
|  | 			return false | ||||||
|  | 		}, | ||||||
|  | 	} | ||||||
|  | 	RetryableMicroErrors = []IsRetryableFunc{ | ||||||
|  | 		func(err error) bool { | ||||||
|  | 			switch verr := err.(type) { | ||||||
|  | 			case *Error: | ||||||
|  | 				switch verr.Code { | ||||||
|  | 				case 401, 403, 408, 500, 501, 502, 503, 504: | ||||||
|  | 					return true | ||||||
|  | 				default: | ||||||
|  | 					return false | ||||||
|  | 				} | ||||||
|  | 			case *retryableError: | ||||||
|  | 				return true | ||||||
|  | 			} | ||||||
|  | 			return false | ||||||
|  | 		}, | ||||||
|  | 	} | ||||||
|  | 	RetryableGoErrors = []IsRetryableFunc{ | ||||||
|  | 		func(err error) bool { | ||||||
|  | 			switch verr := err.(type) { | ||||||
|  | 			case interface{ SafeToRetry() bool }: | ||||||
|  | 				return verr.SafeToRetry() | ||||||
|  | 			case interface{ Timeout() bool }: | ||||||
|  | 				return verr.Timeout() | ||||||
|  | 			} | ||||||
|  | 			switch { | ||||||
|  | 			case errors.Is(err, io.EOF), errors.Is(err, io.ErrUnexpectedEOF): | ||||||
|  | 				return true | ||||||
|  | 			case errors.Is(err, context.DeadlineExceeded): | ||||||
|  | 				return true | ||||||
|  | 			case errors.Is(err, io.ErrClosedPipe), errors.Is(err, io.ErrShortBuffer), errors.Is(err, io.ErrShortWrite): | ||||||
|  | 				return true | ||||||
|  | 			} | ||||||
|  | 			return false | ||||||
|  | 		}, | ||||||
|  | 	} | ||||||
|  | 	RetryableGrpcErrors = []IsRetryableFunc{ | ||||||
|  | 		func(err error) bool { | ||||||
|  | 			st, ok := status.FromError(err) | ||||||
|  | 			if !ok { | ||||||
|  | 				return false | ||||||
|  | 			} | ||||||
|  | 			switch st.Code() { | ||||||
|  | 			case codes.Unavailable, codes.ResourceExhausted: | ||||||
|  | 				return true | ||||||
|  | 			case codes.DeadlineExceeded: | ||||||
|  | 				return true | ||||||
|  | 			case codes.Internal: | ||||||
|  | 				switch { | ||||||
|  | 				case strings.Contains(st.Message(), `transport: received the unexpected content-type "text/html; charset=UTF-8"`): | ||||||
|  | 					return true | ||||||
|  | 				case strings.Contains(st.Message(), io.ErrUnexpectedEOF.Error()): | ||||||
|  | 					return true | ||||||
|  | 				case strings.Contains(st.Message(), `stream terminated by RST_STREAM with error code: INTERNAL_ERROR`): | ||||||
|  | 					return true | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			return false | ||||||
|  | 		}, | ||||||
|  | 	} | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Unwrap provides error wrapping | ||||||
|  | func (e *retryableError) Unwrap() error { | ||||||
|  | 	return e.err | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Error returns the error string | ||||||
|  | func (e *retryableError) Error() string { | ||||||
|  | 	if e.err == nil { | ||||||
|  | 		return "" | ||||||
|  | 	} | ||||||
|  | 	return e.err.Error() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // IsRetryable checks error for ability to retry later | ||||||
|  | func IsRetryable(err error, fns ...IsRetryableFunc) bool { | ||||||
|  | 	for _, fn := range fns { | ||||||
|  | 		if ok := fn(err); ok { | ||||||
|  | 			return true | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|   | |||||||
| @@ -17,7 +17,7 @@ syntax = "proto3"; | |||||||
| package micro.errors; | package micro.errors; | ||||||
|  |  | ||||||
| option cc_enable_arenas = true; | option cc_enable_arenas = true; | ||||||
| option go_package = "go.unistack.org/micro/v3/errors;errors"; | option go_package = "go.unistack.org/micro/v4/errors;errors"; | ||||||
| option java_multiple_files = true; | option java_multiple_files = true; | ||||||
| option java_outer_classname = "MicroErrors"; | option java_outer_classname = "MicroErrors"; | ||||||
| option java_package = "micro.errors"; | option java_package = "micro.errors"; | ||||||
|   | |||||||
| @@ -2,12 +2,19 @@ package errors | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"encoding/json" | 	"encoding/json" | ||||||
| 	er "errors" | 	"errors" | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"net/http" | 	"net/http" | ||||||
| 	"testing" | 	"testing" | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | func TestIsRetrayable(t *testing.T) { | ||||||
|  | 	err := fmt.Errorf("ORA-") | ||||||
|  | 	if !IsRetryable(err, RetrayableOracleErrors...) { | ||||||
|  | 		t.Fatalf("IsRetrayable not works") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| func TestMarshalJSON(t *testing.T) { | func TestMarshalJSON(t *testing.T) { | ||||||
| 	e := InternalServerError("id", "err: %v", fmt.Errorf("err: %v", `xxx: "UNIX_TIMESTAMP": invalid identifier`)) | 	e := InternalServerError("id", "err: %v", fmt.Errorf("err: %v", `xxx: "UNIX_TIMESTAMP": invalid identifier`)) | ||||||
| 	_, err := json.Marshal(e) | 	_, err := json.Marshal(e) | ||||||
| @@ -19,7 +26,7 @@ func TestMarshalJSON(t *testing.T) { | |||||||
| func TestEmpty(t *testing.T) { | func TestEmpty(t *testing.T) { | ||||||
| 	msg := "test" | 	msg := "test" | ||||||
| 	var err *Error | 	var err *Error | ||||||
| 	err = FromError(fmt.Errorf(msg)) | 	err = FromError(errors.New(msg)) | ||||||
| 	if err.Detail != msg { | 	if err.Detail != msg { | ||||||
| 		t.Fatalf("invalid error %v", err) | 		t.Fatalf("invalid error %v", err) | ||||||
| 	} | 	} | ||||||
| @@ -35,7 +42,7 @@ func TestFromError(t *testing.T) { | |||||||
| 	if merr.ID != "go.micro.test" || merr.Code != 404 { | 	if merr.ID != "go.micro.test" || merr.Code != 404 { | ||||||
| 		t.Fatalf("invalid conversation %v != %v", err, merr) | 		t.Fatalf("invalid conversation %v != %v", err, merr) | ||||||
| 	} | 	} | ||||||
| 	err = er.New(err.Error()) | 	err = errors.New(err.Error()) | ||||||
| 	merr = FromError(err) | 	merr = FromError(err) | ||||||
| 	if merr.ID != "go.micro.test" || merr.Code != 404 { | 	if merr.ID != "go.micro.test" || merr.Code != 404 { | ||||||
| 		t.Fatalf("invalid conversation %v != %v", err, merr) | 		t.Fatalf("invalid conversation %v != %v", err, merr) | ||||||
| @@ -50,7 +57,7 @@ func TestEqual(t *testing.T) { | |||||||
| 		t.Fatal("errors must be equal") | 		t.Fatal("errors must be equal") | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	err3 := er.New("my test err") | 	err3 := errors.New("my test err") | ||||||
| 	if Equal(err1, err3) { | 	if Equal(err1, err3) { | ||||||
| 		t.Fatal("errors must be not equal") | 		t.Fatal("errors must be not equal") | ||||||
| 	} | 	} | ||||||
|   | |||||||
							
								
								
									
										27
									
								
								event.go
									
									
									
									
									
								
							
							
						
						
									
										27
									
								
								event.go
									
									
									
									
									
								
							| @@ -1,27 +0,0 @@ | |||||||
| package micro |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"context" |  | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/client" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // Event is used to publish messages to a topic |  | ||||||
| type Event interface { |  | ||||||
| 	// Publish publishes a message to the event topic |  | ||||||
| 	Publish(ctx context.Context, msg interface{}, opts ...client.PublishOption) error |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type event struct { |  | ||||||
| 	c     client.Client |  | ||||||
| 	topic string |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewEvent creates a new event publisher |  | ||||||
| func NewEvent(topic string, c client.Client) Event { |  | ||||||
| 	return &event{c, topic} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (e *event) Publish(ctx context.Context, msg interface{}, opts ...client.PublishOption) error { |  | ||||||
| 	return e.c.Publish(ctx, e.c.NewMessage(e.topic, msg), opts...) |  | ||||||
| } |  | ||||||
| @@ -1,3 +1,5 @@ | |||||||
|  | //go:build ignore | ||||||
|  |  | ||||||
| package flow | package flow | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|   | |||||||
| @@ -8,6 +8,7 @@ import ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
| func TestDeps(t *testing.T) { | func TestDeps(t *testing.T) { | ||||||
|  | 	t.Skip() | ||||||
| 	d := &dag.AcyclicGraph{} | 	d := &dag.AcyclicGraph{} | ||||||
|  |  | ||||||
| 	v0 := d.Add(&node{"v0"}) | 	v0 := d.Add(&node{"v0"}) | ||||||
|   | |||||||
							
								
								
									
										478
									
								
								flow/default.go
									
									
									
									
									
								
							
							
						
						
									
										478
									
								
								flow/default.go
									
									
									
									
									
								
							| @@ -1,17 +1,19 @@ | |||||||
|  | //go:build ignore | ||||||
|  |  | ||||||
| package flow | package flow | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"fmt" | 	"fmt" | ||||||
|  | 	"path/filepath" | ||||||
| 	"sync" | 	"sync" | ||||||
|  |  | ||||||
| 	"github.com/silas/dag" | 	"github.com/heimdalr/dag" | ||||||
| 	"go.unistack.org/micro/v3/client" | 	"go.unistack.org/micro/v4/client" | ||||||
| 	"go.unistack.org/micro/v3/codec" | 	"go.unistack.org/micro/v4/codec" | ||||||
| 	"go.unistack.org/micro/v3/logger" | 	"go.unistack.org/micro/v4/metadata" | ||||||
| 	"go.unistack.org/micro/v3/metadata" | 	"go.unistack.org/micro/v4/store" | ||||||
| 	"go.unistack.org/micro/v3/store" | 	"go.unistack.org/micro/v4/util/id" | ||||||
| 	"go.unistack.org/micro/v3/util/id" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type microFlow struct { | type microFlow struct { | ||||||
| @@ -20,7 +22,7 @@ type microFlow struct { | |||||||
|  |  | ||||||
| type microWorkflow struct { | type microWorkflow struct { | ||||||
| 	opts   Options | 	opts   Options | ||||||
| 	g      *dag.AcyclicGraph | 	g      *dag.DAG | ||||||
| 	steps  map[string]Step | 	steps  map[string]Step | ||||||
| 	id     string | 	id     string | ||||||
| 	status Status | 	status Status | ||||||
| @@ -32,20 +34,20 @@ func (w *microWorkflow) ID() string { | |||||||
| 	return w.id | 	return w.id | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *microWorkflow) Steps() ([][]Step, error) { |  | ||||||
| 	return w.getSteps("", false) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *microWorkflow) Status() Status { | func (w *microWorkflow) Status() Status { | ||||||
| 	return w.status | 	return w.status | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *microWorkflow) AppendSteps(steps ...Step) error { | func (w *microWorkflow) AppendSteps(steps ...Step) error { | ||||||
|  | 	var err error | ||||||
| 	w.Lock() | 	w.Lock() | ||||||
|  | 	defer w.Unlock() | ||||||
|  |  | ||||||
| 	for _, s := range steps { | 	for _, s := range steps { | ||||||
| 		w.steps[s.String()] = s | 		w.steps[s.String()] = s | ||||||
| 		w.g.Add(s) | 		if _, err = w.g.AddVertex(s); err != nil { | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	for _, dst := range steps { | 	for _, dst := range steps { | ||||||
| @@ -54,18 +56,13 @@ func (w *microWorkflow) AppendSteps(steps ...Step) error { | |||||||
| 			if !ok { | 			if !ok { | ||||||
| 				return ErrStepNotExists | 				return ErrStepNotExists | ||||||
| 			} | 			} | ||||||
| 			w.g.Connect(dag.BasicEdge(src, dst)) | 			if err = w.g.AddEdge(src.String(), dst.String()); err != nil { | ||||||
|  | 				return err | ||||||
|  | 			} | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if err := w.g.Validate(); err != nil { | 	w.g.ReduceTransitively() | ||||||
| 		w.Unlock() |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	w.g.TransitiveReduction() |  | ||||||
|  |  | ||||||
| 	w.Unlock() |  | ||||||
|  |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
| @@ -74,10 +71,11 @@ func (w *microWorkflow) RemoveSteps(steps ...Step) error { | |||||||
| 	// TODO: handle case when some step requires or required by removed step | 	// TODO: handle case when some step requires or required by removed step | ||||||
|  |  | ||||||
| 	w.Lock() | 	w.Lock() | ||||||
|  | 	defer w.Unlock() | ||||||
|  |  | ||||||
| 	for _, s := range steps { | 	for _, s := range steps { | ||||||
| 		delete(w.steps, s.String()) | 		delete(w.steps, s.String()) | ||||||
| 		w.g.Remove(s) | 		w.g.DeleteVertex(s.String()) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	for _, dst := range steps { | 	for _, dst := range steps { | ||||||
| @@ -86,91 +84,34 @@ func (w *microWorkflow) RemoveSteps(steps ...Step) error { | |||||||
| 			if !ok { | 			if !ok { | ||||||
| 				return ErrStepNotExists | 				return ErrStepNotExists | ||||||
| 			} | 			} | ||||||
| 			w.g.Connect(dag.BasicEdge(src, dst)) | 			w.g.AddEdge(src.String(), dst.String()) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if err := w.g.Validate(); err != nil { | 	w.g.ReduceTransitively() | ||||||
| 		w.Unlock() |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	w.g.TransitiveReduction() |  | ||||||
|  |  | ||||||
| 	w.Unlock() |  | ||||||
|  |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *microWorkflow) getSteps(start string, reverse bool) ([][]Step, error) { |  | ||||||
| 	var steps [][]Step |  | ||||||
| 	var root dag.Vertex |  | ||||||
| 	var err error |  | ||||||
|  |  | ||||||
| 	fn := func(n dag.Vertex, idx int) error { |  | ||||||
| 		if idx == 0 { |  | ||||||
| 			steps = make([][]Step, 1) |  | ||||||
| 			steps[0] = make([]Step, 0, 1) |  | ||||||
| 		} else if idx >= len(steps) { |  | ||||||
| 			tsteps := make([][]Step, idx+1) |  | ||||||
| 			copy(tsteps, steps) |  | ||||||
| 			steps = tsteps |  | ||||||
| 			steps[idx] = make([]Step, 0, 1) |  | ||||||
| 		} |  | ||||||
| 		steps[idx] = append(steps[idx], n.(Step)) |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if start != "" { |  | ||||||
| 		var ok bool |  | ||||||
| 		w.RLock() |  | ||||||
| 		root, ok = w.steps[start] |  | ||||||
| 		w.RUnlock() |  | ||||||
| 		if !ok { |  | ||||||
| 			return nil, ErrStepNotExists |  | ||||||
| 		} |  | ||||||
| 	} else { |  | ||||||
| 		root, err = w.g.Root() |  | ||||||
| 		if err != nil { |  | ||||||
| 			return nil, err |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if reverse { |  | ||||||
| 		err = w.g.SortedReverseDepthFirstWalk([]dag.Vertex{root}, fn) |  | ||||||
| 	} else { |  | ||||||
| 		err = w.g.SortedDepthFirstWalk([]dag.Vertex{root}, fn) |  | ||||||
| 	} |  | ||||||
| 	if err != nil { |  | ||||||
| 		return nil, err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return steps, nil |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *microWorkflow) Abort(ctx context.Context, id string) error { | func (w *microWorkflow) Abort(ctx context.Context, id string) error { | ||||||
| 	workflowStore := store.NewNamespaceStore(w.opts.Store, "workflows"+w.opts.Store.Options().Separator+id) | 	workflowStore := store.NewNamespaceStore(w.opts.Store, filepath.Join("workflows", id)) | ||||||
| 	return workflowStore.Write(ctx, "status", &codec.Frame{Data: []byte(StatusAborted.String())}) | 	return workflowStore.Write(ctx, "status", &codec.Frame{Data: []byte(StatusAborted.String())}) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *microWorkflow) Suspend(ctx context.Context, id string) error { | func (w *microWorkflow) Suspend(ctx context.Context, id string) error { | ||||||
| 	workflowStore := store.NewNamespaceStore(w.opts.Store, "workflows"+w.opts.Store.Options().Separator+id) | 	workflowStore := store.NewNamespaceStore(w.opts.Store, filepath.Join("workflows", id)) | ||||||
| 	return workflowStore.Write(ctx, "status", &codec.Frame{Data: []byte(StatusSuspend.String())}) | 	return workflowStore.Write(ctx, "status", &codec.Frame{Data: []byte(StatusSuspend.String())}) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *microWorkflow) Resume(ctx context.Context, id string) error { | func (w *microWorkflow) Resume(ctx context.Context, id string) error { | ||||||
| 	workflowStore := store.NewNamespaceStore(w.opts.Store, "workflows"+w.opts.Store.Options().Separator+id) | 	workflowStore := store.NewNamespaceStore(w.opts.Store, filepath.Join("workflows", id)) | ||||||
| 	return workflowStore.Write(ctx, "status", &codec.Frame{Data: []byte(StatusRunning.String())}) | 	return workflowStore.Write(ctx, "status", &codec.Frame{Data: []byte(StatusRunning.String())}) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...ExecuteOption) (string, error) { | func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...ExecuteOption) (string, error) { | ||||||
| 	w.Lock() | 	w.Lock() | ||||||
| 	if !w.init { | 	if !w.init { | ||||||
| 		if err := w.g.Validate(); err != nil { | 		w.g.ReduceTransitively() | ||||||
| 			w.Unlock() |  | ||||||
| 			return "", err |  | ||||||
| 		} |  | ||||||
| 		w.g.TransitiveReduction() |  | ||||||
| 		w.init = true | 		w.init = true | ||||||
| 	} | 	} | ||||||
| 	w.Unlock() | 	w.Unlock() | ||||||
| @@ -180,26 +121,11 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...Execu | |||||||
| 		return "", err | 		return "", err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	stepStore := store.NewNamespaceStore(w.opts.Store, "steps"+w.opts.Store.Options().Separator+eid) | 	//	stepStore := store.NewNamespaceStore(w.opts.Store, filepath.Join("steps", eid)) | ||||||
| 	workflowStore := store.NewNamespaceStore(w.opts.Store, "workflows"+w.opts.Store.Options().Separator+eid) | 	workflowStore := store.NewNamespaceStore(w.opts.Store, filepath.Join("workflows", eid)) | ||||||
|  |  | ||||||
| 	options := NewExecuteOptions(opts...) | 	options := NewExecuteOptions(opts...) | ||||||
|  |  | ||||||
| 	steps, err := w.getSteps(options.Start, options.Reverse) |  | ||||||
| 	if err != nil { |  | ||||||
| 		if werr := workflowStore.Write(w.opts.Context, "status", &codec.Frame{Data: []byte(StatusPending.String())}); werr != nil { |  | ||||||
| 			w.opts.Logger.Errorf(w.opts.Context, "store error: %v", werr) |  | ||||||
| 		} |  | ||||||
| 		return "", err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	var wg sync.WaitGroup |  | ||||||
| 	cherr := make(chan error, 1) |  | ||||||
| 	chstatus := make(chan Status, 1) |  | ||||||
|  |  | ||||||
| 	nctx, cancel := context.WithCancel(ctx) |  | ||||||
| 	defer cancel() |  | ||||||
|  |  | ||||||
| 	nopts := make([]ExecuteOption, 0, len(opts)+5) | 	nopts := make([]ExecuteOption, 0, len(opts)+5) | ||||||
|  |  | ||||||
| 	nopts = append(nopts, | 	nopts = append(nopts, | ||||||
| @@ -209,143 +135,274 @@ func (w *microWorkflow) Execute(ctx context.Context, req *Message, opts ...Execu | |||||||
| 		ExecuteMeter(w.opts.Meter), | 		ExecuteMeter(w.opts.Meter), | ||||||
| 	) | 	) | ||||||
| 	nopts = append(nopts, opts...) | 	nopts = append(nopts, opts...) | ||||||
| 	done := make(chan struct{}) |  | ||||||
|  |  | ||||||
| 	if werr := workflowStore.Write(w.opts.Context, "status", &codec.Frame{Data: []byte(StatusRunning.String())}); werr != nil { | 	if werr := workflowStore.Write(ctx, "status", &codec.Frame{Data: []byte(StatusRunning.String())}); werr != nil { | ||||||
| 		w.opts.Logger.Errorf(w.opts.Context, "store error: %v", werr) | 		w.opts.Logger.Error(ctx, "store error: %v", werr) | ||||||
| 		return eid, werr | 		return eid, werr | ||||||
| 	} | 	} | ||||||
| 	for idx := range steps { |  | ||||||
| 		for nidx := range steps[idx] { | 	var startID string | ||||||
| 			cstep := steps[idx][nidx] | 	if options.Start == "" { | ||||||
| 			if werr := stepStore.Write(ctx, cstep.ID()+w.opts.Store.Options().Separator+"status", &codec.Frame{Data: []byte(StatusPending.String())}); werr != nil { | 		mp := w.g.GetRoots() | ||||||
| 				return eid, werr | 		if len(mp) != 1 { | ||||||
|  | 			return eid, ErrStepNotExists | ||||||
|  | 		} | ||||||
|  | 		for k := range mp { | ||||||
|  | 			startID = k | ||||||
|  | 		} | ||||||
|  | 	} else { | ||||||
|  | 		for k, v := range w.g.GetVertices() { | ||||||
|  | 			if v == options.Start { | ||||||
|  | 				startID = k | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	go func() { | 	if startID == "" { | ||||||
| 		for idx := range steps { | 		return eid, ErrStepNotExists | ||||||
| 			for nidx := range steps[idx] { | 	} | ||||||
| 				wStatus := &codec.Frame{} |  | ||||||
| 				if werr := workflowStore.Read(w.opts.Context, "status", wStatus); werr != nil { | 	if options.Async { | ||||||
| 					cherr <- werr | 		go w.handleWorkflow(startID, nopts...) | ||||||
| 					return | 		return eid, nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return eid, w.handleWorkflow(startID, nopts...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *microWorkflow) handleWorkflow(startID string, opts ...ExecuteOption) error { | ||||||
|  | 	w.RLock() | ||||||
|  | 	defer w.RUnlock() | ||||||
|  |  | ||||||
|  | 	//	stepStore := store.NewNamespaceStore(w.opts.Store, filepath.Join("steps", eid)) | ||||||
|  | 	// workflowStore := store.NewNamespaceStore(w.opts.Store, filepath.Join("workflows", eid)) | ||||||
|  |  | ||||||
|  | 	// Get IDs of all descendant vertices. | ||||||
|  | 	flowIDs, errDes := w.g.GetDescendants(startID) | ||||||
|  | 	if errDes != nil { | ||||||
|  | 		return errDes | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// inputChannels provides for input channels for each of the descendant vertices (+ the start-vertex). | ||||||
|  | 	inputChannels := make(map[string]chan FlowResult, len(flowIDs)+1) | ||||||
|  |  | ||||||
|  | 	// Iterate vertex IDs and create an input channel for each of them and a single | ||||||
|  | 	// output channel for leaves. Note, this "pre-flight" is needed to ensure we | ||||||
|  | 	// really have an input channel regardless of how we traverse the tree and spawn | ||||||
|  | 	// workers. | ||||||
|  | 	leafCount := 0 | ||||||
|  |  | ||||||
|  | 	for id := range flowIDs { | ||||||
|  |  | ||||||
|  | 		// Get all parents of this vertex. | ||||||
|  | 		parents, errPar := w.g.GetParents(id) | ||||||
|  | 		if errPar != nil { | ||||||
|  | 			return errPar | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// Create a buffered input channel that has capacity for all parent results. | ||||||
|  | 		inputChannels[id] = make(chan FlowResult, len(parents)) | ||||||
|  |  | ||||||
|  | 		if ok, err := w.g.IsLeaf(id); ok && err == nil { | ||||||
|  | 			leafCount += 1 | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	// outputChannel caries the results of leaf vertices. | ||||||
|  | 	outputChannel := make(chan FlowResult, leafCount) | ||||||
|  |  | ||||||
|  | 	// To also process the start vertex and to have its results being passed to its | ||||||
|  | 	// children, add it to the vertex IDs. Also add an input channel for the start | ||||||
|  | 	// vertex and feed the inputs to this channel. | ||||||
|  | 	flowIDs[startID] = struct{}{} | ||||||
|  | 	inputChannels[startID] = make(chan FlowResult, len(inputs)) | ||||||
|  | 	for _, i := range inputs { | ||||||
|  | 		inputChannels[startID] <- i | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	wg := sync.WaitGroup{} | ||||||
|  |  | ||||||
|  | 	// Iterate all vertex IDs (now incl. start vertex) and handle each worker (incl. | ||||||
|  | 	// inputs and outputs) in a separate goroutine. | ||||||
|  | 	for id := range flowIDs { | ||||||
|  |  | ||||||
|  | 		// Get all children of this vertex that later need to be notified. Note, we | ||||||
|  | 		// collect all children before the goroutine to be able to release the read | ||||||
|  | 		// lock as early as possible. | ||||||
|  | 		children, errChildren := w.g.GetChildren(id) | ||||||
|  | 		if errChildren != nil { | ||||||
|  | 			return errChildren | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		// Remember to wait for this goroutine. | ||||||
|  | 		wg.Add(1) | ||||||
|  |  | ||||||
|  | 		go func(id string) { | ||||||
|  | 			// Get this vertex's input channel. | ||||||
|  | 			// Note, only concurrent read here, which is fine. | ||||||
|  | 			c := inputChannels[id] | ||||||
|  |  | ||||||
|  | 			// Await all parent inputs and stuff them into a slice. | ||||||
|  | 			parentCount := cap(c) | ||||||
|  | 			parentResults := make([]FlowResult, parentCount) | ||||||
|  | 			for i := 0; i < parentCount; i++ { | ||||||
|  | 				parentResults[i] = <-c | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			// Execute the worker. | ||||||
|  | 			errWorker := callback(w.g, id, parentResults) | ||||||
|  | 			if errWorker != nil { | ||||||
|  | 				return errWorker | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			// Send this worker's FlowResult onto all children's input channels or, if it is | ||||||
|  | 			// a leaf (i.e. no children), send the result onto the output channel. | ||||||
|  | 			if len(children) > 0 { | ||||||
|  | 				for child := range children { | ||||||
|  | 					inputChannels[child] <- flowResult | ||||||
| 				} | 				} | ||||||
| 				if status := StringStatus[string(wStatus.Data)]; status != StatusRunning { | 			} else { | ||||||
| 					chstatus <- status | 				outputChannel <- flowResult | ||||||
| 					return | 			} | ||||||
| 				} |  | ||||||
| 				if w.opts.Logger.V(logger.TraceLevel) { | 			// "Sign off". | ||||||
| 					w.opts.Logger.Tracef(nctx, "will be executed %v", steps[idx][nidx]) | 			wg.Done() | ||||||
| 				} | 		}(id) | ||||||
| 				cstep := steps[idx][nidx] | 	} | ||||||
| 				// nolint: nestif |  | ||||||
| 				if len(cstep.Requires()) == 0 { | 	// Wait for all go routines to finish. | ||||||
| 					wg.Add(1) | 	wg.Wait() | ||||||
| 					go func(step Step) { |  | ||||||
| 						defer wg.Done() | 	// Await all leaf vertex results and stuff them into a slice. | ||||||
| 						if werr := stepStore.Write(ctx, step.ID()+w.opts.Store.Options().Separator+"req", req); werr != nil { | 	resultCount := cap(outputChannel) | ||||||
|  | 	results := make([]FlowResult, resultCount) | ||||||
|  | 	for i := 0; i < resultCount; i++ { | ||||||
|  | 		results[i] = <-outputChannel | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	/* | ||||||
|  | 		go func() { | ||||||
|  | 			for idx := range steps { | ||||||
|  | 				for nidx := range steps[idx] { | ||||||
|  | 					wStatus := &codec.Frame{} | ||||||
|  | 					if werr := workflowStore.Read(w.opts.Context, "status", wStatus); werr != nil { | ||||||
|  | 						cherr <- werr | ||||||
|  | 						return | ||||||
|  | 					} | ||||||
|  | 					if status := StringStatus[string(wStatus.Data)]; status != StatusRunning { | ||||||
|  | 						chstatus <- status | ||||||
|  | 						return | ||||||
|  | 					} | ||||||
|  | 					if w.opts.Logger.V(logger.TraceLevel) { | ||||||
|  | 						w.opts.Logger.Tracef(nctx, "will be executed %v", steps[idx][nidx]) | ||||||
|  | 					} | ||||||
|  | 					cstep := steps[idx][nidx] | ||||||
|  | 					// nolint: nestif | ||||||
|  | 					if len(cstep.Requires()) == 0 { | ||||||
|  | 						wg.Add(1) | ||||||
|  | 						go func(step Step) { | ||||||
|  | 							defer wg.Done() | ||||||
|  | 							if werr := stepStore.Write(ctx, filepath.Join(step.ID(), "req"), req); werr != nil { | ||||||
|  | 								cherr <- werr | ||||||
|  | 								return | ||||||
|  | 							} | ||||||
|  | 							if werr := stepStore.Write(ctx, filepath.Join(step.ID(), "status"), &codec.Frame{Data: []byte(StatusRunning.String())}); werr != nil { | ||||||
|  | 								cherr <- werr | ||||||
|  | 								return | ||||||
|  | 							} | ||||||
|  | 							rsp, serr := step.Execute(nctx, req, nopts...) | ||||||
|  | 							if serr != nil { | ||||||
|  | 								step.SetStatus(StatusFailure) | ||||||
|  | 								if werr := stepStore.Write(ctx, filepath.Join(step.ID(), "rsp"), serr); werr != nil && w.opts.Logger.V(logger.ErrorLevel) { | ||||||
|  | 									w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | ||||||
|  | 								} | ||||||
|  | 								if werr := stepStore.Write(ctx, filepath.Join(step.ID(), "status"), &codec.Frame{Data: []byte(StatusFailure.String())}); werr != nil && w.opts.Logger.V(logger.ErrorLevel) { | ||||||
|  | 									w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | ||||||
|  | 								} | ||||||
|  | 								cherr <- serr | ||||||
|  | 								return | ||||||
|  | 							} | ||||||
|  | 							if werr := stepStore.Write(ctx, filepath.Join(step.ID(), "rsp"), rsp); werr != nil { | ||||||
|  | 								w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | ||||||
|  | 								cherr <- werr | ||||||
|  | 								return | ||||||
|  | 							} | ||||||
|  | 							if werr := stepStore.Write(ctx, filepath.Join(step.ID(), "status"), &codec.Frame{Data: []byte(StatusSuccess.String())}); werr != nil { | ||||||
|  | 								w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | ||||||
|  | 								cherr <- werr | ||||||
|  | 								return | ||||||
|  | 							} | ||||||
|  | 						}(cstep) | ||||||
|  | 						wg.Wait() | ||||||
|  | 					} else { | ||||||
|  | 						if werr := stepStore.Write(ctx, filepath.Join(cstep.ID(), "req"), req); werr != nil { | ||||||
| 							cherr <- werr | 							cherr <- werr | ||||||
| 							return | 							return | ||||||
| 						} | 						} | ||||||
| 						if werr := stepStore.Write(ctx, step.ID()+w.opts.Store.Options().Separator+"status", &codec.Frame{Data: []byte(StatusRunning.String())}); werr != nil { | 						if werr := stepStore.Write(ctx, filepath.Join(cstep.ID(), "status"), &codec.Frame{Data: []byte(StatusRunning.String())}); werr != nil { | ||||||
| 							cherr <- werr | 							cherr <- werr | ||||||
| 							return | 							return | ||||||
| 						} | 						} | ||||||
| 						rsp, serr := step.Execute(nctx, req, nopts...) | 						rsp, serr := cstep.Execute(nctx, req, nopts...) | ||||||
| 						if serr != nil { | 						if serr != nil { | ||||||
| 							step.SetStatus(StatusFailure) | 							cstep.SetStatus(StatusFailure) | ||||||
| 							if werr := stepStore.Write(ctx, step.ID()+w.opts.Store.Options().Separator+"rsp", serr); werr != nil && w.opts.Logger.V(logger.ErrorLevel) { | 							if werr := stepStore.Write(ctx, filepath.Join(cstep.ID(), "rsp"), serr); werr != nil && w.opts.Logger.V(logger.ErrorLevel) { | ||||||
| 								w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | 								w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | ||||||
| 							} | 							} | ||||||
| 							if werr := stepStore.Write(ctx, step.ID()+w.opts.Store.Options().Separator+"status", &codec.Frame{Data: []byte(StatusFailure.String())}); werr != nil && w.opts.Logger.V(logger.ErrorLevel) { | 							if werr := stepStore.Write(ctx, filepath.Join(cstep.ID(), "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.Errorf(ctx, "store write error: %v", werr) | ||||||
| 							} | 							} | ||||||
| 							cherr <- serr | 							cherr <- serr | ||||||
| 							return | 							return | ||||||
| 						} | 						} | ||||||
| 						if werr := stepStore.Write(ctx, step.ID()+w.opts.Store.Options().Separator+"rsp", rsp); werr != nil { | 						if werr := stepStore.Write(ctx, filepath.Join(cstep.ID(), "rsp"), rsp); werr != nil { | ||||||
| 							w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | 							w.opts.Logger.Errorf(ctx, "store write error: %v", werr) | ||||||
| 							cherr <- werr | 							cherr <- werr | ||||||
| 							return | 							return | ||||||
| 						} | 						} | ||||||
| 						if werr := stepStore.Write(ctx, step.ID()+w.opts.Store.Options().Separator+"status", &codec.Frame{Data: []byte(StatusSuccess.String())}); werr != nil { | 						if werr := stepStore.Write(ctx, filepath.Join(cstep.ID(), "status"), &codec.Frame{Data: []byte(StatusSuccess.String())}); werr != nil { | ||||||
| 							w.opts.Logger.Errorf(ctx, "store write error: %v", werr) |  | ||||||
| 							cherr <- werr | 							cherr <- werr | ||||||
| 							return | 							return | ||||||
| 						} | 						} | ||||||
| 					}(cstep) |  | ||||||
| 					wg.Wait() |  | ||||||
| 				} else { |  | ||||||
| 					if werr := stepStore.Write(ctx, cstep.ID()+w.opts.Store.Options().Separator+"req", req); werr != nil { |  | ||||||
| 						cherr <- werr |  | ||||||
| 						return |  | ||||||
| 					} |  | ||||||
| 					if werr := stepStore.Write(ctx, cstep.ID()+w.opts.Store.Options().Separator+"status", &codec.Frame{Data: []byte(StatusRunning.String())}); werr != nil { |  | ||||||
| 						cherr <- werr |  | ||||||
| 						return |  | ||||||
| 					} |  | ||||||
| 					rsp, serr := cstep.Execute(nctx, req, nopts...) |  | ||||||
| 					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) |  | ||||||
| 						} |  | ||||||
| 						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) |  | ||||||
| 						} |  | ||||||
| 						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) |  | ||||||
| 						cherr <- werr |  | ||||||
| 						return |  | ||||||
| 					} |  | ||||||
| 					if werr := stepStore.Write(ctx, cstep.ID()+w.opts.Store.Options().Separator+"status", &codec.Frame{Data: []byte(StatusSuccess.String())}); werr != nil { |  | ||||||
| 						cherr <- werr |  | ||||||
| 						return |  | ||||||
| 					} | 					} | ||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
| 		} | 			close(done) | ||||||
| 		close(done) | 		}() | ||||||
| 	}() |  | ||||||
|  |  | ||||||
| 	if options.Async { | 		if options.Async { | ||||||
| 		return eid, nil | 			return eid, nil | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	logger.Tracef(ctx, "wait for finish or error") |  | ||||||
| 	select { |  | ||||||
| 	case <-nctx.Done(): |  | ||||||
| 		err = nctx.Err() |  | ||||||
| 	case cerr := <-cherr: |  | ||||||
| 		err = cerr |  | ||||||
| 	case <-done: |  | ||||||
| 		close(cherr) |  | ||||||
| 	case <-chstatus: |  | ||||||
| 		close(chstatus) |  | ||||||
| 		return eid, nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	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) |  | ||||||
| 		} | 		} | ||||||
| 	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) |  | ||||||
| 		} |  | ||||||
| 	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) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return eid, err | 		logger.Tracef(ctx, "wait for finish or error") | ||||||
|  | 		select { | ||||||
|  | 		case <-nctx.Done(): | ||||||
|  | 			err = nctx.Err() | ||||||
|  | 		case cerr := <-cherr: | ||||||
|  | 			err = cerr | ||||||
|  | 		case <-done: | ||||||
|  | 			close(cherr) | ||||||
|  | 		case <-chstatus: | ||||||
|  | 			close(chstatus) | ||||||
|  | 			return eid, nil | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		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) | ||||||
|  | 			} | ||||||
|  | 		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) | ||||||
|  | 			} | ||||||
|  | 		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) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	*/ | ||||||
|  | 	return err | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewFlow create new flow | // NewFlow create new flow | ||||||
| @@ -385,11 +442,11 @@ func (f *microFlow) WorkflowList(ctx context.Context) ([]Workflow, error) { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (f *microFlow) WorkflowCreate(ctx context.Context, id string, steps ...Step) (Workflow, error) { | func (f *microFlow) WorkflowCreate(ctx context.Context, id string, steps ...Step) (Workflow, error) { | ||||||
| 	w := µWorkflow{opts: f.opts, id: id, g: &dag.AcyclicGraph{}, steps: make(map[string]Step, len(steps))} | 	w := µWorkflow{opts: f.opts, id: id, g: &dag.DAG{}, steps: make(map[string]Step, len(steps))} | ||||||
|  |  | ||||||
| 	for _, s := range steps { | 	for _, s := range steps { | ||||||
| 		w.steps[s.String()] = s | 		w.steps[s.String()] = s | ||||||
| 		w.g.Add(s) | 		w.g.AddVertex(s) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	for _, dst := range steps { | 	for _, dst := range steps { | ||||||
| @@ -398,14 +455,11 @@ func (f *microFlow) WorkflowCreate(ctx context.Context, id string, steps ...Step | |||||||
| 			if !ok { | 			if !ok { | ||||||
| 				return nil, ErrStepNotExists | 				return nil, ErrStepNotExists | ||||||
| 			} | 			} | ||||||
| 			w.g.Connect(dag.BasicEdge(src, dst)) | 			w.g.AddEdge(src.String(), dst.String()) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if err := w.g.Validate(); err != nil { | 	w.g.ReduceTransitively() | ||||||
| 		return nil, err |  | ||||||
| 	} |  | ||||||
| 	w.g.TransitiveReduction() |  | ||||||
|  |  | ||||||
| 	w.init = true | 	w.init = true | ||||||
|  |  | ||||||
|   | |||||||
| @@ -1,5 +1,5 @@ | |||||||
| // Package flow is an interface used for saga pattern microservice workflow | // Package flow is an interface used for saga pattern microservice workflow | ||||||
| package flow // import "go.unistack.org/micro/v3/flow" | package flow // import "go.unistack.org/micro/v4/flow" | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| @@ -7,7 +7,7 @@ import ( | |||||||
| 	"sync" | 	"sync" | ||||||
| 	"sync/atomic" | 	"sync/atomic" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/metadata" | 	"go.unistack.org/micro/v4/metadata" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| @@ -125,8 +125,6 @@ type Workflow interface { | |||||||
| 	AppendSteps(steps ...Step) error | 	AppendSteps(steps ...Step) error | ||||||
| 	// Status returns workflow status | 	// Status returns workflow status | ||||||
| 	Status() Status | 	Status() Status | ||||||
| 	// Steps returns steps slice where parallel steps returned on the same level |  | ||||||
| 	Steps() ([][]Step, error) |  | ||||||
| 	// Suspend suspends execution | 	// Suspend suspends execution | ||||||
| 	Suspend(ctx context.Context, id string) error | 	Suspend(ctx context.Context, id string) error | ||||||
| 	// Resume resumes execution | 	// Resume resumes execution | ||||||
|   | |||||||
| @@ -4,11 +4,11 @@ import ( | |||||||
| 	"context" | 	"context" | ||||||
| 	"time" | 	"time" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/client" | 	"go.unistack.org/micro/v4/client" | ||||||
| 	"go.unistack.org/micro/v3/logger" | 	"go.unistack.org/micro/v4/logger" | ||||||
| 	"go.unistack.org/micro/v3/meter" | 	"go.unistack.org/micro/v4/meter" | ||||||
| 	"go.unistack.org/micro/v3/store" | 	"go.unistack.org/micro/v4/store" | ||||||
| 	"go.unistack.org/micro/v3/tracer" | 	"go.unistack.org/micro/v4/tracer" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Option func | // Option func | ||||||
| @@ -123,8 +123,6 @@ type ExecuteOptions struct { | |||||||
| 	Start string | 	Start string | ||||||
| 	// Timeout for execution | 	// Timeout for execution | ||||||
| 	Timeout time.Duration | 	Timeout time.Duration | ||||||
| 	// Reverse execution |  | ||||||
| 	Reverse bool |  | ||||||
| 	// Async enables async execution | 	// Async enables async execution | ||||||
| 	Async bool | 	Async bool | ||||||
| } | } | ||||||
| @@ -167,13 +165,6 @@ func ExecuteContext(ctx context.Context) ExecuteOption { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // ExecuteReverse says that dag must be run in reverse order |  | ||||||
| func ExecuteReverse(b bool) ExecuteOption { |  | ||||||
| 	return func(o *ExecuteOptions) { |  | ||||||
| 		o.Reverse = b |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // ExecuteTimeout pass timeout time.Duration for execution | // ExecuteTimeout pass timeout time.Duration for execution | ||||||
| func ExecuteTimeout(td time.Duration) ExecuteOption { | func ExecuteTimeout(td time.Duration) ExecuteOption { | ||||||
| 	return func(o *ExecuteOptions) { | 	return func(o *ExecuteOptions) { | ||||||
|   | |||||||
							
								
								
									
										126
									
								
								fsm/default.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										126
									
								
								fsm/default.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,126 @@ | |||||||
|  | package fsm | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"fmt" | ||||||
|  | 	"sync" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type state struct { | ||||||
|  | 	body interface{} | ||||||
|  | 	name string | ||||||
|  | } | ||||||
|  |  | ||||||
|  | var _ State = &state{} | ||||||
|  |  | ||||||
|  | func (s *state) Name() string { | ||||||
|  | 	return s.name | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (s *state) Body() interface{} { | ||||||
|  | 	return s.body | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // fsm is a finite state machine | ||||||
|  | type fsm struct { | ||||||
|  | 	statesMap   map[string]StateFunc | ||||||
|  | 	current     string | ||||||
|  | 	statesOrder []string | ||||||
|  | 	opts        Options | ||||||
|  | 	mu          sync.Mutex | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewFSM creates a new finite state machine having the specified initial state | ||||||
|  | // with specified options | ||||||
|  | func NewFSM(opts ...Option) FSM { | ||||||
|  | 	return &fsm{ | ||||||
|  | 		statesMap: map[string]StateFunc{}, | ||||||
|  | 		opts:      NewOptions(opts...), | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Current returns the current state | ||||||
|  | func (f *fsm) Current() string { | ||||||
|  | 	f.mu.Lock() | ||||||
|  | 	s := f.current | ||||||
|  | 	f.mu.Unlock() | ||||||
|  | 	return s | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Current returns the current state | ||||||
|  | func (f *fsm) Reset() { | ||||||
|  | 	f.mu.Lock() | ||||||
|  | 	f.current = f.opts.Initial | ||||||
|  | 	f.mu.Unlock() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // State adds state to fsm | ||||||
|  | func (f *fsm) State(state string, fn StateFunc) { | ||||||
|  | 	f.mu.Lock() | ||||||
|  | 	f.statesMap[state] = fn | ||||||
|  | 	f.statesOrder = append(f.statesOrder, state) | ||||||
|  | 	f.mu.Unlock() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Start runs state machine with provided data | ||||||
|  | func (f *fsm) Start(ctx context.Context, args interface{}, opts ...Option) (interface{}, error) { | ||||||
|  | 	var err error | ||||||
|  |  | ||||||
|  | 	f.mu.Lock() | ||||||
|  | 	options := f.opts | ||||||
|  |  | ||||||
|  | 	for _, opt := range opts { | ||||||
|  | 		opt(&options) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	sopts := []StateOption{StateDryRun(options.DryRun)} | ||||||
|  |  | ||||||
|  | 	cstate := options.Initial | ||||||
|  | 	states := make(map[string]StateFunc, len(f.statesMap)) | ||||||
|  | 	for k, v := range f.statesMap { | ||||||
|  | 		states[k] = v | ||||||
|  | 	} | ||||||
|  | 	f.current = cstate | ||||||
|  | 	f.mu.Unlock() | ||||||
|  |  | ||||||
|  | 	var s State | ||||||
|  | 	s = &state{name: cstate, body: args} | ||||||
|  | 	nstate := s.Name() | ||||||
|  |  | ||||||
|  | 	for { | ||||||
|  | 		select { | ||||||
|  | 		case <-ctx.Done(): | ||||||
|  | 			return nil, ctx.Err() | ||||||
|  | 		default: | ||||||
|  | 			fn, ok := states[nstate] | ||||||
|  | 			if !ok { | ||||||
|  | 				return nil, fmt.Errorf(`state "%s" %w`, nstate, ErrInvalidState) | ||||||
|  | 			} | ||||||
|  | 			f.mu.Lock() | ||||||
|  | 			f.current = nstate | ||||||
|  | 			f.mu.Unlock() | ||||||
|  |  | ||||||
|  | 			// wrap the handler func | ||||||
|  | 			for i := len(options.Wrappers); i > 0; i-- { | ||||||
|  | 				fn = options.Wrappers[i-1](fn) | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			s, err = fn(ctx, s, sopts...) | ||||||
|  |  | ||||||
|  | 			switch { | ||||||
|  | 			case err != nil: | ||||||
|  | 				return s.Body(), err | ||||||
|  | 			case s.Name() == StateEnd: | ||||||
|  | 				return s.Body(), nil | ||||||
|  | 			case s.Name() == "": | ||||||
|  | 				for idx := range f.statesOrder { | ||||||
|  | 					if f.statesOrder[idx] == nstate && len(f.statesOrder) > idx+1 { | ||||||
|  | 						nstate = f.statesOrder[idx+1] | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			default: | ||||||
|  | 				nstate = s.Name() | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										176
									
								
								fsm/fsm.go
									
									
									
									
									
								
							
							
						
						
									
										176
									
								
								fsm/fsm.go
									
									
									
									
									
								
							| @@ -1,10 +1,8 @@ | |||||||
| package fsm // import "go.unistack.org/micro/v3/fsm" | package fsm | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"errors" | 	"errors" | ||||||
| 	"fmt" |  | ||||||
| 	"sync" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| @@ -12,170 +10,20 @@ var ( | |||||||
| 	StateEnd        = "end" | 	StateEnd        = "end" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Options struct holding fsm options | type State interface { | ||||||
| type Options struct { | 	Name() string | ||||||
| 	// DryRun mode | 	Body() interface{} | ||||||
| 	DryRun bool |  | ||||||
| 	// Initial state |  | ||||||
| 	Initial string |  | ||||||
| 	// HooksBefore func slice runs in order before state |  | ||||||
| 	HooksBefore []HookBeforeFunc |  | ||||||
| 	// HooksAfter func slice runs in order after state |  | ||||||
| 	HooksAfter []HookAfterFunc |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // HookBeforeFunc func signature | // StateWrapper wraps the StateFunc and returns the equivalent | ||||||
| type HookBeforeFunc func(ctx context.Context, state string, args interface{}) | type StateWrapper func(StateFunc) StateFunc | ||||||
|  |  | ||||||
| // HookAfterFunc func signature |  | ||||||
| type HookAfterFunc func(ctx context.Context, state string, args interface{}) |  | ||||||
|  |  | ||||||
| // Option func signature |  | ||||||
| type Option func(*Options) |  | ||||||
|  |  | ||||||
| // StateOptions holds state options |  | ||||||
| type StateOptions struct { |  | ||||||
| 	DryRun bool |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // StateDryRun says that state executes in dry run mode |  | ||||||
| func StateDryRun(b bool) StateOption { |  | ||||||
| 	return func(o *StateOptions) { |  | ||||||
| 		o.DryRun = b |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // StateOption func signature |  | ||||||
| type StateOption func(*StateOptions) |  | ||||||
|  |  | ||||||
| // InitialState sets init state for state machine |  | ||||||
| func InitialState(initial string) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.Initial = initial |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // HookBefore provides hook func slice |  | ||||||
| func HookBefore(fns ...HookBeforeFunc) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.HooksBefore = fns |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // HookAfter provides hook func slice |  | ||||||
| func HookAfter(fns ...HookAfterFunc) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.HooksAfter = fns |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // StateFunc called on state transition and return next step and error | // StateFunc called on state transition and return next step and error | ||||||
| type StateFunc func(ctx context.Context, args interface{}, opts ...StateOption) (string, interface{}, error) | type StateFunc func(ctx context.Context, state State, opts ...StateOption) (State, error) | ||||||
|  |  | ||||||
| // FSM is a finite state machine | type FSM interface { | ||||||
| type FSM struct { | 	Start(context.Context, interface{}, ...Option) (interface{}, error) | ||||||
| 	mu          sync.Mutex | 	Current() string | ||||||
| 	statesMap   map[string]StateFunc | 	Reset() | ||||||
| 	statesOrder []string | 	State(string, StateFunc) | ||||||
| 	opts        *Options |  | ||||||
| 	current     string |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // New creates a new finite state machine having the specified initial state |  | ||||||
| // with specified options |  | ||||||
| func New(opts ...Option) *FSM { |  | ||||||
| 	options := &Options{} |  | ||||||
|  |  | ||||||
| 	for _, opt := range opts { |  | ||||||
| 		opt(options) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return &FSM{ |  | ||||||
| 		statesMap: map[string]StateFunc{}, |  | ||||||
| 		opts:      options, |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Current returns the current state |  | ||||||
| func (f *FSM) Current() string { |  | ||||||
| 	f.mu.Lock() |  | ||||||
| 	defer f.mu.Unlock() |  | ||||||
| 	return f.current |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Current returns the current state |  | ||||||
| func (f *FSM) Reset() { |  | ||||||
| 	f.mu.Lock() |  | ||||||
| 	f.current = f.opts.Initial |  | ||||||
| 	f.mu.Unlock() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // State adds state to fsm |  | ||||||
| func (f *FSM) State(state string, fn StateFunc) { |  | ||||||
| 	f.mu.Lock() |  | ||||||
| 	f.statesMap[state] = fn |  | ||||||
| 	f.statesOrder = append(f.statesOrder, state) |  | ||||||
| 	f.mu.Unlock() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Init initialize fsm and check states |  | ||||||
|  |  | ||||||
| // Start runs state machine with provided data |  | ||||||
| func (f *FSM) Start(ctx context.Context, args interface{}, opts ...Option) (interface{}, error) { |  | ||||||
| 	var err error |  | ||||||
| 	var ok bool |  | ||||||
| 	var fn StateFunc |  | ||||||
| 	var nstate string |  | ||||||
|  |  | ||||||
| 	f.mu.Lock() |  | ||||||
| 	options := f.opts |  | ||||||
|  |  | ||||||
| 	for _, opt := range opts { |  | ||||||
| 		opt(options) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	sopts := []StateOption{StateDryRun(options.DryRun)} |  | ||||||
|  |  | ||||||
| 	cstate := options.Initial |  | ||||||
| 	states := make(map[string]StateFunc, len(f.statesMap)) |  | ||||||
| 	for k, v := range f.statesMap { |  | ||||||
| 		states[k] = v |  | ||||||
| 	} |  | ||||||
| 	f.current = cstate |  | ||||||
| 	f.mu.Unlock() |  | ||||||
|  |  | ||||||
| 	for { |  | ||||||
| 		select { |  | ||||||
| 		case <-ctx.Done(): |  | ||||||
| 			return nil, ctx.Err() |  | ||||||
| 		default: |  | ||||||
| 			fn, ok = states[cstate] |  | ||||||
| 			if !ok { |  | ||||||
| 				return nil, fmt.Errorf(`state "%s" %w`, cstate, ErrInvalidState) |  | ||||||
| 			} |  | ||||||
| 			f.mu.Lock() |  | ||||||
| 			f.current = cstate |  | ||||||
| 			f.mu.Unlock() |  | ||||||
| 			for _, fn := range options.HooksBefore { |  | ||||||
| 				fn(ctx, cstate, args) |  | ||||||
| 			} |  | ||||||
| 			nstate, args, err = fn(ctx, args, sopts...) |  | ||||||
| 			for _, fn := range options.HooksAfter { |  | ||||||
| 				fn(ctx, cstate, args) |  | ||||||
| 			} |  | ||||||
| 			switch { |  | ||||||
| 			case err != nil: |  | ||||||
| 				return args, err |  | ||||||
| 			case nstate == StateEnd: |  | ||||||
| 				return args, nil |  | ||||||
| 			case nstate == "": |  | ||||||
| 				for idx := range f.statesOrder { |  | ||||||
| 					if f.statesOrder[idx] == cstate && len(f.statesOrder) > idx+1 { |  | ||||||
| 						nstate = f.statesOrder[idx+1] |  | ||||||
| 					} |  | ||||||
| 				} |  | ||||||
| 			} |  | ||||||
| 			cstate = nstate |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,63 +1,72 @@ | |||||||
| package fsm | package fsm | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"bytes" |  | ||||||
| 	"context" | 	"context" | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"testing" | 	"testing" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v4/logger" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| func TestFSMStart(t *testing.T) { | func TestFSMStart(t *testing.T) { | ||||||
| 	ctx := context.TODO() | 	ctx := context.TODO() | ||||||
| 	buf := bytes.NewBuffer(nil) |  | ||||||
| 	pfb := func(_ context.Context, state string, _ interface{}) { | 	if err := logger.DefaultLogger.Init(); err != nil { | ||||||
| 		fmt.Fprintf(buf, "before state %s\n", state) | 		t.Fatal(err) | ||||||
| 	} | 	} | ||||||
| 	pfa := func(_ context.Context, state string, _ interface{}) { |  | ||||||
| 		fmt.Fprintf(buf, "after state %s\n", state) | 	wrapper := func(next StateFunc) StateFunc { | ||||||
|  | 		return func(sctx context.Context, s State, opts ...StateOption) (State, error) { | ||||||
|  | 			sctx = logger.NewContext(sctx, logger.DefaultLogger.Fields("state", s.Name())) | ||||||
|  | 			return next(sctx, s, opts...) | ||||||
|  | 		} | ||||||
| 	} | 	} | ||||||
| 	f := New(InitialState("1"), HookBefore(pfb), HookAfter(pfa)) |  | ||||||
| 	f1 := func(_ context.Context, req interface{}, _ ...StateOption) (string, interface{}, error) { | 	f := NewFSM(InitialState("1"), WrapState(wrapper)) | ||||||
| 		args := req.(map[string]interface{}) | 	f1 := func(sctx context.Context, s State, opts ...StateOption) (State, error) { | ||||||
|  | 		_, ok := logger.FromContext(sctx) | ||||||
|  | 		if !ok { | ||||||
|  | 			t.Fatal("f1 context does not have logger") | ||||||
|  | 		} | ||||||
|  | 		args := s.Body().(map[string]interface{}) | ||||||
| 		if v, ok := args["request"].(string); !ok || v == "" { | 		if v, ok := args["request"].(string); !ok || v == "" { | ||||||
| 			return "", nil, fmt.Errorf("empty request") | 			return nil, fmt.Errorf("empty request") | ||||||
| 		} | 		} | ||||||
| 		return "2", map[string]interface{}{"response": "test2"}, nil | 		return &state{name: "", body: map[string]interface{}{"response": "state1"}}, nil | ||||||
| 	} | 	} | ||||||
| 	f2 := func(_ context.Context, req interface{}, _ ...StateOption) (string, interface{}, error) { | 	f2 := func(sctx context.Context, s State, opts ...StateOption) (State, error) { | ||||||
| 		args := req.(map[string]interface{}) | 		_, ok := logger.FromContext(sctx) | ||||||
| 		if v, ok := args["response"].(string); !ok || v == "" { | 		if !ok { | ||||||
| 			return "", nil, fmt.Errorf("empty response") | 			t.Fatal("f2 context does not have logger") | ||||||
| 		} | 		} | ||||||
| 		return "", map[string]interface{}{"response": "test"}, nil | 		args := s.Body().(map[string]interface{}) | ||||||
|  | 		if v, ok := args["response"].(string); !ok || v == "" { | ||||||
|  | 			return nil, fmt.Errorf("empty response") | ||||||
|  | 		} | ||||||
|  | 		return &state{name: "", body: map[string]interface{}{"response": "state2"}}, nil | ||||||
| 	} | 	} | ||||||
| 	f3 := func(_ context.Context, req interface{}, _ ...StateOption) (string, interface{}, error) { | 	f3 := func(sctx context.Context, s State, opts ...StateOption) (State, error) { | ||||||
| 		args := req.(map[string]interface{}) | 		_, ok := logger.FromContext(sctx) | ||||||
| 		if v, ok := args["response"].(string); !ok || v == "" { | 		if !ok { | ||||||
| 			return "", nil, fmt.Errorf("empty response") | 			t.Fatal("f3 context does not have logger") | ||||||
| 		} | 		} | ||||||
| 		return StateEnd, map[string]interface{}{"response": "test_last"}, nil | 		args := s.Body().(map[string]interface{}) | ||||||
|  | 		if v, ok := args["response"].(string); !ok || v == "" { | ||||||
|  | 			return nil, fmt.Errorf("empty response") | ||||||
|  | 		} | ||||||
|  | 		return &state{name: StateEnd, body: map[string]interface{}{"response": "state3"}}, nil | ||||||
| 	} | 	} | ||||||
| 	f.State("1", f1) | 	f.State("1", f1) | ||||||
| 	f.State("2", f2) | 	f.State("2", f2) | ||||||
| 	f.State("3", f3) | 	f.State("3", f3) | ||||||
| 	rsp, err := f.Start(ctx, map[string]interface{}{"request": "test1"}) | 	rsp, err := f.Start(ctx, map[string]interface{}{"request": "state"}) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		t.Fatal(err) | 		t.Fatal(err) | ||||||
| 	} | 	} | ||||||
| 	args := rsp.(map[string]interface{}) | 	args := rsp.(map[string]interface{}) | ||||||
| 	if v, ok := args["response"].(string); !ok || v == "" { | 	if v, ok := args["response"].(string); !ok || v == "" { | ||||||
| 		t.Fatalf("nil rsp: %#+v", args) | 		t.Fatalf("nil rsp: %#+v", args) | ||||||
| 	} else if v != "test_last" { | 	} else if v != "state3" { | ||||||
| 		t.Fatalf("invalid rsp %#+v", args) | 		t.Fatalf("invalid rsp %#+v", args) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`before state 1`)) || |  | ||||||
| 		!bytes.Contains(buf.Bytes(), []byte(`before state 2`)) || |  | ||||||
| 		!bytes.Contains(buf.Bytes(), []byte(`after state 1`)) || |  | ||||||
| 		!bytes.Contains(buf.Bytes(), []byte(`after state 2`)) || |  | ||||||
| 		!bytes.Contains(buf.Bytes(), []byte(`after state 3`)) || |  | ||||||
| 		!bytes.Contains(buf.Bytes(), []byte(`after state 3`)) { |  | ||||||
| 		t.Fatalf("fsm not works properly or hooks error, buf: %s", buf.Bytes()) |  | ||||||
| 	} |  | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										52
									
								
								fsm/options.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										52
									
								
								fsm/options.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,52 @@ | |||||||
|  | package fsm | ||||||
|  |  | ||||||
|  | // Options struct holding fsm options | ||||||
|  | type Options struct { | ||||||
|  | 	// Initial state | ||||||
|  | 	Initial string | ||||||
|  | 	// Wrappers runs before state | ||||||
|  | 	Wrappers []StateWrapper | ||||||
|  | 	// DryRun mode | ||||||
|  | 	DryRun bool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Option func signature | ||||||
|  | type Option func(*Options) | ||||||
|  |  | ||||||
|  | // StateOptions holds state options | ||||||
|  | type StateOptions struct { | ||||||
|  | 	DryRun bool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // StateDryRun says that state executes in dry run mode | ||||||
|  | func StateDryRun(b bool) StateOption { | ||||||
|  | 	return func(o *StateOptions) { | ||||||
|  | 		o.DryRun = b | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // StateOption func signature | ||||||
|  | type StateOption func(*StateOptions) | ||||||
|  |  | ||||||
|  | // InitialState sets init state for state machine | ||||||
|  | func InitialState(initial string) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Initial = initial | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WrapState adds a state Wrapper to a list of options passed into the fsm | ||||||
|  | func WrapState(w StateWrapper) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.Wrappers = append(o.Wrappers, w) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // NewOptions returns new Options struct filled by passed Option | ||||||
|  | func NewOptions(opts ...Option) Options { | ||||||
|  | 	options := Options{} | ||||||
|  | 	for _, o := range opts { | ||||||
|  | 		o(&options) | ||||||
|  | 	} | ||||||
|  | 	return options | ||||||
|  | } | ||||||
							
								
								
									
										37
									
								
								go.mod
									
									
									
									
									
								
							
							
						
						
									
										37
									
								
								go.mod
									
									
									
									
									
								
							| @@ -1,13 +1,34 @@ | |||||||
| module go.unistack.org/micro/v3 | module go.unistack.org/micro/v4 | ||||||
|  |  | ||||||
| go 1.16 | go 1.22.0 | ||||||
|  |  | ||||||
| require ( | require ( | ||||||
| 	github.com/google/go-cmp v0.5.7 // indirect | 	dario.cat/mergo v1.0.1 | ||||||
| 	github.com/imdario/mergo v0.3.13 | 	github.com/DATA-DOG/go-sqlmock v1.5.2 | ||||||
| 	github.com/kr/pretty v0.2.1 // indirect | 	github.com/KimMachineGun/automemlimit v0.7.0 | ||||||
| 	github.com/kr/text v0.2.0 // indirect | 	github.com/ash3in/uuidv8 v1.2.0 | ||||||
|  | 	github.com/google/uuid v1.6.0 | ||||||
|  | 	github.com/matoous/go-nanoid v1.5.1 | ||||||
| 	github.com/patrickmn/go-cache v2.1.0+incompatible | 	github.com/patrickmn/go-cache v2.1.0+incompatible | ||||||
| 	github.com/silas/dag v0.0.0-20211117232152-9d50aa809f35 | 	github.com/silas/dag v0.0.0-20220518035006-a7e85ada93c5 | ||||||
| 	go.unistack.org/micro-proto/v3 v3.3.1 | 	github.com/spf13/cast v1.7.1 | ||||||
|  | 	go.uber.org/atomic v1.11.0 | ||||||
|  | 	go.uber.org/automaxprocs v1.6.0 | ||||||
|  | 	go.unistack.org/micro-proto/v4 v4.1.0 | ||||||
|  | 	golang.org/x/sync v0.10.0 | ||||||
|  | 	google.golang.org/grpc v1.69.4 | ||||||
|  | 	google.golang.org/protobuf v1.36.3 | ||||||
|  | 	gopkg.in/yaml.v3 v3.0.1 | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | require ( | ||||||
|  | 	github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // 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/stretchr/testify v1.10.0 // indirect | ||||||
|  | 	golang.org/x/net v0.34.0 // indirect | ||||||
|  | 	golang.org/x/sys v0.29.0 // indirect | ||||||
|  | 	google.golang.org/genproto/googleapis/rpc v0.0.0-20241216192217-9240e9c98484 // indirect | ||||||
|  | 	gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c // indirect | ||||||
| ) | ) | ||||||
|   | |||||||
							
								
								
									
										213
									
								
								go.sum
									
									
									
									
									
								
							
							
						
						
									
										213
									
								
								go.sum
									
									
									
									
									
								
							| @@ -1,160 +1,69 @@ | |||||||
| cloud.google.com/go v0.26.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw= | dario.cat/mergo v1.0.1 h1:Ra4+bf83h2ztPIQYNP99R6m+Y7KfnARDfID+a+vLl4s= | ||||||
| cloud.google.com/go v0.34.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw= | dario.cat/mergo v1.0.1/go.mod h1:uNxQE+84aUszobStD9th8a29P2fMDhsBdgRYvZOxGmk= | ||||||
| github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU= | github.com/DATA-DOG/go-sqlmock v1.5.2 h1:OcvFkGmslmlZibjAjaHm3L//6LiuBgolP7OputlJIzU= | ||||||
| github.com/OneOfOne/xxhash v1.2.2/go.mod h1:HSdplMjZKSmBqAxg5vPj2TmRDmfkzw+cTzAElWljhcU= | github.com/DATA-DOG/go-sqlmock v1.5.2/go.mod h1:88MAG/4G7SMwSE3CeA0ZKzrT5CiOU3OJ+JlNzwDqpNU= | ||||||
| github.com/antihax/optional v1.0.0/go.mod h1:uupD/76wgC+ih3iEmQUL+0Ugr19nfwCT1kdvxnR2qWY= | github.com/KimMachineGun/automemlimit v0.7.0 h1:7G06p/dMSf7G8E6oq+f2uOPuVncFyIlDI/pBWK49u88= | ||||||
| github.com/buger/jsonparser v1.1.1/go.mod h1:6RYKKt7H4d4+iWqouImQ9R2FZql3VbhNgx27UK13J/0= | github.com/KimMachineGun/automemlimit v0.7.0/go.mod h1:QZxpHaGOQoYvFhv/r4u3U0JTC2ZcOwbSr11UZF46UBM= | ||||||
| github.com/census-instrumentation/opencensus-proto v0.2.1/go.mod h1:f6KPmirojxKA12rnyqOA5BBL4O983OfeGPqjHWSTneU= | github.com/ash3in/uuidv8 v1.2.0 h1:2oogGdtCPwaVtyvPPGin4TfZLtOGE5F+W++E880G6SI= | ||||||
| github.com/cespare/xxhash v1.1.0/go.mod h1:XrSqR1VqqWfGrhpAt58auRo0WTKS1nRRg3ghfAqPWnc= | github.com/ash3in/uuidv8 v1.2.0/go.mod h1:BnU0wJBxnzdEKmVg4xckBkD+VZuecTFTUP3M0dWgyY4= | ||||||
| github.com/client9/misspell v0.3.4/go.mod h1:qj6jICC3Q7zFZvVWo7KLAzC3yx5G7kyvSDkc90ppPyw= | github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1VwoXQT9A3Wy9MM3WgvqSxFWenqJduM= | ||||||
| github.com/cncf/udpa/go v0.0.0-20191209042840-269d4d468f6f/go.mod h1:M8M6+tZqaGXZJjfX53e64911xZQV5JYwmTeXPW+k8Sc= | github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | ||||||
| github.com/cncf/udpa/go v0.0.0-20201120205902-5459f2c99403/go.mod h1:WmhPx2Nbnhtbo57+VJT5O0JRkEi1Wbu0z5j0R8u5Hbk= | github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8= | ||||||
| github.com/cncf/xds/go v0.0.0-20210312221358-fbca930ec8ed/go.mod h1:eXthEFrGJvWHgFFCl3hGmgk+/aYT6PnTQLykKQRLhEs= | github.com/frankban/quicktest v1.14.6/go.mod h1:4ptaffx2x8+WTWXmUCuVU6aPUX1/Mz7zb5vbUoiM6w0= | ||||||
| github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E= | github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek= | ||||||
| github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | github.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps= | ||||||
| github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | github.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI= | ||||||
| github.com/docopt/docopt-go v0.0.0-20180111231733-ee0de3bc6815/go.mod h1:WwZ+bS3ebgob9U8Nd0kOddGdZWjyMGR8Wziv+TBNwSE= | github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY= | ||||||
| github.com/envoyproxy/go-control-plane v0.9.0/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4= | github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0= | ||||||
| github.com/envoyproxy/go-control-plane v0.9.1-0.20191026205805-5f8ba28d4473/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4= | github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= | ||||||
| github.com/envoyproxy/go-control-plane v0.9.4/go.mod h1:6rpuAdCZL397s3pYoYcLgu1mIlRU8Am5FuJP05cCM98= | github.com/kisielk/sqlstruct v0.0.0-20201105191214-5f3e10d3ab46/go.mod h1:yyMNCyc/Ib3bDTKd379tNMpB/7/H5TjM2Y9QJ5THLbE= | ||||||
| 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/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI= | github.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI= | ||||||
|  | github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE= | ||||||
|  | github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk= | ||||||
| github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= | 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.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 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY= | ||||||
| github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE= | github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE= | ||||||
|  | github.com/matoous/go-nanoid v1.5.1 h1:aCjdvTyO9LLnTIi0fgdXhOPPvOHjpXN6Ik9DaNjIct4= | ||||||
|  | github.com/matoous/go-nanoid v1.5.1/go.mod h1:zyD2a71IubI24efhpvkJz+ZwfwagzgSO6UNiFsZKN7U= | ||||||
| github.com/patrickmn/go-cache v2.1.0+incompatible h1:HRMgzkcYKYpi3C8ajMPV8OFXaaRUnok+kx1WdO15EQc= | github.com/patrickmn/go-cache v2.1.0+incompatible h1:HRMgzkcYKYpi3C8ajMPV8OFXaaRUnok+kx1WdO15EQc= | ||||||
| github.com/patrickmn/go-cache v2.1.0+incompatible/go.mod h1:3Qf8kWWT7OJRJbdiICTKqZju1ZixQ/KpMGzzAfe6+WQ= | github.com/patrickmn/go-cache v2.1.0+incompatible/go.mod h1:3Qf8kWWT7OJRJbdiICTKqZju1ZixQ/KpMGzzAfe6+WQ= | ||||||
| github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= | github.com/pbnjay/memory v0.0.0-20210728143218-7b4eea64cf58 h1:onHthvaw9LFnH4t2DcNVpwGmV9E1BkGknEliJkfwQj0= | ||||||
| github.com/prometheus/client_model v0.0.0-20190812154241-14fe0d1b01d4/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA= | github.com/pbnjay/memory v0.0.0-20210728143218-7b4eea64cf58/go.mod h1:DXv8WO4yhMYhSNPKjeNKa5WY9YCIEBRbNzFFPJbWO6Y= | ||||||
| github.com/rogpeppe/fastuuid v1.2.0/go.mod h1:jVj6XXZzXRy/MSR5jhDC/2q6DgLz+nrA6LYCDYWNEvQ= | github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 h1:Jamvg5psRIccs7FGNTlIRMkT8wgtp5eCXdBlqhYGL6U= | ||||||
| github.com/silas/dag v0.0.0-20211117232152-9d50aa809f35 h1:4mohWoM/UGg1BvFFiqSPRl5uwJY3rVV0HQX0ETqauqQ= | github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= | ||||||
| github.com/silas/dag v0.0.0-20211117232152-9d50aa809f35/go.mod h1:7RTUFBdIRC9nZ7/3RyRNH1bdqIShrDejd1YbLwgPS+I= | github.com/prashantv/gostub v1.1.0 h1:BTyx3RfQjRHnUWaGF9oQos79AlQ5k8WNktv7VGvVH4g= | ||||||
| github.com/spaolacci/murmur3 v0.0.0-20180118202830-f09979ecbc72/go.mod h1:JwIasOWyU6f++ZhiEuf87xNszmSA2myDM2Kzu9HwQUA= | github.com/prashantv/gostub v1.1.0/go.mod h1:A5zLQHz7ieHGG7is6LLXLz7I8+3LZzsrV0P1IAHhP5U= | ||||||
| github.com/stoewer/go-strcase v1.2.0/go.mod h1:IBiWB2sKIp3wVVQ3Y035++gc+knqhUQag1KpM8ahLw8= | github.com/rogpeppe/go-internal v1.13.1 h1:KvO1DLK/DRN07sQ1LQKScxyZJuNnedQ5/wKSR38lUII= | ||||||
| github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= | github.com/rogpeppe/go-internal v1.13.1/go.mod h1:uMEvuHeurkdAXX61udpOXGD/AzZDWNMNyH2VO9fmH0o= | ||||||
| github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= | github.com/silas/dag v0.0.0-20220518035006-a7e85ada93c5 h1:G/FZtUu7a6NTWl3KUHMV9jkLAh/Rvtf03NWMHaEDl+E= | ||||||
| github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA= | github.com/silas/dag v0.0.0-20220518035006-a7e85ada93c5/go.mod h1:7RTUFBdIRC9nZ7/3RyRNH1bdqIShrDejd1YbLwgPS+I= | ||||||
| github.com/xeipuuv/gojsonpointer v0.0.0-20180127040702-4e3ac2762d5f/go.mod h1:N2zxlSyiKSe5eX1tZViRH5QA0qijqEDrYZiPEAiq3wU= | github.com/spf13/cast v1.7.1 h1:cuNEagBQEHWN1FnbGEjCXL2szYEXqfJPbP2HNUaca9Y= | ||||||
| github.com/xeipuuv/gojsonreference v0.0.0-20180127040603-bd5ef7bd5415/go.mod h1:GwrjFmJcFw6At/Gs6z4yjiIwzuJ1/+UwLxMQDVQXShQ= | github.com/spf13/cast v1.7.1/go.mod h1:ancEpBxwJDODSW/UG4rDrAqiKolqNNh2DX3mk86cAdo= | ||||||
| github.com/xeipuuv/gojsonschema v1.2.0/go.mod h1:anYRn/JVcOK2ZgGU+IjEV4nwlhoK5sQluxsYJ78Id3Y= | github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA= | ||||||
| go.opentelemetry.io/proto/otlp v0.7.0/go.mod h1:PqfVotwruBrMGOCsRd/89rSnXhoiJIqeYNgFYFoEGnI= | github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY= | ||||||
| go.unistack.org/micro-proto/v3 v3.3.1 h1:nQ0MtWvP2G3QrpOgawVOPhpZZYkq6umTGDqs8FxJYIo= | go.uber.org/atomic v1.11.0 h1:ZvwS0R+56ePWxUNi+Atn9dWONBPp/AUETXlHW0DxSjE= | ||||||
| go.unistack.org/micro-proto/v3 v3.3.1/go.mod h1:cwRyv8uInM2I7EbU7O8Fx2Ls3N90Uw9UCCcq4olOdfE= | go.uber.org/atomic v1.11.0/go.mod h1:LUxbIzbOniOlMKjJjyPfpl4v+PKK2cNJn91OQbhoJI0= | ||||||
| golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= | go.uber.org/automaxprocs v1.6.0 h1:O3y2/QNTOdbF+e/dpXNNW7Rx2hZ4sTIPyybbxyNqTUs= | ||||||
| golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto= | go.uber.org/automaxprocs v1.6.0/go.mod h1:ifeIMSnPZuznNm6jmdzmU3/bfk01Fe2fotchwEFJ8r8= | ||||||
| golang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA= | go.unistack.org/micro-proto/v4 v4.1.0 h1:qPwL2n/oqh9RE3RTTDgt28XK3QzV597VugQPaw9lKUk= | ||||||
| golang.org/x/lint v0.0.0-20181026193005-c67002cb31c3/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE= | go.unistack.org/micro-proto/v4 v4.1.0/go.mod h1:ArmK7o+uFvxSY3dbJhKBBX4Pm1rhWdLEFf3LxBrMtec= | ||||||
| golang.org/x/lint v0.0.0-20190227174305-5b3e6a55c961/go.mod h1:wehouNa3lNwaWXcvxsM5YxQ5yQlVC4a0KAMCusXpPoU= | golang.org/x/net v0.34.0 h1:Mb7Mrk043xzHgnRM88suvJFwzVrRfHEHJEl5/71CKw0= | ||||||
| golang.org/x/lint v0.0.0-20190313153728-d0100b6bd8b3/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc= | golang.org/x/net v0.34.0/go.mod h1:di0qlW3YNM5oh6GqDGQr92MyTozJPmybPK4Ev/Gm31k= | ||||||
| golang.org/x/net v0.0.0-20180724234803-3673e40ba225/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= | golang.org/x/sync v0.10.0 h1:3NQrjDixjgGwUOCaF8w2+VYHv0Ve/vGYSbdkTa98gmQ= | ||||||
| golang.org/x/net v0.0.0-20180826012351-8a410e7b638d/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= | golang.org/x/sync v0.10.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk= | ||||||
| golang.org/x/net v0.0.0-20190108225652-1e06a53dbb7e/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= | golang.org/x/sys v0.29.0 h1:TPYlXGxvx1MGTn2GiZDhnjPA9wZzZeGKHHmKhHYvgaU= | ||||||
| golang.org/x/net v0.0.0-20190213061140-3a22650c66bd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= | golang.org/x/sys v0.29.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= | ||||||
| golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= | golang.org/x/text v0.21.0 h1:zyQAAkrwaneQ066sspRyJaG9VNi/YJ1NfzcGB3hZ/qo= | ||||||
| golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= | golang.org/x/text v0.21.0/go.mod h1:4IBbMaMmOPCJ8SecivzSH54+73PCFmPWxNTLm+vZkEQ= | ||||||
| golang.org/x/net v0.0.0-20200822124328-c89045814202/go.mod h1:/O7V0waA8r7cgGh81Ro3o1hOxt32SMVPicZroKQ2sZA= | google.golang.org/genproto/googleapis/rpc v0.0.0-20241216192217-9240e9c98484 h1:Z7FRVJPSMaHQxD0uXU8WdgFh8PseLM8Q8NzhnpMrBhQ= | ||||||
| golang.org/x/net v0.0.0-20210405180319-a5a99cb37ef4/go.mod h1:p54w0d4576C0XHj96bSt6lcn1PtDYWL6XObtHCRCNQM= | google.golang.org/genproto/googleapis/rpc v0.0.0-20241216192217-9240e9c98484/go.mod h1:lcTa1sDdWEIHMWlITnIczmw5w60CF9ffkb8Z+DVmmjA= | ||||||
| golang.org/x/net v0.0.0-20210805182204-aaa1db679c0d/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y= | google.golang.org/grpc v1.69.4 h1:MF5TftSMkd8GLw/m0KM6V8CMOCY6NZ1NQDPGFgbTt4A= | ||||||
| golang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U= | google.golang.org/grpc v1.69.4/go.mod h1:vyjdE6jLBI76dgpDojsFGNaHlxdjXN9ghpnd2o7JGZ4= | ||||||
| golang.org/x/oauth2 v0.0.0-20200107190931-bf48bf16ab8d/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw= | google.golang.org/protobuf v1.36.3 h1:82DV7MYdb8anAVi3qge1wSnMDrnKK7ebr+I0hHRN1BU= | ||||||
| golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= | google.golang.org/protobuf v1.36.3/go.mod h1:9fA7Ob0pmnwhb644+1+CVWFRbNajQ6iRojtC/QF5bRE= | ||||||
| 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= |  | ||||||
| gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= | gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= | ||||||
| gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15 h1:YR8cESwS4TdDjEe65xsg0ogRM/Nc3DYOhEAlW+xobZo= | gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk= | ||||||
| gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= | gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q= | ||||||
| gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= | gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= | ||||||
| gopkg.in/yaml.v2 v2.2.3/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= | gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= | ||||||
| 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= |  | ||||||
|   | |||||||
							
								
								
									
										117
									
								
								hooks/metadata/metadata.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										117
									
								
								hooks/metadata/metadata.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,117 @@ | |||||||
|  | package metadata | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v4/client" | ||||||
|  | 	"go.unistack.org/micro/v4/metadata" | ||||||
|  | 	"go.unistack.org/micro/v4/server" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type wrapper struct { | ||||||
|  | 	keys []string | ||||||
|  |  | ||||||
|  | 	client.Client | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func NewClientWrapper(keys ...string) client.Wrapper { | ||||||
|  | 	return func(c client.Client) client.Client { | ||||||
|  | 		handler := &wrapper{ | ||||||
|  | 			Client: c, | ||||||
|  | 			keys:   keys, | ||||||
|  | 		} | ||||||
|  | 		return handler | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func NewClientCallWrapper(keys ...string) client.CallWrapper { | ||||||
|  | 	return func(fn client.CallFunc) client.CallFunc { | ||||||
|  | 		return func(ctx context.Context, addr string, req client.Request, rsp interface{}, opts client.CallOptions) error { | ||||||
|  | 			if keys == nil { | ||||||
|  | 				return fn(ctx, addr, req, rsp, opts) | ||||||
|  | 			} | ||||||
|  | 			if imd, iok := metadata.FromIncomingContext(ctx); iok && imd != nil { | ||||||
|  | 				omd, ook := metadata.FromOutgoingContext(ctx) | ||||||
|  | 				if !ook || omd == nil { | ||||||
|  | 					omd = metadata.New(len(imd)) | ||||||
|  | 				} | ||||||
|  | 				for _, k := range keys { | ||||||
|  | 					if v, ok := imd.Get(k); ok { | ||||||
|  | 						omd.Add(k, v...) | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 				if !ook { | ||||||
|  | 					ctx = metadata.NewOutgoingContext(ctx, omd) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			return fn(ctx, addr, req, rsp, opts) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *wrapper) Call(ctx context.Context, req client.Request, rsp interface{}, opts ...client.CallOption) error { | ||||||
|  | 	if w.keys == nil { | ||||||
|  | 		return w.Client.Call(ctx, req, rsp, opts...) | ||||||
|  | 	} | ||||||
|  | 	if imd, iok := metadata.FromIncomingContext(ctx); iok && imd != nil { | ||||||
|  | 		omd, ook := metadata.FromOutgoingContext(ctx) | ||||||
|  | 		if !ook || omd == nil { | ||||||
|  | 			omd = metadata.New(len(imd)) | ||||||
|  | 		} | ||||||
|  | 		for _, k := range w.keys { | ||||||
|  | 			if v, ok := imd.Get(k); ok { | ||||||
|  | 				omd.Add(k, v...) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		if !ook { | ||||||
|  | 			ctx = metadata.NewOutgoingContext(ctx, omd) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return w.Client.Call(ctx, req, rsp, opts...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *wrapper) Stream(ctx context.Context, req client.Request, opts ...client.CallOption) (client.Stream, error) { | ||||||
|  | 	if w.keys == nil { | ||||||
|  | 		return w.Client.Stream(ctx, req, opts...) | ||||||
|  | 	} | ||||||
|  | 	if imd, iok := metadata.FromIncomingContext(ctx); iok && imd != nil { | ||||||
|  | 		omd, ook := metadata.FromOutgoingContext(ctx) | ||||||
|  | 		if !ook || omd == nil { | ||||||
|  | 			omd = metadata.New(len(imd)) | ||||||
|  | 		} | ||||||
|  | 		for _, k := range w.keys { | ||||||
|  | 			if v, ok := imd.Get(k); ok { | ||||||
|  | 				omd.Add(k, v...) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		if !ook { | ||||||
|  | 			ctx = metadata.NewOutgoingContext(ctx, omd) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return w.Client.Stream(ctx, req, opts...) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func NewServerHandlerWrapper(keys ...string) server.HandlerWrapper { | ||||||
|  | 	return func(fn server.HandlerFunc) server.HandlerFunc { | ||||||
|  | 		return func(ctx context.Context, req server.Request, rsp interface{}) error { | ||||||
|  | 			if keys == nil { | ||||||
|  | 				return fn(ctx, req, rsp) | ||||||
|  | 			} | ||||||
|  | 			if imd, iok := metadata.FromIncomingContext(ctx); iok && imd != nil { | ||||||
|  | 				omd, ook := metadata.FromOutgoingContext(ctx) | ||||||
|  | 				if !ook || omd == nil { | ||||||
|  | 					omd = metadata.New(len(imd)) | ||||||
|  | 				} | ||||||
|  | 				for _, k := range keys { | ||||||
|  | 					if v, ok := imd.Get(k); ok { | ||||||
|  | 						omd.Add(k, v...) | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 				if !ook { | ||||||
|  | 					ctx = metadata.NewOutgoingContext(ctx, omd) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			return fn(ctx, req, rsp) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										63
									
								
								hooks/recovery/recovery.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										63
									
								
								hooks/recovery/recovery.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,63 @@ | |||||||
|  | package recovery | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"fmt" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v4/errors" | ||||||
|  | 	"go.unistack.org/micro/v4/server" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func NewOptions(opts ...Option) Options { | ||||||
|  | 	options := Options{ | ||||||
|  | 		ServerHandlerFn: DefaultServerHandlerFn, | ||||||
|  | 	} | ||||||
|  | 	for _, o := range opts { | ||||||
|  | 		o(&options) | ||||||
|  | 	} | ||||||
|  | 	return options | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type Options struct { | ||||||
|  | 	ServerHandlerFn func(context.Context, server.Request, interface{}, error) error | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type Option func(*Options) | ||||||
|  |  | ||||||
|  | func ServerHandlerFunc(fn func(context.Context, server.Request, interface{}, error) error) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.ServerHandlerFn = fn | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | var DefaultServerHandlerFn = func(ctx context.Context, req server.Request, rsp interface{}, err error) error { | ||||||
|  | 	return errors.BadRequest("", "%v", err) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | var Hook = NewHook() | ||||||
|  |  | ||||||
|  | type hook struct { | ||||||
|  | 	opts Options | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func NewHook(opts ...Option) *hook { | ||||||
|  | 	return &hook{opts: NewOptions(opts...)} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *hook) ServerHandler(next server.FuncHandler) server.FuncHandler { | ||||||
|  | 	return func(ctx context.Context, req server.Request, rsp interface{}) (err error) { | ||||||
|  | 		defer func() { | ||||||
|  | 			r := recover() | ||||||
|  | 			switch verr := r.(type) { | ||||||
|  | 			case nil: | ||||||
|  | 				return | ||||||
|  | 			case error: | ||||||
|  | 				err = w.opts.ServerHandlerFn(ctx, req, rsp, verr) | ||||||
|  | 			default: | ||||||
|  | 				err = w.opts.ServerHandlerFn(ctx, req, rsp, fmt.Errorf("%v", r)) | ||||||
|  | 			} | ||||||
|  | 		}() | ||||||
|  | 		err = next(ctx, req, rsp) | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										114
									
								
								hooks/requestid/requestid.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										114
									
								
								hooks/requestid/requestid.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,114 @@ | |||||||
|  | package requestid | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"net/textproto" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v4/client" | ||||||
|  | 	"go.unistack.org/micro/v4/metadata" | ||||||
|  | 	"go.unistack.org/micro/v4/server" | ||||||
|  | 	"go.unistack.org/micro/v4/util/id" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type XRequestIDKey struct{} | ||||||
|  |  | ||||||
|  | // DefaultMetadataKey contains metadata key | ||||||
|  | var DefaultMetadataKey = textproto.CanonicalMIMEHeaderKey("x-request-id") | ||||||
|  |  | ||||||
|  | // DefaultMetadataFunc wil be used if user not provide own func to fill metadata | ||||||
|  | var DefaultMetadataFunc = func(ctx context.Context) (context.Context, error) { | ||||||
|  | 	var xid string | ||||||
|  |  | ||||||
|  | 	cid, cok := ctx.Value(XRequestIDKey{}).(string) | ||||||
|  | 	if cok && cid != "" { | ||||||
|  | 		xid = cid | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	imd, iok := metadata.FromIncomingContext(ctx) | ||||||
|  | 	if !iok || imd == nil { | ||||||
|  | 		imd = metadata.New(1) | ||||||
|  | 		ctx = metadata.NewIncomingContext(ctx, imd) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	omd, ook := metadata.FromOutgoingContext(ctx) | ||||||
|  | 	if !ook || omd == nil { | ||||||
|  | 		omd = metadata.New(1) | ||||||
|  | 		ctx = metadata.NewOutgoingContext(ctx, omd) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if xid == "" { | ||||||
|  | 		var ids []string | ||||||
|  | 		if ids, iok = imd.Get(DefaultMetadataKey); iok { | ||||||
|  | 			for i := range ids { | ||||||
|  | 				if ids[i] != "" { | ||||||
|  | 					xid = ids[i] | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		if ids, ook = omd.Get(DefaultMetadataKey); ook { | ||||||
|  | 			for i := range ids { | ||||||
|  | 				if ids[i] != "" { | ||||||
|  | 					xid = ids[i] | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if xid == "" { | ||||||
|  | 		var err error | ||||||
|  | 		xid, err = id.New() | ||||||
|  | 		if err != nil { | ||||||
|  | 			return ctx, err | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !cok { | ||||||
|  | 		ctx = context.WithValue(ctx, XRequestIDKey{}, xid) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !iok { | ||||||
|  | 		imd.Set(DefaultMetadataKey, xid) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if !ook { | ||||||
|  | 		omd.Set(DefaultMetadataKey, xid) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return ctx, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type hook struct{} | ||||||
|  |  | ||||||
|  | func NewHook() *hook { | ||||||
|  | 	return &hook{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *hook) ServerHandler(next server.FuncHandler) server.FuncHandler { | ||||||
|  | 	return func(ctx context.Context, req server.Request, rsp interface{}) error { | ||||||
|  | 		var err error | ||||||
|  | 		if ctx, err = DefaultMetadataFunc(ctx); err != nil { | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  | 		return next(ctx, req, rsp) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *hook) ClientCall(next client.FuncCall) client.FuncCall { | ||||||
|  | 	return func(ctx context.Context, req client.Request, rsp interface{}, opts ...client.CallOption) error { | ||||||
|  | 		var err error | ||||||
|  | 		if ctx, err = DefaultMetadataFunc(ctx); err != nil { | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  | 		return next(ctx, req, rsp, opts...) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *hook) ClientStream(next client.FuncStream) client.FuncStream { | ||||||
|  | 	return func(ctx context.Context, req client.Request, opts ...client.CallOption) (client.Stream, error) { | ||||||
|  | 		var err error | ||||||
|  | 		if ctx, err = DefaultMetadataFunc(ctx); err != nil { | ||||||
|  | 			return nil, err | ||||||
|  | 		} | ||||||
|  | 		return next(ctx, req, opts...) | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										33
									
								
								hooks/requestid/requestid_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										33
									
								
								hooks/requestid/requestid_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,33 @@ | |||||||
|  | package requestid | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"testing" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v4/metadata" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | func TestDefaultMetadataFunc(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  |  | ||||||
|  | 	nctx, err := DefaultMetadataFunc(ctx) | ||||||
|  | 	if err != nil { | ||||||
|  | 		t.Fatalf("%v", err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	imd, ok := metadata.FromIncomingContext(nctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		t.Fatalf("md missing in incoming context") | ||||||
|  | 	} | ||||||
|  | 	omd, ok := metadata.FromOutgoingContext(nctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		t.Fatalf("md missing in outgoing context") | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	_, iok := imd.Get(DefaultMetadataKey) | ||||||
|  | 	_, ook := omd.Get(DefaultMetadataKey) | ||||||
|  |  | ||||||
|  | 	if !iok || !ook { | ||||||
|  | 		t.Fatalf("missing metadata key value") | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										133
									
								
								hooks/validator/validator.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										133
									
								
								hooks/validator/validator.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,133 @@ | |||||||
|  | package validator | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v4/client" | ||||||
|  | 	"go.unistack.org/micro/v4/errors" | ||||||
|  | 	"go.unistack.org/micro/v4/server" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | var ( | ||||||
|  | 	DefaultClientErrorFunc = func(req client.Request, rsp interface{}, err error) error { | ||||||
|  | 		if rsp != nil { | ||||||
|  | 			return errors.BadGateway(req.Service(), "%v", err) | ||||||
|  | 		} | ||||||
|  | 		return errors.BadRequest(req.Service(), "%v", err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	DefaultServerErrorFunc = func(req server.Request, rsp interface{}, err error) error { | ||||||
|  | 		if rsp != nil { | ||||||
|  | 			return errors.BadGateway(req.Service(), "%v", err) | ||||||
|  | 		} | ||||||
|  | 		return errors.BadRequest(req.Service(), "%v", err) | ||||||
|  | 	} | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type ( | ||||||
|  | 	ClientErrorFunc func(client.Request, interface{}, error) error | ||||||
|  | 	ServerErrorFunc func(server.Request, interface{}, error) error | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // Options struct holds wrapper options | ||||||
|  | type Options struct { | ||||||
|  | 	ClientErrorFn          ClientErrorFunc | ||||||
|  | 	ServerErrorFn          ServerErrorFunc | ||||||
|  | 	ClientValidateResponse bool | ||||||
|  | 	ServerValidateResponse bool | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Option func signature | ||||||
|  | type Option func(*Options) | ||||||
|  |  | ||||||
|  | func ClientValidateResponse(b bool) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.ClientValidateResponse = b | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func ServerValidateResponse(b bool) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.ClientValidateResponse = b | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func ClientReqErrorFn(fn ClientErrorFunc) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.ClientErrorFn = fn | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func ServerErrorFn(fn ServerErrorFunc) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.ServerErrorFn = fn | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func NewOptions(opts ...Option) Options { | ||||||
|  | 	options := Options{ | ||||||
|  | 		ClientErrorFn: DefaultClientErrorFunc, | ||||||
|  | 		ServerErrorFn: DefaultServerErrorFunc, | ||||||
|  | 	} | ||||||
|  | 	for _, o := range opts { | ||||||
|  | 		o(&options) | ||||||
|  | 	} | ||||||
|  | 	return options | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func NewHook(opts ...Option) *hook { | ||||||
|  | 	return &hook{opts: NewOptions(opts...)} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type validator interface { | ||||||
|  | 	Validate() error | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type hook struct { | ||||||
|  | 	opts Options | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *hook) ClientCall(next client.FuncCall) client.FuncCall { | ||||||
|  | 	return func(ctx context.Context, req client.Request, rsp interface{}, opts ...client.CallOption) error { | ||||||
|  | 		if v, ok := req.Body().(validator); ok { | ||||||
|  | 			if err := v.Validate(); err != nil { | ||||||
|  | 				return w.opts.ClientErrorFn(req, nil, err) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		err := next(ctx, req, rsp, opts...) | ||||||
|  | 		if v, ok := rsp.(validator); ok && w.opts.ClientValidateResponse { | ||||||
|  | 			if verr := v.Validate(); verr != nil { | ||||||
|  | 				return w.opts.ClientErrorFn(req, rsp, verr) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *hook) ClientStream(next client.FuncStream) client.FuncStream { | ||||||
|  | 	return func(ctx context.Context, req client.Request, opts ...client.CallOption) (client.Stream, error) { | ||||||
|  | 		if v, ok := req.Body().(validator); ok { | ||||||
|  | 			if err := v.Validate(); err != nil { | ||||||
|  | 				return nil, w.opts.ClientErrorFn(req, nil, err) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		return next(ctx, req, opts...) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *hook) ServerHandler(next server.FuncHandler) server.FuncHandler { | ||||||
|  | 	return func(ctx context.Context, req server.Request, rsp interface{}) error { | ||||||
|  | 		if v, ok := req.Body().(validator); ok { | ||||||
|  | 			if err := v.Validate(); err != nil { | ||||||
|  | 				return w.opts.ServerErrorFn(req, nil, err) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		err := next(ctx, req, rsp) | ||||||
|  | 		if v, ok := rsp.(validator); ok && w.opts.ServerValidateResponse { | ||||||
|  | 			if verr := v.Validate(); verr != nil { | ||||||
|  | 				return w.opts.ServerErrorFn(req, rsp, verr) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | } | ||||||
| @@ -13,6 +13,15 @@ func FromContext(ctx context.Context) (Logger, bool) { | |||||||
| 	return l, ok | 	return l, ok | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // MustContext returns logger from passed context or DefaultLogger if empty | ||||||
|  | func MustContext(ctx context.Context) Logger { | ||||||
|  | 	l, ok := FromContext(ctx) | ||||||
|  | 	if !ok { | ||||||
|  | 		panic("missing logger") | ||||||
|  | 	} | ||||||
|  | 	return l | ||||||
|  | } | ||||||
|  |  | ||||||
| // NewContext stores logger into passed context | // NewContext stores logger into passed context | ||||||
| func NewContext(ctx context.Context, l Logger) context.Context { | func NewContext(ctx context.Context, l Logger) context.Context { | ||||||
| 	if ctx == nil { | 	if ctx == nil { | ||||||
|   | |||||||
| @@ -1,243 +0,0 @@ | |||||||
| package logger |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"context" |  | ||||||
| 	"encoding/json" |  | ||||||
| 	"fmt" |  | ||||||
| 	"os" |  | ||||||
| 	"runtime" |  | ||||||
| 	"strings" |  | ||||||
| 	"sync" |  | ||||||
| 	"time" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| type defaultLogger struct { |  | ||||||
| 	enc      *json.Encoder |  | ||||||
| 	logFunc  LogFunc |  | ||||||
| 	logfFunc LogfFunc |  | ||||||
| 	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 |  | ||||||
| 	for i := len(l.opts.Wrappers); i > 0; i-- { |  | ||||||
| 		l.logFunc = l.opts.Wrappers[i-1].Log(l.logFunc) |  | ||||||
| 		l.logfFunc = l.opts.Wrappers[i-1].Logf(l.logfFunc) |  | ||||||
| 	} |  | ||||||
| 	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) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	oldopts.Wrappers = newopts.Wrappers |  | ||||||
| 	l.Lock() |  | ||||||
| 	cl := &defaultLogger{opts: oldopts, logFunc: l.logFunc, logfFunc: l.logfFunc, enc: json.NewEncoder(l.opts.Out)} |  | ||||||
| 	l.Unlock() |  | ||||||
|  |  | ||||||
| 	// wrap the Log func |  | ||||||
| 	for i := len(newopts.Wrappers); i > 0; i-- { |  | ||||||
| 		cl.logFunc = newopts.Wrappers[i-1].Log(cl.logFunc) |  | ||||||
| 		cl.logfFunc = newopts.Wrappers[i-1].Logf(cl.logfFunc) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	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 { |  | ||||||
| 	nl := &defaultLogger{opts: l.opts, enc: l.enc} |  | ||||||
| 	if len(fields) == 0 { |  | ||||||
| 		return nl |  | ||||||
| 	} else if len(fields)%2 != 0 { |  | ||||||
| 		fields = fields[:len(fields)-1] |  | ||||||
| 	} |  | ||||||
| 	nl.logFunc = l.logFunc |  | ||||||
| 	nl.logfFunc = l.logfFunc |  | ||||||
| 	nl.opts.Fields = append(nl.opts.Fields, fields...) |  | ||||||
| 	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.logfFunc(ctx, InfoLevel, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) Errorf(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	l.logfFunc(ctx, ErrorLevel, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) Debugf(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	l.logfFunc(ctx, DebugLevel, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) Warnf(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	l.logfFunc(ctx, WarnLevel, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) Tracef(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	l.logfFunc(ctx, TraceLevel, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (l *defaultLogger) Fatalf(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	l.logfFunc(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.logFunc = l.Log |  | ||||||
| 	l.logfFunc = l.Logf |  | ||||||
| 	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 provides a log interface | ||||||
| package logger // import "go.unistack.org/micro/v3/logger" | package logger | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"os" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | type ContextAttrFunc func(ctx context.Context) []interface{} | ||||||
|  |  | ||||||
|  | var DefaultContextAttrFuncs []ContextAttrFunc | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| 	// DefaultLogger variable | 	// DefaultLogger variable | ||||||
| 	DefaultLogger = NewLogger(WithLevel(ParseLevel(os.Getenv("MICRO_LOG_LEVEL")))) | 	DefaultLogger = NewLogger() | ||||||
| 	// DefaultLevel used by logger | 	// DefaultLevel used by logger | ||||||
| 	DefaultLevel = InfoLevel | 	DefaultLevel = InfoLevel | ||||||
| 	// DefaultCallerSkipCount used by logger |  | ||||||
| 	DefaultCallerSkipCount = 2 |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Logger is a generic logging interface | // Logger is a generic logging interface | ||||||
| @@ -30,111 +31,24 @@ type Logger interface { | |||||||
| 	// Fields set fields to always be logged with keyval pairs | 	// Fields set fields to always be logged with keyval pairs | ||||||
| 	Fields(fields ...interface{}) Logger | 	Fields(fields ...interface{}) Logger | ||||||
| 	// Info level message | 	// Info level message | ||||||
| 	Info(ctx context.Context, args ...interface{}) | 	Info(ctx context.Context, msg string, args ...interface{}) | ||||||
| 	// Trace level message | 	// Trace level message | ||||||
| 	Trace(ctx context.Context, args ...interface{}) | 	Trace(ctx context.Context, msg string, args ...interface{}) | ||||||
| 	// Debug level message | 	// Debug level message | ||||||
| 	Debug(ctx context.Context, args ...interface{}) | 	Debug(ctx context.Context, msg string, args ...interface{}) | ||||||
| 	// Warn level message | 	// Warn level message | ||||||
| 	Warn(ctx context.Context, args ...interface{}) | 	Warn(ctx context.Context, msg string, args ...interface{}) | ||||||
| 	// Error level message | 	// Error level message | ||||||
| 	Error(ctx context.Context, args ...interface{}) | 	Error(ctx context.Context, msg string, args ...interface{}) | ||||||
| 	// Fatal level message | 	// Fatal level message | ||||||
| 	Fatal(ctx context.Context, args ...interface{}) | 	Fatal(ctx context.Context, msg string, args ...interface{}) | ||||||
| 	// Infof level message |  | ||||||
| 	Infof(ctx context.Context, msg string, args ...interface{}) |  | ||||||
| 	// Tracef level message |  | ||||||
| 	Tracef(ctx context.Context, msg string, args ...interface{}) |  | ||||||
| 	// Debug level message |  | ||||||
| 	Debugf(ctx context.Context, msg string, args ...interface{}) |  | ||||||
| 	// Warn level message |  | ||||||
| 	Warnf(ctx context.Context, msg string, args ...interface{}) |  | ||||||
| 	// Error level message |  | ||||||
| 	Errorf(ctx context.Context, msg string, args ...interface{}) |  | ||||||
| 	// Fatal level message |  | ||||||
| 	Fatalf(ctx context.Context, msg string, args ...interface{}) |  | ||||||
| 	// Log logs message with needed level | 	// Log logs message with needed level | ||||||
| 	Log(ctx context.Context, level Level, args ...interface{}) | 	Log(ctx context.Context, level Level, msg string, args ...interface{}) | ||||||
| 	// Logf logs message with needed level | 	// Name returns broker instance name | ||||||
| 	Logf(ctx context.Context, level Level, msg string, args ...interface{}) | 	Name() string | ||||||
| 	// String returns the name of logger | 	// String returns the type of logger | ||||||
| 	String() string | 	String() string | ||||||
| } | } | ||||||
|  |  | ||||||
| // Field contains keyval pair | // Field contains keyval pair | ||||||
| type Field interface{} | type Field interface{} | ||||||
|  |  | ||||||
| // Info writes msg to default logger on info level |  | ||||||
| func Info(ctx context.Context, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Info(ctx, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Error writes msg to default logger on error level |  | ||||||
| func Error(ctx context.Context, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Error(ctx, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Debug writes msg to default logger on debug level |  | ||||||
| func Debug(ctx context.Context, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Debug(ctx, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Warn writes msg to default logger on warn level |  | ||||||
| func Warn(ctx context.Context, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Warn(ctx, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Trace writes msg to default logger on trace level |  | ||||||
| func Trace(ctx context.Context, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Trace(ctx, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Fatal writes msg to default logger on fatal level |  | ||||||
| func Fatal(ctx context.Context, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Fatal(ctx, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Infof writes formatted msg to default logger on info level |  | ||||||
| func Infof(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Infof(ctx, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Errorf writes formatted msg to default logger on error level |  | ||||||
| func Errorf(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Errorf(ctx, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Debugf writes formatted msg to default logger on debug level |  | ||||||
| func Debugf(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Debugf(ctx, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Warnf writes formatted msg to default logger on warn level |  | ||||||
| func Warnf(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Warnf(ctx, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Tracef writes formatted msg to default logger on trace level |  | ||||||
| func Tracef(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Tracef(ctx, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Fatalf writes formatted msg to default logger on fatal level |  | ||||||
| func Fatalf(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	DefaultLogger.Fatalf(ctx, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // V returns true if passed level enabled in default logger |  | ||||||
| func V(level Level) bool { |  | ||||||
| 	return DefaultLogger.V(level) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Init initialize logger |  | ||||||
| func Init(opts ...Option) error { |  | ||||||
| 	return DefaultLogger.Init(opts...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Fields create logger with specific fields |  | ||||||
| func Fields(fields ...interface{}) Logger { |  | ||||||
| 	return DefaultLogger.Fields(fields...) |  | ||||||
| } |  | ||||||
|   | |||||||
| @@ -1,148 +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) |  | ||||||
| 	} |  | ||||||
| 	l.Fields("key", "val").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()) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestLoggerWrapper(t *testing.T) { |  | ||||||
| 	ctx := context.TODO() |  | ||||||
| 	buf := bytes.NewBuffer(nil) |  | ||||||
| 	l := NewLogger(WithLevel(TraceLevel), WithOutput(buf)) |  | ||||||
| 	if err := l.Init(WrapLogger(NewOmitWrapper())); err != nil { |  | ||||||
| 		t.Fatal(err) |  | ||||||
| 	} |  | ||||||
| 	type secret struct { |  | ||||||
| 		Name  string |  | ||||||
| 		Passw string `logger:"omit"` |  | ||||||
| 	} |  | ||||||
| 	s := &secret{Name: "name", Passw: "secret"} |  | ||||||
| 	l.Errorf(ctx, "test %#+v", s) |  | ||||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`logger.secret{Name:\"name\", Passw:\"\"}"`)) { |  | ||||||
| 		t.Fatalf("omit not works, struct: %v, output: %s", s, buf.Bytes()) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestOmitLoggerWrapper(t *testing.T) { |  | ||||||
| 	ctx := context.TODO() |  | ||||||
| 	buf := bytes.NewBuffer(nil) |  | ||||||
| 	l := NewOmitLogger(NewLogger(WithLevel(TraceLevel), WithOutput(buf))) |  | ||||||
| 	if err := l.Init(); err != nil { |  | ||||||
| 		t.Fatal(err) |  | ||||||
| 	} |  | ||||||
| 	type secret struct { |  | ||||||
| 		Name  string |  | ||||||
| 		Passw string `logger:"omit"` |  | ||||||
| 	} |  | ||||||
| 	s := &secret{Name: "name", Passw: "secret"} |  | ||||||
| 	l.Errorf(ctx, "test %#+v", s) |  | ||||||
| 	if !bytes.Contains(buf.Bytes(), []byte(`logger.secret{Name:\"name\", Passw:\"\"}"`)) { |  | ||||||
| 		t.Fatalf("omit not works, struct: %v, output: %s", 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,45 +3,116 @@ package logger | |||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"io" | 	"io" | ||||||
|  | 	"log/slog" | ||||||
| 	"os" | 	"os" | ||||||
|  | 	"slices" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v4/meter" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Option func | // Option func signature | ||||||
| type Option func(*Options) | type Option func(*Options) | ||||||
|  |  | ||||||
| // Options holds logger options | // Options holds logger options | ||||||
| type Options struct { | type Options struct { | ||||||
|  | 	// TimeKey is the key used for the time of the log call | ||||||
|  | 	TimeKey string | ||||||
|  | 	// LevelKey is the key used for the level of the log call | ||||||
|  | 	LevelKey string | ||||||
|  | 	// ErroreKey is the key used for the error of the log call | ||||||
|  | 	ErrorKey string | ||||||
|  | 	// MessageKey is the key used for the message of the log call | ||||||
|  | 	MessageKey string | ||||||
|  | 	// SourceKey is the key used for the source file and line of the log call | ||||||
|  | 	SourceKey string | ||||||
|  | 	// StacktraceKey is the key used for the stacktrace | ||||||
|  | 	StacktraceKey string | ||||||
|  | 	// Name holds the logger name | ||||||
|  | 	Name string | ||||||
| 	// Out holds the output writer | 	// Out holds the output writer | ||||||
| 	Out io.Writer | 	Out io.Writer | ||||||
| 	// Context holds exernal options | 	// Context holds exernal options | ||||||
| 	Context context.Context | 	Context context.Context | ||||||
|  | 	// Meter used to count logs for specific level | ||||||
|  | 	Meter meter.Meter | ||||||
|  | 	// TimeFunc used to obtain current time | ||||||
|  | 	TimeFunc func() time.Time | ||||||
| 	// Fields holds additional metadata | 	// Fields holds additional metadata | ||||||
| 	Fields []interface{} | 	Fields []interface{} | ||||||
| 	// Name holds the logger name | 	// ContextAttrFuncs contains funcs that executed before log func on context | ||||||
| 	Name string | 	ContextAttrFuncs []ContextAttrFunc | ||||||
| 	// Wrappers logger wrapper that called before actual Log/Logf function | 	// callerSkipCount number of frmaes to skip | ||||||
| 	Wrappers []Wrapper | 	CallerSkipCount int | ||||||
| 	// The logging level the logger should log | 	// The logging level the logger should log | ||||||
| 	Level Level | 	Level Level | ||||||
| 	// CallerSkipCount number of frmaes to skip | 	// AddSource enabled writing source file and position in log | ||||||
| 	CallerSkipCount int | 	AddSource bool | ||||||
|  | 	// AddStacktrace controls writing of stacktaces on error | ||||||
|  | 	AddStacktrace bool | ||||||
|  | 	// DedupKeys deduplicate keys in log output | ||||||
|  | 	DedupKeys bool | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewOptions creates new options struct | // NewOptions creates new options struct | ||||||
| func NewOptions(opts ...Option) Options { | func NewOptions(opts ...Option) Options { | ||||||
| 	options := Options{ | 	options := Options{ | ||||||
| 		Level:           DefaultLevel, | 		Level:            DefaultLevel, | ||||||
| 		Fields:          make([]interface{}, 0, 6), | 		Fields:           make([]interface{}, 0, 6), | ||||||
| 		Out:             os.Stderr, | 		Out:              os.Stderr, | ||||||
| 		CallerSkipCount: DefaultCallerSkipCount, | 		Context:          context.Background(), | ||||||
| 		Context:         context.Background(), | 		ContextAttrFuncs: DefaultContextAttrFuncs, | ||||||
|  | 		AddSource:        true, | ||||||
|  | 		TimeFunc:         time.Now, | ||||||
|  | 		Meter:            meter.DefaultMeter, | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	WithMicroKeys()(&options) | ||||||
|  |  | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&options) | 		o(&options) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return options | 	return options | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // WithContextAttrFuncs appends default funcs for the context attrs filler | ||||||
|  | func WithContextAttrFuncs(fncs ...ContextAttrFunc) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.ContextAttrFuncs = append(o.ContextAttrFuncs, fncs...) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithDedupKeys dont log duplicate keys | ||||||
|  | func WithDedupKeys(b bool) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		o.DedupKeys = b | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // WithAddFields add fields for the logger | ||||||
|  | func WithAddFields(fields ...interface{}) Option { | ||||||
|  | 	return func(o *Options) { | ||||||
|  | 		if o.DedupKeys { | ||||||
|  | 			for i := 0; i < len(o.Fields); i += 2 { | ||||||
|  | 				for j := 0; j < len(fields); j += 2 { | ||||||
|  | 					iv, iok := o.Fields[i].(string) | ||||||
|  | 					jv, jok := fields[j].(string) | ||||||
|  | 					if iok && jok && iv == jv { | ||||||
|  | 						o.Fields[i+1] = fields[j+1] | ||||||
|  | 						fields = slices.Delete(fields, j, j+2) | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			if len(fields) > 0 { | ||||||
|  | 				o.Fields = append(o.Fields, fields...) | ||||||
|  | 			} | ||||||
|  | 		} else { | ||||||
|  | 			o.Fields = append(o.Fields, fields...) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // WithFields set default fields for the logger | // WithFields set default fields for the logger | ||||||
| func WithFields(fields ...interface{}) Option { | func WithFields(fields ...interface{}) Option { | ||||||
| 	return func(o *Options) { | 	return func(o *Options) { | ||||||
| @@ -63,10 +134,17 @@ func WithOutput(out io.Writer) Option { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // WithCallerSkipCount set frame count to skip | // WithAddStacktrace controls writing stacktrace on error | ||||||
| func WithCallerSkipCount(c int) Option { | func WithAddStacktrace(v bool) Option { | ||||||
| 	return func(o *Options) { | 	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 | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -84,9 +162,69 @@ func WithName(n string) Option { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // WrapLogger adds a logger Wrapper to a list of options passed into the logger | // WithMeter sets the meter | ||||||
| func WrapLogger(w Wrapper) Option { | func WithMeter(m meter.Meter) Option { | ||||||
| 	return func(o *Options) { | 	return func(o *Options) { | ||||||
| 		o.Wrappers = append(o.Wrappers, w) | 		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 | ||||||
|  | 		} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										375
									
								
								logger/slog/slog.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										375
									
								
								logger/slog/slog.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,375 @@ | |||||||
|  | package slog | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"io" | ||||||
|  | 	"log/slog" | ||||||
|  | 	"os" | ||||||
|  | 	"reflect" | ||||||
|  | 	"regexp" | ||||||
|  | 	"runtime" | ||||||
|  | 	"strconv" | ||||||
|  | 	"sync" | ||||||
|  | 	"sync/atomic" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
|  | 	"go.unistack.org/micro/v4/logger" | ||||||
|  | 	"go.unistack.org/micro/v4/semconv" | ||||||
|  | 	"go.unistack.org/micro/v4/tracer" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | const ( | ||||||
|  | 	badKey = "!BADKEY" | ||||||
|  | 	// defaultCallerSkipCount used by logger | ||||||
|  | 	defaultCallerSkipCount = 3 | ||||||
|  | 	timeFormat             = "2006-01-02T15:04:05.000000000Z07:00" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | 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 | ||||||
|  | 		a.Value = slog.StringValue(a.Value.Time().Format(timeFormat)) | ||||||
|  | 	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 || 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) | ||||||
|  | } | ||||||
							
								
								
									
										454
									
								
								logger/slog/slog_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										454
									
								
								logger/slog/slog_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,454 @@ | |||||||
|  | package slog | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"bytes" | ||||||
|  | 	"context" | ||||||
|  | 	"errors" | ||||||
|  | 	"fmt" | ||||||
|  | 	"log" | ||||||
|  | 	"log/slog" | ||||||
|  | 	"strings" | ||||||
|  | 	"testing" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
|  | 	"github.com/google/uuid" | ||||||
|  | 	"go.unistack.org/micro/v4/logger" | ||||||
|  | 	"go.unistack.org/micro/v4/metadata" | ||||||
|  | 	"go.unistack.org/micro/v4/util/buffer" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // always first to have proper check | ||||||
|  | func TestStacktrace(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.DebugLevel), logger.WithOutput(buf), | ||||||
|  | 		WithHandlerFunc(slog.NewTextHandler), | ||||||
|  | 		logger.WithAddStacktrace(true), | ||||||
|  | 	) | ||||||
|  | 	if err := l.Init(logger.WithFields("key1", "val1")); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l.Error(ctx, "msg1", errors.New("err")) | ||||||
|  |  | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`slog_test.go:32`)) { | ||||||
|  | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestDelayedBuffer(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	dbuf := buffer.NewDelayedBuffer(100, 100*time.Millisecond, buf) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.ErrorLevel), logger.WithOutput(dbuf), | ||||||
|  | 		WithHandlerFunc(slog.NewTextHandler), | ||||||
|  | 		logger.WithAddStacktrace(true), | ||||||
|  | 	) | ||||||
|  | 	if err := l.Init(logger.WithFields("key1", "val1")); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l.Error(ctx, "msg1", errors.New("err")) | ||||||
|  | 	time.Sleep(120 * time.Millisecond) | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`key1=val1`)) { | ||||||
|  | 		t.Fatalf("logger delayed buffer not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestTime(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.ErrorLevel), logger.WithOutput(buf), | ||||||
|  | 		WithHandlerFunc(slog.NewTextHandler), | ||||||
|  | 		logger.WithAddStacktrace(true), | ||||||
|  | 		logger.WithTimeFunc(func() time.Time { | ||||||
|  | 			return time.Unix(0, 0) | ||||||
|  | 		}), | ||||||
|  | 	) | ||||||
|  | 	if err := l.Init(logger.WithFields("key1", "val1")); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l.Error(ctx, "msg1", errors.New("err")) | ||||||
|  |  | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`timestamp=1970-01-01T03:00:00.000000000+03:00`)) && | ||||||
|  | 		!bytes.Contains(buf.Bytes(), []byte(`timestamp=1970-01-01T00:00:00.000000000Z`)) { | ||||||
|  | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestWithFields(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.InfoLevel), logger.WithOutput(buf), | ||||||
|  | 		WithHandlerFunc(slog.NewTextHandler), | ||||||
|  | 		logger.WithDedupKeys(true), | ||||||
|  | 	) | ||||||
|  | 	if err := l.Init(logger.WithFields("key1", "val1")); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l.Info(ctx, "msg1") | ||||||
|  |  | ||||||
|  | 	l = l.Fields("key1", "val2") | ||||||
|  |  | ||||||
|  | 	l.Info(ctx, "msg2") | ||||||
|  |  | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`msg=msg2 key1=val1`)) { | ||||||
|  | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestWithDedupKeysWithAddFields(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.InfoLevel), logger.WithOutput(buf), | ||||||
|  | 		WithHandlerFunc(slog.NewTextHandler), | ||||||
|  | 		logger.WithDedupKeys(true), | ||||||
|  | 	) | ||||||
|  | 	if err := l.Init(logger.WithFields("key1", "val1")); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l.Info(ctx, "msg1") | ||||||
|  |  | ||||||
|  | 	if err := l.Init(logger.WithAddFields("key2", "val2")); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l.Info(ctx, "msg2") | ||||||
|  |  | ||||||
|  | 	if err := l.Init(logger.WithAddFields("key2", "val3", "key1", "val4")); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l.Info(ctx, "msg3") | ||||||
|  |  | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`msg=msg3 key1=val4 key2=val3`)) { | ||||||
|  | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestWithHandlerFunc(t *testing.T) { | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.InfoLevel), logger.WithOutput(buf), | ||||||
|  | 		WithHandlerFunc(slog.NewTextHandler), | ||||||
|  | 	) | ||||||
|  | 	if err := l.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l.Info(ctx, "msg1") | ||||||
|  |  | ||||||
|  | 	if !bytes.Contains(buf.Bytes(), []byte(`msg=msg1`)) { | ||||||
|  | 		t.Fatalf("logger error not works, buf contains: %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func TestWithAddFields(t *testing.T) { | ||||||
|  | 	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 { | ||||||
|  | 				key := strings.ToLower(k) | ||||||
|  | 				switch key { | ||||||
|  | 				case "x-request-id", "phone", "external-Id", "source-service", "x-app-install-id", "client-id", "client-ip": | ||||||
|  | 					attrs = append(attrs, key, v[0]) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			return attrs | ||||||
|  | 		}, | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	logger.DefaultContextAttrFuncs = append(logger.DefaultContextAttrFuncs, loggerContextAttrFuncs...) | ||||||
|  |  | ||||||
|  | 	ctx := context.TODO() | ||||||
|  | 	ctx = metadata.AppendIncomingContext(ctx, "X-Request-Id", uuid.New().String(), | ||||||
|  | 		"Source-Service", "Test-System") | ||||||
|  |  | ||||||
|  | 	buf := bytes.NewBuffer(nil) | ||||||
|  | 	l := NewLogger(logger.WithLevel(logger.TraceLevel), logger.WithOutput(buf)) | ||||||
|  | 	if err := l.Init(); err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	l.Info(ctx, "test message") | ||||||
|  | 	if !(bytes.Contains(buf.Bytes(), []byte(`"level":"info"`)) && bytes.Contains(buf.Bytes(), []byte(`"msg":"test message"`))) { | ||||||
|  | 		t.Fatalf("logger info, buf %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | 	if !(bytes.Contains(buf.Bytes(), []byte(`"x-request-id":"`))) { | ||||||
|  | 		t.Fatalf("logger info, buf %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | 	if !(bytes.Contains(buf.Bytes(), []byte(`"source-service":"Test-System"`))) { | ||||||
|  | 		t.Fatalf("logger info, buf %s", buf.Bytes()) | ||||||
|  | 	} | ||||||
|  | 	buf.Reset() | ||||||
|  | 	imd, _ := metadata.FromIncomingContext(ctx) | ||||||
|  | 	l.Info(ctx, "test message1") | ||||||
|  | 	imd.Set("Source-Service", "Test-System2") | ||||||
|  | 	l.Info(ctx, "test message2") | ||||||
|  |  | ||||||
|  | 	// t.Logf("xxx %s", buf.Bytes()) | ||||||
|  | } | ||||||
| @@ -8,7 +8,7 @@ import ( | |||||||
| 	"strconv" | 	"strconv" | ||||||
| 	"strings" | 	"strings" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/codec" | 	"go.unistack.org/micro/v4/codec" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| const sf = "0-+# " | const sf = "0-+# " | ||||||
| @@ -36,23 +36,33 @@ var ( | |||||||
| 	circularShortBytes = []byte("<shown>") | 	circularShortBytes = []byte("<shown>") | ||||||
| 	invalidAngleBytes  = []byte("<invalid>") | 	invalidAngleBytes  = []byte("<invalid>") | ||||||
| 	filteredBytes      = []byte("<filtered>") | 	filteredBytes      = []byte("<filtered>") | ||||||
| 	openBracketBytes   = []byte("[") | 	// openBracketBytes   = []byte("[") | ||||||
| 	closeBracketBytes  = []byte("]") | 	// closeBracketBytes  = []byte("]") | ||||||
| 	percentBytes       = []byte("%") | 	percentBytes    = []byte("%") | ||||||
| 	precisionBytes     = []byte(".") | 	precisionBytes  = []byte(".") | ||||||
| 	openAngleBytes     = []byte("<") | 	openAngleBytes  = []byte("<") | ||||||
| 	closeAngleBytes    = []byte(">") | 	closeAngleBytes = []byte(">") | ||||||
| 	openMapBytes       = []byte("{") | 	openMapBytes    = []byte("{") | ||||||
| 	closeMapBytes      = []byte("}") | 	closeMapBytes   = []byte("}") | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type unwrap struct { | type protoMessage interface { | ||||||
| 	val            interface{} | 	Reset() | ||||||
| 	s              fmt.State | 	ProtoMessage() | ||||||
| 	pointers       map[uintptr]int | } | ||||||
| 	opts           *Options |  | ||||||
| 	depth          int | type Wrapper struct { | ||||||
| 	ignoreNextType bool | 	pointers map[uintptr]int | ||||||
|  | 	takeMap  map[int]bool | ||||||
|  |  | ||||||
|  | 	val  interface{} | ||||||
|  | 	s    fmt.State | ||||||
|  | 	opts *Options | ||||||
|  |  | ||||||
|  | 	depth            int | ||||||
|  | 	ignoreNextType   bool | ||||||
|  | 	protoWrapperType bool | ||||||
|  | 	sqlWrapperType   bool | ||||||
| } | } | ||||||
|  |  | ||||||
| // Options struct | // Options struct | ||||||
| @@ -106,14 +116,14 @@ func Tagged(b bool) Option { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func Unwrap(val interface{}, opts ...Option) *unwrap { | func Unwrap(val interface{}, opts ...Option) *Wrapper { | ||||||
| 	options := NewOptions(opts...) | 	options := NewOptions(opts...) | ||||||
| 	return &unwrap{val: val, opts: &options, pointers: make(map[uintptr]int)} | 	return &Wrapper{val: val, opts: &options, pointers: make(map[uintptr]int), takeMap: make(map[int]bool)} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (f *unwrap) unpackValue(v reflect.Value) reflect.Value { | func (w *Wrapper) unpackValue(v reflect.Value) reflect.Value { | ||||||
| 	if v.Kind() == reflect.Interface { | 	if v.Kind() == reflect.Interface { | ||||||
| 		f.ignoreNextType = false | 		w.ignoreNextType = false | ||||||
| 		if !v.IsNil() { | 		if !v.IsNil() { | ||||||
| 			v = v.Elem() | 			v = v.Elem() | ||||||
| 		} | 		} | ||||||
| @@ -122,19 +132,19 @@ func (f *unwrap) unpackValue(v reflect.Value) reflect.Value { | |||||||
| } | } | ||||||
|  |  | ||||||
| // formatPtr handles formatting of pointers by indirecting them as necessary. | // formatPtr handles formatting of pointers by indirecting them as necessary. | ||||||
| func (f *unwrap) formatPtr(v reflect.Value) { | func (w *Wrapper) formatPtr(v reflect.Value) { | ||||||
| 	// Display nil if top level pointer is nil. | 	// Display nil if top level pointer is nil. | ||||||
| 	showTypes := f.s.Flag('#') | 	showTypes := w.s.Flag('#') | ||||||
| 	if v.IsNil() && (!showTypes || f.ignoreNextType) { | 	if v.IsNil() && (!showTypes || w.ignoreNextType) { | ||||||
| 		_, _ = f.s.Write(nilAngleBytes) | 		_, _ = w.s.Write(nilAngleBytes) | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// Remove pointers at or below the current depth from map used to detect | 	// Remove pointers at or below the current depth from map used to detect | ||||||
| 	// circular refs. | 	// circular refs. | ||||||
| 	for k, depth := range f.pointers { | 	for k, depth := range w.pointers { | ||||||
| 		if depth >= f.depth { | 		if depth >= w.depth { | ||||||
| 			delete(f.pointers, k) | 			delete(w.pointers, k) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -156,12 +166,12 @@ func (f *unwrap) formatPtr(v reflect.Value) { | |||||||
| 		indirects++ | 		indirects++ | ||||||
| 		addr := ve.Pointer() | 		addr := ve.Pointer() | ||||||
| 		pointerChain = append(pointerChain, addr) | 		pointerChain = append(pointerChain, addr) | ||||||
| 		if pd, ok := f.pointers[addr]; ok && pd < f.depth { | 		if pd, ok := w.pointers[addr]; ok && pd < w.depth { | ||||||
| 			cycleFound = true | 			cycleFound = true | ||||||
| 			indirects-- | 			indirects-- | ||||||
| 			break | 			break | ||||||
| 		} | 		} | ||||||
| 		f.pointers[addr] = f.depth | 		w.pointers[addr] = w.depth | ||||||
|  |  | ||||||
| 		ve = ve.Elem() | 		ve = ve.Elem() | ||||||
| 		if ve.Kind() == reflect.Interface { | 		if ve.Kind() == reflect.Interface { | ||||||
| @@ -174,49 +184,49 @@ func (f *unwrap) formatPtr(v reflect.Value) { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// Display type or indirection level depending on flags. | 	// Display type or indirection level depending on flags. | ||||||
| 	if showTypes && !f.ignoreNextType { | 	if showTypes && !w.ignoreNextType { | ||||||
| 		if f.depth > 0 { | 		if w.depth > 0 { | ||||||
| 			_, _ = f.s.Write(openParenBytes) | 			_, _ = w.s.Write(openParenBytes) | ||||||
| 		} | 		} | ||||||
| 		if f.depth > 0 { | 		if w.depth > 0 { | ||||||
| 			_, _ = f.s.Write(bytes.Repeat(asteriskBytes, indirects)) | 			_, _ = w.s.Write(bytes.Repeat(asteriskBytes, indirects)) | ||||||
| 		} else { | 		} else { | ||||||
| 			_, _ = f.s.Write(bytes.Repeat(ampBytes, indirects)) | 			_, _ = w.s.Write(bytes.Repeat(ampBytes, indirects)) | ||||||
| 		} | 		} | ||||||
| 		_, _ = f.s.Write([]byte(ve.Type().String())) | 		_, _ = w.s.Write([]byte(ve.Type().String())) | ||||||
| 		if f.depth > 0 { | 		if w.depth > 0 { | ||||||
| 			_, _ = f.s.Write(closeParenBytes) | 			_, _ = w.s.Write(closeParenBytes) | ||||||
| 		} | 		} | ||||||
| 	} else { | 	} else { | ||||||
| 		if nilFound || cycleFound { | 		if nilFound || cycleFound { | ||||||
| 			indirects += strings.Count(ve.Type().String(), "*") | 			indirects += strings.Count(ve.Type().String(), "*") | ||||||
| 		} | 		} | ||||||
| 		_, _ = f.s.Write(openAngleBytes) | 		_, _ = w.s.Write(openAngleBytes) | ||||||
| 		_, _ = f.s.Write([]byte(strings.Repeat("*", indirects))) | 		_, _ = w.s.Write([]byte(strings.Repeat("*", indirects))) | ||||||
| 		_, _ = f.s.Write(closeAngleBytes) | 		_, _ = w.s.Write(closeAngleBytes) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// Display pointer information depending on flags. | 	// Display pointer information depending on flags. | ||||||
| 	if f.s.Flag('+') && (len(pointerChain) > 0) { | 	if w.s.Flag('+') && (len(pointerChain) > 0) { | ||||||
| 		_, _ = f.s.Write(openParenBytes) | 		_, _ = w.s.Write(openParenBytes) | ||||||
| 		for i, addr := range pointerChain { | 		for i, addr := range pointerChain { | ||||||
| 			if i > 0 { | 			if i > 0 { | ||||||
| 				_, _ = f.s.Write(pointerChainBytes) | 				_, _ = w.s.Write(pointerChainBytes) | ||||||
| 			} | 			} | ||||||
| 			getHexPtr(f.s, addr) | 			getHexPtr(w.s, addr) | ||||||
| 		} | 		} | ||||||
| 		_, _ = f.s.Write(closeParenBytes) | 		_, _ = w.s.Write(closeParenBytes) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// Display dereferenced value. | 	// Display dereferenced value. | ||||||
| 	switch { | 	switch { | ||||||
| 	case nilFound: | 	case nilFound: | ||||||
| 		_, _ = f.s.Write(nilAngleBytes) | 		_, _ = w.s.Write(nilAngleBytes) | ||||||
| 	case cycleFound: | 	case cycleFound: | ||||||
| 		_, _ = f.s.Write(circularShortBytes) | 		_, _ = w.s.Write(circularShortBytes) | ||||||
| 	default: | 	default: | ||||||
| 		f.ignoreNextType = true | 		w.ignoreNextType = true | ||||||
| 		f.format(ve) | 		w.format(ve) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -224,54 +234,73 @@ func (f *unwrap) formatPtr(v reflect.Value) { | |||||||
| // uses the passed reflect value to figure out what kind of object we are | // uses the passed reflect value to figure out what kind of object we are | ||||||
| // dealing with and formats it appropriately.  It is a recursive function, | // dealing with and formats it appropriately.  It is a recursive function, | ||||||
| // however circular data structures are detected and handled properly. | // however circular data structures are detected and handled properly. | ||||||
| func (f *unwrap) format(v reflect.Value) { | func (w *Wrapper) format(v reflect.Value) { | ||||||
| 	if f.opts.Codec != nil { | 	if w.opts.Codec != nil { | ||||||
| 		buf, err := f.opts.Codec.Marshal(v.Interface()) | 		buf, err := w.opts.Codec.Marshal(v.Interface()) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			_, _ = f.s.Write(invalidAngleBytes) | 			_, _ = w.s.Write(invalidAngleBytes) | ||||||
| 			return | 			return | ||||||
| 		} | 		} | ||||||
| 		_, _ = f.s.Write(buf) | 		_, _ = w.s.Write(buf) | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// Handle invalid reflect values immediately. | 	// Handle invalid reflect values immediately. | ||||||
| 	kind := v.Kind() | 	kind := v.Kind() | ||||||
| 	if kind == reflect.Invalid { | 	if kind == reflect.Invalid { | ||||||
| 		_, _ = f.s.Write(invalidAngleBytes) | 		_, _ = w.s.Write(invalidAngleBytes) | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// Handle pointers specially. | 	// Handle pointers specially. | ||||||
| 	if kind == reflect.Ptr { | 	switch kind { | ||||||
| 		f.formatPtr(v) | 	case reflect.Ptr: | ||||||
|  | 		if !v.IsZero() { | ||||||
|  | 			if strings.HasPrefix(reflect.Indirect(v).Type().String(), "wrapperspb.") { | ||||||
|  | 				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) | ||||||
| 		return | 		return | ||||||
|  | 	case reflect.Struct: | ||||||
|  | 		if !v.IsZero() { | ||||||
|  | 			if strings.HasPrefix(reflect.Indirect(v).Type().String(), "sql.Null") { | ||||||
|  | 				w.sqlWrapperType = true | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// get type information unless already handled elsewhere. | 	// get type information unless already handled elsewhere. | ||||||
| 	if !f.ignoreNextType && f.s.Flag('#') { | 	if !w.ignoreNextType && w.s.Flag('#') { | ||||||
| 		if v.Type().Kind() != reflect.Map && | 		if v.Type().Kind() != reflect.Map && | ||||||
| 			v.Type().Kind() != reflect.String && | 			v.Type().Kind() != reflect.String && | ||||||
| 			v.Type().Kind() != reflect.Array && | 			v.Type().Kind() != reflect.Array && | ||||||
| 			v.Type().Kind() != reflect.Slice { | 			v.Type().Kind() != reflect.Slice { | ||||||
| 			_, _ = f.s.Write(openParenBytes) | 			_, _ = w.s.Write(openParenBytes) | ||||||
| 		} | 		} | ||||||
| 		if v.Kind() != reflect.String { | 		if v.Kind() != reflect.String { | ||||||
| 			_, _ = f.s.Write([]byte(v.Type().String())) | 			_, _ = w.s.Write([]byte(v.Type().String())) | ||||||
| 		} | 		} | ||||||
| 		if v.Type().Kind() != reflect.Map && | 		if v.Type().Kind() != reflect.Map && | ||||||
| 			v.Type().Kind() != reflect.String && | 			v.Type().Kind() != reflect.String && | ||||||
| 			v.Type().Kind() != reflect.Array && | 			v.Type().Kind() != reflect.Array && | ||||||
| 			v.Type().Kind() != reflect.Slice { | 			v.Type().Kind() != reflect.Slice { | ||||||
| 			_, _ = f.s.Write(closeParenBytes) | 			_, _ = w.s.Write(closeParenBytes) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 	f.ignoreNextType = false | 	w.ignoreNextType = false | ||||||
|  |  | ||||||
| 	// Call Stringer/error interfaces if they exist and the handle methods | 	// Call Stringer/error interfaces if they exist and the handle methods | ||||||
| 	// flag is enabled. | 	// flag is enabled. | ||||||
| 	if f.opts.Methods { | 	if w.opts.Methods { | ||||||
| 		if (kind != reflect.Invalid) && (kind != reflect.Interface) { | 		if (kind != reflect.Invalid) && (kind != reflect.Interface) { | ||||||
| 			if handled := handleMethods(f.opts, f.s, v); handled { | 			if handled := handleMethods(w.opts, w.s, v); handled { | ||||||
| 				return | 				return | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
| @@ -279,48 +308,48 @@ func (f *unwrap) format(v reflect.Value) { | |||||||
|  |  | ||||||
| 	switch kind { | 	switch kind { | ||||||
| 	case reflect.Invalid: | 	case reflect.Invalid: | ||||||
| 		_, _ = f.s.Write(invalidAngleBytes) | 		_, _ = w.s.Write(invalidAngleBytes) | ||||||
| 	case reflect.Bool: | 	case reflect.Bool: | ||||||
| 		getBool(f.s, v.Bool()) | 		getBool(w.s, v.Bool()) | ||||||
| 	case reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64, reflect.Int: | 	case reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64, reflect.Int: | ||||||
| 		getInt(f.s, v.Int(), 10) | 		getInt(w.s, v.Int(), 10) | ||||||
| 	case reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uint: | 	case reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uint: | ||||||
| 		getUint(f.s, v.Uint(), 10) | 		getUint(w.s, v.Uint(), 10) | ||||||
| 	case reflect.Float32: | 	case reflect.Float32: | ||||||
| 		getFloat(f.s, v.Float(), 32) | 		getFloat(w.s, v.Float(), 32) | ||||||
| 	case reflect.Float64: | 	case reflect.Float64: | ||||||
| 		getFloat(f.s, v.Float(), 64) | 		getFloat(w.s, v.Float(), 64) | ||||||
| 	case reflect.Complex64: | 	case reflect.Complex64: | ||||||
| 		getComplex(f.s, v.Complex(), 32) | 		getComplex(w.s, v.Complex(), 32) | ||||||
| 	case reflect.Complex128: | 	case reflect.Complex128: | ||||||
| 		getComplex(f.s, v.Complex(), 64) | 		getComplex(w.s, v.Complex(), 64) | ||||||
| 	case reflect.Slice: | 	case reflect.Slice: | ||||||
| 		if v.IsNil() { | 		if v.IsNil() { | ||||||
| 			_, _ = f.s.Write(nilAngleBytes) | 			_, _ = w.s.Write(nilAngleBytes) | ||||||
| 			break | 			break | ||||||
| 		} | 		} | ||||||
| 		fallthrough | 		fallthrough | ||||||
| 	case reflect.Array: | 	case reflect.Array: | ||||||
| 		_, _ = f.s.Write(openBraceBytes) | 		_, _ = w.s.Write(openBraceBytes) | ||||||
| 		f.depth++ | 		w.depth++ | ||||||
| 		numEntries := v.Len() | 		numEntries := v.Len() | ||||||
| 		for i := 0; i < numEntries; i++ { | 		for i := 0; i < numEntries; i++ { | ||||||
| 			if i > 0 { | 			if i > 0 { | ||||||
| 				_, _ = f.s.Write(commaBytes) | 				_, _ = w.s.Write(commaBytes) | ||||||
| 				_, _ = f.s.Write(spaceBytes) | 				_, _ = w.s.Write(spaceBytes) | ||||||
| 			} | 			} | ||||||
| 			f.ignoreNextType = true | 			w.ignoreNextType = true | ||||||
| 			f.format(f.unpackValue(v.Index(i))) | 			w.format(w.unpackValue(v.Index(i))) | ||||||
| 		} | 		} | ||||||
| 		f.depth-- | 		w.depth-- | ||||||
| 		_, _ = f.s.Write(closeBraceBytes) | 		_, _ = w.s.Write(closeBraceBytes) | ||||||
| 	case reflect.String: | 	case reflect.String: | ||||||
| 		_, _ = f.s.Write([]byte(`"` + v.String() + `"`)) | 		_, _ = w.s.Write([]byte(`"` + v.String() + `"`)) | ||||||
| 	case reflect.Interface: | 	case reflect.Interface: | ||||||
| 		// The only time we should get here is for nil interfaces due to | 		// The only time we should get here is for nil interfaces due to | ||||||
| 		// unpackValue calls. | 		// unpackValue calls. | ||||||
| 		if v.IsNil() { | 		if v.IsNil() { | ||||||
| 			_, _ = f.s.Write(nilAngleBytes) | 			_, _ = w.s.Write(nilAngleBytes) | ||||||
| 		} | 		} | ||||||
| 	case reflect.Ptr: | 	case reflect.Ptr: | ||||||
| 		// Do nothing.  We should never get here since pointers have already | 		// Do nothing.  We should never get here since pointers have already | ||||||
| @@ -328,34 +357,54 @@ func (f *unwrap) format(v reflect.Value) { | |||||||
| 	case reflect.Map: | 	case reflect.Map: | ||||||
| 		// nil maps should be indicated as different than empty maps | 		// nil maps should be indicated as different than empty maps | ||||||
| 		if v.IsNil() { | 		if v.IsNil() { | ||||||
| 			_, _ = f.s.Write(nilAngleBytes) | 			_, _ = w.s.Write(nilAngleBytes) | ||||||
| 			break | 			break | ||||||
| 		} | 		} | ||||||
| 		_, _ = f.s.Write(openMapBytes) | 		_, _ = w.s.Write(openMapBytes) | ||||||
| 		f.depth++ | 		w.depth++ | ||||||
| 		keys := v.MapKeys() | 		keys := v.MapKeys() | ||||||
| 		for i, key := range keys { | 		for i, key := range keys { | ||||||
| 			if i > 0 { | 			if i > 0 { | ||||||
| 				_, _ = f.s.Write(spaceBytes) | 				_, _ = w.s.Write(spaceBytes) | ||||||
| 			} | 			} | ||||||
| 			f.ignoreNextType = true | 			w.ignoreNextType = true | ||||||
| 			f.format(f.unpackValue(key)) | 			w.format(w.unpackValue(key)) | ||||||
| 			_, _ = f.s.Write(colonBytes) | 			_, _ = w.s.Write(colonBytes) | ||||||
| 			f.ignoreNextType = true | 			w.ignoreNextType = true | ||||||
| 			f.format(f.unpackValue(v.MapIndex(key))) | 			w.format(w.unpackValue(v.MapIndex(key))) | ||||||
| 		} | 		} | ||||||
| 		f.depth-- | 		w.depth-- | ||||||
| 		_, _ = f.s.Write(closeMapBytes) | 		_, _ = w.s.Write(closeMapBytes) | ||||||
| 	case reflect.Struct: | 	case reflect.Struct: | ||||||
|  |  | ||||||
| 		numFields := v.NumField() | 		numFields := v.NumField() | ||||||
| 		numWritten := 0 | 		numWritten := 0 | ||||||
| 		_, _ = f.s.Write(openBraceBytes) | 		_, _ = w.s.Write(openBraceBytes) | ||||||
| 		f.depth++ | 		w.depth++ | ||||||
|  |  | ||||||
| 		vt := v.Type() | 		vt := v.Type() | ||||||
| 		prevSkip := false | 		prevSkip := false | ||||||
|  |  | ||||||
| 		for i := 0; i < numFields; i++ { | 		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 | ||||||
|  | 			} else if w.sqlWrapperType && vt.Field(i).Name == "Valid" { | ||||||
|  | 				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") | 			sv, ok := vt.Field(i).Tag.Lookup("logger") | ||||||
| 			if ok { | 			switch { | ||||||
|  | 			case ok: | ||||||
| 				switch sv { | 				switch sv { | ||||||
| 				case "omit": | 				case "omit": | ||||||
| 					prevSkip = true | 					prevSkip = true | ||||||
| @@ -363,57 +412,68 @@ func (f *unwrap) format(v reflect.Value) { | |||||||
| 				case "take": | 				case "take": | ||||||
| 					break | 					break | ||||||
| 				} | 				} | ||||||
| 			} else if f.opts.Tagged { | 			case !ok && w.opts.Tagged: | ||||||
| 				prevSkip = true | 				// skip top level untagged | ||||||
| 				continue | 				if w.depth == 1 { | ||||||
| 			} | 					prevSkip = true | ||||||
| 			if i > 0 && !prevSkip { | 					continue | ||||||
| 				_, _ = f.s.Write(commaBytes) | 				} | ||||||
| 				_, _ = f.s.Write(spaceBytes) | 				if tv, ok := w.takeMap[w.depth]; ok && !tv { | ||||||
|  | 					prevSkip = true | ||||||
|  | 					continue | ||||||
|  | 				} | ||||||
| 			} | 			} | ||||||
|  |  | ||||||
| 			if prevSkip { | 			if prevSkip { | ||||||
| 				prevSkip = false | 				prevSkip = false | ||||||
| 			} | 			} | ||||||
| 			vtf := vt.Field(i) |  | ||||||
| 			if f.s.Flag('+') || f.s.Flag('#') { | 			if numWritten > 0 { | ||||||
| 				_, _ = f.s.Write([]byte(vtf.Name)) | 				_, _ = w.s.Write(commaBytes) | ||||||
| 				_, _ = f.s.Write(colonBytes) | 				_, _ = w.s.Write(spaceBytes) | ||||||
| 			} | 			} | ||||||
| 			f.format(f.unpackValue(v.Field(i))) |  | ||||||
|  | 			vt := vt.Field(i) | ||||||
|  | 			if w.s.Flag('+') || w.s.Flag('#') { | ||||||
|  | 				_, _ = w.s.Write([]byte(vt.Name)) | ||||||
|  | 				_, _ = w.s.Write(colonBytes) | ||||||
|  | 			} | ||||||
|  | 			w.format(w.unpackValue(v.Field(i))) | ||||||
| 			numWritten++ | 			numWritten++ | ||||||
| 		} | 		} | ||||||
| 		f.depth-- | 		w.depth-- | ||||||
| 		if numWritten == 0 && f.depth < 0 { |  | ||||||
| 			_, _ = f.s.Write(filteredBytes) | 		if numWritten == 0 && w.depth < 0 { | ||||||
|  | 			_, _ = w.s.Write(filteredBytes) | ||||||
| 		} | 		} | ||||||
| 		_, _ = f.s.Write(closeBraceBytes) | 		_, _ = w.s.Write(closeBraceBytes) | ||||||
| 	case reflect.Uintptr: | 	case reflect.Uintptr: | ||||||
| 		getHexPtr(f.s, uintptr(v.Uint())) | 		getHexPtr(w.s, uintptr(v.Uint())) | ||||||
| 	case reflect.UnsafePointer, reflect.Chan, reflect.Func: | 	case reflect.UnsafePointer, reflect.Chan, reflect.Func: | ||||||
| 		getHexPtr(f.s, v.Pointer()) | 		getHexPtr(w.s, v.Pointer()) | ||||||
| 	// There were not any other types at the time this code was written, but | 	// There were not any other types at the time this code was written, but | ||||||
| 	// fall back to letting the default fmt package handle it if any get added. | 	// fall back to letting the default fmt package handle it if any get added. | ||||||
| 	default: | 	default: | ||||||
| 		format := f.buildDefaultFormat() | 		format := w.buildDefaultFormat() | ||||||
| 		if v.CanInterface() { | 		if v.CanInterface() { | ||||||
| 			_, _ = fmt.Fprintf(f.s, format, v.Interface()) | 			_, _ = fmt.Fprintf(w.s, format, v.Interface()) | ||||||
| 		} else { | 		} else { | ||||||
| 			_, _ = fmt.Fprintf(f.s, format, v.String()) | 			_, _ = fmt.Fprintf(w.s, format, v.String()) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (f *unwrap) Format(s fmt.State, verb rune) { | func (w *Wrapper) Format(s fmt.State, verb rune) { | ||||||
| 	f.s = s | 	w.s = s | ||||||
|  |  | ||||||
| 	// Use standard formatting for verbs that are not v. | 	// Use standard formatting for verbs that are not v. | ||||||
| 	if verb != 'v' { | 	if verb != 'v' { | ||||||
| 		format := f.constructOrigFormat(verb) | 		format := w.constructOrigFormat(verb) | ||||||
| 		_, _ = fmt.Fprintf(s, format, f.val) | 		_, _ = fmt.Fprintf(s, format, w.val) | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if f.val == nil { | 	if w.val == nil { | ||||||
| 		if s.Flag('#') { | 		if s.Flag('#') { | ||||||
| 			_, _ = s.Write(interfaceBytes) | 			_, _ = s.Write(interfaceBytes) | ||||||
| 		} | 		} | ||||||
| @@ -421,7 +481,11 @@ func (f *unwrap) Format(s fmt.State, verb rune) { | |||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	f.format(reflect.ValueOf(f.val)) | 	if w.opts.Tagged { | ||||||
|  | 		w.buildTakeMap(reflect.ValueOf(w.val), 1) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	w.format(reflect.ValueOf(w.val)) | ||||||
| } | } | ||||||
|  |  | ||||||
| // handle special methods like error.Error() or fmt.Stringer interface | // handle special methods like error.Error() or fmt.Stringer interface | ||||||
| @@ -537,11 +601,11 @@ func catchPanic(w io.Writer, _ reflect.Value) { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (f *unwrap) buildDefaultFormat() (format string) { | func (w *Wrapper) buildDefaultFormat() (format string) { | ||||||
| 	buf := bytes.NewBuffer(percentBytes) | 	buf := bytes.NewBuffer(percentBytes) | ||||||
|  |  | ||||||
| 	for _, flag := range sf { | 	for _, flag := range sf { | ||||||
| 		if f.s.Flag(int(flag)) { | 		if w.s.Flag(int(flag)) { | ||||||
| 			_, _ = buf.WriteRune(flag) | 			_, _ = buf.WriteRune(flag) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| @@ -552,26 +616,68 @@ func (f *unwrap) buildDefaultFormat() (format string) { | |||||||
| 	return format | 	return format | ||||||
| } | } | ||||||
|  |  | ||||||
| func (f *unwrap) constructOrigFormat(verb rune) (format string) { | func (w *Wrapper) constructOrigFormat(verb rune) string { | ||||||
| 	buf := bytes.NewBuffer(percentBytes) | 	buf := bytes.NewBuffer(percentBytes) | ||||||
|  |  | ||||||
| 	for _, flag := range sf { | 	for _, flag := range sf { | ||||||
| 		if f.s.Flag(int(flag)) { | 		if w.s.Flag(int(flag)) { | ||||||
| 			_, _ = buf.WriteRune(flag) | 			_, _ = buf.WriteRune(flag) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if width, ok := f.s.Width(); ok { | 	if width, ok := w.s.Width(); ok { | ||||||
| 		_, _ = buf.WriteString(strconv.Itoa(width)) | 		_, _ = buf.WriteString(strconv.Itoa(width)) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if precision, ok := f.s.Precision(); ok { | 	if precision, ok := w.s.Precision(); ok { | ||||||
| 		_, _ = buf.Write(precisionBytes) | 		_, _ = buf.Write(precisionBytes) | ||||||
| 		_, _ = buf.WriteString(strconv.Itoa(precision)) | 		_, _ = buf.WriteString(strconv.Itoa(precision)) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	_, _ = buf.WriteRune(verb) | 	_, _ = buf.WriteRune(verb) | ||||||
|  |  | ||||||
| 	format = buf.String() | 	return buf.String() | ||||||
| 	return format | } | ||||||
|  |  | ||||||
|  | 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.takeMap[depth] = true | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  | 	default: | ||||||
|  | 		w.takeMap[depth] = true | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	vt := v.Type() | ||||||
|  |  | ||||||
|  | 	for i := 0; i < v.NumField(); i++ { | ||||||
|  | 		sv, ok := vt.Field(i).Tag.Lookup("logger") | ||||||
|  | 		if ok && sv == "take" { | ||||||
|  | 			w.takeMap[depth] = false | ||||||
|  | 		} | ||||||
|  | 		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 | ||||||
|  | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -5,7 +5,7 @@ import ( | |||||||
| 	"strings" | 	"strings" | ||||||
| 	"testing" | 	"testing" | ||||||
|  |  | ||||||
| 	"go.unistack.org/micro/v3/codec" | 	"go.unistack.org/micro/v4/codec" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| func TestUnwrap(t *testing.T) { | func TestUnwrap(t *testing.T) { | ||||||
| @@ -82,16 +82,16 @@ func TestTagged(t *testing.T) { | |||||||
| func TestTaggedNested(t *testing.T) { | func TestTaggedNested(t *testing.T) { | ||||||
| 	type val struct { | 	type val struct { | ||||||
| 		key string `logger:"take"` | 		key string `logger:"take"` | ||||||
| 		val string `logger:"omit"` | 		// val string `logger:"omit"` | ||||||
| 		unk string | 		unk string | ||||||
| 	} | 	} | ||||||
| 	type str struct { | 	type str struct { | ||||||
| 		val *val   `logger:"take"` | 		// key string `logger:"omit"` | ||||||
| 		key string `logger:"omit"` | 		val *val `logger:"take"` | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	var iface interface{} | 	var iface interface{} | ||||||
| 	v := &str{key: "omit", val: &val{key: "test", val: "omit", unk: "unk"}} | 	v := &str{val: &val{key: "test", unk: "unk"}} | ||||||
| 	iface = v | 	iface = v | ||||||
| 	buf := fmt.Sprintf("%#v", Unwrap(iface, Tagged(true))) | 	buf := fmt.Sprintf("%#v", Unwrap(iface, Tagged(true))) | ||||||
| 	if strings.Compare(buf, `&unwrap.str{val:(*unwrap.val){key:"test"}}`) != 0 { | 	if strings.Compare(buf, `&unwrap.str{val:(*unwrap.val){key:"test"}}`) != 0 { | ||||||
|   | |||||||
| @@ -1,166 +0,0 @@ | |||||||
| package logger // import "go.unistack.org/micro/v3/logger/wrapper" |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"context" |  | ||||||
| 	"reflect" |  | ||||||
|  |  | ||||||
| 	rutil "go.unistack.org/micro/v3/util/reflect" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // LogFunc function used for Log method |  | ||||||
| type LogFunc func(ctx context.Context, level Level, args ...interface{}) |  | ||||||
|  |  | ||||||
| // LogfFunc function used for Logf method |  | ||||||
| type LogfFunc func(ctx context.Context, level Level, msg string, args ...interface{}) |  | ||||||
|  |  | ||||||
| type Wrapper interface { |  | ||||||
| 	// Log logs message with needed level |  | ||||||
| 	Log(LogFunc) LogFunc |  | ||||||
| 	// Logf logs message with needed level |  | ||||||
| 	Logf(LogfFunc) LogfFunc |  | ||||||
| } |  | ||||||
|  |  | ||||||
| var _ Logger = &omitLogger{} |  | ||||||
|  |  | ||||||
| type omitLogger struct { |  | ||||||
| 	l Logger |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func NewOmitLogger(l Logger) Logger { |  | ||||||
| 	return &omitLogger{l: l} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *omitLogger) Init(opts ...Option) error { |  | ||||||
| 	return w.l.Init(append(opts, WrapLogger(NewOmitWrapper()))...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *omitLogger) V(level Level) bool { |  | ||||||
| 	return w.l.V(level) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *omitLogger) Level(level Level) { |  | ||||||
| 	w.l.Level(level) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *omitLogger) Clone(opts ...Option) Logger { |  | ||||||
| 	return w.l.Clone(opts...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *omitLogger) Options() Options { |  | ||||||
| 	return w.l.Options() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *omitLogger) Fields(fields ...interface{}) Logger { |  | ||||||
| 	return w.l.Fields(fields...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *omitLogger) Info(ctx context.Context, args ...interface{}) { |  | ||||||
| 	w.l.Info(ctx, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *omitLogger) Trace(ctx context.Context, args ...interface{}) { |  | ||||||
| 	w.l.Trace(ctx, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *omitLogger) Debug(ctx context.Context, args ...interface{}) { |  | ||||||
| 	w.l.Debug(ctx, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *omitLogger) Warn(ctx context.Context, args ...interface{}) { |  | ||||||
| 	w.l.Warn(ctx, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *omitLogger) Error(ctx context.Context, args ...interface{}) { |  | ||||||
| 	w.l.Error(ctx, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *omitLogger) Fatal(ctx context.Context, args ...interface{}) { |  | ||||||
| 	w.l.Fatal(ctx, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *omitLogger) Infof(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	w.l.Infof(ctx, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *omitLogger) Tracef(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	w.l.Tracef(ctx, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *omitLogger) Debugf(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	w.l.Debugf(ctx, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *omitLogger) Warnf(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	w.l.Warnf(ctx, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *omitLogger) Errorf(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	w.l.Errorf(ctx, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *omitLogger) Fatalf(ctx context.Context, msg string, args ...interface{}) { |  | ||||||
| 	w.l.Fatalf(ctx, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *omitLogger) Log(ctx context.Context, level Level, args ...interface{}) { |  | ||||||
| 	w.l.Log(ctx, level, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *omitLogger) Logf(ctx context.Context, level Level, msg string, args ...interface{}) { |  | ||||||
| 	w.l.Logf(ctx, level, msg, args...) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *omitLogger) String() string { |  | ||||||
| 	return w.l.String() |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type omitWrapper struct{} |  | ||||||
|  |  | ||||||
| func NewOmitWrapper() Wrapper { |  | ||||||
| 	return &omitWrapper{} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func getArgs(args []interface{}) []interface{} { |  | ||||||
| 	nargs := make([]interface{}, 0, len(args)) |  | ||||||
| 	var err error |  | ||||||
| 	for _, arg := range args { |  | ||||||
| 		val := reflect.ValueOf(arg) |  | ||||||
| 		if val.Kind() == reflect.Ptr { |  | ||||||
| 			val = val.Elem() |  | ||||||
| 		} |  | ||||||
| 		narg := arg |  | ||||||
| 		if val.Kind() != reflect.Struct { |  | ||||||
| 			nargs = append(nargs, narg) |  | ||||||
| 			continue |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		if narg, err = rutil.Zero(arg); err != nil { |  | ||||||
| 			nargs = append(nargs, narg) |  | ||||||
| 			continue |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		rutil.CopyDefaults(narg, arg) |  | ||||||
| 		if flds, ferr := rutil.StructFields(narg); ferr == nil { |  | ||||||
| 			for _, fld := range flds { |  | ||||||
| 				if tv, ok := fld.Field.Tag.Lookup("logger"); ok && tv == "omit" { |  | ||||||
| 					fld.Value.Set(reflect.Zero(fld.Value.Type())) |  | ||||||
| 				} |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		nargs = append(nargs, narg) |  | ||||||
| 	} |  | ||||||
| 	return nargs |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *omitWrapper) Log(fn LogFunc) LogFunc { |  | ||||||
| 	return func(ctx context.Context, level Level, args ...interface{}) { |  | ||||||
| 		fn(ctx, level, getArgs(args)...) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func (w *omitWrapper) Logf(fn LogfFunc) LogfFunc { |  | ||||||
| 	return func(ctx context.Context, level Level, msg string, args ...interface{}) { |  | ||||||
| 		fn(ctx, level, msg, getArgs(args)...) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -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 |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| @@ -1,146 +0,0 @@ | |||||||
| // Package metadata is a way of defining message headers |  | ||||||
| package metadata |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"context" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| type ( |  | ||||||
| 	mdIncomingKey struct{} |  | ||||||
| 	mdOutgoingKey struct{} |  | ||||||
| 	mdKey         struct{} |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| // FromIncomingContext returns metadata from incoming ctx |  | ||||||
| // returned metadata shoud not be modified or race condition happens |  | ||||||
| func FromIncomingContext(ctx context.Context) (Metadata, bool) { |  | ||||||
| 	if ctx == nil { |  | ||||||
| 		return nil, false |  | ||||||
| 	} |  | ||||||
| 	md, ok := ctx.Value(mdIncomingKey{}).(*rawMetadata) |  | ||||||
| 	if !ok || md.md == nil { |  | ||||||
| 		return nil, false |  | ||||||
| 	} |  | ||||||
| 	return md.md, ok |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // FromOutgoingContext returns metadata from outgoing ctx |  | ||||||
| // returned metadata shoud not be modified or race condition happens |  | ||||||
| func FromOutgoingContext(ctx context.Context) (Metadata, bool) { |  | ||||||
| 	if ctx == nil { |  | ||||||
| 		return nil, false |  | ||||||
| 	} |  | ||||||
| 	md, ok := ctx.Value(mdOutgoingKey{}).(*rawMetadata) |  | ||||||
| 	if !ok || md.md == nil { |  | ||||||
| 		return nil, false |  | ||||||
| 	} |  | ||||||
| 	return md.md, ok |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // FromContext returns metadata from the given context |  | ||||||
| // returned metadata shoud not be modified or race condition happens |  | ||||||
| // |  | ||||||
| // Deprecated: use FromIncomingContext or FromOutgoingContext |  | ||||||
| func FromContext(ctx context.Context) (Metadata, bool) { |  | ||||||
| 	if ctx == nil { |  | ||||||
| 		return nil, false |  | ||||||
| 	} |  | ||||||
| 	md, ok := ctx.Value(mdKey{}).(*rawMetadata) |  | ||||||
| 	if !ok || md.md == nil { |  | ||||||
| 		return nil, false |  | ||||||
| 	} |  | ||||||
| 	return md.md, ok |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewContext creates a new context with the given metadata |  | ||||||
| // |  | ||||||
| // 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 |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // SetOutgoingContext modify outgoing context with given metadata |  | ||||||
| func SetOutgoingContext(ctx context.Context, md Metadata) bool { |  | ||||||
| 	if ctx == nil { |  | ||||||
| 		return false |  | ||||||
| 	} |  | ||||||
| 	if omd, ok := ctx.Value(mdOutgoingKey{}).(*rawMetadata); ok { |  | ||||||
| 		omd.md = md |  | ||||||
| 		return true |  | ||||||
| 	} |  | ||||||
| 	return false |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // SetIncomingContext modify incoming context with given metadata |  | ||||||
| func SetIncomingContext(ctx context.Context, md Metadata) bool { |  | ||||||
| 	if ctx == nil { |  | ||||||
| 		return false |  | ||||||
| 	} |  | ||||||
| 	if omd, ok := ctx.Value(mdIncomingKey{}).(*rawMetadata); ok { |  | ||||||
| 		omd.md = md |  | ||||||
| 		return true |  | ||||||
| 	} |  | ||||||
| 	return false |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewIncomingContext creates a new context with incoming metadata attached |  | ||||||
| func NewIncomingContext(ctx context.Context, md Metadata) context.Context { |  | ||||||
| 	if ctx == nil { |  | ||||||
| 		ctx = context.Background() |  | ||||||
| 	} |  | ||||||
| 	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 |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewOutgoingContext creates a new context with outcoming metadata attached |  | ||||||
| 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 |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // AppendOutgoingContext apends new md to context |  | ||||||
| func AppendOutgoingContext(ctx context.Context, kv ...string) context.Context { |  | ||||||
| 	md, ok := Pairs(kv...) |  | ||||||
| 	if !ok { |  | ||||||
| 		return ctx |  | ||||||
| 	} |  | ||||||
| 	omd, ok := FromOutgoingContext(ctx) |  | ||||||
| 	if !ok { |  | ||||||
| 		return NewOutgoingContext(ctx, md) |  | ||||||
| 	} |  | ||||||
| 	for k, v := range md { |  | ||||||
| 		omd.Set(k, v) |  | ||||||
| 	} |  | ||||||
| 	return NewOutgoingContext(ctx, omd) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // AppendIncomingContext apends new md to context |  | ||||||
| func AppendIncomingContext(ctx context.Context, kv ...string) context.Context { |  | ||||||
| 	md, ok := Pairs(kv...) |  | ||||||
| 	if !ok { |  | ||||||
| 		return ctx |  | ||||||
| 	} |  | ||||||
| 	omd, ok := FromIncomingContext(ctx) |  | ||||||
| 	if !ok { |  | ||||||
| 		return NewIncomingContext(ctx, md) |  | ||||||
| 	} |  | ||||||
| 	for k, v := range md { |  | ||||||
| 		omd.Set(k, v) |  | ||||||
| 	} |  | ||||||
| 	return NewIncomingContext(ctx, omd) |  | ||||||
| } |  | ||||||
| @@ -1,140 +0,0 @@ | |||||||
| package metadata |  | ||||||
|  |  | ||||||
| import ( |  | ||||||
| 	"context" |  | ||||||
| 	"testing" |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| func TestFromNilContext(t *testing.T) { |  | ||||||
| 	// nolint: staticcheck |  | ||||||
| 	c, ok := FromContext(nil) |  | ||||||
| 	if ok || c != nil { |  | ||||||
| 		t.Fatal("FromContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestNewNilContext(t *testing.T) { |  | ||||||
| 	// nolint: staticcheck |  | ||||||
| 	ctx := NewContext(nil, New(0)) |  | ||||||
|  |  | ||||||
| 	c, ok := FromContext(ctx) |  | ||||||
| 	if c == nil || !ok { |  | ||||||
| 		t.Fatal("NewContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestFromContext(t *testing.T) { |  | ||||||
| 	ctx := context.WithValue(context.TODO(), mdKey{}, &rawMetadata{New(0)}) |  | ||||||
|  |  | ||||||
| 	c, ok := FromContext(ctx) |  | ||||||
| 	if c == nil || !ok { |  | ||||||
| 		t.Fatal("FromContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestNewContext(t *testing.T) { |  | ||||||
| 	ctx := NewContext(context.TODO(), New(0)) |  | ||||||
|  |  | ||||||
| 	c, ok := FromContext(ctx) |  | ||||||
| 	if c == nil || !ok { |  | ||||||
| 		t.Fatal("NewContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestFromIncomingContext(t *testing.T) { |  | ||||||
| 	ctx := context.WithValue(context.TODO(), mdIncomingKey{}, &rawMetadata{New(0)}) |  | ||||||
|  |  | ||||||
| 	c, ok := FromIncomingContext(ctx) |  | ||||||
| 	if c == nil || !ok { |  | ||||||
| 		t.Fatal("FromIncomingContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestFromOutgoingContext(t *testing.T) { |  | ||||||
| 	ctx := context.WithValue(context.TODO(), mdOutgoingKey{}, &rawMetadata{New(0)}) |  | ||||||
|  |  | ||||||
| 	c, ok := FromOutgoingContext(ctx) |  | ||||||
| 	if c == nil || !ok { |  | ||||||
| 		t.Fatal("FromOutgoingContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestSetIncomingContext(t *testing.T) { |  | ||||||
| 	md := New(1) |  | ||||||
| 	md.Set("key", "val") |  | ||||||
| 	ctx := context.WithValue(context.TODO(), mdIncomingKey{}, &rawMetadata{}) |  | ||||||
| 	if !SetIncomingContext(ctx, md) { |  | ||||||
| 		t.Fatal("SetIncomingContext not works") |  | ||||||
| 	} |  | ||||||
| 	md, ok := FromIncomingContext(ctx) |  | ||||||
| 	if md == nil || !ok { |  | ||||||
| 		t.Fatal("SetIncomingContext not works") |  | ||||||
| 	} else if v, ok := md.Get("key"); !ok || v != "val" { |  | ||||||
| 		t.Fatal("SetIncomingContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestSetOutgoingContext(t *testing.T) { |  | ||||||
| 	md := New(1) |  | ||||||
| 	md.Set("key", "val") |  | ||||||
| 	ctx := context.WithValue(context.TODO(), mdOutgoingKey{}, &rawMetadata{}) |  | ||||||
| 	if !SetOutgoingContext(ctx, md) { |  | ||||||
| 		t.Fatal("SetOutgoingContext not works") |  | ||||||
| 	} |  | ||||||
| 	md, ok := FromOutgoingContext(ctx) |  | ||||||
| 	if md == nil || !ok { |  | ||||||
| 		t.Fatal("SetOutgoingContext not works") |  | ||||||
| 	} else if v, ok := md.Get("key"); !ok || v != "val" { |  | ||||||
| 		t.Fatal("SetOutgoingContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestNewIncomingContext(t *testing.T) { |  | ||||||
| 	md := New(1) |  | ||||||
| 	md.Set("key", "val") |  | ||||||
| 	ctx := NewIncomingContext(context.TODO(), md) |  | ||||||
|  |  | ||||||
| 	c, ok := FromIncomingContext(ctx) |  | ||||||
| 	if c == nil || !ok { |  | ||||||
| 		t.Fatal("NewIncomingContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestNewOutgoingContext(t *testing.T) { |  | ||||||
| 	md := New(1) |  | ||||||
| 	md.Set("key", "val") |  | ||||||
| 	ctx := NewOutgoingContext(context.TODO(), md) |  | ||||||
|  |  | ||||||
| 	c, ok := FromOutgoingContext(ctx) |  | ||||||
| 	if c == nil || !ok { |  | ||||||
| 		t.Fatal("NewOutgoingContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestAppendIncomingContext(t *testing.T) { |  | ||||||
| 	md := New(1) |  | ||||||
| 	md.Set("key1", "val1") |  | ||||||
| 	ctx := AppendIncomingContext(context.TODO(), "key2", "val2") |  | ||||||
|  |  | ||||||
| 	nmd, ok := FromIncomingContext(ctx) |  | ||||||
| 	if nmd == nil || !ok { |  | ||||||
| 		t.Fatal("AppendIncomingContext not works") |  | ||||||
| 	} |  | ||||||
| 	if v, ok := nmd.Get("key2"); !ok || v != "val2" { |  | ||||||
| 		t.Fatal("AppendIncomingContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func TestAppendOutgoingContext(t *testing.T) { |  | ||||||
| 	md := New(1) |  | ||||||
| 	md.Set("key1", "val1") |  | ||||||
| 	ctx := AppendOutgoingContext(context.TODO(), "key2", "val2") |  | ||||||
|  |  | ||||||
| 	nmd, ok := FromOutgoingContext(ctx) |  | ||||||
| 	if nmd == nil || !ok { |  | ||||||
| 		t.Fatal("AppendOutgoingContext not works") |  | ||||||
| 	} |  | ||||||
| 	if v, ok := nmd.Get("key2"); !ok || v != "val2" { |  | ||||||
| 		t.Fatal("AppendOutgoingContext not works") |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
							
								
								
									
										19
									
								
								metadata/headers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										19
									
								
								metadata/headers.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,19 @@ | |||||||
|  | // Package metadata is a way of defining message headers | ||||||
|  | package metadata | ||||||
|  |  | ||||||
|  | var ( | ||||||
|  | 	// HeaderTopic is the header name that contains topic name | ||||||
|  | 	HeaderTopic = "Micro-Topic" | ||||||
|  | 	// HeaderContentType specifies content type of message | ||||||
|  | 	HeaderContentType = "Content-Type" | ||||||
|  | 	// HeaderEndpoint specifies endpoint in service | ||||||
|  | 	HeaderEndpoint = "Micro-Endpoint" | ||||||
|  | 	// HeaderService specifies service | ||||||
|  | 	HeaderService = "Micro-Service" | ||||||
|  | 	// HeaderTimeout specifies timeout of operation | ||||||
|  | 	HeaderTimeout = "Micro-Timeout" | ||||||
|  | 	// HeaderAuthorization specifies Authorization header | ||||||
|  | 	HeaderAuthorization = "Authorization" | ||||||
|  | 	// HeaderXRequestID specifies request id | ||||||
|  | 	HeaderXRequestID = "X-Request-Id" | ||||||
|  | ) | ||||||
Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user