org.jboss.weld.util.collections.ArraySetMultimap Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of weld-servlet Show documentation
There is a newer version: 3.0.0.Alpha1
/*
 * JBoss, Home of Professional Open Source
 * Copyright 2008, Red Hat, Inc. and/or its affiliates, and individual contributors
 * by the @authors tag. See the copyright.txt in the distribution for a
 * full listing of individual contributors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * http://www.apache.org/licenses/LICENSE-2.0
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.jboss.weld.util.collections;

import java.util.AbstractMap;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Set;

/**
 * A {@link java.util.Map} that can contain multiple values, where the keys
 * and values are stored in {@link ArraySet} instances.
 *
 * @author David Allen
 */
public class ArraySetMultimap extends AbstractMap> {
    private ArraySet>> entrySet = new ArraySet>>();

    public ArraySetMultimap() {

    }

    public ArraySetMultimap(Map> map) {
        // Add entries from the given map to this one
    }

    @Override
    public List get(Object key) {
        List result = super.get(key);
        if (result == null) {
            result = Collections.emptyList();
        }
        return result;
    }

    // Workaround bug in Helios
    public List putSingleElement(K key, V value) {
        List result = super.get(key);
        if (result == null) {
            result = new ArrayList();
            MapEntry> entry = new MapEntry>(key, result);
            entrySet.add(entry);
        }
        result.add(value);
        return result;
    }

    public ArraySetMultimap trimToSize() {
        for (Map.Entry> entry : this.entrySet()) {
            ((ArrayList) entry.getValue()).trimToSize();
        }
        return this;
    }

    @Override
    public Set>> entrySet() {
        return entrySet;
    }

    static class MapEntry implements Map.Entry {
        private K key;
        private V value;

        MapEntry(K key, V value) {
            this.key = key;
            this.value = value;
        }

        public K getKey() {
            return key;
        }

        public V getValue() {
            return value;
        }

        public V setValue(V value) {
            V previous = this.value;
            this.value = value;
            return previous;
        }

        @Override
        public int hashCode() {
            return (key == null ? 0 : key.hashCode()) ^ (value == null ? 0 : value.hashCode());
        }

        @Override
        public boolean equals(Object o) {
            if (!(o instanceof Map.Entry)) {
                return false;
            }

            Map.Entry e = (Map.Entry) o;

            return (key == null ? e.getKey() == null : key.equals(e.getKey())) &&
                   (value == null ? e.getValue() == null : value.equals(e.getValue()));
        }

        @Override
        public String toString() {
            return key + "=" + value;
        }
    }
}