try {
                    rotateSessionKey(sessionId);
                    rotated++;
                } catch (Exception e) {
                    log.warn("Failed to rotate key for session: {}", sessionId, e);
                }
            }
        }

        log.info("Force rotated {} keys", rotated);
        return rotated;
    }

    /**
     * Clean up old archived key versions
     *

        for (int i = 0; i < 3; i++) {
            keyManager.storeSessionKey("session-" + i, testKey, "AES");
        }

        // Force rotate all
        int rotated = keyManager.forceRotateAllKeys();
        assertEquals(3, rotated, "Should rotate 3 keys");

        // Verify all keys were rotated
        for (int i = 0; i < 3; i++) {
            assertEquals(1, keyManager.getKeyVersion("session-" + i), "Key version should be 1");

        if (closed) {
            throw new IllegalStateException("Cannot rotate keys on closed context");
        }

        log.info("Rotating encryption keys for session: {}", sessionId);

        if (keyManager != null) {
            // Rotate keys in SecureKeyManager
            String encKeyId = sessionId + "-enc";
            String decKeyId = sessionId + "-dec";

ue,\"truncate\":100},\"title\":{}}],\"valueAxes\":[{\"id\":\"ValueAxis-1\",\"name\":\"LeftAxis-1\",\"type\":\"value\",\"position\":\"left\",\"show\":true,\"style\":{},\"scale\":{\"type\":\"linear\",\"mode\":\"normal\"},\"labels\":{\"show\":true,\"rotate\":0,\"filter\":false,\"truncate\":100},\"title\":{\"text\":\"Query Count\"}}],\"seriesParams\":[{\"show\":\"true\",\"type\":\"histogram\",\"mode\":\"stacked\",\"data\":{\"label\":\"Query Count\",\"id\":\"1\"},\"valueAxis\":\"ValueAxis-1\",\"drawL...

        // Should be idempotent
        assertDoesNotThrow(() -> context.close());

        // Cleanup
        keyManager.close();
    }

    @Test
    @DisplayName("Should rotate keys successfully")
    void testKeyRotation() throws Exception {
        // Given
        Smb2EncryptionContext context = new Smb2EncryptionContext(EncryptionNegotiateContext.CIPHER_AES128_GCM, DialectVersion.SMB311,

public class ChannelLoadBalancer {
    private final ChannelManager manager;
    private LoadBalancingStrategy strategy;
    
    public enum LoadBalancingStrategy {
        ROUND_ROBIN,      // Rotate through channels
        LEAST_LOADED,     // Select least busy channel
        WEIGHTED_RANDOM,  // Random selection weighted by score
        AFFINITY_BASED,   // Stick to same channel for related operations