hazelcast-master/hazelcast-tpc-engine/src/main/java/com/hazelcast/internal/tpcengine/nio/NioReactor.java
/*
 * Copyright (c) 2008-2023, Hazelcast, Inc. All Rights Reserved.
 *
 * Licensed 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 com.hazelcast.internal.tpcengine.nio;

import com.hazelcast.internal.tpcengine.net.AcceptRequest;
import com.hazelcast.internal.tpcengine.Reactor;
import com.hazelcast.internal.tpcengine.ReactorBuilder;

import java.nio.channels.Selector;

import static com.hazelcast.internal.tpcengine.util.Preconditions.checkInstanceOf;

/**
 * Nio implementation of the {@link Reactor}.
 */
public final class NioReactor extends Reactor {

    final Selector selector;

    NioReactor(NioReactorBuilder builder) {
        super(builder);
        this.selector = ((NioEventloop) eventloop()).selector;
    }

    @Override
    public NioAsyncSocketBuilder newAsyncSocketBuilder() {
        verifyRunning();

        return new NioAsyncSocketBuilder(this, null);
    }

    @Override
    public NioAsyncSocketBuilder newAsyncSocketBuilder(AcceptRequest acceptRequest) {
        verifyRunning();

        NioAcceptRequest nioAcceptRequest = checkInstanceOf(NioAcceptRequest.class, acceptRequest, "acceptRequest");
        return new NioAsyncSocketBuilder(this, nioAcceptRequest);
    }

    @Override
    public NioAsyncServerSocketBuilder newAsyncServerSocketBuilder() {
        verifyRunning();

        return new NioAsyncServerSocketBuilder(this);
    }

    @Override
    protected NioEventloop newEventloop(ReactorBuilder builder) {
        return new NioEventloop(this, (NioReactorBuilder) builder);
    }

    @Override
    public void wakeup() {
        if (spin || Thread.currentThread() == eventloopThread) {
            return;
        }

        if (wakeupNeeded.get() && wakeupNeeded.compareAndSet(true, false)) {
            selector.wakeup();
        }
    }
}
