feat: stress for helm testing

.github/workflows/publish.yaml

@@ -15,4 +15,4 @@ jobs:
           go-version: '1.21'
       - uses: actions/checkout@v3
       - uses: ko-build/setup-ko@v0.6
-      - run: ko build --platform=linux/amd64 --bare ./cmd/...
+      - run: ko build --platform=linux/amd64 --bare ./cmd/erida/...

Makefile

@@ -3,8 +3,13 @@ ifneq (,$(wildcard ./.env))
 	export
 endif
 
+include helm/Makefile
+
 run:
-	@go run ./cmd/main.go
+	@go run ./cmd/erida/main.go
+
+stress:
+	@go run ./cmd/stress/main.go
 
 integration:
 	@go test -v ./... -tags=integration

README.md

@@ -24,7 +24,7 @@ Configuring Erida is a breeze. All you need to do is set the following environme
 - `SMTP_HOST`: SMTP server host
 - `SMTP_PORT`: SMTP server port
 - `SMTP_USER`: SMTP server username
-- `SMTP_PASSWORD`: SMTP server password
+- `SMTP_PASS`: SMTP server password
 - `SLACK_TOKEN`: Slack bot token
 - `SMTP_TLS` (Optional, default: true): Enable or disable Start TLS usage
 

cmd/stress/main.go

@@ -0,0 +1,61 @@
+package main
+
+import (
+	"fmt"
+	"github.com/emersion/go-smtp"
+	"log/slog"
+	"os"
+	"strings"
+	"sync"
+	"time"
+)
+
+var (
+	eridaAddr   = os.Getenv("ERIDA_ADDR")
+	from        = os.Getenv("ERIDA_FROM")
+	to          = strings.Split(os.Getenv("ERIDA_TO"), ",")
+	bodyPattern = os.Getenv("ERIDA_BODY_PATTERN")
+)
+
+// following code is a stress test for erida
+// it will send 10 messages to erida server
+//
+// they will be launched in the same cluster
+func main() {
+	time.Sleep(5 * time.Second)
+
+	var wg sync.WaitGroup
+	for i := 0; i < 10; i++ {
+		wg.Add(1)
+
+		go func(i int) {
+			defer wg.Done()
+
+			issueRequest(i)
+		}(i)
+	}
+
+	wg.Wait()
+}
+
+func issueRequest(reqNumber int) {
+	erida, err := smtp.Dial(eridaAddr)
+	if err != nil {
+		slog.Error("failed to dial erida server", "error", err)
+		return
+	}
+
+	defer func() {
+		if err = erida.Quit(); err != nil {
+			slog.Error("failed to quit erida server", "error", err)
+		}
+	}()
+
+	var body = strings.NewReader(fmt.Sprintf(bodyPattern, reqNumber))
+	if err = erida.SendMail(from, to, body); err != nil {
+		slog.Error("failed to send email", "error", err)
+		return
+	}
+
+	slog.Info("email sent", "reqNumber", reqNumber)
+}

internal/config.go

@@ -4,7 +4,7 @@ import "fmt"
 
 type Config struct {
 	RelayHost  string `env:"SERVER_HOST" env-default:"localhost"`
-	RelayPort  string `env:"SERVER_PORT" env-default:"1025"`
+	RelayPort  string `env:"SERVER_PORT" env-default:"25"`
 	Host       string `env:"SMTP_HOST" env-default:"smtp.gmail.com"`
 	Port       string `env:"SMTP_PORT" env-default:"587"`
 	User       string `env:"SMTP_USER"`