server/src/main/java/org/onap/usecaseui/server/util/RestfulServices.java

   1 /**
   2  * Copyright 2016-2017 ZTE Corporation.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 package org.onap.usecaseui.server.util;
  17
  18 import okhttp3.RequestBody;
  19 import okhttp3.MediaType;
  20 import retrofit2.Retrofit;
  21 import retrofit2.converter.jackson.JacksonConverterFactory;
  22
  23 import javax.servlet.ServletInputStream;
  24 import javax.servlet.http.HttpServletRequest;
  25 import java.io.IOException;
  26
  27 public class RestfulServices {
  28
  29     public static <T> T create(Class<T> clazz) {
  30         String msbUrl = getMsbAddress();
  31         Retrofit retrofit = new Retrofit.Builder()
  32                 .baseUrl("http://" + msbUrl)
  33                 .addConverterFactory(JacksonConverterFactory.create())
  34                 .build();
  35         return retrofit.create(clazz);
  36     }
  37
  38     public static String getMsbAddress() {
  39         String msbAddress = System.getenv("MSB_ADDR");
  40         if (msbAddress == null) {
  41             return "";
  42         }
  43         return msbAddress;
  44     }
  45
  46     public static RequestBody extractBody(HttpServletRequest request) throws IOException {
  47         int len = request.getContentLength();
  48         ServletInputStream inStream = null;
  49         try {
  50             inStream = request.getInputStream();
  51             byte[] buffer = new byte[len];
  52             inStream.read(buffer, 0, len);
  53             return RequestBody.create(MediaType.parse("application/json"), buffer);
  54         }finally {
  55             if (inStream != null) {
  56                 inStream.close();
  57             }
  58         }
  59     }
  60 }