import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.net.URL;
import java.net.URLConnection;
import java.net.URLEncoder;
import java.util.ArrayList;
import java.util.List;
import org.apache.http.HttpResponse;
import org.apache.http.NameValuePair;
import org.apache.http.client.HttpClient;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.impl.client.HttpClientBuilder;
import org.apache.http.message.BasicNameValuePair;
import org.apache.http.util.EntityUtils;
import org.apache.log4j.Logger;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
/**
* @ClassName: ShortUrlUtil
* @Description: 工具类:借助第三方,将长连接转换为短连接
* @date 2018/07/23 17:21:48
*
*/
public class ShortUrlUtil {
private static Logger log = Logger.getLogger(ShortUrlUtil.class);
public static ShortUrlUtil shortUrlUtil;
public static HttpClient httpclient;
static {
shortUrlUtil = new ShortUrlUtil();
httpclient = HttpClientBuilder.create().build();
}
/**
* 生成短连接信息
*/
public String generateShortUrl(String url) {
try {
String apiUrl = "http://suo.im/api.php";
HttpPost httpost = new HttpPost(apiUrl);
List<NameValuePair> params = new ArrayList<NameValuePair>();
params.add(new BasicNameValuePair("format", "utf-8")); // 编码
params.add(new BasicNameValuePair("url", url)); // 用户名称
httpost.setEntity(new UrlEncodedFormEntity(params, "utf-8"));
HttpResponse response = httpclient.execute(httpost);
String jsonStr = EntityUtils.toString(response.getEntity(), "utf-8");
return jsonStr;
} catch (Exception e) {
shortUrlUtil = new ShortUrlBySinaUtil();
log.info("ShortUrlUtil", e);
return "";
}
}
/**
* 测试生成短连接
*/
public static void main(String[] args) {
String url = ShortUrlUtil.shortUrlUtil.generateShortUrl("https://mp.weixin.qq.com/debug/cgi-bin/sandbox?t=sandbox/login");
System.err.println(url);
}
class ShortUrlBySinaUtil extends ShortUrlUtil {
static final String actionUrl = "http://api.t.sina.com.cn/short_url/shorten.json";
static final String APPKEY = "2024358821,3925598208,3271760578,31641035,2815391962";
public String generateShortUrl(String longUrl) {
try {
longUrl = URLEncoder.encode(longUrl, "UTF-8");
String appkey = APPKEY;
String[] sourceArray = appkey.split(",");
for (String key : sourceArray) {
String shortUrl = sinaShortUrl(key, longUrl);
System.out.println("key====>"+ key + " " + shortUrl);
if (shortUrl != null) {
return shortUrl;
}
}
} catch (Exception e) {
shortUrlUtil = new ShortUrlBySuoUtil();
log.info("ShortUrlBySinaUtil", e);
}
return "";
}
public String sinaShortUrl(String source, String longUrl) {
String result = sendPost(actionUrl, "url_long=" + longUrl + "&source=" + source);
if (result == null || "".equals(result)) {
return "";
}
JSONArray jsonArr = JSON.parseArray(result);
JSONObject json = JSON.parseObject(jsonArr.get(0).toString());
String ret = json.get("url_short").toString();
return ret;
}
/**
* 向指定 URL 发送POST方法的请求
*
* @param url 发送请求的 URL
* @param param 请求参数,请求参数应该是 name1=value1&name2=value2 的形式。
* @return 所代表远程资源的响应结果
*/
public String sendPost(String url, String param) {
PrintWriter out = null;
BufferedReader in = null;
String result = "";
try {
URL realUrl = new URL(url);
// 打开和URL之间的连接
URLConnection conn = realUrl.openConnection();
// 设置通用的请求属性
conn.setRequestProperty("accept", "*/*");
conn.setRequestProperty("connection", "Keep-Alive");
conn.setRequestProperty("user-agent", "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1;SV1)");
// 发送POST请求必须设置如下两行
conn.setDoOutput(true);
conn.setDoInput(true);
// 获取URLConnection对象对应的输出流
out = new PrintWriter(conn.getOutputStream());
// 发送请求参数
out.print(param);
// flush输出流的缓冲
out.flush();
// 定义BufferedReader输入流来读取URL的响应
in = new BufferedReader(new InputStreamReader(conn.getInputStream()));
String line;
while ((line = in.readLine()) != null) {
result += line;
}
} catch (Exception e) {
System.out.println("发送 POST 请求出现异常!" + e);
e.printStackTrace();
}
//使用finally块来关闭输出流、输入流
finally {
try {
if (out != null) {
out.close();
}
if (in != null) {
in.close();
}
} catch (IOException ex) {
ex.printStackTrace();
}
}
return result;
}
}
class ShortUrlBySuoUtil extends ShortUrlUtil {
/**
* 生成短连接信息
*/
public String generateShortUrl(String url) {
try {
String apiUrl = "http://a.bbest.me/short/create";
HttpPost httpost = new HttpPost(apiUrl);
List<NameValuePair> params = new ArrayList<NameValuePair>();
params.add(new BasicNameValuePair("format", "utf-8")); // 编码
params.add(new BasicNameValuePair("l", url)); // 用户名称
httpost.setEntity(new UrlEncodedFormEntity(params, "utf-8"));
HttpResponse response = httpclient.execute(httpost);
String jsonStr = EntityUtils.toString(response.getEntity(), "utf-8");
JSONObject jsonResult = JSON.parseObject(jsonStr);
String shortUrl = jsonResult.getString("short_url");
return "http://bbest.me/"+ shortUrl;
} catch (Exception e) {
log.info("ShortUrlBySuoUtil", e);
return "";
}
}
}
}