创建函数未终止的美元报价字符串

Question

我正在尝试使用postgres（pq lib）数据库与Goose创建此函数。

我的代码如下：

   CREATE OR REPLACE FUNCTION add_userlocation(user_id INT, location_id INT) RETURNS VOID AS
   $BODY$
   BEGIN
       LOOP
           UPDATE userslocations SET count = count+1 WHERE userid = user_id AND locationid = location_id;
        IF found THEN
            RETURN;
        END IF;
        BEGIN
            INSERT INTO userslocations(userid,locationid, count) VALUES (user_id, location_id, 1);
               RETURN;
           EXCEPTION WHEN unique_violation THEN
        END;
       END LOOP;
   END;
   $BODY$
   LANGUAGE plpgsql;

当我尝试goose up时，它提供了一个错误：

(pq: unterminated dollar-quoted string at or near "$BODY$
BEGIN
    LOOP
        -- first try to update the key
        UPDATE userslocations SET count = count+1 WHERE userid = user_id AND locationid = location_id;
"), quitting migration.

Goose 基本上回显了 pq 库错误，所以我不认为它在 Goose 中，而是在 pq 库中。查询在 pgAdmin III 上成功运行。

Answer 1

根据 goose 文档，包含分号的复杂语句必须用 -- +goose StatementBegin 和 -- +goose StatementEnd 进行批注

您的语句包含嵌入其中的分号，因此您需要使用这些注释。否则，goose 会破坏 SQL，以便 libpq 给出错误。