Compare commits
	
		
			201 Commits
		
	
	
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| c757127453 | |||
| 
						 | 
					b2b24e0a9a | ||
| ea64ce0eab | |||
| b89246b18c | |||
| a9829b5da0 | |||
| a770c498e7 | |||
| fa5602486a | |||
| d5690ef2b7 | |||
| 134031e18b | |||
| 9743a7f9b1 | |||
| 2225b95e3e | |||
| b76b8d3114 | |||
| b92b2541e0 | |||
| 9d955fc079 | |||
| 1be75ea264 | |||
| d4a07099a2 | |||
| a648bf77f4 | |||
| 0eb663fffe | |||
| 
						 | 
					fd803b1fe0 | ||
| 
						 | 
					4c2178305d | ||
| 9149aeb3de | |||
| dcadd64941 | |||
| 8207d4154f | |||
| d7524cbe01 | |||
| 0e7b8e73a8 | |||
| 35146aa717 | |||
| 3f44a41d30 | |||
| 090100e522 | |||
| 0bf8053d72 | |||
| 42bdd79b3d | |||
| 5fa952c86a | |||
| 
						 | 
					d861c31a05 | ||
| e7dc59157d | |||
| 
						 | 
					7f7081e6e8 | ||
| d441f151e0 | |||
| 
						 | 
					6a6be26ef1 | ||
| d1d5e0f4ac | |||
| 
						 | 
					3f8976ee62 | ||
| 2078f2e700 | |||
| 
						 | 
					007d45fc7b | ||
| 6c84752ccd | |||
| 
						 | 
					54d21ba6e1 | ||
| 5b1f1df17a | |||
| 
						 | 
					11d7ee54fd | ||
| 41d606b4e8 | |||
| 
						 | 
					5f6f06c239 | ||
| 
						 | 
					f80bb95119 | ||
| 
						 | 
					cd23f93d57 | ||
| f625d5c958 | |||
| 
						 | 
					8b068fb7ab | ||
| 3a1e9ec707 | |||
| 
						 | 
					4d6a98782c | ||
| af838ec6e5 | |||
| 
						 | 
					0c90ea28d9 | ||
| 90e327726b | |||
| 
						 | 
					6e29ac652b | ||
| f8b0fe881f | |||
| 
						 | 
					2941a335f0 | ||
| 712ba79b55 | |||
| 
						 | 
					3df3d6bfa3 | ||
| bed23083cf | |||
| 
						 | 
					79175c96aa | ||
| 
						 | 
					c23192f2b6 | ||
| 2bf129d6ab | |||
| 4bf60486e0 | |||
| d048278b84 | |||
| 
						 | 
					1cf18ac2a9 | ||
| 
						 | 
					251f6f70c4 | ||
| 
						 | 
					26bf526741 | ||
| 
						 | 
					a86c824eb7 | ||
| 
						 | 
					f8cd67b8cd | ||
| feb5fec0dd | |||
| 
						 | 
					37996fbe90 | ||
| 08a22e02f9 | |||
| 
						 | 
					3358ce8aa2 | ||
| 
						 | 
					74158bd60b | ||
| 930530cbae | |||
| 
						 | 
					24a8dbdc4b | ||
| c3c580163d | |||
| 4ad526253a | |||
| 
						 | 
					3eca811f81 | ||
| 
						 | 
					367fe6d579 | ||
| b86a166d08 | |||
| 
						 | 
					ecd2711714 | ||
| 0e82845989 | |||
| 
						 | 
					670a31c9ef | ||
| 6f136a039b | |||
| 
						 | 
					c8516877ee | ||
| 2075368ba3 | |||
| 
						 | 
					fbef71bae8 | ||
| a946d63cae | |||
| 
						 | 
					bc47181f45 | ||
| e29671add6 | |||
| 
						 | 
					dede9e6488 | ||
| 2f658787d4 | |||
| 
						 | 
					b8ad04a42a | ||
| 766cbd8ae0 | |||
| 847352a8d3 | |||
| 147c2d756e | |||
| f42994e5c0 | |||
| 08b3e7e23e | |||
| 2f70890e6c | |||
| e9be505c59 | |||
| 13749e7a4d | |||
| 91cf06abbd | |||
| a1f29249d3 | |||
| db9c686306 | |||
| b74add18c1 | |||
| 5d3cbfc7d3 | |||
| f42e8883ae | |||
| 93344a3f7b | |||
| 2e218748e8 | |||
| 178d68aae9 | |||
| 2061b2dbb6 | |||
| fa9e204032 | |||
| bce6f8c3e7 | |||
| 1ee46a0aac | |||
| 4f749c5bb1 | |||
| 8231f4f7f6 | |||
| 241114aeb1 | |||
| e4541d8b44 | |||
| b658f5091a | |||
| c21d30e4a0 | |||
| 1349df4d97 | |||
| 6dfc8b8a01 | |||
| d280ea260e | |||
| 
						 | 
					e9295b1d12 | ||
| 
						 | 
					75b102666f | ||
| 18b95d7dde | |||
| 
						 | 
					6d3d8ccd14 | ||
| f37c8fd5d4 | |||
| 
						 | 
					c0bfe416b1 | ||
| ce520397c4 | |||
| 
						 | 
					3cd4e16c36 | ||
| 3dccbcba98 | |||
| 
						 | 
					7572ef631d | ||
| 030d906f87 | |||
| 
						 | 
					bfefab250f | ||
| 82b7b5e4fb | |||
| 
						 | 
					980bea6e4f | ||
| a7ce52d564 | |||
| 
						 | 
					4771d5e12b | ||
| 6be01f2cb2 | |||
| 3829766afd | |||
| a2939066a5 | |||
| 
						 | 
					e6718f368a | ||
| 381f3e051e | |||
| 757c458f60 | |||
| a2f089f890 | |||
| 
						 | 
					d85e3fdce2 | ||
| 5f370d1a31 | |||
| 
						 | 
					2b47343e07 | ||
| 00c1d3618f | |||
| 
						 | 
					0f80c9f7ba | ||
| a275ffa001 | |||
| 
						 | 
					2800d4c78b | ||
| a609f15425 | |||
| 
						 | 
					83cc922771 | ||
| 2bba395ba3 | |||
| a360ef0e8f | |||
| 5f0e4915f6 | |||
| f02efaf18c | |||
| 279d111d39 | |||
| 928455a651 | |||
| 6df77d317d | |||
| e8f7fab079 | |||
| ae6f29e540 | |||
| de6cdb4e4d | |||
| 0398d969d5 | |||
| 10419bc74a | |||
| 93ce60f59c | |||
| 61c082cb2f | |||
| 22f1b6551a | |||
| 73b7d998c0 | |||
| 4aca8e1ce0 | |||
| 3592052ac9 | |||
| fd5479f6e7 | |||
| 4fe11b430e | |||
| 5dc57579c3 | |||
| 
						 | 
					f974460ff3 | ||
| 5c54b82aaf | |||
| a5bf3e2edc | |||
| 3a2bac8723 | |||
| 
						 | 
					9a72dba9df | ||
| 00e853e3ee | |||
| 5aff0014bb | |||
| 807b4dc41a | |||
| a1f432fac9 | |||
| dd7e872858 | |||
| 57c2a5b95f | |||
| 73fc3d34f0 | |||
| 09e7282b2b | |||
| 5f029fd432 | |||
| 8555ebdd5c | |||
| 273da35b92 | |||
| 556e8dd568 | |||
| f3573e651b | |||
| 8074f9f617 | |||
| e497b5fa89 | |||
| 520dc29f89 | |||
| 59d6c26003 | 
							
								
								
									
										8
									
								
								.github/ISSUE_TEMPLATE/bug_report.md
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										8
									
								
								.github/ISSUE_TEMPLATE/bug_report.md
									
									
									
									
										vendored
									
									
								
							@@ -1,6 +1,6 @@
 | 
				
			|||||||
---
 | 
					---
 | 
				
			||||||
name: Bug report
 | 
					name: Bug report
 | 
				
			||||||
about: For reporting bugs in go-micro
 | 
					about: For reporting bugs in micro
 | 
				
			||||||
title: "[BUG]"
 | 
					title: "[BUG]"
 | 
				
			||||||
labels: ''
 | 
					labels: ''
 | 
				
			||||||
assignees: ''
 | 
					assignees: ''
 | 
				
			||||||
@@ -16,9 +16,3 @@ assignees: ''
 | 
				
			|||||||
**How to reproduce the bug:**
 | 
					**How to reproduce the bug:**
 | 
				
			||||||
 | 
					
 | 
				
			||||||
If possible, please include a minimal code snippet here.
 | 
					If possible, please include a minimal code snippet here.
 | 
				
			||||||
 | 
					 | 
				
			||||||
**Environment:**
 | 
					 | 
				
			||||||
Go Version: please paste `go version` output here
 | 
					 | 
				
			||||||
```
 | 
					 | 
				
			||||||
please paste `go env` output here
 | 
					 | 
				
			||||||
```
 | 
					 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,6 +1,6 @@
 | 
				
			|||||||
---
 | 
					---
 | 
				
			||||||
name: Feature request / Enhancement
 | 
					name: Feature request / Enhancement
 | 
				
			||||||
about: If you have a need not served by go-micro
 | 
					about: If you have a need not served by micro
 | 
				
			||||||
title: "[FEATURE]"
 | 
					title: "[FEATURE]"
 | 
				
			||||||
labels: ''
 | 
					labels: ''
 | 
				
			||||||
assignees: ''
 | 
					assignees: ''
 | 
				
			||||||
@@ -14,4 +14,4 @@ A clear and concise description of what the problem is. Ex. I'm always frustrate
 | 
				
			|||||||
A clear and concise description of what you want to happen.
 | 
					A clear and concise description of what you want to happen.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
**Additional context**
 | 
					**Additional context**
 | 
				
			||||||
Add any other context or screenshots about the feature request here.
 | 
					Add any other context or screenshots about the feature request here.
 | 
				
			||||||
							
								
								
									
										10
									
								
								.github/ISSUE_TEMPLATE/question.md
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										10
									
								
								.github/ISSUE_TEMPLATE/question.md
									
									
									
									
										vendored
									
									
								
							@@ -1,14 +1,8 @@
 | 
				
			|||||||
---
 | 
					---
 | 
				
			||||||
name: Question
 | 
					name: Question
 | 
				
			||||||
about: Ask a question about go-micro
 | 
					about: Ask a question about micro
 | 
				
			||||||
title: ''
 | 
					title: ''
 | 
				
			||||||
labels: ''
 | 
					labels: ''
 | 
				
			||||||
assignees: ''
 | 
					assignees: ''
 | 
				
			||||||
 | 
					
 | 
				
			||||||
---
 | 
					---
 | 
				
			||||||
 | 
					 | 
				
			||||||
Before asking, please check if your question has already been answered: 
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
1. Check the documentation - https://micro.mu/docs/
 | 
					 | 
				
			||||||
2. Check the examples and plugins - https://github.com/micro/examples & https://github.com/micro/go-plugins
 | 
					 | 
				
			||||||
