feat: database layer + canvas validator/converter + LLM inference activities

- Add pkg/db models and CRUD methods for workflows - Add internal/routing canvas validator (DAG check, connectivity) - Add internal/routing canvas converter (Canvas → WorkflowSpec) - Register LLMInferenceActivity and LLMBatchInferenceActivity - Update api/server and cmd/server with database integration - Add K8s environment variable support - Update activity knowledge base with LLM activities - Add .env.example configuration template
2026-09-05 00:43:30 -07:00
parent 924aa398b6
commit 0da90fdd7a
13 changed files with 2183 additions and 7 deletions
@@ -0,0 +1,124 @@
+package main
+
+import (
+	"context"
+	"flag"
+	"log"
+	"os"
+	"os/signal"
+	"sync"
+	"syscall"
+
+	"go.temporal.io/sdk/client"
+	"go.temporal.io/sdk/worker"
+
+	"github.com/rockliang/poimen/workflows/action"
+	"github.com/rockliang/poimen/workflows/internal/api"
+	"github.com/rockliang/poimen/workflows/internal/config"
+	"github.com/rockliang/poimen/workflows/pkg/db"
+	"github.com/rockliang/poimen/workflows/statemachine"
+)
+
+func main() {
+	var (
+		apiPort = flag.Int("port", 8080, "HTTP API port")
+		verbose = flag.Bool("verbose", false, "verbose logging")
+	)
+	flag.Parse()
+
+	logger := log.New(os.Stdout, "[poimen-server] ", log.LstdFlags|log.Lshortfile)
+
+	// Load configuration
+	cfg, err := config.LoadConfig()
+	if err != nil {
+		logger.Fatalf("failed to load config: %v", err)
+	}
+
+	// Connect to database (memory-db via K8s CNPG)
+	logger.Println("connecting to database...")
+	database, err := db.New(os.Getenv("DATABASE_URL"))
+	if err != nil {
+		logger.Fatalf("failed to connect to database: %v", err)
+	}
+	defer database.Close()
+	logger.Println("✓ Connected to database")
+
+	// Connect to Temporal
+	logger.Printf("connecting to Temporal at %s", cfg.Temporal.HostPort)
+	c, err := client.Dial(client.Options{
+		HostPort:  cfg.Temporal.HostPort,
+		Namespace: cfg.Temporal.Namespace,
+	})
+	if err != nil {
+		logger.Fatalf("failed to connect to temporal: %v", err)
+	}
+	defer c.Close()
+
+	logger.Println("✓ Connected to Temporal")
+
+	// Create and start Temporal worker
+	w := worker.New(c, "default", worker.Options{})
+
+	// Register RoutingWorkflow
+	w.RegisterWorkflow(statemachine.RoutingWorkflow)
+
+	// Register activities
+	w.RegisterActivity(action.CloneRepoActivity)
+	w.RegisterActivity(action.AnalyzeCodeActivity)
+	w.RegisterActivity(action.SecurityScanActivity)
+	w.RegisterActivity(action.GenerateReportActivity)
+	w.RegisterActivity(action.DeploymentPreCheckActivity)
+	w.RegisterActivity(action.NotifyStatusActivity)
+	w.RegisterActivity(action.ApproveWorkflowActivity)
+	w.RegisterActivity(action.ArchiveResultsActivity)
+	w.RegisterActivity(action.RetrieveMemoryActivity)
+	w.RegisterActivity(action.AssumeRoleActivity)
+	w.RegisterActivity(action.LLMInferenceActivity)
+	w.RegisterActivity(action.LLMBatchInferenceActivity)
+
+	var wg sync.WaitGroup
+	errChan := make(chan error, 2)
+
+	// Start Temporal worker
+	wg.Add(1)
+	go func() {
+		defer wg.Done()
+		logger.Println("starting Temporal worker...")
+		if err := w.Run(worker.InterruptCh()); err != nil {
+			errChan <- err
+		}
+	}()
+
+	// Start HTTP API server
+	wg.Add(1)
+	go func() {
+		defer wg.Done()
+		server := api.NewServer(database, c, logger)
+		logger.Printf("starting API server on port %d", *apiPort)
+		if err := server.Start(*apiPort); err != nil {
+			errChan <- err
+		}
+	}()
+
+	// Wait for interrupt signal
+	sigChan := make(chan os.Signal, 1)
+	signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)
+
+	go func() {
+		sig := <-sigChan
+		logger.Printf("received signal: %v", sig)
+		w.Stop()
+	}()
+
+	// Monitor for errors
+	go func() {
+		err := <-errChan
+		if err != nil {
+			logger.Printf("error: %v", err)
+			w.Stop()
+		}
+	}()
+
+	wg.Wait()
+	logger.Println("✓ Server stopped gracefully")
+}