如何有效地检查在haskell中使用HDBC是否存在条目?

时间:2013-05-19 09:01:52

标签: database haskell hdbc

就像在我的问题中提到的那样,我在一个表中有一个唯一的约束,我想在其中插入一些数据。我必须检查数据是否已经存在,如果没有则插入。

这是我似乎无法做到的。 这是我的代码:

import System.Environment (getArgs)
import Database.HDBC
import Database.HDBC.Sqlite3
import Text.Regex.Posix
import qualified Data.ByteString.Char8 as B

getFrom bstr =
  bstr =~ "From:.+@.+\\.(fr|com).?" :: B.ByteString

getTo bstr =
  bstr =~ "To:.+@.+\\.(fr|com).?" :: B.ByteString

getSubject bstr =
  bstr =~ "Subject:.+" :: B.ByteString

main = do
  --[dbPath, rawMail] <- getArgs
  bstr <- B.getContents
  conn <- connectSqlite3 "bmsg.db"
  let
    sqNomDest  = toSql $ getTo bstr
    sqNomExped = toSql $ getFrom bstr
  -- begin of the problematic part

  -- I make a query to know if the entry if already present
  qdbefore <- quickQuery' conn "SELECT d_id FROM dest WHERE maildest LIKE ?" [sqNomDest]
  qebefore <- quickQuery' conn "SELECT e_id FROM exped WHERE mailexped LIKE ?" [sqNomExped]
  -- then if not I insert it and else I return an arbitrary Int
  case qdbefore of --my check is probably wrong since that snippet always go to return 0
    [[SqlNull]] -> run conn "INSERT INTO dest(maildest) VALUES(?)" [sqNomDest] --unique on the column constraint so if I ran it alone on something already present it would raise an exeption
    _           -> return 0
  case qebefore of
    [[SqlNull]] -> run conn "INSERT INTO exped(mailexped) VALUES(?)" [sqNomExped] --same here for the constraint
    _           -> return 0
  commit conn
  --end of the problematic part

  [[qd]] <- quickQuery' conn "SELECT d_id FROM dest WHERE maildest LIKE ?" [sqNomDest]
  [[qe]] <- quickQuery' conn "SELECT e_id FROM exped WHERE mailexped LIKE ?" [sqNomExped]
  run conn "INSERT INTO mails(d_id, e_id, sujet, mail) VALUES (?, ?, ?, ?)" [qd, qe, toSql $ getSubject bstr, toSql bstr]
  commit conn
  disconnect conn


-- CREATE TABLE dest
-- (
-- d_id INTEGER PRIMARY KEY,
-- maildest VARCHAR(64) NOT NULL UNIQUE
-- );

-- CREATE TABLE exped
-- (
-- e_id INTEGER PRIMARY KEY,
-- mailexped VARCHAR(64) NOT NULL UNIQUE
-- );

-- CREATE TABLE mails
-- (
-- m_id    INTEGER PRIMARY KEY,
-- d_id    INTEGER NOT NULL,
-- e_id    INTEGER NOT NULL,
-- sujet   VARCHAR(128),
-- mail    TEXT NOT NULL,
-- CONSTRAINT fk_dest FOREIGN KEY (d_id) REFERENCES dest(d_id),
-- CONSTRAINT fk_exped FOREIGN KEY (e_id) REFERENCES exped(e_id)
-- );

没有错误,它会编译,但是当我给它发送一封尚未存在的电子邮件时,patern匹配失败。

1 个答案:

答案 0 :(得分:0)

如果您的查询没有结果,您将获得一个空列表,而不是一行包含一行和一个空 - 这就是为什么在插入新内容时案例失败的原因。对于失败的情况,只需匹配一个空列表。