feat: Use dbt state to avoid large table rebuilds

tutoraspects/commands_v1.py

@@ -52,16 +52,27 @@ def load_xapi_test_data(config_file) -> list[tuple[str, str]]:
          tutor local do dbt -c "run -m enrollments_by_day --threads 4"
          """,
 )
-def dbt(command: string) -> list[tuple[str, str]]:
+@click.option(
+    "--only_changed",
+    default=True,
+    type=click.UNPROCESSED,
+    help="""Whether to only run models that have changed since the last dbt run. Since
+         re-running materialized views will recreate potentially huge datasets and 
+         incur downtime, this defaults to true.
+         
+         If no prior state is found, the command will run as if this was False. 
+         """,
+)
+def dbt(only_changed: bool, command: string) -> list[tuple[str, str]]:
     """
     Job that proxies dbt commands to a container which runs them against ClickHouse.
     """
     return [
         (
             "aspects",
             "echo 'Making dbt script executable...' && "
-            f"echo 'Running dbt {command}' && "
-            f"bash /app/aspects/scripts/dbt.sh {command} && "
+            f"echo 'Running dbt, only_changed: {only_changed} command: {command}' && "
+            f"bash /app/aspects/scripts/dbt.sh {only_changed} {command} && "
             "echo 'Done!';",
         ),
     ]

tutoraspects/patches/k8s-jobs

@@ -18,6 +18,8 @@ spec:
             value: {{ ASPECTS_XAPI_DATABASE }}
           - name: ASPECTS_ENROLLMENT_EVENTS_TABLE
             value: {{ ASPECTS_ENROLLMENT_EVENTS_TABLE }}
+          - name: DBT_STATE
+            value: {{ DBT_STATE_DIR }}
         image: {{ DOCKER_IMAGE_ASPECTS }}
         securityContext:
           allowPrivilegeEscalation: false
@@ -35,7 +37,12 @@ spec:
             name: alembic
           - mountPath: /app/aspects/migrations/alembic/versions
             name: versions
+          - mountPath: {{ DBT_STATE_DIR }}
+            name: dbt-state
       volumes:
+        - name: dbt-state
+          persistentVolumeClaim:
+            claimName: aspects-dbt
         - name: scripts
           configMap:
             name: aspects-scripts

tutoraspects/patches/k8s-volumes

@@ -1,3 +1,18 @@
+---
+apiVersion: v1
+kind: PersistentVolumeClaim
+metadata:
+  name: aspects-dbt
+  labels:
+    app.kubernetes.io/component: volume
+    app.kubernetes.io/name: aspects-dbt
+spec:
+  accessModes:
+    - ReadWriteOnce
+  resources:
+    requests:
+      storage: 10M
+
 {% if RUN_CLICKHOUSE %}
 ---
 apiVersion: v1

tutoraspects/patches/local-docker-compose-jobs-services

@@ -7,6 +7,7 @@ aspects-job:
   volumes:
     - ../../env/plugins/aspects/apps/aspects:/app/aspects
     - ../../env/plugins/aspects/apps/aspects/scripts/:/app/aspects/scripts:ro
+    - ../../data/aspects-dbt:{{ DBT_STATE_DIR }}
   {% if RUN_SUPERSET or RUN_CLICKHOUSE or RUN_RALPH %}depends_on:{% if RUN_SUPERSET %}
     - superset{% endif %}{% if RUN_CLICKHOUSE%}
     - clickhouse{% endif %}{% if RUN_RALPH %}

tutoraspects/plugin.py

@@ -353,6 +353,7 @@
         ("DBT_REPOSITORY", "https://github.com/openedx/aspects-dbt"),
         ("DBT_BRANCH", "v3.5.0"),
         ("DBT_SSH_KEY", ""),
+        ("DBT_STATE_DIR", "/app/aspects/dbt_state/"),
         # This is a pip compliant list of Python packages to install to run dbt
         # make sure packages with versions are enclosed in double quotes
         ("EXTRA_DBT_PACKAGES", []),

tutoraspects/templates/aspects/apps/aspects/scripts/dbt.sh

@@ -26,11 +26,28 @@ cd aspects-dbt || exit
 
 export ASPECTS_EVENT_SINK_DATABASE={{ASPECTS_EVENT_SINK_DATABASE}}
 export ASPECTS_XAPI_DATABASE={{ASPECTS_XAPI_DATABASE}}
+export DBT_STATE={{ DBT_STATE_DIR }}
 
 echo "Installing dbt dependencies"
 dbt deps --profiles-dir /app/aspects/dbt/
 
-echo "Running dbt $*"
-dbt "$@" --profiles-dir /app/aspects/dbt/
-
+echo "Running dbt ${@:2}"
+
+if [ "$1" == "True" ]
+then
+  echo "Requested to only run modified state, checking for ${DBT_STATE}/manifest.json"
+fi
+
+# If state exists and we've asked to only run changed files, add the flag
+if [ "$1" == "True" ] && [ -e "${DBT_STATE}/manifest.json" ]
+then
+  echo "Found {{DBT_STATE_DIR}}/manifest.json so only running modified items"
+  dbt "${@:2}" --profiles-dir /app/aspects/dbt/ -s state:modified
+else
+  echo "Running command without state:modified+"
+  dbt "${@:2}" --profiles-dir /app/aspects/dbt/
+fi
+
+rm -rf ${DBT_STATE}/*
+cp -r ./target/manifest.json ${DBT_STATE}
 rm -rf aspects-dbt

tutoraspects/templates/aspects/jobs/init/aspects/init-aspects.sh

@@ -2,4 +2,4 @@
 
 bash /app/aspects/scripts/alembic.sh upgrade head
 
-bash /app/aspects/scripts/dbt.sh run
+bash /app/aspects/scripts/dbt.sh True run