80 lines
1.7 KiB
Go
80 lines
1.7 KiB
Go
|
package swaggerset
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
"sync"
|
||
|
|
|
||
|
|
"github.com/getkin/kin-openapi/openapi3"
|
||
|
|
)
|
||
|
|
|
||
|
|
var errNotFound = errors.New("file descriptor not found")
|
||
|
|
|
||
|
|
type SwaggerSet struct {
|
||
|
|
mu sync.Mutex
|
||
|
|
files map[string]*openapi3.T
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewSwaggerSet() *SwaggerSet {
|
||
|
|
return &SwaggerSet{
|
||
|
|
mu: sync.Mutex{},
|
||
|
|
files: make(map[string]*openapi3.T, 0),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *SwaggerSet) GetMessage(addr, svc, mth, typereq string) (*messages, error) {
|
||
|
|
if svc == "" || mth == "" || addr == "" || typereq == "" {
|
||
|
|
return nil, errors.New("addr or service name is empty")
|
||
|
|
}
|
||
|
|
|
||
|
|
messages := newMessages()
|
||
|
|
|
||
|
|
p.mu.Lock()
|
||
|
|
doc := p.files[addr+"|"+svc]
|
||
|
|
p.mu.Unlock()
|
||
|
|
|
||
|
|
pathItem := doc.Paths.Value(mth)
|
||
|
|
if pathItem.Get != nil {
|
||
|
|
reqParam, reqBody, rsp := handleOperation("GET", pathItem.Get)
|
||
|
|
messages.Msgs = append(messages.Msgs, message{
|
||
|
|
Type: "GET",
|
||
|
|
RequestParam: reqParam,
|
||
|
|
RequestBody: reqBody,
|
||
|
|
Response: rsp,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
if pathItem.Post != nil {
|
||
|
|
reqParam, reqBody, rsp := handleOperation("POST", pathItem.Post)
|
||
|
|
messages.Msgs = append(messages.Msgs, message{
|
||
|
|
Type: "POST",
|
||
|
|
RequestParam: reqParam,
|
||
|
|
RequestBody: reqBody,
|
||
|
|
Response: rsp,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
return messages, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *SwaggerSet) AddSwaggerset(addr, svc string, data []byte) error {
|
||
|
|
ctx := context.Background()
|
||
|
|
|
||
|
|
loader := openapi3.NewLoader()
|
||
|
|
loader.IsExternalRefsAllowed = true
|
||
|
|
|
||
|
|
doc, err := loader.LoadFromData(data)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("failed to load data from buf: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err = doc.Validate(ctx); err != nil {
|
||
|
|
return fmt.Errorf("failed to validate data from swagger: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
p.mu.Lock()
|
||
|
|
p.files[addr+"|"+svc] = doc
|
||
|
|
p.mu.Unlock()
|
||
|
|
return nil
|
||
|
|
}
|