Examples with VirtualChannelSelectorImpl com.biglybt.core.networkmanager.impl.tcp.VirtualChannelSelectorImpl used on opensource projects

Example 6 with VirtualChannelSelectorImpl

use of com.biglybt.core.networkmanager.impl.tcp.VirtualChannelSelectorImpl in project BiglyBT by BiglySoftware.

the class VirtualChannelSelector method registerSupport.

/**
 * Register the given selectable channel, using the given listener for notification
 * of completed select operations.
 * NOTE: For OP_CONNECT and OP_WRITE -type selectors, once a selection request op
 * completes, the channel's op registration is automatically disabled (paused); any
 * future wanted selection notification requires re-enabling via resume.  For OP_READ selectors,
 * it stays enabled until actively paused, no matter how many times it is selected.
 * @param channel socket to listen for
 * @param listener op-complete listener
 * @param attachment object to be passed back with listener notification
 */
protected void registerSupport(AbstractSelectableChannel channel, VirtualAbstractSelectorListener listener, Object attachment) {
    if (SAFE_SELECTOR_MODE_ENABLED) {
        try {
            selectors_mon.enter();
            // System.out.println( "register - " + channel.hashCode()  + " - " + Debug.getCompressedStackTrace());
            for (Map.Entry<VirtualChannelSelectorImpl, ArrayList<AbstractSelectableChannel>> entry : selectors.entrySet()) {
                VirtualChannelSelectorImpl sel = entry.getKey();
                ArrayList<AbstractSelectableChannel> channels = entry.getValue();
                if (channels.size() >= (TEST_SAFE_MODE ? 0 : MAX_CHANNELS_PER_SAFE_SELECTOR)) {
                    // it seems that we have a bug somewhere where a selector is being registered
                    // but not cancelled on close. As an interim fix scan channels and remove any
                    // closed ones
                    Iterator<AbstractSelectableChannel> chan_it = channels.iterator();
                    while (chan_it.hasNext()) {
                        AbstractSelectableChannel chan = chan_it.next();
                        if (!chan.isOpen()) {
                            Debug.out("Selector '" + getName() + "' - removing orphaned safe channel registration");
                            chan_it.remove();
                        }
                    }
                }
                if (channels.size() < MAX_CHANNELS_PER_SAFE_SELECTOR) {
                    // there's room in the current selector
                    sel.register(channel, listener, attachment);
                    channels.add(channel);
                    return;
                }
            }
            // max limit to the number of Selectors we are allowed to create
            if (selectors.size() >= MAX_SAFEMODE_SELECTORS) {
                String msg = "Error: MAX_SAFEMODE_SELECTORS reached [" + selectors.size() + "], no more socket channels can be registered. Too many peer connections.";
                Debug.out(msg);
                // reject registration
                selectFailure(listener, channel, attachment, new Throwable(msg));
                return;
            }
            if (destroyed) {
                String msg = "socket registered after controller destroyed";
                Debug.out(msg);
                // reject registration
                selectFailure(listener, channel, attachment, new Throwable(msg));
                return;
            }
            VirtualChannelSelectorImpl sel = new VirtualChannelSelectorImpl(this, op, pause, randomise_keys);
            ArrayList<AbstractSelectableChannel> chans = new ArrayList<>();
            selectors.put(sel, chans);
            sel.register(channel, listener, attachment);
            chans.add(channel);
            selectors_keyset_cow = new HashSet<>(selectors.keySet());
        } finally {
            selectors_mon.exit();
        }
    } else {
        selector_impl.register(channel, listener, attachment);
    }
}