feat: add resource instance

bentoml · Apr 27, 2023 · 044066b · 044066b
1 parent 70d0804
commit 044066b
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/modelschemas/cluster.go b/modelschemas/cluster.go
@@ -13,6 +13,7 @@ type ClusterConfigSchema struct {
 	DefaultDeploymentKubeNamespace string                  `json:"default_deployment_kube_namespace"`
 	IngressIp                      string                  `json:"ingress_ip"`
 	AWS                            *ClusterConfigAWSSchema `json:"aws"`
+	ResourceInstances              []ResourceInstance      `json:"resource_instances"`
 }
 
 func (c *ClusterConfigSchema) Scan(value interface{}) error {

diff --git a/modelschemas/deployment_target.go b/modelschemas/deployment_target.go
@@ -168,6 +168,7 @@ type RunnerBentoDeploymentOverrides struct {
 }
 
 type DeploymentTargetRunnerConfig struct {
+	ResourceInstance                       *string                         `json:"resource_instance,omitempty"`
 	Resources                              *DeploymentTargetResources      `json:"resources,omitempty"`
 	HPAConf                                *DeploymentTargetHPAConf        `json:"hpa_conf,omitempty"`
 	Envs                                   *[]*LabelItemSchema             `json:"envs,omitempty"`
@@ -227,6 +228,7 @@ type RequestQueueConfig struct {
 type DeploymentTargetConfig struct {
 	KubeResourceUid                        string                                  `json:"kubeResourceUid"`
 	KubeResourceVersion                    string                                  `json:"kubeResourceVersion"`
+	ResourceInstance                       *string                                 `json:"resource_instance,omitempty"`
 	Resources                              *DeploymentTargetResources              `json:"resources"`
 	HPAConf                                *DeploymentTargetHPAConf                `json:"hpa_conf,omitempty"`
 	Envs                                   *[]*LabelItemSchema                     `json:"envs,omitempty"`

diff --git a/modelschemas/resource_instance.go b/modelschemas/resource_instance.go
@@ -0,0 +1,10 @@
+package modelschemas
+
+type ResourceInstance struct {
+	ID            string                    `json:"id"`
+	Name          string                    `json:"name"`
+	Group         string                    `json:"group"`
+	Description   string                    `json:"description"`
+	NodeSelectors map[string]string         `json:"node_selectors"`
+	Resources     DeploymentTargetResources `json:"resources"`
+}