Add columnCount function

nurpax · Feb 13, 2017 · 54d4af6 · 54d4af6
1 parent 64c5287
commit 54d4af6
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/Database/SQLite/Simple.hs b/Database/SQLite/Simple.hs
@@ -90,6 +90,7 @@ module Database.SQLite.Simple (
   , bindNamed
   , reset
   , columnName
+  , columnCount
   , withBind
   , nextRow
     -- ** Exceptions
@@ -246,6 +247,10 @@ columnName (Statement stmt) (ColumnIndex n) = BaseD.columnName stmt n >>= takeUt
     takeUtf8 Nothing  =
       throwIO (IndexOutOfBounds ("Column index " ++ show n ++ " out of bounds"))
 
+-- | Return number of columns in the query
+columnCount :: Statement -> IO ColumnIndex
+columnCount (Statement stmt) = ColumnIndex <$> BaseD.columnCount stmt
+
 -- | Binds parameters to a prepared statement, and 'reset's the statement when
 -- the callback completes, even in the presence of exceptions.
 --