首页 > 代码库 > 基于Java的数据采集(三)

基于Java的数据采集(三)

《基于Java的数据采集(一)》:http://www.cnblogs.com/lichenwei/p/3904715.html

《基于Java的数据采集(二)》:http://www.cnblogs.com/lichenwei/p/3905370.html

基于之前2篇Java数据采集入库,做了下功能整合,实现本地的存读取,上个效果图:

直接上代码吧,本程序只是作为"如何用JAVA抓取页面简单采集入库"的入门,在实际做采集工具的时候,还需考虑许多东西,比如当采集一个页面发生卡顿时,发生延迟时怎么办?等一系列的问题,希望这篇文字能够抛砖引玉。

先看下项目结构:

一共有五个类:

Mysql.java  --数据库操作类

RegEX.java   --正则匹配类

GetAllData.java --采集类

Action.java  --功能实现类

FootBallMain.java --主程序类

其他的,直接结合前面2篇文章外加看代码注释吧

Mysql.java

 1 package com.lcw.curl; 2  3  4 import java.sql.Connection; 5 import java.sql.DriverManager; 6 import java.sql.ResultSet; 7 import java.sql.SQLException; 8 import java.sql.Statement; 9 10 11 /**12  * 数据库操作类,一更新,一查询13  * @author Balla_兔子14  *15  */16 public class MySql {17   18     //定义MySql驱动,数据库地址,数据库用户名 密码, 执行语句和数据库连接  19     public String driver = "com.mysql.jdbc.Driver";20     public String url = "jdbc:mysql://127.0.0.1:3306/football";21     public String user = "root";22     public String password = "";23     public Statement stmt = null;24     public Connection conn = null;25     26     //创建一个插入数据的方法27     public void datatoMySql(String insertSQl) {28 29         try {30             try {31                 Class.forName(driver).newInstance();32             } catch (Exception e) {33                 System.out.println("Unable to find the local driver");34                 e.printStackTrace();35             }36             //创建连接37             conn = DriverManager.getConnection(url, user, password);38             //创建一个 Statement 对象来将 SQL 语句发送到数据库39             stmt = conn.createStatement();40         } catch (SQLException e) {41             e.printStackTrace();42         }43         try {44             //执行SQL 插入语句45             stmt.executeUpdate(insertSQl);46         } catch (SQLException e) {47             e.printStackTrace();48         }49         try {50             stmt.close();51             conn.close();52         } catch (SQLException e) {53             e.printStackTrace();54         }55     }56     57     58   //创建一个查找数据的方法59     public ResultSet searchMySql(String selectSQl) {60         61         ResultSet rs=null;62 63         try {64             try {65                 Class.forName(driver).newInstance();66             } catch (Exception e) {67                 System.out.println("Unable to find the local driver");68                 e.printStackTrace();69             }70             //创建连接71             conn = DriverManager.getConnection(url, user, password);72             //创建一个 Statement 对象来将 SQL 语句发送到数据库73             stmt = conn.createStatement();74         } catch (SQLException e) {75             e.printStackTrace();76         }77         try {78             //执行SQL 插入语句79             rs=stmt.executeQuery(selectSQl);80         } catch (SQLException e) {81             e.printStackTrace();82         }83         84         return rs;85     }86     87 }
Mysql.java

RegEX.java

 1 package com.lcw.curl; 2  3 import java.util.regex.Matcher; 4 import java.util.regex.Pattern; 5  6 public class RegEX { 7  8     /** 9      * 10      * @param regex11      * 正则表达式12      * @param content13      * 所要匹配的内容14      * @return15      */16     public String getData(String regex, String content) {17         Pattern pattern = Pattern.compile(regex, Pattern.CASE_INSENSITIVE);// 设定正则表达式,不区分大小写18         Matcher matcher = pattern.matcher(content);19         if (matcher.find()) {20             return matcher.group();//返回正则匹配结果21         } else {22             return "";23         }24     }25 26 }
RegEX.java

GetAllData.java

 1 package com.lcw.curl; 2  3 import java.io.BufferedReader; 4 import java.io.InputStreamReader; 5 import java.net.URL; 6  7 public class GetAllData { 8  9     /**采集类10      * @param Balla_兔子11      */12     public void getAllData() {13 14         try {15             String address = "http://www.footballresults.org/league.php?league=EngDiv1";16             URL url = new URL(address);17             InputStreamReader inputStreamReader = new InputStreamReader(url18                     .openStream(), "utf-8");// 打开地址,以UTF-8编码的形式返回字节并转为字符19             BufferedReader bufferedReader = new BufferedReader(20                     inputStreamReader);// 从字符输入流中读取文本,缓冲各个字符,从而提供字符、数组和行的高效读取。21 22             RegEX data = http://www.mamicode.com/new RegEX();23             MySql mySql = new MySql();24             String content = "";// 用来接受每次读取的行字符25             int flag = 0;// 标志,队伍信息刚好在日期信息后面,则正则相同,用于分离数据26             String dateRegex = "\\d{1,2}\\.\\d{1,2}\\.\\d{4}";// 日期匹配正则表达式27             String teamRegex = ">[^<>]*</a>";// 队伍匹配正则表达式28             String scoreRegex = ">(\\d{1,2}-\\d{1,2})</TD>";// 比分正则表达式29             String tempDate = "";// 存储临时比赛时间30             String teama = "";// 存储临时主队31             String teamb = "";// 存储临时客队32             String score = "";// 存储临时比分33             int i = 0;// 记录信息条数34             String sql = "";// 数据库语句35 36             while ((content = bufferedReader.readLine()) != null) {// 每次读取一行数据37                 // 获取比赛日期信息38                 String dateInfo = data.getData(dateRegex, content);39                 if (!dateInfo.equals("")) {40                     // System.out.println("日期:" + dateInfo);41                     tempDate = dateInfo;42                     flag++;43                 }44                 // 获取队伍信息,需先读到日期信息让标志符自增45                 String teamInfo = data.getData(teamRegex, content);46                 if (!teamInfo.equals("") && flag == 1) {47                     teama = teamInfo.substring(1, teamInfo.indexOf("</a>"));48                     // System.out.println("主队:" + teama);49                     flag++;50                 } else if (!teamInfo.equals("") && flag == 2) {51                     teamb = teamInfo.substring(1, teamInfo.indexOf("</a>"));52                     // System.out.println("客队:" + teamb);53                     flag = 0;54                 }55                 // 获取比分信息56                 String scoreInfo = data.getData(scoreRegex, content);57                 if (!scoreInfo.equals("")) {58                     score = scoreInfo.substring(1, scoreInfo.indexOf("</TD>"));59                     // System.out.println("比分:" + score);60                     // System.out.println();61                     i++;62                     sql = "insert into football(`date`,`teama`,`teamb`,`score`) values(‘"63                             + tempDate64                             + "‘,‘"65                             + teama66                             + "‘,‘"67                             + teamb68                             + "‘,‘"69                             + score + "‘)";70                     mySql.datatoMySql(sql);71                     System.out.println("存储数据成功:" + i + "条");72                 }73 74             }75             bufferedReader.close();76             // System.out.println("一共收集到了" + i + "条信息");77         } catch (Exception e) {78             e.printStackTrace();79         }80 81     }82 83 }
GetAllData.java

Action.java

  1 package com.lcw.curl;  2   3 import java.sql.ResultSet;  4 import java.sql.SQLException;  5 import java.util.ArrayList;  6 import java.util.List;  7 import java.util.Vector;  8   9 public class Action { 10  11     /** 12      * 操作一:初始化数据库数据 13      */ 14     public void initData() { 15         String sql = "delete from football"; 16         MySql doMySql = new MySql(); 17         try { 18             doMySql.datatoMySql(sql); 19             System.out.println("数据初始化完毕!"); 20         } catch (Exception e) { 21             System.out.println("数据初始化失败!"); 22         } 23  24     } 25  26     /** 27      * 获取所有队伍信息 28      *  29      * @return 30      */ 31     public Vector<String> getAllTeam() { 32         ResultSet rs = null; 33         Vector<String> vector = new Vector<String>(); 34         String sql = "select teama,teamb from football"; 35         MySql doMySql = new MySql(); 36         rs = doMySql.searchMySql(sql); 37  38         try { 39             while (rs.next()) { 40                 try { 41                     if (!vector.contains(rs.getString("teama"))) { 42                         vector.add(rs.getString("teama")); 43                     } 44                     if (!vector.contains(rs.getString("teamb"))) { 45                         vector.add(rs.getString("teamb")); 46                     } 47                 } catch (SQLException e) { 48                     e.printStackTrace(); 49                 } 50             } 51         } catch (SQLException e) { 52             e.printStackTrace(); 53         } 54  55         return vector; 56  57     } 58  59     /** 60      * 获取具体某队的比赛信息 61      *  62      * @param team 63      * @return 64      */ 65     public List<String> findTeam(String team) { 66         List<String> list = new ArrayList<String>(); 67         String sql = "select * from football where teama =‘" + team 68                 + "‘ or teamb =‘" + team + "‘"; 69         MySql mysql = new MySql(); 70         ResultSet rs = null; 71         rs = mysql.searchMySql(sql); 72         try { 73             while (rs.next()) { 74                 list.add(rs.getString("date")); 75                 list.add(rs.getString("teama")); 76                 list.add(rs.getString("teamb")); 77                 list.add(rs.getString("score")); 78             } 79         } catch (SQLException e) { 80             e.printStackTrace(); 81         } 82         return list; 83  84     } 85  86     public List<String> findGame(String date) { 87         List<String> list = new ArrayList<String>(); 88         ResultSet rs = null; 89         String sql = "select * from football where date =‘" + date + "‘"; 90         MySql mysql = new MySql(); 91         rs = mysql.searchMySql(sql); 92         try { 93             while (rs.next()) { 94                 list.add(rs.getString("date")); 95                 list.add(rs.getString("teama")); 96                 list.add(rs.getString("teamb")); 97                 list.add(rs.getString("score")); 98             } 99         } catch (SQLException e) {100             // TODO Auto-generated catch block101             e.printStackTrace();102         }103         return list;104     }105 106 }
Action.java

FootBallMain.java

 1 package com.lcw.curl; 2  3 import java.util.List; 4 import java.util.Scanner; 5 import java.util.Vector; 6  7 public class FootBallMain { 8  9     /**主程序类10      * @param Balla_兔子11      */12     public static void main(String[] args) {13         GetAllData allData = http://www.mamicode.com/new GetAllData();14         Action action = new Action();15 16         while (true) {17             System.out.println("①初始化数据库-请按 (1)");18             System.out.println("②自动化采集数据-请按(2)");19             System.out.println("③查询参赛队伍-请按(3)");20             System.out.println("④查询具体球队比赛结果-请按(4)");21             System.out.println("⑤查询具体某天的比赛详情-请按(5)");22             Scanner scanner = new Scanner(System.in);23             String input = scanner.next();24             if (input.equals("1")) {25                 System.out.println();26                 action.initData();27                 System.out28                         .println("-----------------------------------------------------");29             } else if (input.equals("2")) {30                 System.out.println("正在采集数据...请稍后");31                 allData.getAllData();32                 System.out33                         .println("-----------------------------------------------------");34             } else if (input.equals("3")) {35                 Vector<String> allTeam = action.getAllTeam();36                 System.out.println("正在获取数据...请稍后");37                 if (allTeam.size() != 0) {38                     System.out.println("参赛队伍如下:");39                     for (int i = 0; i < allTeam.size(); i++) {40                         System.out.println(allTeam.get(i));41                     }42                 }43                 System.out44                         .println("-----------------------------------------------------");45             } else if (input.equals("4")) {46                 System.out.println("请输入您要查询的队伍名:");47                 String team = scanner.next();48                 List<String> list = action.findTeam(team);49                 System.out.println("比赛日期\t\t\t主队\t\t客队\t\t\t比赛结果");50                 if (list.size() != 0) {51                     for (int i = 0; i < list.size(); i++) {52                         System.out.print(list.get(i) + "\t\t");53                     }54                 } else {55                     System.out.println("暂时没有您所提供队伍的比赛信息,敬请关注...");56                 }57                 System.out.println();58                 System.out59                         .println("-----------------------------------------------------");60             } else if (input.equals("5")) {61                 System.out.println("请输入您要查询日期(格式如下:xx.xx.xxxx):");62                 String date = scanner.next();63                 List<String> info = action.findGame(date);64                 System.out.println("比赛日期\t\t\t主队\t\t客队\t\t\t比赛结果");65                 if (info.size() != 0) {66                     for (int i = 0; i < info.size(); i++) {67                         if (i % 4 == 0 && i != 0) {68                             System.out.println();69                         }70                         System.out.print(info.get(i) + "\t\t");71                     }72                 } else {73                     System.out.println("暂时没有您所提供的比赛信息,敬请关注...");74                 }75                 System.out.println();76                 System.out77                         .println("------------------------------------------------------------------------");78             } else {79                 System.out.println("请输入正确的对应编号..");80                 System.out81                         .println("------------------------------------------------------------------------");82             }83         }84     }85 86 }
FootBallMain.java