Skip to content

Commit 52a7115

Browse files
authored
refactor(ipc)!: merger start and stop power-save-mode api (#48)
Signed-off-by: Kevin Cui <[email protected]>
1 parent 2573e6c commit 52a7115

File tree

1 file changed

+13
-15
lines changed

1 file changed

+13
-15
lines changed

pkg/ipc/restful/restful.go

+13-15
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,10 @@ func New(vz *vz.VirtualMachine, vmC *config.VirtualMachine, log *logger.Context,
4646
}
4747
}
4848

49+
type powerSaveModeBody struct {
50+
Enable bool `json:"enable"`
51+
}
52+
4953
func (s *Restful) mux() *http.ServeMux {
5054
mux := http.NewServeMux()
5155
mux.HandleFunc("/info", func(w http.ResponseWriter, r *http.Request) {
@@ -104,21 +108,20 @@ func (s *Restful) mux() *http.ServeMux {
104108
http.Error(w, err.Error(), http.StatusInternalServerError)
105109
}
106110
})
107-
mux.HandleFunc("/start-power-save-mode", func(w http.ResponseWriter, r *http.Request) {
111+
mux.HandleFunc("/power-save-mode", func(w http.ResponseWriter, r *http.Request) {
108112
if r.Method != http.MethodPut {
109113
http.Error(w, "put only", http.StatusBadRequest)
110114
return
111115
}
112116

113-
s.startPowerSaveMode()
114-
})
115-
mux.HandleFunc("/stop-power-save-mode", func(w http.ResponseWriter, r *http.Request) {
116-
if r.Method != http.MethodPut {
117-
http.Error(w, "put only", http.StatusBadRequest)
117+
var body powerSaveModeBody
118+
if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
119+
s.log.Warnf("Failed to decode request body: %v", err)
120+
http.Error(w, "failed to decode request body", http.StatusBadRequest)
118121
return
119122
}
120123

121-
s.stopPowerSaveMode()
124+
s.powerSaveMode(body.Enable)
122125
})
123126

124127
return mux
@@ -200,12 +203,7 @@ func (s *Restful) stop() error {
200203
return err
201204
}
202205

203-
func (s *Restful) startPowerSaveMode() {
204-
s.log.Info("request /startPowerSaveMode")
205-
s.opt.PowerSaveMode = true
206-
}
207-
208-
func (s *Restful) stopPowerSaveMode() {
209-
s.log.Info("request /stopPowerSaveMode")
210-
s.opt.PowerSaveMode = false
206+
func (s *Restful) powerSaveMode(enable bool) {
207+
s.log.Info("request /powerSaveMode")
208+
s.opt.PowerSaveMode = enable
211209
}

0 commit comments

Comments
 (0)