38 lines
1.5 KiB
TypeScript
38 lines
1.5 KiB
TypeScript
import { task, plugin } from '@ms/api'
|
|
import { inject, provideSingleton } from '@ms/container';
|
|
|
|
const Sponge = Java.type("org.spongepowered.api.Sponge");
|
|
const Task = Java.type("org.spongepowered.api.scheduler.Task");
|
|
const Consumer = Java.type('java.util.function.Consumer');
|
|
const Callable = Java.type('java.util.concurrent.Callable');
|
|
const TimeUnit = Java.type('java.util.concurrent.TimeUnit');
|
|
|
|
@provideSingleton(task.TaskManager)
|
|
export class SpongeTaskManager implements task.TaskManager {
|
|
@inject(plugin.PluginInstance)
|
|
private pluginInstance: any;
|
|
|
|
create(func: Function): task.Task {
|
|
if (Object.prototype.toString.call(func) !== "[object Function]") { throw TypeError('第一个参数 Task 必须为 function !'); };
|
|
return new SpongeTask(this.pluginInstance, func);
|
|
}
|
|
callSyncMethod(func: Function): any {
|
|
return Sponge.getScheduler()
|
|
.createSyncExecutor(this.pluginInstance)
|
|
// @ts-ignore
|
|
.schedule(new Callable({ call: () => func() }), java.lang.Long.valueOf(0), TimeUnit.NANOSECONDS).get()
|
|
}
|
|
}
|
|
|
|
export class SpongeTask extends task.Task {
|
|
submit(): task.Cancelable {
|
|
let run = Task.builder().execute(new Consumer({
|
|
accept: () => this.run()
|
|
}));
|
|
if (this.isAsync) { run.async() }
|
|
if (this.laterTime) { run.delayTicks(this.laterTime) }
|
|
if (this.interval) { run.intervalTicks(this.interval) }
|
|
return run.submit(this.plugin)
|
|
}
|
|
}
|