blob: 25c040e06b600bc14ce39cb269fc4449171b55b6 (
plain)
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
|
package org.onap.pnfsimulator.rest.util;
import java.util.LinkedHashMap;
import java.util.Map;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
public class ResponseBuilder {
public static final String TIMESTAMP = "timestamp";
public static final String MESSAGE = "message";
public static final String SIMULATOR_STATUS = "simulatorStatus";
public static final String REMAINING_TIME = "remainingTime";
private HttpStatus httpStatus;
private Map<String, Object> body = new LinkedHashMap<>();
private ResponseBuilder(HttpStatus httpStatus) {
this.httpStatus = httpStatus;
}
public static ResponseBuilder status(HttpStatus httpStatus) {
return new ResponseBuilder(httpStatus);
}
public ResponseBuilder put(String key, Object value) {
body.put(key, value);
return this;
}
public ResponseEntity build() {
if (body.isEmpty()) {
return ResponseEntity.status(httpStatus).build();
}
return ResponseEntity.status(httpStatus).body(body);
}
}
|