submarine-server/server-core/src/main/java/org/apache/submarine/server/utils/GitHttpRequest.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
package org.apache.submarine.server.utils;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.BufferedOutputStream;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.HttpURLConnection;
import java.net.ProtocolException;
import java.net.URL;
import java.net.URLConnection;
import java.nio.charset.StandardCharsets;
import java.util.Date;
import java.util.Map;

public class GitHttpRequest {
  private static final Logger LOG = LoggerFactory.getLogger(GitHttpRequest.class);

  /**
   * Sends an HTTP request to the specified URL.
   * @param url the URL you want to send request to
   * @param headParams a map of HTTP header parameters
   * @param content the content of request
   * @param method the HTTP method
   * @return String
   */
  public static String sendHttpRequest(String url, Map<String, String> headParams,
                                       byte[] content, String method) {
    // Create a URLConnection
    URLConnection urlConnection = null;
    try {
      urlConnection = (new URL(url)).openConnection();
    } catch (IOException e) {
      LOG.error(e.getMessage(), e);
    }

    // Defines StringBuilder to facilitate string concatenation later
    // when reading web pages and returning byte stream information
    StringBuilder stringBuilder = new StringBuilder();

    // Force URLConnection class to HttpURLConnection class
    HttpURLConnection httpURLConnection = (HttpURLConnection) urlConnection;
    httpURLConnection.setDoInput(true);
    httpURLConnection.setDoOutput(true);

    // Set the request, which could be deleted put post get
    try {
      httpURLConnection.setRequestMethod(method);
    } catch (ProtocolException e) {
      LOG.error(e.getMessage(), e);
    }

    if (content != null) {
      // Sets the length of the content
      httpURLConnection.setRequestProperty("Content-Length", String.valueOf(content.length));
    }

    // Set encoding format
    httpURLConnection.setRequestProperty("Content-Type", "application/json;charset=utf-8");
    // Sets the format of the received return parameter
    httpURLConnection.setRequestProperty("accept", "application/json");
    httpURLConnection.setRequestProperty("connection", "Keep-Alive");
    httpURLConnection.setRequestProperty("user-agent",
        "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1;SV1)");
    // Timestamp (accurate to seconds)
    httpURLConnection.setRequestProperty("timestamp", String.valueOf(new Date().getTime() / 1000));

    if (headParams != null) {
      for (String key : headParams.keySet()) {
        httpURLConnection.setRequestProperty(key, headParams.get(key));
      }
    }

    httpURLConnection.setUseCaches(false);

    if (content != null) {
      try (BufferedOutputStream outputStream =
               new BufferedOutputStream(httpURLConnection.getOutputStream())) {
        outputStream.write(content);
        outputStream.flush();
      } catch (IOException e) {
        LOG.error(e.getMessage(), e);
      }
    }

    try (InputStreamReader inputStreamReader =
             new InputStreamReader(httpURLConnection.getInputStream(), StandardCharsets.UTF_8);
         BufferedReader bufferedReader = new BufferedReader(inputStreamReader)) {
      String line;
      while ((line = bufferedReader.readLine()) != null) {
        stringBuilder.append(line);
      }
    } catch (IOException e) {
      LOG.error(e.getMessage(), e);
    }

    LOG.info("result:{}", stringBuilder);
    return stringBuilder.toString();
  }
}