org.zanata.rest.dto.resource.ExtensionSet Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of zanata-common-api Show documentation

Zanata API definitions

There is a newer version: 4.6.2

package org.zanata.rest.dto.resource;

import java.io.Serializable;
import java.util.AbstractCollection;
import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.Map;

import org.zanata.rest.dto.DTOUtil;
import org.zanata.rest.dto.ExtensionValue;

import javax.annotation.Nonnull;

public class ExtensionSet extends
        AbstractCollection implements Serializable {

    private static final long serialVersionUID = 1L;
    private @Nonnull Map, T> extensions = new LinkedHashMap<>();

    @Override
    public Iterator iterator() {
        return extensions.values().iterator();
    }

    @Override
    public int size() {
        return extensions.size();
    }

    @Override
    public boolean add(T e) {
        this.extensions.put(e.getClass(), e);
        return true;
    };

    @SuppressWarnings("unchecked")
    public  E findByType(Class clz) {
        return (E) this.extensions.get(clz);
    }

    @Override
    public String toString() {
        return DTOUtil.toXML(this);
    }

    public  E findOrAddByType(Class clz) {
        E ext = findByType(clz);
        if (ext == null) {
            try {
                ext = clz.newInstance();
                add(ext);
            } catch (Throwable e) {
                throw new RuntimeException("unable to create instance", e);
            }
        }
        return ext;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) {
            return true;
        }
        if (o == null || !(o instanceof ExtensionSet)) {
            return false;
        }
        ExtensionSet that = (ExtensionSet) o;
        return extensions.equals(that.extensions);
    }

    @Override
    public int hashCode() {
        return extensions.hashCode();
    }

}