healthchecks

.gitignore

@@ -1,2 +1,3 @@
 go-exchange-api
-redis
+redis
+!cmd/*

Dockerfile

@@ -3,13 +3,16 @@ WORKDIR /src/app
 COPY go.mod go.sum ./
 RUN go mod download
 COPY . ./
-RUN go build -ldflags "-s -w" -o go-exchange-api
+RUN go build -ldflags "-s -w" -o go-exchange-api cmd/go-exchange-api/main.go
+RUN go build -ldflags "-s -w" -o healthcheck cmd/healthcheck/main.go
 
 FROM alpine
 ENV GIN_MODE release
 WORKDIR /app
 COPY --from=builder /src/app/go-exchange-api ./go-exchange-api
+COPY --from=builder /src/app/healthcheck ./healthcheck
 RUN addgroup -S appgroup && \
     adduser -S appuser -G appgroup
 USER appuser
+HEALTHCHECK --interval=30s --timeout=10s --retries=3 CMD ["./healthcheck"]
 ENTRYPOINT ["./go-exchange-api"]

api/moex.go

@@ -6,6 +6,7 @@ import (
 	"log"
 	"time"
 
+	"github.com/kiberdruzhinnik/go-exchange-api/constants"
 	custom_errors "github.com/kiberdruzhinnik/go-exchange-api/errors"
 	"github.com/kiberdruzhinnik/go-exchange-api/utils"
 )
@@ -50,7 +51,7 @@ type MoexPriceJSON struct {
 
 func NewMoexAPI(redis utils.RedisClient) MoexAPI {
 	return MoexAPI{
-		BaseURL: "https://iss.moex.com",
+		BaseURL: constants.MoexBaseApiURL,
 		Redis:   redis,
 	}
 }

api/spbex.go

@@ -6,6 +6,7 @@ import (
 	"log"
 	"time"
 
+	"github.com/kiberdruzhinnik/go-exchange-api/constants"
 	custom_errors "github.com/kiberdruzhinnik/go-exchange-api/errors"
 	"github.com/kiberdruzhinnik/go-exchange-api/utils"
 )
@@ -30,7 +31,7 @@ type SpbexSecurityJSON struct {
 
 func NewSpbexAPI() SpbexAPI {
 	return SpbexAPI{
-		BaseURL: "https://investcab.ru/api",
+		BaseURL: constants.SpbexBaseApiURL,
 	}
 }
 

cmd/healthcheck/main.go

@@ -0,0 +1,28 @@
+package main
+
+import (
+	"io"
+	"log"
+	"net/http"
+)
+
+func main() {
+	resp, err := http.Get("http://127.0.0.1:8080/healthcheck")
+
+	if err != nil {
+		log.Fatalln(err)
+	}
+
+	defer resp.Body.Close()
+
+	body, err := io.ReadAll(resp.Body)
+
+	if err != nil {
+		log.Fatalln(err)
+	}
+
+	status := string(body)
+	if status != "ok" {
+		log.Fatalln("status is not ok")
+	}
+}

constants/constants.go

@@ -0,0 +1,4 @@
+package constants
+
+const MoexBaseApiURL = "https://iss.moex.com"
+const SpbexBaseApiURL = "https://investcab.ru/api"

docker-compose.yml

@@ -5,6 +5,12 @@ services:
     image: redis:7-alpine
     volumes:
       - redis:/data
+    healthcheck:
+      test: [ "CMD-SHELL", "redis-cli ping | grep PONG" ]
+      interval: 30s
+      timeout: 10s
+      retries: 3
+      start_period: 30s
 
   exchange-api:
     image: kotasha/go-exchange-api:latest
@@ -14,6 +20,12 @@ services:
       - 8080:8080/tcp
     depends_on:
       - redis-cache
+    healthcheck:
+      test: [ "./healthcheck" ]
+      interval: 30s
+      timeout: 10s
+      retries: 3
+      start_period: 30s
 
 volumes:
   redis:

errors/errors.go

@@ -9,3 +9,5 @@ var ErrorNoData = errors.New("no data")
 
 var ErrorRedisNotConnected = errors.New("redis is not connected")
 var ErrorRedisNotFound = errors.New("not found in redis")
+
+var ErrorNotAllowed = errors.New("not allowed")

utils/utils.go

@@ -5,9 +5,31 @@ import (
 	"net/http"
 	"strings"
 	"unicode"
+
+	"github.com/kiberdruzhinnik/go-exchange-api/constants"
+	"github.com/kiberdruzhinnik/go-exchange-api/errors"
 )
 
+var URLS_ALLOW_LIST []string = []string{
+	constants.MoexBaseApiURL,
+	constants.SpbexBaseApiURL,
+}
+
+func CheckSafeURL(url string) bool {
+	for _, u := range URLS_ALLOW_LIST {
+		if strings.HasPrefix(url, u) {
+			return true
+		}
+	}
+	return false
+}
+
 func HttpGet(url string) ([]byte, error) {
+
+	if !CheckSafeURL(url) {
+		return nil, errors.ErrorNotAllowed
+	}
+
 	resp, err := http.Get(url)
 
 	if err != nil {