`

Java中的HTTP操作的封装(破防盗链)

    博客分类:
  • Java
阅读更多

经常会遇到读取网页的内容,XML,图片等,于是稍花时间封装了下在Java中使用http请求处理的各种操作,实现的功能如下:

  • 添加用户代理
  • 设置超时时间
  • 设置最大长度
  • 控制是否跟踪重定向(默认是)
  • 破解防盗链
  • 自由控制Http响应异常的处理
  • 添加请求参数、标头(header)信息、cookie信息
  • 自动解析页面编码格式,也可指定
  • 获取页面文本信息
  • 将连接所指向的内容保存到本地

示例:

	//获取文本信息
	String str = HttpHelper.connect("http://www.autohome.com.cn/364/").timeout(3000).get().html();
	
	//保存到本地
	HttpHelper.connect("http://www.autohome.com.cn/364/").timeout(3000).get().toFile("D://001.html");
	
	//图片本地化(破解防盗链)
	HttpHelper.connect("http://www.autoimg.cn/upload/2013/2/25/t_201302251823083624136.jpg")
			.referrer("http://www.autohome.com.cn/").timeout(3000).get().toFile("D://001.jpg");

 

其他方法请直接查看源代码。编写过程中参考了开源工具jsoup,为了方便添加到各个项目中,该工具类不依赖任何三方jar包,并且将所有方法都写在一个文件中,因此使用时只需直接添加到项目中即可!下面附上源代码:

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLEncoder;
import java.nio.ByteBuffer;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.Collection;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.zip.GZIPInputStream;

/**
 * @author anly
 *
 */
public class HttpHelper {
	
	/**
	 * 请求方法
	 */
	public enum Method {
        GET, POST
    }
	
	/**
	 * 添加URL地址
	 * @param url
	 * @return
	 */
	public static HttpHelper connect(String url) {
        HttpHelper http = new HttpHelper();
        http.url(url);
        return http;
    }

	/**
	 * 添加URL地址
	 * @param url
	 * @return
	 */
    public static HttpHelper connect(URL url) {
        HttpHelper http = new HttpHelper();
        http.url(url);
        return http;
    }

    private HttpHelper.Request req;
    private HttpHelper.Response res;

	private HttpHelper() {
        req = new Request();
        res = new Response();
    }

	/**
	 * 替换URL地址
	 * @param url
	 * @return
	 */
    public HttpHelper url(URL url) {
        req.url(url);
        return this;
    }

    /**
	 * 替换URL地址
	 * @param url
	 * @return
	 */
    public HttpHelper url(String url) {
        Validate.notEmpty(url, "Must supply a valid URL");
        try {
            req.url(new URL(url));
        } catch (MalformedURLException e) {
            throw new IllegalArgumentException("Malformed URL: " + url, e);
        }
        return this;
    }
    
    /**
     * 设置用户代理
     * @param userAgent
     * @return
     */
    public HttpHelper userAgent(String userAgent) {
        Validate.notNull(userAgent, "User agent must not be null");
        req.header("User-Agent", userAgent);
        return this;
    }

    /**
     * 设置超时时间
     * @param millis
     * @return
     */
    public HttpHelper timeout(int millis) {
        req.timeout(millis);
        return this;
    }

    /**
     * 设置最大长度
     * @param bytes
     * @return
     */
    public HttpHelper maxBodySize(int bytes) {
        req.maxBodySize(bytes);
        return this;
    }

    /**
     * 是否跟踪重定向(默认是)
     * @param followRedirects
     * @return
     */
    public HttpHelper followRedirects(boolean followRedirects) {
        req.followRedirects(followRedirects);
        return this;
    }

    /**
     * 设置网站来路
     * @param referrer
     * @return
     */
    public HttpHelper referrer(String referrer) {
        Validate.notNull(referrer, "Referrer must not be null");
        req.header("Referer", referrer);
        return this;
    }

    /**
     * 设置请求方法
     * @param method
     * @return
     */
    public HttpHelper method(Method method) {
        req.method(method);
        return this;
    }

    /**
     * 是否忽略Http响应异常
     * @param ignoreHttpErrors
     * @return
     */
    public HttpHelper ignoreHttpErrors(boolean ignoreHttpErrors) {
		req.ignoreHttpErrors(ignoreHttpErrors);
		return this;
	}

    /**
     * 添加参数
     * @param key
     * @param value
     * @return
     */
    public HttpHelper data(String key, String value) {
        req.data(KeyVal.create(key, value));
        return this;
    }

    /**
     * 添加参数
     * @param data
     * @return
     */
    public HttpHelper data(Map<String, String> data) {
        Validate.notNull(data, "Data map must not be null");
        for (Map.Entry<String, String> entry : data.entrySet()) {
            req.data(KeyVal.create(entry.getKey(), entry.getValue()));
        }
        return this;
    }

    /**
     * 添加参数
     * @param keyvals
     * @return
     */
    public HttpHelper data(String... keyvals) {
        Validate.notNull(keyvals, "Data key value pairs must not be null");
        Validate.isTrue(keyvals.length %2 == 0, "Must supply an even number of key value pairs");
        for (int i = 0; i < keyvals.length; i += 2) {
            String key = keyvals[i];
            String value = keyvals[i+1];
            Validate.notEmpty(key, "Data key must not be empty");
            Validate.notNull(value, "Data value must not be null");
            req.data(KeyVal.create(key, value));
        }
        return this;
    }

    /**
     * 添加标头(header)信息
     * @param name
     * @param value
     * @return
     */
    public HttpHelper header(String name, String value) {
        req.header(name, value);
        return this;
    }

    /**
     * 添加cookie信息
     * @param name
     * @param value
     * @return
     */
    public HttpHelper cookie(String name, String value) {
        req.cookie(name, value);
        return this;
    }

    /**
     * 添加cookie信息
     * @param cookies
     * @return
     */
    public HttpHelper cookies(Map<String, String> cookies) {
        Validate.notNull(cookies, "Cookie map must not be null");
        for (Map.Entry<String, String> entry : cookies.entrySet()) {
            req.cookie(entry.getKey(), entry.getValue());
        }
        return this;
    }

    /**
     * 通过get方法获取文本信息
     * @return
     * @throws IOException
     */
    public HttpHelper.Response get() throws IOException {
        req.method(Method.GET);
        res = Response.execute(req);
        return res;
    }

    /**
     * 通过post方法获取文本信息
     * @return
     * @throws IOException
     */
    public HttpHelper.Response post() throws IOException {
        req.method(Method.POST);
        res = Response.execute(req);
        return res;
    }

    /**
     * 获取请求对象
     * @return
     */
    public HttpHelper.Request request() {
        return req;
    }

    /**
     * 设置请求对象
     * @return
     */
    public HttpHelper request(HttpHelper.Request request) {
        req = request;
        return this;
    }
    
    /**
     * 设置编码(默认UTF-8)
     * @return
     */
    public HttpHelper charset(String charset) {
    	Validate.isTrue(Charset.isSupported(charset), String.format("The charset of '%s' is not supported", charset));
    	req.charset(charset);
    	return this;
    }

    @SuppressWarnings({ "unchecked", "rawtypes" })
    private static abstract class Base<T extends Base> {
        private URL url;
        private Method method;
        private Map<String, String> headers;
        private Map<String, String> cookies;

        private Base() {
            headers = new LinkedHashMap<String, String>();
            cookies = new LinkedHashMap<String, String>();
        }

        public URL url() {
            return url;
        }

		public T url(URL url) {
            Validate.notNull(url, "URL must not be null");
            this.url = url;
            return (T) this;
        }

        public Method method() {
            return method;
        }

        public T method(Method method) {
            Validate.notNull(method, "Method must not be null");
            this.method = method;
            return (T) this;
        }

        public String header(String name) {
            Validate.notNull(name, "Header name must not be null");
            return getHeaderCaseInsensitive(name);
        }

        public T header(String name, String value) {
            Validate.notEmpty(name, "Header name must not be empty");
            Validate.notNull(value, "Header value must not be null");
            removeHeader(name); // ensures we don't get an "accept-encoding" and a "Accept-Encoding"
            headers.put(name, value);
            return (T) this;
        }

        public boolean hasHeader(String name) {
            Validate.notEmpty(name, "Header name must not be empty");
            return getHeaderCaseInsensitive(name) != null;
        }

        public T removeHeader(String name) {
            Validate.notEmpty(name, "Header name must not be empty");
            Map.Entry<String, String> entry = scanHeaders(name); // remove is case insensitive too
            if (entry != null) {
            	headers.remove(entry.getKey()); // ensures correct case
            }
            return (T) this;
        }

        public Map<String, String> headers() {
            return headers;
        }

        private String getHeaderCaseInsensitive(String name) {
            Validate.notNull(name, "Header name must not be null");
            // quick evals for common case of title case, lower case, then scan for mixed
            String value = headers.get(name);
            if (value == null) {
            	value = headers.get(name.toLowerCase());
            }
            if (value == null) {
                Map.Entry<String, String> entry = scanHeaders(name);
                if (entry != null) {
                	value = entry.getValue();
                }
            }
            return value;
        }

        private Map.Entry<String, String> scanHeaders(String name) {
            String lc = name.toLowerCase();
            for (Map.Entry<String, String> entry : headers.entrySet()) {
                if (entry.getKey().toLowerCase().equals(lc)) {
                	return entry;
                }
            }
            return null;
        }

        public String cookie(String name) {
            Validate.notNull(name, "Cookie name must not be null");
            return cookies.get(name);
        }

        public T cookie(String name, String value) {
            Validate.notEmpty(name, "Cookie name must not be empty");
            Validate.notNull(value, "Cookie value must not be null");
            cookies.put(name, value);
            return (T) this;
        }

        public boolean hasCookie(String name) {
            Validate.notEmpty("Cookie name must not be empty");
            return cookies.containsKey(name);
        }

        public T removeCookie(String name) {
            Validate.notEmpty("Cookie name must not be empty");
            cookies.remove(name);
            return (T) this;
        }

        public Map<String, String> cookies() {
            return cookies;
        }
    }

    public static class Request extends Base<Request> {
    	
        private int timeoutMilliseconds;
        private int maxBodySizeBytes;
        private boolean followRedirects;
        private Collection<HttpHelper.KeyVal> data;
        private boolean ignoreHttpErrors = false;
        private String charset;

      	private Request() {
            timeoutMilliseconds = 3000;
            maxBodySizeBytes = 3 * 1024 * 1024; // 1MB
            followRedirects = true;
            data = new ArrayList<HttpHelper.KeyVal>();
            super.method = Method.GET;
            super.headers.put("Accept-Encoding", "gzip");
        }

        public int timeout() {
            return timeoutMilliseconds;
        }

        public Request timeout(int millis) {
            Validate.isTrue(millis >= 0, "Timeout milliseconds must be 0 (infinite) or greater");
            timeoutMilliseconds = millis;
            return this;
        }

        public int maxBodySize() {
            return maxBodySizeBytes;
        }

        public HttpHelper.Request maxBodySize(int bytes) {
            Validate.isTrue(bytes >= 0, "maxSize must be 0 (unlimited) or larger");
            maxBodySizeBytes = bytes;
            return this;
        }

        public boolean followRedirects() {
            return followRedirects;
        }

        public HttpHelper.Request followRedirects(boolean followRedirects) {
            this.followRedirects = followRedirects;
            return this;
        }

        public boolean ignoreHttpErrors() {
            return ignoreHttpErrors;
        }

        public HttpHelper.Request ignoreHttpErrors(boolean ignoreHttpErrors) {
            this.ignoreHttpErrors = ignoreHttpErrors;
            return this;
        }

        public Request data(HttpHelper.KeyVal keyval) {
            Validate.notNull(keyval, "Key val must not be null");
            data.add(keyval);
            return this;
        }

        public Collection<HttpHelper.KeyVal> data() {
            return data;
        }
        
        public String charset() {
            return charset;
        }
        
        public void charset(String charset) {
        	this.charset = charset;
        }
        
    }

    public static class Response extends Base<Response> {
    	
        private static final String defaultCharset = "UTF-8"; // used if not found in header or meta charset
        
        private static final int MAX_REDIRECTS = 20;
        private int statusCode;
        private String statusMessage;
        private ByteBuffer byteData;
        private String charset;
        private String contentType;
        private boolean executed = false;
        private int numRedirects = 0;
        @SuppressWarnings("unused")
		private HttpHelper.Request req;

        Response() {
            super();
        }

        private Response(Response previousResponse) throws IOException {
            super();
            if (previousResponse != null) {
                numRedirects = previousResponse.numRedirects + 1;
                if (numRedirects >= MAX_REDIRECTS) {
                	throw new IOException(String.format("Too many redirects occurred trying to load URL %s", previousResponse.url()));
                }
            }
        }
        
        private static Response execute(HttpHelper.Request req) throws IOException {
            return execute(req, null);
        }

        private static Response execute(HttpHelper.Request req, Response previousResponse) throws IOException {
            
        	Validate.notNull(req, "Request must not be null");
            String protocol = req.url().getProtocol();
            if (!protocol.equals("http") && !protocol.equals("https")) {
            	throw new MalformedURLException("Only http & https protocols supported");
            }

            // set up the request for execution
            if (req.method() == HttpHelper.Method.GET && req.data().size() > 0) {
            	serialiseRequestUrl(req); // appends query string
            }
            HttpURLConnection conn = createConnection(req);
            Response res;
            try {
                conn.connect();
                if (req.method() == Method.POST) {
                	writePost(req.data(), conn.getOutputStream());
                }

                int status = conn.getResponseCode();
                boolean needsRedirect = false;
                if (status != HttpURLConnection.HTTP_OK) {
                    if (status == HttpURLConnection.HTTP_MOVED_TEMP || status == HttpURLConnection.HTTP_MOVED_PERM || status == HttpURLConnection.HTTP_SEE_OTHER) {
                    	needsRedirect = true;
                    } else if (!req.ignoreHttpErrors()) {
                    	throw new HttpStatusException("HTTP error fetching URL", status, req.url().toString());
                    }
                }
                res = new Response(previousResponse);
                res.setupFromConnection(conn, previousResponse);
                if (needsRedirect && req.followRedirects()) {
                    req.method(Method.GET); // always redirect with a get. any data param from original req are dropped.
                    req.data().clear();
                    req.url(new URL(req.url(), res.header("Location")));
                    for (Map.Entry<String, String> cookie : res.cookies().entrySet()) { // add response cookies to request (for e.g. login posts)
                        req.cookie(cookie.getKey(), cookie.getValue());
                    }
                    return execute(req, res);
                }
                res.req = req;

                InputStream dataStream = null;
                InputStream bodyStream = null;
                try {
                    dataStream = conn.getErrorStream() != null ? conn.getErrorStream() : conn.getInputStream();
                    bodyStream = res.hasHeader("Content-Encoding") && res.header("Content-Encoding").equalsIgnoreCase("gzip") ?
                            new BufferedInputStream(new GZIPInputStream(dataStream)) :
                            new BufferedInputStream(dataStream);

                    res.byteData = readToByteBuffer(bodyStream, req.maxBodySize());
                    if(req.charset() == null) {
                    	res.charset = getCharsetFromContentType(res.contentType); // may be null, readInputStream deals with it
                    } else {
                    	res.charset = req.charset;
                    }
                } finally {
                    if (bodyStream != null) { 
                    	bodyStream.close();
                    }
                    if (dataStream != null) {
                    	dataStream.close();
                    }
                }
            } finally {
                conn.disconnect();
            }

            res.executed = true;
            return res;
        }

        public int statusCode() {
            return statusCode;
        }

        public String statusMessage() {
            return statusMessage;
        }

        public String charset() {
            return charset;
        }
        
        public String contentType() {
            return contentType;
        }

        public String html() throws IOException {
            
        	Validate.isTrue(executed, "Request must be executed (with .execute(), .get(), or .post() before parsing response");
            
            String content = null;
            
            if(charset == null) {
            	
            	content = Charset.forName(defaultCharset).decode(byteData).toString();
            	
            	Pattern pattern = Pattern.compile("<meta[^>]*content=\"(.+?)\"[^>]*>");
            	
            	Matcher matcher = pattern.matcher(content);
            	
            	String foundCharset = null;
            	
            	while (matcher.find()) {
            		foundCharset = getCharsetFromContentType(matcher.group(1));
            		if(foundCharset != null) {
            			break;
            		}
            	}
            	
            	if(foundCharset == null) {
            		pattern = Pattern.compile("<meta[^>]*charset=\"(.+?)\"[^>]*>");
            		matcher = pattern.matcher(content);
            		while (matcher.find()) {
                		foundCharset = matcher.group(1);
                		if(foundCharset != null) {
                			break;
                		}
                	}
            	}
            	
            	if(foundCharset != null) {
            		charset = foundCharset;
            		content = Charset.forName(foundCharset).decode(byteData).toString();
            	}
            } else {
            	content = Charset.forName(charset).decode(byteData).toString();
            }
            
            if (content.length() > 0 && content.charAt(0) == 65279) {
            	content = content.substring(1);
            }
            byteData.rewind();
            
            return content;
        }
        
        public void toFile(String to) {
        	
        	Validate.notNull(to, "Data value must not be null");
        	
        	makeDirs(to);
        	
        	generateFile(to);
        	
        }
        
        private static void makeDirs(String path) {
    		try {
    			int i = path.lastIndexOf("/");
    			if (i < 1) {
    				i = path.lastIndexOf("\\");
    			}
    			path = path.substring(0, i);
    			File file = new File(path);
    			if (!file.exists()) {
    				file.mkdirs();
    			}
    		} catch (Exception e) {
    			e.printStackTrace();
    		}
    	}
        
		private void generateFile(String to) {
    		OutputStream bos = null;
        	try {
        		bos = new BufferedOutputStream(new FileOutputStream(to));
        		if(byteData != null) {
        			bos.write(byteData.array(), 0, byteData.array().length);
        		}
    			bos.flush();
    		} catch (FileNotFoundException e) {
    			e.printStackTrace();
    		} catch (IOException e) {
    			e.printStackTrace();
    		} finally {
    			if (bos != null) {
    				try {
    					bos.close();
    				} catch (IOException e) {
    					e.printStackTrace();
    				}
    			}
    		}
        }

        // set up connection defaults, and details from request
        private static HttpURLConnection createConnection(HttpHelper.Request req) throws IOException {
        	HttpURLConnection conn = (HttpURLConnection) req.url().openConnection();
            conn.setRequestMethod(req.method().name());
            conn.setInstanceFollowRedirects(false); // don't rely on native redirection support
            conn.setConnectTimeout(req.timeout());
            conn.setReadTimeout(req.timeout());
            if (req.method() == Method.POST) {
            	conn.setDoOutput(true);
            }
            if (req.cookies().size() > 0) {
            	conn.addRequestProperty("Cookie", getRequestCookieString(req));
            }
            for (Map.Entry<String, String> header : req.headers().entrySet()) {
                conn.addRequestProperty(header.getKey(), header.getValue());
            }
            return conn;
        }

        // set up url, method, header, cookies
        private void setupFromConnection(HttpURLConnection conn, HttpHelper.Response previousResponse) throws IOException {
            super.method = HttpHelper.Method.valueOf(conn.getRequestMethod());
            super.url = conn.getURL();
            statusCode = conn.getResponseCode();
            statusMessage = conn.getResponseMessage();
            contentType = conn.getContentType();

            Map<String, List<String>> resHeaders = conn.getHeaderFields();
            processResponseHeaders(resHeaders);

            // if from a redirect, map previous response cookies into this response
            if (previousResponse != null) {
                for (Map.Entry<String, String> prevCookie : previousResponse.cookies().entrySet()) {
                    if (!hasCookie(prevCookie.getKey())) {
                    	cookie(prevCookie.getKey(), prevCookie.getValue());
                    }
                }
            }
        }

        private void processResponseHeaders(Map<String, List<String>> resHeaders) {
            for (Map.Entry<String, List<String>> entry : resHeaders.entrySet()) {
                String name = entry.getKey();
                if (name == null) {
                	continue; // http/1.1 line
                }

                List<String> values = entry.getValue();
                if (name.equalsIgnoreCase("Set-Cookie")) {
                    for (String value : values) {
                        if (value == null){
                        	continue;
                        }
                        TokenQueue cd = new TokenQueue(value);
                        String cookieName = cd.chompTo("=").trim();
                        String cookieVal = cd.consumeTo(";").trim();
                        if (cookieVal == null) {
                        	cookieVal = "";
                        }
                        // ignores path, date, domain, secure et al. req'd?
                        // name not blank, value not null
                        if (cookieName != null && cookieName.length() > 0) {
                        	cookie(cookieName, cookieVal);
                        }
                    }
                } else { // only take the first instance of each header
                    if (!values.isEmpty()) {
                    	header(name, values.get(0));
                    }
                }
            }
        }

        private static void writePost(Collection<HttpHelper.KeyVal> data, OutputStream outputStream) throws IOException {
            OutputStreamWriter w = new OutputStreamWriter(outputStream, defaultCharset);
            boolean first = true;
            for (HttpHelper.KeyVal keyVal : data) {
                if (!first) {
                	w.append('&');
                } else {
                	first = false;
                }
                
                w.write(URLEncoder.encode(keyVal.key(), defaultCharset));
                w.write('=');
                w.write(URLEncoder.encode(keyVal.value(), defaultCharset));
            }
            w.close();
        }
        
        private static String getRequestCookieString(HttpHelper.Request req) {
            StringBuilder sb = new StringBuilder();
            boolean first = true;
            for (Map.Entry<String, String> cookie : req.cookies().entrySet()) {
                if (!first) {
                	sb.append("; ");
                } else {
                	first = false;
                }
                sb.append(cookie.getKey()).append('=').append(cookie.getValue());
            }
            return sb.toString();
        }

        // for get url reqs, serialise the data map into the url
        private static void serialiseRequestUrl(HttpHelper.Request req) throws IOException {
            URL in = req.url();
            StringBuilder url = new StringBuilder();
            boolean first = true;
            // reconstitute the query, ready for appends
            url.append(in.getProtocol())
                .append("://")
                .append(in.getAuthority()) // includes host, port
                .append(in.getPath())
                .append("?");
            if (in.getQuery() != null) {
                url.append(in.getQuery());
                first = false;
            }
            for (HttpHelper.KeyVal keyVal : req.data()) {
                if (!first) {
                	url.append('&');
                } else {
                	first = false;
                }
                url.append(URLEncoder.encode(keyVal.key(), defaultCharset))
                    .append('=')
                    .append(URLEncoder.encode(keyVal.value(), defaultCharset));
            }
            req.url(new URL(url.toString()));
            req.data().clear(); // moved into url as get params
        }
    }
    
    private static class TokenQueue {
        private String queue;
        private int pos = 0;
        
        /**
         Create a new TokenQueue.
         @param data string of data to back queue.
         */
        public TokenQueue(String data) {
            Validate.notNull(data);
            queue = data;
        }

        /**
         * Is the queue empty?
         * @return true if no data left in queue.
         */
        public boolean isEmpty() {
            return remainingLength() == 0;
        }
        
        private int remainingLength() {
            return queue.length() - pos;
        }

        /**
         * Tests if the next characters on the queue match the sequence. Case insensitive.
         * @param seq String to check queue for.
         * @return true if the next characters match.
         */
        public boolean matches(String seq) {
            return queue.regionMatches(true, pos, seq, 0, seq.length());
        }

        /**
         * Tests if the queue matches the sequence (as with match), and if they do, removes the matched string from the
         * queue.
         * @param seq String to search for, and if found, remove from queue.
         * @return true if found and removed, false if not found.
         */
        public boolean matchChomp(String seq) {
            if (matches(seq)) {
                pos += seq.length();
                return true;
            } else {
                return false;
            }
        }

        /**
         * Consume one character off queue.
         * @return first character on queue.
         */
        public char consume() {
            return queue.charAt(pos++);
        }

        /**
         * Pulls a string off the queue, up to but exclusive of the match sequence, or to the queue running out.
         * @param seq String to end on (and not include in return, but leave on queue). <b>Case sensitive.</b>
         * @return The matched data consumed from queue.
         */
        public String consumeTo(String seq) {
            int offset = queue.indexOf(seq, pos);
            if (offset != -1) {
                String consumed = queue.substring(pos, offset);
                pos += consumed.length();
                return consumed;
            } else {
                return remainder();
            }
        }
        
        /**
         * Pulls a string off the queue (like consumeTo), and then pulls off the matched string (but does not return it).
         * <p>
         * If the queue runs out of characters before finding the seq, will return as much as it can (and queue will go
         * isEmpty() == true).
         * @param seq String to match up to, and not include in return, and to pull off queue. <b>Case sensitive.</b>
         * @return Data matched from queue.
         */
        public String chompTo(String seq) {
            String data = consumeTo(seq);
            matchChomp(seq);
            return data;
        }

        /**
         Consume and return whatever is left on the queue.
         @return remained of queue.
         */
        public String remainder() {
            StringBuilder accum = new StringBuilder();
            while (!isEmpty()) {
                accum.append(consume());
            }
            return accum.toString();
        }
        
        public String toString() {
            return queue.substring(pos);
        }
    }

    public static class KeyVal {
        
    	private String key;
        private String value;

        public static KeyVal create(String key, String value) {
            Validate.notEmpty(key, "Data key must not be empty");
            Validate.notNull(value, "Data value must not be null");
            return new KeyVal(key, value);
        }

        private KeyVal(String key, String value) {
            this.key = key;
            this.value = value;
        }

        public KeyVal key(String key) {
            Validate.notEmpty(key, "Data key must not be empty");
            this.key = key;
            return this;
        }

        public String key() {
            return key;
        }

        public KeyVal value(String value) {
            Validate.notNull(value, "Data value must not be null");
            this.value = value;
            return this;
        }

        public String value() {
            return value;
        }

    }
    
    @SuppressWarnings("unused")
    private static class HttpStatusException extends IOException {

    	private static final long serialVersionUID = -2926428810498166324L;
		private int statusCode;
        private String url;

        public HttpStatusException(String message, int statusCode, String url) {
            super(message);
            this.statusCode = statusCode;
            this.url = url;
        }

        public int getStatusCode() {
            return statusCode;
        }

		public String getUrl() {
            return url;
        }

        public String toString() {
            return super.toString() + ". Status=" + statusCode + ", URL=" + url;
        }
    }
    
    @SuppressWarnings("unused")
    private static class UnsupportedMimeTypeException extends IOException {
        
		private static final long serialVersionUID = 2535952512520299658L;
		private String mimeType;
        private String url;

        public UnsupportedMimeTypeException(String message, String mimeType, String url) {
            super(message);
            this.mimeType = mimeType;
            this.url = url;
        }

        public String getMimeType() {
            return mimeType;
        }

		public String getUrl() {
            return url;
        }

        public String toString() {
            return super.toString() + ". Mimetype=" + mimeType + ", URL="+url;
        }
    }
    
    private static ByteBuffer readToByteBuffer(InputStream inStream, int maxSize) throws IOException {
        Validate.isTrue(maxSize >= 0, "maxSize must be 0 (unlimited) or larger");
        final boolean capped = maxSize > 0;
        byte[] buffer = new byte[0x20000];
        ByteArrayOutputStream outStream = new ByteArrayOutputStream(0x20000);
        int read;
        int remaining = maxSize;

        while (true) {
            read = inStream.read(buffer);
            if (read == -1) break;
            if (capped) {
                if (read > remaining) {
                    outStream.write(buffer, 0, remaining);
                    break;
                }
                remaining -= read;
            }
            outStream.write(buffer, 0, read);
        }
        ByteBuffer byteData = ByteBuffer.wrap(outStream.toByteArray());
        return byteData;
    }
    
    private static String getCharsetFromContentType(String contentType) {
        if (contentType == null) {
        	return null;
        }
        Pattern charsetPattern = Pattern.compile("(?i)\\bcharset=\\s*\"?([^\\s;\"]*)");
        Matcher m = charsetPattern.matcher(contentType);
        if (m.find()) {
            String charset = m.group(1).trim();
            if (Charset.isSupported(charset)) {
            	return charset;
            }
            charset = charset.toUpperCase(Locale.ENGLISH);
            if (Charset.isSupported(charset)) {
            	return charset;
            }
        }
        return null;
    }
    
    @SuppressWarnings("unused")
    private static class Validate {
        
        private Validate() {}

        /**
         * 验证是否为true
         * @param val
         */
		public static void isTrue(boolean val) {
            if (!val) {
            	throw new IllegalArgumentException("Must be true");
            }
        }

        /**
         * 验证是否为true
         * @param val
         * @param msg
         */
        public static void isTrue(boolean val, String msg) {
            if (!val) {
            	throw new IllegalArgumentException(msg);
            }
        }

        /**
         * 验证是否为false
         * @param val
         */
        public static void isFalse(boolean val) {
            if (val) {
            	throw new IllegalArgumentException("Must be false");
            }
        }

        /**
         * 验证是否为false
         * @param val
         * @param msg
         */
        public static void isFalse(boolean val, String msg) {
            if (val) {
            	throw new IllegalArgumentException(msg);
            }
        }

        /**
         * 验证消息是否为空
         * @param string 
         */
        public static void notNull(Object bean) {
            if (bean == null) {
            	throw new IllegalArgumentException("Object must not be null");
            }
        }

        /**
         * 验证字符串是否为空
         * @param string
         * @param msg
         */
        public static void notNull(Object bean, String msg) {
            if (bean == null){
            	throw new IllegalArgumentException(msg);
            }
        }
        
        /**
         * 验证消息是否为空
         * @param string 
         */
        public static void notEmpty(String string) {
        	if (string == null || string.length() == 0) {
        		throw new IllegalArgumentException("String must not be empty");
        	}
        }
        
        /**
         * 验证字符串是否为空
         * @param string
         * @param msg
         */
        public static void notEmpty(String string, String msg) {
        	if (string == null || string.length() == 0){
        		throw new IllegalArgumentException(msg);
        	}
        }
        
        /**
         * 失败
         * @param string 
         */
        public static void isFail() {
        	throw new IllegalArgumentException("Unknown exception");
        }
        
        /**
         * 失败
         * @param string
         * @param msg
         */
        public static void isFail(String msg) {
        	throw new IllegalArgumentException(msg);
        }

    }
    
}

 

 

 

1
4
分享到:
评论
2 楼 anlystar 2013-06-20  
huxiulei 写道
Validate.notEmpty(key, "Data key must not be empty");  是少了一个类么 
Validate???

不好意思,代码太长,Iteye给我截断了,重新添加了,现在是完整的
1 楼 huxiulei 2013-06-20  
Validate.notEmpty(key, "Data key must not be empty");  是少了一个类么 
Validate???

相关推荐

Global site tag (gtag.js) - Google Analytics