/*
* 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.ignite.internal.util.nio;
import org.apache.ignite.lang.IgniteRunnable;
import org.jetbrains.annotations.Nullable;
/**
* Utility class that allows to ignore back-pressure control for threads that are processing messages.
*/
public class GridNioBackPressureControl {
/** Thread local flag indicating that thread is processing message. */
private static ThreadLocal<Holder> threadProcMsg = new ThreadLocal<Holder>() {
@Override protected Holder initialValue() {
return new Holder();
}
};
/**
* @return Flag indicating whether current thread is processing message.
*/
public static boolean threadProcessingMessage() {
return threadProcMsg.get().procMsg;
}
/**
* @param processing Flag indicating whether current thread is processing message.
* @param tracker Thread local back pressure tracker of messages, associated with one connection.
*/
public static void threadProcessingMessage(boolean processing, @Nullable IgniteRunnable tracker) {
Holder holder = threadProcMsg.get();
holder.procMsg = processing;
holder.tracker = tracker;
}
/**
* @return Thread local back pressure tracker of messages, associated with one connection.
*/
@Nullable public static IgniteRunnable threadTracker() {
return threadProcMsg.get().tracker;
}
/**
*
*/
private static class Holder {
/** Process message. */
private boolean procMsg;
/** Tracker. */
private IgniteRunnable tracker;
}
}