versioned table

azt3k · Feb 7, 2016 · 07aaccb · 07aaccb
1 parent 7f9e09a
commit 07aaccb
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/src/abc/code/Classes/DataObjectHelper.php b/src/abc/code/Classes/DataObjectHelper.php
@@ -1,6 +1,6 @@
 <?php
 
-class DataObjectHelper{
+class DataObjectHelper {
 
 	/*
 	 *	Stores a cache of the extension Map
@@ -15,6 +15,18 @@ protected static function db_dialect() {
 		return $databaseConfig['type'];
 	}
 
+	public static function versioned_table($className) {
+
+		$stagedRes = (
+			singleton($className)->hasExtension('Versioned') &&
+			strtolower(Versioned::current_stage()) != 'stage'
+		)
+			? $className . '_' . ucfirst(strtolower(Versioned::current_stage()))
+			: $className;
+
+		return $stagedRes;
+	}
+
 	/*
 	 *	Returns an array of classnames that have been extended with a Extension
 	 */