PostgreSQLでカウンターを作る

Webで連番のカウンターが必要な用件があったので実装を検討しました。普段使っているPostgreSQLの上にテーブルを作り、カウンターを実装しました。１追加してUPDATEした値が返ってきます。

テーブル実装

Create

テーブルを作成します。

create table mycounter(
 count_key varchar(16) PRIMARY KEY,
 count int8
);

Count

以下のようなQueryで一致するcount_keyが無い場合は1をセット、存在する場合はカウントアップをして、resultを返します。

INSERT INTO mycounter (count_key, count) VALUES ('MYCOUNTERKEY', 1)
ON CONFLICT ON CONSTRAINT mycounter_pkey
DO UPDATE SET count = mycounter.count + 1 RETURNING *;

シーケンスを使う

シーケンスを使っても連番を取得する事ができます。こちらの方がIDなどに使えるので応用がきくかもしれません。

Create

CREATE SEQUENCE my_seq; -- 指定無しで1 

Count等

select nextval('my_seq');  --次の値
select currval('my_seq');  --現在の値 
select setval('my_seq', 1);  --1に設定 

むにゃむにゃ

count_keyを変更する事で複数のカウンターを実装したり、日付別や月別での連番を生成する事ができるので、何かと役に立つかと思います。