多线程(callable+futureTask)去组装数据,并批量入库

2024-06-19 02:32

本文主要是介绍多线程(callable+futureTask)去组装数据,并批量入库,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

      商城项目,收货地址会用到4级地址(省,市,县,镇),我们只用到了特定城市的。 但是我想通过京东的接口把全部的数据拿出来。于是就有 ------多线程(callable+futureTask)去组装数据。

---------------------------

    先贴下controller的代码:

package com.truelore.xunjia.wssc.test.controller;


import com.alibaba.fastjson.JSON;
import com.truelore.common.util.WsscHttpClientUtils;
import com.truelore.xunjia.wssc.dao.ProvinceDao;
import com.truelore.xunjia.wssc.entity.WsscArea;
import com.truelore.xunjia.wssc.entity.WsscCity;
import com.truelore.xunjia.wssc.entity.WsscProvince;
import com.truelore.xunjia.wssc.service.AreaService;
import com.truelore.xunjia.wssc.service.CityService;
import com.truelore.xunjia.wssc.service.ProvinceService;
import com.truelore.xunjia.wssc.service.TownService;
import com.truelore.xunjia.wssc.vo.JdaddressVo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.*;
import java.util.concurrent.Callable;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.FutureTask;

/**
 * 拿到京东全部的4级地址
 *  fujf
 *  201853114:49:49
 */
@RequestMapping("/wssc/test")
@Controller("testGetallJdaddress")
public class TestGetallJdaddress extends SuperToken {@Autowired
    private ProvinceService provinceService;

    @Autowired
    private ProvinceDao provinceDao;

    @Autowired
    private CityService cityService;

    @Autowired
    private AreaService areaService;

    @Autowired
    private TownService townService;

    @RequestMapping("/saveAddress")public void saveAddress(String level, HttpServletResponse resp){//level ="2";
        Long start_time = System.currentTimeMillis();   //开始时间

        List<String> parentids = new ArrayList<String>();
         if("1".equals(level)){//获取省的时候,没有parentid.  直接保存好了
             List<JdaddressVo> jdaddressVos = getaddressList(level, null);
            //save(jdaddressVos,level);
             batsave(jdaddressVos,level);

        }else if("2".equals(level)){List<WsscProvince> ProvinceList = provinceService.getLocalProvince(99);
             for (WsscProvince wsscProvince : ProvinceList) {parentids.add(wsscProvince.getLocalProvinceId());
             }}else if("3".equals(level)){List<WsscCity> cityList = cityService.queryCityByCondition(null,null,null,99);
             for (WsscCity wsscCity : cityList) {parentids.add(wsscCity.getLocalCityId());
             }}else if("4".equals(level)){List<WsscArea> areaList = areaService.queryAreaByCondition(null,null,null,99);
             for (WsscArea wsscArea : areaList) {parentids.add(wsscArea.getLocalAreaId());
             }}if(!("1".equals(level))){         //不是省级的地址获取,我们就用下面的多线程方式
      List<FutureTask<List<JdaddressVo>>> futureTasks = new ArrayList<FutureTask<List<JdaddressVo>>>();
            ExecutorService executorService = Executors.newFixedThreadPool(50);
            MycallableForaddress callable = null;


            System.out.println("****************");
            for (String parentid : parentids) {callable = new MycallableForaddress(level,parentid);
                FutureTask<List<JdaddressVo>> futureTask = new FutureTask<List<JdaddressVo>>(callable);
                futureTasks.add(futureTask);
                executorService.submit(futureTask);
              while(futureTasks.size()==500){for (FutureTask<List<JdaddressVo>> task : futureTasks) {try {List<JdaddressVo> addrlist = task.get();
                          if(null!=addrlist){//save(addrlist,level);
                              batsave(addrlist,level);   //换成批量保存
                          }} catch (Exception e) {e.printStackTrace();
                      }}futureTasks.clear();
              }}//循环结束,最后不满futureTasks.size()的也要保存起来
            while (futureTasks.size() > 0) {for (FutureTask<List<JdaddressVo>> task : futureTasks) {try {List<JdaddressVo> addrlist = task.get();
                        batsave(addrlist,level);
                    } catch (Exception e) {e.printStackTrace();
                    }}futureTasks.clear();
            }executorService.shutdown();
        }try {Long end_time = System.currentTimeMillis();
            resp.setHeader("Content-type", "text/html;charset=UTF-8");
            resp.getWriter().write("ok");
            resp.getWriter().write("共用时:"+(end_time - start_time)+"毫秒");
        } catch (IOException e) {e.printStackTrace();
        }}//批量保存
    private void batsave(List<JdaddressVo> addrlist, String level) {if("1".equals(level)){for (JdaddressVo jdaddressVo : addrlist) {WsscProvince p = new WsscProvince();
                p.setGuid(UUID.randomUUID().toString());
                p.setLocalProvinceId(jdaddressVo.getAddressId());
                p.setProvinceName(jdaddressVo.getAddressName());
                p.setTarget(99);
                p.setTargetProvinceId(jdaddressVo.getAddressId());
                provinceService.save(p);
            }}else if("2".equals(level)){cityService.batsave(addrlist);
        }else if("3".equals(level)){areaService.batsave(addrlist);
        }else{townService.batsave(addrlist);
      }}//内部线程类    根据上级id返回下级的地址list<JdaddressVo>
    class MycallableForaddress implements Callable{private String level;

