Allow migrate to be limited by env variable

City-of-Helsinki · Jan 20, 2021 · 0041cc7 · 0041cc7
1 parent 91a38ac
commit 0041cc7
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/MigrateTrait.php b/src/MigrateTrait.php
@@ -9,6 +9,18 @@
  */
 trait MigrateTrait {
 
+  /**
+   * Gets the limit.
+   *
+   * @return int
+   *   The limit.
+   */
+  public function getLimit() : int {
+    $limit = getenv('MIGRATE_LIMIT') ?: 0;
+
+    return (int) $limit;
+  }
+
   /**
    * Checks if we're doing partial migrate.
    *