io.protostuff.compiler.parser.LocalExtensionRegistry Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of protostuff-parser Show documentation

parser for .proto files

There is a newer version: 3.1.40

package io.protostuff.compiler.parser;

import com.google.common.base.Preconditions;
import com.google.common.collect.HashMultimap;
import com.google.common.collect.SetMultimap;

import java.util.Collection;

import io.protostuff.compiler.model.Extension;

/**
 * @author Kostiantyn Shchepanovskyi
 */
public class LocalExtensionRegistry extends AbstractExtensionRegistry {

    private final SetMultimap extensions = HashMultimap.create();


    @Override
    public void registerExtension(Extension extension) {
        String extendeeFullyQualifiedName = extension.getExtendee().getFullyQualifiedName();
        Preconditions.checkNotNull(extendeeFullyQualifiedName);
        extensions.put(extendeeFullyQualifiedName, extension);
    }

    @Override
    public Collection getExtensions(String messageName) {
        return extensions.get(messageName);
    }

}