1196cf6d9e485553663e0febbe9ede74a3c86185
[policy/apex-pdp.git] / plugins / plugins-context / plugins-context-schema / plugins-context-schema-avro / src / main / java / org / onap / policy / apex / plugins / context / schema / avro / AvroNullableMapper.java
1 /*-
2  * ============LICENSE_START=======================================================
3  *  Copyright (C) 2016-2018 Ericsson. All rights reserved.
4  * ================================================================================
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at
8  * 
9  *      http://www.apache.org/licenses/LICENSE-2.0
10  * 
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  * 
17  * SPDX-License-Identifier: Apache-2.0
18  * ============LICENSE_END=========================================================
19  */
20
21 package org.onap.policy.apex.plugins.context.schema.avro;
22
23 import org.apache.avro.Schema;
24 import org.apache.avro.Schema.Type;
25
26 import org.onap.policy.apex.model.basicmodel.concepts.ApexRuntimeException;
27 import org.onap.policy.apex.model.basicmodel.concepts.AxKey;
28
29 /**
30  * The Class AvroNullableMapper handles Avro null mappings to Java null values.
31  *
32  * @author John Keeney (john.keeney@ericsson.com)
33  */
34 public class AvroNullableMapper extends AvroDirectObjectMapper {
35     // The wrapped mapper for nullables
36     private final AvroObjectMapper wrappedMapper;
37
38     /**
39      * The Constructor.
40      *
41      * @param wrappedMapper the wrapped mapper
42      */
43     public AvroNullableMapper(final AvroObjectMapper wrappedMapper) {
44         this.wrappedMapper = wrappedMapper;
45     }
46
47     /**
48      * {@inheritDoc}.
49      */
50     @Override
51     public Class<?> getJavaClass() {
52         return wrappedMapper.getJavaClass();
53     }
54
55     /**
56      * {@inheritDoc}.
57      */
58     @Override
59     public void init(final AxKey userKey, final Type avroType) {
60         wrappedMapper.init(userKey, avroType);
61     }
62
63     /**
64      * {@inheritDoc}.
65      */
66     @Override
67     public Object createNewInstance(final Schema avroSchema) {
68         return wrappedMapper.createNewInstance(avroSchema);
69     }
70
71     /**
72      * {@inheritDoc}.
73      */
74     @Override
75     public Type getAvroType() {
76         return Schema.Type.UNION;
77     }
78
79     /**
80      * {@inheritDoc}.
81      */
82     @Override
83     public Object mapFromAvro(final Object avroObject) {
84         if (avroObject == null) {
85             return null;
86         } else {
87             return wrappedMapper.mapFromAvro(avroObject);
88         }
89     }
90
91     /**
92      * {@inheritDoc}.
93      */
94     @Override
95     public Object mapToAvro(final Object object) {
96         if (object == null) {
97             return null;
98         } else {
99             throw new ApexRuntimeException("Unions/Nullable is not supported in output event ... Coming soon!");
100         }
101
102     }
103
104 }