• Home
• com.google.api
• gax-httpjson
• 2.52.0
• source code
• ProtoMessageRequestFormatter.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.google.api.gax.httpjson.ProtoMessageRequestFormatter Maven / Gradle / Ivy

/*
 * Copyright 2020 Google LLC
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met:
 *
 *     * Redistributions of source code must retain the above copyright
 * notice, this list of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above
 * copyright notice, this list of conditions and the following disclaimer
 * in the documentation and/or other materials provided with the
 * distribution.
 *     * Neither the name of Google LLC nor the names of its
 * contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
package com.google.api.gax.httpjson;

import com.google.api.core.InternalApi;
import com.google.api.pathtemplate.PathTemplate;
import com.google.protobuf.Message;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/** Creates parts of a HTTP request from a protobuf message. */
public class ProtoMessageRequestFormatter
    implements HttpRequestFormatter {

  private final FieldsExtractor requestBodyExtractor;
  // Using of triple nested generics (which is not pretty) is predetermined by the
  // Map> returned value type of the getQueryParamNames interface method
  // implemented by this class.
  private final FieldsExtractor>> queryParamsExtractor;
  private final String rawPath;
  private final PathTemplate pathTemplate;
  private final FieldsExtractor> pathVarsExtractor;
  private final List additionalRawPaths;
  private final List additionalPathTemplates;

  private ProtoMessageRequestFormatter(
      FieldsExtractor requestBodyExtractor,
      FieldsExtractor>> queryParamsExtractor,
      String rawPath,
      PathTemplate pathTemplate,
      FieldsExtractor> pathVarsExtractor,
      List additionalRawPaths,
      List additionalPathTemplates) {
    this.requestBodyExtractor = requestBodyExtractor;
    this.queryParamsExtractor = queryParamsExtractor;
    this.rawPath = rawPath;
    this.pathTemplate = pathTemplate;
    this.pathVarsExtractor = pathVarsExtractor;
    this.additionalRawPaths = additionalRawPaths;
    this.additionalPathTemplates = additionalPathTemplates;
  }

  public static 
      ProtoMessageRequestFormatter.Builder newBuilder() {
    return new Builder().setAdditionalPaths();
  }

  public Builder toBuilder() {
    return new Builder()
        .setPath(rawPath, pathVarsExtractor)
        .setAdditionalPaths(additionalRawPaths.toArray(new String[] {}))
        .setQueryParamsExtractor(queryParamsExtractor)
        .setRequestBodyExtractor(requestBodyExtractor);
  }

  /* {@inheritDoc} */
  @Override
  public Map> getQueryParamNames(RequestT apiMessage) {
    return queryParamsExtractor.extract(apiMessage);
  }

  /* {@inheritDoc} */
  @Override
  public String getRequestBody(RequestT apiMessage) {
    return requestBodyExtractor.extract(apiMessage);
  }

  /**
   * Returns the relative URL path created from the path parameters from the given message. Attempts
   * to match the with the default PathTemplate. If there is not match, it attempts to match with
   * the templates in the additionalPathTemplates.
   *
   * @param apiMessage Request object to extract fields from
   * @return Path of a matching valid URL or the default Path URL
   */
  @Override
  public String getPath(RequestT apiMessage) {
    Map pathVarsMap = pathVarsExtractor.extract(apiMessage);
    String path = pathTemplate.instantiate(pathVarsMap);
    if (pathTemplate.matches(path)) {
      return path;
    }
    for (PathTemplate additionalPathTemplate : additionalPathTemplates) {
      String additionalPath = additionalPathTemplate.instantiate(pathVarsMap);
      if (additionalPathTemplate.matches(additionalPath)) {
        return additionalPath;
      }
    }
    // If there are no matches, we return the default path, this is for backwards compatibility.
    // TODO: Log this scenario once we implemented the Cloud SDK logging.
    return path;
  }

  @Override
  public List getAdditionalPathTemplates() {
    return additionalPathTemplates;
  }

  /* {@inheritDoc} */
  @Override
  public PathTemplate getPathTemplate() {
    return pathTemplate;
  }

  // This has class has compound setter methods (multiple arguments in setters), that is why not
  // using @AutoValue.
  public static class Builder {
    private FieldsExtractor requestBodyExtractor;
    private FieldsExtractor>> queryParamsExtractor;
    private String rawPath;
    private FieldsExtractor> pathVarsExtractor;
    private List rawAdditionalPaths;

    public Builder setRequestBodyExtractor(
        FieldsExtractor requestBodyExtractor) {
      this.requestBodyExtractor = requestBodyExtractor;
      return this;
    }

    public Builder setQueryParamsExtractor(
        FieldsExtractor>> queryParamsExtractor) {
      this.queryParamsExtractor = queryParamsExtractor;
      return this;
    }

    public Builder setPath(
        String rawPath, FieldsExtractor> pathVarsExtractor) {
      this.rawPath = rawPath;
      this.pathVarsExtractor = pathVarsExtractor;
      return this;
    }

    public Builder setAdditionalPaths(String... rawAdditionalPaths) {
      this.rawAdditionalPaths = Arrays.asList(rawAdditionalPaths);
      return this;
    }

    @InternalApi
    public Builder updateRawPath(String rawPath) {
      this.rawPath = rawPath;
      return this;
    }

    @InternalApi
    public Builder updateRawPath(String target, String replacement) {
      this.rawPath = this.rawPath.replace(target, replacement);
      return this;
    }

    public ProtoMessageRequestFormatter build() {
      return new ProtoMessageRequestFormatter<>(
          requestBodyExtractor,
          queryParamsExtractor,
          rawPath,
          PathTemplate.create(rawPath),
          pathVarsExtractor,
          rawAdditionalPaths,
          rawAdditionalPaths.stream().map(PathTemplate::create).collect(Collectors.toList()));
    }
  }
}