interx/manager/gateway/gateway.go at 5dbf1cd72ef10b5e7b0a3a3348c0e3185b69cc91 · KiraCore/interx · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
package gateway

import (
	"bytes"
	"context"
	"encoding/json"
	"errors"
	"github.com/KiraCore/sai-interx-manager/logger"
	"github.com/KiraCore/sai-service/service"
	"go.uber.org/zap"
	"io"
	"net/http"
	"time"
)

type BaseGateway struct {
	context   *service.Context
	client    *http.Client
	rateLimit *RateLimiter
	retry     *Retrier
}

func NewBaseGateway(ctx *service.Context, retryAttempts int, retryDelay time.Duration, rateLimit int) *BaseGateway {
	return &BaseGateway{
		context: ctx,
		client: &http.Client{
			Timeout: time.Second * 30,
		},
		rateLimit: NewRateLimiter(rateLimit),
		retry:     NewRetrier(retryAttempts, retryDelay),
	}
}

func (g *BaseGateway) makeSaiRequest(ctx context.Context, url string, payload interface{}) (interface{}, error) {
	jsonData, err := json.Marshal(payload)
	if err != nil {
		logger.Logger.Error("BaseGateway - makeRequest", zap.Error(err))
		return nil, err
	}

	req, err := http.NewRequestWithContext(ctx, "POST", url, bytes.NewBuffer(jsonData))
	if err != nil {
		logger.Logger.Error("BaseGateway - makeRequest", zap.Error(err))
		return nil, err
	}
	req.Header.Set("Content-Type", "application/json")

	resp, err := g.client.Do(req)
	if err != nil {
		logger.Logger.Error("BaseGateway - makeRequest", zap.Error(err))
		return nil, err
	}
	defer resp.Body.Close()

	body, err := io.ReadAll(resp.Body)
	if err != nil {
		logger.Logger.Error("BaseGateway - makeRequest", zap.Error(err))
		return nil, err
	}

	if resp.StatusCode != http.StatusOK {
		err = errors.New("non-200 status code: " + resp.Status)
		logger.Logger.Error("BaseGateway - makeRequest", zap.Error(err))
		return nil, err
	}

	var result interface{}
	if err := json.Unmarshal(body, &result); err != nil {
		logger.Logger.Error("BaseGateway - makeRequest", zap.Error(err))
		return nil, err
	}

	return result, nil
}

func (g *BaseGateway) makeRequest(ctx context.Context, method, url string, payload interface{}) (interface{}, error) {
	payloadData, err := json.Marshal(payload)
	if err != nil {
		logger.Logger.Error("BaseGateway - makeRequest", zap.Error(err))
		return nil, err
	}

	req, err := http.NewRequestWithContext(ctx, method, url, bytes.NewBuffer(payloadData))
	if err != nil {
		logger.Logger.Error("BaseGateway - makeRequest", zap.Error(err))
		return nil, err
	}

	resp, err := g.client.Do(req)
	if err != nil {
		logger.Logger.Error("BaseGateway - makeRequest", zap.Error(err))
		return nil, err
	}
	defer resp.Body.Close()

	body, err := io.ReadAll(resp.Body)
	if err != nil {
		logger.Logger.Error("BaseGateway - makeRequest", zap.Error(err))
		return nil, err
	}

	if resp.StatusCode != http.StatusOK {
		err = errors.New("non-200 status code: " + resp.Status)
		logger.Logger.Error("BaseGateway - makeRequest", zap.Error(err))
		return nil, err
	}

	var result interface{}
	if err := json.Unmarshal(body, &result); err != nil {
		logger.Logger.Error("BaseGateway - makeRequest", zap.Error(err))
		return nil, err
	}

	return result, nil
}