feat: 增加连接池配置和只读副本支持，更新驱动和数据源架构

Author: hotlong

packages/spec/src/data/datasource.zod.ts

@@ -118,6 +118,24 @@ export const DatasourceSchema = z.object({
    */
   config: z.record(z.any()).describe('Driver specific configuration'),
 
+  /**
+   * Connection Pool Configuration
+   * Standard connection pooling settings.
+   */
+  pool: z.object({
+    min: z.number().default(0).describe('Minimum connections'),
+    max: z.number().default(10).describe('Maximum connections'),
+    idleTimeoutMillis: z.number().default(30000).describe('Idle timeout'),
+    connectionTimeoutMillis: z.number().default(3000).describe('Connection establishment timeout'),
+  }).optional().describe('Connection pool settings'),
+
+  /**
+   * Read Replicas
+   * Optional list of duplicate configurations for read-only operations.
+   * Useful for scaling read throughput.
+   */
+  readReplicas: z.array(z.record(z.any())).optional().describe('Read-only replica configurations'),
+
   /**
    * Capability Overrides
    * Manually override what the driver claims to support.

packages/spec/src/data/driver.zod.ts

@@ -21,6 +21,18 @@ export const DriverOptionsSchema = z.object({
    * Whether to bypass cache and force a fresh read.
    */
   skipCache: z.boolean().optional().describe('Bypass cache'),
+
+  /**
+   * Distributed Tracing Context.
+   * Used for passing OpenTelemetry span context or request IDs for observability.
+   */
+  traceContext: z.record(z.string()).optional().describe('OpenTelemetry context or request ID'),
+
+  /**
+   * Tenant Identifier.
+   * For multi-tenant databases (row-level security or schema-per-tenant).
+   */
+  tenantId: z.string().optional().describe('Tenant Isolation identifier'),
 });
 
 /**
@@ -109,6 +121,17 @@ export const DriverCapabilitiesSchema = z.object({
    * If false, arrays will be stored as JSON strings or in separate tables.
    */
   arrayFields: z.boolean().describe('Supports array field types'),
+
+  /**
+   * Whether the driver supports vector embeddings and similarity search.
+   * Required for RAG (Retrieval-Augmented Generation) and AI features.
+   */
+  vectorSearch: z.boolean().default(false).describe('Supports vector embeddings and similarity search'),
+
+  /**
+   * Whether the driver supports geospatial queries.
+   */
+  geoSpatial: z.boolean().default(false).describe('Supports geospatial queries'),
 });
 
 /**
@@ -158,6 +181,20 @@ export const DriverInterfaceSchema = z.object({
   checkHealth: z.function()
     .returns(z.promise(z.boolean()))
     .describe('Health check'),
+  
+  /**
+   * Get Connection Pool Statistics.
+   * Useful for monitoring database load.
+   */
+  getPoolStats: z.function()
+    .returns(z.object({
+      total: z.number(),
+      idle: z.number(),
+      active: z.number(),
+      waiting: z.number(),
+    }).optional())
+    .optional()
+    .describe('Get connection pool statistics'),
 
   // ============================================================================
   // Raw Execution (Escape Hatch)
@@ -211,6 +248,20 @@ export const DriverInterfaceSchema = z.object({
     .returns(z.promise(z.array(z.record(z.any()))))
     .describe('Find records'),
 
+  /**
+   * Stream records matching the structured query.
+   * Optimized for large datasets to avoid memory overflow.
+   * 
+   * @param object - The name of the object.
+   * @param query - The structured QueryAST.
+   * @param options - Driver options.
+   * @returns AsyncIterable/ReadableStream of records.
+   */
+  findStream: z.function()
+    .args(z.string(), QuerySchema, DriverOptionsSchema.optional())
+    .returns(z.any()) // Zod cannot easily represent AsyncIterable<T>
+    .describe('Stream records (AsyncIterable)'),
+
   /**
    * Find a single record by query.
    * Similar to find(), but returns only the first match or null.
@@ -255,6 +306,20 @@ export const DriverInterfaceSchema = z.object({
     .returns(z.promise(z.record(z.any())))
     .describe('Update record'),
 
+  /**
+   * Upsert (Update or Insert) a record.
+   * 
+   * @param object - The object name.
+   * @param data - The data to upsert.
+   * @param conflictKeys - Fields to check for conflict (uniqueness).
+   * @param options - Driver options.
+   * @returns The created or updated record.
+   */
+  upsert: z.function()
+    .args(z.string(), z.record(z.any()), z.array(z.string()).optional(), DriverOptionsSchema.optional())
+    .returns(z.promise(z.record(z.any())))
+    .describe('Upsert record'),
+
   /**
    * Delete a record by ID.
    * 
@@ -324,9 +389,13 @@ export const DriverInterfaceSchema = z.object({
 
   /**
    * Begin a new database transaction.
+   * @param options - Isolation level and other settings.
    * @returns A transaction handle to be passed to subsequent operations via `options.transaction`.
    */
   beginTransaction: z.function()
+    .args(z.object({
+      isolationLevel: z.enum(['READ UNCOMMITTED', 'READ COMMITTED', 'REPEATABLE READ', 'SERIALIZABLE', 'SNAPSHOT']).optional()
+    }).optional())
     .returns(z.promise(z.any()))
     .describe('Start transaction'),
 
@@ -375,6 +444,19 @@ export const DriverInterfaceSchema = z.object({
   dropTable: z.function()
     .args(z.string(), DriverOptionsSchema.optional())
     .returns(z.promise(z.void())),
+
+  /**
+   * Analyze query performance.
+   * Returns execution plan without executing the query (where possible).
+   * 
+   * @param object - The object name.
+   * @param query - The query to explain.
+   * @returns The execution plan details.
+   */
+  explain: z.function()
+    .args(z.string(), QuerySchema, DriverOptionsSchema.optional())
+    .returns(z.promise(z.any()))
+    .optional(),
 });
 
 /**