* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */
package jcifs.internal.smb2.multichannel;

/**
 * Load balancing strategies for multi-channel connections
 */
public enum LoadBalancingStrategy {
    /**
     * Round-robin selection through available channels
     */
    ROUND_ROBIN,

    /**
     * Select the least busy channel based on pending operations
     */
    LEAST_LOADED,

    void testStrategyChange() {
        assertEquals(LoadBalancingStrategy.ADAPTIVE, loadBalancer.getStrategy());

        loadBalancer.setStrategy(LoadBalancingStrategy.ROUND_ROBIN);
        assertEquals(LoadBalancingStrategy.ROUND_ROBIN, loadBalancer.getStrategy());
    }

    @Test
    void testZeroScoreChannels() {
        loadBalancer.setStrategy(LoadBalancingStrategy.WEIGHTED_RANDOM);

        // Test setting different strategies
        LoadBalancingStrategy[] strategies = { LoadBalancingStrategy.ROUND_ROBIN, LoadBalancingStrategy.LEAST_LOADED,
                LoadBalancingStrategy.WEIGHTED_RANDOM, LoadBalancingStrategy.AFFINITY_BASED, LoadBalancingStrategy.ADAPTIVE };

        for (LoadBalancingStrategy strategy : strategies) {
            loadBalancer.setStrategy(strategy);

    private final ChannelManager manager;
    private LoadBalancingStrategy strategy;
    private final AtomicInteger roundRobinCounter;

    /**
     * Create channel load balancer
     *
     * @param manager channel manager
     */
    public ChannelLoadBalancer(ChannelManager manager) {
        this.manager = manager;
        this.strategy = LoadBalancingStrategy.ADAPTIVE;
        this.roundRobinCounter = new AtomicInteger(0);

            props.setProperty("jcifs.smb.client.loadBalancingStrategy", strategy);
            PropertyConfiguration config = new PropertyConfiguration(props);
            assertEquals(strategy, config.getLoadBalancingStrategy());
        }

        // Test case insensitivity
        props.setProperty("jcifs.smb.client.loadBalancingStrategy", "ADAPTIVE");