强制hugsql查询函数在返回错误数量的结果时抛出错误

时间:2016-09-26 17:17:46

标签: sql jdbc clojure hugsql

使用Clojure和hugsql。我这样定义我的查询。

-- :name query-should-return-one-but-returns-multiple
-- :result one
-- :command :query
SELECT v.id FROM some_table v;

使用def-db-fns后,这将在我的命名空间中创建一个函数query-should-return-one-but-returns-multiple

但是,如果some_table中有多行,则此函数将返回任意行,并且不会发出错误信号。

如果数据库返回多个结果,我如何强制定义的查询函数返回:one抛出异常?

1 个答案:

答案 0 :(得分:1)

-- :result :one只是获取查询返回结果集的第一行,因此没有验证您的查询正好返回1条记录(hashmap)。

但是,HugSQL使用Clojure多方法来定义其结果类型,因此您可以创建自己的方法或覆盖现有方法以满足您的需求。

我们定义一个函数result-exactly-one,当结果的计数不是1时,它将抛出异常。然后,我们使用hugsql-result-fn关键字定义:exactly-one多方法。请注意引用的名称空间(在本例中为user)符号,指的是我们创建的函数。

(require '[hugsql.core :as hugsql]
         '[hugsql.adapter])

(defn result-exactly-one
  [this result options]
  (let [rs (hugsql.adapter/result-many this result options)]
    (if (= 1 (count rs)) 
        rs 
        (throw (ex-info "Oops!" {})))))

(defmethod hugsql/hugsql-result-fn :exactly-one [sym] 'user/result-exactly-one)
-- :name return-exactly-one
-- :result :exactly-one
-- :command :query
SELECT v.id FROM t1 v;

返回0或多个记录时的结果:

user=> (return-exactly-one db)

ExceptionInfo Oops!  clojure.core/ex-info (core.clj:4617)