需求:
定时任务:每天统计昨天的公司支付宝账户实际收益(扣除手续费)
流程:
1 、调用支付宝接口, 获取zip 下载地址
package com.ycmedia.task;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.alipay.api.AlipayClient;
import com.alipay.api.DefaultAlipayClient;
import com.alipay.api.request.AlipayDataDataserviceBillDownloadurlQueryRequest;
import com.alipay.api.response.AlipayDataDataserviceBillDownloadurlQueryResponse;
import com.google.common.base.Splitter;
import com.ycmedia.constants.Constants;
import com.ycmedia.dao.TaskDao;
import com.ycmedia.entity.RealIncom;
import com.ycmedia.utils.FileUtil;
import com.ycmedia.utils.ReadCsv;
import org.joda.time.DateTime;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.env.Environment;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;
import java.io.File;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;
@Component
public class AliBillTask {
@Autowired
private Environment env;
SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd");
@Autowired
private TaskDao taskDao;
@Scheduled(cron = "0 14 14 ? * *")
public void runAlitask() throws Exception {
downloadBill();
}
/**
* 获取指定日期的账单
*/
public void downloadBill() throws Exception {
// 将订单提交至支付宝
AlipayClient alipayClient = new DefaultAlipayClient(
"https://openapi.alipay.com/gateway.do", Constants.ALI_APP_ID,
Constants.ALI_PRIVATE_KEY, "json", "utf-8",
Constants.ALI_DEV_PLAT_PUBLIC_KEY);
AlipayDataDataserviceBillDownloadurlQueryRequest request = new AlipayDataDataserviceBillDownloadurlQueryRequest();
JSONObject json = new JSONObject();
json.put("bill_type", "trade");
//昨天的数据
json.put("bill_date", new DateTime().minusDays(1).toString("yyyy-MM-dd"));
request.setBizContent(json.toString());
AlipayDataDataserviceBillDownloadurlQueryResponse response = alipayClient
.execute(request);
if(response.isSuccess()){
// 获取下载地址url
String url = response.getBillDownloadUrl();
// 设置下载后生成Zip目录
String filePath = env.getProperty("file.path");
String newZip = filePath + new Date().getTime() + ".zip";
// 开始下载
FileUtil.downloadNet(url, newZip);
// 解压到指定目录
FileUtil.unZip(newZip, env.getProperty("file.zip.path"));
// 遍历文件 获取需要的汇整csv
File[] fs = new File(env.getProperty("file.zip.path")).listFiles();
RealIncom income = null;
for (File file : fs) {
if (file.getAbsolutePath().contains("汇总")) {
Double money = ReadCsv.getMoney("", file.getAbsolutePath());
income = new RealIncom();
income.setDate(new Date());
income.setMoney(money);
taskDao.insertTodayMoney(income);
}
}
// 插入成功, 删除csv 文件
for (File file : fs) {
file.delete();
}
}else{
//如果账单不存在, 插入一条空数据到数据库
RealIncom income= new RealIncom();
income.setDate(new Date());
income.setMoney(0.00);
taskDao.insertTodayMoney(income);
}
if (response.isSuccess()) {
System.out.println("调用成功");
} else {
System.out.println("调用失败");
}
System.out.println(JSON.toJSONString(response));
}
}
工具类代码:
package com.ycmedia.utils; import java.io.BufferedInputStream; import java.io.BufferedOutputStream; import java.io.File; import java.io.FileNotFoundException; import java.io.FileOutputStream; import java.io.IOException; import java.io.InputStream; import java.net.MalformedURLException; import java.net.URL; import java.net.URLConnection; import java.util.Enumeration; import org.apache.tools.zip.ZipEntry; import org.apache.tools.zip.ZipFile; public class FileUtil { /** * 使用GBK编码可以避免压缩中文文件名乱码 */ private static final String CHINESE_CHARSET = "GBK"; /** * 文件读取缓冲区大小 */ private static final int CACHE_SIZE = 1024; /** * 第一步: 把 支付宝生成的账单 下载到本地目录 * * @param path * 支付宝资源url * @param filePath * 生成的zip 包目录 * @throws MalformedURLException */ public static void downloadNet(String path, String filePath) throws MalformedURLException { // 下载网络文件 int bytesum = 0; int byteread = 0; URL url = new URL(path); try { URLConnection conn = url.openConnection(); InputStream inStream = conn.getInputStream(); FileOutputStream fs = new FileOutputStream(filePath); byte[] buffer = new byte[1204]; while ((byteread = inStream.read(buffer)) != -1) { bytesum += byteread; fs.write(buffer, 0, byteread); } } catch (FileNotFoundException e) { e.printStackTrace(); } catch (IOException e) { e.printStackTrace(); } } public static void unZip(String zipFilePath, String destDir) throws Exception { ZipFile zipFile = new ZipFile(zipFilePath, CHINESE_CHARSET); Enumeration<?> emu = zipFile.getEntries(); BufferedInputStream bis; FileOutputStream fos; BufferedOutputStream bos; File file, parentFile; ZipEntry entry; byte[] cache = new byte[CACHE_SIZE]; while (emu.hasMoreElements()) { entry = (ZipEntry) emu.nextElement(); if (entry.isDirectory()) { new File(destDir + entry.getName()).mkdirs(); continue; } bis = new BufferedInputStream(zipFile.getInputStream(entry)); file = new File(destDir + entry.getName()); parentFile = file.getParentFile(); if (parentFile != null && (!parentFile.exists())) { parentFile.mkdirs(); } fos = new FileOutputStream(file); bos = new BufferedOutputStream(fos, CACHE_SIZE); int nRead = 0; while ((nRead = bis.read(cache, 0, CACHE_SIZE)) != -1) { fos.write(cache, 0, nRead); } bos.flush(); bos.close(); fos.close(); bis.close(); } zipFile.close(); } }
目录:
开发环境:
========================================================================================================================
更新实际收益到本地数据库查询
====================更新与2018-11-19 【ReadCsv.java】===================================================
import java.io.BufferedReader; import java.io.FileInputStream; import java.io.InputStreamReader; import java.util.HashMap; public class ReadCsv { // public static void main(String[] args) { // try { // // BufferedReader reader=new BufferedReader(new InputStreamReader(new // FileInputStream("D:\\down\\1476771240197\\20884217298464250156_20160914_业务明细.csv"),"gbk")); // reader.read();//第一行信息,为标题信息,不用,如果需要,注释掉 // String line = null; // // // while((line=reader.readLine())!=null){ // String item[] = line.split("\r\n");//CSV格式文件为逗号分隔符文件,这里根据逗号切分 // // String last = item[item.length-1];//这就是你要的数据了 // // if(last.contains(")")){ // System.out.println(Double.valueOf(last.split(",")[12])-Double.valueOf(last.split(",")[22])); // } // } // } catch (Exception e) { // e.printStackTrace(); // } // } public static HashMap<String, Double> getDetailOrder(String filePath) { HashMap<String, Double> map = new HashMap<String, Double>(); try { BufferedReader reader = new BufferedReader(new InputStreamReader( new FileInputStream(filePath), "gbk")); reader.read();// 第一行信息,为标题信息,不用,如果需要,注释掉 String line = null; while ((line = reader.readLine()) != null) { String item[] = line.split("\r\n");// CSV格式文件为逗号分隔符文件,这里根据逗号切分 String last = item[item.length - 1];// 这就是你要的数据了 if (last.contains(")")) { map.put(last.split(",")[0], Double.valueOf(last.split(",")[12]) - Double.valueOf(last.split(",")[22])); } } } catch (Exception e) { e.printStackTrace(); } return map; } /** * 根绝类型获取指定行的数据 * * @param type * @return */ public static Double getMoney(String type, String filePath) { Double money = null; try { BufferedReader reader = new BufferedReader(new InputStreamReader( new FileInputStream(filePath), "gbk")); reader.read();// String line = null; while ((line = reader.readLine()) != null) { String item[] = line.split("\r\n"); String last = item[item.length - 1]; if (last.contains("合计")) { String[] strs = last.split(","); money = Double.valueOf(strs[strs.length - 1]); } } } catch (Exception e) { e.printStackTrace(); } return money; } }