com.fullcontact.rpc.jersey.yaml.YamlHttpConfig Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of protoc-gen-jersey Show documentation

protoc-gen-jersey developed by FullContact

There is a newer version: 0.1.3

package com.fullcontact.rpc.jersey.yaml;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.dataformat.yaml.YAMLFactory;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.Set;

/**
 *
 * Allows HTTPRules to be generated from a .yml file instead of the .proto files.
 * Any rule defined in a .yml file will override rules in the .proto.
 * https://cloud.google.com/endpoints/docs/grpc-service-config
 *
 * Path to the .yml file should be passed in as part of the optional parameter
 *
 *  @author Kyle Hansen (sypticus)
 */
public class YamlHttpConfig {
  public Map> http;
  public List getRules(){
   return http.get("rules");
  }

  public static Optional getFromOptions(Set options) {
    Optional yamlOption = options.stream().filter(option -> option.startsWith("yaml=")).findFirst();
    if(yamlOption.isPresent()) {
      String yamlPath = yamlOption.get().split("=")[1];
        try {
          File yamlFile = new File(yamlPath);
          if(!yamlFile.exists()) {
            throw new RuntimeException("YAMLs file does not exist: "+ yamlFile.getAbsolutePath());
          }
          InputStream yamlStream = new FileInputStream(yamlFile);

          ObjectMapper mapper = new ObjectMapper(new YAMLFactory());
          return Optional.of(mapper.readValue(yamlStream, YamlHttpConfig.class));
        } catch (IOException e) {
          throw new RuntimeException("Failed to parse YAML", e);
        }
      }
    return Optional.empty();
  }
}