java微信開發之上傳下載多媒體文件

import java.io.BufferedOutputStream;
import java.io.BufferedReader;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URL;
import java.security.KeyManagementException;
import java.security.NoSuchAlgorithmException;
import java.security.NoSuchProviderException;
import java.security.SecureRandom;
import java.util.Calendar;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
 
import javax.net.ssl.HttpsURLConnection;
import javax.net.ssl.SSLContext;
import javax.net.ssl.SSLSocketFactory;
import javax.net.ssl.TrustManager;
 
import net.sf.json.JSONObject;
 
import org.apache.commons.lang.StringUtils;
import org.apache.log4j.Logger;
 
import com.company.project.model.menu.AccessToken;
import com.company.project.model.menu.Menu;
 
public class WeixinUtil {
 private static Logger log = Logger.getLogger(WeixinUtil.class);
 public final static String APPID = "wxb927d4280e6db674";
 public final static String APP_SECRET = "21441e9f3226eee81e14380a768b6d1e";
 // 获取access_token的接口地址（GET） 限200（次/天）
 public final static String access_token_url = "https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid=APPID&secret=APPSECRET";
 // 创建菜单
 public final static String create_menu_url = "https://api.weixin.qq.com/cgi-bin/menu/create?access_token=ACCESS_TOKEN";
 // 存放：1.token，2：获取token的时间,3.过期时间
 public final static Map<String,Object> accessTokenMap = new HashMap<String,Object>();
 /**
 * 发起https请求并获取结果
 * 
 * @param requestUrl 请求地址
 * @param requestMethod 请求方式（GET、POST）
 * @param outputStr 提交的数据
 * @return JSONObject(通过JSONObject.get(key)的方式获取json对象的属性值)
 */
 public static JSONObject handleRequest(String requestUrl,String requestMethod,String outputStr) {
 JSONObject jsonObject = null;
  
 try {
  URL url = new URL(requestUrl);
  HttpsURLConnection conn = (HttpsURLConnection) url.openConnection();
  SSLContext ctx = SSLContext.getInstance("SSL", "SunJSSE");
  TrustManager[] tm = {new MyX509TrustManager()};
  ctx.init(null, tm, new SecureRandom());
  SSLSocketFactory sf = ctx.getSocketFactory();
  conn.setSSLSocketFactory(sf);
  conn.setDoInput(true);
  conn.setDoOutput(true);
  conn.setRequestMethod(requestMethod);
  conn.setUseCaches(false);
   
  if ("GET".equalsIgnoreCase(requestMethod)) {
  conn.connect();
  }
   
  if (StringUtils.isNotEmpty(outputStr)) {
  OutputStream out = conn.getOutputStream();
  out.write(outputStr.getBytes("utf-8"));
  out.close();
  }
   
  InputStream in = conn.getInputStream();
  BufferedReader br = new BufferedReader(new InputStreamReader(in,"utf-8"));
  StringBuffer buffer = new StringBuffer();
  String line = null;
   
  while ((line = br.readLine()) != null) {
  buffer.append(line);
  }
   
  in.close();
  conn.disconnect();
   
  jsonObject = JSONObject.fromObject(buffer.toString());
 } catch (MalformedURLException e) {
  e.printStackTrace();
  log.error("URL错误！");
 } catch (IOException e) {
  e.printStackTrace();
 } catch (NoSuchAlgorithmException e) {
  e.printStackTrace();
 } catch (NoSuchProviderException e) {
  e.printStackTrace();
 } catch (KeyManagementException e) {
  e.printStackTrace();
 }
 return jsonObject;
 }
  
 /**
 * 获取access_token
 *
 * @author qincd
 * @date Nov 6, 2014 9:56:43 AM
 */
 public static AccessToken getAccessToken(String appid,String appSecret) {
 AccessToken at = new AccessToken();
 // 每次获取access_token时，先从accessTokenMap获取，如果过期了就重新从微信获取
 // 没有过期直接返回
 // 从微信获取的token的有效期为2个小时
 if (!accessTokenMap.isEmpty()) {
  Date getTokenTime = (Date) accessTokenMap.get("getTokenTime");
  Calendar c = Calendar.getInstance();
  c.setTime(getTokenTime);
  c.add(Calendar.HOUR_OF_DAY, 2);
   
  getTokenTime = c.getTime();
  if (getTokenTime.after(new Date())) {
  log.info("缓存中发现token未过期，直接从缓存中获取access_token");
  // token未过期，直接从缓存获取返回
  String token = (String) accessTokenMap.get("token");
  Integer expire = (Integer) accessTokenMap.get("expire");
  at.setToken(token);
  at.setExpiresIn(expire);
  return at;
  }
 }
 String requestUrl = access_token_url.replace("APPID", appid).replace("APPSECRET", appSecret);
  
 JSONObject object = handleRequest(requestUrl, "GET", null);
 String access_token = object.getString("access_token");
 int expires_in = object.getInt("expires_in");
  
 log.info("\naccess_token:" + access_token);
 log.info("\nexpires_in:" + expires_in);
  
 at.setToken(access_token);
 at.setExpiresIn(expires_in);
  
 // 每次获取access_token后，存入accessTokenMap
 // 下次获取时，如果没有过期直接从accessTokenMap取。
 accessTokenMap.put("getTokenTime", new Date());
 accessTokenMap.put("token", access_token);
 accessTokenMap.put("expire", expires_in);
  
 return at;
 }
  
