#2 - add swaggerset

pkg/swaggerset/swaggerset.go

@@ -0,0 +1,79 @@
+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
+}