        private String parentId;

       public MycallableForaddress(String level, String parentId) {this.level = level;
           this.parentId = parentId;
       }@Override
        public Object call() throws Exception {List<JdaddressVo> jdaddressVos = getaddressList(level, parentId);
            return jdaddressVos;
        }}//返回map<地区名,编号>
      private List<JdaddressVo> getaddressList(String level,String parentId) {String url = null;
          Map maps = new HashMap<String, String>();
          maps.put("token", token);

          if ("1" .equals(level)) {                      //获取省
              url="https://bizapi.jd.com/api/area/getProvince";
          } else if ("2" .equals(level)) {               //获取市
              url="https://bizapi.jd.com/api/area/getCity";
              maps.put("id", parentId);
          } else if ("3" .equals(level)) {                //获取县
              url="https://bizapi.jd.com/api/area/getCounty";
              maps.put("id", parentId);
          } else if ("4".equals(level)) {                 //获取乡
              url="https://bizapi.jd.com/api/area/getTown";
              maps.put("id", parentId);
          } else {return null;
          }String rev = WsscHttpClientUtils.post(url, maps, null);

          if (null != rev) {Map resultmaps = (Map) JSON.parse(rev);
              System.out.println(resultmaps.get("success"));
              boolean isSuccess = (boolean) resultmaps.get("success");
              if (isSuccess) {Map<String, Integer> resultmap = (Map) resultmaps.get("result");
                  //遍历map,方法1
                  List<JdaddressVo> JdaddressList = new ArrayList<>();
                  for (Object key : resultmap.keySet()) {System.out.println(key + "---->" + resultmap.get(key));
                      JdaddressVo jdaddress = new JdaddressVo();
                      jdaddress.setAddressId(resultmap.get(key).toString());
                      jdaddress.setAddressName(key.toString());
                      jdaddress.setParentAddressId(parentId);
                      JdaddressList.add(jdaddress);
                  }return JdaddressList;
              }}else{return null;
          }return null;

      }//-------------------以下是单元测试,不用理会-----------------------------------------
   // @Test
    public void getProvinceList(){String url = "https://bizapi.jd.com/api/area/getProvince";
        Map maps = new HashMap<String,String>();
        maps.put("token", token);
        String rev = WsscHttpClientUtils.post(url, maps, null);
        System.out.println(rev);
    }// @Test
    public void getCityList(){String url = "https://bizapi.jd.com/api/area/getCity";
        String parentId ="6";
        Map maps = new HashMap<String,String>();
        maps.put("id",parentId);
        maps.put("token", token);

        String rev = WsscHttpClientUtils.post(url, maps, null);
        System.out.println(rev);
    }// @Test
    public void getCountyList(){String url = "https://bizapi.jd.com/api/area/getCounty";
        String parentId ="318";
        Map maps = new HashMap<String,String>();
        maps.put("id",parentId);
        maps.put("token", token);

        String rev = WsscHttpClientUtils.post(url, maps, null);
        System.out.println(rev);
    }// @Test
    public void getTownList() {String url = "https://bizapi.jd.com/api/area/getTown";
        String parentId = "319";
        Map maps = new HashMap<String, String>();
        maps.put("id", parentId);
        maps.put("token", token);

        String rev = WsscHttpClientUtils.post(url, maps, null);
        System.out.println(rev);

        if (null != rev) {Map resultmaps = (Map) JSON.parse(rev);
            System.out.println(resultmaps.get("success"));
            boolean isSuccess = (boolean) resultmaps.get("success");
            if (isSuccess) {Map<String,Integer> resultmap = (Map) resultmaps.get("result");

               //遍历map,方法1
                for (Object key : resultmap.keySet()){System.out.println(key+"---->"+resultmap.get(key));
                }//遍历map,方法2
               /* for (Map.Entry<String,Integer> entry : resultmap.entrySet()){
                    System.out.println(entry.getKey()+"---->"+entry.getValue());
                }*/

                //遍历map,方法3   迭代器
               /* Iterator keys = resultmap.keySet().iterator();
                while (keys.hasNext()){
                   String key = (String) keys.next();
                   System.out.println(key+"--->"+resultmap.get(key));
                }*/

            }}}
}
 

这个是内部线程类的定义。

***DaoImpl中的批量保存代码。

