This repository has been archived on 2024-03-03. You can view files and clone it, but cannot push or open issues or pull requests.
2021-05-09 16:02:23 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
2021-05-09 17:18:48 +00:00
|
|
|
"fmt"
|
|
|
|
|
2021-05-09 16:02:23 +00:00
|
|
|
"github.com/docker/docker/api/types/mount"
|
|
|
|
|
|
|
|
"github.com/nemunaire/minifaas/engine/docker"
|
|
|
|
)
|
|
|
|
|
|
|
|
type Job struct {
|
|
|
|
Image string
|
|
|
|
Cmd []string
|
|
|
|
DataMount bool
|
|
|
|
}
|
|
|
|
|
|
|
|
var jobs = map[string]Job{
|
|
|
|
"counter": {
|
|
|
|
Image: "alpine",
|
|
|
|
Cmd: []string{"sh", "-c", "touch /data/work_done; for i in `seq 10`; do echo $i; sleep 0.5; done"},
|
|
|
|
DataMount: true,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
func RunJob(jobtype string) (string, error) {
|
|
|
|
var mnts []mount.Mount
|
|
|
|
if jobs[jobtype].DataMount {
|
|
|
|
myVolume, err := docker.CreateVolumeDir("/data", false)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
mnts = append(mnts, *myVolume)
|
|
|
|
}
|
|
|
|
|
2021-05-09 16:02:48 +00:00
|
|
|
// Check if the image is here
|
|
|
|
hasimg, err := docker.HasImage(jobs[jobtype].Image)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
if !hasimg {
|
|
|
|
err = docker.PullImage(jobs[jobtype].Image)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-05-09 17:18:48 +00:00
|
|
|
n, err := CountRunningContainers(jobtype)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
if n > 0 {
|
|
|
|
return "", fmt.Errorf("Please wait, there is already a similar job running.")
|
|
|
|
}
|
|
|
|
|
2021-05-09 16:02:23 +00:00
|
|
|
return docker.Run(
|
|
|
|
jobtype,
|
|
|
|
jobs[jobtype].Image,
|
|
|
|
jobs[jobtype].Cmd,
|
|
|
|
mnts,
|
|
|
|
)
|
|
|
|
}
|