feat(ws-probe): тестовый WS echo-сервер для проверки wss на платформе Nubes (v0.1.0, образ naeel/iot-ws-probe)

2026-08-16 08:18:18 +04:00
parent 2a7ecd288e
commit 27cb40e421
6 changed files with 161 additions and 0 deletions
@@ -4,6 +4,7 @@
 /kafka-consumer
 /sqs-consumer
 bin/
 ws-probe/ws-probe
 # Go
 *.exe
@@ -0,0 +1,17 @@
 # ws-probe — проверка WebSocket через платформенный ingress Nubes.
 # Образ: naeel/iot-ws-probe (Docker Hub, публичный — платформа тянет только публичные).
 FROM golang:1.25 AS builder
 WORKDIR /workspace
 COPY go.mod go.sum ./
 RUN go mod download
 COPY main.go ./
 ARG VERSION=dev
 RUN CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -a -ldflags "-X main.version=${VERSION}" -o ws-probe .
 FROM gcr.io/distroless/static:nonroot
 WORKDIR /
 COPY --from=builder /workspace/ws-probe .
 USER 65532:65532
 EXPOSE 8080
 ENTRYPOINT ["/ws-probe"]
@@ -0,0 +1,19 @@
 # ws-probe — тестовый WS echo-сервер для проверки wss на платформе Nubes.
 VERSION ?= v0.1.0
 IMAGE ?= naeel/iot-ws-probe
 .PHONY: build tidy docker-build docker-push
 build:
 	CGO_ENABLED=0 go build -ldflags "-X main.version=$(VERSION)" -o ws-probe .
 tidy:
 	go mod tidy
 docker-build:
 	docker build --build-arg VERSION=$(VERSION) -t $(IMAGE):$(VERSION) -t $(IMAGE):latest .
 docker-push:
 	docker push $(IMAGE):$(VERSION)
 	docker push $(IMAGE):latest
@@ -0,0 +1,5 @@
 module ws-probe
 go 1.25
 require github.com/gorilla/websocket v1.5.3
@@ -0,0 +1,2 @@
 github.com/gorilla/websocket v1.5.3 h1:saDtZ6Pbx/0u+bgYQ3q96pZgCzfhKXGPqt7kZ72aNNg=
 github.com/gorilla/websocket v1.5.3/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
@@ -0,0 +1,117 @@
 // ws-probe — тестовый контейнер для проверки WebSocket (wss) через платформенный ingress Nubes.
 //
 // Назначение: подтвердить, что «Простой HTTP контейнер» платформы пропускает
 // WebSocket upgrade (101 Switching Protocols) и держит долгоживущие соединения.
 //
 // Эндпоинты:
 //
 //	GET /        — инфо-страница (HTML)
 //	GET /health  — {"status":"ok","version":...}
 //	GET /ws      — WebSocket echo: отвечает на сообщения, шлёт ping каждые 20с
 //
 // Без кредов, без внешних зависимостей — только echo.
 package main
 import (
 	"encoding/json"
 	"log"
 	"net/http"
 	"os"
 	"time"
 	"github.com/gorilla/websocket"
 )
 var version = "dev"
 const pageHTML = `<!DOCTYPE html>
 <html><head><meta charset="utf-8"><title>ws-probe</title></head>
 <body style="font-family:system-ui,sans-serif;padding:2rem">
 <h1>ws-probe</h1>
 <p>WebSocket echo: <code>/ws</code></p>
 <p>Health: <code>/health</code></p>
 </body></html>`
 var upgrader = websocket.Upgrader{
 	ReadBufferSize:  1024,
 	WriteBufferSize: 1024,
 	CheckOrigin:     func(r *http.Request) bool { return true },
 }
 func main() {
 	port := os.Getenv("PORT")
 	if port == "" {
 		port = "8080"
 	}
 	mux := http.NewServeMux()
 	mux.HandleFunc("/health", func(w http.ResponseWriter, r *http.Request) {
 		w.Header().Set("Content-Type", "application/json")
 		_ = json.NewEncoder(w).Encode(map[string]string{"status": "ok", "version": version})
 	})
 	mux.HandleFunc("/ws", wsEcho)
 	mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
 		w.Header().Set("Content-Type", "text/html; charset=utf-8")
 		_, _ = w.Write([]byte(pageHTML))
 	})
 	log.Printf("ws-probe %s listening on :%s", version, port)
 	if err := http.ListenAndServe(":"+port, mux); err != nil {
 		log.Fatal(err)
 	}
 }
 // wsEcho — апгрейд до WebSocket, echo сообщений, ping каждые 20с.
 func wsEcho(w http.ResponseWriter, r *http.Request) {
 	conn, err := upgrader.Upgrade(w, r, nil)
 	if err != nil {
 		log.Printf("upgrade failed: %v", err)
 		return
 	}
 	defer conn.Close()
 	log.Printf("ws connected from %s (host=%s)", conn.RemoteAddr(), r.Host)
 	// Read deadline сбрасывается pong'ами — мёртвые соединения закрываются через 60с.
 	conn.SetReadDeadline(time.Now().Add(60 * time.Second))
 	conn.SetPongHandler(func(string) error {
 		conn.SetReadDeadline(time.Now().Add(60 * time.Second))
 		return nil
 	})
 	pingTicker := time.NewTicker(20 * time.Second)
 	defer pingTicker.Stop()
 	done := make(chan struct{})
 	go func() {
 		defer close(done)
 		for {
 			mt, msg, err := conn.ReadMessage()
 			if err != nil {
 				log.Printf("ws read error: %v", err)
 				return
 			}
 			log.Printf("ws recv %d bytes (type %d)", len(msg), mt)
 			if err := conn.WriteMessage(mt, msg); err != nil {
 				log.Printf("ws write error: %v", err)
 				return
 			}
 			log.Printf("ws echo sent")
 		}
 	}()
 	for {
 		select {
 		case <-done:
 			log.Printf("ws closed from %s", conn.RemoteAddr())
 			return
 		case t := <-pingTicker.C:
 			deadline := time.Now().Add(10 * time.Second)
 			if err := conn.WriteControl(websocket.PingMessage, []byte(t.Format(time.RFC3339)), deadline); err != nil {
 				log.Printf("ws ping error: %v", err)
 				return
 			}
 		}
 	}
 }
		`@@ -0,0 +1,2 @@`
							`github.com/gorilla/websocket v1.5.3 h1:saDtZ6Pbx/0u+bgYQ3q96pZgCzfhKXGPqt7kZ72aNNg=`
							`github.com/gorilla/websocket v1.5.3/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=`