@Override
public void batsave(final List<JdaddressVo> addrs) {this.getSession().doWork(new Work() {@Override
                         public void execute(Connection connection) throws SQLException {String sql = "insert into WSSC_CITY(GUID,CITY_NAME,LOCAL_CITYID,LOCAL_PROVINCEID,TARGET_CITYID,TARGET_PROVINCEID,TARGET) values(?,?,?,?,?,?,?)";
                            PreparedStatement ps = connection.prepareStatement(sql);
                            for (JdaddressVo addr : addrs) {ps.setString(1, UUID.randomUUID().toString());
                               ps.setString(2,addr.getAddressName());
                               ps.setString(3,addr.getAddressId());
                               ps.setString(4,addr.getParentAddressId());
                               ps.setString(5,addr.getAddressId());
                               ps.setString(6,addr.getParentAddressId());
                               ps.setInt(7,99);
                               ps.addBatch();
                            }ps.executeBatch();
                         }});

}


-----------------

经过测试,这样处理,5万条数据导入需要几分钟,快了不少。

------

感受:

要培养一种“批量”,“缓存”的思想,比如上面代码中 的 futureTasks(满500再处理);保存数据时,jdbc去批处理等。

"满一定量再去做"


这篇关于多线程(callable+futureTask)去组装数据,并批量入库的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/1073756

相关文章

Python获取中国节假日数据记录入JSON文件

《Python获取中国节假日数据记录入JSON文件》项目系统内置的日历应用为了提升用户体验,特别设置了在调休日期显示“休”的UI图标功能,那么问题是这些调休数据从哪里来呢?我尝试一种更为智能的方法:P... 目录节假日数据获取存入jsON文件节假日数据读取封装完整代码项目系统内置的日历应用为了提升用户体验,

Java利用JSONPath操作JSON数据的技术指南

《Java利用JSONPath操作JSON数据的技术指南》JSONPath是一种强大的工具,用于查询和操作JSON数据,类似于SQL的语法,它为处理复杂的JSON数据结构提供了简单且高效... 目录1、简述2、什么是 jsONPath?3、Java 示例3.1 基本查询3.2 过滤查询3.3 递归搜索3.4

MySQL大表数据的分区与分库分表的实现

《MySQL大表数据的分区与分库分表的实现》数据库的分区和分库分表是两种常用的技术方案,本文主要介绍了MySQL大表数据的分区与分库分表的实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有... 目录1. mysql大表数据的分区1.1 什么是分区?1.2 分区的类型1.3 分区的优点1.4 分

Mysql删除几亿条数据表中的部分数据的方法实现

《Mysql删除几亿条数据表中的部分数据的方法实现》在MySQL中删除一个大表中的数据时,需要特别注意操作的性能和对系统的影响,本文主要介绍了Mysql删除几亿条数据表中的部分数据的方法实现,具有一定... 目录1、需求2、方案1. 使用 DELETE 语句分批删除2. 使用 INPLACE ALTER T

Python Dash框架在数据可视化仪表板中的应用与实践记录

《PythonDash框架在数据可视化仪表板中的应用与实践记录》Python的PlotlyDash库提供了一种简便且强大的方式来构建和展示互动式数据仪表板,本篇文章将深入探讨如何使用Dash设计一... 目录python Dash框架在数据可视化仪表板中的应用与实践1. 什么是Plotly Dash?1.1

Redis 中的热点键和数据倾斜示例详解

《Redis中的热点键和数据倾斜示例详解》热点键是指在Redis中被频繁访问的特定键,这些键由于其高访问频率,可能导致Redis服务器的性能问题,尤其是在高并发场景下,本文给大家介绍Redis中的热... 目录Redis 中的热点键和数据倾斜热点键(Hot Key)定义特点应对策略示例数据倾斜(Data S

Python实现将MySQL中所有表的数据都导出为CSV文件并压缩

《Python实现将MySQL中所有表的数据都导出为CSV文件并压缩》这篇文章主要为大家详细介绍了如何使用Python将MySQL数据库中所有表的数据都导出为CSV文件到一个目录,并压缩为zip文件到... python将mysql数据库中所有表的数据都导出为CSV文件到一个目录,并压缩为zip文件到另一个

SpringBoot整合jasypt实现重要数据加密

《SpringBoot整合jasypt实现重要数据加密》Jasypt是一个专注于简化Java加密操作的开源工具,:本文主要介绍详细介绍了如何使用jasypt实现重要数据加密,感兴趣的小伙伴可... 目录jasypt简介 jasypt的优点SpringBoot使用jasypt创建mapper接口配置文件加密

使用Python高效获取网络数据的操作指南

《使用Python高效获取网络数据的操作指南》网络爬虫是一种自动化程序,用于访问和提取网站上的数据,Python是进行网络爬虫开发的理想语言,拥有丰富的库和工具,使得编写和维护爬虫变得简单高效,本文将... 目录网络爬虫的基本概念常用库介绍安装库Requests和BeautifulSoup爬虫开发发送请求解

Oracle存储过程里操作BLOB的字节数据的办法

《Oracle存储过程里操作BLOB的字节数据的办法》该篇文章介绍了如何在Oracle存储过程中操作BLOB的字节数据,作者研究了如何获取BLOB的字节长度、如何使用DBMS_LOB包进行BLOB操作... 目录一、缘由二、办法2.1 基本操作2.2 DBMS_LOB包2.3 字节级操作与RAW数据类型2.