Merge branch '260-reset-connection' into 'master'

agneum · agneum · commit e6c4bc480a78 · 2023-09-19T02:26:57.000Z
fix: re-init connection on reset (platform#260)

See merge request postgres-ai/joe!178
diff --git a/pkg/bot/command/reset_session.go b/pkg/bot/command/reset_session.go
@@ -50,6 +50,15 @@ func ResetSession(ctx context.Context, cmd *platform.Command, msg *models.Messag
 		idleConnection.Release()
 	}
 
+	cloneConn, err := session.Pool.Acquire(ctx)
+	if err != nil {
+		log.Err("failed to acquire database connection:", err)
+	}
+
+	if cloneConn != nil {
+		session.CloneConnection = cloneConn.Conn()
+	}
+
 	fwData := &foreword.Content{
 		SessionID:  clone.ID,
 		Duration:   time.Duration(clone.Metadata.MaxIdleMinutes) * time.Minute,
