|
| 1 | +{-# LANGUAGE OverloadedStrings #-} |
| 2 | +{-# LANGUAGE FlexibleInstances #-} |
| 3 | +{-# LANGUAGE FlexibleContexts #-} |
| 4 | +{-# LANGUAGE ExistentialQuantification #-} |
| 5 | + |
| 6 | +{-| |
| 7 | +
|
| 8 | + This module provides a very thin wrapper around HDBC |
| 9 | +-} |
| 10 | +module Snap.Snaplet.Hdbc where |
| 11 | + |
| 12 | +import Database.HDBC (IConnection(), SqlValue, Statement, SqlColDesc) |
| 13 | +import qualified Database.HDBC as HDBC |
| 14 | +import Snap.Snaplet |
| 15 | +import Control.Monad.State |
| 16 | + |
| 17 | +data HdbcSnaplet conn = IConnection conn => HdbcSnaplet { |
| 18 | + hdbcConn :: conn |
| 19 | + } |
| 20 | + |
| 21 | +hdbcInit :: IConnection conn => conn -> SnapletInit b (HdbcSnaplet conn) |
| 22 | +hdbcInit conn = makeSnaplet "hdbc" "HDBC abstraction" Nothing $ do |
| 23 | + onUnload $ HDBC.disconnect conn |
| 24 | + return $ HdbcSnaplet conn |
| 25 | + |
| 26 | +getConn :: IConnection conn => Handler b (HdbcSnaplet conn) conn |
| 27 | +getConn = gets hdbcConn |
| 28 | + |
| 29 | +runUnit :: IConnection conn => (conn -> IO ()) |
| 30 | + -> Handler b (HdbcSnaplet conn) () |
| 31 | +runUnit f = do |
| 32 | + conn <- gets hdbcConn |
| 33 | + liftIO $ f conn |
| 34 | + |
| 35 | +disconnect :: IConnection conn => Handler b (HdbcSnaplet conn) () |
| 36 | +disconnect = runUnit HDBC.disconnect |
| 37 | + |
| 38 | +commit :: IConnection conn => Handler b (HdbcSnaplet conn) () |
| 39 | +commit = runUnit HDBC.commit |
| 40 | + |
| 41 | +rollback :: IConnection conn => Handler b (HdbcSnaplet conn) () |
| 42 | +rollback = runUnit HDBC.rollback |
| 43 | + |
| 44 | +runRaw :: IConnection conn => String -> Handler b (HdbcSnaplet conn) () |
| 45 | +runRaw str = do |
| 46 | + conn <- gets hdbcConn |
| 47 | + liftIO $ HDBC.runRaw conn str |
| 48 | + |
| 49 | +run :: IConnection conn => String -> [SqlValue] |
| 50 | + -> Handler b (HdbcSnaplet conn) Integer |
| 51 | +run str vs = do |
| 52 | + conn <- gets hdbcConn |
| 53 | + liftIO $ HDBC.run conn str vs |
| 54 | + |
| 55 | +prepare :: IConnection conn => String -> Handler b (HdbcSnaplet conn) Statement |
| 56 | +prepare str = do |
| 57 | + conn <- gets hdbcConn |
| 58 | + liftIO $ HDBC.prepare conn str |
| 59 | + |
| 60 | +clone :: IConnection conn => Handler b (HdbcSnaplet conn) conn |
| 61 | +clone = do |
| 62 | + conn <- gets hdbcConn |
| 63 | + liftIO $ HDBC.clone conn |
| 64 | + |
| 65 | +pureStr :: IConnection conn => (conn -> String) |
| 66 | + -> Handler b (HdbcSnaplet conn) String |
| 67 | +pureStr f = do |
| 68 | + conn <- gets hdbcConn |
| 69 | + return $ f conn |
| 70 | + |
| 71 | +hdbcDriverName :: IConnection conn => Handler b (HdbcSnaplet conn) String |
| 72 | +hdbcDriverName = pureStr HDBC.hdbcDriverName |
| 73 | + |
| 74 | +hdbcClientVer :: IConnection conn => Handler b (HdbcSnaplet conn) String |
| 75 | +hdbcClientVer = pureStr HDBC.hdbcClientVer |
| 76 | + |
| 77 | +proxiedClientName :: IConnection conn => Handler b (HdbcSnaplet conn) String |
| 78 | +proxiedClientName = pureStr HDBC.proxiedClientName |
| 79 | + |
| 80 | +proxiedClientVer :: IConnection conn => Handler b (HdbcSnaplet conn) String |
| 81 | +proxiedClientVer = pureStr HDBC.proxiedClientVer |
| 82 | + |
| 83 | +dbServerVer :: IConnection conn => Handler b (HdbcSnaplet conn) String |
| 84 | +dbServerVer = pureStr HDBC.dbServerVer |
| 85 | + |
| 86 | +dbTransactionSupport :: IConnection conn => Handler b (HdbcSnaplet conn) Bool |
| 87 | +dbTransactionSupport = do |
| 88 | + conn <- gets hdbcConn |
| 89 | + return $ HDBC.dbTransactionSupport conn |
| 90 | + |
| 91 | +getTables :: IConnection conn => Handler b (HdbcSnaplet conn) [String] |
| 92 | +getTables = do |
| 93 | + conn <- gets hdbcConn |
| 94 | + liftIO $ HDBC.getTables conn |
| 95 | + |
| 96 | +describeTable :: IConnection conn => String |
| 97 | + -> Handler b (HdbcSnaplet conn) [(String, SqlColDesc)] |
| 98 | +describeTable str = do |
| 99 | + conn <- gets hdbcConn |
| 100 | + liftIO $ HDBC.describeTable conn str |
| 101 | + |
| 102 | +quickQuery' :: IConnection conn => String -> [SqlValue] |
| 103 | + -> Handler b (HdbcSnaplet conn) [[SqlValue]] |
| 104 | +quickQuery' str vs = do |
| 105 | + conn <- gets hdbcConn |
| 106 | + liftIO $ HDBC.quickQuery' conn str vs |
| 107 | + |
| 108 | +quickQuery :: IConnection conn => String -> [SqlValue] |
| 109 | + -> Handler b (HdbcSnaplet conn) [[SqlValue]] |
| 110 | +quickQuery str vs = do |
| 111 | + conn <- gets hdbcConn |
| 112 | + liftIO $ HDBC.quickQuery conn str vs |
| 113 | + |
| 114 | +sRun :: IConnection conn => String -> [Maybe String] |
| 115 | + -> Handler b (HdbcSnaplet conn) Integer |
| 116 | +sRun str mstrs = do |
| 117 | + conn <- gets hdbcConn |
| 118 | + liftIO $ HDBC.sRun conn str mstrs |
| 119 | + |
| 120 | +withTransaction :: IConnection conn => (conn -> IO a) -> Handler b (HdbcSnaplet conn) a |
| 121 | +withTransaction f = do |
| 122 | + conn <- gets hdbcConn |
| 123 | + liftIO $ HDBC.withTransaction conn f |
| 124 | + |
0 commit comments