这个问题似乎只能通过一些特定的函数来解决,然后最明显的是 do 语法。下面的函数解决了在 direct-sqlite3 包中访问 SQLite 数据库的问题(并且还插入了一个 REGEXP 处理程序)。
import Text.Regex.Base.RegexLike
import qualified Text.Regex.PCRE.ByteString as PCRE
import qualified Data.ByteString as BS
import Data.Text (pack,Text,append)
import Data.Text.Encoding (encodeUtf8)
import Data.Int (Int64)
import Database.SQLite3
pcreRegex :: BS.ByteString -> BS.ByteString -> IO Int64
pcreRegex reg b = do
reC <- pcreCompile reg
re <- case reC of
(Right r) -> return r
(Left (off,e) ) -> fail e
reE <- PCRE.execute re b
case reE of
(Right (Just _)) -> return (1 :: Int64)
(Right (Nothing)) -> return (0 :: Int64)
(Left (c,e)) -> fail e -- Not a good idea maybe, but I have no use for error messages.
where pcreCompile = PCRE.compile defaultCompOpt defaultExecOpt
sqlRegexp :: FuncContext -> FuncArgs -> IO ()
sqlRegexp ctx args = do
r <- fmap encodeUtf8 $ funcArgText args 0
t <- fmap encodeUtf8 $ funcArgText args 1
res <- pcreRegex r t
funcResultInt64 ctx res
getRows :: Statement -> [Maybe ColumnType] -> [[SQLData]] -> IO [[SQLData]]
getRows stmt coltypes acc = do
r <- step stmt
case r of
Done -> return acc
Row -> do
out <- typedColumns stmt coltypes
getRows stmt coltypes (out:acc)
runQuery q args columntypes dbFile = do
conn <- open $ pack dbFile
createFunction conn "regexp" (Just 2) True sqlRegexp
statement <- prepare conn q
bind statement args
res <- fmap reverse $ getRows statement (fmap Just columntypes) [[]]
Database.SQLite3.finalize statement
deleteFunction conn "regexp" (Just 2)
close conn
return $ res
希望这可以帮助这里的人。