*

cmd/main.go

@@ -3,39 +3,44 @@ package main
 import (
 	"context"
 	"log"
+	"net"
 	"os"
 	"os/signal"
+	"strconv"
+	"syscall"
+	server "test3k/authPostman/internal"
 	"test3k/authPostman/internal/config"
 
 	"github.com/segmentio/kafka-go"
 )
 
 func main() {
-	r := kafka.NewReader(kafka.ReaderConfig{
-		Topic:     "registrations",
-		Brokers:   []string{"localhost:9092"},
-		GroupID:   "consumer-group-id",
-		Partition: 0,
-		MinBytes:  10e3, // 10KB
-		MaxBytes:  10e6, // 10MB
-	})
-	defer r.Close()
+	config := config.NewConfig()
+	ctx, _ := context.WithCancel(context.Background())
+	srv := server.NewServer(ctx, config)
 
 	//
-	// r.SetOffset(0)
+	signalChannel := make(chan os.Signal, 1)
+	signal.Notify(signalChannel, syscall.SIGINT)
+	signal.Notify(signalChannel, syscall.SIGTERM)
+	defer stop(signalChannel, srv)
+
+	// Запуск сервера
+	go start(config, srv)
 
+	//
 	for {
-		m, err := r.ReadMessage(context.Background())
-		if err != nil {
-			break
+		select {
+		case <-signalChannel:
+			return
+		case <-ctx.Done():
+			return
 		}
-
-		log.Printf("message at offset %d: %s = %s\n", m.Offset, string(m.Key), string(m.Value))
 	}
 }
 
 // Остановка сервера
-func stop(signalChannel chan os.Signal, srv *server.AuthDBServer) {
+func stop(signalChannel chan os.Signal, srv *server.AuthPostmanServer) {
 	defer srv.GracefulStop()
 	defer signal.Stop(signalChannel)
 
@@ -43,9 +48,32 @@ func stop(signalChannel chan os.Signal, srv *server.AuthDBServer) {
 }
 
 // Запуск сервера
-func start(config *config.Config, srv *server.AuthDBServer) {
-	// connStr := net.JoinHostPort("", strconv.Itoa(config.App.Port))
+func start(config *config.Config, srv *server.AuthPostmanServer) {
+	connStr := net.JoinHostPort(config.Kafka.Host, strconv.Itoa(config.Kafka.Port))
+
+	//
+	log.Printf("authPostmanServer starting (listening to %s)\n", connStr)
 
 	//
-	log.Printf("authPostmanServer starting (%s)\n", connStr)
+	r := kafka.NewReader(kafka.ReaderConfig{
+		Topic:     "registrations",
+		Brokers:   []string{connStr},
+		GroupID:   "consumer-group-id",
+		Partition: 0,
+		MinBytes:  10e3, // 10KB
+		MaxBytes:  10e6, // 10MB
+	})
+	defer r.Close()
+
+	// ...
+	r.SetOffset(0)
+
+	for {
+		m, err := r.ReadMessage(context.Background())
+		if err != nil {
+			break
+		}
+
+		log.Printf("message at offset %d: %s = %s\n", m.Offset, string(m.Key), string(m.Value))
+	}
 }

internal/postman.go

@@ -0,0 +1,18 @@
+package postman
+
+import (
+	"context"
+	"test3k/authPostman/internal/config"
+)
+
+type AuthPostmanServer struct {
+}
+
+func NewServer(ctx context.Context, config *config.Config) *AuthPostmanServer {
+	return &AuthPostmanServer{}
+}
+
+func (s *AuthPostmanServer) GracefulStop() error {
+	// return s.db.Close()
+	return nil
+}