selenium 可以動態爬取網頁數據,就像真實用戶操作瀏覽器一樣,從終端用戶的角度測試應用程序,WebDriver通過原生瀏覽器支持或者瀏覽器擴展直接控制瀏覽器
webdriver下載
因為selenuim對瀏覽器的版本存在兼容問題,顧需要針對指定瀏覽器下載指定版本。
1、添加依賴
<dependency><groupId>org.seleniumhq.selenium</groupId><artifactId>selenium-java</artifactId><version>4.11.0</version></dependency><dependency><groupId>com.google.guava</groupId><artifactId>guava</artifactId><version>32.1.2-jre</version></dependency>
2、工具類
import cn.hutool.core.collection.CollectionUtil;
import com.google.common.collect.Lists;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.chrome.ChromeOptions;
import org.openqa.selenium.edge.EdgeDriver;
import org.openqa.selenium.edge.EdgeOptions;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.openqa.selenium.firefox.FirefoxOptions;
import org.springframework.stereotype.Component;import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;/*** Selenium 工具類** @author kou*/
@Slf4j
@RequiredArgsConstructor
@Component
public class SeleniumUtil {private final ReptileProperties reptileProperties;/*** 獲取chromeDriver** @return chromeDriver*/public WebDriver chromeDriver() {// 加載驅動路徑System.setProperty("webdriver.chrome.driver", "D:/chromedriver.exe");// Chrome默認不允許跨機器調試,需要給啟動命令加上白名單System.setProperty("webdriver.chrome.whitelistedIps", "");ChromeOptions options = new ChromeOptions();// 開啟一個實驗性參數excludeSwitches,用來隱藏window.navigator.webdriver返回true,這個參數必須是Listoptions.setExperimentalOption("useAutomationExtension", false);// 開啟開發者模式options.setExperimentalOption("excludeSwitches", Lists.newArrayList("enable-automation"));// 發現主要是這句是關鍵options.addArguments("--disable-blink-features=AutomationControlled");// options.addArguments("--incognito");// options.addArguments("--disable-infobars");//options.addArguments("user-agent=Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/86.0.4240.198 Safari/537.36");options.addArguments("user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/115.0.0.0 Safari/537.36");// 禁用沙箱options.addArguments("--no-sandbox");// 無頭瀏覽器,這樣不會打開瀏覽器窗口// options.addArguments("--headless");// options.addArguments("--disable-gpu");options.addArguments("--remote-allow-origins=*");// 初始化一個谷歌瀏覽器實例,實例名稱叫driverWebDriver driver = new ChromeDriver(options);return driver;}/*** 獲取edgeDriver** @return edgeDriver*/public WebDriver edgeDriver() {// 加載驅動路徑System.setProperty("webdriver.edge.driver", "D:/msedgedriver.exe");EdgeOptions options = new EdgeOptions();// 開啟一個實驗性參數excludeSwitches,用來隱藏window.navigator.webdriver返回true,這個參數必須是Listoptions.setExperimentalOption("useAutomationExtension", false);//開啟開發者模式options.setExperimentalOption("excludeSwitches", Lists.newArrayList("enable-automation"));// 發現主要是這句是關鍵options.addArguments("--disable-blink-features=AutomationControlled");options.addArguments("--incognito", "--disable-infobars");// options.addArguments("user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/115.0.0.0 Safari/537.36");options.addArguments("user-agent=Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/86.0.4240.198 Safari/537.36");// 禁用沙箱options.addArguments("--no-sandbox");// 無頭瀏覽器,這樣不會打開瀏覽器窗口// options.addArguments("--headless");options.addArguments("--disable-gpu");options.addArguments("--remote-allow-origins=*");// 初始化一個谷歌瀏覽器實例,實例名稱叫driverWebDriver driver = new EdgeDriver(options);return driver;}/*** 獲取firefoxDriver** @return firefoxDriver*/public WebDriver firefoxDriver() {// 加載驅動路徑System.setProperty("webdriver.gecko.driver", "D:/geckodriver.exe");System.setProperty("webdriver.chrome.whitelistedIps", "");FirefoxOptions options = new FirefoxOptions();options.addArguments("user-agent=Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/86.0.4240.198 Safari/537.36");// 無頭瀏覽器,這樣不會打開瀏覽器窗口options.addArguments("--headless");// 初始化一個谷歌瀏覽器實例,實例名稱叫driverWebDriver driver = new FirefoxDriver(options);return driver;}/*** 獲取表頭** @param table 表格* @return 表頭*/public List<String> getTableHead(WebElement table) {log.info("開始解析表頭...");// 獲取表頭WebElement head = table.findElement(By.tagName("thead"));if (null == head) {return Collections.emptyList();}List<WebElement> headths = head.findElements(By.tagName("th"));List<String> headList = new ArrayList<>(headths.size());headths.forEach(t -> {headList.add(t.getText());});log.info("表頭解析完成!!!");return headList;}/*** 獲取表數據** @param table 表格* @return 表頭*/public List<List<String>> getTableBody(WebElement table) {log.info("開始解析表數據...");// 獲取表頭WebElement tbody = table.findElement(By.tagName("tbody"));if (null == tbody) {return Collections.emptyList();}// 獲取body數據行List<WebElement> bodyTrs = tbody.findElements(By.tagName("tr"));if (CollectionUtil.isEmpty(bodyTrs)) {return Collections.emptyList();}List<List<String>> bodyDatas = new ArrayList<>(bodyTrs.size());bodyTrs.stream().forEach(r -> {List<WebElement> tds = r.findElements(By.tagName("td"));List<String> rows = new ArrayList<>(tds.size());tds.forEach(d -> {rows.add(d.getText());});bodyDatas.add(rows);});log.info("表數據解析完成!!!");return bodyDatas;}/*** 將參數轉化為路徑參數** @param params 參數* @return 路徑參數*/public String convertPathParams(Map<String, Object> params) {if (CollectionUtil.isEmpty(params)) {return "";}StringBuffer path = new StringBuffer();for (Map.Entry<String, Object> p : params.entrySet()) {path.append(p.getKey()).append("=").append(p.getValue().toString()).append("&");}return path.substring(0, path.length() - 1);}}
3、爬取數據
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.support.ui.ExpectedCondition;
import org.openqa.selenium.support.ui.WebDriverWait;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;import java.time.Duration;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;/*** 數據接口實現類** @author kou*/
@Slf4j
@RequiredArgsConstructor
@Service
public class DataServiceImpl {private final SeleniumUtil seleniumUtil;/*** 獲取頁面數據** @return 數據*/@Overridepublic Map<String, Object> getHtmlData() {try {Map<String, Object> data = new HashMap<>();String url = "url";Map<String, Object> params = new HashMap<>();params.put("pageNum", 1);params.put("pageSize", 1000);String fullUrl = url + seleniumUtil.convertPathParams(params);WebDriver driver = seleniumUtil.firefoxDriver();driver.get(fullUrl);// 打開一個站點log.info("開始訪問:{}", fullUrl);driver.get(fullUrl);String title = driver.getTitle();log.info("網頁:{}", title);// 獲取表格數據WebElement table = driver.findElement(By.id("table"));//顯式等待,針對某個元素等待,等待超時時間100s,2s檢測一次WebDriverWait wait = new WebDriverWait(driver, Duration.ofSeconds(100), Duration.ofSeconds(2));// wait.until(ExpectedConditions.presenceOfElementLocated(By.id("table")));wait.until(new ExpectedCondition<WebElement>() {@Overridepublic WebElement apply(WebDriver text) {log.info("開始檢查tbody數據是否已加載");WebElement table = text.findElement(By.id("table")).findElement(By.tagName("tbody"));if (!table.isDisplayed()) {log.info("檢查結果:tbody數據未加載完,等待加載...");return null;}log.info("檢查結果:tbody數據加載完成!!!");return table;}});// 獲取表頭List<String> headList = seleniumUtil.getTableHead(table);List<List<String>> bodyList = seleniumUtil.getTableBody(table);data.put("header", headList);data.put("body", bodyList);driver.close();return data;} catch (Exception e) {throw new RuntimeException(e);}}}