bulltrackers-module 1.0.116 → 1.0.118

package/functions/computation-system/helpers/orchestration_helpers.js

@@ -688,6 +688,7 @@ async function runMetaComputation(
 
     try {
         const resultsCollectionRef = db.collection(config.resultsCollection).doc(dateStr).collection(config.resultsSubcollection);
+        const compsSub = config.computationsSubcollection || 'computations';
         let successCount = 0;
         
         const dependenciesForMetaCalc = {
@@ -772,11 +773,51 @@ async function runMetaComputation(
                 const summaryData = {};
 
                 if (result && Object.keys(result).length > 0) {
-                    const computationDocRef = resultsCollectionRef.doc(category)
-                                                      .collection(config.computationsSubcollection)
-                                                      .doc(calcName);
-                    pendingWrites.push({ ref: computationDocRef, data: result });
+
+                    // --- START SHARDING FIX ---
+                    let isSharded = false;
+                    const shardedCollections = {
+                        // Add keys from sharded meta-calcs here
+                        'sharded_user_profile': config.shardedUserProfileCollection,
+                        'sharded_user_profitability': config.shardedProfitabilityCollection 
+                    };
+
+                    for (const resultKey in shardedCollections) {
+                        if (result[resultKey]) {
+                            isSharded = true;
+                            const shardCollectionName = shardedCollections[resultKey];
+                            if (!shardCollectionName) {
+                                logger.log('ERROR', `[${passName}] Missing config key for sharded collection: ${resultKey}`);
+                                continue;
+                            }
+                            
+                            const shardedData = result[resultKey];
+                            
+                            for (const shardId in shardedData) {
+                                const shardDocData = shardedData[shardId];
+                                if (shardDocData && (Object.keys(shardDocData).length > 0)) {
+                                    const shardRef = db.collection(shardCollectionName).doc(shardId);
+                                    // Use { merge: true } to safely write to sharded docs
+                                    pendingWrites.push({ ref: shardRef, data: shardDocData, merge: true });
+                                }
+                            }
+                            
+                            // De-structure the sharded key from the result
+                            const { [resultKey]: _, ...otherResults } = result;
+                            result = otherResults; // Re-assign 'result' to be only the non-sharded data
+                        }
+                    }
                     
+                    // After all sharding is handled, save any *remaining* data
+                    // (e.g., 'daily_investor_scores' from user-investment-profile)
+                    if (result && Object.keys(result).length > 0) {
+                        const computationDocRef = resultsCollectionRef.doc(category)
+                                                          .collection(compsSub)
+                                                          .doc(calcName);
+                        pendingWrites.push({ ref: computationDocRef, data: result });
+                    }
+                    // --- END SHARDING FIX ---
+
                     if (!summaryData[category]) summaryData[category] = {};
                     summaryData[category][calcName] = true;
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "bulltrackers-module",
-  "version": "1.0.116",
+  "version": "1.0.118",
   "description": "Helper Functions for Bulltrackers.",
   "main": "index.js",
   "files": [