通过百度api处理交通数据

1、读取excel获取道路数据

• //道路名称
  
  @Data
  @EqualsAndHashCode
  public class RoadName {
      @ExcelProperty("Name")
      private String name;
  }
  

• /**
           * 获取excel中的道路名称
           */
          private static List<String> getRoadName() {
          // 定义文件路径，指向包含道路名称的Excel文件
          String fileName = "src\\main\\resources\\static\\cd_road_name.xlsx";
  
          // 创建一个RoadName对象的列表，用于存储从Excel文件中读取的数据
          List<RoadName> nameDataList = new ArrayList<>();
  
          // 使用EasyExcel库读取Excel文件
          EasyExcel.read(fileName, RoadName.class, new AnalysisEventListener<RoadName>() {
              // 重写invoke方法，该方法会在读取每一行数据时被调用
              @Override
              public void invoke(RoadName data, AnalysisContext context) {
                  // 将读取到的每一行数据添加到nameDataList列表中
                  nameDataList.add(data);
              }
  
              // 重写doAfterAllAnalysed方法，该方法会在所有数据读取完成后被调用
              @Override
              public void doAfterAllAnalysed(AnalysisContext context) {
                  // 所有数据解析完成后的操作，这里只是打印一条消息
                  System.out.println("所有数据解析完成！");
              }
          }).sheet().doRead(); // 指定读取第一个sheet并开始读取
  
          // 创建一个字符串列表，用于存储道路名称
          List<String> RoadNameList = new ArrayList<>();
  
          // 遍历nameDataList列表，提取每条记录的道路名称
          for (RoadName roadName : nameDataList) {
              // 将每条记录的道路名称添加到RoadNameList列表中
              RoadNameList.add(roadName.getName());
          }
  
          // 返回包含所有道路名称的列表
          return RoadNameList;
          }
  

2、获取百度api数据

• 代码官网有：https://lbsyun.baidu.com/faq/api?title=webapi/traffic-roadseek

• package xyz.zzj.traffic_zzj.utils;
  
  import org.springframework.web.util.UriUtils;
  import xyz.zzj.traffic_zzj.constant.BaiduConstant;
  
  import java.net.URL;
  import java.io.BufferedReader;
  import java.io.InputStreamReader;
  import java.net.HttpURLConnection;
  import java.net.URLConnection;
  
  import java.util.Map;
  
  //用于获取百度api
  
  public class BaiduApi {
      public static String URL = BaiduConstant.BAIDU_URL;
  
      public static String AK = BaiduConstant.BAIDU_AK;
      public String requestGetAK(String strUrl, Map<String, String> param) throws Exception {
          if (strUrl == null || strUrl.length() <= 0 || param == null || param.size() <= 0) {
              return null;
          }
  
          StringBuffer queryString = new StringBuffer();
          queryString.append(strUrl);
          for (Map.Entry<?, ?> pair : param.entrySet()) {
              queryString.append(pair.getKey() + "=");
              //    第一种方式使用的 jdk 自带的转码方式  第二种方式使用的 spring 的转码方法 两种均可
              //    queryString.append(URLEncoder.encode((String) pair.getValue(), "UTF-8").replace("+", "%20") + "&");
              queryString.append(UriUtils.encode((String) pair.getValue(), "UTF-8") + "&");
          }
  
          if (queryString.length() > 0) {
              queryString.deleteCharAt(queryString.length() - 1);
          }
  
          java.net.URL url = new URL(queryString.toString());
          System.out.println(queryString.toString());
          URLConnection httpConnection = (HttpURLConnection) url.openConnection();
          httpConnection.connect();
  
          InputStreamReader isr = new InputStreamReader(httpConnection.getInputStream());
          BufferedReader reader = new BufferedReader(isr);
          StringBuffer buffer = new StringBuffer();
          String line;
          while ((line = reader.readLine()) != null) {
              buffer.append(line);
          }
          reader.close();
          isr.close();
          //数据类型为json字符串
          return buffer.toString();
      }
  }
  

3、使用获取的道路信息，请求百度api获取每条道路的拥堵信息，并设置定时任务，没半小时请求一次