3. Search existing issues
 | 
					 | 
				
			||||||
							
								
								
									
										20
									
								
								.github/renovate.json
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										20
									
								
								.github/renovate.json
									
									
									
									
										vendored
									
									
								
							@@ -1,20 +0,0 @@
 | 
				
			|||||||
{
 | 
					 | 
				
			||||||
  "extends": [
 | 
					 | 
				
			||||||
    "config:base"
 | 
					 | 
				
			||||||
  ],
 | 
					 | 
				
			||||||
  "postUpdateOptions": ["gomodTidy"],
 | 
					 | 
				
			||||||
  "packageRules": [
 | 
					 | 
				
			||||||
    {
 | 
					 | 
				
			||||||
      "matchUpdateTypes": ["minor", "patch", "pin", "digest"],
 | 
					 | 
				
			||||||
      "automerge": true
 | 
					 | 
				
			||||||
    },
 | 
					 | 
				
			||||||
    {
 | 
					 | 
				
			||||||
      "groupName": "all deps",
 | 
					 | 
				
			||||||
      "separateMajorMinor": true,
 | 
					 | 
				
			||||||
      "groupSlug": "all",
 | 
					 | 
				
			||||||
      "packagePatterns": [
 | 
					 | 
				
			||||||
        "*"
 | 
					 | 
				
			||||||
      ]
 | 
					 | 
				
			||||||
    }
 | 
					 | 
				
			||||||
  ]
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
							
								
								
									
										45
									
								
								.github/workflows/build.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										45
									
								
								.github/workflows/build.yml
									
									
									
									
										vendored
									
									
								
							@@ -1,45 +0,0 @@
 | 
				
			|||||||
name: build
 | 
					 | 
				
			||||||
on:
 | 
					 | 
				
			||||||
 push:
 | 
					 | 
				
			||||||
    branches:
 | 
					 | 
				
			||||||
    - master
 | 
					 | 
				
			||||||
jobs:
 | 
					 | 
				
			||||||
  test:
 | 
					 | 
				
			||||||
    name: test
 | 
					 | 
				
			||||||
    runs-on: ubuntu-latest
 | 
					 | 
				
			||||||
    steps:
 | 
					 | 
				
			||||||
    - name: setup
 | 
					 | 
				
			||||||
      uses: actions/setup-go@v2
 | 
					 | 
				
			||||||
      with:
 | 
					 | 
				
			||||||
        go-version: 1.16
 | 
					 | 
				
			||||||
    - name: checkout
 | 
					 | 
				
			||||||
      uses: actions/checkout@v2
 | 
					 | 
				
			||||||
    - name: cache
 | 
					 | 
				
			||||||
      uses: actions/cache@v2
 | 
					 | 
				
			||||||
      with:
 | 
					 | 
				
			||||||
        path: ~/go/pkg/mod
 | 
					 | 
				
			||||||
        key: ${{ runner.os }}-go-${{ hashFiles('**/go.sum') }}
 | 
					 | 
				
			||||||
        restore-keys: ${{ runner.os }}-go-
 | 
					 | 
				
			||||||
    - name: deps
 | 
					 | 
				
			||||||
      run: go get -v -t -d ./...
 | 
					 | 
				
			||||||
    - name: test
 | 
					 | 
				
			||||||
      env:
 | 
					 | 
				
			||||||
        INTEGRATION_TESTS: yes
 | 
					 | 
				
			||||||
      run: go test -mod readonly -v ./...
 | 
					 | 
				
			||||||
  lint:
 | 
					 | 
				
			||||||
    name: lint
 | 
					 | 
				
			||||||
    runs-on: ubuntu-latest
 | 
					 | 
				
			||||||
    steps:
 | 
					 | 
				
			||||||
      - name: checkout
 | 
					 | 
				
			||||||
        uses: actions/checkout@v2
 | 
					 | 
				
			||||||
      - name: lint
 | 
					 | 
				
			||||||
        uses: golangci/golangci-lint-action@v2
 | 
					 | 
				
			||||||
        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.39
 | 
					 | 
				
			||||||
          # 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
 | 
					 | 
				
			||||||
							
								
								
									
										53
									
								
								.github/workflows/job_coverage.yml
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										53
									
								
								.github/workflows/job_coverage.yml
									
									
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,53 @@
 | 
				
			|||||||
 | 
					name: coverage
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					on:
 | 
				
			||||||
 | 
					  push:
 | 
				
			||||||
 | 
					    branches: [ main, v3, v4 ]
 | 
				
			||||||
 | 
					    paths-ignore:
 | 
				
			||||||
 | 
					      - '.github/**'
 | 
				
			||||||
 | 
					      - '.gitea/**'
 | 
				
			||||||
 | 
					  pull_request:
 | 
				
			||||||
 | 
					    branches: [ main, v3, v4 ]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					jobs:
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  build:
 | 
				
			||||||
 | 
					    if: github.server_url != 'https://github.com'
 | 
				
			||||||
 | 
					    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: false
 | 
				
			||||||
 | 
					        skip_checkout: false
 | 
				
			||||||
 | 
					        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
									
								
								.github/workflows/job_lint.yml
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										29
									
								
								.github/workflows/job_lint.yml
									
									
									
									
										vendored
									
									
										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: golangci/golangci-lint-action@v6
 | 
				
			||||||
 | 
					      with:
 | 
				
			||||||
 | 
					        version: 'latest'
 | 
				
			||||||
							
								
								
									
										31
									
								
								.github/workflows/job_test.yml
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										31
									
								
								.github/workflows/job_test.yml
									
									
									
									
										vendored
									
									
										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
									
								
								.github/workflows/job_tests.yml
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										50
									
								
								.github/workflows/job_tests.yml
									
									
									
									
										vendored
									
									
										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: ${{ github.workspace }}/go.work
 | 
				
			||||||
 | 
					      run: |
 | 
				
			||||||
 | 
					        go work init
 | 
				
			||||||
 | 
					        go work use .
 | 
				
			||||||
 | 
					        go work use micro-tests
 | 
				
			||||||
 | 
					    - name: setup deps
 | 
				
			||||||
 | 
					      env:
 | 
				
			||||||
 | 
					        GOWORK: ${{ github.workspace }}/go.work
 | 
				
			||||||
 | 
					      run: go get -v ./...
 | 
				
			||||||
 | 
					    - name: run tests
 | 
				
			||||||
 | 
					      env:
 | 
				
			||||||
 | 
					        INTEGRATION_TESTS: yes
 | 
				
			||||||
 | 
					        GOWORK: ${{ github.workspace }}/go.work
 | 
				
			||||||
 | 
					      run: |
 | 
				
			||||||
 | 
					        cd micro-tests
 | 
				
			||||||
 | 
					        go test -mod readonly -v ./... || true
 | 
				
			||||||
							
								
								
									
										45
									
								
								.github/workflows/pr.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										45
									
								
								.github/workflows/pr.yml
									
									
									
									
										vendored
									
									
								
							@@ -1,45 +0,0 @@
 | 
				
			|||||||
name: prbuild
 | 
					 | 
				
			||||||
on:
 | 
					 | 
				
			||||||
  pull_request:
 | 
					 | 
				
			||||||
    branches:
 | 
					 | 
				
			||||||
    - master
 | 
					 | 
				
			||||||
jobs:
 | 
					 | 
				
			||||||
  test:
 | 
					 | 
				
			||||||
    name: test
 | 
					 | 
				
			||||||
    runs-on: ubuntu-latest
 | 
					 | 
				
			||||||
    steps:
 | 
					 | 
				
			||||||
    - name: setup
 | 
					 | 
				
			||||||
      uses: actions/setup-go@v2
 | 
					 | 
				
			||||||
      with:
 | 
					 | 
				
			||||||
        go-version: 1.16
 | 
					 | 
				
			||||||
    - name: checkout
 | 
					 | 
				
			||||||
      uses: actions/checkout@v2
 | 
					 | 
				
			||||||
    - name: cache
 | 
					 | 
				
			||||||
      uses: actions/cache@v2
 | 
					 | 
				
			||||||
      with:
 | 
					 | 
				
			||||||
        path: ~/go/pkg/mod
 | 
					 | 
				
			||||||
        key: ${{ runner.os }}-go-${{ hashFiles('**/go.sum') }}
 | 
					 | 
				
			||||||
        restore-keys: ${{ runner.os }}-go-
 | 
					 | 
				
			||||||
    - name: deps
 | 
					 | 
				
			||||||
      run: go get -v -t -d ./...
 | 
					 | 
				
			||||||
    - name: test
 | 
					 | 
				
			||||||
      env:
 | 
					 | 
				
			||||||
        INTEGRATION_TESTS: yes
 | 
					 | 
				
			||||||
      run: go test -mod readonly -v ./...
 | 
					 | 
				
			||||||
  lint:
 | 
					 | 
				
			||||||
    name: lint
 | 
					 | 
				
			||||||
    runs-on: ubuntu-latest
 | 
					 | 
				
			||||||
    steps:
 | 
					 | 
				
			||||||
      - name: checkout
 | 
					 | 
				
			||||||
        uses: actions/checkout@v2
 | 
					 | 
				
			||||||
      - name: lint
 | 
					 | 
				
			||||||
        uses: golangci/golangci-lint-action@v2
 | 
					 | 
				
			||||||
        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.39
 | 
					 | 
				
			||||||
          # 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
 | 
					 | 
				
			||||||
							
								
								
									
										39
									
								
								.gitignore
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										39
									
								
								.gitignore
									
									
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,39 @@
 | 
				
			|||||||
 | 
					# Develop tools
 | 
				
			||||||
 | 
					/.vscode/
 | 
				
			||||||
 | 
					/.idea/
 | 
				
			||||||
 | 
					.idea
 | 
				
			||||||
 | 
					.vscode
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# Binaries for programs and plugins
 | 
				
			||||||
 | 
					*.exe
 | 
				
			||||||
 | 
					*.exe~
 | 
				
			||||||
 | 
					*.dll
 | 
				
			||||||
 | 
					*.so
 | 
				
			||||||
 | 
					*.dylib
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# Folders
 | 
				
			||||||
 | 
					_obj
 | 
				
			||||||
 | 
					_test
 | 
				
			||||||
 | 
					_build
 | 
				
			||||||
 | 
					.DS_Store
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# Architecture specific extensions/prefixes
 | 
				
			||||||
 | 
					*.[568vq]
 | 
				
			||||||
 | 
					[568vq].out
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					*.cgo1.go
 | 
				
			||||||
 | 
					*.cgo2.c
 | 
				
			||||||
 | 
					_cgo_defun.c
 | 
				
			||||||
 | 
					_cgo_gotypes.go
 | 
				
			||||||
 | 
					_cgo_export.*
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# Test binary, build with `go test -c`
 | 
				
			||||||
 | 
					*.test
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# Output of the go coverage tool, specifically when used with LiteIDE
 | 
				
			||||||
 | 
					*.out
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# vim temp files
 | 
				
			||||||
 | 
					*~
 | 
				
			||||||
 | 
					*.swp
 | 
				
			||||||
 | 
					*.swo
 | 
				
			||||||
@@ -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
 | 
					 | 
				
			||||||
    - golint
 | 
					 | 
				
			||||||
    - gosec
 | 
					 | 
				
			||||||
    - makezero
 | 
					 | 
				
			||||||
    - misspell
 | 
					 | 
				
			||||||
    - nakedret
 | 
					 | 
				
			||||||
    - nestif
 | 
					 | 
				
			||||||
    - nilerr
 | 
					 | 
				
			||||||
    - noctx
 | 
					 | 
				
			||||||
    - prealloc
 | 
					 | 
				
			||||||
    - unconvert
 | 
					 | 
				
			||||||
    - unparam
 | 
					 | 
				
			||||||
  disable-all: false
 | 
					 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										160
									
								
								README.md
									
									
									
									
									
								
							
							
						
						
									
										160
									
								
								README.md
									
									
									
									
									
								
							@@ -1,18 +1,26 @@
 | 
				
			|||||||
# HTTP Client
 | 
					# HTTP Client
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
This plugin is a http client for micro.
 | 
					This plugin is an HTTP client for [Micro](https://pkg.go.dev/go.unistack.org/micro/v3).
 | 
				
			||||||
 | 
					It implements the [micro.Client](https://pkg.go.dev/go.unistack.org/micro/v3/client#Client) interface.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
## Overview
 | 
					## Overview
 | 
				
			||||||
 | 
					
 | 
				
			||||||
The http client wraps `net/http` to provide a robust micro client with service discovery, load balancing and streaming. 
 | 
					The HTTP client wraps `net/http` to provide a robust client with service discovery, load balancing and 
 | 
				
			||||||
It complies with the [micro.Client](https://godoc.org/github.com/unistack-org/micro-client-http#Client) interface.
 | 
					implements HTTP rules defined in the [google/api/http.proto](https://github.com/googleapis/googleapis/blob/master/google/api/http.proto) specification.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					## Limitations
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					* Streaming is not yet implemented.
 | 
				
			||||||
 | 
					* Only protobuf-generated messages are supported.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
## Usage
 | 
					## Usage
 | 
				
			||||||
 | 
					
 | 
				
			||||||
### Use directly
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
```go
 | 
					```go
 | 
				
			||||||
import "github.com/unistack-org/micro-client-http"
 | 
					import (
 | 
				
			||||||
 | 
					    "go.unistack.org/micro/v3"
 | 
				
			||||||
 | 
					    http "go.unistack.org/micro-client-http/v3"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
service := micro.NewService(
 | 
					service := micro.NewService(
 | 
				
			||||||
	micro.Name("my.service"),
 | 
						micro.Name("my.service"),
 | 
				
			||||||
@@ -20,43 +28,137 @@ service := micro.NewService(
 | 
				
			|||||||
)
 | 
					)
 | 
				
			||||||
```
 | 
					```
 | 
				
			||||||
 | 
					
 | 
				
			||||||
### Call Service
 | 
					### Simple call
 | 
				
			||||||
 | 
					
 | 
				
			||||||
Assuming you have a http service "my.service" with path "/foo/bar"
 | 
					 | 
				
			||||||
```go
 | 
					```go
 | 
				
			||||||
// new client
 | 
					import (
 | 
				
			||||||
client := http.NewClient()
 | 
					    "go.unistack.org/micro/v3/client"
 | 
				
			||||||
 | 
					    http "go.unistack.org/micro-client-http/v3"
 | 
				
			||||||
 | 
					    jsoncodec "go.unistack.org/micro-codec-json/v3"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// create request/response
 | 
					c := http.NewClient(
 | 
				
			||||||
request := client.NewRequest("my.service", "/foo/bar", protoRequest{})
 | 
					    client.Codec("application/json", jsoncodec.NewCodec()),
 | 
				
			||||||
response := new(protoResponse)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// call service
 | 
					req := c.NewRequest(
 | 
				
			||||||
err := client.Call(context.TODO(), request, response)
 | 
					    "user-service",
 | 
				
			||||||
 | 
					    "/user/{user_id}/order/{order_id}",
 | 
				
			||||||
 | 
					    &protoReq{UserId: "123", OrderId: 456},
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					rsp := new(protoRsp)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					err := c.Call(
 | 
				
			||||||
 | 
					    ctx,
 | 
				
			||||||
 | 
					    req,
 | 
				
			||||||
 | 
					    rsp,
 | 
				
			||||||
 | 
					    client.WithAddress("example.com"),
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
```
 | 
					```
 | 
				
			||||||
 | 
					
 | 
				
			||||||
or you can call any rest api or site and unmarshal to response struct
 | 
					### Call with specific options
 | 
				
			||||||
 | 
					
 | 
				
			||||||
```go
 | 
					```go
 | 
				
			||||||
// new client
 | 
					import (
 | 
				
			||||||
client := client.NewClientCallOptions(http.NewClient(), http.Address("https://api.github.com"))
 | 
					    "go.unistack.org/micro/v3/client"
 | 
				
			||||||
 | 
					    http "go.unistack.org/micro-client-http/v3"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
req := client.NewRequest("github", "/users/vtolstov", nil)
 | 
					err := c.Call(
 | 
				
			||||||
rsp := make(map[string]interface{})
 | 
					    ctx,
 | 
				
			||||||
 | 
					    req,
 | 
				
			||||||
err := c.Call(context.TODO(), req, &rsp, mhttp.Method(http.MethodGet)) 
 | 
					    rsp,
 | 
				
			||||||
 | 
					    client.WithAddress("example.com"),
 | 
				
			||||||
 | 
					    http.Method("POST"),
 | 
				
			||||||
 | 
					    http.Path("/user/{user_id}/order/{order_id}"),
 | 
				
			||||||
 | 
					    http.Body("*"), // <- use all fields from the proto request as HTTP request body or specify a single field name to use only that field (see Google API HTTP spec: google/api/http.proto)
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
```
 | 
					```
 | 
				
			||||||
 | 
					
 | 
				
			||||||
Look at http_test.go for detailed use.
 | 
					### Call with request headers
 | 
				
			||||||
 | 
					
 | 
				
			||||||
### Encoding
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
Default protobuf with content-type application/proto
 | 
					 | 
				
			||||||
```go
 | 
					```go
 | 
				
			||||||
client.NewRequest("service", "/path", protoRequest{})
 | 
					import (
 | 
				
			||||||
 | 
					    "go.unistack.org/micro/v3/metadata"
 | 
				
			||||||
 | 
					    http "go.unistack.org/micro-client-http/v3"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					ctx := metadata.NewOutgoingContext(ctx, metadata.Pairs(
 | 
				
			||||||
 | 
					    "Authorization", "Bearer token",
 | 
				
			||||||
 | 
					    "My-Header", "My-Header-Value",
 | 
				
			||||||
 | 
					))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					err := c.Call(
 | 
				
			||||||
 | 
					    ctx,
 | 
				
			||||||
 | 
					    req,
 | 
				
			||||||
 | 
					    rsp,
 | 
				
			||||||
 | 
					    http.Header("Authorization", "true", "My-Header", "false"), // <- call option that declares required/optional headers
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
```
 | 
					```
 | 
				
			||||||
 | 
					
 | 
				
			||||||
Json with content-type application/json
 | 
					### Call with response headers
 | 
				
			||||||
 | 
					
 | 
				
			||||||
```go
 | 
					```go
 | 
				
			||||||
client.NewJsonRequest("service", "/path", jsonRequest{})
 | 
					import (
 | 
				
			||||||
 | 
					    "go.unistack.org/micro/v3/metadata"
 | 
				
			||||||
 | 
					    http "go.unistack.org/micro-client-http/v3"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					respMetadata := metadata.Metadata{}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					err := c.Call(
 | 
				
			||||||
 | 
					    ctx,
 | 
				
			||||||
 | 
					    req,
 | 
				
			||||||
 | 
					    rsp,
 | 
				
			||||||
 | 
					    client.WithResponseMetadata(&respMetadata), // <- metadata with response headers
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
```
 | 
					```
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					### Call with cookies
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					```go
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
					    "go.unistack.org/micro/v3/metadata"
 | 
				
			||||||
 | 
					    http "go.unistack.org/micro-client-http/v3"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					ctx := metadata.NewOutgoingContext(context.Background(), metadata.Pairs(
 | 
				
			||||||
 | 
					    "Cookie", "session_id=abc123; theme=dark",
 | 
				
			||||||
 | 
					))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					err := c.Call(
 | 
				
			||||||
 | 
					    ctx,
 | 
				
			||||||
 | 
					    req,
 | 
				
			||||||
 | 
					    rsp,
 | 
				
			||||||
 | 
					    http.Cookie("session_id", "true", "theme", "false"), // <- call option that declares required/optional cookies
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					```
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					### Call with error mapping
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					```go
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
					    http "go.unistack.org/micro-client-http/v3"
 | 
				
			||||||
 | 
					    status "go.unistack.org/micro-client-http/v3/status"
 | 
				
			||||||
 | 
					    jsoncodec "go.unistack.org/micro-codec-json/v3"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					err := c.Call(
 | 
				
			||||||
 | 
					    ctx,
 | 
				
			||||||
 | 
					    req,
 | 
				
			||||||
 | 
					    rsp,
 | 
				
			||||||
 | 
					    http.ErrorMap(map[string]any{
 | 
				
			||||||
 | 
					        "default": &protoDefaultError{}, // <- default case
 | 
				
			||||||
 | 
					        "403":     &protoSpecialError{}, // <- key is the HTTP status code that is mapped to this error
 | 
				
			||||||
 | 
					    }),
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					if err != nil {
 | 
				
			||||||
 | 
					    s, ok := status.FromError(err)
 | 
				
			||||||
 | 
					    if !ok {...}
 | 
				
			||||||
 | 
					    
 | 
				
			||||||
 | 
					    code    := s.Code()     // HTTP status code
 | 
				
			||||||
 | 
					    message := s.Message()  // HTTP status text
 | 
				
			||||||
 | 
					    details := s.Details()  // Error type mapped from ErrorMap
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					```
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										116
									
								
								builder/body.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										116
									
								
								builder/body.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,116 @@
 | 
				
			|||||||
 | 
					package builder
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"fmt"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"google.golang.org/protobuf/proto"
 | 
				
			||||||
 | 
						"google.golang.org/protobuf/reflect/protoreflect"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func buildSingleFieldBody(msg proto.Message, fieldName string) (proto.Message, error) {
 | 
				
			||||||
 | 
						msgReflect := msg.ProtoReflect()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						fd, found := findFieldByName(msgReflect, fieldName)
 | 
				
			||||||
 | 
						if !found || fd == nil {
 | 
				
			||||||
 | 
							return nil, fmt.Errorf("field %s not found", fieldName)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if !msgReflect.Has(fd) {
 | 
				
			||||||
 | 
							return nil, fmt.Errorf("field %s is not set", fieldName)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						val := msgReflect.Get(fd)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if fd.Kind() == protoreflect.MessageKind {
 | 
				
			||||||
 | 
							return val.Message().Interface(), nil
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						newMsg := proto.Clone(msg)
 | 
				
			||||||
 | 
						newMsgReflect := newMsg.ProtoReflect()
 | 
				
			||||||
 | 
						newMsgReflect.Range(func(f protoreflect.FieldDescriptor, _ protoreflect.Value) bool {
 | 
				
			||||||
 | 
							if f != fd {
 | 
				
			||||||
 | 
								newMsgReflect.Clear(f)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							return true
 | 
				
			||||||
 | 
						})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return newMsg, nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func buildFullBody(msg proto.Message, usedFieldsPath *usedFields) (proto.Message, error) {
 | 
				
			||||||
 | 
						var (
 | 
				
			||||||
 | 
							msgReflect    = msg.ProtoReflect()
 | 
				
			||||||
 | 
							newMsg        = msgReflect.New().Interface()
 | 
				
			||||||
 | 
							newMsgReflect = newMsg.ProtoReflect()
 | 
				
			||||||
 | 
						)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						fields := msgReflect.Descriptor().Fields()
 | 
				
			||||||
 | 
						for i := 0; i < fields.Len(); i++ {
 | 
				
			||||||
 | 
							fd := fields.Get(i)
 | 
				
			||||||
 | 
							fieldName := fd.JSONName()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							if usedFieldsPath.hasTopLevelKey(fieldName) {
 | 
				
			||||||
 | 
								continue
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							val := msgReflect.Get(fd)
 | 
				
			||||||
 | 
							if !val.IsValid() {
 | 
				
			||||||
 | 
								continue
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							// Note: order of the cases is important!
 | 
				
			||||||
 | 
							switch {
 | 
				
			||||||
 | 
							case fd.IsList():
 | 
				
			||||||
 | 
								list := val.List()
 | 
				
			||||||
 | 
								newList := newMsgReflect.Mutable(fd).List()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								if fd.Kind() == protoreflect.MessageKind {
 | 
				
			||||||
 | 
									for j := 0; j < list.Len(); j++ {
 | 
				
			||||||
 | 
										elem, err := buildFullBody(list.Get(j).Message().Interface(), usedFieldsPath)
 | 
				
			||||||
 | 
										if err != nil {
 | 
				
			||||||
 | 
											return nil, fmt.Errorf("recursive build full body: %w", err)
 | 
				
			||||||
 | 
										}
 | 
				
			||||||
 | 
										newList.Append(protoreflect.ValueOfMessage(elem.ProtoReflect()))
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								} else {
 | 
				
			||||||
 | 
									for j := 0; j < list.Len(); j++ {
 | 
				
			||||||
 | 
										newList.Append(list.Get(j))
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							case fd.IsMap():
 | 
				
			||||||
 | 
								var (
 | 
				
			||||||
 | 
									m        = val.Map()
 | 
				
			||||||
 | 
									newMap   = newMsgReflect.Mutable(fd).Map()
 | 
				
			||||||
 | 
									rangeErr error
 | 
				
			||||||
 | 
								)
 | 
				
			||||||
 | 
								m.Range(func(k protoreflect.MapKey, v protoreflect.Value) bool {
 | 
				
			||||||
 | 
									if fd.MapValue().Kind() == protoreflect.MessageKind {
 | 
				
			||||||
 | 
										elem, err := buildFullBody(v.Message().Interface(), usedFieldsPath)
 | 
				
			||||||
 | 
										if err != nil {
 | 
				
			||||||
 | 
											rangeErr = fmt.Errorf("recursive build full body: %w", err)
 | 
				
			||||||
 | 
											return false
 | 
				
			||||||
 | 
										}
 | 
				
			||||||
 | 
										newMap.Set(k, protoreflect.ValueOfMessage(elem.ProtoReflect()))
 | 
				
			||||||
 | 
									} else {
 | 
				
			||||||
 | 
										newMap.Set(k, v)
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									return true
 | 
				
			||||||
 | 
								})
 | 
				
			||||||
 | 
								if rangeErr != nil {
 | 
				
			||||||
 | 
									return nil, fmt.Errorf("map range error: %w", rangeErr)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							case fd.Kind() == protoreflect.MessageKind:
 | 
				
			||||||
 | 
								elem, err := buildFullBody(val.Message().Interface(), usedFieldsPath)
 | 
				
			||||||
 | 
								if err != nil {
 | 
				
			||||||
 | 
									return nil, fmt.Errorf("recursive build full body: %w", err)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								newMsgReflect.Set(fd, protoreflect.ValueOfMessage(elem.ProtoReflect()))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							default:
 | 
				
			||||||
 | 
								newMsgReflect.Set(fd, val)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return newMsg, nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										22
									
								
								builder/body_option.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										22
									
								
								builder/body_option.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,22 @@
 | 
				
			|||||||
 | 
					package builder
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					const (
 | 
				
			||||||
 | 
						singleWildcard string = "*"
 | 
				
			||||||
 | 
						doubleWildcard string = "**"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					type bodyOption string
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (o bodyOption) String() string { return string(o) }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (o bodyOption) isFullBody() bool {
 | 
				
			||||||
 | 
						return o.String() == singleWildcard
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (o bodyOption) isWithoutBody() bool {
 | 
				
			||||||
 | 
						return o == ""
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (o bodyOption) isSingleField() bool {
 | 
				
			||||||
 | 
						return o != "" && o.String() != singleWildcard
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										79
									
								
								builder/body_option_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										79
									
								
								builder/body_option_test.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,79 @@
 | 
				
			|||||||
 | 
					package builder
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"testing"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"github.com/stretchr/testify/require"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestBodyOption_String(t *testing.T) {
 | 
				
			||||||
 | 
						tests := []struct {
 | 
				
			||||||
 | 
							name string
 | 
				
			||||||
 | 
							opt  bodyOption
 | 
				
			||||||
 | 
							want string
 | 
				
			||||||
 | 
						}{
 | 
				
			||||||
 | 
							{"empty", bodyOption(""), ""},
 | 
				
			||||||
 | 
							{"star", bodyOption("*"), "*"},
 | 
				
			||||||
 | 
							{"field", bodyOption("field"), "field"},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, tt := range tests {
 | 
				
			||||||
 | 
							t.Run(tt.name, func(t *testing.T) {
 | 
				
			||||||
 | 
								require.Equal(t, tt.want, tt.opt.String())
 | 
				
			||||||
 | 
							})
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestBodyOption_isFullBody(t *testing.T) {
 | 
				
			||||||
 | 
						tests := []struct {
 | 
				
			||||||
 | 
							name string
 | 
				
			||||||
 | 
							opt  bodyOption
 | 
				
			||||||
 | 
							want bool
 | 
				
			||||||
 | 
						}{
 | 
				
			||||||
 | 
							{"empty", bodyOption(""), false},
 | 
				
			||||||
 | 
							{"star", bodyOption("*"), true},
 | 
				
			||||||
 | 
							{"field", bodyOption("field"), false},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, tt := range tests {
 | 
				
			||||||
 | 
							t.Run(tt.name, func(t *testing.T) {
 | 
				
			||||||
 | 
								require.Equal(t, tt.want, tt.opt.isFullBody())
 | 
				
			||||||
 | 
							})
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestBodyOption_isWithoutBody(t *testing.T) {
 | 
				
			||||||
 | 
						tests := []struct {
 | 
				
			||||||
 | 
							name string
 | 
				
			||||||
 | 
							opt  bodyOption
 | 
				
			||||||
 | 
							want bool
 | 
				
			||||||
 | 
						}{
 | 
				
			||||||
 | 
							{"empty", bodyOption(""), true},
 | 
				
			||||||
 | 
							{"star", bodyOption("*"), false},
 | 
				
			||||||
 | 
							{"field", bodyOption("field"), false},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, tt := range tests {
 | 
				
			||||||
 | 
							t.Run(tt.name, func(t *testing.T) {
 | 
				
			||||||
 | 
								require.Equal(t, tt.want, tt.opt.isWithoutBody())
 | 
				
			||||||
 | 
							})
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestBodyOption_isSingleField(t *testing.T) {
 | 
				
			||||||
 | 
						tests := []struct {
 | 
				
			||||||
 | 
							name string
 | 
				
			||||||
 | 
							opt  bodyOption
 | 
				
			||||||
 | 
							want bool
 | 
				
			||||||
 | 
						}{
 | 
				
			||||||
 | 
							{"empty", bodyOption(""), false},
 | 
				
			||||||
 | 
							{"star", bodyOption("*"), false},
 | 
				
			||||||
 | 
							{"field", bodyOption("field"), true},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, tt := range tests {
 | 
				
			||||||
 | 
							t.Run(tt.name, func(t *testing.T) {
 | 
				
			||||||
 | 
								require.Equal(t, tt.want, tt.opt.isSingleField())
 | 
				
			||||||
 | 
							})
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										114
									
								
								builder/helpers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										114
									
								
								builder/helpers.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,114 @@
 | 
				
			|||||||
 | 
					package builder
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"encoding/base64"
 | 
				
			||||||
 | 
						"fmt"
 | 
				
			||||||
 | 
						"strconv"
 | 
				
			||||||
 | 
						"strings"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"google.golang.org/protobuf/reflect/protoreflect"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// findFieldByPath resolves a dot-separated field path in a protobuf message and returns the protoreflect value and its descriptor.
 | 
				
			||||||
 | 
					func findFieldByPath(msg protoreflect.Message, fieldPath string) (protoreflect.Value, protoreflect.FieldDescriptor, bool) {
 | 
				
			||||||
 | 
						var (
 | 
				
			||||||
 | 
							current    = msg
 | 
				
			||||||
 | 
							parts      = strings.Split(fieldPath, ".")
 | 
				
			||||||
 | 
							partsCount = len(parts) - 1
 | 
				
			||||||
 | 
						)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for i, part := range parts {
 | 
				
			||||||
 | 
							fd, ok := findFieldByName(current, part)
 | 
				
			||||||
 | 
							if !ok {
 | 
				
			||||||
 | 
								return protoreflect.Value{}, nil, false
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							val := current.Get(fd)
 | 
				
			||||||
 | 
							if i == partsCount { // it's last part
 | 
				
			||||||
 | 
								return val, fd, true
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							if fd.Kind() != protoreflect.MessageKind {
 | 
				
			||||||
 | 
								return protoreflect.Value{}, nil, false
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							current = val.Message()
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return protoreflect.Value{}, nil, false
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// findFieldByName find a field name in a protobuf message and returns the protoreflect field descriptor.
 | 
				
			||||||
 | 
					func findFieldByName(msg protoreflect.Message, fieldName string) (protoreflect.FieldDescriptor, bool) {
 | 
				
			||||||
 | 
						fields := msg.Descriptor().Fields()
 | 
				
			||||||
 | 
						for i := 0; i < fields.Len(); i++ {
 | 
				
			||||||
 | 
							fd := fields.Get(i)
 | 
				
			||||||
 | 
							if fd.JSONName() == fieldName {
 | 
				
			||||||
 | 
								return fd, true
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return nil, false
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// isZeroValue checks if protoreflect.Value is zero for the field.
 | 
				
			||||||
 | 
					func isZeroValue(val protoreflect.Value, fd protoreflect.FieldDescriptor) bool {
 | 
				
			||||||
 | 
						if fd.IsList() {
 | 
				
			||||||
 | 
							return val.List().Len() == 0
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if fd.IsMap() {
 | 
				
			||||||
 | 
							return val.Map().Len() == 0
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						switch fd.Kind() {
 | 
				
			||||||
 | 
						case protoreflect.StringKind:
 | 
				
			||||||
 | 
							return val.String() == ""
 | 
				
			||||||
 | 
						case protoreflect.BytesKind:
 | 
				
			||||||
 | 
							return len(val.Bytes()) == 0
 | 
				
			||||||
 | 
						case protoreflect.BoolKind:
 | 
				
			||||||
 | 
							return !val.Bool()
 | 
				
			||||||
 | 
						case protoreflect.Int32Kind, protoreflect.Sint32Kind, protoreflect.Sfixed32Kind,
 | 
				
			||||||
 | 
							protoreflect.Int64Kind, protoreflect.Sint64Kind, protoreflect.Sfixed64Kind:
 | 
				
			||||||
 | 
							return val.Int() == 0
 | 
				
			||||||
 | 
						case protoreflect.Uint32Kind, protoreflect.Fixed32Kind,
 | 
				
			||||||
 | 
							protoreflect.Uint64Kind, protoreflect.Fixed64Kind:
 | 
				
			||||||
 | 
							return val.Uint() == 0
 | 
				
			||||||
 | 
						case protoreflect.FloatKind, protoreflect.DoubleKind:
 | 
				
			||||||
 | 
							return val.Float() == 0
 | 
				
			||||||
 | 
						case protoreflect.EnumKind:
 | 
				
			||||||
 | 
							return val.Enum() == 0
 | 
				
			||||||
 | 
						case protoreflect.MessageKind:
 | 
				
			||||||
 | 
							return !val.Message().IsValid()
 | 
				
			||||||
 | 
						default:
 | 
				
			||||||
 | 
							return !val.IsValid()
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// stringifyValue converts protoreflect.Value to string for path/query substitution.
 | 
				
			||||||
 | 
					func stringifyValue(val protoreflect.Value, fd protoreflect.FieldDescriptor) (string, error) {
 | 
				
			||||||
 | 
						switch fd.Kind() {
 | 
				
			||||||
 | 
						case protoreflect.StringKind:
 | 
				
			||||||
 | 
							return val.String(), nil
 | 
				
			||||||
 | 
						case protoreflect.BytesKind:
 | 
				
			||||||
 | 
							return base64.StdEncoding.EncodeToString(val.Bytes()), nil
 | 
				
			||||||
 | 
						case protoreflect.BoolKind:
 | 
				
			||||||
 | 
							if val.Bool() {
 | 
				
			||||||
 | 
								return "true", nil
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							return "false", nil
 | 
				
			||||||
 | 
						case protoreflect.Int32Kind, protoreflect.Sint32Kind, protoreflect.Sfixed32Kind,
 | 
				
			||||||
 | 
							protoreflect.Int64Kind, protoreflect.Sint64Kind, protoreflect.Sfixed64Kind:
 | 
				
			||||||
 | 
							return fmt.Sprintf("%d", val.Int()), nil
 | 
				
			||||||
 | 
						case protoreflect.Uint32Kind, protoreflect.Fixed32Kind,
 | 
				
			||||||
 | 
							protoreflect.Uint64Kind, protoreflect.Fixed64Kind:
 | 
				
			||||||
 | 
							return fmt.Sprintf("%d", val.Uint()), nil
 | 
				
			||||||
 | 
						case protoreflect.FloatKind, protoreflect.DoubleKind:
 | 
				
			||||||
 | 
							return strconv.FormatFloat(val.Float(), 'g', -1, 64), nil
 | 
				
			||||||
 | 
						case protoreflect.EnumKind:
 | 
				
			||||||
 | 
							ed := fd.Enum().Values().ByNumber(val.Enum())
 | 
				
			||||||
 | 
							if ed != nil {
 | 
				
			||||||
 | 
								return string(ed.Name()), nil
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							return fmt.Sprintf("%d", val.Enum()), nil
 | 
				
			||||||
 | 
						default:
 | 
				
			||||||
 | 
							return "", fmt.Errorf("unsupported field kind: %s", fd.Kind())
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										313
									
								
								builder/path_template.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										313
									
								
								builder/path_template.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,313 @@
 | 
				
			|||||||
 | 
					package builder
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"errors"
 | 
				
			||||||
 | 
						"fmt"
 | 
				
			||||||
 | 
						"net/url"
 | 
				
			||||||
 | 
						"strings"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"google.golang.org/protobuf/proto"
 | 
				
			||||||
 | 
						"google.golang.org/protobuf/reflect/protoreflect"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// -------------------------- Path template representation -------------------------
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// pathSegment is a helper interface for elements of a path.
 | 
				
			||||||
 | 
					type pathSegment interface {
 | 
				
			||||||
 | 
						isSegment() bool
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// pathTemplate represents a parsed URL path template.
 | 
				
			||||||
 | 
					type pathTemplate struct {
 | 
				
			||||||
 | 
						// literalPrefix is the fixed part of the path before the first {var},
 | 
				
			||||||
 | 
						// e.g. "/v1/users/" for "/v1/users/{user_id}/orders:get".
 | 
				
			||||||
 | 
						// It is removed from segments, so segments contain only the remaining path literals and variables.
 | 
				
			||||||
 | 
						literalPrefix string
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// segments is a sequence of pathLiteral or pathVar representing the rest of the path after literalPrefix.
 | 
				
			||||||
 | 
						segments []pathSegment
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// customVerb is an optional ":verb" suffix, e.g. ":get".
 | 
				
			||||||
 | 
						customVerb string
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// pathLiteral represents a fixed literal segment in a path template, e.g., "/v1/users/".
 | 
				
			||||||
 | 
					type pathLiteral struct {
 | 
				
			||||||
 | 
						text string
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (p pathLiteral) isSegment() bool { return true }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// pathVar represents a variable segment in a path template, e.g., "{user.id}".
 | 
				
			||||||
 | 
					type pathVar struct {
 | 
				
			||||||
 | 
						// fieldPath is the dotted path to the field in the struct, e.g., "user.id".
 | 
				
			||||||
 | 
						fieldPath string
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// pattern is the optional pattern after '=', e.g., "*" or "**/orders".
 | 
				
			||||||
 | 
						// It specifies how the variable can match parts of the URL path.
 | 
				
			||||||
 | 
						pattern string
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// multiSegment is true if the pattern can match multiple path segments
 | 
				
			||||||
 | 
						// (contains '/' or "**").
 | 
				
			||||||
 | 
						multiSegment bool
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (p pathVar) isSegment() bool { return true }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// ----------------------------- Path template parsing -----------------------------
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// parsePathTemplate parses a URL path template into a pathTemplate.
 | 
				
			||||||
 | 
					// It extracts:
 | 
				
			||||||
 | 
					//  1. literalPrefix — fixed part before the first variable,
 | 
				
			||||||
 | 
					//  2. segments — sequence of pathLiteral and pathVar,
 | 
				
			||||||
 | 
					//  3. customVerb — optional ":verb" suffix.
 | 
				
			||||||
 | 
					//
 | 
				
			||||||
 | 
					// Complexity: time O(n), memory O(n).
 | 
				
			||||||
 | 
					//
 | 
				
			||||||
 | 
					// Example:
 | 
				
			||||||
 | 
					//
 | 
				
			||||||
 | 
					//	input: "/v1/users/{user_id}/orders:get"
 | 
				
			||||||
 | 
					//	output: pathTemplate{
 | 
				
			||||||
 | 
					//	  literalPrefix: "/v1/users/",
 | 
				
			||||||
 | 
					//	  segments:      [{user_id}, "/orders"],
 | 
				
			||||||
 | 
					//	  customVerb:    ":get",
 | 
				
			||||||
 | 
					//	}
 | 
				
			||||||
 | 
					func parsePathTemplate(input string) (*pathTemplate, error) {
 | 
				
			||||||
 | 
						// Step 1: extract custom verb after the last colon, e.g. ":get"
 | 
				
			||||||
 | 
						var customVerb string
 | 
				
			||||||
 | 
						if i := strings.LastIndex(input, ":"); i >= 0 && i > strings.LastIndex(input, "/") {
 | 
				
			||||||
 | 
							customVerb = input[i:]
 | 
				
			||||||
 | 
							input = input[:i]
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						var (
 | 
				
			||||||
 | 
							segments []pathSegment
 | 
				
			||||||
 | 
							buf      strings.Builder
 | 
				
			||||||
 | 
						)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// Step 2: iterate over the input and split into segments
 | 
				
			||||||
 | 
						for i := 0; i < len(input); {
 | 
				
			||||||
 | 
							if input[i] != '{' {
 | 
				
			||||||
 | 
								buf.WriteByte(input[i])
 | 
				
			||||||
 | 
								i++
 | 
				
			||||||
 | 
								continue
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							// Add literal before '{' if any
 | 
				
			||||||
 | 
							if buf.Len() > 0 {
 | 
				
			||||||
 | 
								segments = append(segments, pathLiteral{text: buf.String()})
 | 
				
			||||||
 | 
								buf.Reset()
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							// Find closing '}'
 | 
				
			||||||
 | 
							start := i + 1
 | 
				
			||||||
 | 
							offset := strings.IndexByte(input[start:], '}') // relative offset from start
 | 
				
			||||||
 | 
							if offset < 0 {
 | 
				
			||||||
 | 
								return nil, fmt.Errorf("unclosed '{' in path: %s", input)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							end := start + offset
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							token := input[start:end]
 | 
				
			||||||
 | 
							i = end + 1 // jump past '}'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							// Split field path and optional pattern
 | 
				
			||||||
 | 
							var fieldPath, pattern string
 | 
				
			||||||
 | 
							if k := strings.IndexByte(token, '='); k >= 0 {
 | 
				
			||||||
 | 
								fieldPath = strings.TrimSpace(token[:k])
 | 
				
			||||||
 | 
								pattern = strings.TrimSpace(token[k+1:])
 | 
				
			||||||
 | 
							} else {
 | 
				
			||||||
 | 
								fieldPath = strings.TrimSpace(token)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							if fieldPath == "" {
 | 
				
			||||||
 | 
								return nil, fmt.Errorf("empty variable in path: %s", input)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							pv := pathVar{
 | 
				
			||||||
 | 
								fieldPath:    fieldPath,
 | 
				
			||||||
 | 
								pattern:      pattern,
 | 
				
			||||||
 | 
								multiSegment: isMultiSegmentPattern(pattern),
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							segments = append(segments, pv)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// Step 3: add any trailing literal after last '}'
 | 
				
			||||||
 | 
						if buf.Len() > 0 {
 | 
				
			||||||
 | 
							segments = append(segments, pathLiteral{text: buf.String()})
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// Step 4: extract literalPrefix if the first segment is a literal
 | 
				
			||||||
 | 
						var literalPrefix string
 | 
				
			||||||
 | 
						if len(segments) > 0 {
 | 
				
			||||||
 | 
							if pl, ok := segments[0].(pathLiteral); ok {
 | 
				
			||||||
 | 
								literalPrefix = pl.text
 | 
				
			||||||
 | 
								segments = segments[1:] // remove from segments to avoid duplication
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// Step 5: return fully parsed pathTemplate
 | 
				
			||||||
 | 
						return &pathTemplate{
 | 
				
			||||||
 | 
							literalPrefix: literalPrefix,
 | 
				
			||||||
 | 
							segments:      segments,
 | 
				
			||||||
 | 
							customVerb:    customVerb,
 | 
				
			||||||
 | 
						}, nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// isMultiSegmentPattern returns true if pattern can match multiple path segments (contains '/' or '**').
 | 
				
			||||||
 | 
					// Examples:
 | 
				
			||||||
 | 
					// | 	Pattern     | Result | 					Usecase					|
 | 
				
			||||||
 | 
					// |----------------|--------|------------------------------------------|
 | 
				
			||||||
 | 
					// | ""             | false  | {var} 			=> single segment		|
 | 
				
			||||||
 | 
					// | "*"            | false  | {var=*} 			=> single segment		|
 | 
				
			||||||
 | 
					// | "**"           | true   | {var=**} 		=> multiple segments	|
 | 
				
			||||||
 | 
					// | "foo/*"        | true   | {var=foo/*} 		=> multiple segments	|
 | 
				
			||||||
 | 
					// | "foo/**"       | true   | {var=foo/**} 	=> multiple segments	|
 | 
				
			||||||
 | 
					// | "users/*/orders"| true  | {users/*/orders} => multiple segments	|
 | 
				
			||||||
 | 
					func isMultiSegmentPattern(pattern string) bool {
 | 
				
			||||||
 | 
						if pattern == "" {
 | 
				
			||||||
 | 
							return false
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if pattern == singleWildcard {
 | 
				
			||||||
 | 
							return false
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return strings.Contains(pattern, "/") || strings.Contains(pattern, doubleWildcard)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// ----------------------------- Path template resolving -----------------------------
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// resolvePathPlaceholders expands placeholders in a path template using values from proto.Message.
 | 
				
			||||||
 | 
					// Placeholders must be bound to non-repeated scalar fields (not lists, maps, or messages).
 | 
				
			||||||
 | 
					//
 | 
				
			||||||
 | 
					// Example:
 | 
				
			||||||
 | 
					//
 | 
				
			||||||
 | 
					//	tmpl:       "/v1/users/{user_id}/orders:get"
 | 
				
			||||||
 | 
					//	msg:        &pb.Message{UserId: 12345}
 | 
				
			||||||
 | 
					//
 | 
				
			||||||
 | 
					//	path:       "/v1/users/12345/orders:get"
 | 
				
			||||||
 | 
					//	usedFields: {"user_id"}
 | 
				
			||||||
 | 
					func resolvePathPlaceholders(tmpl *pathTemplate, msg proto.Message) (path string, usedFields *usedFields, err error) {
 | 
				
			||||||
 | 
						usedFields = newUsedFields()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						var sb strings.Builder
 | 
				
			||||||
 | 
						sb.WriteString(tmpl.literalPrefix)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						msgReflect := msg.ProtoReflect()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, segment := range tmpl.segments {
 | 
				
			||||||
 | 
							switch s := segment.(type) {
 | 
				
			||||||
 | 
							case pathLiteral:
 | 
				
			||||||
 | 
								sb.WriteString(s.text)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							case pathVar:
 | 
				
			||||||
 | 
								val, fd, ok := findFieldByPath(msgReflect, s.fieldPath)
 | 
				
			||||||
 | 
								if !ok {
 | 
				
			||||||
 | 
									return "", nil, fmt.Errorf("path placeholder %s not found", s.fieldPath)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								if isZeroValue(val, fd) {
 | 
				
			||||||
 | 
									// it's the only case that allows zero-value matches.
 | 
				
			||||||
 | 
									if s.pattern == doubleWildcard {
 | 
				
			||||||
 | 
										usedFields.add(s.fieldPath)
 | 
				
			||||||
 | 
										continue
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									return "", nil, fmt.Errorf("path placeholder %s has zero value", s.fieldPath)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								// must be scalar (non-repeated, non-map, non-message)
 | 
				
			||||||
 | 
								if fd.IsList() || fd.IsMap() || fd.Kind() == protoreflect.MessageKind {
 | 
				
			||||||
 | 
									return "", nil, fmt.Errorf("path placeholder %s must be scalar", s.fieldPath)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								usedFields.add(s.fieldPath)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								var strVal string
 | 
				
			||||||
 | 
								strVal, err = stringifyValue(val, fd)
 | 
				
			||||||
 | 
								if err != nil {
 | 
				
			||||||
 | 
									return "", nil, fmt.Errorf("stringify placeholder %s: %w", s.fieldPath, err)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								if err = validatePattern(s.pattern, strVal); err != nil {
 | 
				
			||||||
 | 
									return "", nil, fmt.Errorf("validate pattern, %s:%s: %w", s.fieldPath, strVal, err)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								parts := strings.Split(strVal, "/")
 | 
				
			||||||
 | 
								for i := range parts {
 | 
				
			||||||
 | 
									parts[i] = url.PathEscape(parts[i])
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								sb.WriteString(strings.Join(parts, "/"))
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						sb.WriteString(tmpl.customVerb)
 | 
				
			||||||
 | 
						return sb.String(), usedFields, nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// validatePattern checks whether input matches the given path pattern.
 | 
				
			||||||
 | 
					//
 | 
				
			||||||
 | 
					// Rules:
 | 
				
			||||||
 | 
					//   - "" or "*" => exactly one segment, no "/" allowed
 | 
				
			||||||
 | 
					//   - "**"      => zero or more segments (may include "/")
 | 
				
			||||||
 | 
					//   - composite patterns like "*/orders/*" must match literally
 | 
				
			||||||
 | 
					//
 | 
				
			||||||
 | 
					// Example for composite pattern case:
 | 
				
			||||||
 | 
					//
 | 
				
			||||||
 | 
					//	pattern: "*/orders/*"
 | 
				
			||||||
 | 
					//	input:   "42/orders/123"
 | 
				
			||||||
 | 
					//
 | 
				
			||||||
 | 
					//	patternSegments = ["*", "orders", "*"]
 | 
				
			||||||
 | 
					//	valueParts      = ["42", "orders", "123"]
 | 
				
			||||||
 | 
					//
 | 
				
			||||||
 | 
					//	Match:
 | 
				
			||||||
 | 
					//	  "*"      -> "42"
 | 
				
			||||||
 | 
					//	  "orders" -> "orders"
 | 
				
			||||||
 | 
					//	  "*"      -> "123"
 | 
				
			||||||
 | 
					func validatePattern(pattern, input string) error {
 | 
				
			||||||
 | 
						var (
 | 
				
			||||||
 | 
							parts    = strings.Split(input, "/")
 | 
				
			||||||
 | 
							lenParts = len(parts)
 | 
				
			||||||
 | 
						)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if pattern == "" || pattern == singleWildcard {
 | 
				
			||||||
 | 
							if lenParts != 1 {
 | 
				
			||||||
 | 
								return errors.New("must be a single path segment")
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							return nil
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if pattern == doubleWildcard {
 | 
				
			||||||
 | 
							if lenParts < 1 {
 | 
				
			||||||
 | 
								return errors.New("must contain at least one segment")
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							return nil
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						var (
 | 
				
			||||||
 | 
							patternSegments = strings.Split(pattern, "/")
 | 
				
			||||||
 | 
							patternIndex    int
 | 
				
			||||||
 | 
						)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for i := 0; i < len(patternSegments); i++ {
 | 
				
			||||||
 | 
							switch patternSegments[i] {
 | 
				
			||||||
 | 
							case singleWildcard:
 | 
				
			||||||
 | 
								if patternIndex >= lenParts || parts[patternIndex] == "" {
 | 
				
			||||||
 | 
									return fmt.Errorf("segment %d must not be empty", patternIndex)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								patternIndex++
 | 
				
			||||||
 | 
							case doubleWildcard:
 | 
				
			||||||
 | 
								if patternIndex >= lenParts {
 | 
				
			||||||
 | 
									return fmt.Errorf("must contain at least one segment at position %d", patternIndex)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								return nil
 | 
				
			||||||
 | 
							default:
 | 
				
			||||||
 | 
								if patternIndex >= lenParts || parts[patternIndex] != patternSegments[i] {
 | 
				
			||||||
 | 
									return fmt.Errorf("expected literal %s at position %d", patternSegments[i], patternIndex)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								patternIndex++
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if patternIndex != lenParts {
 | 
				
			||||||
 | 
							return errors.New("extra segments in value")
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										21
									
								
								builder/path_template_cache.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										21
									
								
								builder/path_template_cache.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,21 @@
 | 
				
			|||||||
 | 
					package builder
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import "sync"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					var (
 | 
				
			||||||
 | 
						pathTemplateCache   = make(map[string]*pathTemplate)
 | 
				
			||||||
 | 
						pathTemplateCacheMu sync.RWMutex
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func getCachedPathTemplate(path string) (*pathTemplate, bool) {
 | 
				
			||||||
 | 
						pathTemplateCacheMu.RLock()
 | 
				
			||||||
 | 
						defer pathTemplateCacheMu.RUnlock()
 | 
				
			||||||
 | 
						tmpl, ok := pathTemplateCache[path]
 | 
				
			||||||
 | 
						return tmpl, ok
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func setPathTemplateCache(path string, tmpl *pathTemplate) {
 | 
				
			||||||
 | 
						pathTemplateCacheMu.Lock()
 | 
				
			||||||
 | 
						defer pathTemplateCacheMu.Unlock()
 | 
				
			||||||
 | 
						pathTemplateCache[path] = tmpl
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										15
									
								
								builder/proto/errors.pb.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										15
									
								
								builder/proto/errors.pb.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,15 @@
 | 
				
			|||||||
 | 
					package proto
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import "google.golang.org/protobuf/encoding/protojson"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					var marshaler = protojson.MarshalOptions{}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (m *Test_Client_Call_DefaultError) Error() string {
 | 
				
			||||||
 | 
						buf, _ := marshaler.Marshal(m)
 | 
				
			||||||
 | 
						return string(buf)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (m *Test_Client_Call_SpecialError) Error() string {
 | 
				
			||||||
 | 
						buf, _ := marshaler.Marshal(m)
 | 
				
			||||||
 | 
						return string(buf)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										3906
									
								
								builder/proto/test_messages.pb.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										3906
									
								
								builder/proto/test_messages.pb.go
									
									
									
									
									
										Normal file
									
								
							
										
											
												File diff suppressed because it is too large
												Load Diff
											
										
									
								
							
							
								
								
									
										330
									
								
								builder/proto/test_messages.proto
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										330
									
								
								builder/proto/test_messages.proto
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,330 @@
 | 
				
			|||||||
 | 
					syntax = "proto3";
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					package proto;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					option go_package = "go.unistack.org/micro-client-http/v3/proto;proto";
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					message TestRequestBuilder {}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					message Test_PathOnly {
 | 
				
			||||||
 | 
					  message PrimitiveCase {
 | 
				
			||||||
 | 
					    string userId = 1 [json_name = "user_id"];
 | 
				
			||||||
 | 
					    int64  orderId = 2 [json_name = "order_id"];
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  message NestedCase {
 | 
				
			||||||
 | 
					    User user = 1;
 | 
				
			||||||
 | 
					    Order order = 2;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    message User {
 | 
				
			||||||
 | 
					      string id = 1;
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					    message Order {
 | 
				
			||||||
 | 
					      int64 id = 1;
 | 
				
			||||||
 | 
					      Product product = 2;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					      message Product {
 | 
				
			||||||
 | 
					        int64 id = 1;
 | 
				
			||||||
 | 
					      }
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  message MultipleCase {
 | 
				
			||||||
 | 
					    string userId = 1 [json_name = "user_id"];
 | 
				
			||||||
 | 
					    Order order = 2;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    message Order {
 | 
				
			||||||
 | 
					      string id = 1;
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  message RepeatedCase {
 | 
				
			||||||
 | 
					    repeated string userId = 1 [json_name = "user_id"];
 | 
				
			||||||
 | 
					    int64  orderId = 2 [json_name = "order_id"];
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  message NonPrimitiveMessageCase {
 | 
				
			||||||
 | 
					    User userId = 1 [json_name = "user_id"];
 | 
				
			||||||
 | 
					    int64  orderId = 2 [json_name = "order_id"];
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    message User {
 | 
				
			||||||
 | 
					      string id = 1;
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  message NonPrimitiveMapCase {
 | 
				
			||||||
 | 
					    map<string, string> userId = 1 [json_name = "user_id"];
 | 
				
			||||||
 | 
					    int64  orderId = 2 [json_name = "order_id"];
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  message PatternCase {
 | 
				
			||||||
 | 
					    string pattern = 1;
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  message CompositePatternCase {
 | 
				
			||||||
 | 
					    string pattern = 1;
 | 
				
			||||||
 | 
					    string orderId = 2 [json_name = "order_id"];
 | 
				
			||||||
 | 
					    string productId = 3 [json_name = "product_id"];
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					message Test_QueryOnly {
 | 
				
			||||||
 | 
					  message PrimitiveCase {
 | 
				
			||||||
 | 
					    string userId = 1 [json_name = "user_id"];
 | 
				
			||||||
 | 
					    int64  orderId = 2 [json_name = "order_id"];
 | 
				
			||||||
 | 
					    bool flag = 3;
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  message RepeatedCase {
 | 
				
			||||||
 | 
					    repeated string strings = 1;
 | 
				
			||||||
 | 
					    repeated int64 integers = 2;
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  message NestedMessageCase {
 | 
				
			||||||
 | 
					    string userId = 1 [json_name = "user_id"];
 | 
				
			||||||
 | 
					    Filter filter = 2;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    message Filter {
 | 
				
			||||||
 | 
					      int64 age = 1;
 | 
				
			||||||
 | 
					      string name = 2;
 | 
				
			||||||
 | 
					      SubFilter subFilter = 3 [json_name = "sub_filter"];
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					      message SubFilter {
 | 
				
			||||||
 | 
					        int64 subAge = 1 [json_name = "sub_age"];
 | 
				
			||||||
 | 
					        string subName = 2 [json_name = "sub_name"];
 | 
				
			||||||
 | 
					      }
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  message NestedMapCase {
 | 
				
			||||||
 | 
					    string userId = 1 [json_name = "user_id"];
 | 
				
			||||||
 | 
					    map<string, string> firstFilter = 2 [json_name = "first_filter"];
 | 
				
			||||||
 | 
					    map<string, SubFilter> secondFilter = 4 [json_name = "second_filter"];
 | 
				
			||||||
 | 
					    message SubFilter {
 | 
				
			||||||
 | 
					      int64 subAge = 1 [json_name = "sub_age"];
 | 
				
			||||||
 | 
					      string subName = 2 [json_name = "sub_name"];
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  message MultipleCase {
 | 
				
			||||||
 | 
					    string userId = 1 [json_name = "user_id"];
 | 
				
			||||||
 | 
					    repeated string strings = 2;
 | 
				
			||||||
 | 
					    Filter firstFilter = 3 [json_name = "first_filter"];
 | 
				
			||||||
 | 
					    map<string, SubFilter> secondFilter = 4 [json_name = "second_filter"];
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    message Filter {
 | 
				
			||||||
 | 
					      int64 age = 1;
 | 
				
			||||||
 | 
					      SubFilter subFilter = 2 [json_name = "sub_filter"];
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					    message SubFilter {
 | 
				
			||||||
 | 
					      int64 subAge = 1 [json_name = "sub_age"];
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  message RepeatedMessageCase {
 | 
				
			||||||
 | 
					    repeated Filter filters = 1;
 | 
				
			||||||
 | 
					    message Filter {
 | 
				
			||||||
 | 
					      int64 age = 1;
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					message Test_BodyOnly {
 | 
				
			||||||
 | 
					  message PrimitiveCase {
 | 
				
			||||||
 | 
					    string userId = 1 [json_name = "user_id"];
 | 
				
			||||||
 | 
					    int64  orderId = 2 [json_name = "order_id"];
 | 
				
			||||||
 | 
					    bool flag = 3;
 | 
				
			||||||
 | 
					    repeated string strings = 4;
 | 
				
			||||||
 | 
					    Product product = 6;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    message Product {
 | 
				
			||||||
 | 
					      string id = 1;
 | 
				
			||||||
 | 
					      string name = 2;
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  message NestedCase {
 | 
				
			||||||
 | 
					    string userId = 1 [json_name = "user_id"];
 | 
				
			||||||
 | 
					    Filter first_filter = 2 [json_name = "first_filter"];
 | 
				
			||||||
 | 
					    Filter second_filter = 3 [json_name = "second_filter"];
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    message Filter {
 | 
				
			||||||
 | 
					      int64 age = 1;
 | 
				
			||||||
 | 
					      string name = 2;
 | 
				
			||||||
 | 
					      SubFilter subFilter = 3 [json_name = "sub_filter"];
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					      message SubFilter {
 | 
				
			||||||
 | 
					        int64 subAge = 1 [json_name = "sub_age"];
 | 
				
			||||||
 | 
					        string subName = 2 [json_name = "sub_name"];
 | 
				
			||||||
 | 
					      }
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  message RepeatedMessageCase {
 | 
				
			||||||
 | 
					    string userId = 1 [json_name = "user_id"];
 | 
				
			||||||
 | 
					    repeated Product products = 2 [json_name = "products"];
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    message Product {
 | 
				
			||||||
 | 
					      string id = 1;
 | 
				
			||||||
 | 
					      string name = 2;
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  message MapCase {
 | 
				
			||||||
 | 
					    map<string, string> firstFilter = 1 [json_name = "first_filter"];
 | 
				
			||||||
 | 
					    map<string, SubFilter> secondFilter = 2 [json_name = "second_filter"];
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    message SubFilter {
 | 
				
			||||||
 | 
					      int64 subAge = 1 [json_name = "sub_age"];
 | 
				
			||||||
 | 
					      string subName = 2 [json_name = "sub_name"];
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  message MultipleCase {
 | 
				
			||||||
 | 
					    string userId = 1 [json_name = "user_id"];
 | 
				
			||||||
 | 
					    repeated SubFilter firstFilter = 2 [json_name = "first_filter"];
 | 
				
			||||||
 | 
					    map<string, SubFilter> secondFilter = 3 [json_name = "second_filter"];
 | 
				
			||||||
 | 
					    SubFilter thirdFilter = 4 [json_name = "third_filter"];
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    message SubFilter {
 | 
				
			||||||
 | 
					      int64 subAge = 1 [json_name = "sub_age"];
 | 
				
			||||||
 | 
					      string subName = 2 [json_name = "sub_name"];
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					message Test_Mixed {
 | 
				
			||||||
 | 
					  message PrimitiveCase {
 | 
				
			||||||
 | 
					    string userId = 1 [json_name = "user_id"];
 | 
				
			||||||
 | 
					    int64  orderId = 2 [json_name = "order_id"];
 | 
				
			||||||
 | 
					    Product product = 3;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    message Product {
 | 
				
			||||||
 | 
					      string id = 1;
 | 
				
			||||||
 | 
					      string name = 2;
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  message NestedCase {
 | 
				
			||||||
 | 
					    string userId = 1 [json_name = "user_id"];
 | 
				
			||||||
 | 
					    Filter first_filter = 2 [json_name = "first_filter"];
 | 
				
			||||||
 | 
					    Filter second_filter = 3 [json_name = "second_filter"];
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    message Filter {
 | 
				
			||||||
 | 
					      int64 age = 1;
 | 
				
			||||||
 | 
					      string name = 2;
 | 
				
			||||||
 | 
					      SubFilter subFilter = 3 [json_name = "sub_filter"];
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					      message SubFilter {
 | 
				
			||||||
 | 
					        int64 subAge = 1 [json_name = "sub_age"];
 | 
				
			||||||
 | 
					        string subName = 2 [json_name = "sub_name"];
 | 
				
			||||||
 | 
					      }
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  message RepeatedMessageCase {
 | 
				
			||||||
 | 
					    string userId = 1 [json_name = "user_id"];
 | 
				
			||||||
 | 
					    repeated Product products = 2 [json_name = "products"];
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    message Product {
 | 
				
			||||||
 | 
					      string id = 1;
 | 
				
			||||||
 | 
					      string name = 2;
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  message MapCase {
 | 
				
			||||||
 | 
					    map<string, string> firstFilter = 1 [json_name = "first_filter"];
 | 
				
			||||||
 | 
					    map<string, SubFilter> secondFilter = 2 [json_name = "second_filter"];
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    message SubFilter {
 | 
				
			||||||
 | 
					      int64 subAge = 1 [json_name = "sub_age"];
 | 
				
			||||||
 | 
					      string subName = 2 [json_name = "sub_name"];
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					  message MultipleCase {
 | 
				
			||||||
 | 
					    string userId = 1 [json_name = "user_id"];
 | 
				
			||||||
 | 
					    repeated SubFilter firstFilter = 2 [json_name = "first_filter"];
 | 
				
			||||||
 | 
					    map<string, SubFilter> secondFilter = 3 [json_name = "second_filter"];
 | 
				
			||||||
 | 
					    SubFilter thirdFilter = 4 [json_name = "third_filter"];
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    message SubFilter {
 | 
				
			||||||
 | 
					      int64 subAge = 1 [json_name = "sub_age"];
 | 
				
			||||||
 | 
					      string subName = 2 [json_name = "sub_name"];
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					message Benchmark {
 | 
				
			||||||
 | 
					  message Case5 {
 | 
				
			||||||
 | 
					    string field1 = 1;
 | 
				
			||||||
 | 
					    string field2 = 2;
 | 
				
			||||||
 | 
					    string field3 = 3;
 | 
				
			||||||
 | 
					    string field4 = 4;
 | 
				
			||||||
 | 
					    string field5 = 5;
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					  message Case10 {
 | 
				
			||||||
 | 
					    string field1 = 1;
 | 
				
			||||||
 | 
					    string field2 = 2;
 | 
				
			||||||
 | 
					    string field3 = 3;
 | 
				
			||||||
 | 
					    string field4 = 4;
 | 
				
			||||||
 | 
					    string field5 = 5;
 | 
				
			||||||
 | 
					    string field6 = 6;
 | 
				
			||||||
 | 
					    string field7 = 7;
 | 
				
			||||||
 | 
					    string field8 = 8;
 | 
				
			||||||
 | 
					    string field9 = 9;
 | 
				
			||||||
 | 
					    string field10 = 10;
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					  message Case30 {
 | 
				
			||||||
 | 
					    string field1 = 1;
 | 
				
			||||||
 | 
					    string field2 = 2;
 | 
				
			||||||
 | 
					    string field3 = 3;
 | 
				
			||||||
 | 
					    string field4 = 4;
 | 
				
			||||||
 | 
					    string field5 = 5;
 | 
				
			||||||
 | 
					    string field6 = 6;
 | 
				
			||||||
 | 
					    string field7 = 7;
 | 
				
			||||||
 | 
					    string field8 = 8;
 | 
				
			||||||
 | 
					    string field9 = 9;
 | 
				
			||||||
 | 
					    string field10 = 10;
 | 
				
			||||||
 | 
					    string field11 = 11;
 | 
				
			||||||
 | 
					    string field12 = 12;
 | 
				
			||||||
 | 
					    string field13 = 13;
 | 
				
			||||||
 | 
					    string field14 = 14;
 | 
				
			||||||
 | 
					    string field15 = 15;
 | 
				
			||||||
 | 
					    string field16 = 16;
 | 
				
			||||||
 | 
					    string field17 = 17;
 | 
				
			||||||
 | 
					    string field18 = 18;
 | 
				
			||||||
 | 
					    string field19 = 19;
 | 
				
			||||||
 | 
					    string field20 = 20;
 | 
				
			||||||
 | 
					    string field21 = 21;
 | 
				
			||||||
 | 
					    string field22 = 22;
 | 
				
			||||||
 | 
					    string field23 = 23;
 | 
				
			||||||
 | 
					    string field24 = 24;
 | 
				
			||||||
 | 
					    string field25 = 25;
 | 
				
			||||||
 | 
					    string field26 = 26;
 | 
				
			||||||
 | 
					    string field27 = 27;
 | 
				
			||||||
 | 
					    string field28 = 28;
 | 
				
			||||||
 | 
					    string field29 = 29;
 | 
				
			||||||
 | 
					    string field30 = 30;
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					message Test_Client_Call {
 | 
				
			||||||
 | 
					  message Request {
 | 
				
			||||||
 | 
					    string userId = 1 [json_name = "user_id"];
 | 
				
			||||||
 | 
					    int64  orderId = 2 [json_name = "order_id"];
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					  message Response {
 | 
				
			||||||
 | 
					    string id = 1;
 | 
				
			||||||
 | 
					    string name = 2;
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					  message DefaultError {
 | 
				
			||||||
 | 
					    string code = 1;
 | 
				
			||||||
 | 
					    string msg = 2;
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					  message SpecialError {
 | 
				
			||||||
 | 
					    string code = 1;
 | 
				
			||||||
 | 
					    string msg = 2;
 | 
				
			||||||
 | 
					    string warning = 3;
 | 
				
			||||||
 | 
					  }
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										194
									
								
								builder/query.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										194
									
								
								builder/query.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,194 @@
 | 
				
			|||||||
 | 
					package builder
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"fmt"
 | 
				
			||||||
 | 
						"net/url"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"google.golang.org/protobuf/proto"
 | 
				
			||||||
 | 
						"google.golang.org/protobuf/reflect/protoreflect"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func buildQuery(msg proto.Message, usedFieldsPath *usedFields, usedFieldBody string) (url.Values, error) {
 | 
				
			||||||
 | 
						var (
 | 
				
			||||||
 | 
							query      = url.Values{}
 | 
				
			||||||
 | 
							msgReflect = msg.ProtoReflect()
 | 
				
			||||||
 | 
						)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						fields := msgReflect.Descriptor().Fields()
 | 
				
			||||||
 | 
						for i := 0; i < fields.Len(); i++ {
 | 
				
			||||||
 | 
							var (
 | 
				
			||||||
 | 
								fd        = fields.Get(i)
 | 
				
			||||||
 | 
								fieldName = fd.JSONName()
 | 
				
			||||||
 | 
							)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							if usedFieldsPath.hasFullKey(fieldName) {
 | 
				
			||||||
 | 
								continue
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							if fieldName == usedFieldBody {
 | 
				
			||||||
 | 
								continue
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							val := msgReflect.Get(fd)
 | 
				
			||||||
 | 
							if isZeroValue(val, fd) {
 | 
				
			||||||
 | 
								continue
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							// Note: order of the cases is important!
 | 
				
			||||||
 | 
							switch {
 | 
				
			||||||
 | 
							case fd.IsList():
 | 
				
			||||||
 | 
								if fd.Kind() == protoreflect.MessageKind {
 | 
				
			||||||
 | 
									return nil, fmt.Errorf("repeated message field %s cannot be mapped to URL query parameters", fieldName)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								list := val.List()
 | 
				
			||||||
 | 
								for j := 0; j < list.Len(); j++ {
 | 
				
			||||||
 | 
									strVal, err := stringifyValue(list.Get(j), fd)
 | 
				
			||||||
 | 
									if err != nil {
 | 
				
			||||||
 | 
										return nil, fmt.Errorf("stringify value for query %s: %w", fieldName, err)
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									query.Add(fieldName, strVal)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							case fd.IsMap():
 | 
				
			||||||
 | 
								var (
 | 
				
			||||||
 | 
									m        = val.Map()
 | 
				
			||||||
 | 
									rangeErr error
 | 
				
			||||||
 | 
								)
 | 
				
			||||||
 | 
								m.Range(func(k protoreflect.MapKey, v protoreflect.Value) bool {
 | 
				
			||||||
 | 
									key := fmt.Sprintf("%s.%s", fieldName, k.String())
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
									if fd.MapValue().Kind() == protoreflect.MessageKind {
 | 
				
			||||||
 | 
										flattened, err := flattenMsgForQuery(key, v.Message())
 | 
				
			||||||
 | 
										if err != nil {
 | 
				
			||||||
 | 
											rangeErr = fmt.Errorf("flatten msg for query %s: %w", fieldName, err)
 | 
				
			||||||
 | 
											return false
 | 
				
			||||||
 | 
										}
 | 
				
			||||||
 | 
										for _, item := range flattened {
 | 
				
			||||||
 | 
											if item.val == "" {
 | 
				
			||||||
 | 
												continue
 | 
				
			||||||
 | 
											}
 | 
				
			||||||
 | 
											if usedFieldsPath.hasFullKey(item.key) {
 | 
				
			||||||
 | 
												continue
 | 
				
			||||||
 | 
											}
 | 
				
			||||||
 | 
											query.Add(item.key, item.val)
 | 
				
			||||||
 | 
										}
 | 
				
			||||||
 | 
									} else {
 | 
				
			||||||
 | 
										strVal, err := stringifyValue(v, fd.MapValue())
 | 
				
			||||||
 | 
										if err != nil {
 | 
				
			||||||
 | 
											rangeErr = fmt.Errorf("stringify value for map %s: %w", fieldName, err)
 | 
				
			||||||
 | 
											return false
 | 
				
			||||||
 | 
										}
 | 
				
			||||||
 | 
										query.Add(key, strVal)
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									return true
 | 
				
			||||||
 | 
								})
 | 
				
			||||||
 | 
								if rangeErr != nil {
 | 
				
			||||||
 | 
									return nil, fmt.Errorf("map range error: %w", rangeErr)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							case fd.Kind() == protoreflect.MessageKind:
 | 
				
			||||||
 | 
								flattened, err := flattenMsgForQuery(fieldName, val.Message())
 | 
				
			||||||
 | 
								if err != nil {
 | 
				
			||||||
 | 
									return nil, fmt.Errorf("flatten msg for query %s: %w", fieldName, err)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								for _, item := range flattened {
 | 
				
			||||||
 | 
									if item.val == "" {
 | 
				
			||||||
 | 
										continue
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									if usedFieldsPath.hasFullKey(item.key) {
 | 
				
			||||||
 | 
										continue
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									query.Add(item.key, item.val)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							default:
 | 
				
			||||||
 | 
								strVal, err := stringifyValue(val, fd)
 | 
				
			||||||
 | 
								if err != nil {
 | 
				
			||||||
 | 
									return nil, fmt.Errorf("stringify value for primitive %s: %w", fieldName, err)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								query.Add(fieldName, strVal)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return query, nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					type flattenItem struct {
 | 
				
			||||||
 | 
						key string
 | 
				
			||||||
 | 
						val string
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// flattenMsgForQuery flattens a non-repeated message value under a given prefix.
 | 
				
			||||||
 | 
					func flattenMsgForQuery(prefix string, msg protoreflect.Message) ([]flattenItem, error) {
 | 
				
			||||||
 | 
						var out []flattenItem
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						fields := msg.Descriptor().Fields()
 | 
				
			||||||
 | 
						for i := 0; i < fields.Len(); i++ {
 | 
				
			||||||
 | 
							var (
 | 
				
			||||||
 | 
								fd  = fields.Get(i)
 | 
				
			||||||
 | 
								val = msg.Get(fd)
 | 
				
			||||||
 | 
							)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							if isZeroValue(val, fd) {
 | 
				
			||||||
 | 
								continue
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							key := fmt.Sprintf("%s.%s", prefix, fd.JSONName())
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							switch {
 | 
				
			||||||
 | 
							case fd.IsList():
 | 
				
			||||||
 | 
								if fd.Kind() == protoreflect.MessageKind {
 | 
				
			||||||
 | 
									return nil, fmt.Errorf("repeated message field %s cannot be flattened for query", key)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								list := val.List()
 | 
				
			||||||
 | 
								for j := 0; j < list.Len(); j++ {
 | 
				
			||||||
 | 
									strVal, err := stringifyValue(list.Get(j), fd)
 | 
				
			||||||
 | 
									if err != nil {
 | 
				
			||||||
 | 
										return nil, fmt.Errorf("stringify query %s: %w", key, err)
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									out = append(out, flattenItem{key: key, val: strVal})
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							case fd.Kind() == protoreflect.MessageKind:
 | 
				
			||||||
 | 
								nested, err := flattenMsgForQuery(key, val.Message())
 | 
				
			||||||
 | 
								if err != nil {
 | 
				
			||||||
 | 
									return nil, fmt.Errorf("flatten msg for query %s: %w", key, err)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								out = append(out, nested...)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							case fd.IsMap():
 | 
				
			||||||
 | 
								var mapErr error
 | 
				
			||||||
 | 
								val.Map().Range(func(k protoreflect.MapKey, v protoreflect.Value) bool {
 | 
				
			||||||
 | 
									keyStr := k.String()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
									if fd.MapValue().Kind() == protoreflect.MessageKind {
 | 
				
			||||||
 | 
										child, err := flattenMsgForQuery(keyStr, v.Message())
 | 
				
			||||||
 | 
										if err != nil {
 | 
				
			||||||
 | 
											mapErr = fmt.Errorf("flatten map value %s: %w", key, err)
 | 
				
			||||||
 | 
											return false
 | 
				
			||||||
 | 
										}
 | 
				
			||||||
 | 
										out = append(out, child...)
 | 
				
			||||||
 | 
									} else {
 | 
				
			||||||
 | 
										strVal, err := stringifyValue(v, fd.MapValue())
 | 
				
			||||||
 | 
										if err != nil {
 | 
				
			||||||
 | 
											mapErr = fmt.Errorf("stringify query %s: %w", keyStr, err)
 | 
				
			||||||
 | 
											return false
 | 
				
			||||||
 | 
										}
 | 
				
			||||||
 | 
										out = append(out, flattenItem{key: keyStr, val: strVal})
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									return true
 | 
				
			||||||
 | 
								})
 | 
				
			||||||
 | 
								if mapErr != nil {
 | 
				
			||||||
 | 
									return nil, mapErr
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							default:
 | 
				
			||||||
 | 
								strVal, err := stringifyValue(val, fd)
 | 
				
			||||||
 | 
								if err != nil {
 | 
				
			||||||
 | 
									return nil, fmt.Errorf("stringify query %s: %w", key, err)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								out = append(out, flattenItem{key: key, val: strVal})
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return out, nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										180
									
								
								builder/request_builder.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										180
									
								
								builder/request_builder.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,180 @@
 | 
				
			|||||||
 | 
					// Package builder implements google.api.http-style request building (gRPC JSON transcoding)
 | 
				
			||||||
 | 
					// for HTTP requests, closely following the google.api.http spec.
 | 
				
			||||||
 | 
					// See full spec for details: https://github.com/googleapis/googleapis/blob/master/google/api/http.proto
 | 
				
			||||||
 | 
					package builder
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"errors"
 | 
				
			||||||
 | 
						"fmt"
 | 
				
			||||||
 | 
						"net/http"
 | 
				
			||||||
 | 
						"net/url"
 | 
				
			||||||
 | 
						"strings"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"google.golang.org/protobuf/proto"
 | 
				
			||||||
 | 
						"google.golang.org/protobuf/reflect/protoreflect"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					type RequestBuilder struct {
 | 
				
			||||||
 | 
						path       string        // e.g. "/v1/{name=projects/*/topics/*}:publish" or "/users/{user.id}"
 | 
				
			||||||
 | 
						method     string        // GET, POST, PATCH, etc. (not used in mapping rules, but convenient for callers)
 | 
				
			||||||
 | 
						bodyOption bodyOption    // "", "*", or top-level field name
 | 
				
			||||||
 | 
						msg        proto.Message // request struct
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func NewRequestBuilder(
 | 
				
			||||||
 | 
						path string,
 | 
				
			||||||
 | 
						method string,
 | 
				
			||||||
 | 
						bodyOpt string,
 | 
				
			||||||
 | 
						msg proto.Message,
 | 
				
			||||||
 | 
					) (
 | 
				
			||||||
 | 
						*RequestBuilder,
 | 
				
			||||||
 | 
						error,
 | 
				
			||||||
 | 
					) {
 | 
				
			||||||
 | 
						rb := &RequestBuilder{
 | 
				
			||||||
 | 
							path:       path,
 | 
				
			||||||
 | 
							method:     method,
 | 
				
			||||||
 | 
							bodyOption: bodyOption(bodyOpt),
 | 
				
			||||||
 | 
							msg:        msg,
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if err := rb.validate(); err != nil {
 | 
				
			||||||
 | 
							return nil, fmt.Errorf("validate: %w", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return rb, nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// Build applies mapping rules and returns:
 | 
				
			||||||
 | 
					//
 | 
				
			||||||
 | 
					//	resolvedPath — path with placeholders substituted and query appended
 | 
				
			||||||
 | 
					//	newMsg       — same concrete type as input, filtered to contain only the body fields
 | 
				
			||||||
 | 
					//	err          — if mapping/validation failed
 | 
				
			||||||
 | 
					func (b *RequestBuilder) Build() (resolvedPath string, newMsg proto.Message, err error) {
 | 
				
			||||||
 | 
						tmpl, isCached := getCachedPathTemplate(b.path)
 | 
				
			||||||
 | 
						if !isCached {
 | 
				
			||||||
 | 
							tmpl, err = parsePathTemplate(b.path)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								return "", nil, fmt.Errorf("parse path template: %w", err)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							setPathTemplateCache(b.path, tmpl)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						var usedFieldsPath *usedFields
 | 
				
			||||||
 | 
						resolvedPath, usedFieldsPath, err = resolvePathPlaceholders(tmpl, b.msg)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return "", nil, fmt.Errorf("resolve path placeholders: %w", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// if all set fields are already used in path, no need to process query/body
 | 
				
			||||||
 | 
						if allFieldsUsed(b.msg, usedFieldsPath) {
 | 
				
			||||||
 | 
							return resolvedPath, initZeroMsg(b.msg), nil
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						switch {
 | 
				
			||||||
 | 
						case b.bodyOption.isWithoutBody():
 | 
				
			||||||
 | 
							var query url.Values
 | 
				
			||||||
 | 
							query, err = buildQuery(b.msg, usedFieldsPath, "")
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								return "", nil, fmt.Errorf("build query: %w", err)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							return resolvedPath + encodeQuery(query), initZeroMsg(b.msg), nil
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						case b.bodyOption.isSingleField():
 | 
				
			||||||
 | 
							fieldBody := b.bodyOption.String()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							newMsg, err = buildSingleFieldBody(b.msg, fieldBody)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								return "", nil, fmt.Errorf("build single field body: %w", err)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							var query url.Values
 | 
				
			||||||
 | 
							query, err = buildQuery(b.msg, usedFieldsPath, fieldBody)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								return "", nil, fmt.Errorf("build query: %w", err)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							return resolvedPath + encodeQuery(query), newMsg, nil
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						case b.bodyOption.isFullBody():
 | 
				
			||||||
 | 
							newMsg, err = buildFullBody(b.msg, usedFieldsPath)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								return "", nil, fmt.Errorf("build full body: %w", err)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							return resolvedPath, newMsg, nil
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						default:
 | 
				
			||||||
 | 
							return "", nil, fmt.Errorf("unsupported body option %s", b.bodyOption.String())
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (b *RequestBuilder) validate() error {
 | 
				
			||||||
 | 
						if b.path == "" {
 | 
				
			||||||
 | 
							return errors.New("path is empty")
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if err := validateHTTPMethod(b.method); err != nil {
 | 
				
			||||||
 | 
							return fmt.Errorf("validate http method: %w", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if err := validateHTTPMethodAndBody(b.method, b.bodyOption); err != nil {
 | 
				
			||||||
 | 
							return fmt.Errorf("validate http method and body: %w", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if b.msg == nil {
 | 
				
			||||||
 | 
							return errors.New("msg is nil")
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func validateHTTPMethod(method string) error {
 | 
				
			||||||
 | 
						switch strings.ToUpper(method) {
 | 
				
			||||||
 | 
						case http.MethodGet,
 | 
				
			||||||
 | 
							http.MethodHead,
 | 
				
			||||||
 | 
							http.MethodPost,
 | 
				
			||||||
 | 
							http.MethodPut,
 | 
				
			||||||
 | 
							http.MethodPatch,
 | 
				
			||||||
 | 
							http.MethodDelete,
 | 
				
			||||||
 | 
							http.MethodConnect,
 | 
				
			||||||
 | 
							http.MethodOptions,
 | 
				
			||||||
 | 
							http.MethodTrace:
 | 
				
			||||||
 | 
							return nil
 | 
				
			||||||
 | 
						default:
 | 
				
			||||||
 | 
							return errors.New("invalid http method")
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func validateHTTPMethodAndBody(method string, bodyOpt bodyOption) error {
 | 
				
			||||||
 | 
						switch method {
 | 
				
			||||||
 | 
						case http.MethodGet, http.MethodDelete, http.MethodHead, http.MethodOptions:
 | 
				
			||||||
 | 
							if !bodyOpt.isWithoutBody() {
 | 
				
			||||||
 | 
								return fmt.Errorf("%s method must not have a body", method)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func allFieldsUsed(msg proto.Message, used *usedFields) bool {
 | 
				
			||||||
 | 
						if used.len() == 0 {
 | 
				
			||||||
 | 
							return false
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						count := 0
 | 
				
			||||||
 | 
						msg.ProtoReflect().Range(func(protoreflect.FieldDescriptor, protoreflect.Value) bool {
 | 
				
			||||||
 | 
							count++
 | 
				
			||||||
 | 
							return true
 | 
				
			||||||
 | 
						})
 | 
				
			||||||
 | 
						return used.len() == count
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func encodeQuery(query url.Values) string {
 | 
				
			||||||
 | 
						if len(query) == 0 {
 | 
				
			||||||
 | 
							return ""
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						enc := query.Encode()
 | 
				
			||||||
 | 
						if enc == "" {
 | 
				
			||||||
 | 
							return ""
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return "?" + enc
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func initZeroMsg(msg proto.Message) proto.Message {
 | 
				
			||||||
 | 
						return msg.ProtoReflect().New().Interface()
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										149
									
								
								builder/request_builder_bench_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										149
									
								
								builder/request_builder_bench_test.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,149 @@
 | 
				
			|||||||
 | 
					package builder
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"fmt"
 | 
				
			||||||
 | 
						"math/rand"
 | 
				
			||||||
 | 
						"strings"
 | 
				
			||||||
 | 
						"testing"
 | 
				
			||||||
 | 
						"time"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"google.golang.org/protobuf/proto"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						pb "go.unistack.org/micro-client-http/v3/builder/proto"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// sink prevents the compiler from optimizing away parsePathTemplate results.
 | 
				
			||||||
 | 
					var sink *pathTemplate
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func BenchmarkParsePathTemplate(b *testing.B) {
 | 
				
			||||||
 | 
						r := rand.New(rand.NewSource(1))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						benchInput := func(size int) string {
 | 
				
			||||||
 | 
							sb := strings.Builder{}
 | 
				
			||||||
 | 
							sb.Grow(size * 10)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							for i := 0; i < size; i++ {
 | 
				
			||||||
 | 
								name := fmt.Sprintf("var%d", r.Intn(1000))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								if r.Intn(5) == 0 {
 | 
				
			||||||
 | 
									sb.WriteString(fmt.Sprintf("{%s=**}", name))
 | 
				
			||||||
 | 
								} else {
 | 
				
			||||||
 | 
									sb.WriteString(fmt.Sprintf("{%s}", name))
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							return sb.String()
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						sizes := []int{1_000, 10_000, 50_000, 100_000}
 | 
				
			||||||
 | 
						for _, size := range sizes {
 | 
				
			||||||
 | 
							input := benchInput(size)
 | 
				
			||||||
 | 
							b.Run(fmt.Sprintf("N=%d", size), func(b *testing.B) {
 | 
				
			||||||
 | 
								b.ReportAllocs()
 | 
				
			||||||
 | 
								for i := 0; i < b.N; i++ {
 | 
				
			||||||
 | 
									var err error
 | 
				
			||||||
 | 
									sink, err = parsePathTemplate(input)
 | 
				
			||||||
 | 
									if err != nil && testing.Verbose() {
 | 
				
			||||||
 | 
										b.Fatal(err)
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							})
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func BenchmarkRequestBuilder(b *testing.B) {
 | 
				
			||||||
 | 
						r := rand.New(rand.NewSource(time.Now().UnixNano()))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						makeMsg := func(fieldCount int) proto.Message {
 | 
				
			||||||
 | 
							switch fieldCount {
 | 
				
			||||||
 | 
							case 5:
 | 
				
			||||||
 | 
								return &pb.Benchmark_Case5{
 | 
				
			||||||
 | 
									Field1: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field2: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field3: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field4: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field5: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							case 10:
 | 
				
			||||||
 | 
								return &pb.Benchmark_Case10{
 | 
				
			||||||
 | 
									Field1:  fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field2:  fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field3:  fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field4:  fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field5:  fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field6:  fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field7:  fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field8:  fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field9:  fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field10: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							case 30:
 | 
				
			||||||
 | 
								return &pb.Benchmark_Case30{
 | 
				
			||||||
 | 
									Field1:  fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field2:  fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field3:  fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field4:  fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field5:  fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field6:  fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field7:  fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field8:  fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field9:  fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field10: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field11: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field12: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field13: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field14: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field15: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field16: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field17: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field18: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field19: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field20: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field21: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field22: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field23: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field24: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field25: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field26: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field27: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field28: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field29: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
									Field30: fmt.Sprintf("value%d", r.Intn(1000)),
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							default:
 | 
				
			||||||
 | 
								b.Fatal("undefined field count")
 | 
				
			||||||
 | 
								return nil
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						tests := []struct {
 | 
				
			||||||
 | 
							name       string
 | 
				
			||||||
 | 
							pathTmpl   string
 | 
				
			||||||
 | 
							bodyOption string
 | 
				
			||||||
 | 
						}{
 | 
				
			||||||
 | 
							{"all fields in path", "/resource/{field1}/{field2}", ""},
 | 
				
			||||||
 | 
							{"single field body", "/resource/{field1}", "field4"},
 | 
				
			||||||
 | 
							{"full body", "/resource", "*"},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, fields := range []int{5, 10, 30} {
 | 
				
			||||||
 | 
							for _, tt := range tests {
 | 
				
			||||||
 | 
								b.Run(fmt.Sprintf("%s_%d_fields", tt.name, fields), func(b *testing.B) {
 | 
				
			||||||
 | 
									b.ReportAllocs()
 | 
				
			||||||
 | 
									b.ResetTimer()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
									for i := 0; i < b.N; i++ {
 | 
				
			||||||
 | 
										msg := makeMsg(fields)
 | 
				
			||||||
 | 
										rb, err := NewRequestBuilder(tt.pathTmpl, "POST", tt.bodyOption, msg)
 | 
				
			||||||
 | 
										if err != nil {
 | 
				
			||||||
 | 
											b.Fatalf("new request builder: %v", err)
 | 
				
			||||||
 | 
										}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
										_, _, err = rb.Build()
 | 
				
			||||||
 | 
										if err != nil {
 | 
				
			||||||
 | 
											b.Fatalf("build: %v", err)
 | 
				
			||||||
 | 
										}
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								})
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										1302
									
								
								builder/request_builder_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1302
									
								
								builder/request_builder_test.go
									
									
									
									
									
										Normal file
									
								
							
										
											
												File diff suppressed because it is too large
												Load Diff
											
										
									
								
							
							
								
								
									
										44
									
								
								builder/used_fields.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										44
									
								
								builder/used_fields.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,44 @@
 | 
				
			|||||||
 | 
					package builder
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import "strings"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// usedFields stores keys and their top-level parts,
 | 
				
			||||||
 | 
					// turning top-level lookups from O(N) into O(1).
 | 
				
			||||||
 | 
					type usedFields struct {
 | 
				
			||||||
 | 
						full map[string]struct{}
 | 
				
			||||||
 | 
						top  map[string]struct{}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func newUsedFields() *usedFields {
 | 
				
			||||||
 | 
						return &usedFields{
 | 
				
			||||||
 | 
							full: make(map[string]struct{}),
 | 
				
			||||||
 | 
							top:  make(map[string]struct{}),
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// add inserts a new key and updates the top-level index.
 | 
				
			||||||
 | 
					func (u *usedFields) add(key string) {
 | 
				
			||||||
 | 
						u.full[key] = struct{}{}
 | 
				
			||||||
 | 
						top := key
 | 
				
			||||||
 | 
						if i := strings.IndexByte(key, '.'); i != -1 {
 | 
				
			||||||
 | 
							top = key[:i]
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						u.top[top] = struct{}{}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// hasTopLevelKey checks if a top-level key exists.
 | 
				
			||||||
 | 
					func (u *usedFields) hasTopLevelKey(top string) bool {
 | 
				
			||||||
 | 
						_, ok := u.top[top]
 | 
				
			||||||
 | 
						return ok
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// hasFullKey checks if an exact key exists.
 | 
				
			||||||
 | 
					func (u *usedFields) hasFullKey(key string) bool {
 | 
				
			||||||
 | 
						_, ok := u.full[key]
 | 
				
			||||||
 | 
						return ok
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// len returns the number of full keys stored in the set.
 | 
				
			||||||
 | 
					func (u *usedFields) len() int {
 | 
				
			||||||
 | 
						return len(u.full)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										78
									
								
								builder/used_fields_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										78
									
								
								builder/used_fields_test.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,78 @@
 | 
				
			|||||||
 | 
					package builder
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"testing"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"github.com/stretchr/testify/require"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestNewUsedFields(t *testing.T) {
 | 
				
			||||||
 | 
						u := newUsedFields()
 | 
				
			||||||
 | 
						require.NotNil(t, u)
 | 
				
			||||||
 | 
						require.NotNil(t, u.full)
 | 
				
			||||||
 | 
						require.NotNil(t, u.top)
 | 
				
			||||||
 | 
						require.Len(t, u.full, 0)
 | 
				
			||||||
 | 
						require.Len(t, u.top, 0)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestUsedFields_Add(t *testing.T) {
 | 
				
			||||||
 | 
						u := newUsedFields()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						u.add("user.name")
 | 
				
			||||||
 | 
						u.add("profile")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						_, ok := u.full["user.name"]
 | 
				
			||||||
 | 
						require.True(t, ok)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						_, ok = u.full["profile"]
 | 
				
			||||||
 | 
						require.True(t, ok)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						_, ok = u.top["user"]
 | 
				
			||||||
 | 
						require.True(t, ok)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						_, ok = u.top["profile"]
 | 
				
			||||||
 | 
						require.True(t, ok)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						require.Len(t, u.full, 2)
 | 
				
			||||||
 | 
						require.Len(t, u.top, 2)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestUsedFields_HasFullKey(t *testing.T) {
 | 
				
			||||||
 | 
						u := newUsedFields()
 | 
				
			||||||
 | 
						u.add("user.name")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						require.True(t, u.hasFullKey("user.name"))
 | 
				
			||||||
 | 
						require.False(t, u.hasFullKey("user.email"))
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestUsedFields_HasTopLevelKey(t *testing.T) {
 | 
				
			||||||
 | 
						u := newUsedFields()
 | 
				
			||||||
 | 
						u.add("user.name")
 | 
				
			||||||
 | 
						u.add("settings.theme")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						require.True(t, u.hasTopLevelKey("user"))
 | 
				
			||||||
 | 
						require.True(t, u.hasTopLevelKey("settings"))
 | 
				
			||||||
 | 
						require.False(t, u.hasTopLevelKey("profile"))
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestUsedFields_AddDuplicate(t *testing.T) {
 | 
				
			||||||
 | 
						u := newUsedFields()
 | 
				
			||||||
 | 
						u.add("user.name")
 | 
				
			||||||
 | 
						u.add("user.name")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						require.True(t, u.hasFullKey("user.name"))
 | 
				
			||||||
 | 
						require.True(t, u.hasTopLevelKey("user"))
 | 
				
			||||||
 | 
						require.Len(t, u.full, 1)
 | 
				
			||||||
 | 
						require.Len(t, u.top, 1)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestUsedFields_Len(t *testing.T) {
 | 
				
			||||||
 | 
						u := newUsedFields()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						u.add("user.name")
 | 
				
			||||||
 | 
						u.add("profile")
 | 
				
			||||||
 | 
						u.add("user.name")
 | 
				
			||||||
 | 
						u.add("profile")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						require.Equal(t, u.len(), 2)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										122
									
								
								client.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										122
									
								
								client.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,122 @@
 | 
				
			|||||||
 | 
					package http
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"context"
 | 
				
			||||||
 | 
						"net/http"
 | 
				
			||||||
 | 
						"strconv"
 | 
				
			||||||
 | 
						"sync"
 | 
				
			||||||
 | 
						"time"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"go.unistack.org/micro/v3/client"
 | 
				
			||||||
 | 
						"go.unistack.org/micro/v3/errors"
 | 
				
			||||||
 | 
						"go.unistack.org/micro/v3/options"
 | 
				
			||||||
 | 
						"go.unistack.org/micro/v3/semconv"
 | 
				
			||||||
 | 
						"go.unistack.org/micro/v3/tracer"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					var DefaultContentType = "application/json"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					type Client struct {
 | 
				
			||||||
 | 
						funcCall   client.FuncCall
 | 
				
			||||||
 | 
						funcStream client.FuncStream
 | 
				
			||||||
 | 
						httpClient *http.Client
 | 
				
			||||||
 | 
						opts       client.Options
 | 
				
			||||||
 | 
						mu         sync.RWMutex
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func NewClient(opts ...client.Option) *Client {
 | 
				
			||||||
 | 
						clientOpts := client.NewOptions(opts...)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if len(clientOpts.ContentType) == 0 {
 | 
				
			||||||
 | 
							clientOpts.ContentType = DefaultContentType
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						c := &Client{opts: clientOpts}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						dialer, ok := httpDialerFromOpts(clientOpts)
 | 
				
			||||||
 | 
						if !ok {
 | 
				
			||||||
 | 
							dialer = defaultHTTPDialer()
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						c.httpClient, ok = httpClientFromOpts(clientOpts)
 | 
				
			||||||
 | 
						if !ok {
 | 
				
			||||||
 | 
							c.httpClient = defaultHTTPClient(dialer, clientOpts.TLSConfig)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						c.funcCall = c.fnCall
 | 
				
			||||||
 | 
						c.funcStream = c.fnStream
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return c
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (c *Client) Name() string {
 | 
				
			||||||
 | 
						return c.opts.Name
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (c *Client) Init(opts ...client.Option) error {
 | 
				
			||||||
 | 
						for _, o := range opts {
 | 
				
			||||||
 | 
							o(&c.opts)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						c.opts.Hooks.EachPrev(func(hook options.Hook) {
 | 
				
			||||||
 | 
							switch h := hook.(type) {
 | 
				
			||||||
 | 
							case client.HookCall:
 | 
				
			||||||
 | 
								c.funcCall = h(c.funcCall)
 | 
				
			||||||
 | 
							case client.HookStream:
 | 
				
			||||||
 | 
								c.funcStream = h(c.funcStream)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (c *Client) Options() client.Options {
 | 
				
			||||||
 | 
						return c.opts
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (c *Client) NewRequest(service, method string, req any, opts ...client.RequestOption) client.Request {
 | 
				
			||||||
 | 
						reqOpts := client.NewRequestOptions(opts...)
 | 
				
			||||||
 | 
						if reqOpts.ContentType == "" {
 | 
				
			||||||
 | 
							reqOpts.ContentType = c.opts.ContentType
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return &httpRequest{
 | 
				
			||||||
 | 
							service: service,
 | 
				
			||||||
 | 
							method:  method,
 | 
				
			||||||
 | 
							request: req,
 | 
				
			||||||
 | 
							opts:    reqOpts,
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (c *Client) Call(ctx context.Context, req client.Request, rsp any, opts ...client.CallOption) error {
 | 
				
			||||||
 | 
						ts := time.Now()
 | 
				
			||||||
 | 
						c.opts.Meter.Counter(semconv.ClientRequestInflight, "endpoint", req.Endpoint()).Inc()
 | 
				
			||||||
 | 
						var sp tracer.Span
 | 
				
			||||||
 | 
						ctx, sp = c.opts.Tracer.Start(ctx, req.Endpoint()+" rpc-client",
 | 
				
			||||||
 | 
							tracer.WithSpanKind(tracer.SpanKindClient),
 | 
				
			||||||
 | 
							tracer.WithSpanLabels("endpoint", req.Endpoint()),
 | 
				
			||||||
 | 
						)
 | 
				
			||||||
 | 
						err := c.funcCall(ctx, req, rsp, opts...)
 | 
				
			||||||
 | 
						c.opts.Meter.Counter(semconv.ClientRequestInflight, "endpoint", req.Endpoint()).Dec()
 | 
				
			||||||
 | 
						te := time.Since(ts)
 | 
				
			||||||
 | 
						c.opts.Meter.Summary(semconv.ClientRequestLatencyMicroseconds, "endpoint", req.Endpoint()).Update(te.Seconds())
 | 
				
			||||||
 | 
						c.opts.Meter.Histogram(semconv.ClientRequestDurationSeconds, "endpoint", req.Endpoint()).Update(te.Seconds())
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if me := errors.FromError(err); me == nil {
 | 
				
			||||||
 | 
							sp.Finish()
 | 
				
			||||||
 | 
							c.opts.Meter.Counter(semconv.ClientRequestTotal, "endpoint", req.Endpoint(), "status", "success", "code", strconv.Itoa(int(200))).Inc()
 | 
				
			||||||
 | 
						} else {
 | 
				
			||||||
 | 
							sp.SetStatus(tracer.SpanStatusError, err.Error())
 | 
				
			||||||
 | 
							c.opts.Meter.Counter(semconv.ClientRequestTotal, "endpoint", req.Endpoint(), "status", "failure", "code", strconv.Itoa(int(me.Code))).Inc()
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return err
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (c *Client) Stream(ctx context.Context, req client.Request, opts ...client.CallOption) (client.Stream, error) {
 | 
				
			||||||
 | 
						return c.funcStream(ctx, req, opts...)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (c *Client) String() string {
 | 
				
			||||||
 | 
						return "http"
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										261
									
								
								client_helpers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										261
									
								
								client_helpers.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,261 @@
 | 
				
			|||||||
 | 
					package http
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"bytes"
 | 
				
			||||||
 | 
						"context"
 | 
				
			||||||
 | 
						"errors"
 | 
				
			||||||
 | 
						"fmt"
 | 
				
			||||||
 | 
						"io"
 | 
				
			||||||
 | 
						"net/http"
 | 
				
			||||||
 | 
						"net/url"
 | 
				
			||||||
 | 
						"strings"
 | 
				
			||||||
 | 
						"time"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"go.unistack.org/micro/v3/client"
 | 
				
			||||||
 | 
						"go.unistack.org/micro/v3/codec"
 | 
				
			||||||
 | 
						"go.unistack.org/micro/v3/logger"
 | 
				
			||||||
 | 
						"go.unistack.org/micro/v3/metadata"
 | 
				
			||||||
 | 
						"google.golang.org/protobuf/proto"
 | 
				
			||||||
 | 
						"google.golang.org/protobuf/reflect/protoreflect"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"go.unistack.org/micro-client-http/v3/builder"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func buildHTTPRequest(
 | 
				
			||||||
 | 
						ctx context.Context,
 | 
				
			||||||
 | 
						addr string,
 | 
				
			||||||
 | 
						path string,
 | 
				
			||||||
 | 
						ct string,
 | 
				
			||||||
 | 
						cf codec.Codec,
 | 
				
			||||||
 | 
						msg any,
 | 
				
			||||||
 | 
						opts client.CallOptions,
 | 
				
			||||||
 | 
						log logger.Logger,
 | 
				
			||||||
 | 
					) (
 | 
				
			||||||
 | 
						*http.Request,
 | 
				
			||||||
 | 
						error,
 | 
				
			||||||
 | 
					) {
 | 
				
			||||||
 | 
						protoMsg, ok := msg.(proto.Message)
 | 
				
			||||||
 | 
						if !ok {
 | 
				
			||||||
 | 
							return nil, errors.New("msg must be a proto message type")
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						var (
 | 
				
			||||||
 | 
							method  = http.MethodPost
 | 
				
			||||||
 | 
							bodyOpt string
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							parameters = map[string]map[string]string{}
 | 
				
			||||||
 | 
						)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if opts.Context != nil {
 | 
				
			||||||
 | 
							if v, ok := methodFromOpts(opts); ok {
 | 
				
			||||||
 | 
								method = v
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							if v, ok := pathFromOpts(opts); ok {
 | 
				
			||||||
 | 
								path = v
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							if v, ok := bodyFromOpts(opts); ok {
 | 
				
			||||||
 | 
								bodyOpt = v
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							if h, ok := headerFromOpts(opts); ok && len(h) > 0 {
 | 
				
			||||||
 | 
								m, ok := parameters["header"]
 | 
				
			||||||
 | 
								if !ok {
 | 
				
			||||||
 | 
									m = make(map[string]string)
 | 
				
			||||||
 | 
									parameters["header"] = m
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								for idx := 0; idx+1 < len(h); idx += 2 {
 | 
				
			||||||
 | 
									m[h[idx]] = h[idx+1]
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							if c, ok := cookieFromOpts(opts); ok && len(c) > 0 {
 | 
				
			||||||
 | 
								m, ok := parameters["cookie"]
 | 
				
			||||||
 | 
								if !ok {
 | 
				
			||||||
 | 
									m = make(map[string]string)
 | 
				
			||||||
 | 
									parameters["cookie"] = m
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								for idx := 0; idx+1 < len(c); idx += 2 {
 | 
				
			||||||
 | 
									m[c[idx]] = c[idx+1]
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						reqBuilder, err := builder.NewRequestBuilder(path, method, bodyOpt, protoMsg)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return nil, fmt.Errorf("new request builder: %w", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						resolvedPath, newMsg, err := reqBuilder.Build()
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return nil, fmt.Errorf("build request: %w", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						resolvedURL := joinURL(addr, resolvedPath)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						u, err := normalizeURL(resolvedURL)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return nil, fmt.Errorf("normalize url: %w", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						body, err := marshallMsg(cf, newMsg)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return nil, fmt.Errorf("marshal msg: %w", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						var hreq *http.Request
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if len(body) > 0 {
 | 
				
			||||||
 | 
							hreq, err = http.NewRequestWithContext(ctx, method, u.String(), io.NopCloser(bytes.NewBuffer(body)))
 | 
				
			||||||
 | 
							hreq.ContentLength = int64(len(body))
 | 
				
			||||||
 | 
						} else {
 | 
				
			||||||
 | 
							hreq, err = http.NewRequestWithContext(ctx, method, u.String(), nil)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return nil, fmt.Errorf("new http request: %w", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						setHeadersAndCookies(ctx, hreq, ct, opts)
 | 
				
			||||||
 | 
						if err = validateHeadersAndCookies(hreq, parameters); err != nil {
 | 
				
			||||||
 | 
							return nil, fmt.Errorf("validate headers and cookies: %w", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if log.V(logger.DebugLevel) {
 | 
				
			||||||
 | 
							log.Debug(
 | 
				
			||||||
 | 
								ctx,
 | 
				
			||||||
 | 
								fmt.Sprintf("request %s to %s with headers %v body %s", method, u.String(), hreq.Header, body),
 | 
				
			||||||
 | 
							)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return hreq, nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func joinURL(addr, resolvedPath string) string {
 | 
				
			||||||
 | 
						if addr == "" {
 | 
				
			||||||
 | 
							return resolvedPath
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if resolvedPath == "" {
 | 
				
			||||||
 | 
							return addr
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						switch {
 | 
				
			||||||
 | 
						case strings.HasSuffix(addr, "/") && strings.HasPrefix(resolvedPath, "/"):
 | 
				
			||||||
 | 
							return addr + resolvedPath[1:]
 | 
				
			||||||
 | 
						case !strings.HasSuffix(addr, "/") && !strings.HasPrefix(resolvedPath, "/"):
 | 
				
			||||||
 | 
							return addr + "/" + resolvedPath
 | 
				
			||||||
 | 
						default:
 | 
				
			||||||
 | 
							return addr + resolvedPath
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func normalizeURL(raw string) (*url.URL, error) {
 | 
				
			||||||
 | 
						if !strings.Contains(raw, "://") {
 | 
				
			||||||
 | 
							raw = "http://" + raw
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						u, err := url.Parse(raw)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return nil, fmt.Errorf("parse url: %w", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if u.Scheme != "http" && u.Scheme != "https" {
 | 
				
			||||||
 | 
							return nil, fmt.Errorf("invalid scheme: %q (must be http or https)", u.Scheme)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if u.Host == "" {
 | 
				
			||||||
 | 
							return nil, errors.New("missing host in url")
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return u, nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func marshallMsg(cf codec.Codec, msg proto.Message) ([]byte, error) {
 | 
				
			||||||
 | 
						if msg == nil {
 | 
				
			||||||
 | 
							return nil, nil
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						isEmpty := true
 | 
				
			||||||
 | 
						msg.ProtoReflect().Range(func(protoreflect.FieldDescriptor, protoreflect.Value) bool {
 | 
				
			||||||
 | 
							isEmpty = false
 | 
				
			||||||
 | 
							return false
 | 
				
			||||||
 | 
						})
 | 
				
			||||||
 | 
						if isEmpty {
 | 
				
			||||||
 | 
							return nil, nil
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return cf.Marshal(msg)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func setHeadersAndCookies(ctx context.Context, r *http.Request, ct string, opts client.CallOptions) {
 | 
				
			||||||
 | 
						r.Header = make(http.Header)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						r.Header.Set(metadata.HeaderContentType, ct)
 | 
				
			||||||
 | 
						r.Header.Set("Content-Length", fmt.Sprintf("%d", r.ContentLength))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if opts.AuthToken != "" {
 | 
				
			||||||
 | 
							r.Header.Set(metadata.HeaderAuthorization, opts.AuthToken)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if opts.StreamTimeout > time.Duration(0) {
 | 
				
			||||||
 | 
							r.Header.Set(metadata.HeaderTimeout, fmt.Sprintf("%d", opts.StreamTimeout))
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if opts.RequestTimeout > time.Duration(0) {
 | 
				
			||||||
 | 
							r.Header.Set(metadata.HeaderTimeout, fmt.Sprintf("%d", opts.RequestTimeout))
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if opts.RequestMetadata != nil {
 | 
				
			||||||
 | 
							for k, v := range opts.RequestMetadata {
 | 
				
			||||||
 | 
								if k == "Cookie" {
 | 
				
			||||||
 | 
									applyCookies(r, v)
 | 
				
			||||||
 | 
									continue
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								r.Header[k] = append(r.Header[k], v)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if md, ok := metadata.FromOutgoingContext(ctx); ok {
 | 
				
			||||||
 | 
							for k, v := range md {
 | 
				
			||||||
 | 
								if k == "Cookie" {
 | 
				
			||||||
 | 
									applyCookies(r, v)
 | 
				
			||||||
 | 
									continue
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								r.Header[k] = append(r.Header[k], v)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func applyCookies(r *http.Request, rawCookies string) {
 | 
				
			||||||
 | 
						if len(rawCookies) == 0 {
 | 
				
			||||||
 | 
							return
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						tmp := http.Request{Header: http.Header{}}
 | 
				
			||||||
 | 
						tmp.Header.Set("Cookie", rawCookies)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, c := range tmp.Cookies() {
 | 
				
			||||||
 | 
							r.AddCookie(c)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func validateHeadersAndCookies(r *http.Request, parameters map[string]map[string]string) error {
 | 
				
			||||||
 | 
						if headers, ok := parameters["header"]; ok {
 | 
				
			||||||
 | 
							for name, required := range headers {
 | 
				
			||||||
 | 
								if required == "true" && r.Header.Get(name) == "" {
 | 
				
			||||||
 | 
									return fmt.Errorf("missing required header: %s", name)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if cookies, ok := parameters["cookie"]; ok {
 | 
				
			||||||
 | 
							cookieMap := map[string]string{}
 | 
				
			||||||
 | 
							for _, c := range r.Cookies() {
 | 
				
			||||||
 | 
								cookieMap[c.Name] = c.Value
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							for name, required := range cookies {
 | 
				
			||||||
 | 
								if required == "true" {
 | 
				
			||||||
 | 
									if _, ok := cookieMap[name]; !ok {
 | 
				
			||||||
 | 
										return fmt.Errorf("missing required cookie: %s", name)
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										401
									
								
								client_helpers_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										401
									
								
								client_helpers_test.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,401 @@
 | 
				
			|||||||
 | 
					package http
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"net/http"
 | 
				
			||||||
 | 
						"net/http/httptest"
 | 
				
			||||||
 | 
						"testing"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"github.com/stretchr/testify/require"
 | 
				
			||||||
 | 
						jsoncodec "go.unistack.org/micro-codec-json/v3"
 | 
				
			||||||
 | 
						"google.golang.org/protobuf/proto"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						pb "go.unistack.org/micro-client-http/v3/builder/proto"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestJoinURL(t *testing.T) {
 | 
				
			||||||
 | 
						tests := []struct {
 | 
				
			||||||
 | 
							name string
 | 
				
			||||||
 | 
							addr string
 | 
				
			||||||
 | 
							path string
 | 
				
			||||||
 | 
							want string
 | 
				
			||||||
 | 
						}{
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name: "both without slash",
 | 
				
			||||||
 | 
								addr: "http://example.com",
 | 
				
			||||||
 | 
								path: "api/v1",
 | 
				
			||||||
 | 
								want: "http://example.com/api/v1",
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name: "addr with slash, path without slash",
 | 
				
			||||||
 | 
								addr: "http://example.com/",
 | 
				
			||||||
 | 
								path: "api/v1",
 | 
				
			||||||
 | 
								want: "http://example.com/api/v1",
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name: "addr without slash, path with slash",
 | 
				
			||||||
 | 
								addr: "http://example.com",
 | 
				
			||||||
 | 
								path: "/api/v1",
 | 
				
			||||||
 | 
								want: "http://example.com/api/v1",
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name: "both with slash",
 | 
				
			||||||
 | 
								addr: "http://example.com/",
 | 
				
			||||||
 | 
								path: "/api/v1",
 | 
				
			||||||
 | 
								want: "http://example.com/api/v1",
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name: "empty addr",
 | 
				
			||||||
 | 
								addr: "",
 | 
				
			||||||
 | 
								path: "/api/v1",
 | 
				
			||||||
 | 
								want: "/api/v1",
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name: "empty path",
 | 
				
			||||||
 | 
								addr: "http://example.com",
 | 
				
			||||||
 | 
								path: "",
 | 
				
			||||||
 | 
								want: "http://example.com",
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name: "both empty",
 | 
				
			||||||
 | 
								addr: "",
 | 
				
			||||||
 | 
								path: "",
 | 
				
			||||||
 | 
								want: "",
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, tt := range tests {
 | 
				
			||||||
 | 
							t.Run(tt.name, func(t *testing.T) {
 | 
				
			||||||
 | 
								require.Equal(t, tt.want, joinURL(tt.addr, tt.path))
 | 
				
			||||||
 | 
							})
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestNormalizeURL(t *testing.T) {
 | 
				
			||||||
 | 
						tests := []struct {
 | 
				
			||||||
 | 
							name    string
 | 
				
			||||||
 | 
							input   string
 | 
				
			||||||
 | 
							want    string
 | 
				
			||||||
 | 
							wantErr bool
 | 
				
			||||||
 | 
						}{
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:    "host with port",
 | 
				
			||||||
 | 
								input:   "localhost:8080",
 | 
				
			||||||
 | 
								want:    "http://localhost:8080",
 | 
				
			||||||
 | 
								wantErr: false,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:    "http with host",
 | 
				
			||||||
 | 
								input:   "http://example.com",
 | 
				
			||||||
 | 
								want:    "http://example.com",
 | 
				
			||||||
 | 
								wantErr: false,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:    "http with no host",
 | 
				
			||||||
 | 
								input:   "http://",
 | 
				
			||||||
 | 
								want:    "",
 | 
				
			||||||
 | 
								wantErr: true,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:    "https with host",
 | 
				
			||||||
 | 
								input:   "https://example.com",
 | 
				
			||||||
 | 
								want:    "https://example.com",
 | 
				
			||||||
 | 
								wantErr: false,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:    "https with no host",
 | 
				
			||||||
 | 
								input:   "https://",
 | 
				
			||||||
 | 
								want:    "",
 | 
				
			||||||
 | 
								wantErr: true,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:    "invalid scheme",
 | 
				
			||||||
 | 
								input:   "ftp://example.com",
 | 
				
			||||||
 | 
								want:    "",
 | 
				
			||||||
 | 
								wantErr: true,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:    "IPv4 without scheme",
 | 
				
			||||||
 | 
								input:   "127.0.0.1:9000",
 | 
				
			||||||
 | 
								want:    "http://127.0.0.1:9000",
 | 
				
			||||||
 | 
								wantErr: false,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:    "IPv4 with scheme",
 | 
				
			||||||
 | 
								input:   "http://127.0.0.1:8080",
 | 
				
			||||||
 | 
								want:    "http://127.0.0.1:8080",
 | 
				
			||||||
 | 
								wantErr: false,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:    "IPv6 without scheme",
 | 
				
			||||||
 | 
								input:   "[::1]:8080",
 | 
				
			||||||
 | 
								want:    "http://[::1]:8080",
 | 
				
			||||||
 | 
								wantErr: false,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:    "IPv6 with scheme",
 | 
				
			||||||
 | 
								input:   "https://[::1]:443",
 | 
				
			||||||
 | 
								want:    "https://[::1]:443",
 | 
				
			||||||
 | 
								wantErr: false,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:    "hostname only",
 | 
				
			||||||
 | 
								input:   "my-service",
 | 
				
			||||||
 | 
								want:    "http://my-service",
 | 
				
			||||||
 | 
								wantErr: false,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:    "hostname with path",
 | 
				
			||||||
 | 
								input:   "service.local/api/v1",
 | 
				
			||||||
 | 
								want:    "http://service.local/api/v1",
 | 
				
			||||||
 | 
								wantErr: false,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:    "hostname with dash and port",
 | 
				
			||||||
 | 
								input:   "api-service.local:8080",
 | 
				
			||||||
 | 
								want:    "http://api-service.local:8080",
 | 
				
			||||||
 | 
								wantErr: false,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:    "just path",
 | 
				
			||||||
 | 
								input:   "/api/v1",
 | 
				
			||||||
 | 
								want:    "",
 | 
				
			||||||
 | 
								wantErr: true,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:    "empty string",
 | 
				
			||||||
 | 
								input:   "",
 | 
				
			||||||
 | 
								want:    "",
 | 
				
			||||||
 | 
								wantErr: true,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:    "http with query params",
 | 
				
			||||||
 | 
								input:   "http://example.com?x=1&y=2",
 | 
				
			||||||
 | 
								want:    "http://example.com?x=1&y=2",
 | 
				
			||||||
 | 
								wantErr: false,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:    "http with fragment",
 | 
				
			||||||
 | 
								input:   "http://example.com/path#section1",
 | 
				
			||||||
 | 
								want:    "http://example.com/path#section1",
 | 
				
			||||||
 | 
								wantErr: false,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, tt := range tests {
 | 
				
			||||||
 | 
							t.Run(tt.name, func(t *testing.T) {
 | 
				
			||||||
 | 
								result, err := normalizeURL(tt.input)
 | 
				
			||||||
 | 
								if tt.wantErr {
 | 
				
			||||||
 | 
									require.Error(t, err)
 | 
				
			||||||
 | 
									require.Nil(t, result)
 | 
				
			||||||
 | 
								} else {
 | 
				
			||||||
 | 
									require.NoError(t, err)
 | 
				
			||||||
 | 
									require.Equal(t, tt.want, result.String())
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							})
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestMarshallMsg(t *testing.T) {
 | 
				
			||||||
 | 
						type request = pb.Test_Client_Call_Request
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						tests := []struct {
 | 
				
			||||||
 | 
							name     string
 | 
				
			||||||
 | 
							msg      proto.Message
 | 
				
			||||||
 | 
							expected string
 | 
				
			||||||
 | 
						}{
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:     "empty",
 | 
				
			||||||
 | 
								msg:      &request{},
 | 
				
			||||||
 | 
								expected: "",
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:     "nil",
 | 
				
			||||||
 | 
								msg:      nil,
 | 
				
			||||||
 | 
								expected: "",
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:     "valid",
 | 
				
			||||||
 | 
								msg:      &request{UserId: "123", OrderId: 456},
 | 
				
			||||||
 | 
								expected: `{"userId":"123","orderId":456}`,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, tt := range tests {
 | 
				
			||||||
 | 
							t.Run(tt.name, func(t *testing.T) {
 | 
				
			||||||
 | 
								result, err := marshallMsg(jsoncodec.NewCodec(), tt.msg)
 | 
				
			||||||
 | 
								require.NoError(t, err)
 | 
				
			||||||
 | 
								require.Equal(t, tt.expected, string(result))
 | 
				
			||||||
 | 
							})
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestApplyCookies(t *testing.T) {
 | 
				
			||||||
 | 
						tests := []struct {
 | 
				
			||||||
 | 
							name       string
 | 
				
			||||||
 | 
							rawCookies string
 | 
				
			||||||
 | 
							want       []*http.Cookie
 | 
				
			||||||
 | 
						}{
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:       "empty",
 | 
				
			||||||
 | 
								rawCookies: "",
 | 
				
			||||||
 | 
								want:       []*http.Cookie{},
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:       "single cookie",
 | 
				
			||||||
 | 
								rawCookies: "session=abc123",
 | 
				
			||||||
 | 
								want: []*http.Cookie{
 | 
				
			||||||
 | 
									{Name: "session", Value: "abc123"},
 | 
				
			||||||
 | 
								},
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:       "multiple cookies separate items",
 | 
				
			||||||
 | 
								rawCookies: "session=abc123; user=john",
 | 
				
			||||||
 | 
								want: []*http.Cookie{
 | 
				
			||||||
 | 
									{Name: "session", Value: "abc123"},
 | 
				
			||||||
 | 
									{Name: "user", Value: "john"},
 | 
				
			||||||
 | 
								},
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:       "multiple cookies in one item",
 | 
				
			||||||
 | 
								rawCookies: "a=1; b=2",
 | 
				
			||||||
 | 
								want: []*http.Cookie{
 | 
				
			||||||
 | 
									{Name: "a", Value: "1"},
 | 
				
			||||||
 | 
									{Name: "b", Value: "2"},
 | 
				
			||||||
 | 
								},
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:       "mix of combined and separate cookies",
 | 
				
			||||||
 | 
								rawCookies: "a=1; b=2; c=3",
 | 
				
			||||||
 | 
								want: []*http.Cookie{
 | 
				
			||||||
 | 
									{Name: "a", Value: "1"},
 | 
				
			||||||
 | 
									{Name: "b", Value: "2"},
 | 
				
			||||||
 | 
									{Name: "c", Value: "3"},
 | 
				
			||||||
 | 
								},
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:       "duplicate cookies",
 | 
				
			||||||
 | 
								rawCookies: "session=abc123; session=xyz",
 | 
				
			||||||
 | 
								want: []*http.Cookie{
 | 
				
			||||||
 | 
									{Name: "session", Value: "abc123"},
 | 
				
			||||||
 | 
									{Name: "session", Value: "xyz"},
 | 
				
			||||||
 | 
								},
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:       "cookie with spaces",
 | 
				
			||||||
 | 
								rawCookies: "token=abc 123",
 | 
				
			||||||
 | 
								want: []*http.Cookie{
 | 
				
			||||||
 | 
									{Name: "token", Value: "abc 123", Quoted: true},
 | 
				
			||||||
 | 
								},
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, tt := range tests {
 | 
				
			||||||
 | 
							t.Run(tt.name, func(t *testing.T) {
 | 
				
			||||||
 | 
								req := httptest.NewRequest("GET", "/", nil)
 | 
				
			||||||
 | 
								applyCookies(req, tt.rawCookies)
 | 
				
			||||||
 | 
								require.Equal(t, tt.want, req.Cookies())
 | 
				
			||||||
 | 
							})
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestValidateHeadersAndCookies(t *testing.T) {
 | 
				
			||||||
 | 
						tests := []struct {
 | 
				
			||||||
 | 
							name           string
 | 
				
			||||||
 | 
							prepareRequest func() *http.Request
 | 
				
			||||||
 | 
							parameters     map[string]map[string]string
 | 
				
			||||||
 | 
							wantErr        bool
 | 
				
			||||||
 | 
						}{
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name: "all required headers and cookies present",
 | 
				
			||||||
 | 
								prepareRequest: func() *http.Request {
 | 
				
			||||||
 | 
									req := httptest.NewRequest("GET", "/", nil)
 | 
				
			||||||
 | 
									req.Header.Set("My-Header-1", "Header-Value-1")
 | 
				
			||||||
 | 
									req.Header.Set("My-Header-2", "Header-Value-2")
 | 
				
			||||||
 | 
									req.AddCookie(&http.Cookie{Name: "session-1", Value: "abc-1"})
 | 
				
			||||||
 | 
									req.AddCookie(&http.Cookie{Name: "session-2", Value: "abc-2"})
 | 
				
			||||||
 | 
									return req
 | 
				
			||||||
 | 
								},
 | 
				
			||||||
 | 
								parameters: map[string]map[string]string{
 | 
				
			||||||
 | 
									"header": {"My-Header-1": "true", "My-Header-2": "true"},
 | 
				
			||||||
 | 
									"cookie": {"session-1": "true", "session-2": "true"},
 | 
				
			||||||
 | 
								},
 | 
				
			||||||
 | 
								wantErr: false,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name: "missing required header",
 | 
				
			||||||
 | 
								prepareRequest: func() *http.Request {
 | 
				
			||||||
 | 
									req := httptest.NewRequest("GET", "/", nil)
 | 
				
			||||||
 | 
									req.Header.Set("My-Header-1", "Header-Value-1")
 | 
				
			||||||
 | 
									req.AddCookie(&http.Cookie{Name: "session-1", Value: "abc-1"})
 | 
				
			||||||
 | 
									req.AddCookie(&http.Cookie{Name: "session-2", Value: "abc-2"})
 | 
				
			||||||
 | 
									return req
 | 
				
			||||||
 | 
								},
 | 
				
			||||||
 | 
								parameters: map[string]map[string]string{
 | 
				
			||||||
 | 
									"header": {"My-Header-1": "true", "My-Header-2": "true"},
 | 
				
			||||||
 | 
									"cookie": {"session-1": "true", "session-2": "true"},
 | 
				
			||||||
 | 
								},
 | 
				
			||||||
 | 
								wantErr: true,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name: "missing required cookie",
 | 
				
			||||||
 | 
								prepareRequest: func() *http.Request {
 | 
				
			||||||
 | 
									req := httptest.NewRequest("GET", "/", nil)
 | 
				
			||||||
 | 
									req.Header.Set("My-Header-1", "Header-Value-1")
 | 
				
			||||||
 | 
									req.Header.Set("My-Header-2", "Header-Value-2")
 | 
				
			||||||
 | 
									req.AddCookie(&http.Cookie{Name: "session-1", Value: "abc-1"})
 | 
				
			||||||
 | 
									return req
 | 
				
			||||||
 | 
								},
 | 
				
			||||||
 | 
								parameters: map[string]map[string]string{
 | 
				
			||||||
 | 
									"header": {"My-Header-1": "true", "My-Header-2": "true"},
 | 
				
			||||||
 | 
									"cookie": {"session-1": "true", "session-2": "true"},
 | 
				
			||||||
 | 
								},
 | 
				
			||||||
 | 
								wantErr: true,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name: "optional header and cookie not provided partially",
 | 
				
			||||||
 | 
								prepareRequest: func() *http.Request {
 | 
				
			||||||
 | 
									req := httptest.NewRequest("GET", "/", nil)
 | 
				
			||||||
 | 
									req.Header.Set("My-Header-1", "Header-Value-1")
 | 
				
			||||||
 | 
									req.AddCookie(&http.Cookie{Name: "session-1", Value: "abc-1"})
 | 
				
			||||||
 | 
									return req
 | 
				
			||||||
 | 
								},
 | 
				
			||||||
 | 
								parameters: map[string]map[string]string{
 | 
				
			||||||
 | 
									"header": {"My-Header-1": "true", "My-Header-2": "false"},
 | 
				
			||||||
 | 
									"cookie": {"session-1": "true", "session-2": "false"},
 | 
				
			||||||
 | 
								},
 | 
				
			||||||
 | 
								wantErr: false,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name: "optional header and cookie not provided",
 | 
				
			||||||
 | 
								prepareRequest: func() *http.Request {
 | 
				
			||||||
 | 
									req := httptest.NewRequest("GET", "/", nil)
 | 
				
			||||||
 | 
									return req
 | 
				
			||||||
 | 
								},
 | 
				
			||||||
 | 
								parameters: map[string]map[string]string{
 | 
				
			||||||
 | 
									"header": {"My-Header-1": "false", "My-Header-2": "false"},
 | 
				
			||||||
 | 
									"cookie": {"session-1": "false", "session-2": "false"},
 | 
				
			||||||
 | 
								},
 | 
				
			||||||
 | 
								wantErr: false,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name: "no headers or cookies required",
 | 
				
			||||||
 | 
								prepareRequest: func() *http.Request {
 | 
				
			||||||
 | 
									req := httptest.NewRequest("GET", "/", nil)
 | 
				
			||||||
 | 
									return req
 | 
				
			||||||
 | 
								},
 | 
				
			||||||
 | 
								parameters: map[string]map[string]string{},
 | 
				
			||||||
 | 
								wantErr:    false,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, tt := range tests {
 | 
				
			||||||
 | 
							t.Run(tt.name, func(t *testing.T) {
 | 
				
			||||||
 | 
								err := validateHeadersAndCookies(tt.prepareRequest(), tt.parameters)
 | 
				
			||||||
 | 
								if tt.wantErr {
 | 
				
			||||||
 | 
									require.Error(t, err)
 | 
				
			||||||
 | 
								} else {
 | 
				
			||||||
 | 
									require.NoError(t, err)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							})
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										13
									
								
								client_stream.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										13
									
								
								client_stream.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,13 @@
 | 
				
			|||||||
 | 
					package http
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"context"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"go.unistack.org/micro/v3/client"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// TODO: Add stream support in the future.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (c *Client) fnStream(context.Context, client.Request, ...client.CallOption) (client.Stream, error) {
 | 
				
			||||||
 | 
						panic("not implemented")
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										272
									
								
								client_unary_call.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										272
									
								
								client_unary_call.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,272 @@
 | 
				
			|||||||
 | 
					package http
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"context"
 | 
				
			||||||
 | 
						"fmt"
 | 
				
			||||||
 | 
						"io"
 | 
				
			||||||
 | 
						"net"
 | 
				
			||||||
 | 
						"net/http"
 | 
				
			||||||
 | 
						"net/url"
 | 
				
			||||||
 | 
						"strings"
 | 
				
			||||||
 | 
						"time"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"go.unistack.org/micro/v3/client"
 | 
				
			||||||
 | 
						"go.unistack.org/micro/v3/codec"
 | 
				
			||||||
 | 
						"go.unistack.org/micro/v3/errors"
 | 
				
			||||||
 | 
						"go.unistack.org/micro/v3/logger"
 | 
				
			||||||
 | 
						"go.unistack.org/micro/v3/metadata"
 | 
				
			||||||
 | 
						"go.unistack.org/micro/v3/selector"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"go.unistack.org/micro-client-http/v3/status"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (c *Client) fnCall(ctx context.Context, req client.Request, rsp any, opts ...client.CallOption) error {
 | 
				
			||||||
 | 
						// make a copy of call opts
 | 
				
			||||||
 | 
						callOpts := c.opts.CallOptions
 | 
				
			||||||
 | 
						for _, opt := range opts {
 | 
				
			||||||
 | 
							opt(&callOpts)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// check if we already have a deadline
 | 
				
			||||||
 | 
						d, ok := ctx.Deadline()
 | 
				
			||||||
 | 
						if !ok {
 | 
				
			||||||
 | 
							var cancel context.CancelFunc
 | 
				
			||||||
 | 
							// no deadline so we create a new one
 | 
				
			||||||
 | 
							ctx, cancel = context.WithTimeout(ctx, callOpts.RequestTimeout)
 | 
				
			||||||
 | 
							defer cancel()
 | 
				
			||||||
 | 
						} else {
 | 
				
			||||||
 | 
							// got a deadline so no need to setup context,
 | 
				
			||||||
 | 
							// but we need to set the timeout we pass along
 | 
				
			||||||
 | 
							opt := client.WithRequestTimeout(time.Until(d))
 | 
				
			||||||
 | 
							opt(&callOpts)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// should we noop right here?
 | 
				
			||||||
 | 
						select {
 | 
				
			||||||
 | 
						case <-ctx.Done():
 | 
				
			||||||
 | 
							return errors.New("go.micro.client", fmt.Sprintf("%v", ctx.Err()), 408)
 | 
				
			||||||
 | 
						default:
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// make copy of call method
 | 
				
			||||||
 | 
						hcall := c.call
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// use the router passed as a call option, or fallback to the rpc clients router
 | 
				
			||||||
 | 
						if callOpts.Router == nil {
 | 
				
			||||||
 | 
							callOpts.Router = c.opts.Router
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if callOpts.Selector == nil {
 | 
				
			||||||
 | 
							callOpts.Selector = c.opts.Selector
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// inject proxy address
 | 
				
			||||||
 | 
						// TODO: don't even bother using Lookup/Select in this case
 | 
				
			||||||
 | 
						if len(c.opts.Proxy) > 0 {
 | 
				
			||||||
 | 
							callOpts.Address = []string{c.opts.Proxy}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						var next selector.Next
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						call := func(i int) error {
 | 
				
			||||||
 | 
							// call backoff first. Someone may want an initial start delay
 | 
				
			||||||
 | 
							t, err := callOpts.Backoff(ctx, req, i)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								return errors.InternalServerError("go.micro.client", "%+v", err)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							// only sleep if greater than 0
 | 
				
			||||||
 | 
							if t.Seconds() > 0 {
 | 
				
			||||||
 | 
								time.Sleep(t)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							if next == nil {
 | 
				
			||||||
 | 
								var routes []string
 | 
				
			||||||
 | 
								// lookup the route to send the reques to
 | 
				
			||||||
 | 
								// TODO apply any filtering here
 | 
				
			||||||
 | 
								routes, err = c.opts.Lookup(ctx, req, callOpts)
 | 
				
			||||||
 | 
								if err != nil {
 | 
				
			||||||
 | 
									return errors.InternalServerError("go.micro.client", "%+v", err)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								// balance the list of nodes
 | 
				
			||||||
 | 
								next, err = callOpts.Selector.Select(routes)
 | 
				
			||||||
 | 
								if err != nil {
 | 
				
			||||||
 | 
									return errors.InternalServerError("go.micro.client", "%+v", err)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							node := next()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							// make the call
 | 
				
			||||||
 | 
							err = hcall(ctx, node, req, rsp, callOpts)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							// record the result of the call to inform future routing decisions
 | 
				
			||||||
 | 
							if verr := c.opts.Selector.Record(node, err); verr != nil {
 | 
				
			||||||
 | 
								return errors.InternalServerError("go.micro.client", "%+v", verr)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							// try and transform the error to micro error
 | 
				
			||||||
 | 
							if verr, ok := err.(*errors.Error); ok {
 | 
				
			||||||
 | 
								return verr
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							return err
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						ch := make(chan error, callOpts.Retries)
 | 
				
			||||||
 | 
						var gerr error
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for i := 0; i <= callOpts.Retries; i++ {
 | 
				
			||||||
 | 
							go func() {
 | 
				
			||||||
 | 
								ch <- call(i)
 | 
				
			||||||
 | 
							}()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							select {
 | 
				
			||||||
 | 
							case <-ctx.Done():
 | 
				
			||||||
 | 
								return errors.New("go.micro.client", fmt.Sprintf("%v", ctx.Err()), 408)
 | 
				
			||||||
 | 
							case err := <-ch:
 | 
				
			||||||
 | 
								// if the call succeeded lets bail early
 | 
				
			||||||
 | 
								if err == nil {
 | 
				
			||||||
 | 
									return nil
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								retry, rerr := callOpts.Retry(ctx, req, i, err)
 | 
				
			||||||
 | 
								if rerr != nil {
 | 
				
			||||||
 | 
									return rerr
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								if !retry {
 | 
				
			||||||
 | 
									return err
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								gerr = err
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return gerr
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (c *Client) call(ctx context.Context, addr string, req client.Request, rsp any, opts client.CallOptions) error {
 | 
				
			||||||
 | 
						ct := req.ContentType()
 | 
				
			||||||
 | 
						if len(opts.ContentType) > 0 {
 | 
				
			||||||
 | 
							ct = opts.ContentType
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						cf, err := c.newCodec(ct)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return errors.BadRequest("go.micro.client", "%+v", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						hreq, err := buildHTTPRequest(ctx, addr, req.Endpoint(), ct, cf, req.Body(), opts, c.opts.Logger)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return errors.BadRequest("go.micro.client", "%+v", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						hrsp, err := c.httpClient.Do(hreq)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							switch err := err.(type) {
 | 
				
			||||||
 | 
							case *url.Error:
 | 
				
			||||||
 | 
								if err, ok := err.Err.(net.Error); ok && err.Timeout() {
 | 
				
			||||||
 | 
									return errors.Timeout("go.micro.client", "%+v", err)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							case net.Error:
 | 
				
			||||||
 | 
								if err.Timeout() {
 | 
				
			||||||
 | 
									return errors.Timeout("go.micro.client", "%+v", err)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							return errors.InternalServerError("go.micro.client", "%+v", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						defer hrsp.Body.Close()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return c.parseRsp(ctx, hrsp, rsp, opts)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (c *Client) newCodec(ct string) (codec.Codec, error) {
 | 
				
			||||||
 | 
						c.mu.RLock()
 | 
				
			||||||
 | 
						defer c.mu.RUnlock()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if idx := strings.IndexRune(ct, ';'); idx >= 0 {
 | 
				
			||||||
 | 
							ct = ct[:idx]
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if cf, ok := c.opts.Codecs[ct]; ok {
 | 
				
			||||||
 | 
							return cf, nil
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return nil, codec.ErrUnknownContentType
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (c *Client) parseRsp(ctx context.Context, hrsp *http.Response, rsp any, opts client.CallOptions) error {
 | 
				
			||||||
 | 
						log := c.opts.Logger
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						select {
 | 
				
			||||||
 | 
						case <-ctx.Done():
 | 
				
			||||||
 | 
							return ctx.Err()
 | 
				
			||||||
 | 
						default:
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						var buf []byte
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if opts.ResponseMetadata != nil {
 | 
				
			||||||
 | 
							for k, v := range hrsp.Header {
 | 
				
			||||||
 | 
								opts.ResponseMetadata.Set(k, strings.Join(v, ","))
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if hrsp.StatusCode == http.StatusNoContent {
 | 
				
			||||||
 | 
							return nil
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						ct := DefaultContentType
 | 
				
			||||||
 | 
						if htype := hrsp.Header.Get(metadata.HeaderContentType); htype != "" {
 | 
				
			||||||
 | 
							ct = htype
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if hrsp.Body != nil {
 | 
				
			||||||
 | 
							var err error
 | 
				
			||||||
 | 
							buf, err = io.ReadAll(hrsp.Body)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								return errors.InternalServerError("go.micro.client", "read body: %v", err)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						cf, err := c.newCodec(ct)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return errors.InternalServerError("go.micro.client", "unknown content-type %s: %v", ct, err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if log.V(logger.DebugLevel) {
 | 
				
			||||||
 | 
							log.Debug(ctx, fmt.Sprintf("response with headers: %v and body: %s", hrsp.Header, buf))
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if hrsp.StatusCode < http.StatusBadRequest {
 | 
				
			||||||
 | 
							if err = cf.Unmarshal(buf, rsp); err != nil {
 | 
				
			||||||
 | 
								return errors.InternalServerError("go.micro.client", "unmarshal response: %v", err)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							return nil
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						s := status.New(hrsp.StatusCode)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						var mappedErr any
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						errMap, ok := errorMapFromOpts(opts)
 | 
				
			||||||
 | 
						if ok && errMap != nil {
 | 
				
			||||||
 | 
							mappedErr, ok = errMap[fmt.Sprintf("%d", hrsp.StatusCode)]
 | 
				
			||||||
 | 
							if !ok {
 | 
				
			||||||
 | 
								mappedErr, ok = errMap["default"]
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if !ok || mappedErr == nil {
 | 
				
			||||||
 | 
							return s.Err()
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if err = cf.Unmarshal(buf, mappedErr); err != nil {
 | 
				
			||||||
 | 
							return errors.InternalServerError("go.micro.client", "unmarshal response: %v", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return s.WithDetails(mappedErr).Err()
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										1416
									
								
								client_unary_call_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1416
									
								
								client_unary_call_test.go
									
									
									
									
									
										Normal file
									
								
							
										
											
												File diff suppressed because it is too large
												Load Diff
											
										
									
								
							
							
								
								
									
										27
									
								
								go.mod
									
									
									
									
									
								
							
							
						
						
									
										27
									
								
								go.mod
									
									
									
									
									
								
							@@ -1,5 +1,26 @@
 | 
				
			|||||||
module github.com/unistack-org/micro-client-http/v3
 | 
					module go.unistack.org/micro-client-http/v3
 | 
				
			||||||
 | 
					
 | 
				
			||||||
go 1.16
 | 
					go 1.24.0
 | 
				
			||||||
 | 
					
 | 
				
			||||||
require github.com/unistack-org/micro/v3 v3.4.5
 | 
					toolchain go1.24.3
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					require (
 | 
				
			||||||
 | 
						github.com/stretchr/testify v1.11.1
 | 
				
			||||||
 | 
						go.unistack.org/micro-codec-json/v3 v3.10.3
 | 
				
			||||||
 | 
						go.unistack.org/micro/v3 v3.11.45
 | 
				
			||||||
 | 
						google.golang.org/protobuf v1.36.10
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					require (
 | 
				
			||||||
 | 
						github.com/ash3in/uuidv8 v1.2.0 // indirect
 | 
				
			||||||
 | 
						github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect
 | 
				
			||||||
 | 
						github.com/google/uuid v1.6.0 // indirect
 | 
				
			||||||
 | 
						github.com/kr/pretty v0.3.1 // indirect
 | 
				
			||||||
 | 
						github.com/matoous/go-nanoid v1.5.1 // indirect
 | 
				
			||||||
 | 
						github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 // indirect
 | 
				
			||||||
 | 
						go.unistack.org/micro-proto/v3 v3.4.1 // indirect
 | 
				
			||||||
 | 
						golang.org/x/sys v0.36.0 // indirect
 | 
				
			||||||
 | 
						google.golang.org/genproto/googleapis/rpc v0.0.0-20250929231259-57b25ae835d4 // indirect
 | 
				
			||||||
 | 
						google.golang.org/grpc v1.75.1 // indirect
 | 
				
			||||||
 | 
						gopkg.in/yaml.v3 v3.0.1 // indirect
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										65
									
								
								go.sum
									
									
									
									
									
								
							
							
						
						
									
										65
									
								
								go.sum
									
									
									
									
									
								
							@@ -1,17 +1,50 @@
 | 
				
			|||||||
github.com/dgrijalva/jwt-go v3.2.0+incompatible/go.mod h1:E3ru+11k8xSBh+hMPgOLZmtrrCbhqsmaPHjLKYnJCaQ=
 | 
					github.com/DATA-DOG/go-sqlmock v1.5.2 h1:OcvFkGmslmlZibjAjaHm3L//6LiuBgolP7OputlJIzU=
 | 
				
			||||||
github.com/ef-ds/deque v1.0.4/go.mod h1:gXDnTC3yqvBcHbq2lcExjtAcVrOnJCbMcZXmuj8Z4tg=
 | 
					github.com/DATA-DOG/go-sqlmock v1.5.2/go.mod h1:88MAG/4G7SMwSE3CeA0ZKzrT5CiOU3OJ+JlNzwDqpNU=
 | 
				
			||||||
github.com/google/uuid v1.2.0 h1:qJYtXnJRWmpe7m/3XlyhrsLrEURqHRM2kxzoxXqyUDs=
 | 
					github.com/ash3in/uuidv8 v1.2.0 h1:2oogGdtCPwaVtyvPPGin4TfZLtOGE5F+W++E880G6SI=
 | 
				
			||||||
github.com/google/uuid v1.2.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 | 
					github.com/ash3in/uuidv8 v1.2.0/go.mod h1:BnU0wJBxnzdEKmVg4xckBkD+VZuecTFTUP3M0dWgyY4=
 | 
				
			||||||
github.com/imdario/mergo v0.3.12/go.mod h1:jmQim1M+e3UYxmgPu/WyfjB3N3VflVyUjjjwH0dnCYA=
 | 
					github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=
 | 
				
			||||||
github.com/patrickmn/go-cache v2.1.0+incompatible/go.mod h1:3Qf8kWWT7OJRJbdiICTKqZju1ZixQ/KpMGzzAfe6+WQ=
 | 
					github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1VwoXQT9A3Wy9MM3WgvqSxFWenqJduM=
 | 
				
			||||||
github.com/silas/dag v0.0.0-20210121180416-41cf55125c34/go.mod h1:7RTUFBdIRC9nZ7/3RyRNH1bdqIShrDejd1YbLwgPS+I=
 | 
					github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 | 
				
			||||||
github.com/unistack-org/micro/v3 v3.4.5 h1:4Rn6EkJz/web43XGZAPt4NtqDLV2SrYXZxdAfBKF9Ic=
 | 
					github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek=
 | 
				
			||||||
github.com/unistack-org/micro/v3 v3.4.5/go.mod h1:LXmPfbJnJNvL0kQs8HfnkV3Wya2Wb+C7keVq++RCZnk=
 | 
					github.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps=
 | 
				
			||||||
golang.org/x/net v0.0.0-20210510120150-4163338589ed/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
 | 
					github.com/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8=
 | 
				
			||||||
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 | 
					github.com/google/go-cmp v0.7.0/go.mod h1:pXiqmnSA92OHEEa9HXL2W4E7lf9JzCmGVUdgjX3N/iU=
 | 
				
			||||||
golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 | 
					github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
 | 
				
			||||||
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
 | 
					github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 | 
				
			||||||
golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
 | 
					github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE=
 | 
				
			||||||
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
 | 
					github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk=
 | 
				
			||||||
 | 
					github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
 | 
				
			||||||
 | 
					github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
 | 
				
			||||||
 | 
					github.com/matoous/go-nanoid v1.5.1 h1:aCjdvTyO9LLnTIi0fgdXhOPPvOHjpXN6Ik9DaNjIct4=
 | 
				
			||||||
 | 
					github.com/matoous/go-nanoid v1.5.1/go.mod h1:zyD2a71IubI24efhpvkJz+ZwfwagzgSO6UNiFsZKN7U=
 | 
				
			||||||
 | 
					github.com/pkg/diff v0.0.0-20210226163009-20ebb0f2a09e/go.mod h1:pJLUxLENpZxwdsKMEsNbx1VGcRFpLqf3715MtcvvzbA=
 | 
				
			||||||
 | 
					github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 h1:Jamvg5psRIccs7FGNTlIRMkT8wgtp5eCXdBlqhYGL6U=
 | 
				
			||||||
 | 
					github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 | 
				
			||||||
 | 
					github.com/rogpeppe/go-internal v1.9.0/go.mod h1:WtVeX8xhTBvf0smdhujwtBcq4Qrzq/fJaraNFVN+nFs=
 | 
				
			||||||
 | 
					github.com/rogpeppe/go-internal v1.13.1 h1:KvO1DLK/DRN07sQ1LQKScxyZJuNnedQ5/wKSR38lUII=
 | 
				
			||||||
 | 
					github.com/rogpeppe/go-internal v1.13.1/go.mod h1:uMEvuHeurkdAXX61udpOXGD/AzZDWNMNyH2VO9fmH0o=
 | 
				
			||||||
 | 
					github.com/stretchr/testify v1.11.1 h1:7s2iGBzp5EwR7/aIZr8ao5+dra3wiQyKjjFuvgVKu7U=
 | 
				
			||||||
 | 
					github.com/stretchr/testify v1.11.1/go.mod h1:wZwfW3scLgRK+23gO65QZefKpKQRnfz6sD981Nm4B6U=
 | 
				
			||||||
 | 
					go.unistack.org/micro-codec-json/v3 v3.10.3 h1:FwSBfJswov30Dyqxp1XfQW1EG4h77uTEe/VGflg6XlY=
 | 
				
			||||||
 | 
					go.unistack.org/micro-codec-json/v3 v3.10.3/go.mod h1:26OK5MizMNKhspGC6PRVwpDIp5w1GmRb0nE5eRWWDxA=
 | 
				
			||||||
 | 
					go.unistack.org/micro-proto/v3 v3.4.1 h1:UTjLSRz2YZuaHk9iSlVqqsA50JQNAEK2ZFboGqtEa9Q=
 | 
				
			||||||
 | 
					go.unistack.org/micro-proto/v3 v3.4.1/go.mod h1:okx/cnOhzuCX0ggl/vToatbCupi0O44diiiLLsZ93Zo=
 | 
				
			||||||
 | 
					go.unistack.org/micro/v3 v3.11.45 h1:fjTLZYWgsVf9FIMZBxOg8ios2/tmyimnjZrsrxEUeXU=
 | 
				
			||||||
 | 
					go.unistack.org/micro/v3 v3.11.45/go.mod h1:fDQ8Mu9wubaFP0L8hNQlpzHiEnWN0wbOlawN9HYo0N4=
 | 
				
			||||||
 | 
					golang.org/x/net v0.41.0 h1:vBTly1HeNPEn3wtREYfy4GZ/NECgw2Cnl+nK6Nz3uvw=
 | 
				
			||||||
 | 
					golang.org/x/net v0.41.0/go.mod h1:B/K4NNqkfmg07DQYrbwvSluqCJOOXwUjeb/5lOisjbA=
 | 
				
			||||||
 | 
					golang.org/x/sys v0.36.0 h1:KVRy2GtZBrk1cBYA7MKu5bEZFxQk4NIDV6RLVcC8o0k=
 | 
				
			||||||
 | 
					golang.org/x/sys v0.36.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
 | 
				
			||||||
 | 
					golang.org/x/text v0.26.0 h1:P42AVeLghgTYr4+xUnTRKDMqpar+PtX7KWuNQL21L8M=
 | 
				
			||||||
 | 
					golang.org/x/text v0.26.0/go.mod h1:QK15LZJUUQVJxhz7wXgxSy/CJaTFjd0G+YLonydOVQA=
 | 
				
			||||||
 | 
					google.golang.org/genproto/googleapis/rpc v0.0.0-20250929231259-57b25ae835d4 h1:i8QOKZfYg6AbGVZzUAY3LrNWCKF8O6zFisU9Wl9RER4=
 | 
				
			||||||
 | 
					google.golang.org/genproto/googleapis/rpc v0.0.0-20250929231259-57b25ae835d4/go.mod h1:HSkG/KdJWusxU1F6CNrwNDjBMgisKxGnc5dAZfT0mjQ=
 | 
				
			||||||
 | 
					google.golang.org/grpc v1.75.1 h1:/ODCNEuf9VghjgO3rqLcfg8fiOP0nSluljWFlDxELLI=
 | 
				
			||||||
 | 
					google.golang.org/grpc v1.75.1/go.mod h1:JtPAzKiq4v1xcAB2hydNlWI2RnF85XXcV0mhKXr2ecQ=
 | 
				
			||||||
 | 
					google.golang.org/protobuf v1.36.10 h1:AYd7cD/uASjIL6Q9LiTjz8JLcrh/88q5UObnmY3aOOE=
 | 
				
			||||||
 | 
					google.golang.org/protobuf v1.36.10/go.mod h1:HTf+CrKn2C3g5S8VImy6tdcUvCska2kB7j23XfzDpco=
 | 
				
			||||||
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/yaml.v2 v2.3.0/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
 | 
					gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=
 | 
				
			||||||
 | 
					gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=
 | 
				
			||||||
 | 
					gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
 | 
				
			||||||
 | 
					gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										634
									
								
								http.go
									
									
									
									
									
								
							
							
						
						
									
										634
									
								
								http.go
									
									
									
									
									
								
							@@ -1,634 +0,0 @@
 | 
				
			|||||||
// Package http provides a http client
 | 
					 | 
				
			||||||
package http
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
import (
 | 
					 | 
				
			||||||
	"bufio"
 | 
					 | 
				
			||||||
	"bytes"
 | 
					 | 
				
			||||||
	"context"
 | 
					 | 
				
			||||||
	"fmt"
 | 
					 | 
				
			||||||
	"io/ioutil"
 | 
					 | 
				
			||||||
	"net"
 | 
					 | 
				
			||||||
	"net/http"
 | 
					 | 
				
			||||||
	"net/url"
 | 
					 | 
				
			||||||
	"os"
 | 
					 | 
				
			||||||
	"strings"
 | 
					 | 
				
			||||||
	"sync"
 | 
					 | 
				
			||||||
	"time"
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	"github.com/unistack-org/micro/v3/broker"
 | 
					 | 
				
			||||||
	"github.com/unistack-org/micro/v3/client"
 | 
					 | 
				
			||||||
	"github.com/unistack-org/micro/v3/codec"
 | 
					 | 
				
			||||||
	"github.com/unistack-org/micro/v3/errors"
 | 
					 | 
				
			||||||
	"github.com/unistack-org/micro/v3/metadata"
 | 
					 | 
				
			||||||
)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
var DefaultContentType = "application/json"
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
/*
 | 
					 | 
				
			||||||
func filterLabel(r []router.Route) []router.Route {
 | 
					 | 
				
			||||||
	//				selector.FilterLabel("protocol", "http")
 | 
					 | 
				
			||||||
	return r
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
*/
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
type httpClient struct {
 | 
					 | 
				
			||||||
	httpcli *http.Client
 | 
					 | 
				
			||||||
	opts    client.Options
 | 
					 | 
				
			||||||
	sync.RWMutex
 | 
					 | 
				
			||||||
	init bool
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func newRequest(addr string, req client.Request, ct string, cf codec.Codec, msg interface{}, opts client.CallOptions) (*http.Request, error) {
 | 
					 | 
				
			||||||
	hreq := &http.Request{Method: http.MethodPost}
 | 
					 | 
				
			||||||
	body := "*" // as like google api http annotation
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	var tags []string
 | 
					 | 
				
			||||||
	var scheme string
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	u, err := url.Parse(addr)
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		hreq.URL = &url.URL{
 | 
					 | 
				
			||||||
			Scheme: "http",
 | 
					 | 
				
			||||||
			Host:   addr,
 | 
					 | 
				
			||||||
			Path:   req.Endpoint(),
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		hreq.Host = addr
 | 
					 | 
				
			||||||
		scheme = "http"
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// nolint: nestif
 | 
					 | 
				
			||||||
	if scheme == "" {
 | 
					 | 
				
			||||||
		ep := req.Endpoint()
 | 
					 | 
				
			||||||
		if opts.Context != nil {
 | 
					 | 
				
			||||||
			if m, ok := opts.Context.Value(methodKey{}).(string); ok {
 | 
					 | 
				
			||||||
				hreq.Method = m
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
			if p, ok := opts.Context.Value(pathKey{}).(string); ok {
 | 
					 | 
				
			||||||
				ep = p
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
			if b, ok := opts.Context.Value(bodyKey{}).(string); ok {
 | 
					 | 
				
			||||||
				body = b
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
			if t, ok := opts.Context.Value(structTagsKey{}).([]string); ok && len(t) > 0 {
 | 
					 | 
				
			||||||
				tags = t
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		hreq.URL, err = u.Parse(ep)
 | 
					 | 
				
			||||||
		if err != nil {
 | 
					 | 
				
			||||||
			return nil, errors.BadRequest("go.micro.client", err.Error())
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if len(tags) == 0 {
 | 
					 | 
				
			||||||
		switch ct {
 | 
					 | 
				
			||||||
		default:
 | 
					 | 
				
			||||||
			tags = append(tags, "json", "protobuf")
 | 
					 | 
				
			||||||
		case "text/xml":
 | 
					 | 
				
			||||||
			tags = append(tags, "xml")
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	path, nmsg, err := newPathRequest(hreq.URL.Path, hreq.Method, body, msg, tags)
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return nil, errors.BadRequest("go.micro.client", err.Error())
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if scheme != "" {
 | 
					 | 
				
			||||||
		hreq.URL, err = url.Parse(scheme + "://" + addr + path)
 | 
					 | 
				
			||||||
	} else {
 | 
					 | 
				
			||||||
		hreq.URL, err = url.Parse(addr + path)
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return nil, errors.BadRequest("go.micro.client", err.Error())
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	b, err := cf.Marshal(nmsg)
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return nil, errors.BadRequest("go.micro.client", err.Error())
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if len(b) > 0 {
 | 
					 | 
				
			||||||
		hreq.Body = ioutil.NopCloser(bytes.NewBuffer(b))
 | 
					 | 
				
			||||||
		hreq.ContentLength = int64(len(b))
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	return hreq, nil
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpClient) call(ctx context.Context, addr string, req client.Request, rsp interface{}, opts client.CallOptions) error {
 | 
					 | 
				
			||||||
	header := make(http.Header, 2)
 | 
					 | 
				
			||||||
	if md, ok := metadata.FromOutgoingContext(ctx); ok {
 | 
					 | 
				
			||||||
		for k, v := range md {
 | 
					 | 
				
			||||||
			header.Set(k, v)
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	ct := req.ContentType()
 | 
					 | 
				
			||||||
	if len(opts.ContentType) > 0 {
 | 
					 | 
				
			||||||
		ct = opts.ContentType
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// set timeout in nanoseconds
 | 
					 | 
				
			||||||
	header.Set("Timeout", fmt.Sprintf("%d", opts.RequestTimeout))
 | 
					 | 
				
			||||||
	// set the content type for the request
 | 
					 | 
				
			||||||
	header.Set("Content-Type", ct)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	cf, err := h.newCodec(ct)
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return errors.InternalServerError("go.micro.client", err.Error())
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	hreq, err := newRequest(addr, req, ct, cf, req.Body(), opts)
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	hreq.Header = header
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// make the request
 | 
					 | 
				
			||||||
	hrsp, err := h.httpcli.Do(hreq.WithContext(ctx))
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		switch err := err.(type) {
 | 
					 | 
				
			||||||
		case *url.Error:
 | 
					 | 
				
			||||||
			if err, ok := err.Err.(net.Error); ok && err.Timeout() {
 | 
					 | 
				
			||||||
				return errors.Timeout("go.micro.client", err.Error())
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
		case net.Error:
 | 
					 | 
				
			||||||
			if err.Timeout() {
 | 
					 | 
				
			||||||
				return errors.Timeout("go.micro.client", err.Error())
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		return errors.InternalServerError("go.micro.client", err.Error())
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	defer hrsp.Body.Close()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	return h.parseRsp(ctx, hrsp, rsp, opts)
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpClient) stream(ctx context.Context, addr string, req client.Request, opts client.CallOptions) (client.Stream, error) {
 | 
					 | 
				
			||||||
	var header http.Header
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if md, ok := metadata.FromOutgoingContext(ctx); ok {
 | 
					 | 
				
			||||||
		header = make(http.Header, len(md)+2)
 | 
					 | 
				
			||||||
		for k, v := range md {
 | 
					 | 
				
			||||||
			header.Set(k, v)
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	} else {
 | 
					 | 
				
			||||||
		header = make(http.Header, 2)
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	ct := req.ContentType()
 | 
					 | 
				
			||||||
	if len(opts.ContentType) > 0 {
 | 
					 | 
				
			||||||
		ct = opts.ContentType
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	// set timeout in nanoseconds
 | 
					 | 
				
			||||||
	if opts.StreamTimeout > time.Duration(0) {
 | 
					 | 
				
			||||||
		header.Set("Timeout", fmt.Sprintf("%d", opts.StreamTimeout))
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	// set the content type for the request
 | 
					 | 
				
			||||||
	header.Set("Content-Type", ct)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// get codec
 | 
					 | 
				
			||||||
	cf, err := h.newCodec(ct)
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return nil, errors.InternalServerError("go.micro.client", err.Error())
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	cc, err := (h.httpcli.Transport).(*http.Transport).DialContext(ctx, "tcp", addr)
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return nil, errors.InternalServerError("go.micro.client", fmt.Sprintf("Error dialing: %v", err))
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	return &httpStream{
 | 
					 | 
				
			||||||
		address: addr,
 | 
					 | 
				
			||||||
		context: ctx,
 | 
					 | 
				
			||||||
		closed:  make(chan bool),
 | 
					 | 
				
			||||||
		opts:    opts,
 | 
					 | 
				
			||||||
		conn:    cc,
 | 
					 | 
				
			||||||
		ct:      ct,
 | 
					 | 
				
			||||||
		cf:      cf,
 | 
					 | 
				
			||||||
		header:  header,
 | 
					 | 
				
			||||||
		reader:  bufio.NewReader(cc),
 | 
					 | 
				
			||||||
		request: req,
 | 
					 | 
				
			||||||
	}, nil
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpClient) newCodec(ct string) (codec.Codec, error) {
 | 
					 | 
				
			||||||
	h.RLock()
 | 
					 | 
				
			||||||
	defer h.RUnlock()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if idx := strings.IndexRune(ct, ';'); idx >= 0 {
 | 
					 | 
				
			||||||
		ct = ct[:idx]
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if c, ok := h.opts.Codecs[ct]; ok {
 | 
					 | 
				
			||||||
		return c, nil
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	return nil, codec.ErrUnknownContentType
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpClient) Init(opts ...client.Option) error {
 | 
					 | 
				
			||||||
	if len(opts) == 0 && h.init {
 | 
					 | 
				
			||||||
		return nil
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	for _, o := range opts {
 | 
					 | 
				
			||||||
		o(&h.opts)
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if err := h.opts.Broker.Init(); err != nil {
 | 
					 | 
				
			||||||
		return err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	if err := h.opts.Tracer.Init(); err != nil {
 | 
					 | 
				
			||||||
		return err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	if err := h.opts.Router.Init(); err != nil {
 | 
					 | 
				
			||||||
		return err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	if err := h.opts.Logger.Init(); err != nil {
 | 
					 | 
				
			||||||
		return err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	if err := h.opts.Meter.Init(); err != nil {
 | 
					 | 
				
			||||||
		return err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	if err := h.opts.Transport.Init(); err != nil {
 | 
					 | 
				
			||||||
		return err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	return nil
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpClient) Options() client.Options {
 | 
					 | 
				
			||||||
	return h.opts
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpClient) NewMessage(topic string, msg interface{}, opts ...client.MessageOption) client.Message {
 | 
					 | 
				
			||||||
	return newHTTPMessage(topic, msg, h.opts.ContentType, opts...)
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpClient) NewRequest(service, method string, req interface{}, opts ...client.RequestOption) client.Request {
 | 
					 | 
				
			||||||
	return newHTTPRequest(service, method, req, h.opts.ContentType, opts...)
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpClient) Call(ctx context.Context, req client.Request, rsp interface{}, opts ...client.CallOption) error {
 | 
					 | 
				
			||||||
	// make a copy of call opts
 | 
					 | 
				
			||||||
	callOpts := h.opts.CallOptions
 | 
					 | 
				
			||||||
	for _, opt := range opts {
 | 
					 | 
				
			||||||
		opt(&callOpts)
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// check if we already have a deadline
 | 
					 | 
				
			||||||
	d, ok := ctx.Deadline()
 | 
					 | 
				
			||||||
	if !ok {
 | 
					 | 
				
			||||||
		var cancel context.CancelFunc
 | 
					 | 
				
			||||||
		// no deadline so we create a new one
 | 
					 | 
				
			||||||
		ctx, cancel = context.WithTimeout(ctx, callOpts.RequestTimeout)
 | 
					 | 
				
			||||||
		defer cancel()
 | 
					 | 
				
			||||||
	} else {
 | 
					 | 
				
			||||||
		// got a deadline so no need to setup context
 | 
					 | 
				
			||||||
		// but we need to set the timeout we pass along
 | 
					 | 
				
			||||||
		opt := client.WithRequestTimeout(time.Until(d))
 | 
					 | 
				
			||||||
		opt(&callOpts)
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// should we noop right here?
 | 
					 | 
				
			||||||
	select {
 | 
					 | 
				
			||||||
	case <-ctx.Done():
 | 
					 | 
				
			||||||
		return errors.New("go.micro.client", fmt.Sprintf("%v", ctx.Err()), 408)
 | 
					 | 
				
			||||||
	default:
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// make copy of call method
 | 
					 | 
				
			||||||
	hcall := h.call
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// 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
 | 
					 | 
				
			||||||
	if callOpts.Router == nil {
 | 
					 | 
				
			||||||
		callOpts.Router = h.opts.Router
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if callOpts.Selector == nil {
 | 
					 | 
				
			||||||
		callOpts.Selector = h.opts.Selector
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// inject proxy address
 | 
					 | 
				
			||||||
	// TODO: don't even bother using Lookup/Select in this case
 | 
					 | 
				
			||||||
	if len(h.opts.Proxy) > 0 {
 | 
					 | 
				
			||||||
		callOpts.Address = []string{h.opts.Proxy}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// lookup the route to send the reques to
 | 
					 | 
				
			||||||
	// TODO apply any filtering here
 | 
					 | 
				
			||||||
	routes, err := h.opts.Lookup(ctx, req, callOpts)
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return errors.InternalServerError("go.micro.client", err.Error())
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// balance the list of nodes
 | 
					 | 
				
			||||||
	next, err := callOpts.Selector.Select(routes)
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// return errors.New("go.micro.client", "request timeout", 408)
 | 
					 | 
				
			||||||
	call := func(i int) error {
 | 
					 | 
				
			||||||
		// call backoff first. Someone may want an initial start delay
 | 
					 | 
				
			||||||
		t, err := callOpts.Backoff(ctx, req, i)
 | 
					 | 
				
			||||||
		if err != nil {
 | 
					 | 
				
			||||||
			return errors.InternalServerError("go.micro.client", err.Error())
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		// only sleep if greater than 0
 | 
					 | 
				
			||||||
		if t.Seconds() > 0 {
 | 
					 | 
				
			||||||
			time.Sleep(t)
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		node := next()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		// make the call
 | 
					 | 
				
			||||||
		err = hcall(ctx, node, req, rsp, callOpts)
 | 
					 | 
				
			||||||
		// record the result of the call to inform future routing decisions
 | 
					 | 
				
			||||||
		if verr := h.opts.Selector.Record(node, err); verr != nil {
 | 
					 | 
				
			||||||
			return verr
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		// try and transform the error to a go-micro error
 | 
					 | 
				
			||||||
		if verr, ok := err.(*errors.Error); ok {
 | 
					 | 
				
			||||||
			return verr
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		return err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	ch := make(chan error, callOpts.Retries)
 | 
					 | 
				
			||||||
	var gerr error
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	for i := 0; i <= callOpts.Retries; i++ {
 | 
					 | 
				
			||||||
		go func() {
 | 
					 | 
				
			||||||
			ch <- call(i)
 | 
					 | 
				
			||||||
		}()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		select {
 | 
					 | 
				
			||||||
		case <-ctx.Done():
 | 
					 | 
				
			||||||
			return errors.New("go.micro.client", fmt.Sprintf("%v", ctx.Err()), 408)
 | 
					 | 
				
			||||||
		case err := <-ch:
 | 
					 | 
				
			||||||
			// if the call succeeded lets bail early
 | 
					 | 
				
			||||||
			if err == nil {
 | 
					 | 
				
			||||||
				return nil
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
			retry, rerr := callOpts.Retry(ctx, req, i, err)
 | 
					 | 
				
			||||||
			if rerr != nil {
 | 
					 | 
				
			||||||
				return rerr
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
			if !retry {
 | 
					 | 
				
			||||||
				return err
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
			gerr = err
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	return gerr
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpClient) Stream(ctx context.Context, req client.Request, opts ...client.CallOption) (client.Stream, error) {
 | 
					 | 
				
			||||||
	// make a copy of call opts
 | 
					 | 
				
			||||||
	callOpts := h.opts.CallOptions
 | 
					 | 
				
			||||||
	for _, o := range opts {
 | 
					 | 
				
			||||||
		o(&callOpts)
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// check if we already have a deadline
 | 
					 | 
				
			||||||
	d, ok := ctx.Deadline()
 | 
					 | 
				
			||||||
	if !ok && callOpts.StreamTimeout > time.Duration(0) {
 | 
					 | 
				
			||||||
		var cancel context.CancelFunc
 | 
					 | 
				
			||||||
		// no deadline so we create a new one
 | 
					 | 
				
			||||||
		ctx, cancel = context.WithTimeout(ctx, callOpts.StreamTimeout)
 | 
					 | 
				
			||||||
		defer cancel()
 | 
					 | 
				
			||||||
	} else {
 | 
					 | 
				
			||||||
		// got a deadline so no need to setup context
 | 
					 | 
				
			||||||
		// but we need to set the timeout we pass along
 | 
					 | 
				
			||||||
		o := client.WithStreamTimeout(time.Until(d))
 | 
					 | 
				
			||||||
		o(&callOpts)
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// should we noop right here?
 | 
					 | 
				
			||||||
	select {
 | 
					 | 
				
			||||||
	case <-ctx.Done():
 | 
					 | 
				
			||||||
		return nil, errors.New("go.micro.client", fmt.Sprintf("%v", ctx.Err()), 408)
 | 
					 | 
				
			||||||
	default:
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	/*
 | 
					 | 
				
			||||||
		// make copy of call method
 | 
					 | 
				
			||||||
		hstream := h.stream
 | 
					 | 
				
			||||||
		// wrap the call in reverse
 | 
					 | 
				
			||||||
		for i := len(callOpts.CallWrappers); i > 0; i-- {
 | 
					 | 
				
			||||||
			hstream = callOpts.CallWrappers[i-1](hstream)
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	*/
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// use the router passed as a call option, or fallback to the rpc clients router
 | 
					 | 
				
			||||||
	if callOpts.Router == nil {
 | 
					 | 
				
			||||||
		callOpts.Router = h.opts.Router
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if callOpts.Selector == nil {
 | 
					 | 
				
			||||||
		callOpts.Selector = h.opts.Selector
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// inject proxy address
 | 
					 | 
				
			||||||
	// TODO: don't even bother using Lookup/Select in this case
 | 
					 | 
				
			||||||
	if len(h.opts.Proxy) > 0 {
 | 
					 | 
				
			||||||
		callOpts.Address = []string{h.opts.Proxy}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// lookup the route to send the reques to
 | 
					 | 
				
			||||||
	// TODO apply any filtering here
 | 
					 | 
				
			||||||
	routes, err := h.opts.Lookup(ctx, req, callOpts)
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return nil, errors.InternalServerError("go.micro.client", err.Error())
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// balance the list of nodes
 | 
					 | 
				
			||||||
	next, err := callOpts.Selector.Select(routes)
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return nil, err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	call := func(i int) (client.Stream, error) {
 | 
					 | 
				
			||||||
		// call backoff first. Someone may want an initial start delay
 | 
					 | 
				
			||||||
		t, cerr := callOpts.Backoff(ctx, req, i)
 | 
					 | 
				
			||||||
		if cerr != nil {
 | 
					 | 
				
			||||||
			return nil, errors.InternalServerError("go.micro.client", cerr.Error())
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		// only sleep if greater than 0
 | 
					 | 
				
			||||||
		if t.Seconds() > 0 {
 | 
					 | 
				
			||||||
			time.Sleep(t)
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		node := next()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		stream, cerr := h.stream(ctx, node, req, callOpts)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		// record the result of the call to inform future routing decisions
 | 
					 | 
				
			||||||
		if verr := h.opts.Selector.Record(node, cerr); verr != nil {
 | 
					 | 
				
			||||||
			return nil, verr
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		// try and transform the error to a go-micro error
 | 
					 | 
				
			||||||
		if verr, ok := cerr.(*errors.Error); ok {
 | 
					 | 
				
			||||||
			return nil, verr
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		return stream, cerr
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	type response struct {
 | 
					 | 
				
			||||||
		stream client.Stream
 | 
					 | 
				
			||||||
		err    error
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	ch := make(chan response, callOpts.Retries)
 | 
					 | 
				
			||||||
	var grr error
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	for i := 0; i <= callOpts.Retries; i++ {
 | 
					 | 
				
			||||||
		go func() {
 | 
					 | 
				
			||||||
			s, cerr := call(i)
 | 
					 | 
				
			||||||
			ch <- response{s, cerr}
 | 
					 | 
				
			||||||
		}()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		select {
 | 
					 | 
				
			||||||
		case <-ctx.Done():
 | 
					 | 
				
			||||||
			return nil, errors.New("go.micro.client", fmt.Sprintf("%v", ctx.Err()), 408)
 | 
					 | 
				
			||||||
		case rsp := <-ch:
 | 
					 | 
				
			||||||
			// if the call succeeded lets bail early
 | 
					 | 
				
			||||||
			if rsp.err == nil {
 | 
					 | 
				
			||||||
				return rsp.stream, nil
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
			retry, rerr := callOpts.Retry(ctx, req, i, err)
 | 
					 | 
				
			||||||
			if rerr != nil {
 | 
					 | 
				
			||||||
				return nil, rerr
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
			if !retry {
 | 
					 | 
				
			||||||
				return nil, rsp.err
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
			grr = rsp.err
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	return nil, grr
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpClient) Publish(ctx context.Context, p client.Message, opts ...client.PublishOption) error {
 | 
					 | 
				
			||||||
	options := client.NewPublishOptions(opts...)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	md, ok := metadata.FromOutgoingContext(ctx)
 | 
					 | 
				
			||||||
	if !ok {
 | 
					 | 
				
			||||||
		md = metadata.New(2)
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	md["Content-Type"] = p.ContentType()
 | 
					 | 
				
			||||||
	md["Micro-Topic"] = p.Topic()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	cf, err := h.newCodec(p.ContentType())
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return errors.InternalServerError("go.micro.client", err.Error())
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	var body []byte
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// passed in raw data
 | 
					 | 
				
			||||||
	if d, ok := p.Payload().(*codec.Frame); ok {
 | 
					 | 
				
			||||||
		body = d.Data
 | 
					 | 
				
			||||||
	} else {
 | 
					 | 
				
			||||||
		b := bytes.NewBuffer(nil)
 | 
					 | 
				
			||||||
		if err := cf.Write(b, &codec.Message{Type: codec.Event}, p.Payload()); err != nil {
 | 
					 | 
				
			||||||
			return errors.InternalServerError("go.micro.client", err.Error())
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		body = b.Bytes()
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	topic := p.Topic()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// get proxy
 | 
					 | 
				
			||||||
	if prx := os.Getenv("MICRO_PROXY"); len(prx) > 0 {
 | 
					 | 
				
			||||||
		options.Exchange = prx
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// get the exchange
 | 
					 | 
				
			||||||
	if len(options.Exchange) > 0 {
 | 
					 | 
				
			||||||
		topic = options.Exchange
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	return h.opts.Broker.Publish(ctx, topic, &broker.Message{
 | 
					 | 
				
			||||||
		Header: md,
 | 
					 | 
				
			||||||
		Body:   body,
 | 
					 | 
				
			||||||
	},
 | 
					 | 
				
			||||||
		broker.PublishContext(ctx),
 | 
					 | 
				
			||||||
		broker.PublishBodyOnly(options.BodyOnly),
 | 
					 | 
				
			||||||
	)
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpClient) String() string {
 | 
					 | 
				
			||||||
	return "http"
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpClient) Name() string {
 | 
					 | 
				
			||||||
	return h.opts.Name
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func NewClient(opts ...client.Option) client.Client {
 | 
					 | 
				
			||||||
	options := client.NewOptions(opts...)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if len(options.ContentType) == 0 {
 | 
					 | 
				
			||||||
		options.ContentType = DefaultContentType
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	rc := &httpClient{
 | 
					 | 
				
			||||||
		opts: options,
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	dialer, ok := options.Context.Value(httpDialerKey{}).(*net.Dialer)
 | 
					 | 
				
			||||||
	if !ok {
 | 
					 | 
				
			||||||
		dialer = &net.Dialer{
 | 
					 | 
				
			||||||
			Timeout:   30 * time.Second,
 | 
					 | 
				
			||||||
			KeepAlive: 30 * time.Second,
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if httpcli, ok := options.Context.Value(httpClientKey{}).(*http.Client); ok {
 | 
					 | 
				
			||||||
		rc.httpcli = httpcli
 | 
					 | 
				
			||||||
	} else {
 | 
					 | 
				
			||||||
		// TODO customTransport := http.DefaultTransport.(*http.Transport).Clone()
 | 
					 | 
				
			||||||
		tr := &http.Transport{
 | 
					 | 
				
			||||||
			Proxy:                 http.ProxyFromEnvironment,
 | 
					 | 
				
			||||||
			DialContext:           dialer.DialContext,
 | 
					 | 
				
			||||||
			ForceAttemptHTTP2:     true,
 | 
					 | 
				
			||||||
			MaxConnsPerHost:       100,
 | 
					 | 
				
			||||||
			MaxIdleConns:          20,
 | 
					 | 
				
			||||||
			IdleConnTimeout:       60 * time.Second,
 | 
					 | 
				
			||||||
			TLSHandshakeTimeout:   10 * time.Second,
 | 
					 | 
				
			||||||
			ExpectContinueTimeout: 1 * time.Second,
 | 
					 | 
				
			||||||
			TLSClientConfig:       options.TLSConfig,
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		rc.httpcli = &http.Client{Transport: tr}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	c := client.Client(rc)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// wrap in reverse
 | 
					 | 
				
			||||||
	for i := len(options.Wrappers); i > 0; i-- {
 | 
					 | 
				
			||||||
		c = options.Wrappers[i-1](c)
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	return c
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
							
								
								
									
										40
									
								
								http_test.go
									
									
									
									
									
								
							
							
						
						
									
										40
									
								
								http_test.go
									
									
									
									
									
								
							@@ -1,40 +0,0 @@
 | 
				
			|||||||
package http
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
import (
 | 
					 | 
				
			||||||
	"net/url"
 | 
					 | 
				
			||||||
	"strings"
 | 
					 | 
				
			||||||
	"testing"
 | 
					 | 
				
			||||||
)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
type Request struct {
 | 
					 | 
				
			||||||
	Name   string `json:"name"`
 | 
					 | 
				
			||||||
	Field1 string
 | 
					 | 
				
			||||||
	Field2 string
 | 
					 | 
				
			||||||
	Field3 int64
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func TestValidPath(t *testing.T) {
 | 
					 | 
				
			||||||
	req := &Request{Name: "vtolstov", Field1: "field1", Field2: "field2", Field3: 10}
 | 
					 | 
				
			||||||
	p, m, err := newPathRequest("/api/v1/{name}/list", "GET", "", req, nil)
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		t.Fatal(err)
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	u, err := url.Parse(p)
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		t.Fatal(err)
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	_ = m
 | 
					 | 
				
			||||||
	parts := strings.Split(u.RawQuery, "&")
 | 
					 | 
				
			||||||
	if len(parts) != 3 {
 | 
					 | 
				
			||||||
		t.Fatalf("invalid path: %v", parts)
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func TestInvalidPath(t *testing.T) {
 | 
					 | 
				
			||||||
	req := &Request{Name: "vtolstov", Field1: "field1", Field2: "field2", Field3: 10}
 | 
					 | 
				
			||||||
	p, m, err := newPathRequest("/api/v1/{xname}/list", "GET", "", req, nil)
 | 
					 | 
				
			||||||
	if err == nil {
 | 
					 | 
				
			||||||
		t.Fatalf("path param must not be filled")
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	_, _ = p, m
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
							
								
								
									
										36
									
								
								message.go
									
									
									
									
									
								
							
							
						
						
									
										36
									
								
								message.go
									
									
									
									
									
								
							@@ -1,36 +0,0 @@
 | 
				
			|||||||
package http
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
import (
 | 
					 | 
				
			||||||
	"github.com/unistack-org/micro/v3/client"
 | 
					 | 
				
			||||||
)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
type httpMessage struct {
 | 
					 | 
				
			||||||
	payload     interface{}
 | 
					 | 
				
			||||||
	topic       string
 | 
					 | 
				
			||||||
	contentType string
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func newHTTPMessage(topic string, payload interface{}, contentType string, opts ...client.MessageOption) client.Message {
 | 
					 | 
				
			||||||
	options := client.NewMessageOptions(opts...)
 | 
					 | 
				
			||||||
	if len(options.ContentType) == 0 {
 | 
					 | 
				
			||||||
		options.ContentType = contentType
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	return &httpMessage{
 | 
					 | 
				
			||||||
		payload:     payload,
 | 
					 | 
				
			||||||
		topic:       topic,
 | 
					 | 
				
			||||||
		contentType: options.ContentType,
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpMessage) ContentType() string {
 | 
					 | 
				
			||||||
	return h.contentType
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpMessage) Topic() string {
 | 
					 | 
				
			||||||
	return h.topic
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpMessage) Payload() interface{} {
 | 
					 | 
				
			||||||
	return h.payload
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
							
								
								
									
										158
									
								
								options.go
									
									
									
									
									
								
							
							
						
						
									
										158
									
								
								options.go
									
									
									
									
									
								
							@@ -1,98 +1,148 @@
 | 
				
			|||||||
package http
 | 
					package http
 | 
				
			||||||
 | 
					
 | 
				
			||||||
import (
 | 
					import (
 | 
				
			||||||
 | 
						"context"
 | 
				
			||||||
 | 
						"crypto/tls"
 | 
				
			||||||
	"net"
 | 
						"net"
 | 
				
			||||||
	"net/http"
 | 
						"net/http"
 | 
				
			||||||
 | 
						"time"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	"github.com/unistack-org/micro/v3/client"
 | 
						"go.unistack.org/micro/v3/client"
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
var (
 | 
					// --------------------------------------------- HTTPClient option -----------------------------------------------------
 | 
				
			||||||
	// DefaultPoolMaxStreams maximum streams on a connectioin
 | 
					 | 
				
			||||||
	// (20)
 | 
					 | 
				
			||||||
	DefaultPoolMaxStreams = 20
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// DefaultPoolMaxIdle maximum idle conns of a pool
 | 
					 | 
				
			||||||
	// (50)
 | 
					 | 
				
			||||||
	DefaultPoolMaxIdle = 50
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// DefaultMaxRecvMsgSize maximum message that client can receive
 | 
					 | 
				
			||||||
	// (4 MB).
 | 
					 | 
				
			||||||
	DefaultMaxRecvMsgSize = 1024 * 1024 * 4
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// DefaultMaxSendMsgSize maximum message that client can send
 | 
					 | 
				
			||||||
	// (4 MB).
 | 
					 | 
				
			||||||
	DefaultMaxSendMsgSize = 1024 * 1024 * 4
 | 
					 | 
				
			||||||
)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
type poolMaxStreams struct{}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
// PoolMaxStreams maximum streams on a connectioin
 | 
					 | 
				
			||||||
func PoolMaxStreams(n int) client.Option {
 | 
					 | 
				
			||||||
	return client.SetOption(poolMaxStreams{}, n)
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
type poolMaxIdle struct{}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
// PoolMaxIdle maximum idle conns of a pool
 | 
					 | 
				
			||||||
func PoolMaxIdle(d int) client.Option {
 | 
					 | 
				
			||||||
	return client.SetOption(poolMaxIdle{}, d)
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
type maxRecvMsgSizeKey struct{}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
// MaxRecvMsgSize set the maximum size of message that client can receive.
 | 
					 | 
				
			||||||
func MaxRecvMsgSize(s int) client.Option {
 | 
					 | 
				
			||||||
	return client.SetOption(maxRecvMsgSizeKey{}, s)
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
type maxSendMsgSizeKey struct{}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
// MaxSendMsgSize set the maximum size of message that client can send.
 | 
					 | 
				
			||||||
func MaxSendMsgSize(s int) client.Option {
 | 
					 | 
				
			||||||
	return client.SetOption(maxSendMsgSizeKey{}, s)
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
type httpClientKey struct{}
 | 
					type httpClientKey struct{}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// nolint: golint
 | 
					 | 
				
			||||||
func HTTPClient(c *http.Client) client.Option {
 | 
					func HTTPClient(c *http.Client) client.Option {
 | 
				
			||||||
	return client.SetOption(httpClientKey{}, c)
 | 
						return client.SetOption(httpClientKey{}, c)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func httpClientFromOpts(opts client.Options) (*http.Client, bool) {
 | 
				
			||||||
 | 
						httpClient, ok := opts.Context.Value(httpClientKey{}).(*http.Client)
 | 
				
			||||||
 | 
						return httpClient, ok
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func defaultHTTPClient(
 | 
				
			||||||
 | 
						dialer func(ctx context.Context, addr string) (net.Conn, error),
 | 
				
			||||||
 | 
						tlsConfig *tls.Config,
 | 
				
			||||||
 | 
					) *http.Client {
 | 
				
			||||||
 | 
						tr := &http.Transport{
 | 
				
			||||||
 | 
							Proxy: http.ProxyFromEnvironment,
 | 
				
			||||||
 | 
							DialContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
 | 
				
			||||||
 | 
								return dialer(ctx, addr)
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							ForceAttemptHTTP2:     true,
 | 
				
			||||||
 | 
							MaxConnsPerHost:       100,
 | 
				
			||||||
 | 
							MaxIdleConns:          20,
 | 
				
			||||||
 | 
							IdleConnTimeout:       60 * time.Second,
 | 
				
			||||||
 | 
							TLSHandshakeTimeout:   10 * time.Second,
 | 
				
			||||||
 | 
							ExpectContinueTimeout: 1 * time.Second,
 | 
				
			||||||
 | 
							TLSClientConfig:       tlsConfig,
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return &http.Client{Transport: tr}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// --------------------------------------------- HTTPDialer option -----------------------------------------------------
 | 
				
			||||||
type httpDialerKey struct{}
 | 
					type httpDialerKey struct{}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// nolint: golint
 | 
					 | 
				
			||||||
func HTTPDialer(d *net.Dialer) client.Option {
 | 
					func HTTPDialer(d *net.Dialer) client.Option {
 | 
				
			||||||
	return client.SetOption(httpDialerKey{}, d)
 | 
						return client.SetOption(httpDialerKey{}, d)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func httpDialerFromOpts(opts client.Options) (dialerFunc func(context.Context, string) (net.Conn, error), ok bool) {
 | 
				
			||||||
 | 
						var d *net.Dialer
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if d, ok = opts.Context.Value(httpDialerKey{}).(*net.Dialer); ok {
 | 
				
			||||||
 | 
							dialerFunc = func(ctx context.Context, addr string) (net.Conn, error) {
 | 
				
			||||||
 | 
								return d.DialContext(ctx, "tcp", addr)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if opts.ContextDialer != nil {
 | 
				
			||||||
 | 
							dialerFunc, ok = opts.ContextDialer, true
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return dialerFunc, ok
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func defaultHTTPDialer() func(ctx context.Context, addr string) (net.Conn, error) {
 | 
				
			||||||
 | 
						return func(ctx context.Context, addr string) (net.Conn, error) {
 | 
				
			||||||
 | 
							d := &net.Dialer{
 | 
				
			||||||
 | 
								Timeout:   30 * time.Second,
 | 
				
			||||||
 | 
								KeepAlive: 30 * time.Second,
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							return d.DialContext(ctx, "tcp", addr)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// ----------------------------------------------- Method option -------------------------------------------------------
 | 
				
			||||||
type methodKey struct{}
 | 
					type methodKey struct{}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func Method(m string) client.CallOption {
 | 
					func Method(m string) client.CallOption {
 | 
				
			||||||
	return client.SetCallOption(methodKey{}, m)
 | 
						return client.SetCallOption(methodKey{}, m)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func methodFromOpts(opts client.CallOptions) (string, bool) {
 | 
				
			||||||
 | 
						m, ok := opts.Context.Value(methodKey{}).(string)
 | 
				
			||||||
 | 
						return m, ok
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// ------------------------------------------------ Path option --------------------------------------------------------
 | 
				
			||||||
type pathKey struct{}
 | 
					type pathKey struct{}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func Path(p string) client.CallOption {
 | 
					func Path(p string) client.CallOption {
 | 
				
			||||||
	return client.SetCallOption(pathKey{}, p)
 | 
						return client.SetCallOption(pathKey{}, p)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func pathFromOpts(opts client.CallOptions) (string, bool) {
 | 
				
			||||||
 | 
						p, ok := opts.Context.Value(pathKey{}).(string)
 | 
				
			||||||
 | 
						return p, ok
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// ------------------------------------------------ Body option --------------------------------------------------------
 | 
				
			||||||
type bodyKey struct{}
 | 
					type bodyKey struct{}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func Body(b string) client.CallOption {
 | 
					func Body(b string) client.CallOption {
 | 
				
			||||||
	return client.SetCallOption(bodyKey{}, b)
 | 
						return client.SetCallOption(bodyKey{}, b)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func bodyFromOpts(opts client.CallOptions) (string, bool) {
 | 
				
			||||||
 | 
						b, ok := opts.Context.Value(bodyKey{}).(string)
 | 
				
			||||||
 | 
						return b, ok
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// ---------------------------------------------- ErrorMap option ------------------------------------------------------
 | 
				
			||||||
type errorMapKey struct{}
 | 
					type errorMapKey struct{}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func ErrorMap(m map[string]interface{}) client.CallOption {
 | 
					func ErrorMap(m map[string]any) client.CallOption {
 | 
				
			||||||
	return client.SetCallOption(errorMapKey{}, m)
 | 
						return client.SetCallOption(errorMapKey{}, m)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
type structTagsKey struct{}
 | 
					func errorMapFromOpts(opts client.CallOptions) (map[string]any, bool) {
 | 
				
			||||||
 | 
						errMap, ok := opts.Context.Value(errorMapKey{}).(map[string]any)
 | 
				
			||||||
func StructTags(tags []string) client.CallOption {
 | 
						return errMap, ok
 | 
				
			||||||
	return client.SetCallOption(structTagsKey{}, tags)
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// ------------------------------------------------ Cookie option ------------------------------------------------------
 | 
				
			||||||
 | 
					type cookieKey struct{}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func Cookie(cookies ...string) client.CallOption {
 | 
				
			||||||
 | 
						return client.SetCallOption(cookieKey{}, cookies)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func cookieFromOpts(opts client.CallOptions) ([]string, bool) {
 | 
				
			||||||
 | 
						c, ok := opts.Context.Value(cookieKey{}).([]string)
 | 
				
			||||||
 | 
						return c, ok
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// ------------------------------------------------ Header option ------------------------------------------------------
 | 
				
			||||||
 | 
					type headerKey struct{}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func Header(headers ...string) client.CallOption {
 | 
				
			||||||
 | 
						return client.SetCallOption(headerKey{}, headers)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func headerFromOpts(opts client.CallOptions) ([]string, bool) {
 | 
				
			||||||
 | 
						h, ok := opts.Context.Value(headerKey{}).([]string)
 | 
				
			||||||
 | 
						return h, ok
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										42
									
								
								request.go
									
									
									
									
									
								
							
							
						
						
									
										42
									
								
								request.go
									
									
									
									
									
								
							@@ -1,35 +1,15 @@
 | 
				
			|||||||
package http
 | 
					package http
 | 
				
			||||||
 | 
					
 | 
				
			||||||
import (
 | 
					import (
 | 
				
			||||||
	"github.com/unistack-org/micro/v3/client"
 | 
						"go.unistack.org/micro/v3/client"
 | 
				
			||||||
	"github.com/unistack-org/micro/v3/codec"
 | 
						"go.unistack.org/micro/v3/codec"
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
type httpRequest struct {
 | 
					type httpRequest struct {
 | 
				
			||||||
	service     string
 | 
						service string
 | 
				
			||||||
	method      string
 | 
						method  string
 | 
				
			||||||
	contentType string
 | 
						request any
 | 
				
			||||||
	request     interface{}
 | 
						opts    client.RequestOptions
 | 
				
			||||||
	opts        client.RequestOptions
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func newHTTPRequest(service, method string, request interface{}, contentType string, opts ...client.RequestOption) client.Request {
 | 
					 | 
				
			||||||
	options := client.NewRequestOptions(opts...)
 | 
					 | 
				
			||||||
	if len(options.ContentType) == 0 {
 | 
					 | 
				
			||||||
		options.ContentType = contentType
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	return &httpRequest{
 | 
					 | 
				
			||||||
		service:     service,
 | 
					 | 
				
			||||||
		method:      method,
 | 
					 | 
				
			||||||
		request:     request,
 | 
					 | 
				
			||||||
		contentType: options.ContentType,
 | 
					 | 
				
			||||||
		opts:        options,
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpRequest) ContentType() string {
 | 
					 | 
				
			||||||
	return h.contentType
 | 
					 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (h *httpRequest) Service() string {
 | 
					func (h *httpRequest) Service() string {
 | 
				
			||||||
@@ -44,14 +24,18 @@ func (h *httpRequest) Endpoint() string {
 | 
				
			|||||||
	return h.method
 | 
						return h.method
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (h *httpRequest) Codec() codec.Codec {
 | 
					func (h *httpRequest) ContentType() string {
 | 
				
			||||||
	return nil
 | 
						return h.opts.ContentType
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (h *httpRequest) Body() interface{} {
 | 
					func (h *httpRequest) Body() any {
 | 
				
			||||||
	return h.request
 | 
						return h.request
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (h *httpRequest) Codec() codec.Codec {
 | 
				
			||||||
 | 
						return nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (h *httpRequest) Stream() bool {
 | 
					func (h *httpRequest) Stream() bool {
 | 
				
			||||||
	return h.opts.Stream
 | 
						return h.opts.Stream
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										14
									
								
								status/error.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								status/error.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,14 @@
 | 
				
			|||||||
 | 
					package status
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// Error is a thin wrapper around Status that implements the error interface.
 | 
				
			||||||
 | 
					type Error struct {
 | 
				
			||||||
 | 
						s *Status
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (e *Error) Error() string {
 | 
				
			||||||
 | 
						return e.s.String()
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (e *Error) HTTPStatus() *Status {
 | 
				
			||||||
 | 
						return e.s
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										54
									
								
								status/status.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										54
									
								
								status/status.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,54 @@
 | 
				
			|||||||
 | 
					package status
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"errors"
 | 
				
			||||||
 | 
						"fmt"
 | 
				
			||||||
 | 
						"net/http"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// Status represents the outcome of an HTTP request in a style similar to gRPC status.
 | 
				
			||||||
 | 
					type Status struct {
 | 
				
			||||||
 | 
						code    int    // HTTP status code
 | 
				
			||||||
 | 
						message string // HTTP status text
 | 
				
			||||||
 | 
						details any    // parsed error object
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func New(statusCode int) *Status {
 | 
				
			||||||
 | 
						return &Status{code: statusCode, message: http.StatusText(statusCode)}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func FromError(err error) (*Status, bool) {
 | 
				
			||||||
 | 
						if err == nil {
 | 
				
			||||||
 | 
							return nil, false
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						var e *Error
 | 
				
			||||||
 | 
						if errors.As(err, &e) {
 | 
				
			||||||
 | 
							return e.HTTPStatus(), true
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return nil, false
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (s *Status) Code() int {
 | 
				
			||||||
 | 
						return s.code
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (s *Status) Message() string {
 | 
				
			||||||
 | 
						return s.message
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (s *Status) Details() any {
 | 
				
			||||||
 | 
						return s.details
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (s *Status) WithDetails(details any) *Status {
 | 
				
			||||||
 | 
						s.details = details
 | 
				
			||||||
 | 
						return s
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (s *Status) String() string {
 | 
				
			||||||
 | 
						return fmt.Sprintf("http error: code = %d desc = %s", s.Code(), s.Message())
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (s *Status) Err() error {
 | 
				
			||||||
 | 
						return &Error{s: s}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										97
									
								
								status/status_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										97
									
								
								status/status_test.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,97 @@
 | 
				
			|||||||
 | 
					package status_test
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"errors"
 | 
				
			||||||
 | 
						"fmt"
 | 
				
			||||||
 | 
						"net/http"
 | 
				
			||||||
 | 
						"testing"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"github.com/stretchr/testify/require"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"go.unistack.org/micro-client-http/v3/status"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					type fakeError struct{ s *status.Status }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (fe *fakeError) Error() string {
 | 
				
			||||||
 | 
						return fe.s.String()
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestNew(t *testing.T) {
 | 
				
			||||||
 | 
						s := status.New(http.StatusNotFound)
 | 
				
			||||||
 | 
						require.Equal(t, http.StatusNotFound, s.Code())
 | 
				
			||||||
 | 
						require.Equal(t, "Not Found", s.Message())
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestFromError(t *testing.T) {
 | 
				
			||||||
 | 
						tests := []struct {
 | 
				
			||||||
 | 
							name       string
 | 
				
			||||||
 | 
							input      error
 | 
				
			||||||
 | 
							wantStatus *status.Status
 | 
				
			||||||
 | 
							wantOK     bool
 | 
				
			||||||
 | 
						}{
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:       "nil error",
 | 
				
			||||||
 | 
								input:      nil,
 | 
				
			||||||
 | 
								wantStatus: nil,
 | 
				
			||||||
 | 
								wantOK:     false,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:       "simple error",
 | 
				
			||||||
 | 
								input:      errors.New("some error"),
 | 
				
			||||||
 | 
								wantStatus: nil,
 | 
				
			||||||
 | 
								wantOK:     false,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name: "unexpected type of error",
 | 
				
			||||||
 | 
								input: func() error {
 | 
				
			||||||
 | 
									return &fakeError{s: status.New(http.StatusNotFound)}
 | 
				
			||||||
 | 
								}(),
 | 
				
			||||||
 | 
								wantStatus: nil,
 | 
				
			||||||
 | 
								wantOK:     false,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
							{
 | 
				
			||||||
 | 
								name:       "expected type of error",
 | 
				
			||||||
 | 
								input:      status.New(http.StatusNotFound).Err(),
 | 
				
			||||||
 | 
								wantStatus: status.New(http.StatusNotFound),
 | 
				
			||||||
 | 
								wantOK:     true,
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, tt := range tests {
 | 
				
			||||||
 | 
							t.Run(tt.name, func(t *testing.T) {
 | 
				
			||||||
 | 
								result, ok := status.FromError(tt.input)
 | 
				
			||||||
 | 
								require.Equal(t, tt.wantStatus, result)
 | 
				
			||||||
 | 
								require.Equal(t, tt.wantOK, ok)
 | 
				
			||||||
 | 
							})
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestStatus_Code(t *testing.T) {
 | 
				
			||||||
 | 
						s := status.New(http.StatusNotFound)
 | 
				
			||||||
 | 
						require.Equal(t, http.StatusNotFound, s.Code())
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestStatus_Message(t *testing.T) {
 | 
				
			||||||
 | 
						s := status.New(http.StatusNotFound)
 | 
				
			||||||
 | 
						require.Equal(t, "Not Found", s.Message())
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestStatus_WithDetails(t *testing.T) {
 | 
				
			||||||
 | 
						s := status.New(http.StatusNotFound).WithDetails(errors.New("some error"))
 | 
				
			||||||
 | 
						require.Equal(t, errors.New("some error"), s.Details())
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestStatus_String(t *testing.T) {
 | 
				
			||||||
 | 
						s := status.New(http.StatusInternalServerError)
 | 
				
			||||||
 | 
						expected := fmt.Sprintf("http error: code = %d desc = %s", 500, "Internal Server Error")
 | 
				
			||||||
 | 
						require.Equal(t, expected, s.String())
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestStatus_Err(t *testing.T) {
 | 
				
			||||||
 | 
						var e *status.Error
 | 
				
			||||||
 | 
						s := status.New(http.StatusForbidden)
 | 
				
			||||||
 | 
						require.Error(t, s.Err())
 | 
				
			||||||
 | 
						require.ErrorAs(t, s.Err(), &e)
 | 
				
			||||||
 | 
						require.Equal(t, status.New(http.StatusForbidden), e.HTTPStatus())
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										148
									
								
								stream.go
									
									
									
									
									
								
							
							
						
						
									
										148
									
								
								stream.go
									
									
									
									
									
								
							@@ -1,148 +0,0 @@
 | 
				
			|||||||
package http
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
import (
 | 
					 | 
				
			||||||
	"bufio"
 | 
					 | 
				
			||||||
	"context"
 | 
					 | 
				
			||||||
	"fmt"
 | 
					 | 
				
			||||||
	"io"
 | 
					 | 
				
			||||||
	"net"
 | 
					 | 
				
			||||||
	"net/http"
 | 
					 | 
				
			||||||
	"sync"
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	"github.com/unistack-org/micro/v3/client"
 | 
					 | 
				
			||||||
	"github.com/unistack-org/micro/v3/codec"
 | 
					 | 
				
			||||||
	"github.com/unistack-org/micro/v3/errors"
 | 
					 | 
				
			||||||
)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
// Implements the streamer interface
 | 
					 | 
				
			||||||
type httpStream struct {
 | 
					 | 
				
			||||||
	err     error
 | 
					 | 
				
			||||||
	conn    net.Conn
 | 
					 | 
				
			||||||
	cf      codec.Codec
 | 
					 | 
				
			||||||
	context context.Context
 | 
					 | 
				
			||||||
	request client.Request
 | 
					 | 
				
			||||||
	header  http.Header
 | 
					 | 
				
			||||||
	closed  chan bool
 | 
					 | 
				
			||||||
	reader  *bufio.Reader
 | 
					 | 
				
			||||||
	address string
 | 
					 | 
				
			||||||
	ct      string
 | 
					 | 
				
			||||||
	opts    client.CallOptions
 | 
					 | 
				
			||||||
	sync.RWMutex
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
var errShutdown = fmt.Errorf("connection is shut down")
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpStream) isClosed() bool {
 | 
					 | 
				
			||||||
	select {
 | 
					 | 
				
			||||||
	case <-h.closed:
 | 
					 | 
				
			||||||
		return true
 | 
					 | 
				
			||||||
	default:
 | 
					 | 
				
			||||||
		return false
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpStream) Context() context.Context {
 | 
					 | 
				
			||||||
	return h.context
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpStream) Request() client.Request {
 | 
					 | 
				
			||||||
	return h.request
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpStream) Response() client.Response {
 | 
					 | 
				
			||||||
	return nil
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpStream) Send(msg interface{}) error {
 | 
					 | 
				
			||||||
	h.Lock()
 | 
					 | 
				
			||||||
	defer h.Unlock()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if h.isClosed() {
 | 
					 | 
				
			||||||
		h.err = errShutdown
 | 
					 | 
				
			||||||
		return errShutdown
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	hreq, err := newRequest(h.address, h.request, h.ct, h.cf, msg, h.opts)
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	hreq.Header = h.header
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	return hreq.Write(h.conn)
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpStream) Recv(msg interface{}) error {
 | 
					 | 
				
			||||||
	h.Lock()
 | 
					 | 
				
			||||||
	defer h.Unlock()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if h.isClosed() {
 | 
					 | 
				
			||||||
		h.err = errShutdown
 | 
					 | 
				
			||||||
		return errShutdown
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	hrsp, err := http.ReadResponse(h.reader, new(http.Request))
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return errors.InternalServerError("go.micro.client", err.Error())
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	defer hrsp.Body.Close()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	return h.parseRsp(h.context, hrsp, h.cf, msg, h.opts)
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpStream) Error() error {
 | 
					 | 
				
			||||||
	h.RLock()
 | 
					 | 
				
			||||||
	defer h.RUnlock()
 | 
					 | 
				
			||||||
	return h.err
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpStream) Close() error {
 | 
					 | 
				
			||||||
	select {
 | 
					 | 
				
			||||||
	case <-h.closed:
 | 
					 | 
				
			||||||
		return nil
 | 
					 | 
				
			||||||
	default:
 | 
					 | 
				
			||||||
		close(h.closed)
 | 
					 | 
				
			||||||
		return h.conn.Close()
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpStream) parseRsp(ctx context.Context, hrsp *http.Response, cf codec.Codec, rsp interface{}, opts client.CallOptions) error {
 | 
					 | 
				
			||||||
	var err error
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	select {
 | 
					 | 
				
			||||||
	case <-ctx.Done():
 | 
					 | 
				
			||||||
		err = ctx.Err()
 | 
					 | 
				
			||||||
	default:
 | 
					 | 
				
			||||||
		// fast path return
 | 
					 | 
				
			||||||
		if hrsp.StatusCode == http.StatusNoContent {
 | 
					 | 
				
			||||||
			return nil
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		if hrsp.StatusCode < 400 {
 | 
					 | 
				
			||||||
			if err = cf.ReadBody(hrsp.Body, rsp); err != nil {
 | 
					 | 
				
			||||||
				return errors.InternalServerError("go.micro.client", err.Error())
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
			return nil
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		errmap, ok := opts.Context.Value(errorMapKey{}).(map[string]interface{})
 | 
					 | 
				
			||||||
		if ok && errmap != nil {
 | 
					 | 
				
			||||||
			if err, ok = errmap[fmt.Sprintf("%d", hrsp.StatusCode)].(error); !ok {
 | 
					 | 
				
			||||||
				err, ok = errmap["default"].(error)
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		if !ok || err == nil {
 | 
					 | 
				
			||||||
			buf, cerr := io.ReadAll(hrsp.Body)
 | 
					 | 
				
			||||||
			if cerr != nil {
 | 
					 | 
				
			||||||
				return errors.InternalServerError("go.micro.client", cerr.Error())
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
			return errors.New("go.micro.client", string(buf), int32(hrsp.StatusCode))
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		if cerr := cf.ReadBody(hrsp.Body, err); cerr != nil {
 | 
					 | 
				
			||||||
			err = errors.InternalServerError("go.micro.client", cerr.Error())
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	return err
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
							
								
								
									
										258
									
								
								util.go
									
									
									
									
									
								
							
							
						
						
									
										258
									
								
								util.go
									
									
									
									
									
								
							@@ -1,258 +0,0 @@
 | 
				
			|||||||
package http
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
import (
 | 
					 | 
				
			||||||
	"context"
 | 
					 | 
				
			||||||
	"fmt"
 | 
					 | 
				
			||||||
	"io"
 | 
					 | 
				
			||||||
	"net/http"
 | 
					 | 
				
			||||||
	"net/url"
 | 
					 | 
				
			||||||
	"reflect"
 | 
					 | 
				
			||||||
	"strings"
 | 
					 | 
				
			||||||
	"sync"
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	"github.com/unistack-org/micro/v3/client"
 | 
					 | 
				
			||||||
	"github.com/unistack-org/micro/v3/errors"
 | 
					 | 
				
			||||||
	rutil "github.com/unistack-org/micro/v3/util/reflect"
 | 
					 | 
				
			||||||
	util "github.com/unistack-org/micro/v3/util/router"
 | 
					 | 
				
			||||||
)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
var (
 | 
					 | 
				
			||||||
	templateCache = make(map[string]util.Template)
 | 
					 | 
				
			||||||
	mu            sync.RWMutex
 | 
					 | 
				
			||||||
)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
// Error struct holds error
 | 
					 | 
				
			||||||
type Error struct {
 | 
					 | 
				
			||||||
	err interface{}
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
// Error func for error interface
 | 
					 | 
				
			||||||
func (err *Error) Error() string {
 | 
					 | 
				
			||||||
	return fmt.Sprintf("%v", err.err)
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func GetError(err error) interface{} {
 | 
					 | 
				
			||||||
	if rerr, ok := err.(*Error); ok {
 | 
					 | 
				
			||||||
		return rerr.err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	return err
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func newPathRequest(path string, method string, body string, msg interface{}, tags []string) (string, interface{}, error) {
 | 
					 | 
				
			||||||
	// parse via https://github.com/googleapis/googleapis/blob/master/google/api/http.proto definition
 | 
					 | 
				
			||||||
	tpl, err := newTemplate(path)
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return "", nil, err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if len(tpl.Fields) > 0 && msg == nil {
 | 
					 | 
				
			||||||
		return "", nil, fmt.Errorf("nil message but path params requested: %v", path)
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	fieldsmapskip := make(map[string]struct{})
 | 
					 | 
				
			||||||
	fieldsmap := make(map[string]string, len(tpl.Fields))
 | 
					 | 
				
			||||||
	for _, v := range tpl.Fields {
 | 
					 | 
				
			||||||
		fieldsmap[v] = ""
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	nmsg, err := rutil.Zero(msg)
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return "", nil, err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// we cant switch on message and use proto helpers, to avoid dependency to protobuf
 | 
					 | 
				
			||||||
	tmsg := reflect.ValueOf(msg)
 | 
					 | 
				
			||||||
	if tmsg.Kind() == reflect.Ptr {
 | 
					 | 
				
			||||||
		tmsg = tmsg.Elem()
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	tnmsg := reflect.ValueOf(nmsg)
 | 
					 | 
				
			||||||
	if tnmsg.Kind() == reflect.Ptr {
 | 
					 | 
				
			||||||
		tnmsg = tnmsg.Elem()
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	values := url.Values{}
 | 
					 | 
				
			||||||
	// copy cycle
 | 
					 | 
				
			||||||
	for i := 0; i < tmsg.NumField(); i++ {
 | 
					 | 
				
			||||||
		val := tmsg.Field(i)
 | 
					 | 
				
			||||||
		if val.IsZero() {
 | 
					 | 
				
			||||||
			continue
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		fld := tmsg.Type().Field(i)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		t := &tag{}
 | 
					 | 
				
			||||||
		for _, tn := range tags {
 | 
					 | 
				
			||||||
			ts, ok := fld.Tag.Lookup(tn)
 | 
					 | 
				
			||||||
			if !ok {
 | 
					 | 
				
			||||||
				continue
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
			tp := strings.Split(ts, ",")
 | 
					 | 
				
			||||||
			// special
 | 
					 | 
				
			||||||
			switch tn {
 | 
					 | 
				
			||||||
			case "protobuf": // special
 | 
					 | 
				
			||||||
				for _, p := range tp {
 | 
					 | 
				
			||||||
					if idx := strings.Index(p, "name="); idx > 0 {
 | 
					 | 
				
			||||||
						t = &tag{key: tn, name: p[idx:]}
 | 
					 | 
				
			||||||
					}
 | 
					 | 
				
			||||||
				}
 | 
					 | 
				
			||||||
			default:
 | 
					 | 
				
			||||||
				t = &tag{key: tn, name: tp[0]}
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
			if t.name != "" {
 | 
					 | 
				
			||||||
				break
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		if t.name == "" {
 | 
					 | 
				
			||||||
			// fallback to lowercase
 | 
					 | 
				
			||||||
			t.name = strings.ToLower(fld.Name)
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		if !val.IsValid() || val.IsZero() {
 | 
					 | 
				
			||||||
			continue
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		// nolint: gocritic
 | 
					 | 
				
			||||||
		if _, ok := fieldsmap[t.name]; ok {
 | 
					 | 
				
			||||||
			switch val.Type().Kind() {
 | 
					 | 
				
			||||||
			case reflect.Slice:
 | 
					 | 
				
			||||||
				for idx := 0; idx < val.Len(); idx++ {
 | 
					 | 
				
			||||||
					values.Add(t.name, fmt.Sprintf("%v", val.Index(idx).Interface()))
 | 
					 | 
				
			||||||
				}
 | 
					 | 
				
			||||||
				fieldsmapskip[t.name] = struct{}{}
 | 
					 | 
				
			||||||
			default:
 | 
					 | 
				
			||||||
				fieldsmap[t.name] = fmt.Sprintf("%v", val.Interface())
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
		} else if (body == "*" || body == t.name) && method != http.MethodGet {
 | 
					 | 
				
			||||||
			tnmsg.Field(i).Set(val)
 | 
					 | 
				
			||||||
		} else {
 | 
					 | 
				
			||||||
			if val.Type().Kind() == reflect.Slice {
 | 
					 | 
				
			||||||
				for idx := 0; idx < val.Len(); idx++ {
 | 
					 | 
				
			||||||
					values.Add(t.name, fmt.Sprintf("%v", val.Index(idx).Interface()))
 | 
					 | 
				
			||||||
				}
 | 
					 | 
				
			||||||
			} else {
 | 
					 | 
				
			||||||
				values.Add(t.name, fmt.Sprintf("%v", val.Interface()))
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// check not filled stuff
 | 
					 | 
				
			||||||
	for k, v := range fieldsmap {
 | 
					 | 
				
			||||||
		_, ok := fieldsmapskip[k]
 | 
					 | 
				
			||||||
		if !ok && v == "" {
 | 
					 | 
				
			||||||
			return "", nil, fmt.Errorf("path param %s not filled", k)
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	var b strings.Builder
 | 
					 | 
				
			||||||
	for _, fld := range tpl.Pool {
 | 
					 | 
				
			||||||
		_, _ = b.WriteRune('/')
 | 
					 | 
				
			||||||
		if v, ok := fieldsmap[fld]; ok {
 | 
					 | 
				
			||||||
			if v != "" {
 | 
					 | 
				
			||||||
				_, _ = b.WriteString(v)
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
		} else {
 | 
					 | 
				
			||||||
			_, _ = b.WriteString(fld)
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if len(values) > 0 {
 | 
					 | 
				
			||||||
		_, _ = b.WriteRune('?')
 | 
					 | 
				
			||||||
		_, _ = b.WriteString(values.Encode())
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if rutil.IsZero(nmsg) {
 | 
					 | 
				
			||||||
		return b.String(), nil, nil
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	return b.String(), nmsg, nil
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func newTemplate(path string) (util.Template, error) {
 | 
					 | 
				
			||||||
	mu.RLock()
 | 
					 | 
				
			||||||
	tpl, ok := templateCache[path]
 | 
					 | 
				
			||||||
	if ok {
 | 
					 | 
				
			||||||
		mu.RUnlock()
 | 
					 | 
				
			||||||
		return tpl, nil
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	mu.RUnlock()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	rule, err := util.Parse(path)
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return tpl, err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	tpl = rule.Compile()
 | 
					 | 
				
			||||||
	mu.Lock()
 | 
					 | 
				
			||||||
	templateCache[path] = tpl
 | 
					 | 
				
			||||||
	mu.Unlock()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	return tpl, nil
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (h *httpClient) parseRsp(ctx context.Context, hrsp *http.Response, rsp interface{}, opts client.CallOptions) error {
 | 
					 | 
				
			||||||
	var err error
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	select {
 | 
					 | 
				
			||||||
	case <-ctx.Done():
 | 
					 | 
				
			||||||
		err = ctx.Err()
 | 
					 | 
				
			||||||
	default:
 | 
					 | 
				
			||||||
		// fast path return
 | 
					 | 
				
			||||||
		if hrsp.StatusCode == http.StatusNoContent {
 | 
					 | 
				
			||||||
			return nil
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		ct := DefaultContentType
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		if htype := hrsp.Header.Get("Content-Type"); htype != "" {
 | 
					 | 
				
			||||||
			ct = htype
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		cf, cerr := h.newCodec(ct)
 | 
					 | 
				
			||||||
		if cerr != nil {
 | 
					 | 
				
			||||||
			return errors.InternalServerError("go.micro.client", cerr.Error())
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		// succeseful response
 | 
					 | 
				
			||||||
		if hrsp.StatusCode < 400 {
 | 
					 | 
				
			||||||
			if err = cf.ReadBody(hrsp.Body, rsp); err != nil {
 | 
					 | 
				
			||||||
				return errors.InternalServerError("go.micro.client", err.Error())
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
			return nil
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		// response with error
 | 
					 | 
				
			||||||
		var rerr interface{}
 | 
					 | 
				
			||||||
		errmap, ok := opts.Context.Value(errorMapKey{}).(map[string]interface{})
 | 
					 | 
				
			||||||
		if ok && errmap != nil {
 | 
					 | 
				
			||||||
			rerr, ok = errmap[fmt.Sprintf("%d", hrsp.StatusCode)]
 | 
					 | 
				
			||||||
			if !ok {
 | 
					 | 
				
			||||||
				rerr, ok = errmap["default"]
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		if !ok || rerr == nil {
 | 
					 | 
				
			||||||
			buf, rerr := io.ReadAll(hrsp.Body)
 | 
					 | 
				
			||||||
			if rerr != nil {
 | 
					 | 
				
			||||||
				return errors.InternalServerError("go.micro.client", rerr.Error())
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
			return errors.New("go.micro.client", string(buf), int32(hrsp.StatusCode))
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		if cerr := cf.ReadBody(hrsp.Body, rerr); cerr != nil {
 | 
					 | 
				
			||||||
			return errors.InternalServerError("go.micro.client", cerr.Error())
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		if err, ok = rerr.(error); !ok {
 | 
					 | 
				
			||||||
			err = &Error{rerr}
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	return err
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
type tag struct {
 | 
					 | 
				
			||||||
	key  string
 | 
					 | 
				
			||||||
	name string
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
							
								
								
									
										81
									
								
								util_test.go
									
									
									
									
									
								
							
							
						
						
									
										81
									
								
								util_test.go
									
									
									
									
									
								
							@@ -1,81 +0,0 @@
 | 
				
			|||||||
package http
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
import (
 | 
					 | 
				
			||||||
	"net/url"
 | 
					 | 
				
			||||||
	"testing"
 | 
					 | 
				
			||||||
)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func TestTemplate(t *testing.T) {
 | 
					 | 
				
			||||||
	tpl, err := newTemplate("/v1/{ClientID}/list")
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		t.Fatal(err)
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	_ = tpl
 | 
					 | 
				
			||||||
	//	fmt.Printf("%#+v\n", tpl.Pool)
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func TestNewPathRequest(t *testing.T) {
 | 
					 | 
				
			||||||
	type Message struct {
 | 
					 | 
				
			||||||
		Name string `json:"name"`
 | 
					 | 
				
			||||||
		Val1 string `protobuf:"bytes,1,opt,name=val1,proto3" json:"val1"`
 | 
					 | 
				
			||||||
		Val3 []string
 | 
					 | 
				
			||||||
		Val2 int64
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	omsg := &Message{Name: "test_name", Val1: "test_val1", Val2: 100, Val3: []string{"slice"}}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	for _, m := range []string{"POST", "PUT", "PATCH", "GET", "DELETE"} {
 | 
					 | 
				
			||||||
		body := ""
 | 
					 | 
				
			||||||
		path, nmsg, err := newPathRequest("/v1/test", m, body, omsg, []string{"protobuf", "json"})
 | 
					 | 
				
			||||||
		if err != nil {
 | 
					 | 
				
			||||||
			t.Fatal(err)
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		u, err := url.Parse(path)
 | 
					 | 
				
			||||||
		if err != nil {
 | 
					 | 
				
			||||||
			t.Fatal(err)
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		vals := u.Query()
 | 
					 | 
				
			||||||
		if v, ok := vals["name"]; !ok || v[0] != "test_name" {
 | 
					 | 
				
			||||||
			t.Fatalf("invalid path: %v nmsg: %v", path, nmsg)
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func TestNewPathVarRequest(t *testing.T) {
 | 
					 | 
				
			||||||
	type Message struct {
 | 
					 | 
				
			||||||
		Name string `json:"name"`
 | 
					 | 
				
			||||||
		Val1 string `protobuf:"bytes,1,opt,name=val1,proto3" json:"val1"`
 | 
					 | 
				
			||||||
		Val3 []string
 | 
					 | 
				
			||||||
		Val2 int64
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	omsg := &Message{Name: "test_name", Val1: "test_val1", Val2: 100, Val3: []string{"slice"}}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	for _, m := range []string{"POST", "PUT", "PATCH", "GET", "DELETE"} {
 | 
					 | 
				
			||||||
		body := ""
 | 
					 | 
				
			||||||
		if m != "GET" {
 | 
					 | 
				
			||||||
			body = "*"
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		path, nmsg, err := newPathRequest("/v1/test/{val1}", m, body, omsg, []string{"protobuf", "json"})
 | 
					 | 
				
			||||||
		if err != nil {
 | 
					 | 
				
			||||||
			t.Fatal(err)
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		u, err := url.Parse(path)
 | 
					 | 
				
			||||||
		if err != nil {
 | 
					 | 
				
			||||||
			t.Fatal(err)
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		if m != "GET" {
 | 
					 | 
				
			||||||
			if _, ok := nmsg.(*Message); !ok {
 | 
					 | 
				
			||||||
				t.Fatalf("invalid nmsg: %#+v\n", nmsg)
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
			if nmsg.(*Message).Name != "test_name" {
 | 
					 | 
				
			||||||
				t.Fatalf("invalid nmsg: %v nmsg: %v", path, nmsg)
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
		} else {
 | 
					 | 
				
			||||||
			vals := u.Query()
 | 
					 | 
				
			||||||
			if v, ok := vals["val2"]; !ok || v[0] != "100" {
 | 
					 | 
				
			||||||
				t.Fatalf("invalid path: %v nmsg: %v", path, nmsg)
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
		Reference in New Issue
	
	Block a user