io.usethesource.criterion.impl.immutable.guava.ImmutableGuavaSetMultimapBuilder.java Source code

Introduction

Here is the source code for io.usethesource.criterion.impl.immutable.guava.ImmutableGuavaSetMultimapBuilder.java

Source

/**
 * Copyright (c) Michael Steindorfer <Centrum Wiskunde & Informatica> and Contributors.
 * All rights reserved.
 *
 * This file is licensed under the BSD 2-Clause License, which accompanies this project
 * and is available under https://opensource.org/licenses/BSD-2-Clause.
 */
package io.usethesource.criterion.impl.immutable.guava;

import com.google.common.collect.ImmutableSetMultimap;
import io.usethesource.criterion.api.JmhSetMultimap;
import io.usethesource.criterion.api.JmhValue;

final class ImmutableGuavaSetMultimapBuilder implements JmhSetMultimap.Builder {

    protected ImmutableSetMultimap.Builder<JmhValue, JmhValue> mapContent;
    protected JmhSetMultimap constructedMap;

    ImmutableGuavaSetMultimapBuilder() {
        mapContent = ImmutableSetMultimap.builder();
        constructedMap = null;
    }

    @Override
    public void insert(JmhValue key, JmhValue value) {
        checkMutation();
        mapContent.put(key, value);
    }

    protected void checkMutation() {
        if (constructedMap != null) {
            throw new UnsupportedOperationException("Mutation of a finalized map is not supported.");
        }
    }

    @Override
    public JmhSetMultimap done() {
        if (constructedMap == null) {
            constructedMap = new ImmutableGuavaSetMultimap(mapContent.build());
        }

        return constructedMap;
    }

}