import java.io.IOException;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.Arrays;
import java.util.Set;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.stream.Collectors;
/**
* 修改hosts文件
* @author yk.luo
*/
public abstract class HostUtils {
private static final Path HOSTS_PATH = Paths.get("C:\\Windows\\System32\\drivers\\etc\\hosts");
private static final String ENDL = String.format("%n");
public static Set<String> read() {
return Arrays.stream(LocalFileIOUtils.getInstance().read(HOSTS_PATH).split(ENDL))
.filter(it->!it.trim().matches("(^#.*)|(\\s*)"))
.map(it->it.replaceAll("#.*", "").trim().replaceAll("\\s+", "\t"))
.collect(Collectors.toSet());
}
public static boolean append(String ip, String url) {
Set<String> sets = HostUtils.read();
sets.add(String.format("%s\t%s", ip, url));
return LocalFileIOUtils.getInstance().write(HOSTS_PATH, sets.stream().collect(Collectors.joining(ENDL))) && flushdns();
}
public static boolean exists(String ip, String url) {
return HostUtils.read().contains(String.format("%s\t%s", ip, url));
}
public static boolean flushdns() {
try {
Runtime.getRuntime().exec("ipconfig /flushdns");
return true;
} catch (IOException e) {
e.printStackTrace();
return false;
}
}
public static void main(String[] args) {
int nThreads = 16;
ExecutorService executorService = Executors.newCachedThreadPool();
while (nThreads-- > 0) {
executorService.execute(()->{
boolean rst = HostUtils.append("0.0.0.0", "www.baidu.com");
System.out.println(rst);
HostUtils.read().forEach(System.out::println);
});
}
executorService.shutdown();
}
}