Merge "Refactor Appc LCM Model"
[policy/models.git] / models-interactions / model-impl / events / src / main / java / org / onap / policy / controlloop / util / Serialization.java
1 /*-
2  * ============LICENSE_START=======================================================
3  * controlloop
4  * ================================================================================
5  * Copyright (C) 2017-2019 AT&T Intellectual Property. All rights reserved.
6  * Modifications Copyright (C) 2019 Nordix Foundation.
7  * ================================================================================
8  * Licensed under the Apache License, Version 2.0 (the "License");
9  * you may not use this file except in compliance with the License.
10  * You may obtain a copy of the License at
11  *
12  *      http://www.apache.org/licenses/LICENSE-2.0
13  *
14  * Unless required by applicable law or agreed to in writing, software
15  * distributed under the License is distributed on an "AS IS" BASIS,
16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17  * See the License for the specific language governing permissions and
18  * limitations under the License.
19  * ============LICENSE_END=========================================================
20  */
21
22 package org.onap.policy.controlloop.util;
23
24 import com.google.gson.Gson;
25 import com.google.gson.GsonBuilder;
26 import com.google.gson.JsonDeserializationContext;
27 import com.google.gson.JsonDeserializer;
28 import com.google.gson.JsonElement;
29 import com.google.gson.JsonPrimitive;
30 import com.google.gson.JsonSerializationContext;
31 import com.google.gson.JsonSerializer;
32
33 import java.lang.reflect.Type;
34 import java.time.Instant;
35 import java.time.ZonedDateTime;
36 import java.time.format.DateTimeFormatter;
37
38 import org.onap.policy.controlloop.ControlLoopNotificationType;
39 import org.slf4j.Logger;
40 import org.slf4j.LoggerFactory;
41
42 public final class Serialization {
43     public static final Gson gson =
44             new GsonBuilder().disableHtmlEscaping().registerTypeAdapter(ZonedDateTime.class, new GsonUtcAdapter())
45                     .registerTypeAdapter(Instant.class, new GsonInstantAdapter())
46                     .registerTypeAdapter(ControlLoopNotificationType.class, new NotificationTypeAdapter())
47                     .create();
48
49
50     public static final Gson gsonPretty = new GsonBuilder().disableHtmlEscaping().setPrettyPrinting()
51             .registerTypeAdapter(ZonedDateTime.class, new GsonUtcAdapter())
52             .registerTypeAdapter(Instant.class, new GsonInstantAdapter())
53             .registerTypeAdapter(ControlLoopNotificationType.class, new NotificationTypeAdapter())
54             .create();
55
56     public static final Gson gsonJunit = new GsonBuilder().disableHtmlEscaping().setPrettyPrinting()
57             .registerTypeAdapter(ZonedDateTime.class, new GsonUtcAdapter())
58             .registerTypeAdapter(Instant.class, new GsonInstantAdapter())
59             .create();
60
61     private Serialization() {}
62
63     public static class NotificationTypeAdapter
64             implements JsonSerializer<ControlLoopNotificationType>, JsonDeserializer<ControlLoopNotificationType> {
65         @Override
66         public JsonElement serialize(ControlLoopNotificationType src, Type typeOfSrc,
67                 JsonSerializationContext context) {
68             return new JsonPrimitive(src.toString());
69         }
70
71         @Override
72         public ControlLoopNotificationType deserialize(JsonElement json, Type typeOfT,
73                 JsonDeserializationContext context) {
74             return ControlLoopNotificationType.toType(json.getAsString());
75         }
76     }
77
78     public static class GsonUtcAdapter implements JsonSerializer<ZonedDateTime>, JsonDeserializer<ZonedDateTime> {
79         private static final Logger logger = LoggerFactory.getLogger(GsonUtcAdapter.class);
80         public static final DateTimeFormatter format = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss.SSSSSSxxx");
81
82         @Override
83         public ZonedDateTime deserialize(JsonElement element, Type type, JsonDeserializationContext context) {
84             try {
85                 return ZonedDateTime.parse(element.getAsString(), format);
86             } catch (Exception e) {
87                 logger.error(e.getMessage(), e);
88             }
89             return null;
90         }
91
92         @Override
93         public JsonElement serialize(ZonedDateTime datetime, Type type, JsonSerializationContext context) {
94             return new JsonPrimitive(datetime.format(format));
95         }
96     }
97
98     public static class GsonInstantAdapter implements JsonSerializer<Instant>, JsonDeserializer<Instant> {
99
100         @Override
101         public Instant deserialize(JsonElement json, Type typeOfT, JsonDeserializationContext context) {
102             return Instant.ofEpochMilli(json.getAsLong());
103         }
104
105         @Override
106         public JsonElement serialize(Instant src, Type typeOfSrc, JsonSerializationContext context) {
107             return new JsonPrimitive(src.toEpochMilli());
108         }
109
110     }
111
112 }