feat(db): add team delete RPS fields

GitOrigin-RevId: b58e7fd40d714529d47c540997ae33d6d63f91c8

Author: nalekseev-e2b

packages/auth/pkg/types/limits.go

@@ -14,5 +14,6 @@ type TeamLimits struct {
 	DefaultFreeDiskSizeMb int64
 	MaxFreeDiskSizeMb     int64
 
-	APITeamRPSList int64
+	APITeamRPSList   int64
+	APITeamRPSDelete int64
 }

packages/auth/pkg/types/teams.go

@@ -26,6 +26,7 @@ func newTeamLimits(
 		DiskMb:             teamLimits.DiskMb,
 		EventsTTLDays:      teamLimits.EventsTtlDays,
 		APITeamRPSList:     teamLimits.ApiTeamRpsList,
+		APITeamRPSDelete:   teamLimits.ApiTeamRpsDelete,
 
 		DefaultFreeDiskSizeMb: teamLimits.DefaultFreeDiskSizeMb,
 		MaxFreeDiskSizeMb:     teamLimits.MaxFreeDiskSizeMb,

packages/auth/pkg/types/teams_test.go

@@ -8,9 +8,10 @@ import (
 	authqueries "github.com/e2b-dev/infra/packages/db/pkg/auth/queries"
 )
 
-func TestNewTeamPreservesAPIListRate(t *testing.T) {
+func TestNewTeamPreservesAPIRates(t *testing.T) {
 	t.Parallel()
 
-	team := NewTeam(&authqueries.Team{}, &authqueries.TeamLimit{ApiTeamRpsList: 1 << 32})
+	team := NewTeam(&authqueries.Team{}, &authqueries.TeamLimit{ApiTeamRpsList: 1 << 32, ApiTeamRpsDelete: 1<<32 + 1})
 	assert.Equal(t, int64(1<<32), team.Limits.APITeamRPSList)
+	assert.Equal(t, int64(1<<32+1), team.Limits.APITeamRPSDelete)
 }

packages/db/migrations/20261005085240_add_api_delete_rate_limit.sql

@@ -0,0 +1,86 @@
+-- +goose Up
+-- +goose StatementBegin
+ALTER TABLE public.tiers
+    ADD COLUMN api_team_rps_delete bigint NOT NULL DEFAULT 0
+        CHECK (api_team_rps_delete >= 0);
+
+ALTER TABLE public.addons
+    ADD COLUMN extra_api_team_rps_delete bigint NOT NULL DEFAULT 0
+        CHECK (extra_api_team_rps_delete >= 0);
+
+CREATE OR REPLACE VIEW public.team_limits
+WITH (security_invoker=on) AS
+SELECT
+    t.id,
+    COALESCE(pl.max_length_hours, tier.max_length_hours) AS max_length_hours,
+    COALESCE(pl.concurrent_sandboxes, tier.concurrent_instances + a.extra_concurrent_sandboxes) AS concurrent_sandboxes,
+    COALESCE(pl.concurrent_template_builds, tier.concurrent_template_builds + a.extra_concurrent_template_builds) AS concurrent_template_builds,
+    COALESCE(pl.max_vcpu, tier.max_vcpu + a.extra_max_vcpu) AS max_vcpu,
+    COALESCE(pl.max_ram_mb, tier.max_ram_mb + a.extra_max_ram_mb) AS max_ram_mb,
+    COALESCE(pl.disk_mb, tier.disk_mb + a.extra_disk_mb) AS disk_mb,
+    COALESCE(pl.events_ttl_days, tier.events_ttl_days + a.extra_events_ttl_days) AS events_ttl_days,
+    COALESCE(pl.default_free_disk_size_mb, (tier.default_free_disk_size_mb + a.extra_disk_mb))::bigint AS default_free_disk_size_mb,
+    COALESCE(pl.max_disk_size_mb, (tier.max_disk_size_mb + a.extra_max_disk_size_mb))::bigint AS max_disk_size_mb,
+    COALESCE(pl.max_disk_size_mb, (tier.max_disk_size_mb + a.extra_max_disk_size_mb))::bigint AS max_free_disk_size_mb,
+    COALESCE(pl.api_team_rps_list, tier.api_team_rps_list + a.extra_api_team_rps_list)::bigint AS api_team_rps_list,
+    (tier.api_team_rps_delete + a.extra_api_team_rps_delete)::bigint AS api_team_rps_delete
+FROM public.teams t
+JOIN public.tiers tier ON t.tier = tier.id
+LEFT JOIN public.project_limits pl ON pl.team_id = t.id
+LEFT JOIN LATERAL (
+    SELECT COALESCE(SUM(extra_concurrent_sandboxes), 0)::bigint AS extra_concurrent_sandboxes,
+           COALESCE(SUM(extra_concurrent_template_builds), 0)::bigint AS extra_concurrent_template_builds,
+           COALESCE(SUM(extra_max_vcpu), 0)::bigint AS extra_max_vcpu,
+           COALESCE(SUM(extra_max_ram_mb), 0)::bigint AS extra_max_ram_mb,
+           COALESCE(SUM(extra_disk_mb), 0)::bigint AS extra_disk_mb,
+           COALESCE(SUM(extra_events_ttl_days), 0)::bigint AS extra_events_ttl_days,
+           COALESCE(SUM(COALESCE(extra_max_disk_size_mb, extra_disk_mb)), 0)::bigint AS extra_max_disk_size_mb,
+           COALESCE(SUM(extra_api_team_rps_list), 0)::bigint AS extra_api_team_rps_list,
+           COALESCE(SUM(extra_api_team_rps_delete), 0)::bigint AS extra_api_team_rps_delete
+    FROM public.addons addon
+    WHERE addon.team_id = t.id
+      AND addon.valid_from <= now()
+      AND (addon.valid_to IS NULL OR addon.valid_to > now())
+) a ON true;
+-- +goose StatementEnd
+
+-- +goose Down
+-- +goose StatementBegin
+DROP VIEW public.team_limits;
+
+CREATE VIEW public.team_limits
+WITH (security_invoker=on) AS
+SELECT
+    t.id,
+    COALESCE(pl.max_length_hours, tier.max_length_hours) AS max_length_hours,
+    COALESCE(pl.concurrent_sandboxes, tier.concurrent_instances + a.extra_concurrent_sandboxes) AS concurrent_sandboxes,
+    COALESCE(pl.concurrent_template_builds, tier.concurrent_template_builds + a.extra_concurrent_template_builds) AS concurrent_template_builds,
+    COALESCE(pl.max_vcpu, tier.max_vcpu + a.extra_max_vcpu) AS max_vcpu,
+    COALESCE(pl.max_ram_mb, tier.max_ram_mb + a.extra_max_ram_mb) AS max_ram_mb,
+    COALESCE(pl.disk_mb, tier.disk_mb + a.extra_disk_mb) AS disk_mb,
+    COALESCE(pl.events_ttl_days, tier.events_ttl_days + a.extra_events_ttl_days) AS events_ttl_days,
+    COALESCE(pl.default_free_disk_size_mb, (tier.default_free_disk_size_mb + a.extra_disk_mb))::bigint AS default_free_disk_size_mb,
+    COALESCE(pl.max_disk_size_mb, (tier.max_disk_size_mb + a.extra_max_disk_size_mb))::bigint AS max_disk_size_mb,
+    COALESCE(pl.max_disk_size_mb, (tier.max_disk_size_mb + a.extra_max_disk_size_mb))::bigint AS max_free_disk_size_mb,
+    COALESCE(pl.api_team_rps_list, tier.api_team_rps_list + a.extra_api_team_rps_list)::bigint AS api_team_rps_list
+FROM public.teams t
+JOIN public.tiers tier ON t.tier = tier.id
+LEFT JOIN public.project_limits pl ON pl.team_id = t.id
+LEFT JOIN LATERAL (
+    SELECT COALESCE(SUM(extra_concurrent_sandboxes), 0)::bigint AS extra_concurrent_sandboxes,
+           COALESCE(SUM(extra_concurrent_template_builds), 0)::bigint AS extra_concurrent_template_builds,
+           COALESCE(SUM(extra_max_vcpu), 0)::bigint AS extra_max_vcpu,
+           COALESCE(SUM(extra_max_ram_mb), 0)::bigint AS extra_max_ram_mb,
+           COALESCE(SUM(extra_disk_mb), 0)::bigint AS extra_disk_mb,
+           COALESCE(SUM(extra_events_ttl_days), 0)::bigint AS extra_events_ttl_days,
+           COALESCE(SUM(COALESCE(extra_max_disk_size_mb, extra_disk_mb)), 0)::bigint AS extra_max_disk_size_mb,
+           COALESCE(SUM(extra_api_team_rps_list), 0)::bigint AS extra_api_team_rps_list
+    FROM public.addons addon
+    WHERE addon.team_id = t.id
+      AND addon.valid_from <= now()
+      AND (addon.valid_to IS NULL OR addon.valid_to > now())
+) a ON true;
+
+ALTER TABLE public.addons DROP COLUMN extra_api_team_rps_delete;
+ALTER TABLE public.tiers DROP COLUMN api_team_rps_delete;
+-- +goose StatementEnd