• //启动类和定时任务上都得加上
  @EnableScheduling
  //提供给spring管理
  @Component
  public class Baidu {
  

定时任务主方法代码

• 将json映射Java实体类：实体类如下

  • //接收返回的百度api数据的总实体类
    @Data
    public class BaiduReturn {
        private int status;
        private String message;
        private String description;
        private Evaluation evaluation;
        private List<RoadTraffic> road_traffic;
    
    }
    

  • //拥堵数据
    @Data
    public class CongestionSection{
        private String congestion_distance;
        private String speed;
        private int status;
        private String congestion_trend;
        private String section_desc;
    }
    

  • //拥堵状态
    @Data
    public class Evaluation {
        private int status;
        private String status_desc;
    
    }
    

  • //路况数据
    @Data
    public class RoadTraffic {
        private List<CongestionSection> congestion_sections;
        private String road_name;
    }
    

• 	@Scheduled(cron = "0 0/30 * * * ?")
          public void scheduledTask() throws Exception {            
  	    List<String> nameList;
              List<BaiduReturn> resultList = new ArrayList<>();
              nameList = getRoadName();
              for(int i = 0;i<nameList.size();i++){
                  BaiduApi snCal = new BaiduApi();
                  // 创建一个Map对象，用于存储请求参数
                  Map params = new LinkedHashMap<String, String>();
                  params.put("road_name", nameList.get(i));
                  params.put("city", "成都市");
                  params.put("ak", AK);
                  String result = snCal.requestGetAK(URL, params);
                  ObjectMapper objectMapper = new ObjectMapper();
                  // 使用 readValue 方法解析 JSON 字符串为 BaiduReturn 对象
                  BaiduReturn baiduReturn = objectMapper.readValue(result, BaiduReturn.class);
                  if (baiduReturn.getStatus() == 0) {
                      resultList.add(baiduReturn);
                  }
              }
  

4、将获取的所有道路拥堵存入excel

• 主方法代码

• 写入excel表用的实体类如下

  • //用于写入excel的数据字段
    @Data
    public class ExcelData {
            //城市名称
            private String cityName;
            //道路名称
            private String roadName;
            //道路详情
            private String description;
            //道路状态
            private String evalDesc;
            //道路拥堵距离
            private String congestionDistance;
            //道路拥堵车速
            private String speed;
            //道路拥堵地段
            private String congestionTrend;
            //道路拥堵的时间
            private String nowTime;
    }
    

•  //将数据写入excel
              for (BaiduReturn baiduReturn : resultList) {
                  RoadTraffic roadTraffic = new RoadTraffic();
                  roadTraffic.setRoad_name(baiduReturn.getRoad_traffic().get(0).getRoad_name());
                  roadTraffic.setCongestion_sections(baiduReturn.getRoad_traffic().get(0).getCongestion_sections());
                  //转化为excel格式
                  ExcelData excelData = new ExcelData();
                  excelData.setCityName("成都市");
                  excelData.setDescription(baiduReturn.getDescription());
                  excelData.setEvalDesc(baiduReturn.getEvaluation().getStatus_desc());
                  excelData.setRoadName(roadTraffic.getRoad_name());
                  // 获取当前时间戳
                  long timestamp = System.currentTimeMillis();
                  // 将时间戳转换为Instant对象
                  Instant instant = Instant.ofEpochMilli(timestamp);
                  // 转换为本地时间，指定时区
                  LocalDateTime localDateTime = LocalDateTime.ofInstant(instant, ZoneId.systemDefault());
                  // 定义日期时间格式
                  DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy年MM月dd日 HH:mm:ss");
                  // 格式化日期时间
                  String formattedDateTime = localDateTime.format(formatter);
                  excelData.setNowTime(formattedDateTime);
                  //不拥堵时，这几个字段是空，要进行判空处理
                  if (roadTraffic.getCongestion_sections() !=null) {
                      excelData.setCongestionDistance(roadTraffic.getCongestion_sections().get(0).getCongestion_distance());
                      excelData.setSpeed(roadTraffic.getCongestion_sections().get(0).getSpeed());
                      excelData.setCongestionTrend(roadTraffic.getCongestion_sections().get(0).getSection_desc());
                  }else {
                      excelData.setCongestionDistance("");
                      excelData.setSpeed("");
                      excelData.setCongestionTrend("");
                  }
      //            // 写入Excel文件
                  String fileName = "src\\main\\resources\\static\\cd_road_traffic.xlsx";
                  writeBaiduReturnToExcel(excelData, fileName);
  
          }
  }
  

• 写入excel的代码

  • /**
             * 将数据写入excel
             */
            public static void writeBaiduReturnToExcel(ExcelData excelData, String fileName) throws IOException, IllegalAccessException {
                FileInputStream fis = new FileInputStream(new File(fileName));
                Workbook workbook = new XSSFWorkbook(fis);
    
                Sheet sheet = workbook.getSheetAt(0); // 获取第一个工作表
                int rowCount = sheet.getLastRowNum(); // 行数从0开始计数，所以需要加1
                int rowIndexToAppendAfter = rowCount; // 指定要在第几行之后追加数据（从0开始计数）
    
                fis = new FileInputStream(new File(fileName));
                workbook = new XSSFWorkbook(fis);
    
                sheet = workbook.getSheetAt(0); // 获取第一个工作表
    
                // 创建新行
                Row newRow = sheet.createRow(rowIndexToAppendAfter + 1);
                List<String> list = new ArrayList<>();
                // 通过反射获取ExcelData类的所有字段
                Class<?> clazz = excelData.getClass();
                // 遍历所有字段
                for (Field field : clazz.getDeclaredFields()) {
                    field.setAccessible(true);
                    Object value = field.get(excelData);
                    list.add(value != null ? value.toString() : null);
                }
                // 将list转换为字符串数组方便填充
                String[] array = list.toArray(new String[0]);
                // 在新行中填充数据
                for (int i = 0; i < array.length; i++) {
                    Cell cell = newRow.createCell(i);
                    cell.setCellValue(array[i]);
                }
                // 将更改写回文件
                try (FileOutputStream fos = new FileOutputStream(fileName)) {
                    workbook.write(fos);
                }
    
            }