fix: balance snapshot indexes

This should help with the usage snapshot queries.

Author: turip

openmeter/ent/db/migrate/schema.go

@@ -3,6 +3,7 @@ package schema
 import (
 	"entgo.io/ent"
 	"entgo.io/ent/dialect"
+	"entgo.io/ent/dialect/entsql"
 	"entgo.io/ent/schema/edge"
 	"entgo.io/ent/schema/field"
 	"entgo.io/ent/schema/index"
@@ -45,9 +46,9 @@ func (BalanceSnapshot) Fields() []ent.Field {
 
 func (BalanceSnapshot) Indexes() []ent.Index {
 	return []ent.Index{
-		index.Fields("namespace", "at"),
-		index.Fields("namespace", "balance"),
-		index.Fields("namespace", "balance", "at"),
+		index.Fields("namespace", "owner_id", "at").Annotations(
+			entsql.IndexWhere("deleted_at IS NULL"),
+		),
 	}
 }
 

openmeter/ent/schema/balance_snapshot.go

@@ -0,0 +1,8 @@
+-- reverse: create index "balancesnapshot_namespace_owner_id_at" to table: "balance_snapshots"
+DROP INDEX "balancesnapshot_namespace_owner_id_at";
+-- reverse: drop index "balancesnapshot_namespace_balance_at" from table: "balance_snapshots"
+CREATE INDEX "balancesnapshot_namespace_balance_at" ON "balance_snapshots" ("namespace", "balance", "at");
+-- reverse: drop index "balancesnapshot_namespace_balance" from table: "balance_snapshots"
+CREATE INDEX "balancesnapshot_namespace_balance" ON "balance_snapshots" ("namespace", "balance");
+-- reverse: drop index "balancesnapshot_namespace_at" from table: "balance_snapshots"
+CREATE INDEX "balancesnapshot_namespace_at" ON "balance_snapshots" ("namespace", "at");

tools/migrate/migrations/20250311193204_balance-snapshot-indexes.down.sql

@@ -0,0 +1,8 @@
+-- drop index "balancesnapshot_namespace_at" from table: "balance_snapshots"
+DROP INDEX "balancesnapshot_namespace_at";
+-- drop index "balancesnapshot_namespace_balance" from table: "balance_snapshots"
+DROP INDEX "balancesnapshot_namespace_balance";
+-- drop index "balancesnapshot_namespace_balance_at" from table: "balance_snapshots"
+DROP INDEX "balancesnapshot_namespace_balance_at";
+-- create index "balancesnapshot_namespace_owner_id_at" to table: "balance_snapshots"
+CREATE INDEX "balancesnapshot_namespace_owner_id_at" ON "balance_snapshots" ("namespace", "owner_id", "at") WHERE (deleted_at IS NULL);

tools/migrate/migrations/20250311193204_balance-snapshot-indexes.up.sql

@@ -1,4 +1,4 @@
-h1:GBSDL939Zna5rKl2wFhpUqGQJUHddDKFReBELgNGbV0=
+h1:fDFD8PZ8j8vPRxmMYjAeFv9ai8AXG8pP7VA0oDQL2Hg=
 20240826120919_init.down.sql h1:AIbgwwngjkJEYa3yRZsIXQyBa2+qoZttwMXHxXEbHLI=
 20240826120919_init.up.sql h1:/hYHWF3Z3dab8SMKnw99ixVktCuJe2bAw5wstCZIEN8=
 20240903155435_entitlement-expired-index.down.sql h1:np2xgYs3KQ2z7qPBcobtGNhqWQ3V8NwEP9E5U3TmpSA=
@@ -129,3 +129,5 @@ h1:GBSDL939Zna5rKl2wFhpUqGQJUHddDKFReBELgNGbV0=
 20250306040407_resource_meter.up.sql h1:L0bhDCSzStPWsIvx6hULDKK7eurpBD3bIv9b18mSWqc=
 20250307151454_balance-snapshot-usage.down.sql h1:UprCI9cWSumzY1NXmlO055r65VEd2M8+vvMzkJc6Q3g=
 20250307151454_balance-snapshot-usage.up.sql h1:j2x/+rYBLsHNpUCmeKwvyXe1TwXTSDMlddESgUFUqMM=
+20250311193204_balance-snapshot-indexes.down.sql h1:8EImnDEROfVw7dS2S6iIsbKH7L2VKBPdjKmfXV4bv4c=
+20250311193204_balance-snapshot-indexes.up.sql h1:pOm1/hjT5JNIKY6HGoE9cKdm7sEktvGoe70aXIXdtas=