forked from vitessio/vitess
-
Notifications
You must be signed in to change notification settings - Fork 6
/
ping.go
87 lines (73 loc) · 2.21 KB
/
ping.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
76
77
78
79
80
81
82
83
84
85
86
87
// Copyright 2015, Google Inc. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package worker
import (
"html/template"
"golang.org/x/net/context"
"github.com/youtube/vitess/go/vt/wrangler"
)
// PingWorker will log a message with level CONSOLE.
type PingWorker struct {
StatusWorker
// We use the Wrangler's logger to print the message.
wr *wrangler.Wrangler
message string
}
// NewPingWorker returns a new PingWorker object.
func NewPingWorker(wr *wrangler.Wrangler, message string) (Worker, error) {
return &PingWorker{
StatusWorker: NewStatusWorker(),
wr: wr,
message: message,
}, nil
}
// StatusAsHTML implements the Worker interface
func (pw *PingWorker) StatusAsHTML() template.HTML {
pw.Mu.Lock()
defer pw.Mu.Unlock()
result := "<b>Ping Command with message:</b> '" + pw.message + "'</br>\n"
result += "<b>State:</b> " + pw.State.String() + "</br>\n"
switch pw.State {
case WorkerStateCopy:
result += "<b>Running</b>:</br>\n"
result += "Logging message: '" + pw.message + "'</br>\n"
case WorkerStateDone:
result += "<b>Success</b>:</br>\n"
result += "Logged message: '" + pw.message + "'</br>\n"
}
return template.HTML(result)
}
// StatusAsText implements the Worker interface.
func (pw *PingWorker) StatusAsText() string {
pw.Mu.Lock()
defer pw.Mu.Unlock()
result := "Ping Command with message: '" + pw.message + "'\n"
result += "State: " + pw.State.String() + "\n"
switch pw.State {
case WorkerStateCopy:
result += "Logging message: '" + pw.message + "'\n"
case WorkerStateDone:
result += "Logged message: '" + pw.message + "'\n"
}
return result
}
// Run implements the Worker interface.
func (pw *PingWorker) Run(ctx context.Context) error {
resetVars()
err := pw.run(ctx)
pw.SetState(WorkerStateCleanUp)
if err != nil {
pw.SetState(WorkerStateError)
return err
}
pw.SetState(WorkerStateDone)
return nil
}
func (pw *PingWorker) run(ctx context.Context) error {
// We reuse the Copy state to reflect that the logging is in progress.
pw.SetState(WorkerStateCopy)
pw.wr.Logger().Printf("Ping command was called with message: '%v'.\n", pw.message)
pw.SetState(WorkerStateDone)
return nil
}