aboutsummaryrefslogtreecommitdiffstats
path: root/vid-automation/src/main/java/org/onap/vid/model/mso/MsoResponseWrapper2.java
blob: 40bf86b264f0c05cc31ec7d54f57525cd31c3e3a (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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
package org.onap.vid.model.mso;

import com.fasterxml.jackson.annotation.JsonIgnore;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.fasterxml.jackson.annotation.JsonPropertyOrder;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;

@JsonPropertyOrder({
	    "status",
	    "entity"
})

/*
This is a brother of MsoResponseWrapper. I (Ittay) think it's better.
It is generic, immutable, and has some knowledge about RestObject.
The serialized "entity" field may be either String or nested object.
 */
public class MsoResponseWrapper2<T> implements MsoResponseWrapperInterface {

    final static ObjectMapper objectMapper = new ObjectMapper();

	private final int status;
	private final T entity;
    private final String raw;

    public MsoResponseWrapper2(RestObject<T> msoResponse) {
        this.status = msoResponse.getStatusCode();
        this.entity = msoResponse.get();
        this.raw = msoResponse.getRaw();
    }

    public MsoResponseWrapper2(
            @JsonProperty(value = "status", required = true) int status,
            @JsonProperty(value = "entity", required = true) T entity) {
        this.status = status;
        this.entity = entity;
        this.raw = null;
    }

    public int getStatus() {
		return status;
	}

    @Override
    @JsonIgnore
    public String getResponse() {
        try {
            return objectMapper.writeValueAsString(this);
        } catch (JsonProcessingException e) {
            return getEntity() != null ? getEntity().toString() : null;
        }
    }

    @JsonProperty
	public Object getEntity() {
		return entity != null ? entity : raw;
	}

}