Add payers table

xmtp · Feb 21, 2025 · 1758eb6 · 1758eb6
1 parent 4e86897
commit 1758eb6
Show file tree

Hide file tree

Showing 6 changed files with 61 additions and 0 deletions.
diff --git a/pkg/db/queries.sql b/pkg/db/queries.sql
@@ -171,3 +171,12 @@ FROM (
 WHERE
 	bm.block_number = locked_rows.block_number;
 
+-- name: FindOrCreatePayer :one
+INSERT INTO payers(address)
+	VALUES (@address)
+ON CONFLICT (address)
+	DO UPDATE SET
+		address = @address
+	RETURNING
+		id;
+
diff --git a/pkg/db/queries/models.go b/pkg/db/queries/models.go
diff --git a/pkg/db/queries/queries.sql.go b/pkg/db/queries/queries.sql.go
diff --git a/pkg/db/queries_test.go b/pkg/db/queries_test.go
@@ -135,3 +135,26 @@ func TestRevokeAddressLog(t *testing.T) {
 	require.NotNil(t, addressLog)
 	require.Equal(t, addressLog.AssociationSequenceID.Int64, int64(3))
 }
+
+func TestFindOrCreatePayer(t *testing.T) {
+	ctx := context.Background()
+	db, _, cleanup := testutils.NewDB(t, ctx)
+	defer cleanup()
+
+	querier := queries.New(db)
+
+	address1 := testutils.RandomString(42)
+	address2 := testutils.RandomString(42)
+
+	id1, err := querier.FindOrCreatePayer(ctx, address1)
+	require.NoError(t, err)
+
+	id2, err := querier.FindOrCreatePayer(ctx, address2)
+	require.NoError(t, err)
+
+	require.NotEqual(t, id1, id2)
+
+	reinsertId, err := querier.FindOrCreatePayer(ctx, address1)
+	require.NoError(t, err)
+	require.Equal(t, id1, reinsertId)
+}
diff --git a/pkg/migrations/00005_add-payers-table.down.sql b/pkg/migrations/00005_add-payers-table.down.sql
@@ -0,0 +1,2 @@
+DROP TABLE IF EXISTS payers;
+
diff --git a/pkg/migrations/00005_add-payers-table.up.sql b/pkg/migrations/00005_add-payers-table.up.sql
@@ -0,0 +1,5 @@
+CREATE TABLE payers(
+	id SERIAL PRIMARY KEY,
+	address TEXT NOT NULL UNIQUE
+);
+