17
17
package com .navercorp .pinpoint .collector .manage .controller ;
18
18
19
19
import com .navercorp .pinpoint .collector .manage .HandlerManager ;
20
+ import com .navercorp .pinpoint .common .server .response .MapResponse ;
21
+ import com .navercorp .pinpoint .common .server .response .Response ;
22
+ import com .navercorp .pinpoint .common .server .response .Result ;
20
23
import org .springframework .http .HttpStatus ;
21
24
import org .springframework .http .ResponseEntity ;
22
25
import org .springframework .web .bind .annotation .GetMapping ;
@@ -39,10 +42,10 @@ public HandlerManagerController(HandlerManager handlerManager) {
39
42
}
40
43
41
44
@ GetMapping (value = "/enableAccess" )
42
- public ResponseEntity <SimpleResponse > enableAccess () {
45
+ public ResponseEntity <Response > enableAccess () {
43
46
try {
44
47
handlerManager .enableAccess ();
45
- return ResponseEntity .ok (SimpleResponse . success ());
48
+ return ResponseEntity .ok (MapResponse . ok ());
46
49
} catch (Exception e ) {
47
50
return unauthorizedResponse (e .getMessage ());
48
51
}
@@ -51,27 +54,27 @@ public ResponseEntity<SimpleResponse> enableAccess() {
51
54
52
55
53
56
@ GetMapping (value = "/disableAccess" )
54
- public ResponseEntity <SimpleResponse > disableAccess () {
57
+ public ResponseEntity <Response > disableAccess () {
55
58
try {
56
59
handlerManager .disableAccess ();
57
- return ResponseEntity .ok (SimpleResponse . success ());
60
+ return ResponseEntity .ok (MapResponse . ok ());
58
61
} catch (Exception e ) {
59
62
return unauthorizedResponse (e .getMessage ());
60
63
}
61
64
}
62
65
63
66
@ GetMapping (value = "/isEnable" )
64
- public ResponseEntity <SimpleResponse > isEnable () {
67
+ public ResponseEntity <Response > isEnable () {
65
68
boolean isEnable = handlerManager .isEnable ();
66
69
67
- SimpleResponse simpleResponse = SimpleResponse . success ( );
68
- simpleResponse .addAttribute ("isEnable" , isEnable );
70
+ MapResponse response = new MapResponse ( Result . SUCCESS );
71
+ response .addAttribute ("isEnable" , isEnable );
69
72
70
- return ResponseEntity .ok (simpleResponse );
73
+ return ResponseEntity .ok (response );
71
74
}
72
75
73
- private ResponseEntity <SimpleResponse > unauthorizedResponse (String errorMessage ) {
74
- SimpleResponse body = new SimpleResponse ( false , errorMessage );
76
+ private ResponseEntity <Response > unauthorizedResponse (String errorMessage ) {
77
+ MapResponse body = new MapResponse ( Result . FAIL , errorMessage );
75
78
return ResponseEntity .status (HttpStatus .FORBIDDEN ).body (body );
76
79
}
77
80
}
0 commit comments