Fix/microgateway postgres blob schema (#381)

Author: MohammedELKheir

microgateway/internal/database/connection_test.go

@@ -2,11 +2,14 @@
 package database
 
 import (
+	"sync"
 	"testing"
 	"time"
 
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
+	"gorm.io/driver/postgres"
+	"gorm.io/gorm/schema"
 )
 
 func TestConnect_SQLite(t *testing.T) {
@@ -37,7 +40,7 @@ func TestConnect_SQLite(t *testing.T) {
 	t.Run("InvalidDSN", func(t *testing.T) {
 		invalidConfig := config
 		invalidConfig.DSN = "/invalid/path/to/db.sqlite"
-		
+
 		_, err := Connect(invalidConfig)
 		assert.Error(t, err)
 	})
@@ -54,7 +57,27 @@ func TestConnect_UnsupportedDatabase(t *testing.T) {
 	assert.Contains(t, err.Error(), "unsupported database type")
 }
 
-// TestMigrate removed - GORM handles migrations automatically
+func TestControlPayload_PostgresPayloadColumnType(t *testing.T) {
+	modelSchema, err := schema.Parse(&ControlPayload{}, &sync.Map{}, schema.NamingStrategy{})
+	require.NoError(t, err)
+
+	payloadField := modelSchema.LookUpField("Payload")
+	require.NotNil(t, payloadField)
+
+	columnType := postgres.Dialector{}.DataTypeOf(payloadField)
+	assert.Equal(t, "bytea", columnType)
+}
+
+func TestAnalyticsEvent_PostgresInteractionTypeColumnType(t *testing.T) {
+	modelSchema, err := schema.Parse(&AnalyticsEvent{}, &sync.Map{}, schema.NamingStrategy{})
+	require.NoError(t, err)
+
+	interactionTypeField := modelSchema.LookUpField("InteractionType")
+	require.NotNil(t, interactionTypeField)
+
+	columnType := postgres.Dialector{}.DataTypeOf(interactionTypeField)
+	assert.Equal(t, "text", columnType)
+}
 
 func TestDatabaseConfig_Validation(t *testing.T) {
 	tests := []struct {
@@ -90,7 +113,7 @@ func TestDatabaseConfig_Validation(t *testing.T) {
 		t.Run(tt.name, func(t *testing.T) {
 			// Test connection (may fail due to actual DB not available, but config should be valid)
 			_, err := Connect(tt.config)
-			
+
 			if tt.hasError {
 				assert.Error(t, err)
 			} else {
@@ -124,4 +147,4 @@ func TestGormLogger(t *testing.T) {
 			// but we can verify the function doesn't panic
 		})
 	}
-}
+}

microgateway/internal/database/models.go

@@ -174,7 +174,7 @@ type AnalyticsEvent struct {
 	UserID              uint   `gorm:"index:idx_analytics_user"`      // User who made the request
 	Name                string                                        // Model name (e.g., "gpt-4", "claude-3-opus")
 	Vendor              string                                        // LLM vendor (e.g., "openai", "anthropic")
-	InteractionType     string `gorm:"type:string;default:'proxy'"`  // "chat" or "proxy"
+	InteractionType     string `gorm:"type:text;default:'proxy'"`    // "chat" or "proxy"
 	Choices             int                                           // Number of choices in response
 	ToolCalls           int                                           // Number of tool calls made
 	ChatID              string                                        // Chat session identifier
@@ -383,7 +383,7 @@ func (pkv *PluginKV) IsExpired() bool {
 type ControlPayload struct {
 	ID            uint           `gorm:"primaryKey"`
 	PluginID      uint           `gorm:"not null;index:idx_control_payload_plugin"`
-	Payload       []byte         `gorm:"type:blob;not null"`
+	Payload       []byte         `gorm:"not null"`
 	CorrelationID string         `gorm:"size:255;index:idx_control_payload_correlation"`
 	Metadata      datatypes.JSON `gorm:"type:json"`
 	Sent          bool           `gorm:"default:false;index:idx_control_payload_sent"`
@@ -560,4 +560,4 @@ func (OAuthClientEdge) TableName() string   { return "oauth_clients" }
 func (AccessTokenEdge) TableName() string   { return "access_tokens" }
 func (AppTool) TableName() string           { return "app_tools" }
 func (AppDatasource) TableName() string     { return "app_datasources" }
-func (ToolFilter) TableName() string        { return "tool_filters" }
+func (ToolFilter) TableName() string        { return "tool_filters" }