io.protostuff.compiler.parser.ImporterImpl 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

The newest version!

package io.protostuff.compiler.parser;

import java.util.HashMap;
import java.util.Map;
import javax.inject.Inject;
import javax.inject.Singleton;

/**
 * Caching implementation of an {@link Importer}.
 *
 * @author Kostiantyn Shchepanovskyi
 */
@Singleton
public class ImporterImpl implements Importer {

    private final FileDescriptorLoader loader;

    private Map cachedImports = new HashMap<>();

    @Inject
    public ImporterImpl(FileDescriptorLoader loader) {
        this.loader = loader;
    }

    @Override
    public ProtoContext importFile(FileReader reader, String fileName) {
        ProtoContext cachedInstance = cachedImports.get(fileName);
        if (cachedInstance != null) {
            if (cachedInstance.isInitialized()) {
                return cachedInstance;
            }
            throw new ParserException("Can not load proto: imports cycle found");
        }
        ProtoContext context = loader.load(reader, fileName);
        cachedImports.put(fileName, context);
        return context;
    }

}