added a new withPivot() overload

silverqx · Sep 1, 2022 · 97a12a9 · 97a12a9
1 parent 6d6ad32
commit 97a12a9
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/include/orm/tiny/relations/concerns/interactswithpivottable.hpp b/include/orm/tiny/relations/concerns/interactswithpivottable.hpp
@@ -12,6 +12,7 @@ TINY_SYSTEM_HEADER
 #include <range/v3/view/set_algorithm.hpp>
 
 #include "orm/exceptions/domainerror.hpp"
+#include "orm/ormconcepts.hpp"
 #include "orm/tiny/types/syncchanges.hpp"
 #include "orm/tiny/utils/attribute.hpp"
 #include "orm/utils/query.hpp"
@@ -83,6 +84,9 @@ namespace Concerns
         BelongsToManyType &withPivot(QStringList &&columns);
         /*! Set the columns on the pivot table to retrieve. */
         inline BelongsToManyType &withPivot(QString &&column);
+        /*! Set the columns on the pivot table to retrieve. */
+        template<QStringConcept ...Args>
+        BelongsToManyType &withPivot(Args &&...columns);
 
         /*! Determine whether the given column is defined as a pivot column. */
         inline bool hasPivotColumn(const QString &column) const;
@@ -347,6 +351,14 @@ namespace Concerns
         return withPivot(QStringList {std::move(column)});
     }
 
+    template<class Model, class Related, class PivotType>
+    template<QStringConcept ...Args>
+    BelongsToMany<Model, Related, PivotType> &
+    InteractsWithPivotTable<Model, Related, PivotType>::withPivot(Args &&...columns)
+    {
+        return withPivot(QStringList {std::forward<Args>(columns)...});
+    }
+
     template<class Model, class Related, class PivotType>
     bool InteractsWithPivotTable<Model, Related, PivotType>::hasPivotColumn(
             const QString &column) const