 /**
 * 创建菜单
 *
 * @author qincd
 * @date Nov 6, 2014 9:56:36 AM
 */
 public static boolean createMenu(Menu menu,String accessToken) {
 String requestUrl = create_menu_url.replace("ACCESS_TOKEN", accessToken);
 String menuJsonString = JSONObject.fromObject(menu).toString();
 JSONObject jsonObject = handleRequest(requestUrl, "POST", menuJsonString);
 String errorCode = jsonObject.getString("errcode");
 if (!"0".equals(errorCode)) {
  log.error(String.format("菜单创建失败！errorCode:%d,errorMsg:%s",jsonObject.getInt("errcode"),jsonObject.getString("errmsg")));
  return false;
 }
  
 log.info("菜单创建成功！");
  
 return true;
 }
  
 // 上传多媒体文件到微信服务器
 public static final String upload_media_url = "http://file.api.weixin.qq.com/cgi-bin/media/upload?access_token=ACCESS_TOKEN&type=TYPE";
 /**
 * 上传多媒体文件到微信服务器<br>
 * @see http://www.oschina.net/code/snippet_1029535_23824
 * @see http://mp.weixin.qq.com/wiki/index.php?title=上传下载多媒体文件
 *
 * @author qincd
 * @date Nov 6, 2014 4:11:22 PM
 */
 public static JSONObject uploadMediaToWX(String filePath,String type,String accessToken) throws IOException{
 File file = new File(filePath);
 if (!file.exists()) {
  log.error("文件不存在！");
  return null;
 }
  
 String url = upload_media_url.replace("ACCESS_TOKEN", accessToken).replace("TYPE", type);
 URL urlObj = new URL(url);
 HttpURLConnection conn = (HttpURLConnection) urlObj.openConnection();
 conn.setDoInput(true);
 conn.setDoOutput(true);
 conn.setUseCaches(false);
  
 conn.setRequestProperty("Connection", "Keep-Alive");
    conn.setRequestProperty("Charset", "UTF-8");
  
    // 设置边界
    String BOUNDARY = "----------" + System.currentTimeMillis();
    conn.setRequestProperty("Content-Type", "multipart/form-data; boundary="
        + BOUNDARY);
  
    // 请求正文信息
  
    // 第一部分：
    StringBuilder sb = new StringBuilder();
    sb.append("--"); // ////////必须多两道线
    sb.append(BOUNDARY);
    sb.append("\r\n");
    sb.append("Content-Disposition: form-data;name=\"file\";filename=\""
        + file.getName() + "\"\r\n");
    sb.append("Content-Type:application/octet-stream\r\n\r\n");
  
    byte[] head = sb.toString().getBytes("utf-8");
  
    // 获得输出流
    OutputStream out = new DataOutputStream(conn.getOutputStream());
    out.write(head);
  
    // 文件正文部分
    DataInputStream in = new DataInputStream(new FileInputStream(file));
    int bytes = 0;
    byte[] bufferOut = new byte[1024];
    while ((bytes = in.read(bufferOut)) != -1) {
      out.write(bufferOut, 0, bytes);
    }
    in.close();
  
    // 结尾部分
    byte[] foot = ("\r\n--" + BOUNDARY + "--\r\n").getBytes("utf-8");// 定义最后数据分隔线
  
    out.write(foot);
  
    out.flush();
    out.close();
  
    /**
     * 读取服务器响应，必须读取,否则提交不成功
     */
     try {
     // 定义BufferedReader输入流来读取URL的响应
     StringBuffer buffer = new StringBuffer();
     BufferedReader reader = new BufferedReader(new InputStreamReader(
     conn.getInputStream()));
     String line = null;
     while ((line = reader.readLine()) != null) {
      buffer.append(line);
     }
      
     reader.close();
     conn.disconnect();
      
     return JSONObject.fromObject(buffer.toString());
     } catch (Exception e) {
     log.error("发送POST请求出现异常！" + e);
     e.printStackTrace();
     }
 return null;
 }
  
