feat: support imageaccelerate

serverless-tencent · May 31, 2023 · 72156eb · 72156eb
1 parent c4acb51
commit 72156eb
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 28 deletions.
diff --git a/src/modules/scf/interface.ts b/src/modules/scf/interface.ts
@@ -12,6 +12,7 @@ export interface FunctionCode {
     RegistryId?: string;
     Command?: string;
     Args?: string;
+    ContainerImageAccelerate?: boolean;
   };
 }
 
@@ -69,11 +70,11 @@ export interface BaseFunctionConfig {
   NodeSpec?: string;
   SFType?: string;
   GpuReservedQuota?: number;
-  //请求并发
+  // 请求并发
   InstanceConcurrencyConfig?: {
-    DynamicEnabled?: boolean
-    MaxConcurrency?: number
-  }
+    DynamicEnabled?: boolean;
+    MaxConcurrency?: number;
+  };
 }
 
 export interface TriggerType {
@@ -235,6 +236,8 @@ export interface ScfCreateFunctionInputs {
     command?: string;
     // 启动命令参数
     args?: string;
+    // 镜像加速
+    containerImageAccelerate?: boolean;
   };
 
   // 异步调用重试配置
@@ -244,15 +247,15 @@ export interface ScfCreateFunctionInputs {
   protocolType?: string;
   protocolParams?: ProtocolParams;
 
-  //sd应用类型
-  sFType?: string
-  //gpu并发数，默认是0
-  gpuReservedQuota?: number
-  //请求并发配置
-  instanceConcurrencyConfig?:{
-    dynamicEnabled?: boolean
-    maxConcurrency?: number
-  }
+  // sd应用类型
+  sFType?: string;
+  // gpu并发数，默认是0
+  gpuReservedQuota?: number;
+  // 请求并发配置
+  instanceConcurrencyConfig?: {
+    dynamicEnabled?: boolean;
+    maxConcurrency?: number;
+  };
 }
 
 export interface ScfUpdateAliasTrafficInputs {
@@ -408,25 +411,25 @@ export interface GetRequestStatusOptions {
   /**
    * 函数名称
    */
-  functionName: string
+  functionName: string;
 
   /**
    * 需要查询状态的请求id
    */
-  functionRequestId: string
+  functionRequestId: string;
 
   /**
    * 函数的所在的命名空间
    */
-  namespace?: string
+  namespace?: string;
 
   /**
    * 查询的开始时间，例如：2017-05-16 20:00:00，不填默认为当前时间 - 15min
    */
-  startTime?: string
+  startTime?: string;
 
   /**
    * 查询的结束时间，例如：2017-05-16 20:59:59，不填默认为当前时间。EndTime 需要晚于 StartTime。
    */
-  endTime?: string
+  endTime?: string;
 }
diff --git a/src/modules/scf/utils.ts b/src/modules/scf/utils.ts
@@ -52,6 +52,11 @@ export const formatInputs = (inputs: ScfCreateFunctionInputs) => {
     if (imageConfig.args) {
       functionInputs.Code!.ImageConfig!.Args = imageConfig.args;
     }
+
+    if (imageConfig?.containerImageAccelerate !== undefined) {
+      functionInputs.Code!.ImageConfig!.ContainerImageAccelerate =
+        imageConfig.containerImageAccelerate;
+    }
   } else {
     // 基于 COS 代码部署
     functionInputs.Code = {
@@ -83,23 +88,22 @@ export const formatInputs = (inputs: ScfCreateFunctionInputs) => {
         functionInputs.ProtocolParams = protocolParams;
       }
     }
-
   }
-  
-  //sd应用类型
-  if(inputs?.sFType){
+
+  // sd应用类型
+  if (inputs?.sFType) {
     functionInputs.SFType = inputs?.sFType;
   }
-  //sd gpu并发数
-  if(inputs?.gpuReservedQuota){
+  // sd gpu并发数
+  if (inputs?.gpuReservedQuota) {
     functionInputs.GpuReservedQuota = inputs?.gpuReservedQuota;
   }
-  //请求并发
-  if(inputs?.instanceConcurrencyConfig){
+  // 请求并发
+  if (inputs?.instanceConcurrencyConfig) {
     functionInputs.InstanceConcurrencyConfig = {
       DynamicEnabled: inputs?.instanceConcurrencyConfig?.dynamicEnabled,
-      MaxConcurrency: inputs?.instanceConcurrencyConfig?.maxConcurrency
-    }
+      MaxConcurrency: inputs?.instanceConcurrencyConfig?.maxConcurrency,
+    };
   }
 
   if (inputs.role) {