Compare commits
202 Commits
Author | SHA1 | Date | |
---|---|---|---|
|
9bb2f8cffa | ||
24801750a7 | |||
|
b37fca95cf | ||
4c4a024e19 | |||
b93e634873 | |||
4a6414b2b8 | |||
938ddd2ab3 | |||
04e0597b73 | |||
4d55c7b90a | |||
fbe82a3905 | |||
|
14490a419e | ||
d434261168 | |||
2e384a5b75 | |||
d9e7ac75ef | |||
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 |
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: ''
|
||||||
|
8
.github/ISSUE_TEMPLATE/question.md
vendored
8
.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
|
|
||||||
|
28
.github/autoapprove.yml
vendored
Normal file
28
.github/autoapprove.yml
vendored
Normal file
@@ -0,0 +1,28 @@
|
|||||||
|
name: "autoapprove"
|
||||||
|
|
||||||
|
on:
|
||||||
|
pull_request_target:
|
||||||
|
types: [assigned, opened, synchronize, reopened]
|
||||||
|
workflow_run:
|
||||||
|
workflows: ["prbuild"]
|
||||||
|
types:
|
||||||
|
- completed
|
||||||
|
|
||||||
|
permissions:
|
||||||
|
pull-requests: write
|
||||||
|
contents: write
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
autoapprove:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- name: approve
|
||||||
|
run: [ "curl -o tea https://dl.gitea.com/tea/main/tea-main-linux-amd64",
|
||||||
|
"chmod +x ./tea",
|
||||||
|
"./tea login add --name unistack --token ${{ secrets.GITHUB_TOKEN }} --url https://git.unistack.org",
|
||||||
|
"./tea pr --repo ${{ github.event.repository.name }}"
|
||||||
|
]
|
||||||
|
if: github.actor == 'vtolstov'
|
||||||
|
id: approve
|
||||||
|
with:
|
||||||
|
github-token: ${{ secrets.GITHUB_TOKEN }}
|
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'
|
94
.github/workflows/job_sync.yml
vendored
Normal file
94
.github/workflows/job_sync.yml
vendored
Normal file
@@ -0,0 +1,94 @@
|
|||||||
|
name: sync
|
||||||
|
|
||||||
|
on:
|
||||||
|
schedule:
|
||||||
|
- cron: '*/5 * * * *'
|
||||||
|
# Allows you to run this workflow manually from the Actions tab
|
||||||
|
workflow_dispatch:
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
sync:
|
||||||
|
if: github.server_url != 'https://github.com'
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- name: init
|
||||||
|
run: |
|
||||||
|
git config --global user.email "vtolstov <vtolstov@users.noreply.github.com>"
|
||||||
|
git config --global user.name "github-actions[bot]"
|
||||||
|
echo "machine git.unistack.org login vtolstov password ${{ secrets.TOKEN_GITEA }}" >> /root/.netrc
|
||||||
|
echo "machine github.com login vtolstov password ${{ secrets.TOKEN_GITHUB }}" >> /root/.netrc
|
||||||
|
|
||||||
|
- name: check master
|
||||||
|
id: check_master
|
||||||
|
run: |
|
||||||
|
src_hash=$(git ls-remote https://github.com/${GITHUB_REPOSITORY} refs/heads/master | cut -f1)
|
||||||
|
dst_hash=$(git ls-remote ${GITHUB_SERVER_URL}/${GITHUB_REPOSITORY} refs/heads/master | cut -f1)
|
||||||
|
echo "src_hash=$src_hash"
|
||||||
|
echo "dst_hash=$dst_hash"
|
||||||
|
if [ "$src_hash" != "$dst_hash" ]; then
|
||||||
|
echo "sync_needed=true" >> $GITHUB_OUTPUT
|
||||||
|
else
|
||||||
|
echo "sync_needed=false" >> $GITHUB_OUTPUT
|
||||||
|
fi
|
||||||
|
|
||||||
|
- name: sync master
|
||||||
|
if: steps.check_master.outputs.sync_needed == 'true'
|
||||||
|
run: |
|
||||||
|
git clone --filter=blob:none --filter=tree:0 --branch master --single-branch ${GITHUB_SERVER_URL}/${GITHUB_REPOSITORY} repo
|
||||||
|
cd repo
|
||||||
|
git remote add --no-tags --fetch --track master upstream https://github.com/${GITHUB_REPOSITORY}
|
||||||
|
git pull --rebase upstream master
|
||||||
|
git push upstream master --progress
|
||||||
|
git push origin master --progress
|
||||||
|
cd ../
|
||||||
|
rm -rf repo
|
||||||
|
|
||||||
|
- name: check v3
|
||||||
|
id: check_v3
|
||||||
|
run: |
|
||||||
|
src_hash=$(git ls-remote https://github.com/${GITHUB_REPOSITORY} refs/heads/v3 | cut -f1)
|
||||||
|
dst_hash=$(git ls-remote ${GITHUB_SERVER_URL}/${GITHUB_REPOSITORY} refs/heads/v3 | cut -f1)
|
||||||
|
echo "src_hash=$src_hash"
|
||||||
|
echo "dst_hash=$dst_hash"
|
||||||
|
if [ "$src_hash" != "$dst_hash" ]; then
|
||||||
|
echo "sync_needed=true" >> $GITHUB_OUTPUT
|
||||||
|
else
|
||||||
|
echo "sync_needed=false" >> $GITHUB_OUTPUT
|
||||||
|
fi
|
||||||
|
|
||||||
|
- name: sync v3
|
||||||
|
if: steps.check_v3.outputs.sync_needed == 'true'
|
||||||
|
run: |
|
||||||
|
git clone --filter=blob:none --filter=tree:0 --branch v3 --single-branch ${GITHUB_SERVER_URL}/${GITHUB_REPOSITORY} repo
|
||||||
|
cd repo
|
||||||
|
git remote add --no-tags --fetch --track v3 upstream https://github.com/${GITHUB_REPOSITORY}
|
||||||
|
git pull --rebase upstream v3
|
||||||
|
git push upstream v3 --progress
|
||||||
|
git push origin v3 --progress
|
||||||
|
cd ../
|
||||||
|
rm -rf repo
|
||||||
|
|
||||||
|
- name: check v4
|
||||||
|
id: check_v4
|
||||||
|
run: |
|
||||||
|
src_hash=$(git ls-remote https://github.com/${GITHUB_REPOSITORY} refs/heads/v4 | cut -f1)
|
||||||
|
dst_hash=$(git ls-remote ${GITHUB_SERVER_URL}/${GITHUB_REPOSITORY} refs/heads/v4 | cut -f1)
|
||||||
|
echo "src_hash=$src_hash"
|
||||||
|
echo "dst_hash=$dst_hash"
|
||||||
|
if [ "$src_hash" != "$dst_hash" ]; then
|
||||||
|
echo "sync_needed=true" >> $GITHUB_OUTPUT
|
||||||
|
else
|
||||||
|
echo "sync_needed=false" >> $GITHUB_OUTPUT
|
||||||
|
fi
|
||||||
|
|
||||||
|
- name: sync v4
|
||||||
|
if: steps.check_v4.outputs.sync_needed == 'true'
|
||||||
|
run: |
|
||||||
|
git clone --filter=blob:none --filter=tree:0 --branch v4 --single-branch ${GITHUB_SERVER_URL}/${GITHUB_REPOSITORY} repo
|
||||||
|
cd repo
|
||||||
|
git remote add --no-tags --fetch --track v4 upstream https://github.com/${GITHUB_REPOSITORY}
|
||||||
|
git pull --rebase upstream v4
|
||||||
|
git push upstream v4 --progress
|
||||||
|
git push origin v4 --progress
|
||||||
|
cd ../
|
||||||
|
rm -rf repo
|
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
|
|
||||||
|
150
README.md
150
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/v4).
|
||||||
|
It implements the [micro.Client](https://pkg.go.dev/go.unistack.org/micro/v4/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/v4"
|
||||||
|
http "go.unistack.org/micro-client-http/v4"
|
||||||
|
)
|
||||||
|
|
||||||
service := micro.NewService(
|
service := micro.NewService(
|
||||||
micro.Name("my.service"),
|
micro.Name("my.service"),
|
||||||
@@ -20,43 +28,127 @@ 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/v4/client"
|
||||||
|
http "go.unistack.org/micro-client-http/v4"
|
||||||
|
jsoncodec "go.unistack.org/micro-codec-json/v4"
|
||||||
|
)
|
||||||
|
|
||||||
// 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/v4/client"
|
||||||
|
http "go.unistack.org/micro-client-http/v4"
|
||||||
|
)
|
||||||
|
|
||||||
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/v4/metadata"
|
||||||
|
http "go.unistack.org/micro-client-http/v4"
|
||||||
|
)
|
||||||
|
|
||||||
|
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/v4/metadata"
|
||||||
|
http "go.unistack.org/micro-client-http/v4"
|
||||||
|
)
|
||||||
|
|
||||||
|
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/v4/metadata"
|
||||||
|
http "go.unistack.org/micro-client-http/v4"
|
||||||
|
)
|
||||||
|
|
||||||
|
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/v4"
|
||||||
|
jsoncodec "go.unistack.org/micro-codec-json/v4"
|
||||||
|
)
|
||||||
|
|
||||||
|
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
|
||||||
|
}),
|
||||||
|
)
|
||||||
|
```
|
||||||
|
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/v4/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/v4/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/v4/client"
|
||||||
|
"go.unistack.org/micro/v4/errors"
|
||||||
|
"go.unistack.org/micro/v4/options"
|
||||||
|
"go.unistack.org/micro/v4/semconv"
|
||||||
|
"go.unistack.org/micro/v4/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"
|
||||||
|
}
|
263
client_helpers.go
Normal file
263
client_helpers.go
Normal file
@@ -0,0 +1,263 @@
|
|||||||
|
package http
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"net/url"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"go.unistack.org/micro/v4/client"
|
||||||
|
"go.unistack.org/micro/v4/codec"
|
||||||
|
"go.unistack.org/micro/v4/logger"
|
||||||
|
"go.unistack.org/micro/v4/metadata"
|
||||||
|
"google.golang.org/protobuf/proto"
|
||||||
|
"google.golang.org/protobuf/reflect/protoreflect"
|
||||||
|
|
||||||
|
"go.unistack.org/micro-client-http/v4/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
|
||||||
|
}
|
||||||
|
|
||||||
|
raw := strings.Join(rawCookies, "; ")
|
||||||
|
|
||||||
|
tmp := http.Request{Header: http.Header{}}
|
||||||
|
tmp.Header.Set("Cookie", raw)
|
||||||
|
|
||||||
|
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/v4"
|
||||||
|
"google.golang.org/protobuf/proto"
|
||||||
|
|
||||||
|
pb "go.unistack.org/micro-client-http/v4/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: []string{},
|
||||||
|
want: []*http.Cookie{},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "single cookie",
|
||||||
|
rawCookies: []string{"session=abc123"},
|
||||||
|
want: []*http.Cookie{
|
||||||
|
{Name: "session", Value: "abc123"},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "multiple cookies separate items",
|
||||||
|
rawCookies: []string{"session=abc123", "user=john"},
|
||||||
|
want: []*http.Cookie{
|
||||||
|
{Name: "session", Value: "abc123"},
|
||||||
|
{Name: "user", Value: "john"},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "multiple cookies in one item",
|
||||||
|
rawCookies: []string{"a=1; b=2"},
|
||||||
|
want: []*http.Cookie{
|
||||||
|
{Name: "a", Value: "1"},
|
||||||
|
{Name: "b", Value: "2"},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "mix of combined and separate cookies",
|
||||||
|
rawCookies: []string{"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: []string{"session=abc123", "session=xyz"},
|
||||||
|
want: []*http.Cookie{
|
||||||
|
{Name: "session", Value: "abc123"},
|
||||||
|
{Name: "session", Value: "xyz"},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "cookie with spaces",
|
||||||
|
rawCookies: []string{"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/v4/client"
|
||||||
|
)
|
||||||
|
|
||||||
|
// TODO: Add stream support in the future.
|
||||||
|
|
||||||
|
func (c *Client) fnStream(context.Context, client.Request, ...client.CallOption) (client.Stream, error) {
|
||||||
|
panic("not implemented")
|
||||||
|
}
|
273
client_unary_call.go
Normal file
273
client_unary_call.go
Normal file
@@ -0,0 +1,273 @@
|
|||||||
|
package http
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net"
|
||||||
|
"net/http"
|
||||||
|
"net/url"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"go.unistack.org/micro/v4/client"
|
||||||
|
"go.unistack.org/micro/v4/codec"
|
||||||
|
"go.unistack.org/micro/v4/errors"
|
||||||
|
"go.unistack.org/micro/v4/logger"
|
||||||
|
"go.unistack.org/micro/v4/metadata"
|
||||||
|
"go.unistack.org/micro/v4/selector"
|
||||||
|
)
|
||||||
|
|
||||||
|
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
|
||||||
|
}
|
||||||
|
|
||||||
|
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 errors.New("go.micro.client", string(buf), int32(hrsp.StatusCode))
|
||||||
|
}
|
||||||
|
|
||||||
|
if err = cf.Unmarshal(buf, mappedErr); err != nil {
|
||||||
|
return errors.InternalServerError("go.micro.client", "unmarshal response: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if v, ok := mappedErr.(error); ok {
|
||||||
|
return v
|
||||||
|
}
|
||||||
|
|
||||||
|
// if the error map item does not implement the error interface, wrap it
|
||||||
|
return &Error{err: mappedErr}
|
||||||
|
}
|
1284
client_unary_call_test.go
Normal file
1284
client_unary_call_test.go
Normal file
File diff suppressed because it is too large
Load Diff
12
error.go
Normal file
12
error.go
Normal file
@@ -0,0 +1,12 @@
|
|||||||
|
package http
|
||||||
|
|
||||||
|
import "fmt"
|
||||||
|
|
||||||
|
// Error is used when items in the error map do not implement the error interface and need to be wrapped.
|
||||||
|
type Error struct {
|
||||||
|
err any
|
||||||
|
}
|
||||||
|
|
||||||
|
func (e *Error) Error() string {
|
||||||
|
return fmt.Sprintf("%+v", e.err)
|
||||||
|
}
|
26
go.mod
26
go.mod
@@ -1,5 +1,25 @@
|
|||||||
module github.com/unistack-org/micro-client-http/v3
|
module go.unistack.org/micro-client-http/v4
|
||||||
|
|
||||||
go 1.16
|
go 1.23.0
|
||||||
|
|
||||||
require github.com/unistack-org/micro/v3 v3.5.2
|
toolchain go1.24.2
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/stretchr/testify v1.11.1
|
||||||
|
go.unistack.org/micro-codec-json/v4 v4.1.0
|
||||||
|
go.unistack.org/micro/v4 v4.1.19
|
||||||
|
google.golang.org/protobuf v1.36.9
|
||||||
|
)
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect
|
||||||
|
github.com/google/uuid v1.6.0 // indirect
|
||||||
|
github.com/matoous/go-nanoid v1.5.1 // indirect
|
||||||
|
github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 // indirect
|
||||||
|
github.com/spf13/cast v1.9.2 // indirect
|
||||||
|
go.unistack.org/micro-proto/v4 v4.1.0 // indirect
|
||||||
|
golang.org/x/sys v0.35.0 // indirect
|
||||||
|
google.golang.org/genproto/googleapis/rpc v0.0.0-20250818200422-3122310a409c // indirect
|
||||||
|
google.golang.org/grpc v1.75.0 // indirect
|
||||||
|
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||||
|
)
|
||||||
|
62
go.sum
62
go.sum
@@ -1,17 +1,47 @@
|
|||||||
github.com/dgrijalva/jwt-go v3.2.0+incompatible/go.mod h1:E3ru+11k8xSBh+hMPgOLZmtrrCbhqsmaPHjLKYnJCaQ=
|
github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1VwoXQT9A3Wy9MM3WgvqSxFWenqJduM=
|
||||||
github.com/ef-ds/deque v1.0.4/go.mod h1:gXDnTC3yqvBcHbq2lcExjtAcVrOnJCbMcZXmuj8Z4tg=
|
github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
github.com/google/uuid v1.3.0 h1:t6JiXgmwXMjEs8VusXIJk2BXHsn+wx8BZdTaoZ5fu7I=
|
github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8=
|
||||||
github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
github.com/frankban/quicktest v1.14.6/go.mod h1:4ptaffx2x8+WTWXmUCuVU6aPUX1/Mz7zb5vbUoiM6w0=
|
||||||
github.com/imdario/mergo v0.3.12/go.mod h1:jmQim1M+e3UYxmgPu/WyfjB3N3VflVyUjjjwH0dnCYA=
|
github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek=
|
||||||
github.com/patrickmn/go-cache v2.1.0+incompatible/go.mod h1:3Qf8kWWT7OJRJbdiICTKqZju1ZixQ/KpMGzzAfe6+WQ=
|
github.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps=
|
||||||
github.com/silas/dag v0.0.0-20210121180416-41cf55125c34/go.mod h1:7RTUFBdIRC9nZ7/3RyRNH1bdqIShrDejd1YbLwgPS+I=
|
github.com/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8=
|
||||||
github.com/unistack-org/micro/v3 v3.5.2 h1:8b9Mk4FLWRLp8SduBh5Xs6g/3xJ+ZIBOnH82eHuLWnw=
|
github.com/google/go-cmp v0.7.0/go.mod h1:pXiqmnSA92OHEEa9HXL2W4E7lf9JzCmGVUdgjX3N/iU=
|
||||||
github.com/unistack-org/micro/v3 v3.5.2/go.mod h1:1ZkwpEqpiHiVhM2hiF9DamtpsF04oFybFhEQ4zEMcro=
|
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
|
||||||
golang.org/x/net v0.0.0-20210510120150-4163338589ed/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
|
github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
||||||
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE=
|
||||||
golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk=
|
||||||
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
|
||||||
golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
|
||||||
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
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/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.13.1 h1:KvO1DLK/DRN07sQ1LQKScxyZJuNnedQ5/wKSR38lUII=
|
||||||
|
github.com/rogpeppe/go-internal v1.13.1/go.mod h1:uMEvuHeurkdAXX61udpOXGD/AzZDWNMNyH2VO9fmH0o=
|
||||||
|
github.com/spf13/cast v1.9.2 h1:SsGfm7M8QOFtEzumm7UZrZdLLquNdzFYfIbEXntcFbE=
|
||||||
|
github.com/spf13/cast v1.9.2/go.mod h1:jNfB8QC9IA6ZuY2ZjDp0KtFO2LZZlg4S/7bzP6qqeHo=
|
||||||
|
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/v4 v4.1.0 h1:iydeSkt3ee7IPU0dHHKlGN97lw+YFQasBk9rdv0woYA=
|
||||||
|
go.unistack.org/micro-codec-json/v4 v4.1.0/go.mod h1:aUg86elSlURSynTAetDAAXj/VzFDwwcg92QNrRzcvrM=
|
||||||
|
go.unistack.org/micro-proto/v4 v4.1.0 h1:qPwL2n/oqh9RE3RTTDgt28XK3QzV597VugQPaw9lKUk=
|
||||||
|
go.unistack.org/micro-proto/v4 v4.1.0/go.mod h1:ArmK7o+uFvxSY3dbJhKBBX4Pm1rhWdLEFf3LxBrMtec=
|
||||||
|
go.unistack.org/micro/v4 v4.1.19 h1:LKpmSPYvX5B9AkFD7JqMU/U06v5yEWn2bsCG/YKZtZI=
|
||||||
|
go.unistack.org/micro/v4 v4.1.19/go.mod h1:xleO2M5Yxh4s6I+RUcLrEpUjobefh+71ctrdIfn7TUs=
|
||||||
|
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.35.0 h1:vz1N37gP5bs89s7He8XuIYXpyY0+QlsKmzipCbUtyxI=
|
||||||
|
golang.org/x/sys v0.35.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=
|
||||||
|
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-20250818200422-3122310a409c h1:qXWI/sQtv5UKboZ/zUk7h+mrf/lXORyI+n9DKDAusdg=
|
||||||
|
google.golang.org/genproto/googleapis/rpc v0.0.0-20250818200422-3122310a409c/go.mod h1:gw1tLEfykwDz2ET4a12jcXt4couGAm7IwsVaTy0Sflo=
|
||||||
|
google.golang.org/grpc v1.75.0 h1:+TW+dqTd2Biwe6KKfhE5JpiYIBWq865PhKGSXiivqt4=
|
||||||
|
google.golang.org/grpc v1.75.0/go.mod h1:JtPAzKiq4v1xcAB2hydNlWI2RnF85XXcV0mhKXr2ecQ=
|
||||||
|
google.golang.org/protobuf v1.36.9 h1:w2gp2mA27hUeUzj9Ex9FBjsBm40zfaDtEWow293U7Iw=
|
||||||
|
google.golang.org/protobuf v1.36.9/go.mod h1:fuxRtAxBytpl4zzqUh6/eyUujkJdNiuEkXntxiD/uRU=
|
||||||
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=
|
||||||
|
636
http.go
636
http.go
@@ -1,636 +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(ctx context.Context, addr string, req client.Request, ct string, cf codec.Codec, msg interface{}, opts client.CallOptions) (*http.Request, error) {
|
|
||||||
var tags []string
|
|
||||||
scheme := "http"
|
|
||||||
method := http.MethodPost
|
|
||||||
body := "*" // as like google api http annotation
|
|
||||||
host := addr
|
|
||||||
path := req.Endpoint()
|
|
||||||
|
|
||||||
u, err := url.Parse(addr)
|
|
||||||
if err == nil {
|
|
||||||
scheme = u.Scheme
|
|
||||||
path = u.Path
|
|
||||||
host = u.Host
|
|
||||||
} else {
|
|
||||||
u = &url.URL{Scheme: scheme, Path: path, Host: host}
|
|
||||||
}
|
|
||||||
|
|
||||||
if opts.Context != nil {
|
|
||||||
if m, ok := opts.Context.Value(methodKey{}).(string); ok {
|
|
||||||
method = m
|
|
||||||
}
|
|
||||||
if p, ok := opts.Context.Value(pathKey{}).(string); ok {
|
|
||||||
path += 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
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(tags) == 0 {
|
|
||||||
switch ct {
|
|
||||||
default:
|
|
||||||
tags = append(tags, "json", "protobuf")
|
|
||||||
case "text/xml":
|
|
||||||
tags = append(tags, "xml")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if path == "" {
|
|
||||||
path = req.Endpoint()
|
|
||||||
}
|
|
||||||
|
|
||||||
u, err = u.Parse(path)
|
|
||||||
if err != nil {
|
|
||||||
return nil, errors.BadRequest("go.micro.client", err.Error())
|
|
||||||
}
|
|
||||||
|
|
||||||
path, nmsg, err := newPathRequest(u.Path, method, body, msg, tags)
|
|
||||||
if err != nil {
|
|
||||||
return nil, errors.BadRequest("go.micro.client", err.Error())
|
|
||||||
}
|
|
||||||
|
|
||||||
u, err = url.Parse(fmt.Sprintf("%s://%s%s", scheme, host, 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())
|
|
||||||
}
|
|
||||||
|
|
||||||
var hreq *http.Request
|
|
||||||
if len(b) > 0 {
|
|
||||||
hreq, err = http.NewRequestWithContext(ctx, method, u.String(), ioutil.NopCloser(bytes.NewBuffer(b)))
|
|
||||||
hreq.ContentLength = int64(len(b))
|
|
||||||
} else {
|
|
||||||
hreq, err = http.NewRequestWithContext(ctx, method, u.String(), nil)
|
|
||||||
}
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
return nil, errors.BadRequest("go.micro.client", err.Error())
|
|
||||||
}
|
|
||||||
|
|
||||||
header := make(http.Header)
|
|
||||||
|
|
||||||
if opts.Context != nil {
|
|
||||||
if md, ok := opts.Context.Value(metadataKey{}).(metadata.Metadata); ok {
|
|
||||||
for k, v := range md {
|
|
||||||
header.Set(k, v)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if opts.AuthToken != "" {
|
|
||||||
hreq.Header.Set(metadata.HeaderAuthorization, opts.AuthToken)
|
|
||||||
}
|
|
||||||
|
|
||||||
if md, ok := metadata.FromOutgoingContext(ctx); ok {
|
|
||||||
for k, v := range md {
|
|
||||||
hreq.Header.Set(k, v)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// set timeout in nanoseconds
|
|
||||||
if opts.StreamTimeout > time.Duration(0) {
|
|
||||||
hreq.Header.Set(metadata.HeaderTimeout, fmt.Sprintf("%d", opts.StreamTimeout))
|
|
||||||
}
|
|
||||||
if opts.RequestTimeout > time.Duration(0) {
|
|
||||||
hreq.Header.Set(metadata.HeaderTimeout, fmt.Sprintf("%d", opts.RequestTimeout))
|
|
||||||
}
|
|
||||||
|
|
||||||
// set the content type for the request
|
|
||||||
hreq.Header.Set(metadata.HeaderContentType, ct)
|
|
||||||
|
|
||||||
return hreq, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *httpClient) call(ctx context.Context, addr string, req client.Request, rsp interface{}, opts client.CallOptions) error {
|
|
||||||
ct := req.ContentType()
|
|
||||||
if len(opts.ContentType) > 0 {
|
|
||||||
ct = opts.ContentType
|
|
||||||
}
|
|
||||||
|
|
||||||
cf, err := h.newCodec(ct)
|
|
||||||
if err != nil {
|
|
||||||
return errors.InternalServerError("go.micro.client", err.Error())
|
|
||||||
}
|
|
||||||
hreq, err := newRequest(ctx, addr, req, ct, cf, req.Body(), opts)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
// make the request
|
|
||||||
hrsp, err := h.httpcli.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", 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) {
|
|
||||||
ct := req.ContentType()
|
|
||||||
if len(opts.ContentType) > 0 {
|
|
||||||
ct = opts.ContentType
|
|
||||||
}
|
|
||||||
|
|
||||||
// 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,
|
|
||||||
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[metadata.HeaderContentType] = p.ContentType()
|
|
||||||
md[metadata.HeaderTopic] = 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
|
|
||||||
}
|
|
159
options.go
159
options.go
@@ -1,105 +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/v4/client"
|
||||||
"github.com/unistack-org/micro/v3/metadata"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
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)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
type metadataKey struct{}
|
// ------------------------------------------------ Cookie option ------------------------------------------------------
|
||||||
|
type cookieKey struct{}
|
||||||
|
|
||||||
func Metadata(md metadata.Metadata) client.CallOption {
|
func Cookie(cookies ...string) client.CallOption {
|
||||||
return client.SetCallOption(metadataKey{}, md)
|
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/v4/client"
|
||||||
"github.com/unistack-org/micro/v3/codec"
|
"go.unistack.org/micro/v4/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
|
||||||
}
|
}
|
||||||
|
145
stream.go
145
stream.go
@@ -1,145 +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
|
|
||||||
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.context, h.address, h.request, h.ct, h.cf, msg, h.opts)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
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
|
|
||||||
}
|
|
271
util.go
271
util.go
@@ -1,271 +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"
|
|
||||||
"github.com/unistack-org/micro/v3/logger"
|
|
||||||
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, nestif
|
|
||||||
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 {
|
|
||||||
if tnmsg.Field(i).CanSet() {
|
|
||||||
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 hrsp.StatusCode >= 400 && cerr != nil {
|
|
||||||
var buf []byte
|
|
||||||
if hrsp.Body != nil {
|
|
||||||
buf, err = io.ReadAll(hrsp.Body)
|
|
||||||
if err != nil && h.opts.Logger.V(logger.ErrorLevel) {
|
|
||||||
h.opts.Logger.Errorf(ctx, "failed to read body: %v", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// response like text/plain or something else, return original error
|
|
||||||
return errors.New("go.micro.client", string(buf), int32(hrsp.StatusCode))
|
|
||||||
} else 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