Compare commits
	
		
			13 Commits
		
	
	
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 581fecd8f1 | |||
| d5d1e26d7b | |||
| aabfecfd3d | |||
| ff7a88fcce | |||
| 397f5414b3 | |||
| 9dd6efddf9 | |||
| 1e587b348a | |||
| 8475922d42 | |||
| 69fe6994ae | |||
| 6360feb351 | |||
| 0845f4873b | |||
| 747e35148f | |||
| 21f06fee6c | 
							
								
								
									
										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 }} | ||||
| @@ -3,10 +3,10 @@ name: lint | ||||
| on: | ||||
|   pull_request: | ||||
|     types: [opened, reopened, synchronize] | ||||
|     branches: | ||||
|     - master | ||||
|     - v3 | ||||
|     - v4 | ||||
|     branches: [ master, v3, v4 ] | ||||
|     paths-ignore: | ||||
|       - '.github/**' | ||||
|       - '.gitea/**' | ||||
| 
 | ||||
| jobs: | ||||
|   lint: | ||||
| @@ -24,6 +24,6 @@ jobs: | ||||
|     - name: setup deps | ||||
|       run: go get -v ./... | ||||
|     - name: run lint | ||||
|       uses: https://github.com/golangci/golangci-lint-action@v6 | ||||
|       uses: golangci/golangci-lint-action@v6 | ||||
|       with: | ||||
|         version: 'latest' | ||||
| @@ -3,15 +3,12 @@ name: test | ||||
| on: | ||||
|   pull_request: | ||||
|     types: [opened, reopened, synchronize] | ||||
|     branches: | ||||
|     - master | ||||
|     - v3 | ||||
|     - v4 | ||||
|     branches: [ master, v3, v4 ] | ||||
|   push: | ||||
|     branches: | ||||
|     - master | ||||
|     - v3 | ||||
|     - v4 | ||||
|     branches: [ master, v3, v4 ] | ||||
|     paths-ignore: | ||||
|       - '.github/**' | ||||
|       - '.gitea/**' | ||||
| 
 | ||||
| jobs: | ||||
|   test: | ||||
| @@ -3,15 +3,12 @@ name: test | ||||
| on: | ||||
|   pull_request: | ||||
|     types: [opened, reopened, synchronize] | ||||
|     branches: | ||||
|     - master | ||||
|     - v3 | ||||
|     - v4 | ||||
|     branches: [ master, v3, v4 ] | ||||
|   push: | ||||
|     branches: | ||||
|     - master | ||||
|     - v3 | ||||
|     - v4 | ||||
|     branches: [ master, v3, v4 ] | ||||
|     paths-ignore: | ||||
|       - '.github/**' | ||||
|       - '.gitea/**' | ||||
| 
 | ||||
| jobs: | ||||
|   test: | ||||
| @@ -35,19 +32,19 @@ jobs: | ||||
|         go-version: 'stable' | ||||
|     - name: setup go work | ||||
|       env: | ||||
|         GOWORK: /workspace/${{ github.repository_owner }}/go.work | ||||
|         GOWORK: ${{ github.workspace }}/go.work | ||||
|       run: | | ||||
|         go work init | ||||
|         go work use . | ||||
|         go work use micro-tests | ||||
|     - name: setup deps | ||||
|       env: | ||||
|         GOWORK: /workspace/${{ github.repository_owner }}/go.work | ||||
|         GOWORK: ${{ github.workspace }}/go.work | ||||
|       run: go get -v ./... | ||||
|     - name: run tests | ||||
|       env: | ||||
|         INTEGRATION_TESTS: yes | ||||
|         GOWORK: /workspace/${{ github.repository_owner }}/go.work | ||||
|         GOWORK: ${{ github.workspace }}/go.work | ||||
|       run: | | ||||
|         cd micro-tests | ||||
|         go test -mod readonly -v ./... || true | ||||
| @@ -1,21 +1,5 @@ | ||||
| run: | ||||
|   concurrency: 8 | ||||
|   deadline: 5m | ||||
|   timeout: 5m | ||||
|   issues-exit-code: 1 | ||||
|   tests: true | ||||
|  | ||||
| linters: | ||||
|   enable: | ||||
|     - staticcheck | ||||
|     - unused | ||||
|     - gosimple | ||||
|     - govet | ||||
|     - prealloc | ||||
|     - unconvert | ||||
|     - nakedret | ||||
|  | ||||
| linters-settings: | ||||
|   govet: | ||||
|     check-all: true | ||||
|     enable: | ||||
|       - fieldalignment | ||||
							
								
								
									
										10
									
								
								errors.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										10
									
								
								errors.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,10 @@ | ||||
| package kgo | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"errors" | ||||
| ) | ||||
|  | ||||
| func isContextError(err error) bool { | ||||
| 	return errors.Is(err, context.Canceled) || errors.Is(err, context.DeadlineExceeded) | ||||
| } | ||||
							
								
								
									
										27
									
								
								go.mod
									
									
									
									
									
								
							
							
						
						
									
										27
									
								
								go.mod
									
									
									
									
									
								
							| @@ -1,27 +1,32 @@ | ||||
| module go.unistack.org/micro-broker-kgo/v3 | ||||
|  | ||||
| go 1.23.0 | ||||
| go 1.24.0 | ||||
|  | ||||
| toolchain go1.24.3 | ||||
|  | ||||
| require ( | ||||
| 	github.com/google/uuid v1.6.0 | ||||
| 	github.com/twmb/franz-go v1.18.1 | ||||
| 	github.com/twmb/franz-go/pkg/kadm v1.15.0 | ||||
| 	github.com/twmb/franz-go/pkg/kmsg v1.9.0 | ||||
| 	go.opentelemetry.io/otel v1.34.0 | ||||
| 	go.unistack.org/micro/v3 v3.11.41 | ||||
| 	github.com/stretchr/testify v1.11.1 | ||||
| 	github.com/twmb/franz-go v1.19.5 | ||||
| 	github.com/twmb/franz-go/pkg/kadm v1.16.1 | ||||
| 	github.com/twmb/franz-go/pkg/kmsg v1.11.2 | ||||
| 	go.opentelemetry.io/otel v1.38.0 | ||||
| 	go.unistack.org/micro/v3 v3.11.45 | ||||
| ) | ||||
|  | ||||
| require ( | ||||
| 	github.com/ash3in/uuidv8 v1.2.0 // indirect | ||||
| 	github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect | ||||
| 	github.com/klauspost/compress v1.18.0 // indirect | ||||
| 	github.com/kr/pretty v0.3.1 // indirect | ||||
| 	github.com/matoous/go-nanoid v1.5.1 // indirect | ||||
| 	github.com/pierrec/lz4/v4 v4.1.22 // indirect | ||||
| 	github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 // indirect | ||||
| 	go.unistack.org/micro-proto/v3 v3.4.1 // indirect | ||||
| 	golang.org/x/crypto v0.35.0 // indirect | ||||
| 	golang.org/x/sys v0.30.0 // indirect | ||||
| 	google.golang.org/genproto/googleapis/rpc v0.0.0-20250224174004-546df14abb99 // indirect | ||||
| 	google.golang.org/grpc v1.70.0 // indirect | ||||
| 	google.golang.org/protobuf v1.36.5 // indirect | ||||
| 	golang.org/x/crypto v0.43.0 // indirect | ||||
| 	golang.org/x/sys v0.37.0 // indirect | ||||
| 	google.golang.org/genproto/googleapis/rpc v0.0.0-20251007200510-49b9836ed3ff // indirect | ||||
| 	google.golang.org/grpc v1.76.0 // indirect | ||||
| 	google.golang.org/protobuf v1.36.10 // indirect | ||||
| 	gopkg.in/yaml.v3 v3.0.1 // indirect | ||||
| ) | ||||
|   | ||||
							
								
								
									
										76
									
								
								go.sum
									
									
									
									
									
								
							
							
						
						
									
										76
									
								
								go.sum
									
									
									
									
									
								
							| @@ -7,12 +7,10 @@ github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1 | ||||
