Java中利用函数式接口动态创建与重初始化WebDriver实例

// 假设 driver 是之前崩溃的 WebDriver 实例
// ...

// 这是一个简化的 getFunction 示例
@SneakyThrows // Lombok 注解，用于简化异常处理
static Function<Class<? extends RemoteWebDriver>, RemoteWebDriver> getFunction(Class<? extends RemoteWebDriver> driverClass){
   return c -> {
            try {
               return c.getConstructor().newInstance();
            } catch (IllegalAccessException | InstantiationException e) {
               throw new RuntimeException(e);
            }
         };
}

// 尝试使用 Map 来获取并创建实例
// 错误用法示例：
/*
driver = Map.of(
            ChromeDriver.class, getFunction(ChromeDriver.class),
            EdgeDriver.class, getFunction(EdgeDriver.class),
            FirefoxDriver.class, getFunction(FirefoxDriver.class),
            OperaDriver.class, getFunction(OperaDriver.class)
      ).entrySet().stream().filter((e) -> e.getKey().isInstance(driver))
      .map((e)->e.getValue().identity()).findFirst().orElseThrow(() -> new RuntimeException("WebDriver not detected"));
*/

// 正确使用 Function 的方式
RemoteWebDriver newDriver = Map.of(
            ChromeDriver.class, getFunction(ChromeDriver.class),
            EdgeDriver.class, getFunction(EdgeDriver.class),
            FirefoxDriver.class, getFunction(FirefoxDriver.class),
            OperaDriver.class, getFunction(OperaDriver.class)
      ).entrySet().stream()
      .filter((e) -> e.getKey().isInstance(driver)) // 查找与崩溃 WebDriver 类型匹配的条目
      .map((e)->e.getValue().apply(driver.getClass())) // 调用 Function 的 apply 方法并传入类型
      .findFirst()
      .orElseThrow(() -> new RuntimeException("WebDriver type not detected"));

import java.util.function.Supplier;
import org.openqa.selenium.remote.RemoteWebDriver;
import lombok.SneakyThrows; // 假设使用 Lombok

@SneakyThrows
static Supplier<RemoteWebDriver> getSupplier(Class<? extends RemoteWebDriver> driverClass){
  return () -> { // Supplier 的 lambda 表达式不接受参数
        try {
           // 由于 driverClass 是 getSupplier 方法的参数，它在 lambda 表达式中是“effectively final”的，可以直接引用
           return driverClass.getConstructor().newInstance();
        } catch (IllegalAccessException | InstantiationException e) {
           throw new RuntimeException(e);
        }
     };
}

RemoteWebDriver newDriver = Map.of(
            ChromeDriver.class, getSupplier(ChromeDriver.class),
            EdgeDriver.class, getSupplier(EdgeDriver.class),
            FirefoxDriver.class, getSupplier(FirefoxDriver.class),
            OperaDriver.class, getSupplier(OperaDriver.class)
      ).entrySet().stream()
      .filter((e) -> e.getKey().isInstance(driver))
      .map((e)->e.getValue().get()) // 调用 Supplier 的 get 方法
      .findFirst()
      .orElseThrow(() -> new RuntimeException("WebDriver type not detected"));

import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.edge.EdgeDriver;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.openqa.selenium.opera.OperaDriver; // 如果需要
import java.util.Map;
import java.util.function.Supplier;
import org.openqa.selenium.remote.RemoteWebDriver;

// 假设 driver 是之前崩溃的 WebDriver 实例
// ...

// 注意：Map.of 的泛型类型声明有助于编译器推断正确的类型
Map<Class<? extends RemoteWebDriver>, Supplier<? extends RemoteWebDriver>> driverSuppliers =
      Map.of(
            ChromeDriver.class, ChromeDriver::new,
            EdgeDriver.class, EdgeDriver::new,
            FirefoxDriver.class, FirefoxDriver::new,
            // OperaDriver.class, OperaDriver::new // 如果需要，请确保 OperaDriver 可用
            // 添加其他 WebDriver 类型...
      );

RemoteWebDriver newDriver = driverSuppliers.entrySet().stream()
      .filter((e) -> e.getKey().isInstance(driver))
      .map((e)->e.getValue().get())
      .findFirst()
      .orElseThrow(() -> new RuntimeException("WebDriver type not detected"));

import java.util.function.Supplier;
import org.openqa.selenium.remote.RemoteWebDriver;

// 辅助方法，用于类型推断
static Supplier<? extends RemoteWebDriver> supply(Supplier<? extends RemoteWebDriver> s) {
    return s;
}

// 使用辅助方法
Map<Class<? extends RemoteWebDriver>, Supplier<? extends RemoteWebDriver>> driverSuppliersWithHelper =
      Map.of(
            ChromeDriver.class, supply(ChromeDriver::new),
            EdgeDriver.class, supply(EdgeDriver::new),
            FirefoxDriver.class, supply(FirefoxDriver::new)
            // ...
      );

// 后续使用方式相同
// RemoteWebDriver newDriver = driverSuppliersWithHelper.entrySet().stream()...