blob: f597669815abb6b3a20476d71a2cfb08d1be9334 [file] [log] [blame]
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.iotdb.commons.concurrent.threadpool;
import org.apache.iotdb.commons.concurrent.WrappedCallable;
import org.apache.iotdb.commons.concurrent.WrappedRunnable;
import org.apache.iotdb.commons.conf.IoTDBConstant;
import org.apache.iotdb.commons.service.JMXService;
import java.util.Collection;
import java.util.List;
import java.util.concurrent.Callable;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;
import java.util.stream.Collectors;
public class WrappedSingleThreadExecutorService
implements ExecutorService, WrappedSingleThreadExecutorServiceMBean {
private final String mbeanName;
ExecutorService service;
public WrappedSingleThreadExecutorService(ExecutorService service, String mbeanName) {
this.service = service;
this.mbeanName =
String.format(
"%s:%s=%s", IoTDBConstant.IOTDB_THREADPOOL_PACKAGE, IoTDBConstant.JMX_TYPE, mbeanName);
JMXService.registerMBean(this, this.mbeanName);
}
@Override
public void shutdown() {
service.shutdown();
JMXService.deregisterMBean(mbeanName);
}
@Override
public List<Runnable> shutdownNow() {
JMXService.deregisterMBean(mbeanName);
return service.shutdownNow();
}
@Override
public boolean isShutdown() {
return service.isShutdown();
}
@Override
public boolean isTerminated() {
return service.isTerminated();
}
@Override
public boolean awaitTermination(long timeout, TimeUnit unit) throws InterruptedException {
return service.awaitTermination(timeout, unit);
}
@Override
public <T> Future<T> submit(Callable<T> task) {
return service.submit(WrappedCallable.wrap(task));
}
@Override
public <T> Future<T> submit(Runnable task, T result) {
return service.submit(WrappedRunnable.wrap(task), result);
}
@Override
public Future<?> submit(Runnable task) {
return service.submit(WrappedRunnable.wrap(task));
}
@Override
public <T> List<Future<T>> invokeAll(Collection<? extends Callable<T>> tasks)
throws InterruptedException {
return service.invokeAll(
tasks.stream().map(WrappedCallable::wrap).collect(Collectors.toList()));
}
@Override
public <T> List<Future<T>> invokeAll(
Collection<? extends Callable<T>> tasks, long timeout, TimeUnit unit)
throws InterruptedException {
return service.invokeAll(
tasks.stream().map(WrappedCallable::wrap).collect(Collectors.toList()), timeout, unit);
}
@Override
public <T> T invokeAny(Collection<? extends Callable<T>> tasks)
throws InterruptedException, ExecutionException {
return service.invokeAny(
tasks.stream().map(WrappedCallable::wrap).collect(Collectors.toList()));
}
@Override
public <T> T invokeAny(Collection<? extends Callable<T>> tasks, long timeout, TimeUnit unit)
throws InterruptedException, ExecutionException, TimeoutException {
return service.invokeAny(
tasks.stream().map(WrappedCallable::wrap).collect(Collectors.toList()), timeout, unit);
}
@Override
public void execute(Runnable command) {
service.execute(WrappedRunnable.wrap(command));
}
}