introduce compiledJinjaCache

Author: KSDaemon

packages/cubejs-schema-compiler/src/compiler/DataSchemaCompiler.ts

@@ -89,6 +89,7 @@ export type DataSchemaCompilerOptions = {
   allowNodeRequire?: boolean;
   compiledScriptCache: LRUCache<string, vm.Script>;
   compiledYamlCache: LRUCache<string, string>;
+  compiledJinjaCache: LRUCache<string, string>;
 };
 
 export type TranspileOptions = {
@@ -166,6 +167,8 @@ export class DataSchemaCompiler {
 
   private readonly compiledYamlCache: LRUCache<string, string>;
 
+  private readonly compiledJinjaCache: LRUCache<string, string>;
+
   private compileV8ContextCache: vm.Context | null = null;
 
   // FIXME: Is public only because of tests, should be private
@@ -200,6 +203,7 @@ export class DataSchemaCompiler {
     this.compilerId = options.compilerId || 'default';
     this.compiledScriptCache = options.compiledScriptCache;
     this.compiledYamlCache = options.compiledYamlCache;
+    this.compiledJinjaCache = options.compiledJinjaCache;
   }
 
   public compileObjects(compileServices: CompilerInterface[], objects, errorsReport: ErrorReporter) {
@@ -770,13 +774,24 @@ export class DataSchemaCompiler {
     errorsReport: ErrorReporter,
     options: TranspileOptions
   ): Promise<(FileContent | undefined)> {
-    const renderedFile = await this.yamlCompiler.renderTemplate(
-      file,
-      this.standalone ? {} : this.cloneCompileContextWithGetterAlias(this.compileContext),
-      this.pythonContext!
-    );
+    const cacheKey = crypto.createHash('md5').update(JSON.stringify(file.content)).digest('hex');
+
+    let renderedFileContent: string;
+
+    if (this.compiledJinjaCache.has(cacheKey)) {
+      renderedFileContent = this.compiledJinjaCache.get(cacheKey)!;
+    } else {
+      const renderedFile = await this.yamlCompiler.renderTemplate(
+        file,
+        this.standalone ? {} : this.cloneCompileContextWithGetterAlias(this.compileContext),
+        this.pythonContext!
+      );
+      renderedFileContent = renderedFile.content;
+
+      this.compiledJinjaCache.set(cacheKey, renderedFileContent);
+    }
 
-    return this.transpileYamlFile(renderedFile, errorsReport, options);
+    return this.transpileYamlFile({ ...file, content: renderedFileContent }, errorsReport, options);
   }
 
   public withQuery(query, fn) {

packages/cubejs-schema-compiler/src/compiler/PrepareCompiler.ts

@@ -38,6 +38,7 @@ export type PrepareCompilerOptions = {
   adapter?: string;
   compiledScriptCache?: LRUCache<string, vm.Script>;
   compiledYamlCache?: LRUCache<string, string>;
+  compiledJinjaCache?: LRUCache<string, string>;
 };
 
 export interface CompilerInterface {
@@ -61,6 +62,7 @@ export const prepareCompiler = (repo: SchemaFileRepository, options: PrepareComp
 
   const compiledScriptCache = options.compiledScriptCache || new LRUCache<string, vm.Script>({ max: 250 });
   const compiledYamlCache = options.compiledYamlCache || new LRUCache<string, string>({ max: 250 });
+  const compiledJinjaCache = options.compiledJinjaCache || new LRUCache<string, string>({ max: 250 });
 
   const transpilers: TranspilerInterface[] = [
     new ValidationTranspiler(),
@@ -82,6 +84,7 @@ export const prepareCompiler = (repo: SchemaFileRepository, options: PrepareComp
     viewCompilationGate,
     compiledScriptCache,
     compiledYamlCache,
+    compiledJinjaCache,
     viewCompilers: [viewCompiler],
     cubeCompilers: [cubeEvaluator, joinGraph, metaTransformer],
     contextCompilers: [contextEvaluator],