pivot name fix

vapor · Oct 17, 2016 · b85adc3 · b85adc3
1 parent 5c30919
commit b85adc3
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/Sources/Fluent/Entity/Entity.swift b/Sources/Fluent/Entity/Entity.swift
@@ -11,6 +11,15 @@ public protocol Entity: Preparation, NodeConvertible {
     */
     static var entity: String { get }
 
+    /**
+        The name to use for internal storage.
+
+        This should be left as the default 
+        implementation except for special cases
+        like pivots.
+    */
+    static var name: String { get }
+
     /**
         The entity's primary identifier.
         This is the same value used for

diff --git a/Sources/Fluent/Query/Join.swift b/Sources/Fluent/Query/Join.swift
@@ -29,6 +29,10 @@ public final class Pivot<
         return "\(left.name)_\(right.name)"
     }
 
+    public static var name: String {
+        return entity
+    }
+
     public static var left: Entity.Type {
         if First.entity < Second.entity {
             return First.self