package com.cdu.yige.filter;
import java.io.UnsupportedEncodingException;
import java.util.Map;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
/**
* 對GET請求引數加以處理！
* @author lucien
*
*/
public class GetRequest extends HttpServletRequestWrapper {
     private HttpServletRequest request;
     private String charset;
     
     public GetRequest(HttpServletRequest request, String charset) {
          super(request);
          this.request = request;
          this.charset = charset;
     }

     @Override
     public String getParameter(String name) {
          // 獲取引數
          String value = request.getParameter(name);
          if(value == null) return null;//如果為null，直接返回null
          try {
               // 對引數進行編碼處理後返回
               return new String(value.getBytes("ISO-8859-1"), charset);
          } catch (UnsupportedEncodingException e) {
               throw new RuntimeException(e);
          }
     }
     
     @SuppressWarnings({ "unchecked", "rawtypes" })
     @Override
     public Map getParameterMap() {
          Map<String,String[]> map = request.getParameterMap();
          if(map == null) return map;
          // 遍歷map，對每個值進行編碼處理
          for(String key : map.keySet()) {
               String[] values = map.get(key);
               for(int i = 0; i < values.length; i++) {
                    try {
                         values[i] = new String(values[i].getBytes("ISO-8859-1"), charset);
                    } catch (UnsupportedEncodingException e) {
                         throw new RuntimeException(e);
                    }
               }
          }
          // 處理後返回
          return map;
     }
     @Override
     public String[] getParameterValues(String name) {
          String[] values = super.getParameterValues(name);
          for(int i = 0; i < values.length; i++) {
               try {
                    values[i] = new String(values[i].getBytes("ISO-8859-1"), charset);
               } catch (UnsupportedEncodingException e) {
                    throw new RuntimeException(e);
               }
          }
          return values;
     }
}