Update

Signed-off-by: Gyúróczki Gergő <gergonoorbi@gmail.com>

src/main/java/org/eclipse/yasson/internal/model/JsonbAnnotatedElement.java

@@ -13,15 +13,24 @@
 
 package org.eclipse.yasson.internal.model;
 
+import java.lang.annotation.Annotation;
 import java.lang.reflect.AnnotatedElement;
 
+import java.util.Map;
+import java.util.HashMap;
+
+import javax.json.bind.JsonbException;
+
+import org.eclipse.yasson.internal.properties.Messages;
+import org.eclipse.yasson.internal.properties.MessageKeys;
+
 /**
  * Annotation holder for fields, getters and setters.
  *
  * @param <T> annotated element
  */
-public class JsonbAnnotatedElement<T extends AnnotatedElement> extends JsonbAnnotated {
-
+public class JsonbAnnotatedElement<T extends AnnotatedElement> {
+    private final Map<Class<? extends Annotation>, Annotation> annotations;
     private final T element;
 
     /**
@@ -30,7 +39,12 @@ public class JsonbAnnotatedElement<T extends AnnotatedElement> extends JsonbAnno
      * @param element Element.
      */
     public JsonbAnnotatedElement(T element) {
-        super(element.getAnnotations());
+        Map<Class<? extends Annotation>, Annotation> annotations = new HashMap<>();
+
+        for (Annotation ann : element.getAnnotations()) {
+            annotations.put(ann.annotationType(), ann);
+        }
+        this.annotations = annotations;
         this.element = element;
     }
 
@@ -42,4 +56,25 @@ public JsonbAnnotatedElement(T element) {
     public T getElement() {
         return element;
     }
+
+    public <T extends Annotation> T getAnnotation(Class<T> annotationClass) {
+        return annotationClass.cast(annotations.get(annotationClass));
+    }
+
+    public Annotation[] getAnnotations() {
+        return annotations.values().toArray(new Annotation[0]);
+    }
+
+    /**
+     * Adds annotation.
+     *
+     * @param annotation Annotation to add.
+     */
+    public void putAnnotation(Annotation annotation) {
+        if (annotations.containsKey(annotation.annotationType())) {
+            throw new JsonbException(Messages.getMessage(MessageKeys.INTERNAL_ERROR,
+                                                         "Annotation already present: " + annotation));
+        }
+        annotations.put(annotation.annotationType(), annotation);
+    }
 }