11 "git.arvados.org/arvados.git/sdk/go/arvados"
12 "git.arvados.org/arvados.git/sdk/go/arvadosclient"
13 "git.arvados.org/arvados.git/sdk/go/keepclient"
14 "golang.org/x/crypto/blake2b"
17 type arvadosContainerRunner struct {
18 Client *arvados.Client
24 Mounts map[string]string
28 collectionInPathRe = regexp.MustCompile(`^(.*/)?([0-9a-f]{32}\+[0-9]+|[0-9a-z]{5}-[0-9a-z]{5}-[0-9a-z]{15})(/.*)?$`)
31 func (runner *arvadosContainerRunner) Run() error {
32 if runner.ProjectUUID == "" {
33 return errors.New("cannot run arvados container: ProjectUUID not provided")
35 prog := "/mnt/cmd/lightning"
36 cmdUUID, err := runner.makeCommandCollection()
40 command := append([]string{prog}, runner.Args...)
41 mounts := map[string]map[string]interface{}{
49 "capacity": 100000000000,
52 for uuid, mnt := range runner.Mounts {
53 mounts[mnt] = map[string]interface{}{
58 rc := arvados.RuntimeConstraints{
61 KeepCacheRAM: (1 << 26) * 2 * int64(runner.VCPUs),
63 var cr arvados.ContainerRequest
64 err = runner.Client.RequestAndDecode(&cr, "POST", "arvados/v1/container_requests", nil, map[string]interface{}{
65 "container_request": map[string]interface{}{
66 "owner_uuid": runner.ProjectUUID,
68 "container_image": "lightning-runtime",
72 "output_path": "/mnt/output",
73 "runtime_constraints": rc,
75 "state": arvados.ContainerRequestStateCommitted,
82 func (runner *arvadosContainerRunner) TranslatePaths(paths ...*string) error {
83 if runner.Mounts == nil {
84 runner.Mounts = make(map[string]string)
86 for _, path := range paths {
90 m := collectionInPathRe.FindStringSubmatch(*path)
92 return fmt.Errorf("cannot find uuid in path: %q", *path)
95 mnt, ok := runner.Mounts[uuid]
98 runner.Mounts[uuid] = mnt
105 func (runner *arvadosContainerRunner) makeCommandCollection() (string, error) {
106 exe, err := ioutil.ReadFile("/proc/self/exe")
110 b2 := blake2b.Sum256(exe)
111 cname := fmt.Sprintf("lightning-%x", b2)
112 var existing arvados.CollectionList
113 err = runner.Client.RequestAndDecode(&existing, "GET", "arvados/v1/collections", nil, arvados.ListOptions{
116 Filters: []arvados.Filter{
117 {Attr: "name", Operator: "=", Operand: cname},
118 {Attr: "owner_uuid", Operator: "=", Operand: runner.ProjectUUID},
124 if len(existing.Items) > 0 {
125 uuid := existing.Items[0].UUID
126 log.Printf("using existing collection %q named %q (did not verify whether content matches)", uuid, cname)
129 log.Printf("writing lightning binary to new collection %q", cname)
130 ac, err := arvadosclient.New(runner.Client)
134 kc := keepclient.New(ac)
135 var coll arvados.Collection
136 fs, err := coll.FileSystem(runner.Client, kc)
140 f, err := fs.OpenFile("lightning", os.O_CREATE|os.O_WRONLY, 0777)
144 _, err = f.Write(exe)
152 mtxt, err := fs.MarshalManifest(".")
156 err = runner.Client.RequestAndDecode(&coll, "POST", "arvados/v1/collections", nil, map[string]interface{}{
157 "collection": map[string]interface{}{
158 "owner_uuid": runner.ProjectUUID,
159 "manifest_text": mtxt,
166 log.Printf("collection: %#v", coll)
167 return coll.UUID, nil