137 lines
4.3 KiB
PL/PgSQL
137 lines
4.3 KiB
PL/PgSQL
/*
|
|
Warnings:
|
|
|
|
- You are about to drop the `Vote` table. If the table is not empty, all the data it contains will be lost.
|
|
|
|
*/
|
|
-- CreateEnum
|
|
CREATE TYPE "ItemActType" AS ENUM ('VOTE', 'BOOST', 'TIP');
|
|
|
|
-- DropForeignKey
|
|
ALTER TABLE "Vote" DROP CONSTRAINT "Vote_itemId_fkey";
|
|
|
|
-- DropForeignKey
|
|
ALTER TABLE "Vote" DROP CONSTRAINT "Vote_userId_fkey";
|
|
|
|
ALTER TABLE "Vote" RENAME TO "ItemAct";
|
|
|
|
ALTER TABLE "ItemAct" RENAME COLUMN "boost" TO "act";
|
|
|
|
ALTER TABLE "ItemAct"
|
|
ALTER COLUMN "act" DROP DEFAULT,
|
|
ALTER COLUMN "act" SET DATA TYPE "ItemActType"
|
|
USING (
|
|
CASE
|
|
WHEN "act" THEN 'BOOST'
|
|
ELSE 'VOTE'
|
|
END
|
|
)::"ItemActType";
|
|
|
|
-- CreateIndex
|
|
CREATE INDEX "ItemAct.itemId_index" ON "ItemAct"("itemId");
|
|
|
|
-- CreateIndex
|
|
CREATE INDEX "ItemAct.userId_index" ON "ItemAct"("userId");
|
|
|
|
-- CreateIndex
|
|
CREATE INDEX "ItemAct.act_index" ON "ItemAct"("act");
|
|
|
|
-- AddForeignKey
|
|
ALTER TABLE "ItemAct" ADD FOREIGN KEY ("itemId") REFERENCES "Item"("id") ON DELETE CASCADE ON UPDATE CASCADE;
|
|
|
|
-- AddForeignKey
|
|
ALTER TABLE "ItemAct" ADD FOREIGN KEY ("userId") REFERENCES "users"("id") ON DELETE CASCADE ON UPDATE CASCADE;
|
|
|
|
CREATE OR REPLACE FUNCTION item_act(item_id INTEGER, user_id INTEGER, act "ItemActType", act_sats INTEGER)
|
|
RETURNS INTEGER
|
|
LANGUAGE plpgsql
|
|
AS $$
|
|
DECLARE
|
|
user_sats INTEGER;
|
|
BEGIN
|
|
PERFORM ASSERT_SERIALIZED();
|
|
|
|
SELECT (msats / 1000) INTO user_sats FROM users WHERE id = user_id;
|
|
IF act_sats > user_sats THEN
|
|
RAISE EXCEPTION 'SN_INSUFFICIENT_FUNDS';
|
|
END IF;
|
|
|
|
UPDATE users SET msats = msats - (act_sats * 1000) WHERE id = user_id;
|
|
|
|
CASE act
|
|
when 'VOTE', 'BOOST' then
|
|
-- if we've already voted, then this is boost (doing this here prevents any potential
|
|
-- race)
|
|
IF EXISTS (SELECT 1 FROM "ItemAct" WHERE "itemId" = item_id AND "userId" = user_id) THEN
|
|
INSERT INTO "ItemAct" (sats, "itemId", "userId", act, created_at, updated_at)
|
|
VALUES (act_sats, item_id, user_id, 'BOOST', now_utc(), now_utc());
|
|
|
|
-- this is a vote
|
|
ELSE
|
|
-- only 1 sat votes are allowed
|
|
IF act_sats > 1 THEN
|
|
RAISE EXCEPTION 'SN_EXCEEDS_ACT_SAT_LIMIT';
|
|
END IF;
|
|
|
|
INSERT INTO "ItemAct" (sats, "itemId", "userId", act, created_at, updated_at)
|
|
VALUES (1, item_id, user_id, 'VOTE', now_utc(), now_utc());
|
|
|
|
-- give the item's user 1 sat
|
|
UPDATE users SET msats = msats + 1000 WHERE id = (SELECT "userId" FROM "Item" WHERE id = item_id);
|
|
END IF;
|
|
when 'TIP' then
|
|
INSERT INTO "ItemAct" (sats, "itemId", "userId", act, created_at, updated_at)
|
|
VALUES (act_sats, item_id, user_id, 'TIP', now_utc(), now_utc());
|
|
-- give the item's user act_sats
|
|
UPDATE users SET msats = msats + (act_sats * 1000) WHERE id = (SELECT "userId" FROM "Item" WHERE id = item_id);
|
|
END case;
|
|
|
|
RETURN act_sats;
|
|
END;
|
|
$$;
|
|
|
|
-- make sure we nuke the old versions
|
|
DROP FUNCTION create_item(text,text,text,integer,text);
|
|
|
|
CREATE OR REPLACE FUNCTION create_item(title TEXT, url TEXT, text TEXT, parent_id INTEGER, user_id INTEGER)
|
|
RETURNS "Item"
|
|
LANGUAGE plpgsql
|
|
AS $$
|
|
DECLARE
|
|
user_sats INTEGER;
|
|
free_posts INTEGER;
|
|
free_comments INTEGER;
|
|
freebie BOOLEAN;
|
|
item "Item";
|
|
BEGIN
|
|
PERFORM ASSERT_SERIALIZED();
|
|
|
|
SELECT (msats / 1000), "freePosts", "freeComments"
|
|
INTO user_sats, free_posts, free_comments
|
|
FROM users WHERE id = user_id;
|
|
|
|
freebie := (parent_id IS NULL AND free_posts > 0) OR (parent_id IS NOT NULL AND free_comments > 0);
|
|
|
|
IF NOT freebie AND 1 > user_sats THEN
|
|
RAISE EXCEPTION 'SN_INSUFFICIENT_FUNDS';
|
|
END IF;
|
|
|
|
INSERT INTO "Item" (title, url, text, "userId", "parentId", created_at, updated_at)
|
|
VALUES (title, url, text, user_id, parent_id, now_utc(), now_utc()) RETURNING * INTO item;
|
|
|
|
IF freebie THEN
|
|
IF parent_id IS NULL THEN
|
|
UPDATE users SET "freePosts" = "freePosts" - 1 WHERE id = user_id;
|
|
ELSE
|
|
UPDATE users SET "freeComments" = "freeComments" - 1 WHERE id = user_id;
|
|
END IF;
|
|
ELSE
|
|
UPDATE users SET msats = msats - 1000 WHERE id = user_id;
|
|
|
|
INSERT INTO "ItemAct" (sats, "itemId", "userId", act, created_at, updated_at)
|
|
VALUES (1, item.id, user_id, 'VOTE', now_utc(), now_utc());
|
|
END IF;
|
|
|
|
RETURN item;
|
|
END;
|
|
$$; |