Skip to content

Instantly share code, notes, and snippets.

@abenezerBelachew
Created April 26, 2024 02:00
Show Gist options
  • Save abenezerBelachew/6387760ecddef2a5b27c30c22ed992d3 to your computer and use it in GitHub Desktop.
Save abenezerBelachew/6387760ecddef2a5b27c30c22ed992d3 to your computer and use it in GitHub Desktop.
CREATE OR REPLACE FUNCTION insta5.next_id(OUT result bigint) AS $$
DECLARE
our_epoch bigint := 1314220021721;
seq_id bigint;
now_millis bigint;
shard_id int := 5;
BEGIN
SELECT nextval('insta5.table_id_seq') %% 1024 INTO seq_id;
SELECT FLOOR(EXTRACT(EPOCH FROM clock_timestamp()) * 1000) INTO now_millis;
result := (now_millis - our_epoch) << 23;
result := result | (shard_id <<10);
result := result | (seq_id);
END;
$$ LANGUAGE PLPGSQL;
@abenezerBelachew
Copy link
Author

CREATE TABLE insta5.our_table (
"id" bigint NOT NULL DEFAULT insta5.next_id(),
...rest of table schema...
)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment