diff --git a/cmd/servicechecker/main.go b/cmd/servicechecker/main.go index fc40ce5..1a30316 100644 --- a/cmd/servicechecker/main.go +++ b/cmd/servicechecker/main.go @@ -7,7 +7,6 @@ import ( "os/signal" "time" - openapi_v3 "github.com/google/gnostic/openapiv3" "github.com/google/uuid" grpccli "go.unistack.org/micro-client-grpc/v3" httpcli "go.unistack.org/micro-client-http/v3" @@ -31,6 +30,7 @@ import ( "go.unistack.org/servicechecker/pkg/httpconn" "go.unistack.org/servicechecker/pkg/protoset" "go.unistack.org/servicechecker/pkg/scheduler" + "go.unistack.org/servicechecker/pkg/swaggerset" ) var ( @@ -213,8 +213,10 @@ func newHTTPTask(ctx context.Context, l logger.Logger, m meter.Meter, check stri var treq client.Request var opts []client.CallOption var labels []string + swaggerSet := swaggerset.NewSwaggerSet() if task.HTTP.OpenAPI != "" { + var svc string openapiBuf, err := os.ReadFile(task.HTTP.OpenAPI) if err != nil { @@ -222,12 +224,11 @@ func newHTTPTask(ctx context.Context, l logger.Logger, m meter.Meter, check stri return nil, nil, err } - doc, err := openapi_v3.ParseDocument(openapiBuf) + err = swaggerSet.AddSwaggerset(task.HTTP.Addr, svc, openapiBuf) if err != nil { - l.Error(ctx, "failed to unmarshal openapi file", err) + l.Error(ctx, "failed to add openApi spec", err) return nil, nil, err } - _ = doc errmap := make(map[string]interface{}, 1) errmap["default"] = &codecpb.Frame{} @@ -239,8 +240,17 @@ func newHTTPTask(ctx context.Context, l logger.Logger, m meter.Meter, check stri // client.WithContentType("application/json"), } + msg, err := swaggerSet.GetMessage(task.HTTP.Addr, svc, task.HTTP.Method) + if err != nil { + l.Error(ctx, "failed to get message from swagger spec", err) + return nil, nil, err + } + req = &codecpb.Frame{Data: []byte(task.HTTP.Data)} - rsp = &codecpb.Frame{} + if task.HTTP.Data == "" { + req = msg.Request + } + rsp = msg.Response treq = c.NewRequest(task.Name, task.Name, req) diff --git a/go.mod b/go.mod index d284214..aee1f3d 100644 --- a/go.mod +++ b/go.mod @@ -3,9 +3,10 @@ module go.unistack.org/servicechecker go 1.23.3 require ( + github.com/getkin/kin-openapi v0.128.0 github.com/go-co-op/gocron/v2 v2.12.3 - github.com/google/gnostic v0.7.0 github.com/google/uuid v1.6.0 + github.com/ompluscator/dynamic-struct v1.4.0 github.com/stretchr/testify v1.9.0 go.unistack.org/micro-client-grpc/v3 v3.11.10 go.unistack.org/micro-client-http/v3 v3.9.14 @@ -15,6 +16,7 @@ require ( go.unistack.org/micro-codec-yaml/v3 v3.10.2 go.unistack.org/micro-meter-victoriametrics/v3 v3.8.9 go.unistack.org/micro-proto/v3 v3.4.1 + go.unistack.org/micro-proto/v4 v4.1.0 go.unistack.org/micro-server-http/v3 v3.11.34 go.unistack.org/micro/v3 v3.10.100 google.golang.org/protobuf v1.35.2 @@ -22,8 +24,16 @@ require ( require ( github.com/davecgh/go-spew v1.1.1 // indirect + github.com/go-openapi/jsonpointer v0.21.0 // indirect + github.com/go-openapi/swag v0.23.0 // indirect + github.com/google/gnostic v0.7.0 // indirect github.com/google/gnostic-models v0.6.9-0.20230804172637-c7be7c783f49 // indirect + github.com/invopop/yaml v0.3.1 // indirect github.com/jonboulle/clockwork v0.4.0 // indirect + github.com/josharian/intern v1.0.0 // indirect + github.com/mailru/easyjson v0.7.7 // indirect + github.com/mohae/deepcopy v0.0.0-20170929034955-c48cc78d4826 // indirect + github.com/perimeterx/marshmallow v1.1.5 // indirect github.com/pmezard/go-difflib v1.0.0 // indirect github.com/robfig/cron/v3 v3.0.1 // indirect github.com/valyala/fastrand v1.1.0 // indirect diff --git a/go.sum b/go.sum index 34360c3..f7f07a3 100644 --- a/go.sum +++ b/go.sum @@ -660,6 +660,8 @@ github.com/envoyproxy/protoc-gen-validate v0.10.1/go.mod h1:DRjgyB0I43LtJapqN6Ni github.com/flowstack/go-jsonschema v0.1.1/go.mod h1:yL7fNggx1o8rm9RlgXv7hTBWxdBM0rVwpMwimd3F3N0= github.com/fogleman/gg v1.2.1-0.20190220221249-0403632d5b90/go.mod h1:R/bRT+9gY/C5z7JzPU0zXsXHKM4/ayA+zqcVNZzPa1k= github.com/fogleman/gg v1.3.0/go.mod h1:R/bRT+9gY/C5z7JzPU0zXsXHKM4/ayA+zqcVNZzPa1k= +github.com/getkin/kin-openapi v0.128.0 h1:jqq3D9vC9pPq1dGcOCv7yOp1DaEe7c/T1vzcLbITSp4= +github.com/getkin/kin-openapi v0.128.0/go.mod h1:OZrfXzUfGrNbsKj+xmFBx6E5c6yH3At/tAKSc2UszXM= github.com/ghodss/yaml v1.0.0/go.mod h1:4dBDuWmgqj2HViK6kFavaiC9ZROes6MMH2rRYeMEF04= github.com/go-co-op/gocron/v2 v2.12.3 h1:3JkKjkFoAPp/i0YE+sonlF5gi+xnBChwYh75nX16MaE= github.com/go-co-op/gocron/v2 v2.12.3/go.mod h1:xY7bJxGazKam1cz04EebrlP4S9q4iWdiAylMGP3jY9w= @@ -673,8 +675,14 @@ github.com/go-gl/glfw/v3.3/glfw v0.0.0-20191125211704-12ad95a8df72/go.mod h1:tQ2 github.com/go-gl/glfw/v3.3/glfw v0.0.0-20200222043503-6f7a984d4dc4/go.mod h1:tQ2UAYgL5IevRw8kRxooKSPJfGvJ9fJQFa0TUsXzTg8= github.com/go-latex/latex v0.0.0-20210118124228-b3d85cf34e07/go.mod h1:CO1AlKB2CSIqUrmQPqA0gdRIlnLEY0gK5JGjh37zN5U= github.com/go-latex/latex v0.0.0-20210823091927-c0d11ff05a81/go.mod h1:SX0U8uGpxhq9o2S/CELCSUxEWWAuoCUcVCQWv7G2OCk= +github.com/go-openapi/jsonpointer v0.21.0 h1:YgdVicSA9vH5RiHs9TZW5oyafXZFc6+2Vc1rr/O9oNQ= +github.com/go-openapi/jsonpointer v0.21.0/go.mod h1:IUyH9l/+uyhIYQ/PXVA41Rexl+kOkAPDdXEYns6fzUY= +github.com/go-openapi/swag v0.23.0 h1:vsEVJDUo2hPJ2tu0/Xc+4noaxyEffXNIs3cOULZ+GrE= +github.com/go-openapi/swag v0.23.0/go.mod h1:esZ8ITTYEsH1V2trKHjAN8Ai7xHb8RV+YSZ577vPjgQ= github.com/go-pdf/fpdf v0.5.0/go.mod h1:HzcnA+A23uwogo0tp9yU+l3V+KXhiESpt1PMayhOh5M= github.com/go-pdf/fpdf v0.6.0/go.mod h1:HzcnA+A23uwogo0tp9yU+l3V+KXhiESpt1PMayhOh5M= +github.com/go-test/deep v1.0.8 h1:TDsG77qcSprGbC6vTN8OuXp5g+J+b5Pcguhf7Zt61VM= +github.com/go-test/deep v1.0.8/go.mod h1:5C2ZWiW0ErCdrYzpqxLbTX7MG14M9iiw8DgHncVwcsE= github.com/goccy/go-json v0.9.11/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I= github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0/go.mod h1:E/TSTwGwJL78qG/PmXZO1EjYhfJinVAhrmmHX6Z8B9k= github.com/golang/glog v0.0.0-20160126235308-23def4e6c14b/go.mod h1:SBH7ygxi8pfUlaOkMMuAQtPIUF8ecWP5IEl/CR7VP2Q= @@ -790,8 +798,12 @@ github.com/hashicorp/golang-lru v0.5.1/go.mod h1:/m3WP610KZHVQ1SGc6re/UDhFvYD7pJ github.com/iancoleman/strcase v0.2.0/go.mod h1:iwCmte+B7n89clKwxIoIXy/HfoL7AsD47ZCWhYzw7ho= github.com/ianlancetaylor/demangle v0.0.0-20181102032728-5e5cf60278f6/go.mod h1:aSSvb/t6k1mPoxDqO4vJh6VOCGPwU4O0C2/Eqndh1Sc= github.com/ianlancetaylor/demangle v0.0.0-20200824232613-28f6c0f3b639/go.mod h1:aSSvb/t6k1mPoxDqO4vJh6VOCGPwU4O0C2/Eqndh1Sc= +github.com/invopop/yaml v0.3.1 h1:f0+ZpmhfBSS4MhG+4HYseMdJhoeeopbSKbq5Rpeelso= +github.com/invopop/yaml v0.3.1/go.mod h1:PMOp3nn4/12yEZUFfmOuNHJsZToEEOwoWsT+D81KkeA= github.com/jonboulle/clockwork v0.4.0 h1:p4Cf1aMWXnXAUh8lVfewRBx1zaTSYKrKMF2g3ST4RZ4= github.com/jonboulle/clockwork v0.4.0/go.mod h1:xgRqUGwRcjKCO1vbZUEtSLrqKoPSsUpK7fnezOII0kc= +github.com/josharian/intern v1.0.0 h1:vlS4z54oSdjm0bgjRigI+G1HpF+tI+9rE5LLzOg8HmY= +github.com/josharian/intern v1.0.0/go.mod h1:5DoeVV0s6jJacbCEi61lwdGj/aVlrQvzHFFd8Hwg//Y= github.com/jstemmer/go-junit-report v0.0.0-20190106144839-af01ea7f8024/go.mod h1:6v2b51hI/fHJwM22ozAgKL4VKDeJcHhJFhtBdhmNjmU= github.com/jstemmer/go-junit-report v0.9.1/go.mod h1:Brl9GWCQeLvo8nXZwPNNblvFj/XSXhF0NWZEnDohbsk= github.com/jung-kurt/gofpdf v1.0.0/go.mod h1:7Id9E/uU8ce6rXgefFLlgrJj/GYY22cpxn+r32jIOes= @@ -805,8 +817,9 @@ github.com/kr/fs v0.1.0/go.mod h1:FFnZGqtBN9Gxj7eW1uZ42v5BccTP0vu6NEaFoC2HwRg= github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo= github.com/kr/pretty v0.2.0/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI= github.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI= -github.com/kr/pretty v0.3.0 h1:WgNl7dwNpEZ6jJ9k1snq4pZsg7DOEN8hP9Xw0Tsjwk0= github.com/kr/pretty v0.3.0/go.mod h1:640gp4NfQd8pI5XOwp5fnNeVWj67G7CFk/SaSQn7NBk= +github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE= +github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk= github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY= @@ -814,12 +827,20 @@ github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE= github.com/lyft/protoc-gen-star v0.6.0/go.mod h1:TGAoBVkt8w7MPG72TrKIu85MIdXwDuzJYeZuUPFPNwA= github.com/lyft/protoc-gen-star v0.6.1/go.mod h1:TGAoBVkt8w7MPG72TrKIu85MIdXwDuzJYeZuUPFPNwA= github.com/lyft/protoc-gen-star/v2 v2.0.1/go.mod h1:RcCdONR2ScXaYnQC5tUzxzlpA3WVYF7/opLeUgcQs/o= +github.com/mailru/easyjson v0.7.7 h1:UGYAvKxe3sBsEDzO8ZeWOSlIQfWFlxbzLZe7hwFURr0= +github.com/mailru/easyjson v0.7.7/go.mod h1:xzfreul335JAWq5oZzymOObrkdz5UnU4kGfJJLY9Nlc= github.com/mattn/go-isatty v0.0.12/go.mod h1:cbi8OIDigv2wuxKPP5vlRcQ1OAZbq2CE4Kysco4FUpU= github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM= github.com/mattn/go-sqlite3 v1.14.14/go.mod h1:NyWgC/yNuGj7Q9rpYnZvas74GogHl5/Z4A/KQRfk6bU= github.com/minio/asm2plan9s v0.0.0-20200509001527-cdd76441f9d8/go.mod h1:mC1jAcsrzbxHt8iiaC+zU4b1ylILSosueou12R++wfY= github.com/minio/c2goasm v0.0.0-20190812172519-36a3d3bbc4f3/go.mod h1:RagcQ7I8IeTMnF8JTXieKnO4Z6JCsikNEzj0DwauVzE= +github.com/mohae/deepcopy v0.0.0-20170929034955-c48cc78d4826 h1:RWengNIwukTxcDr9M+97sNutRR1RKhG96O6jWumTTnw= +github.com/mohae/deepcopy v0.0.0-20170929034955-c48cc78d4826/go.mod h1:TaXosZuwdSHYgviHp1DAtfrULt5eUgsSMsZf+YrPgl8= +github.com/ompluscator/dynamic-struct v1.4.0 h1:I/Si9LZtItSwiTMe7vosEuIu2TKdOvWbE3R/lokpN4Q= +github.com/ompluscator/dynamic-struct v1.4.0/go.mod h1:ADQ1+6Ox1D+ntuNwTHyl1NvpAqY2lBXPSPbcO4CJdeA= github.com/patrickmn/go-cache v2.1.0+incompatible/go.mod h1:3Qf8kWWT7OJRJbdiICTKqZju1ZixQ/KpMGzzAfe6+WQ= +github.com/perimeterx/marshmallow v1.1.5 h1:a2LALqQ1BlHM8PZblsDdidgv1mWi1DgC2UmX50IvK2s= +github.com/perimeterx/marshmallow v1.1.5/go.mod h1:dsXbUu8CRzfYP5a87xpp0xq9S3u0Vchtcl8we9tYaXw= github.com/phpdave11/gofpdf v1.4.2/go.mod h1:zpO6xFn9yxo3YLyMvW8HcKWVdbNqgIfOOp2dXMnm1mY= github.com/phpdave11/gofpdi v1.0.12/go.mod h1:vBmVV0Do6hSBHC8uKUQ71JGW+ZGQq74llk/7bXwjDoI= github.com/phpdave11/gofpdi v1.0.13/go.mod h1:vBmVV0Do6hSBHC8uKUQ71JGW+ZGQq74llk/7bXwjDoI= @@ -840,8 +861,9 @@ github.com/robfig/cron/v3 v3.0.1/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzG github.com/rogpeppe/fastuuid v1.2.0/go.mod h1:jVj6XXZzXRy/MSR5jhDC/2q6DgLz+nrA6LYCDYWNEvQ= github.com/rogpeppe/go-internal v1.3.0/go.mod h1:M8bDsm7K2OlrFYOpmOWEs/qY81heoFRclV5y23lUDJ4= github.com/rogpeppe/go-internal v1.6.1/go.mod h1:xXDCJY+GAPziupqXw64V24skbSoqbTEfhy4qGm1nDQc= -github.com/rogpeppe/go-internal v1.9.0 h1:73kH8U+JUqXU8lRuOHeVHaa/SZPifC7BkcraZVejAe8= github.com/rogpeppe/go-internal v1.9.0/go.mod h1:WtVeX8xhTBvf0smdhujwtBcq4Qrzq/fJaraNFVN+nFs= +github.com/rogpeppe/go-internal v1.12.0 h1:exVL4IDcn6na9z1rAb56Vxr+CgyK3nn3O+epU5NdKM8= +github.com/rogpeppe/go-internal v1.12.0/go.mod h1:E+RYuTGaKKdloAfM02xzb0FW3Paa99yedzYV+kq4uf4= github.com/ruudk/golang-pdf417 v0.0.0-20181029194003-1af4ab5afa58/go.mod h1:6lfFZQK844Gfx8o5WFuvpxWRwnSoipWe/p622j1v06w= github.com/ruudk/golang-pdf417 v0.0.0-20201230142125-a7e3863a1245/go.mod h1:pQAZKsJ8yyVxGRWYNEm9oFB8ieLgKFnamEyDmSA0BRk= github.com/silas/dag v0.0.0-20220518035006-a7e85ada93c5/go.mod h1:7RTUFBdIRC9nZ7/3RyRNH1bdqIShrDejd1YbLwgPS+I= @@ -865,6 +887,8 @@ github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o github.com/stretchr/testify v1.8.3/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo= github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg= github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY= +github.com/ugorji/go/codec v1.2.7 h1:YPXUKf7fYbp/y8xloBqZOw2qaVggbfwMlI8WM3wZUJ0= +github.com/ugorji/go/codec v1.2.7/go.mod h1:WGN1fab3R1fzQlVQTkfxVtIBhWDRqOviHU95kRgeqEY= github.com/valyala/fastrand v1.1.0 h1:f+5HkLW4rsgzdNoleUOB69hyT9IlD2ZQh9GyDMfb5G8= github.com/valyala/fastrand v1.1.0/go.mod h1:HWqCzkrkg6QXT8V2EXWvXCoow7vLwOFN002oeRzjapQ= github.com/valyala/histogram v1.2.0 h1:wyYGAZZt3CpwUiIb9AU/Zbllg1llXyrtApRS815OLoQ= @@ -912,6 +936,8 @@ go.unistack.org/micro-meter-victoriametrics/v3 v3.8.9 h1:ZXCS0eFiSdvcFYxpxV2Q77g go.unistack.org/micro-meter-victoriametrics/v3 v3.8.9/go.mod h1:xODJQ0Nu/F8k34D/z2ITL91OskI/C674XCkugAxmc3Q= 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-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-server-http/v3 v3.11.34 h1:Cu44/IWyTBa0I1LIt8TZu/z130uDs2t/OZJwV5H5NX0= go.unistack.org/micro-server-http/v3 v3.11.34/go.mod h1:3hlTQj3e6HQziJa2Coz/BwvVbt3oRejOevyAPUCuBEk= go.unistack.org/micro/v3 v3.10.94/go.mod h1:erMgt3Bl7vQQ0e9UpQyR5NlLiZ9pKeEJ9+1tfYFaqUg= diff --git a/pkg/swaggerset/swagger.yaml b/pkg/swaggerset/swagger.yaml new file mode 100644 index 0000000..56dade6 --- /dev/null +++ b/pkg/swaggerset/swagger.yaml @@ -0,0 +1,81 @@ +openapi: 3.0.3 +info: + title: platform/services/domain/service-proto + description: Domain Service + version: 3.6.0 +paths: + /domain-service/v1/push_mail/enabled: + get: + tags: + - DomainService + description: Получение статуса подключения PUSH (глобального) + operationId: IsPushTokenEnabled + parameters: + - name: Phone + in: header + schema: + type: string + - name: app_name + in: query + schema: + type: string + - name: device_id.value + in: query + description: The string value. + schema: + type: string + responses: + "200": + description: OK + content: + application/json: + schema: + $ref: '#/components/schemas/IsPushTokenEnabledRsp' + post: + tags: + - DomainService + description: Сохранение статуса подключения PUSH (глобального) + operationId: SetPushTokenEnabled + requestBody: + content: + application/json: + schema: + $ref: '#/components/schemas/SetPushTokenEnabledReq' + required: true + responses: + "200": + description: OK + content: + application/json: + schema: + $ref: '#/components/schemas/SetPushTokenEnabledRsp' +components: + schemas: + IsPushTokenEnabledRsp: + type: object + properties: + result: + type: boolean + setPushTokenEnabledRsp: + $ref: '#/components/schemas/SetPushTokenEnabledRsp' + SetPushTokenEnabledReq: + type: object + properties: + app_name: + type: string + enabled: + type: boolean + device_id: + $ref: '#/components/schemas/StringValue' + StringValue: + type: object + properties: + value: + type: string + description: The string value. + description: Wrapper message for `string`. The JSON representation for `StringValue` is JSON string. + SetPushTokenEnabledRsp: + type: object + properties: + result: + type: boolean \ No newline at end of file diff --git a/pkg/swaggerset/swaggerset.go b/pkg/swaggerset/swaggerset.go new file mode 100644 index 0000000..d3f9752 --- /dev/null +++ b/pkg/swaggerset/swaggerset.go @@ -0,0 +1,82 @@ +package swaggerset + +import ( + "errors" + "net/http" + "sync" + + openapi "go.unistack.org/micro-proto/v4/openapiv3" +) + +var errNotFound = errors.New("file descriptor not found") + +type SwaggerSet struct { + mu sync.Mutex + files map[string]*openapi.Document +} + +func NewSwaggerSet() *SwaggerSet { + return &SwaggerSet{ + mu: sync.Mutex{}, + files: make(map[string]*openapi.Document, 0), + } +} + +func (p *SwaggerSet) GetMessage(addr, svc, mth string) (*Message, error) { + if svc == "" || mth == "" || addr == "" { + return nil, errors.New("addr or service name is empty") + } + + p.mu.Lock() + doc := p.files[addr+"|"+svc] + p.mu.Unlock() + + var reqParam, reqBody, rsp interface{} + var typeReq string + for _, path := range doc.Paths.GetPath() { + if path.GetName() == mth { + if path.GetValue().GetGet() != nil { + typeReq = http.MethodGet + reqParam, reqBody, rsp = handleOperation(path.GetValue().GetGet(), doc) + } + if path.GetValue().GetPost() != nil { + typeReq = http.MethodPost + reqParam, reqBody, rsp = handleOperation(path.GetValue().GetPost(), doc) + } + if path.GetValue().GetDelete() != nil { + typeReq = http.MethodDelete + reqParam, reqBody, rsp = handleOperation(path.GetValue().GetDelete(), doc) + } + if path.GetValue().GetPatch() != nil { + typeReq = http.MethodPatch + reqParam, reqBody, rsp = handleOperation(path.GetValue().GetPatch(), doc) + } + if path.GetValue().GetPut() != nil { + typeReq = http.MethodPut + reqParam, reqBody, rsp = handleOperation(path.GetValue().GetPut(), doc) + } + } + } + msg := &Message{ + Type: typeReq, + Request: httpRequest{ + Header: reqParam, + Body: reqBody, + }, + Response: rsp, + } + + return msg, nil +} + +func (p *SwaggerSet) AddSwaggerset(addr, svc string, data []byte) error { + doc, err := openapi.ParseDocument(data) + if err != nil { + return err + } + + p.mu.Lock() + p.files[addr+"|"+svc] = doc + p.mu.Unlock() + return nil +} diff --git a/pkg/swaggerset/swaggerset_test.go b/pkg/swaggerset/swaggerset_test.go new file mode 100644 index 0000000..513a39c --- /dev/null +++ b/pkg/swaggerset/swaggerset_test.go @@ -0,0 +1,33 @@ +package swaggerset + +import ( + "encoding/json" + "fmt" + "os" + "testing" + + "github.com/stretchr/testify/assert" +) + +func TestSwaggerSet_1(t *testing.T) { + s := NewSwaggerSet() + data, err := os.ReadFile("swagger.yaml") + assert.Nil(t, err) + + err = s.AddSwaggerset("localhost:8080", "service", data) + assert.Nil(t, err) + + msg, err := s.GetMessage( + "localhost:8080", + "service", + "/domain-service/v1/push_mail/enabled", + "GET") + assert.Nil(t, err) + assert.NotNil(t, msg) + req, err := json.Marshal(msg.Request) + assert.Nil(t, err) + rsp, err := json.Marshal(msg.Response) + assert.Nil(t, err) + fmt.Printf("JSON: type: %s, req: %s, rsp: %s \n", msg.Type, req, rsp) + fmt.Printf("Struct: type: %s, req: %+v, rsp: %+v \n", msg.Type, msg.Request, msg.Response) +} diff --git a/pkg/swaggerset/util.go b/pkg/swaggerset/util.go new file mode 100644 index 0000000..6291e66 --- /dev/null +++ b/pkg/swaggerset/util.go @@ -0,0 +1,189 @@ +package swaggerset + +import ( + "fmt" + "reflect" + "strings" + + dynamicstruct "github.com/ompluscator/dynamic-struct" + openapi "go.unistack.org/micro-proto/v4/openapiv3" +) + +type Message struct { + Type string + Request httpRequest + Response interface{} +} + +type httpRequest struct { + Header interface{} + Body interface{} +} + +func handleOperation(operation *openapi.Operation, doc *openapi.Document) (reqParam, reqBody, rsp interface{}) { + // Обработка параметров (GET) + if len(operation.Parameters) > 0 { + paramsStruct := dynamicstruct.NewStruct() + for _, paramRef := range operation.Parameters { + var param *openapi.Parameter + param = paramRef.GetParameter() + fieldName := capitalize(param.Name) + jsonName := strings.ToLower(param.Name[:1]) + param.Name[1:] + goType := getGoType(doc, param.Schema) + + // В зависимости от того, где параметр находится (header, query, path, etc.), добавляем соответствующий тег + switch param.In { + case "query": + paramsStruct = paramsStruct.AddField(fieldName, goType, fmt.Sprintf(`json:"%s" query:"%s"`, jsonName, jsonName)) + case "header": + paramsStruct = paramsStruct.AddField(fieldName, goType, fmt.Sprintf(`json:"%s" header:"%s"`, jsonName, jsonName)) + default: + paramsStruct = paramsStruct.AddField(fieldName, goType, fmt.Sprintf(`json:"%s"`, jsonName)) + } + } + // Получили структуру запроса для методов, где есть параметры в header, query, path, etc., добавили теги + reqParam = paramsStruct.Build().New() + } + + // Обработка тела запроса (POST) + if operation.GetRequestBody() != nil { + if operation.GetRequestBody().GetRequestBody() != nil { + bodyFields := buildDynamicStruct(doc, operation.GetRequestBody().GetRequestBody().GetContent().GetAdditionalProperties()[0].GetValue().GetSchema()) + bodyStruct := reflect.StructOf(bodyFields) + bodyInstance := reflect.New(bodyStruct).Interface() + // Получили тело запроса + reqBody = bodyInstance + } + if operation.GetRequestBody().GetReference() != nil { + schemOrRef := findReference(doc, operation.GetRequestBody().GetReference().GetXRef()) + bodyFields := buildDynamicStruct(doc, schemOrRef) + bodyStruct := reflect.StructOf(bodyFields) + bodyInstance := reflect.New(bodyStruct).Interface() + // Получили тело запроса + reqBody = bodyInstance + } + } + + // Обработка ответов + for _, rspOrRef := range operation.Responses.ResponseOrReference { + if rspOrRef.GetValue().GetResponse() != nil { + for _, prop := range rspOrRef.Value.GetResponse().GetContent().GetAdditionalProperties() { + responseFields := buildDynamicStruct(doc, prop.GetValue().GetSchema()) + responseStruct := reflect.StructOf(responseFields) + responseInstance := reflect.New(responseStruct).Interface() + // Получили структуру ответа + rsp = responseInstance + } + } + if rspOrRef.GetValue().GetReference() != nil { + schemaOrRef := findReference(doc, rspOrRef.GetValue().GetReference().GetXRef()) + responseFields := buildDynamicStruct(doc, schemaOrRef) + responseStruct := reflect.StructOf(responseFields) + responseInstance := reflect.New(responseStruct).Interface() + // Получили структуру ответа + rsp = responseInstance + } + } + + return +} + +// Рекурсивное создание структуры из схемы с учетом $ref +func buildDynamicStruct(doc *openapi.Document, schemaOrRef *openapi.SchemaOrReference) []reflect.StructField { + var sfields []reflect.StructField + var schema *openapi.Schema + if schemaOrRef.GetSchema() != nil { + schema = schemaOrRef.GetSchema() + } + if schemaOrRef.GetReference() != nil { + name := strings.Split(schemaOrRef.GetReference().GetXRef(), "#/components/schemas/")[1] + fieldName := capitalize(name) + subBuilder := buildDynamicStruct(doc, findReference(doc, schemaOrRef.GetReference().GetXRef())) + sfield := reflect.StructField{ + Name: fieldName, + Type: reflect.StructOf(subBuilder), + Tag: reflect.StructTag(fmt.Sprintf(`json:"%s"`, name)), + } + sfields = append(sfields, sfield) + } + + for _, prop := range schema.GetProperties().GetAdditionalProperties() { + fieldName := capitalize(prop.GetName()) + if prop.GetValue().GetReference() != nil { + subBuilder := buildDynamicStruct(doc, prop.GetValue()) + + sfield := reflect.StructField{ + Name: fieldName, + Type: reflect.StructOf(subBuilder), + Tag: reflect.StructTag(fmt.Sprintf(`json:"%s"`, prop.GetName())), + } + + sfields = append(sfields, sfield) + } + if prop.GetValue().GetSchema() != nil { + sfield := reflect.StructField{ + Name: fieldName, + Type: reflect.TypeOf(getGoType(doc, prop.GetValue())), + Tag: reflect.StructTag(fmt.Sprintf(`json:"%s"`, prop.GetName())), + } + + sfields = append(sfields, sfield) + } + } + + return sfields +} + +func findReference(doc *openapi.Document, ref string) *openapi.SchemaOrReference { + var result *openapi.SchemaOrReference + ref = strings.Split(ref, "#/components/schemas/")[1] + + for _, prop := range doc.Components.Schemas.GetAdditionalProperties() { + if prop.Name == ref { + result = prop.Value + } + } + + return result +} + +// Преобразование типа OpenAPI в тип Go +func getGoType(doc *openapi.Document, schema *openapi.SchemaOrReference) interface{} { + switch schema.GetSchema().Type { + case "string": + return "" + case "integer": + return 0 + case "boolean": + return false + case "array": + return []interface{}{} + case "object": + return buildDynamicStruct(doc, schema) + default: + return nil + } +} + +func capitalize(fieldName string) string { + if fieldName == "" { + return fieldName + } + // Заменяем точки на подчеркивания для унификации + fieldName = strings.ReplaceAll(fieldName, ".", "_") + + // Разделяем строку по подчеркиваниям + parts := strings.Split(fieldName, "_") + if len(parts) == 1 { + return strings.ToUpper(fieldName[:1]) + fieldName[1:] + } + + // Обрабатываем каждый фрагмент + for i := 0; i < len(parts); i++ { + // Капитализируем первые буквы всех частей, кроме первой + parts[i] = strings.Title(parts[i]) //cases.Title(language.English).String(parts[i]) + } + + // Собираем строку обратно, соединяя части без подчеркиваний + return strings.Join(parts, "") +}