private static void startNioServer(String... args) throws IOException, InterruptedException {
final ServerConfiger configer = new ServerConfiger();
configer.setDataFile(new File(args[1]));
configer.setPort(Integer.valueOf(args[2]));
final LaserOptions options = new LaserOptions(new File(args[3]));
final DataSource dataSource =
options.isServerDebug() ? new MockDataSource() : new PageDataSource(configer.getDataFile());
dataSource.init();
final CountDownLatch countDown = new CountDownLatch(1);
final ExecutorService executorService = Executors.newCachedThreadPool();
final NioLaserServer server =
new NioLaserServer(dataSource, executorService, configer, options);
server.startup();
// registe shutdown
getRuntime()
.addShutdownHook(
new Thread(
() -> {
currentThread().setName("server-shutdown-hook");
try {
dataSource.destroy();
server.shutdown();
executorService.shutdown();
} catch (IOException e) {
// do nothing...
}
}));
countDown.await();
}
private static void startNioClient(String... args) throws IOException, InterruptedException {
final long startTime = System.currentTimeMillis();
final ClientConfiger configer = new ClientConfiger();
configer.setServerAddress(new InetSocketAddress(args[1], Integer.valueOf(args[2])));
configer.setDataFile(new File(args[3]));
final LaserOptions options = new LaserOptions(new File(args[4]));
final int worksNum = options.getClientWorkNumbers();
// read+writer+persistence
final CyclicBarrier workCyclicBarrier = new CyclicBarrier(worksNum * 2 + 1);
final ExecutorService executorService =
Executors.newCachedThreadPool(
(r) -> {
final Thread t = new Thread(r);
t.setDaemon(true);
return t;
});
final DataPersistence dataPersistence = new PageDataPersistence(configer.getDataFile());
// 异步初始化数据源
executorService.execute(
() -> {
try {
dataPersistence.init();
try {
workCyclicBarrier.await();
} catch (Exception e) {
// ingore...
}
} catch (IOException e) {
log.warn("DataPersistence.init failed.");
}
});
final CountDownLatch countDown = new CountDownLatch(worksNum);
// 异步创建建立链接
final Set<NioLaserClient> clients = new HashSet<>();
for (int i = 0; i < worksNum; i++) {
executorService.execute(
() -> {
final NioLaserClient client =
new NioLaserClient(
countDown,
workCyclicBarrier,
executorService,
dataPersistence,
configer,
options);
try {
client.connect();
clients.add(client);
client.work();
} catch (IOException e) {
log.warn("client connect failed.", e);
}
});
}
// 等待所有Client完成
countDown.await();
final long endTime = System.currentTimeMillis();
System.out.println("cost=" + (endTime - startTime));
// 刷新结果
dataPersistence.flush();
dataPersistence.destroy();
// registe shutdown
getRuntime()
.addShutdownHook(
new Thread(
() -> {
try {
currentThread().setName("client-shutdown-hook");
for (NioLaserClient client : clients) {
client.disconnect();
}
executorService.shutdown();
} catch (IOException e) {
// do nothing...
}
}));
}