| github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | ||||
| github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek= | ||||
| github.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps= | ||||
| github.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI= | ||||
| github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY= | ||||
| github.com/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8= | ||||
| github.com/google/go-cmp v0.7.0/go.mod h1:pXiqmnSA92OHEEa9HXL2W4E7lf9JzCmGVUdgjX3N/iU= | ||||
| github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0= | ||||
| github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= | ||||
| github.com/klauspost/compress v1.17.11 h1:In6xLpyWOi1+C7tXUUWv2ot1QvBjxevKAaI6IXrJmUc= | ||||
| github.com/klauspost/compress v1.17.11/go.mod h1:pMDklpSncoRMuLFrf1W9Ss9KT+0rH90U12bZKk7uwG0= | ||||
| github.com/klauspost/compress v1.18.0 h1:c/Cqfb0r+Yi+JtIEq73FWXVkRonBlf0CRNYc8Zttxdo= | ||||
| github.com/klauspost/compress v1.18.0/go.mod h1:2Pp+KzxcywXVXMr50+X0Q/Lsb43OQHYWRCY2AiWywWQ= | ||||
| github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE= | ||||
| @@ -29,52 +27,34 @@ github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2/go.mod h1:iKH | ||||
| github.com/rogpeppe/go-internal v1.9.0/go.mod h1:WtVeX8xhTBvf0smdhujwtBcq4Qrzq/fJaraNFVN+nFs= | ||||
| github.com/rogpeppe/go-internal v1.13.1 h1:KvO1DLK/DRN07sQ1LQKScxyZJuNnedQ5/wKSR38lUII= | ||||
| github.com/rogpeppe/go-internal v1.13.1/go.mod h1:uMEvuHeurkdAXX61udpOXGD/AzZDWNMNyH2VO9fmH0o= | ||||
| github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA= | ||||
| github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY= | ||||
| github.com/twmb/franz-go v1.18.0 h1:25FjMZfdozBywVX+5xrWC2W+W76i0xykKjTdEeD2ejw= | ||||
| github.com/twmb/franz-go v1.18.0/go.mod h1:zXCGy74M0p5FbXsLeASdyvfLFsBvTubVqctIaa5wQ+I= | ||||
| github.com/twmb/franz-go v1.18.1 h1:D75xxCDyvTqBSiImFx2lkPduE39jz1vaD7+FNc+vMkc= | ||||
| github.com/twmb/franz-go v1.18.1/go.mod h1:Uzo77TarcLTUZeLuGq+9lNpSkfZI+JErv7YJhlDjs9M= | ||||
| github.com/twmb/franz-go/pkg/kadm v1.14.0 h1:nAn1co1lXzJQocpzyIyOFOjUBf4WHWs5/fTprXy2IZs= | ||||
| github.com/twmb/franz-go/pkg/kadm v1.14.0/go.mod h1:XjOPz6ZaXXjrW2jVCfLuucP8H1w2TvD6y3PT2M+aAM4= | ||||
| github.com/twmb/franz-go/pkg/kadm v1.15.0 h1:Yo3NAPfcsx3Gg9/hdhq4vmwO77TqRRkvpUcGWzjworc= | ||||
| github.com/twmb/franz-go/pkg/kadm v1.15.0/go.mod h1:MUdcUtnf9ph4SFBLLA/XxE29rvLhWYLM9Ygb8dfSCvw= | ||||
| github.com/twmb/franz-go/pkg/kmsg v1.9.0 h1:JojYUph2TKAau6SBtErXpXGC7E3gg4vGZMv9xFU/B6M= | ||||
| github.com/twmb/franz-go/pkg/kmsg v1.9.0/go.mod h1:CMbfazviCyY6HM0SXuG5t9vOwYDHRCSrJJyBAe5paqg= | ||||
| go.opentelemetry.io/otel v1.33.0 h1:/FerN9bax5LoK51X/sI0SVYrjSE0/yUL7DpxW4K3FWw= | ||||
| go.opentelemetry.io/otel v1.33.0/go.mod h1:SUUkR6csvUQl+yjReHu5uM3EtVV7MBm5FHKRlNx4I8I= | ||||
| go.opentelemetry.io/otel v1.34.0 h1:zRLXxLCgL1WyKsPVrgbSdMN4c0FMkDAskSTQP+0hdUY= | ||||
| go.opentelemetry.io/otel v1.34.0/go.mod h1:OWFPOQ+h4G8xpyjgqo4SxJYdDQ/qmRH+wivy7zzx9oI= | ||||
| github.com/stretchr/testify v1.11.1 h1:7s2iGBzp5EwR7/aIZr8ao5+dra3wiQyKjjFuvgVKu7U= | ||||
| github.com/stretchr/testify v1.11.1/go.mod h1:wZwfW3scLgRK+23gO65QZefKpKQRnfz6sD981Nm4B6U= | ||||
| github.com/twmb/franz-go v1.19.5 h1:W7+o8D0RsQsedqib71OVlLeZ0zI6CbFra7yTYhZTs5Y= | ||||
| github.com/twmb/franz-go v1.19.5/go.mod h1:4kFJ5tmbbl7asgwAGVuyG1ZMx0NNpYk7EqflvWfPCpM= | ||||
| github.com/twmb/franz-go/pkg/kadm v1.16.1 h1:IEkrhTljgLHJ0/hT/InhXGjPdmWfFvxp7o/MR7vJ8cw= | ||||
| github.com/twmb/franz-go/pkg/kadm v1.16.1/go.mod h1:Ue/ye1cc9ipsQFg7udFbbGiFNzQMqiH73fGC2y0rwyc= | ||||
| github.com/twmb/franz-go/pkg/kmsg v1.11.2 h1:hIw75FpwcAjgeyfIGFqivAvwC5uNIOWRGvQgZhH4mhg= | ||||
| github.com/twmb/franz-go/pkg/kmsg v1.11.2/go.mod h1:CFfkkLysDNmukPYhGzuUcDtf46gQSqCZHMW1T4Z+wDE= | ||||
| go.opentelemetry.io/otel v1.38.0 h1:RkfdswUDRimDg0m2Az18RKOsnI8UDzppJAtj01/Ymk8= | ||||
| go.opentelemetry.io/otel v1.38.0/go.mod h1:zcmtmQ1+YmQM9wrNsTGV/q/uyusom3P8RxwExxkZhjM= | ||||
| go.unistack.org/micro-proto/v3 v3.4.1 h1:UTjLSRz2YZuaHk9iSlVqqsA50JQNAEK2ZFboGqtEa9Q= | ||||
| go.unistack.org/micro-proto/v3 v3.4.1/go.mod h1:okx/cnOhzuCX0ggl/vToatbCupi0O44diiiLLsZ93Zo= | ||||
| go.unistack.org/micro/v3 v3.11.37 h1:ZcpnXAYEMcAwmnVb5b7o8/PylGnILxXMHaUlRrPmRI0= | ||||
| go.unistack.org/micro/v3 v3.11.37/go.mod h1:POGU5hstnAT9LH70m8FalyQSNi2GfIew71K75JenIZk= | ||||
| go.unistack.org/micro/v3 v3.11.41 h1:dP4sBLIZpMo+MWGe5bbESewK8wBzYm4Yik/67x4dEtQ= | ||||
| go.unistack.org/micro/v3 v3.11.41/go.mod h1:POGU5hstnAT9LH70m8FalyQSNi2GfIew71K75JenIZk= | ||||
| golang.org/x/crypto v0.31.0 h1:ihbySMvVjLAeSH1IbfcRTkD/iNscyz8rGzjF/E5hV6U= | ||||
| golang.org/x/crypto v0.31.0/go.mod h1:kDsLvtWBEx7MV9tJOj9bnXsPbxwJQ6csT/x4KIN4Ssk= | ||||
| golang.org/x/crypto v0.35.0 h1:b15kiHdrGCHrP6LvwaQ3c03kgNhhiMgvlhxHQhmg2Xs= | ||||
| golang.org/x/crypto v0.35.0/go.mod h1:dy7dXNW32cAb/6/PRuTNsix8T+vJAqvuIy5Bli/x0YQ= | ||||
| golang.org/x/net v0.33.0 h1:74SYHlV8BIgHIFC/LrYkOGIwL19eTYXQ5wc6TBuO36I= | ||||
| golang.org/x/net v0.33.0/go.mod h1:HXLR5J+9DxmrqMwG9qjGCxZ+zKXxBru04zlTvWlWuN4= | ||||
| golang.org/x/sys v0.28.0 h1:Fksou7UEQUWlKvIdsqzJmUmCX3cZuD2+P3XyyzwMhlA= | ||||
| golang.org/x/sys v0.28.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= | ||||
| golang.org/x/sys v0.30.0 h1:QjkSwP/36a20jFYWkSue1YwXzLmsV5Gfq7Eiy72C1uc= | ||||
| golang.org/x/sys v0.30.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= | ||||
| golang.org/x/text v0.21.0 h1:zyQAAkrwaneQ066sspRyJaG9VNi/YJ1NfzcGB3hZ/qo= | ||||
| golang.org/x/text v0.21.0/go.mod h1:4IBbMaMmOPCJ8SecivzSH54+73PCFmPWxNTLm+vZkEQ= | ||||
| google.golang.org/genproto/googleapis/rpc v0.0.0-20241216192217-9240e9c98484 h1:Z7FRVJPSMaHQxD0uXU8WdgFh8PseLM8Q8NzhnpMrBhQ= | ||||
| google.golang.org/genproto/googleapis/rpc v0.0.0-20241216192217-9240e9c98484/go.mod h1:lcTa1sDdWEIHMWlITnIczmw5w60CF9ffkb8Z+DVmmjA= | ||||
| google.golang.org/genproto/googleapis/rpc v0.0.0-20250224174004-546df14abb99 h1:ZSlhAUqC4r8TPzqLXQ0m3upBNZeF+Y8jQ3c4CR3Ujms= | ||||
| google.golang.org/genproto/googleapis/rpc v0.0.0-20250224174004-546df14abb99/go.mod h1:LuRYeWDFV6WOn90g357N17oMCaxpgCnbi/44qJvDn2I= | ||||
| google.golang.org/grpc v1.69.2 h1:U3S9QEtbXC0bYNvRtcoklF3xGtLViumSYxWykJS+7AU= | ||||
| google.golang.org/grpc v1.69.2/go.mod h1:vyjdE6jLBI76dgpDojsFGNaHlxdjXN9ghpnd2o7JGZ4= | ||||
| google.golang.org/grpc v1.70.0 h1:pWFv03aZoHzlRKHWicjsZytKAiYCtNS0dHbXnIdq7jQ= | ||||
| google.golang.org/grpc v1.70.0/go.mod h1:ofIJqVKDXx/JiXrwr2IG4/zwdH9txy3IlF40RmcJSQw= | ||||
| google.golang.org/protobuf v1.36.1 h1:yBPeRvTftaleIgM3PZ/WBIZ7XM/eEYAaEyCwvyjq/gk= | ||||
| google.golang.org/protobuf v1.36.1/go.mod h1:9fA7Ob0pmnwhb644+1+CVWFRbNajQ6iRojtC/QF5bRE= | ||||
| google.golang.org/protobuf v1.36.5 h1:tPhr+woSbjfYvY6/GPufUoYizxw1cF/yFoxJ2fmpwlM= | ||||
| google.golang.org/protobuf v1.36.5/go.mod h1:9fA7Ob0pmnwhb644+1+CVWFRbNajQ6iRojtC/QF5bRE= | ||||
| go.unistack.org/micro/v3 v3.11.45 h1:fjTLZYWgsVf9FIMZBxOg8ios2/tmyimnjZrsrxEUeXU= | ||||
| go.unistack.org/micro/v3 v3.11.45/go.mod h1:fDQ8Mu9wubaFP0L8hNQlpzHiEnWN0wbOlawN9HYo0N4= | ||||
| golang.org/x/crypto v0.43.0 h1:dduJYIi3A3KOfdGOHX8AVZ/jGiyPa3IbBozJ5kNuE04= | ||||
| golang.org/x/crypto v0.43.0/go.mod h1:BFbav4mRNlXJL4wNeejLpWxB7wMbc79PdRGhWKncxR0= | ||||
| golang.org/x/net v0.45.0 h1:RLBg5JKixCy82FtLJpeNlVM0nrSqpCRYzVU1n8kj0tM= | ||||
| golang.org/x/net v0.45.0/go.mod h1:ECOoLqd5U3Lhyeyo/QDCEVQ4sNgYsqvCZ722XogGieY= | ||||
| golang.org/x/sys v0.37.0 h1:fdNQudmxPjkdUTPnLn5mdQv7Zwvbvpaxqs831goi9kQ= | ||||
| golang.org/x/sys v0.37.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks= | ||||
| golang.org/x/text v0.30.0 h1:yznKA/E9zq54KzlzBEAWn1NXSQ8DIp/NYMy88xJjl4k= | ||||
| golang.org/x/text v0.30.0/go.mod h1:yDdHFIX9t+tORqspjENWgzaCVXgk0yYnYuSZ8UzzBVM= | ||||
| google.golang.org/genproto/googleapis/rpc v0.0.0-20251007200510-49b9836ed3ff h1:A90eA31Wq6HOMIQlLfzFwzqGKBTuaVztYu/g8sn+8Zc= | ||||
| google.golang.org/genproto/googleapis/rpc v0.0.0-20251007200510-49b9836ed3ff/go.mod h1:7i2o+ce6H/6BluujYR+kqX3GKH+dChPTQU19wjRPiGk= | ||||
| google.golang.org/grpc v1.76.0 h1:UnVkv1+uMLYXoIz6o7chp59WfQUYA2ex/BXQ9rHZu7A= | ||||
| google.golang.org/grpc v1.76.0/go.mod h1:Ju12QI8M6iQJtbcsV+awF5a4hfJMLi4X0JLo94ULZ6c= | ||||
| google.golang.org/protobuf v1.36.10 h1:AYd7cD/uASjIL6Q9LiTjz8JLcrh/88q5UObnmY3aOOE= | ||||
| google.golang.org/protobuf v1.36.10/go.mod h1:HTf+CrKn2C3g5S8VImy6tdcUvCska2kB7j23XfzDpco= | ||||
| gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= | ||||
| gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk= | ||||
| gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q= | ||||
|   | ||||
| @@ -12,8 +12,8 @@ import ( | ||||
| 
 | ||||
| type hookEvent struct { | ||||
| 	log          logger.Logger | ||||
| 	fatalOnError bool | ||||
| 	connected    *atomic.Uint32 | ||||
| 	fatalOnError bool | ||||
| } | ||||
| 
 | ||||
| var ( | ||||
| @@ -26,44 +26,48 @@ var ( | ||||
| ) | ||||
| 
 | ||||
| func (m *hookEvent) OnGroupManageError(err error) { | ||||
| 	if err != nil { | ||||
| 		m.connected.Store(0) | ||||
| 		if m.fatalOnError { | ||||
| 			m.log.Fatal(context.TODO(), "kgo.OnGroupManageError", err) | ||||
| 		} | ||||
| 	switch { | ||||
| 	case err == nil || isContextError(err) || kgo.IsRetryableBrokerErr(err): | ||||
| 		return | ||||
| 	default: | ||||
| 		m.log.Error(context.TODO(), "kgo.OnGroupManageError", err) | ||||
| 	} | ||||
| } | ||||
| 
 | ||||
| func (m *hookEvent) OnBrokerConnect(_ kgo.BrokerMetadata, _ time.Duration, _ net.Conn, err error) { | ||||
| 	if err != nil { | ||||
| 		m.connected.Store(0) | ||||
| 		if m.fatalOnError { | ||||
| 			m.log.Fatal(context.TODO(), "kgo.OnBrokerConnect", err) | ||||
| 		} | ||||
| 	switch { | ||||
| 	case err == nil || isContextError(err) || kgo.IsRetryableBrokerErr(err): | ||||
| 		return | ||||
| 	default: | ||||
| 		m.log.Error(context.TODO(), "kgo.OnBrokerConnect", err) | ||||
| 	} | ||||
| } | ||||
| 
 | ||||
| func (m *hookEvent) OnBrokerDisconnect(_ kgo.BrokerMetadata, _ net.Conn) { | ||||
| 	m.connected.Store(0) | ||||
| } | ||||
| func (m *hookEvent) OnBrokerDisconnect(_ kgo.BrokerMetadata, _ net.Conn) {} | ||||
| 
 | ||||
| func (m *hookEvent) OnBrokerWrite(_ kgo.BrokerMetadata, _ int16, _ int, _ time.Duration, _ time.Duration, err error) { | ||||
| 	if err != nil { | ||||
| 		m.connected.Store(0) | ||||
| 		if m.fatalOnError { | ||||
| 			m.log.Fatal(context.TODO(), "kgo.OnBrokerWrite", err) | ||||
| 		} | ||||
| 	switch { | ||||
| 	case err == nil || isContextError(err) || kgo.IsRetryableBrokerErr(err): | ||||
| 		return | ||||
| 	default: | ||||
| 		m.log.Error(context.TODO(), "kgo.OnBrokerWrite", err) | ||||
| 	} | ||||
| } | ||||
| 
 | ||||
| func (m *hookEvent) OnBrokerRead(_ kgo.BrokerMetadata, _ int16, _ int, _ time.Duration, _ time.Duration, err error) { | ||||
| 	if err != nil { | ||||
| 		m.connected.Store(0) | ||||
| 	switch { | ||||
| 	case err == nil || isContextError(err) || kgo.IsRetryableBrokerErr(err): | ||||
| 		return | ||||
| 	default: | ||||
| 		m.log.Error(context.TODO(), "kgo.OnBrokerRead", err) | ||||
| 	} | ||||
| } | ||||
| 
 | ||||
| func (m *hookEvent) OnProduceRecordUnbuffered(_ *kgo.Record, err error) { | ||||
| 	if err != nil { | ||||
| 		m.connected.Store(0) | ||||
| 	switch { | ||||
| 	case err == nil || isContextError(err) || kgo.IsRetryableBrokerErr(err): | ||||
| 		return | ||||
| 	default: | ||||
| 		m.log.Error(context.TODO(), "kgo.OnProduceRecordUnbuffered", err) | ||||
| 	} | ||||
| } | ||||
							
								
								
									
										362
									
								
								hook_event_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										362
									
								
								hook_event_test.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,362 @@ | ||||
| package kgo | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"errors" | ||||
| 	"io" | ||||
| 	"net" | ||||
| 	"os" | ||||
| 	"testing" | ||||
|  | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 	"github.com/twmb/franz-go/pkg/kgo" | ||||
| 	"go.unistack.org/micro/v3/logger" | ||||
| ) | ||||
|  | ||||
| func TestHookEvent_OnGroupManageError(t *testing.T) { | ||||
| 	tests := []struct { | ||||
| 		name                  string | ||||
| 		inputErr              error | ||||
| 		expectedErrorIsCalled bool | ||||
| 		expectedErrorMsg      string | ||||
| 	}{ | ||||
| 		{ | ||||
| 			name:                  "error is nil", | ||||
| 			inputErr:              nil, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "context canceled", | ||||
| 			inputErr:              context.Canceled, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "context deadline exceeded", | ||||
| 			inputErr:              context.DeadlineExceeded, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "retryable error: deadline exceeded (os package)", | ||||
| 			inputErr:              os.ErrDeadlineExceeded, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "retryable error: EOF (io package)", | ||||
| 			inputErr:              io.EOF, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "retryable error: closed network connection (net package)", | ||||
| 			inputErr:              net.ErrClosed, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "some error", | ||||
| 			inputErr:              errors.New("some error"), | ||||
| 			expectedErrorIsCalled: true, | ||||
| 			expectedErrorMsg:      "kgo.OnGroupManageError", | ||||
| 		}, | ||||
| 	} | ||||
|  | ||||
| 	for _, tt := range tests { | ||||
| 		t.Run(tt.name, func(t *testing.T) { | ||||
| 			log := &mockLogger{} | ||||
| 			he := &hookEvent{log: log} | ||||
| 			he.OnGroupManageError(tt.inputErr) | ||||
| 			require.Equal(t, tt.expectedErrorIsCalled, log.errorIsCalled) | ||||
| 			require.Equal(t, tt.expectedErrorMsg, log.errorMsg) | ||||
| 		}) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestHookEvent_OnBrokerConnect(t *testing.T) { | ||||
| 	tests := []struct { | ||||
| 		name                  string | ||||
| 		inputErr              error | ||||
| 		expectedErrorIsCalled bool | ||||
| 		expectedErrorMsg      string | ||||
| 	}{ | ||||
| 		{ | ||||
| 			name:                  "error is nil", | ||||
| 			inputErr:              nil, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "context canceled", | ||||
| 			inputErr:              context.Canceled, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "context deadline exceeded", | ||||
| 			inputErr:              context.DeadlineExceeded, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "retryable error: deadline exceeded (os package)", | ||||
| 			inputErr:              os.ErrDeadlineExceeded, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "retryable error: EOF (io package)", | ||||
| 			inputErr:              io.EOF, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "retryable error: closed network connection (net package)", | ||||
| 			inputErr:              net.ErrClosed, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "some error", | ||||
| 			inputErr:              errors.New("some error"), | ||||
| 			expectedErrorIsCalled: true, | ||||
| 			expectedErrorMsg:      "kgo.OnBrokerConnect", | ||||
| 		}, | ||||
| 	} | ||||
|  | ||||
| 	for _, tt := range tests { | ||||
| 		t.Run(tt.name, func(t *testing.T) { | ||||
| 			log := &mockLogger{} | ||||
| 			he := &hookEvent{log: log} | ||||
| 			he.OnBrokerConnect(kgo.BrokerMetadata{}, 0, nil, tt.inputErr) | ||||
| 			require.Equal(t, tt.expectedErrorIsCalled, log.errorIsCalled) | ||||
| 			require.Equal(t, tt.expectedErrorMsg, log.errorMsg) | ||||
| 		}) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestHookEvent_OnBrokerWrite(t *testing.T) { | ||||
| 	tests := []struct { | ||||
| 		name                  string | ||||
| 		inputErr              error | ||||
| 		expectedErrorIsCalled bool | ||||
| 		expectedErrorMsg      string | ||||
| 	}{ | ||||
| 		{ | ||||
| 			name:                  "error is nil", | ||||
| 			inputErr:              nil, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "context canceled", | ||||
| 			inputErr:              context.Canceled, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "context deadline exceeded", | ||||
| 			inputErr:              context.DeadlineExceeded, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "retryable error: deadline exceeded (os package)", | ||||
| 			inputErr:              os.ErrDeadlineExceeded, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "retryable error: EOF (io package)", | ||||
| 			inputErr:              io.EOF, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "retryable error: closed network connection (net package)", | ||||
| 			inputErr:              net.ErrClosed, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "some error", | ||||
| 			inputErr:              errors.New("some error"), | ||||
| 			expectedErrorIsCalled: true, | ||||
| 			expectedErrorMsg:      "kgo.OnBrokerWrite", | ||||
| 		}, | ||||
| 	} | ||||
|  | ||||
| 	for _, tt := range tests { | ||||
| 		t.Run(tt.name, func(t *testing.T) { | ||||
| 			log := &mockLogger{} | ||||
| 			he := &hookEvent{log: log} | ||||
| 			he.OnBrokerWrite(kgo.BrokerMetadata{}, 0, 0, 0, 0, tt.inputErr) | ||||
| 			require.Equal(t, tt.expectedErrorIsCalled, log.errorIsCalled) | ||||
| 			require.Equal(t, tt.expectedErrorMsg, log.errorMsg) | ||||
| 		}) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestHookEvent_OnBrokerRead(t *testing.T) { | ||||
| 	tests := []struct { | ||||
| 		name                  string | ||||
| 		inputErr              error | ||||
| 		expectedErrorIsCalled bool | ||||
| 		expectedErrorMsg      string | ||||
| 	}{ | ||||
| 		{ | ||||
| 			name:                  "error is nil", | ||||
| 			inputErr:              nil, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "context canceled", | ||||
| 			inputErr:              context.Canceled, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "context deadline exceeded", | ||||
| 			inputErr:              context.DeadlineExceeded, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "retryable error: deadline exceeded (os package)", | ||||
| 			inputErr:              os.ErrDeadlineExceeded, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "retryable error: EOF (io package)", | ||||
| 			inputErr:              io.EOF, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "retryable error: closed network connection (net package)", | ||||
| 			inputErr:              net.ErrClosed, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "some error", | ||||
| 			inputErr:              errors.New("some error"), | ||||
| 			expectedErrorIsCalled: true, | ||||
| 			expectedErrorMsg:      "kgo.OnBrokerRead", | ||||
| 		}, | ||||
| 	} | ||||
|  | ||||
| 	for _, tt := range tests { | ||||
| 		t.Run(tt.name, func(t *testing.T) { | ||||
| 			log := &mockLogger{} | ||||
| 			he := &hookEvent{log: log} | ||||
| 			he.OnBrokerRead(kgo.BrokerMetadata{}, 0, 0, 0, 0, tt.inputErr) | ||||
| 			require.Equal(t, tt.expectedErrorIsCalled, log.errorIsCalled) | ||||
| 			require.Equal(t, tt.expectedErrorMsg, log.errorMsg) | ||||
| 		}) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestHookEvent_OnProduceRecordUnbuffered(t *testing.T) { | ||||
| 	tests := []struct { | ||||
| 		name                  string | ||||
| 		inputErr              error | ||||
| 		expectedErrorIsCalled bool | ||||
| 		expectedErrorMsg      string | ||||
| 	}{ | ||||
| 		{ | ||||
| 			name:                  "error is nil", | ||||
| 			inputErr:              nil, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "context canceled", | ||||
| 			inputErr:              context.Canceled, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "context deadline exceeded", | ||||
| 			inputErr:              context.DeadlineExceeded, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "retryable error: deadline exceeded (os package)", | ||||
| 			inputErr:              os.ErrDeadlineExceeded, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "retryable error: EOF (io package)", | ||||
| 			inputErr:              io.EOF, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "retryable error: closed network connection (net package)", | ||||
| 			inputErr:              net.ErrClosed, | ||||
| 			expectedErrorIsCalled: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:                  "some error", | ||||
| 			inputErr:              errors.New("some error"), | ||||
| 			expectedErrorIsCalled: true, | ||||
| 			expectedErrorMsg:      "kgo.OnProduceRecordUnbuffered", | ||||
| 		}, | ||||
| 	} | ||||
|  | ||||
| 	for _, tt := range tests { | ||||
| 		t.Run(tt.name, func(t *testing.T) { | ||||
| 			log := &mockLogger{} | ||||
| 			he := &hookEvent{log: log} | ||||
| 			he.OnProduceRecordUnbuffered(&kgo.Record{}, tt.inputErr) | ||||
| 			require.Equal(t, tt.expectedErrorIsCalled, log.errorIsCalled) | ||||
| 			require.Equal(t, tt.expectedErrorMsg, log.errorMsg) | ||||
| 		}) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Mocks | ||||
|  | ||||
| type mockLogger struct { | ||||
| 	errorIsCalled bool | ||||
| 	errorMsg      string | ||||
| } | ||||
|  | ||||
| func (m *mockLogger) Init(...logger.Option) error { | ||||
| 	panic("implement me") | ||||
| } | ||||
|  | ||||
| func (m *mockLogger) Clone(...logger.Option) logger.Logger { | ||||
| 	panic("implement me") | ||||
| } | ||||
|  | ||||
| func (m *mockLogger) V(logger.Level) bool { | ||||
| 	panic("implement me") | ||||
| } | ||||
|  | ||||
| func (m *mockLogger) Level(logger.Level) { | ||||
| 	panic("implement me") | ||||
| } | ||||
|  | ||||
| func (m *mockLogger) Options() logger.Options { | ||||
| 	panic("implement me") | ||||
| } | ||||
|  | ||||
| func (m *mockLogger) Fields(...interface{}) logger.Logger { | ||||
| 	panic("implement me") | ||||
| } | ||||
|  | ||||
| func (m *mockLogger) Info(context.Context, string, ...interface{}) { | ||||
| 	panic("implement me") | ||||
| } | ||||
|  | ||||
| func (m *mockLogger) Trace(context.Context, string, ...interface{}) { | ||||
| 	panic("implement me") | ||||
| } | ||||
|  | ||||
| func (m *mockLogger) Debug(context.Context, string, ...interface{}) { | ||||
| 	panic("implement me") | ||||
| } | ||||
|  | ||||
| func (m *mockLogger) Warn(context.Context, string, ...interface{}) { | ||||
| 	panic("implement me") | ||||
| } | ||||
|  | ||||
| func (m *mockLogger) Error(ctx context.Context, msg string, args ...interface{}) { | ||||
| 	m.errorIsCalled = true | ||||
| 	m.errorMsg = msg | ||||
| } | ||||
|  | ||||
| func (m *mockLogger) Fatal(context.Context, string, ...interface{}) { | ||||
| 	panic("implement me") | ||||
| } | ||||
|  | ||||
| func (m *mockLogger) Log(context.Context, logger.Level, string, ...interface{}) { | ||||
| 	panic("implement me") | ||||
| } | ||||
|  | ||||
| func (m *mockLogger) Name() string { | ||||
| 	panic("implement me") | ||||
| } | ||||
|  | ||||
| func (m *mockLogger) String() string { | ||||
| 	panic("implement me") | ||||
| } | ||||
							
								
								
									
										124
									
								
								kgo.go
									
									
									
									
									
								
							
							
						
						
									
										124
									
								
								kgo.go
									
									
									
									
									
								
							| @@ -6,12 +6,14 @@ import ( | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"math/rand/v2" | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
| 	"sync" | ||||
| 	"sync/atomic" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/google/uuid" | ||||
| 	"github.com/twmb/franz-go/pkg/kadm" | ||||
| 	"github.com/twmb/franz-go/pkg/kgo" | ||||
| 	"github.com/twmb/franz-go/pkg/kmsg" | ||||
| 	"go.unistack.org/micro/v3/broker" | ||||
| @@ -19,6 +21,7 @@ import ( | ||||
| 	"go.unistack.org/micro/v3/metadata" | ||||
| 	"go.unistack.org/micro/v3/semconv" | ||||
| 	"go.unistack.org/micro/v3/tracer" | ||||
| 	mjitter "go.unistack.org/micro/v3/util/jitter" | ||||
| 	mrand "go.unistack.org/micro/v3/util/rand" | ||||
| ) | ||||
|  | ||||
| @@ -59,12 +62,14 @@ type Broker struct { | ||||
| 	c         *kgo.Client | ||||
| 	connected *atomic.Uint32 | ||||
|  | ||||
| 	done chan struct{} | ||||
|  | ||||
| 	kopts []kgo.Opt | ||||
| 	subs  []*Subscriber | ||||
|  | ||||
| 	opts broker.Options | ||||
| 	mu   sync.RWMutex | ||||
|  | ||||
| 	sync.RWMutex | ||||
| 	init bool | ||||
| } | ||||
|  | ||||
| @@ -143,6 +148,25 @@ func (k *Broker) connect(ctx context.Context, opts ...kgo.Opt) (*kgo.Client, *ho | ||||
| 			return nil, nil, err | ||||
| 		} | ||||
| 		k.connected.Store(1) | ||||
|  | ||||
| 		if fatalOnError { | ||||
| 			go func() { | ||||
| 				c := 3 | ||||
| 				n := 0 | ||||
| 				tc := mjitter.NewTicker(500*time.Millisecond, 1*time.Second) | ||||
| 				defer tc.Stop() | ||||
| 				for range tc.C { | ||||
| 					if k.connected.Load() == 0 { | ||||
| 						if n > c { | ||||
| 							k.opts.Logger.Fatal(context.Background(), "broker fatal error") | ||||
| 						} | ||||
| 						n++ | ||||
| 					} else { | ||||
| 						n = 0 | ||||
| 					} | ||||
| 				} | ||||
| 			}() | ||||
| 		} | ||||
| 		return c, htracer, nil | ||||
| 	} | ||||
| } | ||||
| @@ -162,10 +186,72 @@ func (k *Broker) Connect(ctx context.Context) error { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	k.Lock() | ||||
| 	k.mu.Lock() | ||||
| 	k.c = c | ||||
| 	k.connected.Store(1) | ||||
| 	k.Unlock() | ||||
| 	k.mu.Unlock() | ||||
|  | ||||
| 	exposeLag := false | ||||
| 	if k.opts.Context != nil { | ||||
| 		if v, ok := k.opts.Context.Value(exposeLagKey{}).(bool); ok && v { | ||||
| 			exposeLag = v | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	if exposeLag { | ||||
| 		var mu sync.Mutex | ||||
| 		var lastUpdate time.Time | ||||
| 		type pl struct { | ||||
| 			p string | ||||
| 			l float64 | ||||
| 		} | ||||
|  | ||||
| 		lag := make(map[string]map[string]pl) // topic => group => partition => lag | ||||
| 		ac := kadm.NewClient(k.c) | ||||
|  | ||||
| 		updateStats := func() { | ||||
| 			mu.Lock() | ||||
| 			if time.Since(lastUpdate) < DefaultStatsInterval { | ||||
| 				return | ||||
| 			} | ||||
| 			mu.Unlock() | ||||
|  | ||||
| 			k.mu.Lock() | ||||
| 			groups := make([]string, 0, len(k.subs)) | ||||
| 			for _, g := range k.subs { | ||||
| 				groups = append(groups, g.opts.Group) | ||||
| 			} | ||||
| 			k.mu.Unlock() | ||||
|  | ||||
| 			dgls, err := ac.Lag(ctx, groups...) | ||||
| 			if err != nil || !dgls.Ok() { | ||||
| 				k.opts.Logger.Error(k.opts.Context, "kgo describe group lag error", err) | ||||
| 				return | ||||
| 			} | ||||
|  | ||||
| 			for gn, dgl := range dgls { | ||||
| 				for tn, lmap := range dgl.Lag { | ||||
| 					if _, ok := lag[tn]; !ok { | ||||
| 						lag[tn] = make(map[string]pl) | ||||
| 					} | ||||
| 					for p, l := range lmap { | ||||
| 						lag[tn][gn] = pl{p: strconv.Itoa(int(p)), l: float64(l.Lag)} | ||||
| 					} | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		for tn, dg := range lag { | ||||
| 			for gn, gl := range dg { | ||||
| 				k.opts.Meter.Gauge(semconv.BrokerGroupLag, | ||||
| 					func() float64 { updateStats(); return gl.l }, | ||||
| 					"topic", tn, | ||||
| 					"group", gn, | ||||
| 					"partition", gl.p) | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 	} | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
| @@ -183,8 +269,8 @@ func (k *Broker) Disconnect(ctx context.Context) error { | ||||
| 	ctx, span = k.opts.Tracer.Start(ctx, "Disconnect") | ||||
| 	defer span.Finish() | ||||
|  | ||||
| 	k.Lock() | ||||
| 	defer k.Unlock() | ||||
| 	k.mu.Lock() | ||||
| 	defer k.mu.Unlock() | ||||
| 	select { | ||||
| 	case <-nctx.Done(): | ||||
| 		return nctx.Err() | ||||
| @@ -204,12 +290,13 @@ func (k *Broker) Disconnect(ctx context.Context) error { | ||||
| 	} | ||||
|  | ||||
| 	k.connected.Store(0) | ||||
| 	close(k.done) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (k *Broker) Init(opts ...broker.Option) error { | ||||
| 	k.Lock() | ||||
| 	defer k.Unlock() | ||||
| 	k.mu.Lock() | ||||
| 	defer k.mu.Unlock() | ||||
|  | ||||
| 	if len(opts) == 0 && k.init { | ||||
| 		return nil | ||||
| @@ -257,18 +344,6 @@ func (k *Broker) Publish(ctx context.Context, topic string, msg *broker.Message, | ||||
| } | ||||
|  | ||||
| func (k *Broker) publish(ctx context.Context, msgs []*broker.Message, opts ...broker.PublishOption) error { | ||||
| 	k.Lock() | ||||
| 	if k.connected.Load() == 0 { | ||||
| 		c, _, err := k.connect(ctx, k.kopts...) | ||||
| 		if err != nil { | ||||
| 			k.Unlock() | ||||
| 			return err | ||||
| 		} | ||||
| 		k.c = c | ||||
| 		k.connected.Store(1) | ||||
| 	} | ||||
| 	k.Unlock() | ||||
|  | ||||
| 	options := broker.NewPublishOptions(opts...) | ||||
| 	records := make([]*kgo.Record, 0, len(msgs)) | ||||
| 	var errs []string | ||||
| @@ -391,12 +466,6 @@ func (k *Broker) Subscribe(ctx context.Context, topic string, handler broker.Han | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	if options.Context != nil { | ||||
| 		if v, ok := options.Context.Value(fatalOnErrorKey{}).(bool); ok && v { | ||||
| 			fatalOnError = v | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	sub := &Subscriber{ | ||||
| 		topic:        topic, | ||||
| 		opts:         options, | ||||
| @@ -450,9 +519,9 @@ func (k *Broker) Subscribe(ctx context.Context, topic string, handler broker.Han | ||||
|  | ||||
| 	go sub.poll(ctx) | ||||
|  | ||||
| 	k.Lock() | ||||
| 	k.mu.Lock() | ||||
| 	k.subs = append(k.subs, sub) | ||||
| 	k.Unlock() | ||||
| 	k.mu.Unlock() | ||||
| 	return sub, nil | ||||
| } | ||||
|  | ||||
| @@ -492,5 +561,6 @@ func NewBroker(opts ...broker.Option) *Broker { | ||||
| 		connected: &atomic.Uint32{}, | ||||
| 		opts:      options, | ||||
| 		kopts:     kopts, | ||||
| 		done:      make(chan struct{}), | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -117,6 +117,10 @@ func TestFail(t *testing.T) { | ||||
| } | ||||
|  | ||||
| func TestConnect(t *testing.T) { | ||||
| 	if tr := os.Getenv("INTEGRATION_TESTS"); len(tr) > 0 { | ||||
| 		t.Skip() | ||||
| 	} | ||||
|  | ||||
| 	var addrs []string | ||||
| 	ctx := context.TODO() | ||||
| 	b := kgo.NewBroker( | ||||
|   | ||||
| @@ -15,7 +15,7 @@ var ( | ||||
| 	DefaultCommitInterval = 5 * time.Second | ||||
|  | ||||
| 	// DefaultStatsInterval specifies how fast check consumer lag | ||||
| 	DefaultStatsInterval = 5 * time.Second | ||||
| 	DefaultStatsInterval = 30 * time.Second | ||||
|  | ||||
| 	// DefaultSubscribeMaxInflight specifies how much messages keep inflight | ||||
| 	DefaultSubscribeMaxInflight = 100 | ||||
| @@ -120,3 +120,10 @@ func PublishPromise(fn func(*kgo.Record, error)) broker.PublishOption { | ||||
| func ClientPublishPromise(fn func(*kgo.Record, error)) client.PublishOption { | ||||
| 	return client.SetPublishOption(publishPromiseKey{}, fn) | ||||
| } | ||||
|  | ||||
| type exposeLagKey struct{} | ||||
|  | ||||
| // ExposeLag enabled subscriber lag via [meter.Meter] | ||||
| func ExposeLag(b bool) broker.Option { | ||||
| 	return broker.SetOption(exposeLagKey{}, b) | ||||
| } | ||||
|   | ||||
							
								
								
									
										140
									
								
								subscriber.go
									
									
									
									
									
								
							
							
						
						
									
										140
									
								
								subscriber.go
									
									
									
									
									
								
							| @@ -3,12 +3,10 @@ package kgo | ||||
| import ( | ||||
| 	"context" | ||||
| 	"fmt" | ||||
| 	"strconv" | ||||
| 	"sync" | ||||
| 	"sync/atomic" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/twmb/franz-go/pkg/kadm" | ||||
| 	"github.com/twmb/franz-go/pkg/kgo" | ||||
| 	"github.com/twmb/franz-go/pkg/kmsg" | ||||
| 	"go.unistack.org/micro/v3/broker" | ||||
| @@ -24,34 +22,44 @@ type tp struct { | ||||
| } | ||||
|  | ||||
| type consumer struct { | ||||
| 	topic     string | ||||
| 	topic string | ||||
|  | ||||
| 	c         *kgo.Client | ||||
| 	htracer   *hookTracer | ||||
| 	quit      chan struct{} | ||||
| 	done      chan struct{} | ||||
| 	recs      chan kgo.FetchTopicPartition | ||||
| 	kopts     broker.Options | ||||
| 	partition int32 | ||||
| 	opts      broker.SubscribeOptions | ||||
| 	handler   broker.Handler | ||||
| 	connected *atomic.Uint32 | ||||
|  | ||||
| 	quit chan struct{} | ||||
| 	done chan struct{} | ||||
| 	recs chan kgo.FetchTopicPartition | ||||
|  | ||||
| 	handler broker.Handler | ||||
|  | ||||
| 	kopts broker.Options | ||||
| 	opts  broker.SubscribeOptions | ||||
|  | ||||
| 	partition int32 | ||||
| } | ||||
|  | ||||
| type Subscriber struct { | ||||
| 	topic string | ||||
|  | ||||
| 	consumers map[tp]*consumer | ||||
|  | ||||
| 	c         *kgo.Client | ||||
| 	htracer   *hookTracer | ||||
| 	topic     string | ||||
| 	connected *atomic.Uint32 | ||||
|  | ||||
| 	handler broker.Handler | ||||
| 	done    chan struct{} | ||||
| 	kopts   broker.Options | ||||
| 	opts    broker.SubscribeOptions | ||||
|  | ||||
| 	connected *atomic.Uint32 | ||||
| 	sync.RWMutex | ||||
| 	done chan struct{} | ||||
|  | ||||
| 	kopts broker.Options | ||||
| 	opts  broker.SubscribeOptions | ||||
|  | ||||
| 	closed       bool | ||||
| 	fatalOnError bool | ||||
|  | ||||
| 	sync.RWMutex | ||||
| } | ||||
|  | ||||
| func (s *Subscriber) Client() *kgo.Client { | ||||
| @@ -87,46 +95,13 @@ func (s *Subscriber) Unsubscribe(ctx context.Context) error { | ||||
|  | ||||
| func (s *Subscriber) poll(ctx context.Context) { | ||||
| 	maxInflight := DefaultSubscribeMaxInflight | ||||
|  | ||||
| 	if s.opts.Context != nil { | ||||
| 		if n, ok := s.opts.Context.Value(subscribeMaxInflightKey{}).(int); n > 0 && ok { | ||||
| 			maxInflight = n | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	go func() { | ||||
| 		ac := kadm.NewClient(s.c) | ||||
| 		ticker := time.NewTicker(DefaultStatsInterval) | ||||
|  | ||||
| 		for { | ||||
| 			select { | ||||
| 			case <-ctx.Done(): | ||||
| 				ticker.Stop() | ||||
| 				return | ||||
| 			case <-ticker.C: | ||||
| 				dgls, err := ac.Lag(ctx, s.opts.Group) | ||||
| 				if err != nil || !dgls.Ok() { | ||||
| 					continue | ||||
| 				} | ||||
|  | ||||
| 				dgl, ok := dgls[s.opts.Group] | ||||
| 				if !ok { | ||||
| 					continue | ||||
| 				} | ||||
| 				lmap, ok := dgl.Lag[s.topic] | ||||
| 				if !ok { | ||||
| 					continue | ||||
| 				} | ||||
|  | ||||
| 				s.Lock() | ||||
| 				for p, l := range lmap { | ||||
| 					s.kopts.Meter.Counter(semconv.BrokerGroupLag, "topic", s.topic, "group", s.opts.Group, "partition", strconv.Itoa(int(p)), "lag", strconv.Itoa(int(l.Lag))) | ||||
| 				} | ||||
| 				s.Unlock() | ||||
|  | ||||
| 			} | ||||
| 		} | ||||
| 	}() | ||||
|  | ||||
| 	for { | ||||
| 		select { | ||||
| 		case <-ctx.Done(): | ||||
| @@ -141,7 +116,11 @@ func (s *Subscriber) poll(ctx context.Context) { | ||||
| 				return | ||||
| 			} | ||||
| 			fetches.EachError(func(t string, p int32, err error) { | ||||
| 				s.kopts.Logger.Fatal(ctx, fmt.Sprintf("[kgo] fetch topic %s partition %d error", t, p), err) | ||||
| 				if kgo.IsRetryableBrokerErr(err) { | ||||
| 					s.kopts.Logger.Error(ctx, fmt.Sprintf("[kgo] fetch topic %s partition %d error", t, p), err) | ||||
| 				} else { | ||||
| 					s.kopts.Logger.Fatal(ctx, fmt.Sprintf("[kgo] fetch topic %s partition %d error", t, p), err) | ||||
| 				} | ||||
| 			}) | ||||
|  | ||||
| 			fetches.EachPartition(func(p kgo.FetchTopicPartition) { | ||||
| @@ -160,7 +139,10 @@ func (s *Subscriber) killConsumers(ctx context.Context, lost map[string][]int32) | ||||
| 	for topic, partitions := range lost { | ||||
| 		for _, partition := range partitions { | ||||
| 			tps := tp{topic, partition} | ||||
| 			pc := s.consumers[tps] | ||||
| 			pc, ok := s.consumers[tps] | ||||
| 			if !ok { | ||||
| 				continue | ||||
| 			} | ||||
| 			delete(s.consumers, tps) | ||||
| 			close(pc.quit) | ||||
| 			if s.kopts.Logger.V(logger.DebugLevel) { | ||||
| @@ -174,7 +156,7 @@ func (s *Subscriber) killConsumers(ctx context.Context, lost map[string][]int32) | ||||
|  | ||||
| func (s *Subscriber) autocommit(_ *kgo.Client, _ *kmsg.OffsetCommitRequest, _ *kmsg.OffsetCommitResponse, err error) { | ||||
| 	if err != nil { | ||||
| 		s.connected.Store(0) | ||||
| 		//		s.connected.Store(0) | ||||
| 		if s.fatalOnError { | ||||
| 			s.kopts.Logger.Fatal(context.TODO(), "kgo.AutoCommitCallback error", err) | ||||
| 		} | ||||
| @@ -261,11 +243,17 @@ func (pc *consumer) consume() { | ||||
| 				} else if pc.opts.BodyOnly { | ||||
| 					p.msg.Body = record.Value | ||||
| 				} else { | ||||
| 					sp.AddEvent("codec unmarshal start") | ||||
| 					if sp != nil { | ||||
| 						sp.AddEvent("codec unmarshal start") | ||||
| 					} | ||||
| 					err := pc.kopts.Codec.Unmarshal(record.Value, p.msg) | ||||
| 					sp.AddEvent("codec unmarshal stop") | ||||
| 					if sp != nil { | ||||
| 						sp.AddEvent("codec unmarshal stop") | ||||
| 					} | ||||
| 					if err != nil { | ||||
| 						sp.SetStatus(tracer.SpanStatusError, err.Error()) | ||||
| 						if sp != nil { | ||||
| 							sp.SetStatus(tracer.SpanStatusError, err.Error()) | ||||
| 						} | ||||
| 						pc.kopts.Meter.Counter(semconv.SubscribeMessageTotal, "endpoint", record.Topic, "topic", record.Topic, "status", "failure").Inc() | ||||
| 						p.err = err | ||||
| 						p.msg.Body = record.Value | ||||
| @@ -276,7 +264,7 @@ func (pc *consumer) consume() { | ||||
| 								pc.c.MarkCommitRecords(record) | ||||
| 							} else { | ||||
| 								eventPool.Put(p) | ||||
| 								pc.connected.Store(0) | ||||
| 								//								pc.connected.Store(0) | ||||
| 								pc.kopts.Logger.Fatal(pc.kopts.Context, "[kgo] ErrLostMessage wtf?") | ||||
| 								return | ||||
| 							} | ||||
| @@ -293,19 +281,27 @@ func (pc *consumer) consume() { | ||||
| 						pc.kopts.Meter.Summary(semconv.SubscribeMessageLatencyMicroseconds, "endpoint", record.Topic, "topic", record.Topic).Update(te.Seconds()) | ||||
| 						pc.kopts.Meter.Histogram(semconv.SubscribeMessageDurationSeconds, "endpoint", record.Topic, "topic", record.Topic).Update(te.Seconds()) | ||||
| 						eventPool.Put(p) | ||||
| 						pc.connected.Store(0) | ||||
| 						//						pc.connected.Store(0) | ||||
| 						pc.kopts.Logger.Fatal(pc.kopts.Context, "[kgo] Unmarshal err not handled wtf?") | ||||
| 						sp.Finish() | ||||
| 						if sp != nil { | ||||
| 							sp.Finish() | ||||
| 						} | ||||
| 						return | ||||
| 					} | ||||
| 				} | ||||
| 				sp.AddEvent("handler start") | ||||
| 				if sp != nil { | ||||
| 					sp.AddEvent("handler start") | ||||
| 				} | ||||
| 				err := pc.handler(p) | ||||
| 				sp.AddEvent("handler stop") | ||||
| 				if sp != nil { | ||||
| 					sp.AddEvent("handler stop") | ||||
| 				} | ||||
| 				if err == nil { | ||||
| 					pc.kopts.Meter.Counter(semconv.SubscribeMessageTotal, "endpoint", record.Topic, "topic", record.Topic, "status", "success").Inc() | ||||
| 				} else { | ||||
| 					sp.SetStatus(tracer.SpanStatusError, err.Error()) | ||||
| 					if sp != nil { | ||||
| 						sp.SetStatus(tracer.SpanStatusError, err.Error()) | ||||
| 					} | ||||
| 					pc.kopts.Meter.Counter(semconv.SubscribeMessageTotal, "endpoint", record.Topic, "topic", record.Topic, "status", "failure").Inc() | ||||
| 				} | ||||
| 				pc.kopts.Meter.Counter(semconv.SubscribeMessageInflight, "endpoint", record.Topic, "topic", record.Topic).Dec() | ||||
| @@ -314,9 +310,13 @@ func (pc *consumer) consume() { | ||||
| 				} else if err != nil { | ||||
| 					p.err = err | ||||
| 					if eh != nil { | ||||
| 						sp.AddEvent("error handler start") | ||||
| 						if sp != nil { | ||||
| 							sp.AddEvent("error handler start") | ||||
| 						} | ||||
| 						_ = eh(p) | ||||
| 						sp.AddEvent("error handler stop") | ||||
| 						if sp != nil { | ||||
| 							sp.AddEvent("error handler stop") | ||||
| 						} | ||||
| 					} else { | ||||
| 						if pc.kopts.Logger.V(logger.ErrorLevel) { | ||||
| 							pc.kopts.Logger.Error(pc.kopts.Context, "[kgo]: subscriber error", err) | ||||
| @@ -331,13 +331,17 @@ func (pc *consumer) consume() { | ||||
| 					pc.c.MarkCommitRecords(record) | ||||
| 				} else { | ||||
| 					eventPool.Put(p) | ||||
| 					pc.connected.Store(0) | ||||
| 					//					pc.connected.Store(0) | ||||
| 					pc.kopts.Logger.Fatal(pc.kopts.Context, "[kgo] ErrLostMessage wtf?") | ||||
| 					sp.SetStatus(tracer.SpanStatusError, "ErrLostMessage") | ||||
| 					sp.Finish() | ||||
| 					if sp != nil { | ||||
| 						sp.SetStatus(tracer.SpanStatusError, "ErrLostMessage") | ||||
| 						sp.Finish() | ||||
| 					} | ||||
| 					return | ||||
| 				} | ||||
| 				sp.Finish() | ||||
| 				if sp != nil { | ||||
| 					sp.Finish() | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
|   | ||||
							
								
								
									
										19
									
								
								tracer.go
									
									
									
									
									
								
							
							
						
						
									
										19
									
								
								tracer.go
									
									
									
									
									
								
							| @@ -32,6 +32,9 @@ var ( | ||||
| // the record's context, so it can be ended in the OnProduceRecordUnbuffered | ||||
| // hook. | ||||
| func (m *hookTracer) OnProduceRecordBuffered(r *kgo.Record) { | ||||
| 	if !m.tracer.Enabled() { | ||||
| 		return | ||||
| 	} | ||||
| 	// Set up span options. | ||||
| 	attrs := []interface{}{ | ||||
| 		messagingSystem, | ||||
| @@ -77,6 +80,9 @@ func (m *hookTracer) OnProduceRecordBuffered(r *kgo.Record) { | ||||
| // It sets attributes with values unset when producing and records any error | ||||
| // that occurred during the publish operation. | ||||
| func (m *hookTracer) OnProduceRecordUnbuffered(r *kgo.Record, err error) { | ||||
| 	if !m.tracer.Enabled() { | ||||
| 		return | ||||
| 	} | ||||
| 	if span, ok := tracer.SpanFromContext(r.Context); ok { | ||||
| 		span.AddLabels( | ||||
| 			semconv.MessagingKafkaDestinationPartition(int(r.Partition)), | ||||
| @@ -96,6 +102,9 @@ func (m *hookTracer) OnProduceRecordUnbuffered(r *kgo.Record, err error) { | ||||
| // OnFetchRecordUnbuffered hook and can be used in downstream consumer | ||||
| // processing. | ||||
| func (m *hookTracer) OnFetchRecordBuffered(r *kgo.Record) { | ||||
| 	if !m.tracer.Enabled() { | ||||
| 		return | ||||
| 	} | ||||
| 	// Set up the span options. | ||||
| 	attrs := []interface{}{ | ||||
| 		messagingSystem, | ||||
| @@ -141,6 +150,9 @@ func (m *hookTracer) OnFetchRecordBuffered(r *kgo.Record) { | ||||
| // OnFetchRecordUnbuffered continues and ends the "receive" span for an | ||||
| // unbuffered record. | ||||
| func (m *hookTracer) OnFetchRecordUnbuffered(r *kgo.Record, _ bool) { | ||||
| 	if !m.tracer.Enabled() { | ||||
| 		return | ||||
| 	} | ||||
| 	span, _ := tracer.SpanFromContext(r.Context) | ||||
| 	span.Finish() | ||||
| } | ||||
| @@ -155,6 +167,13 @@ func (m *hookTracer) OnFetchRecordUnbuffered(r *kgo.Record, _ bool) { | ||||
| // not a record which has been created for producing, so call this at the start of each | ||||
| // iteration of your processing for the record. | ||||
| func (m *hookTracer) WithProcessSpan(r *kgo.Record) (context.Context, tracer.Span) { | ||||
| 	if r.Context == nil { | ||||
| 		r.Context = context.Background() | ||||
| 	} | ||||
|  | ||||
| 	if !m.tracer.Enabled() { | ||||
| 		return r.Context, nil | ||||
| 	} | ||||
| 	// Set up the span options. | ||||
| 	attrs := []interface{}{ | ||||
| 		messagingSystem, | ||||
|   | ||||
		Reference in New Issue
	
	Block a user