kubeflow
/
examples
mirror of https://github.com/kubeflow/examples.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Refactor job and deployment specs into different functions

This commit is contained in:
Sanyam Kapoor 2018-08-08 17:24:43 -07:00
parent e34f9aca75
commit 133e054033
1 changed files with 64 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

95
code_search/kubeflow/components/nms.libsonnet
View File

@ -1,9 +1,9 @@
local baseParams = std.extVar("__ksonnet/params").components["nmslib"];
{
spec(params, env, apiVersion="extensions/v1beta1", kind="Deployment"):: {
apiVersion: apiVersion,
kind: kind,
deploymentSpec(params, env, containers):: {
apiVersion: "extensions/v1beta1",
kind: "Deployment",
metadata: {
name: params.name,
namespace: env.namespace,
@ -13,7 +13,7 @@ local baseParams = std.extVar("__ksonnet/params").components["nmslib"];
},
spec: {
replicas: params.replicas,
[if kind == "Deployment" then "selector"]: {
selector: {
matchLabels: {
app: params.name,
},
@ -25,31 +25,7 @@ local baseParams = std.extVar("__ksonnet/params").components["nmslib"];
}
},
spec: {
[if kind == "Job" then "restartPolicy"]: "OnFailure",
containers: [
{
name: params.name,
image: params.image,
args: params.args,
ports: [
{
containerPort: 8008,
}
],
env: [
{
name: "GOOGLE_APPLICATION_CREDENTIALS",
value: "/secret/gcp-credentials/user-gcp-sa.json",
}
],
volumeMounts: [
{
mountPath: "/secret/gcp-credentials",
name: "gcp-credentials",
},
],
}
],
containers: containers,
volumes: [
{
name: "gcp-credentials",
@ -63,6 +39,63 @@ local baseParams = std.extVar("__ksonnet/params").components["nmslib"];
},
},
jobSpec(params, env, containers):: {
apiVersion: "batch/v1",
kind: "Job",
metadata: {
name: params.name,
namespace: env.namespace,
labels: {
app: params.name,
}
},
spec: {
replicas: params.replicas,
template: {
metadata: {
labels: {
app: params.name,
}
},
spec: {
"restartPolicy": "OnFailure",
containers: containers,
volumes: [
{
name: "gcp-credentials",
secret: {
secretName: "user-gcp-sa",
},
},
],
},
},
},
},
containerSpec(params):: {
name: params.name,
image: params.image,
args: params.args,
ports: [
{
containerPort: 8008,
}
],
env: [
{
name: "GOOGLE_APPLICATION_CREDENTIALS",
value: "/secret/gcp-credentials/user-gcp-sa.json",
}
],
volumeMounts: [
{
mountPath: "/secret/gcp-credentials",
name: "gcp-credentials",
},
],
},
service(params, env):: {
apiVersion: "v1",
kind: "Service",
@ -116,7 +149,7 @@ local baseParams = std.extVar("__ksonnet/params").components["nmslib"];
},
all: [
$.spec(creatorParams, env, apiVersion="batch/v1", kind="Job"),
$.jobSpec(creatorParams, env, [ $.containerSpec(creatorParams) ]),
],
}.all,
@ -135,7 +168,7 @@ local baseParams = std.extVar("__ksonnet/params").components["nmslib"];
all: [
$.service(serverParams, env),
$.spec(serverParams, env),
$.deploymentSpec(serverParams, env, [ $.containerSpec(serverParams) ]),
],
}.all,
}