ehlxr/budd
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

update at 2021-09-12 21:03:19 by ehlxr

Browse Source
This commit is contained in:
ehlxr
2021-09-12 21:03:19 +08:00
parent 42da65cf47
commit dba3de0be2
1 changed files with 260 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

260
budd-demo/src/main/java/io/github/ehlxr/ManagedBlockerTest.java Normal file
View File

@@ -0,0 +1,260 @@
/*
* The MIT License (MIT)
*
* Copyright © 2021 xrv <xrg@live.com>
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
package io.github.ehlxr;
import com.google.common.base.Stopwatch;
import java.time.LocalTime;
import java.time.format.DateTimeFormatter;
import java.util.List;
import java.util.concurrent.*;
import java.util.function.Consumer;
import java.util.function.Supplier;
import java.util.stream.Collectors;
import java.util.stream.Stream;
/**
* @author ehlxr
* @since 2021-09-02 15:52.
*/
public class ManagedBlockerTest {
static String threadDateTimeInfo() {
return DateTimeFormatter.ISO_TIME.format(LocalTime.now()) + Thread.currentThread().getName();
}
static void test1() {
List<RecursiveTask<String>> tasks = Stream.generate(() -> new RecursiveTask<String>() {
private static final long serialVersionUID = 1303648247573881735L;
@Override
protected String compute() {
System.out.println(threadDateTimeInfo() + ":simulate io task blocking for 2 seconds···");
try {
//线程休眠2秒模拟IO调用阻塞
TimeUnit.SECONDS.sleep(2);
} catch (InterruptedException e) {
throw new Error(e);
}
return threadDateTimeInfo() + ": io blocking task returns successfully";
}
}).limit(28).collect(Collectors.toList());
tasks.forEach(ForkJoinTask::fork);
tasks.forEach(e -> {
try {
System.out.println(e.get());
} catch (Exception ex) {
ex.printStackTrace();
}
});
}
public static void main(String[] args) {
Stopwatch started = Stopwatch.createStarted();
test4();
System.out.println(started);
}
static void test2() {
List<IoBlockerTask<String>> tasks = Stream.generate(() -> new IoBlockerTask<>(() -> {
System.out.println(threadDateTimeInfo() + ":simulate io task blocking for 2 seconds···");
try {
TimeUnit.SECONDS.sleep(2);
} catch (InterruptedException e) {
throw new Error(e);
}
return threadDateTimeInfo() + ": io blocking task returns successfully";
})).limit(28).collect(Collectors.toList());
tasks.forEach(ForkJoinTask::fork);
tasks.forEach(e -> {
try {
System.out.println(e.get());
} catch (Exception ex) {
ex.printStackTrace();
}
});
}
static void test3() {
List<IoBlockerAction> actions = Stream.generate(() -> new IoBlockerAction(() -> {
System.out.println(threadDateTimeInfo() + ":simulate io action blocking for 2 seconds···");
try {
TimeUnit.SECONDS.sleep(2);
} catch (InterruptedException e) {
throw new Error(e);
}
System.out.println(threadDateTimeInfo() + ": io blocking action successfully");
})).limit(56).collect(Collectors.toList());
//如果不使用 ManagedBlocker 耗时估算： 56/7（8 cpu - 1） = 8 * 2 =16s，实际 约等于 2s
actions.forEach(ForkJoinTask::fork);
actions.forEach(ForkJoinTask::join);
}
static void test4() {
List<IoBlockerAction2<Integer>> actions = Stream.generate(() -> new IoBlockerAction2<>((x) -> {
System.out.println(threadDateTimeInfo() + ":simulate io action blocking for 2 seconds···" + x);
try {
TimeUnit.SECONDS.sleep(2);
} catch (InterruptedException e) {
throw new Error(e);
}
System.out.println(threadDateTimeInfo() + ": io blocking action successfully " + x);
}, 1)).limit(56).collect(Collectors.toList());
//如果不使用 ManagedBlocker 耗时估算： 56/7（8 cpu - 1） = 8 * 2 =16s，实际 约等于 2s
actions.forEach(x -> {
x.fork();
System.out.println("fork {}"+ x);
});
actions.forEach(x -> {
x.join();
System.out.println("join {}"+ x);
});
}
}
class IoBlockerAction2<T> extends RecursiveAction {
private static final long serialVersionUID = -4114798302790287199L;
private final MyManagedBlockerImpl<T> blocker;
public IoBlockerAction2(Consumer<T> consumer, T t) {
this.blocker = new MyManagedBlockerImpl<>(consumer, t);
}
static class MyManagedBlockerImpl<T> implements ForkJoinPool.ManagedBlocker {
private final Consumer<T> consumer;
private final T t;
public MyManagedBlockerImpl(Consumer<T> consumer, T t) {
this.consumer = consumer;
this.t = t;
}
@Override
public boolean block() {
consumer.accept(t);
return true;
}
@Override
public boolean isReleasable() {
return false;
}
}
@Override
protected void compute() {
try {
ForkJoinPool.managedBlock(blocker);
} catch (InterruptedException e) {
throw new RuntimeException(e);
}
}
}
class IoBlockerAction extends RecursiveAction {
private static final long serialVersionUID = -4114798302790287199L;
private final MyManagedBlockerImpl blocker;
public IoBlockerAction(Runnable runnable) {
this.blocker = new MyManagedBlockerImpl(runnable);
}
static class MyManagedBlockerImpl implements ForkJoinPool.ManagedBlocker {
private final Runnable runnable;
public MyManagedBlockerImpl(Runnable runnable) {
this.runnable = runnable;
}
@Override
public boolean block() {
runnable.run();
return true;
}
@Override
public boolean isReleasable() {
return false;
}
}
@Override
protected void compute() {
try {
ForkJoinPool.managedBlock(blocker);
// setRawResult(blocker.result);
// return getRawResult();
} catch (InterruptedException e) {
throw new RuntimeException(e);
}
}
}
class IoBlockerTask<T> extends RecursiveTask<T> {
private static final long serialVersionUID = -4114798302790287199L;
private final MyManagedBlockerImpl<T> blocker;
public IoBlockerTask(Supplier<T> supplier) {
this.blocker = new MyManagedBlockerImpl<>(supplier);
}
static class MyManagedBlockerImpl<T> implements ForkJoinPool.ManagedBlocker {
private final Supplier<T> supplier;
private T result;
public MyManagedBlockerImpl(Supplier<T> supplier) {
this.supplier = supplier;
}
@Override
public boolean block() {
result = supplier.get();
return true;
}
@Override
public boolean isReleasable() {
return false;
}
}
@Override
protected T compute() {
try {
ForkJoinPool.managedBlock(blocker);
setRawResult(blocker.result);
return getRawResult();
} catch (InterruptedException e) {
throw new RuntimeException(e);
}
}
}