 public static final String download_media_url = "http://file.api.weixin.qq.com/cgi-bin/media/get?access_token=ACCESS_TOKEN&media_id=MEDIA_ID";
 /**
 * 从微信服务器下载多媒体文件
 *
 * @author qincd
 * @date Nov 6, 2014 4:32:12 PM
 */
 public static String downloadMediaFromWx(String accessToken,String mediaId,String fileSavePath) throws IOException {
 if (StringUtils.isEmpty(accessToken) || StringUtils.isEmpty(mediaId)) return null;
  
 String requestUrl = download_media_url.replace("ACCESS_TOKEN", accessToken).replace("MEDIA_ID", mediaId);
 URL url = new URL(requestUrl);
 HttpURLConnection conn = (HttpURLConnection) url.openConnection();
 conn.setRequestMethod("GET");
 conn.setDoInput(true);
 conn.setDoOutput(true);
 InputStream in = conn.getInputStream();
  
 File dir = new File(fileSavePath);
 if (!dir.exists()) {
  dir.mkdirs();
 }
 if (!fileSavePath.endsWith("/")) {
  fileSavePath += "/";
 }
  
 String ContentDisposition = conn.getHeaderField("Content-disposition");
 String weixinServerFileName = ContentDisposition.substring(ContentDisposition.indexOf("filename")+10, ContentDisposition.length() -1);
 fileSavePath += weixinServerFileName; 
 BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(fileSavePath));
 byte[] data = new byte[1024];
 int len = -1;
  
 while ((len = in.read(data)) != -1) {
  bos.write(data,0,len);
 }
  
 bos.close();
 in.close();
 conn.disconnect();
  
 return fileSavePath;
 }
}

public class WeixinUtilTest {
 
 /**
 *
 * @author qincd
 * @date Nov 6, 2014 9:57:54 AM
 */
 public static void main(String[] args) {
 // 1).获取access_token
 AccessToken accessToken = WeixinUtil.getAccessToken(WeixinUtil.APPID, WeixinUtil.APP_SECRET);
 String filePath = "C:\\Users\\qince\\Pictures\\壁纸20141029091612.jpg";
 JSONObject uploadJsonObj = testUploadMedia(filePath,"image",accessToken.getToken());
 if (uploadJsonObj == null) {
  System.out.println("上传图片失败");
  return;
 }
  
 int errcode = 0;
 if (uploadJsonObj.containsKey("errcode")) {
  errcode = uploadJsonObj.getInt("errcode");
 }
 if (errcode == 0) {
  System.out.println("图片上传成功");
   
  String mediaId = uploadJsonObj.getString("media_id");
  long createAt = uploadJsonObj.getLong("created_at");
  System.out.println("--Details:");
  System.out.println("media_id:" + mediaId);
  System.out.println("created_at:" + createAt);
 }
 else {
  System.out.println("图片上传失败！");
   
  String errmsg = uploadJsonObj.getString("errmsg");
  System.out.println("--Details:");
  System.out.println("errcode:" + errcode);
  System.out.println("errmsg:" + errmsg);
 }
  
 String mediaId = "6W-UvSrQ2hkdSdVQJJXShwtFDPLfbGI1qnbNFy8weZyb9Jac2xxxcAUwt8p4sYPH";
 String filepath = testDownloadMedia(accessToken.getToken(), mediaId, "d:/test");
 System.out.println(filepath);
 }
 
 
 /**
 * 上传多媒体文件到微信
 *
 * @author qincd
 * @date Nov 6, 2014 4:15:14 PM
 */
 public static JSONObject testUploadMedia(String filePath,String type,String accessToken) {
 try {
  return WeixinUtil.uploadMediaToWX(filePath, type, accessToken);
 } catch (IOException e) {
  e.printStackTrace();
 }
  
 return null;
 }
  
 /**
 * 从微信下载多媒体文件
 *
 * @author qincd
 * @date Nov 6, 2014 4:56:25 PM
 */
 public static String testDownloadMedia(String accessToken,String mediaId,String fileSaveDir) {
 try {
  return WeixinUtil.downloadMediaFromWx(accessToken, mediaId, fileSaveDir);
 } catch (IOException e) {
  e.printStackTrace();
 }
  
 return null;
 }
}