org.infinispan.protostream.impl.parser.SquareProtoParser Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of infinispan-embedded
Infinispan Embedded All-in-One module
There is a newer version: 9.1.7.Final
package org.infinispan.protostream.impl.parser;

import java.io.CharArrayReader;
import java.io.IOException;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.Map;

import org.infinispan.protostream.DescriptorParser;
import org.infinispan.protostream.DescriptorParserException;
import org.infinispan.protostream.FileDescriptorSource;
import org.infinispan.protostream.config.Configuration;
import org.infinispan.protostream.descriptors.FileDescriptor;
import org.infinispan.protostream.descriptors.GenericDescriptor;
import org.infinispan.protostream.impl.parser.mappers.ProtofileMapper;

import com.squareup.protoparser.ProtoFile;
import com.squareup.protoparser.ProtoSchemaParser;

/**
 * Parser for .proto files based on the Protoparser.
 *
 * @author gustavonalle
 * @author [email protected]
 * @since 2.0
 */
public final class SquareProtoParser implements DescriptorParser {

   private static final ProtofileMapper PROTOFILE_MAPPER = new ProtofileMapper();

   private final Configuration configuration;

   public SquareProtoParser(Configuration configuration) {
      this.configuration = configuration;
   }

   @Override
   public Map parseAndResolve(FileDescriptorSource fileDescriptorSource) throws DescriptorParserException {
      // parse the input
      Map fileDescriptorMap = parse(fileDescriptorSource);

      // resolve imports and types
      Map types = new HashMap();
      for (FileDescriptor fileDescriptor : fileDescriptorMap.values()) {
         fileDescriptor.resolveDependencies(null, fileDescriptorMap, types);
         types.putAll(fileDescriptor.getTypes());
      }
      return fileDescriptorMap;
   }

   @Override
   public Map parse(FileDescriptorSource fileDescriptorSource) throws DescriptorParserException {
      Map input = fileDescriptorSource.getFileDescriptors();
      Map fileDescriptorMap = new LinkedHashMap(input.size());
      for (Map.Entry entry : input.entrySet()) {
         String fileName = entry.getKey();
         try {
            ProtoFile protoFile = ProtoSchemaParser.parse(fileName, new CharArrayReader(entry.getValue()));
            FileDescriptor fileDescriptor = PROTOFILE_MAPPER.map(protoFile);
            fileDescriptor.setConfiguration(configuration);
            fileDescriptorMap.put(fileName, fileDescriptor);
         } catch (IOException e) {
            throw new DescriptorParserException("Internal parsing error : " + e.getMessage());
         } catch (DescriptorParserException e) {
            throw e;
         } catch (RuntimeException e) {
            throw new DescriptorParserException(e);
         }
      }
      return fileDescriptorMap;
   }
}