通过百度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方法，该方法会在读取每一行数据时被调用@Overridepublic void invoke(RoadName data, AnalysisContext context) {// 将读取到的每一行数据添加到nameDataList列表中nameDataList.add(data);}// 重写doAfterAllAnalysed方法，该方法会在所有数据读取完成后被调用@Overridepublic 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;//用于获取百度apipublic 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; }
-
-
//将数据写入excelfor (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开始计数，所以需要加1int 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);}}
-