com.litongjava.table.utils.ServletRequestParamUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of api-table Show documentation
auto curd for db table
There is a newer version: 1.4.8
package com.litongjava.table.utils;

import java.util.ArrayList;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.servlet.http.HttpServletRequest;

import com.jfinal.kit.StrKit;
import com.litongjava.tio.http.common.HttpRequest;

public class ServletRequestParamUtils {
  public static List types = new ArrayList<>();

  static {
    types.add("int");
    types.add("long");
  }

  public static Map getRequestMap(HttpServletRequest request) {
    Map map = new HashMap<>();
    String contentType = request.getContentType();

    if (contentType != null && contentType.contains("application/json")) {
      throw new RuntimeException("unspupport: application/json");
    }

    // Form data handling
    Map> arrayParams = new HashMap<>();
    Map paramType = new HashMap<>();
    Enumeration parameterNames = request.getParameterNames();
    while (parameterNames.hasMoreElements()) {
      String paramName = parameterNames.nextElement();
      String paramValue = request.getParameter(paramName);
      if (paramName.contains("[")) {
        // This is an array paramValue
        String arrayName = paramName.substring(0, paramName.indexOf('['));
        if (!arrayParams.containsKey(arrayName)) {
          arrayParams.put(arrayName, new ArrayList<>());
        }
        arrayParams.get(arrayName).add(paramValue);
      } else if (paramName.endsWith("Type") && types.contains(paramValue)) {
        // 前端传递参数缺定数据类型
        paramType.put(paramName, paramValue);
      } else {
        // This is a regular paramValue
        map.put(paramName, paramValue);
      }

    }


    // Convert the lists to arrays and add them to the map
    listArrayToMap(map, arrayParams, paramType);
    return map;
  }

  public static Map getRequestMap(HttpRequest request) {
    Map map = new HashMap<>();
    //String contentType = request.getHeader(HttpConst.RequestHeaderKey.Content_Type);
    String contentType = request.getContentType();

    if (contentType != null && contentType.contains("application/json")) {
      throw new RuntimeException("unspupport: application/json");
    }

    // Form data handling
    Map> arrayParams = new HashMap<>();
    Map paramType = new HashMap<>();
    Enumeration parameterNames = request.getParameterNames();
    while (parameterNames.hasMoreElements()) {
      String paramName = parameterNames.nextElement();
      String paramValue = request.getParameter(paramName);
      if (paramName.contains("[")) {
        // This is an array paramValue
        String arrayName = paramName.substring(0, paramName.indexOf('['));
        if (!arrayParams.containsKey(arrayName)) {
          arrayParams.put(arrayName, new ArrayList<>());
        }
        arrayParams.get(arrayName).add(paramValue);
      } else if (paramName.endsWith("Type") && types.contains(paramValue)) {
        // 前端传递参数缺定数据类型
        paramType.put(paramName, paramValue);
      } else {
        // This is a regular paramValue
        map.put(paramName, paramValue);
      }

    }


    // Convert the lists to arrays and add them to the map
    listArrayToMap(map, arrayParams, paramType);
    return map;
  }
  
  private static void listArrayToMap(Map map, Map> arrayParams,
      Map paramType) {
    for (Map.Entry> entry : arrayParams.entrySet()) {
      map.put(entry.getKey(), entry.getValue().toArray(new String[0]));
    }
    for (Map.Entry entry : paramType.entrySet()) {
      String typeKey = entry.getKey();
      int lastIndexOf = typeKey.lastIndexOf("Type");
      String paramKey = typeKey.substring(0, lastIndexOf);
      Object paramValue = map.get(paramKey);
      if (StrKit.notNull(paramValue)) {
        String value = entry.getValue();
        if ("int".equals(value)) {
          map.put(paramKey, Integer.parseInt((String) map.get(paramKey)));
        } else if ("long".equals(value)) {
          map.put(paramKey, Long.parseLong((String) map.get(paramKey)));
        }
      }
    }
  }
}