-
Notifications
You must be signed in to change notification settings - Fork 22
/
sessionhandle.go
75 lines (67 loc) · 1.51 KB
/
sessionhandle.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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
package kickvmhandle
import (
"errors"
"github.com/aliyun/aliyun_assist_client/agent/log"
"github.com/aliyun/aliyun_assist_client/agent/taskengine"
)
var sessionRoute map[string]handleFunc
func init () {
sessionRoute = map[string]handleFunc{
"start": startSession,
"stop": stopSession,
}
}
func stopSession(params []string) error {
go func() {
if len(params) < 1 {
log.GetLogger().Errorln("params invalid", params)
return
}
ret := taskengine.GetSessionFactory().ContainsTask(params[0])
if ret == true {
log.GetLogger().Println("stop session ", params[0])
task,_ := taskengine.GetSessionFactory().GetTask(params[0])
task.StopTask()
} else {
log.GetLogger().Errorln("stop session failed")
}
}()
return nil
}
func startSession(params []string) error {
go func() {
// kick_vm session start task_id
// kick_vm session stop task_id
if len(params) < 1 {
log.GetLogger().Errorln("params invalid", params)
return
}
taskengine.Fetch(true, params[0], taskengine.SessionTaskType)
}()
return nil
}
type SessionHandle struct {
action string
params []string
}
func NewSessionHandle(action string, params []string) *SessionHandle{
return &SessionHandle{
action: action,
params: params,
}
}
func (h *SessionHandle) DoAction() error{
if v, ok := sessionRoute[h.action]; ok {
v(h.params)
} else {
return errors.New("no action found")
}
return nil
}
func (h *SessionHandle) CheckAction() bool{
if _, ok := sessionRoute[h.action]; ok {
return true
} else {
return false
}
}