forked from apptainer/singularity
/
instance_stop.go
60 lines (49 loc) · 1.96 KB
/
instance_stop.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
// Copyright (c) 2018, Sylabs Inc. All rights reserved.
// This software is licensed under a 3-clause BSD license. Please consult the
// LICENSE.md file distributed with the sources of this project regarding your
// rights to use or distribute this software.
// +build linux
package cli
import (
"errors"
"github.com/spf13/cobra"
"github.com/sylabs/singularity/src/docs"
)
func init() {
InstanceStopCmd.Flags().SetInterspersed(false)
// -u|--user
InstanceStopCmd.Flags().StringVarP(&username, "user", "u", "", `if running as root, list instances from "<username>"`)
InstanceStopCmd.Flags().SetAnnotation("user", "argtag", []string{"<username>"})
InstanceStopCmd.Flags().SetAnnotation("user", "envkey", []string{"USER"})
// -a|--all
InstanceStopCmd.Flags().BoolVarP(&stopAll, "all", "a", false, "stop all user's instances")
InstanceStopCmd.Flags().SetAnnotation("all", "envkey", []string{"ALL"})
// -f|--force
InstanceStopCmd.Flags().BoolVarP(&forceStop, "force", "F", false, "force kill instance")
InstanceStopCmd.Flags().SetAnnotation("force", "envkey", []string{"FORCE"})
// -s|--signal
InstanceStopCmd.Flags().StringVarP(&stopSignal, "signal", "s", "", "signal sent to the instance")
InstanceStopCmd.Flags().SetAnnotation("signal", "argtag", []string{"<signal>"})
InstanceStopCmd.Flags().SetAnnotation("signal", "envkey", []string{"SIGNAL"})
// -t|--timeout
InstanceStopCmd.Flags().IntVarP(&stopTimeout, "timeout", "t", 10, "force kill non stopped instances after X seconds")
}
// InstanceStopCmd singularity instance stop
var InstanceStopCmd = &cobra.Command{
DisableFlagsInUseLine: true,
RunE: func(cmd *cobra.Command, args []string) error {
if len(args) > 0 && !stopAll {
stopInstance(args[0])
return nil
} else if stopAll {
stopInstance("*")
return nil
} else {
return errors.New("Invalid command")
}
},
Use: docs.InstanceStopUse,
Short: docs.InstanceStopShort,
Long: docs.InstanceStopLong,
Example: docs.InstanceStopExample,
}