如何修复使用 Goose 和 PostgreSQL 创建存储过程时的'未终止字符串”错误？

Postgres Goose 引发未终止字符串错误

问题

尝试使用 PostgreSQL 数据库通过 Goose 创建存储过程时，该过程遇到以下情况错误：

(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.

原因

从 pq 库回显的 Goose 错误表明 SQL 格式错误。具体来说，包含分号的复杂语句需要在 Goose 中进行特殊注释：

解决方案

要纠正错误，请按以下方式注释 SQL：

CREATE OR REPLACE FUNCTION add_userlocation(user_id INT, location_id INT) RETURNS VOID AS
$BODY$
-- +goose StatementBegin
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;
-- +goose StatementEnd
END;
$BODY$
LANGUAGE plpgsql;

通过声明每个语句的开头和结尾，指示 Goose 正确处理 SQL，解决未终止的字符串错误。

以上是如何修复使用 Goose 和 PostgreSQL 创建存储过程时的'未终止字符串”错误？的详细内容。更多信息请关注PHP中文网其他相关文章！