dzql 0.5.8 → 0.5.9

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dzql",
-  "version": "0.5.8",
+  "version": "0.5.9",
   "description": "PostgreSQL-powered framework with zero boilerplate CRUD operations and real-time WebSocket synchronization",
   "type": "module",
   "main": "src/server/index.js",

package/src/compiler/codegen/subscribable-codegen.js

@@ -238,13 +238,20 @@ $$ LANGUAGE plpgsql STABLE SECURITY DEFINER;`;
     // Build relation subqueries
     const relationSelects = this._generateRelationSelects();
 
+    // Build schema with path mapping (baked in at compile time)
+    const pathMapping = this.buildPathMapping();
+    const schemaJson = JSON.stringify({
+      root: this.rootEntity,
+      paths: pathMapping
+    });
+
     return `CREATE OR REPLACE FUNCTION get_${this.name}(
   p_params JSONB,
   p_user_id INT
 ) RETURNS JSONB AS $$
 DECLARE
 ${paramDeclarations}
-  v_result JSONB;
+  v_data JSONB;
 BEGIN
   -- Extract parameters
 ${paramExtractions}
@@ -258,11 +265,15 @@ ${paramExtractions}
   SELECT jsonb_build_object(
     '${this.rootEntity}', row_to_json(root.*)${relationSelects}
   )
-  INTO v_result
+  INTO v_data
   FROM ${this.rootEntity} root
   WHERE ${rootFilter};
 
-  RETURN v_result;
+  -- Return data with embedded schema for atomic updates
+  RETURN jsonb_build_object(
+    'data', v_data,
+    'schema', '${schemaJson}'::jsonb
+  );
 END;
 $$ LANGUAGE plpgsql SECURITY DEFINER;`;
   }

package/src/server/ws.js

@@ -322,14 +322,28 @@ export function createRPCHandler(customHandlers = {}) {
         try {
           // Execute initial query (this also checks permissions)
           const queryResult = await sql.unsafe(
-            `SELECT get_${subscribableName}($1, $2) as data`,
+            `SELECT get_${subscribableName}($1, $2) as result`,
             [params, ws.data.user_id]
           );
 
-          const data = queryResult[0]?.data;
-
-          // Get subscribable metadata for schema (path mapping for atomic updates)
-          const metadata = await getSubscribableMetadata(subscribableName, sql);
+          const queryData = queryResult[0]?.result;
+
+          // Check if compiled function returned embedded schema
+          // Compiled: { data, schema } | Interpreted: just the data object
+          let data, schema;
+          if (queryData && queryData.schema && queryData.data !== undefined) {
+            // Compiled mode - schema is embedded
+            data = queryData.data;
+            schema = queryData.schema;
+          } else {
+            // Interpreted mode - fetch schema from metadata table
+            data = queryData;
+            const metadata = await getSubscribableMetadata(subscribableName, sql);
+            schema = {
+              root: metadata.rootEntity,
+              paths: metadata.pathMapping
+            };
+          }
 
           // Register subscription in memory
           const subscriptionId = registerSubscription(
@@ -343,11 +357,7 @@ export function createRPCHandler(customHandlers = {}) {
           const result = {
             subscription_id: subscriptionId,
             data,
-            // Include schema for atomic update support
-            schema: {
-              root: metadata.rootEntity,
-              paths: metadata.pathMapping
-            }
+            schema
           };
 
           wsLogger.response(method, result, Date.now() - startTime);