feat: add resource_multiplier to function config

supabase · Nov 14, 2024 · 6a2da87 · 6a2da87
1 parent bbb06f1
commit 6a2da87
Show file tree

Hide file tree

Showing 4 changed files with 30 additions and 9 deletions.
diff --git a/api/beta.yaml b/api/beta.yaml
@@ -2012,6 +2012,11 @@ paths:
           in: query
           schema:
             type: string
+        - name: resource_multiplier
+          required: false
+          in: query
+          schema:
+            type: string
       requestBody:
         required: true
         content:
@@ -2153,6 +2158,11 @@ paths:
           in: query
           schema:
             type: string
+        - name: resource_multiplier
+          required: false
+          in: query
+          schema:
+            type: string
       requestBody:
         required: true
         content:
@@ -5090,6 +5100,8 @@ components:
           type: string
         verify_jwt:
           type: boolean
+        resource_multiplier:
+          type: string
       required:
         - slug
         - name
@@ -5125,6 +5137,8 @@ components:
           type: string
         import_map_path:
           type: string
+        resource_multiplier:
+          type: string
       required:
         - version
         - created_at
@@ -5164,6 +5178,8 @@ components:
           type: string
         import_map_path:
           type: string
+        resource_multiplier:
+          type: string
       required:
         - version
         - created_at

diff --git a/internal/functions/serve/templates/main.ts b/internal/functions/serve/templates/main.ts
@@ -149,7 +149,8 @@ Deno.serve({
     console.error(`serving the request with ${servicePath}`);
 
     // Ref: https://supabase.com/docs/guides/functions/limits
-    const memoryLimitMb = 256;
+    const resourceMultiplier = Math.max(Math.min(parseFloat(functionsConfig[functionName].resourceMultiplier ?? '1'), 4), 1)
+    const memoryLimitMb = 256 * resourceMultiplier;
     const workerTimeoutMs = isFinite(WALLCLOCK_LIMIT_SEC) ? WALLCLOCK_LIMIT_SEC * 1000 : 400 * 1000;
     const noModuleCache = false;
     const envVarsObj = Deno.env.toObject();
@@ -161,7 +162,7 @@ Deno.serve({
     const forceCreate = false;
     const customModuleRoot = ""; // empty string to allow any local path
     const cpuTimeSoftLimitMs = 1000;
-    const cpuTimeHardLimitMs = 2000;
+    const cpuTimeHardLimitMs = 2000 * resourceMultiplier;
 
     // NOTE(Nyannyacha): Decorator type has been set to tc39 by Lakshan's request,
     // but in my opinion, we should probably expose this to customers at some
@@ -187,7 +188,9 @@ Deno.serve({
         maybeEntrypoint
       });
 
-      return await worker.fetch(req);
+      const res = await worker.fetch(req);
+      res.headers.set('x-sb-resource-multiplier', resourceMultiplier)
+      return res
     } catch (e) {
       console.error(e);
 

diff --git a/pkg/config/config.go b/pkg/config/config.go
@@ -176,10 +176,11 @@ type (
 	FunctionConfig map[string]function
 
 	function struct {
-		Enabled    *bool  `toml:"enabled" json:"-"`
-		VerifyJWT  *bool  `toml:"verify_jwt" json:"verifyJWT"`
-		ImportMap  string `toml:"import_map" json:"importMapPath,omitempty"`
-		Entrypoint string `toml:"entrypoint" json:"entrypointPath,omitempty"`
+		Enabled            *bool   `toml:"enabled" json:"-"`
+		VerifyJWT          *bool   `toml:"verify_jwt" json:"verifyJWT"`
+		ImportMap          string  `toml:"import_map" json:"importMapPath,omitempty"`
+		Entrypoint         string  `toml:"entrypoint" json:"entrypointPath,omitempty"`
+		ResourceMultiplier *string `toml:"resource_multiplier" json:"resourceMultiplier,omitempty"`
 	}
 
 	analytics struct {

diff --git a/pkg/function/batch.go b/pkg/function/batch.go
@@ -44,6 +44,7 @@ func (s *EdgeRuntimeAPI) UpsertFunctions(ctx context.Context, functionConfig con
 				continue
 			}
 		}
+		resourceMultiplier := function.ResourceMultiplier
 		var body bytes.Buffer
 		if err := s.eszip.Bundle(ctx, function.Entrypoint, function.ImportMap, &body); err != nil {
 			return err
@@ -55,7 +56,7 @@ func (s *EdgeRuntimeAPI) UpsertFunctions(ctx context.Context, functionConfig con
 					VerifyJwt:          function.VerifyJWT,
 					ImportMapPath:      toFileURL(function.ImportMap),
 					EntrypointPath:     toFileURL(function.Entrypoint),
-					ResourceMultiplier: function.ResourceMultiplier,
+					ResourceMultiplier: resourceMultiplier,
 				}, eszipContentType, bytes.NewReader(body.Bytes())); err != nil {
 					return errors.Errorf("failed to update function: %w", err)
 				} else if resp.JSON200 == nil {
@@ -68,7 +69,7 @@ func (s *EdgeRuntimeAPI) UpsertFunctions(ctx context.Context, functionConfig con
 					VerifyJwt:          function.VerifyJWT,
 					ImportMapPath:      toFileURL(function.ImportMap),
 					EntrypointPath:     toFileURL(function.Entrypoint),
-					ResourceMultiplier: function.ResourceMultiplier,
+					ResourceMultiplier: resourceMultiplier,
 				}, eszipContentType, bytes.NewReader(body.Bytes())); err != nil {
 					return errors.Errorf("failed to create function: %w", err)
 				} else if resp.JSON201 == nil {