@@ -71,7 +71,7 @@ const runAggregation = async (aggregationPipeline, collection) => {
71
71
72
72
const coll = client . db ( "river_runner" ) . collection ( collection ) ;
73
73
74
- coll . aggregate ( aggregationPipeline , async ( cmdErr , result ) => {
74
+ coll . aggregate ( aggregationPipeline , { allowDiskUse : true , cursor : { batchSize : 1000 } } , async ( cmdErr , result ) => {
75
75
assert . equal ( null , cmdErr ) ;
76
76
77
77
const aggregationData = await result . toArray ( ) ;
@@ -88,43 +88,43 @@ const main = async () => {
88
88
const csvFilepath = "../../public/data/name_overrides.csv" ;
89
89
90
90
const existingOverrides = await getExistingOverrides ( csvFilepath ) ;
91
- const groupedOccurences = await runAggregation ( unnamedFeatureCounts , "unnamed_features" ) ;
91
+ // const groupedOccurences = await runAggregation(unnamedFeatureCounts, "unnamed_features");
92
92
const groupedSuggestions = await runAggregation ( suggestionCounts , "suggestions" ) ;
93
93
94
- // const overridenIDs = existingOverrides.map((d) => Number(d.levelpathid));
95
- // const unhandledSuggestions = groupedSuggestions
96
- // .filter((d) => d.count > 3)
97
- // .filter((d) => !overridenIDs.includes(d._id));
98
-
99
- // console.log(unhandledSuggestions);
100
- const joinedData = groupedOccurences . map ( ( d ) => {
101
- const suggestions = groupedSuggestions . find ( ( a ) => a . _id === d . _id ) || {
102
- count : 0 ,
103
- route_url : [ ] ,
104
- suggestions : null ,
105
- } ;
106
-
107
- const override = existingOverrides . find ( ( a ) => Number ( a . levelpathid ) === d . _id ) ;
108
-
109
- return {
110
- levelpathid : d . _id ,
111
- sample_route_url : [ ...suggestions . route_url , d . route_url ] . filter ( ( d ) => d ) [ 0 ] ,
112
- num_suggestions : suggestions . count ,
113
- num_occurences : d . count ,
114
- suggestions : suggestions . suggestions ,
115
- overriden : override ? true : false ,
116
- overriden_val : override ? override . feature_name : null ,
117
- } ;
118
- } ) ;
119
-
120
- const outputFile = fs . createWriteStream ( "data/aggregated_suggestions.csv" ) ;
121
-
122
- csv
123
- . write ( joinedData , { headers : true } )
124
- . on ( "finish" , function ( ) {
125
- console . log ( "Write to CSV successfully!" ) ;
126
- } )
127
- . pipe ( outputFile ) ;
94
+ const overridenIDs = existingOverrides . map ( ( d ) => Number ( d . levelpathid ) ) ;
95
+ const unhandledSuggestions = groupedSuggestions
96
+ . filter ( ( d ) => d . count > 3 )
97
+ . filter ( ( d ) => ! overridenIDs . includes ( d . _id ) ) ;
98
+
99
+ console . log ( unhandledSuggestions ) ;
100
+ // const joinedData = groupedOccurences.map((d) => {
101
+ // const suggestions = groupedSuggestions.find((a) => a._id === d._id) || {
102
+ // count: 0,
103
+ // route_url: [],
104
+ // suggestions: null,
105
+ // };
106
+
107
+ // const override = existingOverrides.find((a) => Number(a.levelpathid) === d._id);
108
+
109
+ // return {
110
+ // levelpathid: d._id,
111
+ // sample_route_url: [...suggestions.route_url, d.route_url].filter((d) => d)[0],
112
+ // num_suggestions: suggestions.count,
113
+ // num_occurences: d.count,
114
+ // suggestions: suggestions.suggestions,
115
+ // overriden: override ? true : false,
116
+ // overriden_val: override ? override.feature_name : null,
117
+ // };
118
+ // });
119
+
120
+ // const outputFile = fs.createWriteStream("data/aggregated_suggestions.csv");
121
+
122
+ // csv
123
+ // .write(joinedData, { headers: true })
124
+ // .on("finish", function () {
125
+ // console.log("Write to CSV successfully!");
126
+ // })
127
+ // .pipe(outputFile);
128
128
} ;
129
129
130
130
main ( ) ;
0 commit comments