-
Notifications
You must be signed in to change notification settings - Fork 10
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
c578754
commit 2f86cbd
Showing
25 changed files
with
964 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1 @@ | ||
| clickhouseConnUrl: http://localhost:9000?username=admin&password=admin |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,31 @@ | ||
| module github.com/kube-tarian/quality-trace | ||
|
|
||
| go 1.19 | ||
|
|
||
| require ( | ||
| github.com/ClickHouse/clickhouse-go/v2 v2.3.0 | ||
| github.com/gorilla/mux v1.8.0 | ||
| github.com/stretchr/testify v1.8.0 | ||
| gopkg.in/yaml.v2 v2.4.0 | ||
| ) | ||
|
|
||
| require ( | ||
| github.com/ClickHouse/ch-go v0.47.3 // indirect | ||
| github.com/andybalholm/brotli v1.0.4 // indirect | ||
| github.com/davecgh/go-spew v1.1.1 // indirect | ||
| github.com/go-faster/city v1.0.1 // indirect | ||
| github.com/go-faster/errors v0.6.1 // indirect | ||
| github.com/google/uuid v1.3.0 // indirect | ||
| github.com/klauspost/compress v1.15.9 // indirect | ||
| github.com/kr/text v0.2.0 // indirect | ||
| github.com/paulmach/orb v0.7.1 // indirect | ||
| github.com/pierrec/lz4/v4 v4.1.15 // indirect | ||
| github.com/pkg/errors v0.9.1 // indirect | ||
| github.com/pmezard/go-difflib v1.0.0 // indirect | ||
| github.com/segmentio/asm v1.2.0 // indirect | ||
| github.com/shopspring/decimal v1.3.1 // indirect | ||
| go.opentelemetry.io/otel v1.9.0 // indirect | ||
| go.opentelemetry.io/otel/trace v1.9.0 // indirect | ||
| golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab // indirect | ||
| gopkg.in/yaml.v3 v3.0.1 // indirect | ||
| ) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,89 @@ | ||
| github.com/ClickHouse/ch-go v0.47.3 h1:bBKid8DRELKRf4/oXqrEks7Cc4DLb5Giwm9uazM6h3M= | ||
| github.com/ClickHouse/ch-go v0.47.3/go.mod h1:m3LHc5FeQ1Jjee5EEay5e7hQmSk4SuKyMfifNUz8l3g= | ||
| github.com/ClickHouse/clickhouse-go/v2 v2.3.0 h1:v0iT0yZspjjNgnLyPUa0WoGMme0Y/sNjCtOAFcyBkkA= | ||
| github.com/ClickHouse/clickhouse-go/v2 v2.3.0/go.mod h1:f2kb1LPopJdIyt0Y0vxNk9aiQCyhCmeVcyvOOaPCT4Q= | ||
| github.com/andybalholm/brotli v1.0.4 h1:V7DdXeJtZscaqfNuAdSRuRFzuiKlHSC/Zh3zl9qY3JY= | ||
| github.com/andybalholm/brotli v1.0.4/go.mod h1:fO7iG3H7G2nSZ7m0zPUDn85XEX2GTukHGRSepvi9Eig= | ||
| github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E= | ||
| github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | ||
| github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= | ||
| github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | ||
| github.com/go-faster/city v1.0.1 h1:4WAxSZ3V2Ws4QRDrscLEDcibJY8uf41H6AhXDrNDcGw= | ||
| github.com/go-faster/city v1.0.1/go.mod h1:jKcUJId49qdW3L1qKHH/3wPeUstCVpVSXTM6vO3VcTw= | ||
| github.com/go-faster/errors v0.6.1 h1:nNIPOBkprlKzkThvS/0YaX8Zs9KewLCOSFQS5BU06FI= | ||
| github.com/go-faster/errors v0.6.1/go.mod h1:5MGV2/2T9yvlrbhe9pD9LO5Z/2zCSq2T8j+Jpi2LAyY= | ||
| github.com/gogo/protobuf v1.3.2/go.mod h1:P1XiOD3dCwIKUDQYPy72D8LYyHL2YPYrpS2s69NZV8Q= | ||
| github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk= | ||
| github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= | ||
| github.com/google/go-cmp v0.5.8 h1:e6P7q2lk1O+qJJb4BtCQXlK8vWEO8V1ZeuEdJNOqZyg= | ||
| github.com/google/uuid v1.3.0 h1:t6JiXgmwXMjEs8VusXIJk2BXHsn+wx8BZdTaoZ5fu7I= | ||
| github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= | ||
| github.com/gorilla/mux v1.8.0 h1:i40aqfkR1h2SlN9hojwV5ZA91wcXFOvkdNIeFDP5koI= | ||
| github.com/gorilla/mux v1.8.0/go.mod h1:DVbg23sWSpFRCP0SfiEN6jmj59UnW/n46BH5rLB71So= | ||
| github.com/kisielk/errcheck v1.5.0/go.mod h1:pFxgyoBC7bSaBwPgfKdkLd5X25qrDl4LWUI2bnpBCr8= | ||
| github.com/kisielk/gotool v1.0.0/go.mod h1:XhKaO+MFFWcvkIS/tQcRk01m1F5IRFswLeQ+oQHNcck= | ||
| github.com/klauspost/compress v1.15.9 h1:wKRjX6JRtDdrE9qwa4b/Cip7ACOshUI4smpCQanqjSY= | ||
| github.com/klauspost/compress v1.15.9/go.mod h1:PhcZ0MbTNciWF3rruxRgKxI5NkcHHrHUDtV4Yw2GlzU= | ||
| github.com/kr/pretty v0.1.0 h1:L/CwN0zerZDmRFUapSPitk6f+Q3+0za1rQkzVuMiMFI= | ||
| github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY= | ||
| github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE= | ||
| github.com/paulmach/orb v0.7.1 h1:Zha++Z5OX/l168sqHK3k4z18LDvr+YAO/VjK0ReQ9rU= | ||
| github.com/paulmach/orb v0.7.1/go.mod h1:FWRlTgl88VI1RBx/MkrwWDRhQ96ctqMCh8boXhmqB/A= | ||
| github.com/paulmach/protoscan v0.2.1/go.mod h1:SpcSwydNLrxUGSDvXvO0P7g7AuhJ7lcKfDlhJCDw2gY= | ||
| github.com/pierrec/lz4/v4 v4.1.15 h1:MO0/ucJhngq7299dKLwIMtgTfbkoSPF6AoMYDd8Q4q0= | ||
| github.com/pierrec/lz4/v4 v4.1.15/go.mod h1:gZWDp/Ze/IJXGXf23ltt2EXimqmTUXEy0GFuRQyBid4= | ||
| github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4= | ||
| github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= | ||
| github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= | ||
| github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= | ||
| github.com/segmentio/asm v1.2.0 h1:9BQrFxC+YOHJlTlHGkTrFWf59nbL3XnCoFLTwDCI7ys= | ||
| github.com/segmentio/asm v1.2.0/go.mod h1:BqMnlJP91P8d+4ibuonYZw9mfnzI9HfxselHZr5aAcs= | ||
| github.com/shopspring/decimal v1.3.1 h1:2Usl1nmF/WZucqkFZhnfFYxxxu8LG21F6nPQBE5gKV8= | ||
| github.com/shopspring/decimal v1.3.1/go.mod h1:DKyhrW/HYNuLGql+MJL6WCR6knT2jwCFRcu2hWCYk4o= | ||
| github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= | ||
| github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw= | ||
| github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= | ||
| github.com/stretchr/testify v1.8.0 h1:pSgiaMZlXftHpm5L7V1+rVB+AZJydKsMxsQBIJw4PKk= | ||
| github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU= | ||
| github.com/yuin/goldmark v1.1.27/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74= | ||
| github.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74= | ||
| go.opentelemetry.io/otel v1.9.0 h1:8WZNQFIB2a71LnANS9JeyidJKKGOOremcUtb/OtHISw= | ||
| go.opentelemetry.io/otel v1.9.0/go.mod h1:np4EoPGzoPs3O67xUVNoPPcmSvsfOxNlNA4F4AC+0Eo= | ||
| go.opentelemetry.io/otel/trace v1.9.0 h1:oZaCNJUjWcg60VXWee8lJKlqhPbXAPB51URuR47pQYc= | ||
| go.opentelemetry.io/otel/trace v1.9.0/go.mod h1:2737Q0MuG8q1uILYm2YYVkAyLtOofiTNGg6VODnOiPo= | ||
| golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= | ||
| golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI= | ||
| golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto= | ||
| golang.org/x/mod v0.2.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA= | ||
| golang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA= | ||
| golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= | ||
| golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= | ||
| golang.org/x/net v0.0.0-20200226121028-0de0cce0169b/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= | ||
| golang.org/x/net v0.0.0-20201021035429-f5854403a974/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU= | ||
| golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= | ||
| golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= | ||
| golang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= | ||
| golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= | ||
| golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= | ||
| golang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= | ||
| golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab h1:2QkjZIsXupsJbJIdSjjUOgWK3aEtzyuh2mPt3l/CkeU= | ||
| golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||
| golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= | ||
| golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= | ||
| golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= | ||
| golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= | ||
| golang.org/x/tools v0.0.0-20200619180055-7c47624df98f/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE= | ||
| golang.org/x/tools v0.0.0-20210106214847-113979e3529a/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA= | ||
| golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= | ||
| golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= | ||
| golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= | ||
| golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= | ||
| google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw= | ||
| google.golang.org/protobuf v1.27.1/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc= | ||
| gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= | ||
| gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 h1:qIbj1fsPNlZgppZ+VLlY7N33q108Sa+fhmuc+sWQYwY= | ||
| gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY= | ||
| gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ= | ||
| gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= | ||
| gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= | ||
| gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,45 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "context" | ||
| "flag" | ||
| "fmt" | ||
| "log" | ||
| "net/http" | ||
|
|
||
| "github.com/gorilla/mux" | ||
| "github.com/kube-tarian/quality-trace/server/adapters/clickhousereader" | ||
| "github.com/kube-tarian/quality-trace/server/config" | ||
| "github.com/kube-tarian/quality-trace/server/model" | ||
| "github.com/kube-tarian/quality-trace/server/tests" | ||
| ) | ||
|
|
||
| var cfg = flag.String("config", "config.yaml", "path to the config file") | ||
|
|
||
| func main() { | ||
| ctx := context.Background() | ||
| fmt.Println("Context in main func", ctx) | ||
|
|
||
| flag.Parse() | ||
| cfg, err := config.FromFile(*cfg) | ||
| if err != nil { | ||
| log.Fatal(err) | ||
| } | ||
|
|
||
| // Initialize reader with connection to clickhouse | ||
| connUrl := cfg.ClickhouseConnUrl | ||
| reader := clickhousereader.NewReader(connUrl) | ||
|
|
||
| // Initializing mux router and performing handler function | ||
| s := mux.NewRouter() | ||
|
|
||
| r := &tests.TestHandler{Ctx: ctx, Reader: *reader, Tests: map[int]*model.Test{}} | ||
|
|
||
| s.HandleFunc("/test/", r.CreateTest).Methods("POST") | ||
| // s.HandleFunc("/test/delete", r.DeleteTest) | ||
| s.HandleFunc("/test/{id:[0-9]+}", r.GetTest).Methods("GET") | ||
| s.HandleFunc("/test/{id:[0-9]+}/run", r.RunTest).Methods("POST") | ||
|
|
||
| fmt.Printf("Server started at :8080...") | ||
| http.ListenAndServe(":8080", s) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,21 @@ | ||
| name: POST New Book | ||
| trigger: | ||
| type: http | ||
| httpRequest: | ||
| url: localhost:8090 | ||
| route: /books | ||
| method: GET | ||
| # contentType: application/json | ||
| # body: '{ "title": "foo", "author": "bar"}' | ||
| specs: | ||
| - name: Assert that status code is 200 | ||
| selectors: | ||
| serviceName: goApp | ||
| httpMethod: GET | ||
| httpRoute: /books | ||
| httpHost: localhost:8090 | ||
| assertions: | ||
| responseStatusCode: 200 | ||
| maxRetries: 3 | ||
| retryInterval: 4 | ||
|
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,21 @@ | ||
| name: POST New Book | ||
| trigger: | ||
| type: http | ||
| httpRequest: | ||
| url: localhost:8090 | ||
| route: /books | ||
| method: POST | ||
| contentType: application/json | ||
| body: '{ "title": "foo", "author": "bar"}' | ||
| specs: | ||
| - name: Assert that status code is 200 | ||
| selectors: | ||
| serviceName: goApp | ||
| httpMethod: POST | ||
| httpRoute: /books | ||
| httpHost: localhost:8090 | ||
| assertions: | ||
| responseStatusCode: 200 | ||
| maxRetries: 3 | ||
| retryInterval: 4 | ||
|
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,3 @@ | ||
| # Clickhouse Reader | ||
|
|
||
| This package has been copied from Signoz and adapted, see <https://github.com/SigNoz/signoz/tree/9c4521b34a82d5fcd01cde39ea3de6a796c54ea6/pkg/query-service/app/clickhouseReader>. |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1 @@ | ||
| package clickhousereader |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,148 @@ | ||
| package clickhousereader | ||
|
|
||
| import ( | ||
| "context" | ||
| "fmt" | ||
| "net/url" | ||
| "time" | ||
|
|
||
| clickhouse "github.com/ClickHouse/clickhouse-go/v2" | ||
| ) | ||
|
|
||
| type Encoding string | ||
|
|
||
| const ( | ||
| // EncodingJSON is used for spans encoded as JSON. | ||
| EncodingJSON Encoding = "json" | ||
| // EncodingProto is used for spans encoded as Protobuf. | ||
| EncodingProto Encoding = "protobuf" | ||
| ) | ||
|
|
||
| const ( | ||
| defaultDatasource string = "tcp://localhost:9000" | ||
| defaultTraceDB string = "signoz_traces" | ||
| defaultSpansTable string = "signoz_spans" | ||
| defaultLiveTailRefreshSeconds int = 10 | ||
| defaultWriteBatchDelay time.Duration = 5 * time.Second | ||
| defaultWriteBatchSize int = 10000 | ||
| defaultEncoding Encoding = EncodingJSON | ||
| ) | ||
|
|
||
| // NamespaceConfig is Clickhouse's internal configuration data | ||
| type namespaceConfig struct { | ||
| namespace string | ||
| Enabled bool | ||
| Datasource string | ||
| TraceDB string | ||
| OperationsTable string | ||
| IndexTable string | ||
| DurationTable string | ||
| UsageExplorerTable string | ||
| SpansTable string | ||
| ErrorTable string | ||
| DependencyGraphTable string | ||
| TopLevelOperationsTable string | ||
| LogsDB string | ||
| LogsTable string | ||
| LogsAttributeKeysTable string | ||
| LogsResourceKeysTable string | ||
| LiveTailRefreshSeconds int | ||
| WriteBatchDelay time.Duration | ||
| WriteBatchSize int | ||
| Encoding Encoding | ||
| Connector Connector | ||
| } | ||
|
|
||
| // Connecto defines how to connect to the database | ||
| type Connector func(cfg *namespaceConfig) (clickhouse.Conn, error) | ||
|
|
||
| func defaultConnector(cfg *namespaceConfig) (clickhouse.Conn, error) { | ||
| ctx := context.Background() | ||
| dsnURL, err := url.Parse(cfg.Datasource) | ||
| fmt.Println("URL:", dsnURL) | ||
| options := &clickhouse.Options{ | ||
| Addr: []string{dsnURL.Host}, | ||
| } | ||
| if dsnURL.Query().Get("username") != "" { | ||
| auth := clickhouse.Auth{ | ||
| Username: "admin", | ||
| Password: "admin", | ||
| } | ||
| options.Auth = auth | ||
| } | ||
|
|
||
| fmt.Printf("Options:%v\n", options) | ||
|
|
||
| db, err := clickhouse.Open(options) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| if err := db.Ping(ctx); err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| return db, nil | ||
| } | ||
|
|
||
| // Options store storage plugin related configs | ||
| type Options struct { | ||
| primary *namespaceConfig | ||
|
|
||
| others map[string]*namespaceConfig | ||
| } | ||
|
|
||
| // NewOptions creates a new Options struct. | ||
| func NewOptions(datasource string, primaryNamespace string, otherNamespaces ...string) *Options { | ||
|
|
||
| if datasource == "" { | ||
| datasource = defaultDatasource | ||
| } | ||
|
|
||
| options := &Options{ | ||
| primary: &namespaceConfig{ | ||
| namespace: primaryNamespace, | ||
| Enabled: true, | ||
| Datasource: datasource, | ||
| TraceDB: defaultTraceDB, | ||
| SpansTable: defaultSpansTable, | ||
| LiveTailRefreshSeconds: defaultLiveTailRefreshSeconds, | ||
| WriteBatchDelay: defaultWriteBatchDelay, | ||
| WriteBatchSize: defaultWriteBatchSize, | ||
| Encoding: defaultEncoding, | ||
| Connector: defaultConnector, | ||
| }, | ||
| others: make(map[string]*namespaceConfig, len(otherNamespaces)), | ||
| } | ||
|
|
||
| for _, namespace := range otherNamespaces { | ||
| if namespace == archiveNamespace { | ||
| options.others[namespace] = &namespaceConfig{ | ||
| namespace: namespace, | ||
| Datasource: datasource, | ||
| TraceDB: "", | ||
| OperationsTable: "", | ||
| IndexTable: "", | ||
| ErrorTable: "", | ||
| LogsDB: "", | ||
| LogsTable: "", | ||
| LogsAttributeKeysTable: "", | ||
| LogsResourceKeysTable: "", | ||
| LiveTailRefreshSeconds: defaultLiveTailRefreshSeconds, | ||
| WriteBatchDelay: defaultWriteBatchDelay, | ||
| WriteBatchSize: defaultWriteBatchSize, | ||
| Encoding: defaultEncoding, | ||
| Connector: defaultConnector, | ||
| } | ||
| } else { | ||
| options.others[namespace] = &namespaceConfig{namespace: namespace} | ||
| } | ||
| } | ||
|
|
||
| return options | ||
| } | ||
|
|
||
| // GetPrimary returns the primary namespace configuration | ||
| func (opt *Options) getPrimary() *namespaceConfig { | ||
| return opt.primary | ||
| } |
Oops, something went wrong.