first commit

cmd/main.go

@@ -0,0 +1,122 @@
+package main
+
+import (
+	"context"
+	"database/sql"
+	"fmt"
+	"log"
+	"os"
+
+	"go-watermill-template/internal/config"
+	"go-watermill-template/internal/db"
+	"go-watermill-template/internal/pubsub"
+
+	"github.com/ThreeDotsLabs/watermill"
+	"github.com/ThreeDotsLabs/watermill/message"
+)
+
+// -------------------------------------------------------------
+// Main Entry
+// -------------------------------------------------------------
+func main() {
+	cfg := config.Load()
+	logger := watermill.NewStdLogger(cfg.LogDebug, cfg.LogDebug)
+
+	// Connect ke Postgres
+	conn, err := db.Connect(cfg.DBHost, cfg.DBPort, cfg.DBUser, cfg.DBPassword, cfg.DBName, cfg.DBSSLMode)
+	config.Must(err)
+	defer conn.Close()
+
+	args := os.Args
+	if len(args) > 1 {
+		switch args[1] {
+		case "create-tables":
+			// buat semua topik dari .env
+			for _, t := range cfg.Topics {
+				pubsub.CreateTopicTables(conn, t)
+			}
+			fmt.Println("✅ All tables created successfully from .env")
+			return
+
+		case "create-table":
+			if len(args) < 3 {
+				fmt.Println("❌ Usage: go run cmd/main.go create-table <topic_name>")
+				return
+			}
+			topic := args[2]
+			pubsub.CreateTopicTables(conn, topic)
+			fmt.Printf("✅ Table created for topic: %s\n", topic)
+			return
+		}
+	}
+
+	// kalau tidak ada argumen, jalankan service utama
+	runService(cfg, conn, logger)
+}
+
+// -------------------------------------------------------------
+// Watermill Runtime Service
+// -------------------------------------------------------------
+func runService(cfg *config.Config, conn *sql.DB, logger watermill.LoggerAdapter) {
+	log.Println("🚀 Starting Watermill PostgreSQL Service...")
+
+	// Inisialisasi Publisher dan Subscriber
+	publisher, err := pubsub.NewPublisher(conn, logger)
+	config.Must(err)
+
+	subscriber, err := pubsub.NewSubscriber(conn, logger)
+	config.Must(err)
+
+	log.Println("✅ Watermill PostgreSQL Pub/Sub initialized")
+
+	// Pastikan semua tabel ada
+	for _, t := range cfg.Topics {
+		pubsub.CreateTopicTables(conn, t)
+	}
+
+	// Jalankan subscriber untuk semua topik
+	for _, topic := range cfg.Topics {
+		go startDynamicSubscriber(subscriber, topic)
+	}
+
+	// Kirim contoh pesan ke topik pertama
+	if len(cfg.Topics) > 0 {
+		sendExampleMessage(publisher, cfg.Topics[0])
+	}
+
+	select {} // block forever
+}
+
+// -------------------------------------------------------------
+// Contoh Publisher
+// -------------------------------------------------------------
+func sendExampleMessage(publisher message.Publisher, topic string) {
+	msg := message.NewMessage(
+		watermill.NewUUID(),
+		[]byte(fmt.Sprintf(`{"hello": "from topic %s"}`, topic)),
+	)
+
+	if err := publisher.Publish(topic, msg); err != nil {
+		log.Printf("❌ Failed to publish to %s: %v", topic, err)
+		return
+	}
+
+	log.Printf("📤 Published message to %s", topic)
+}
+
+// -------------------------------------------------------------
+// Contoh Subscriber
+// -------------------------------------------------------------
+func startDynamicSubscriber(sub message.Subscriber, topic string) {
+	messages, err := sub.Subscribe(context.Background(), topic)
+	if err != nil {
+		panic(err)
+	}
+
+	go func() {
+		for msg := range messages {
+			log.Printf("📥 [%s] Received: %s", topic, string(msg.Payload))
+			msg.Ack()
+		}
+	}()
+}