c9aaa93d83
Motivation: Sometimes it is desirable to be able to use a different Queue implementation for the EventLoop of a Channel. This is currently not possible without resort to reflection. Modifications: - Add a new constructor to Nio|Epoll|KQueueEventLoopGroup which allows to specify a factory which is used to create the task queue. This was the user can override the default implementation. - Add test Result: Be able to change Queue that is used for the EventLoop.
173 lines
5.9 KiB
Java
173 lines
5.9 KiB
Java
/*
|
|
* Copyright 2012 The Netty Project
|
|
*
|
|
* The Netty Project 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 io.netty.channel;
|
|
|
|
import io.netty.util.concurrent.RejectedExecutionHandler;
|
|
import io.netty.util.concurrent.RejectedExecutionHandlers;
|
|
import io.netty.util.concurrent.SingleThreadEventExecutor;
|
|
import io.netty.util.internal.ObjectUtil;
|
|
import io.netty.util.internal.SystemPropertyUtil;
|
|
import io.netty.util.internal.UnstableApi;
|
|
|
|
import java.util.Queue;
|
|
import java.util.concurrent.Executor;
|
|
import java.util.concurrent.ThreadFactory;
|
|
|
|
/**
|
|
* Abstract base class for {@link EventLoop}s that execute all its submitted tasks in a single thread.
|
|
*
|
|
*/
|
|
public abstract class SingleThreadEventLoop extends SingleThreadEventExecutor implements EventLoop {
|
|
|
|
protected static final int DEFAULT_MAX_PENDING_TASKS = Math.max(16,
|
|
SystemPropertyUtil.getInt("io.netty.eventLoop.maxPendingTasks", Integer.MAX_VALUE));
|
|
|
|
private final Queue<Runnable> tailTasks;
|
|
|
|
protected SingleThreadEventLoop(EventLoopGroup parent, ThreadFactory threadFactory, boolean addTaskWakesUp) {
|
|
this(parent, threadFactory, addTaskWakesUp, DEFAULT_MAX_PENDING_TASKS, RejectedExecutionHandlers.reject());
|
|
}
|
|
|
|
protected SingleThreadEventLoop(EventLoopGroup parent, Executor executor, boolean addTaskWakesUp) {
|
|
this(parent, executor, addTaskWakesUp, DEFAULT_MAX_PENDING_TASKS, RejectedExecutionHandlers.reject());
|
|
}
|
|
|
|
protected SingleThreadEventLoop(EventLoopGroup parent, ThreadFactory threadFactory,
|
|
boolean addTaskWakesUp, int maxPendingTasks,
|
|
RejectedExecutionHandler rejectedExecutionHandler) {
|
|
super(parent, threadFactory, addTaskWakesUp, maxPendingTasks, rejectedExecutionHandler);
|
|
tailTasks = newTaskQueue(maxPendingTasks);
|
|
}
|
|
|
|
protected SingleThreadEventLoop(EventLoopGroup parent, Executor executor,
|
|
boolean addTaskWakesUp, int maxPendingTasks,
|
|
RejectedExecutionHandler rejectedExecutionHandler) {
|
|
super(parent, executor, addTaskWakesUp, maxPendingTasks, rejectedExecutionHandler);
|
|
tailTasks = newTaskQueue(maxPendingTasks);
|
|
}
|
|
|
|
protected SingleThreadEventLoop(EventLoopGroup parent, Executor executor,
|
|
boolean addTaskWakesUp, Queue<Runnable> taskQueue, Queue<Runnable> tailTaskQueue,
|
|
RejectedExecutionHandler rejectedExecutionHandler) {
|
|
super(parent, executor, addTaskWakesUp, taskQueue, rejectedExecutionHandler);
|
|
tailTasks = ObjectUtil.checkNotNull(tailTaskQueue, "tailTaskQueue");
|
|
}
|
|
|
|
@Override
|
|
public EventLoopGroup parent() {
|
|
return (EventLoopGroup) super.parent();
|
|
}
|
|
|
|
@Override
|
|
public EventLoop next() {
|
|
return (EventLoop) super.next();
|
|
}
|
|
|
|
@Override
|
|
public ChannelFuture register(Channel channel) {
|
|
return register(new DefaultChannelPromise(channel, this));
|
|
}
|
|
|
|
@Override
|
|
public ChannelFuture register(final ChannelPromise promise) {
|
|
ObjectUtil.checkNotNull(promise, "promise");
|
|
promise.channel().unsafe().register(this, promise);
|
|
return promise;
|
|
}
|
|
|
|
@Deprecated
|
|
@Override
|
|
public ChannelFuture register(final Channel channel, final ChannelPromise promise) {
|
|
if (channel == null) {
|
|
throw new NullPointerException("channel");
|
|
}
|
|
if (promise == null) {
|
|
throw new NullPointerException("promise");
|
|
}
|
|
|
|
channel.unsafe().register(this, promise);
|
|
return promise;
|
|
}
|
|
|
|
/**
|
|
* Adds a task to be run once at the end of next (or current) {@code eventloop} iteration.
|
|
*
|
|
* @param task to be added.
|
|
*/
|
|
@UnstableApi
|
|
public final void executeAfterEventLoopIteration(Runnable task) {
|
|
ObjectUtil.checkNotNull(task, "task");
|
|
if (isShutdown()) {
|
|
reject();
|
|
}
|
|
|
|
if (!tailTasks.offer(task)) {
|
|
reject(task);
|
|
}
|
|
|
|
if (wakesUpForTask(task)) {
|
|
wakeup(inEventLoop());
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Removes a task that was added previously via {@link #executeAfterEventLoopIteration(Runnable)}.
|
|
*
|
|
* @param task to be removed.
|
|
*
|
|
* @return {@code true} if the task was removed as a result of this call.
|
|
*/
|
|
@UnstableApi
|
|
final boolean removeAfterEventLoopIterationTask(Runnable task) {
|
|
return tailTasks.remove(ObjectUtil.checkNotNull(task, "task"));
|
|
}
|
|
|
|
@Override
|
|
protected boolean wakesUpForTask(Runnable task) {
|
|
return !(task instanceof NonWakeupRunnable);
|
|
}
|
|
|
|
@Override
|
|
protected void afterRunningAllTasks() {
|
|
runAllTasksFrom(tailTasks);
|
|
}
|
|
|
|
@Override
|
|
protected boolean hasTasks() {
|
|
return super.hasTasks() || !tailTasks.isEmpty();
|
|
}
|
|
|
|
@Override
|
|
public int pendingTasks() {
|
|
return super.pendingTasks() + tailTasks.size();
|
|
}
|
|
|
|
/**
|
|
* Returns the number of {@link Channel}s registered with this {@link EventLoop} or {@code -1}
|
|
* if operation is not supported. The returned value is not guaranteed to be exact accurate and
|
|
* should be viewed as a best effort.
|
|
*/
|
|
@UnstableApi
|
|
public int registeredChannels() {
|
|
return -1;
|
|
}
|
|
|
|
/**
|
|
* Marker interface for {@link Runnable} that will not trigger an {@link #wakeup(boolean)} in all cases.
|
|
*/
|
|
interface NonWakeupRunnable extends Runnable { }
|
|
}
|