Project: CouchbaseMock
/**
 *     Copyright 2012 Couchbase, Inc. 
 * 
 *   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 org.couchbase.mock.control; 
 
import java.util.HashSet; 
import java.util.List; 
import java.util.Set; 
 
import org.couchbase.mock.Bucket; 
import org.couchbase.mock.CouchbaseMock; 
import org.couchbase.mock.memcached.MemcachedServer; 
 
public abstract class ServersCommandHandler implements MockControlCommandHandler { 
 
    abstract void doServerCommand(MemcachedServer server); 
 
    abstract void extractParams(List<String> tokens); 
 
    @Override 
    public void execute(CouchbaseMock mock, List<String> tokens) { 
        extractParams(tokens); 
        Set<MemcachedServer> servers = new HashSet<MemcachedServer>(); 
 
        for (Bucket bucket : mock.getBuckets().values()) { 
            for (MemcachedServer server : bucket.getServers()) { 
                if (servers.contains(server) == false) { 
                    doServerCommand(server); 
                    servers.add(server); 
                } 
            } 
        } 
 
    } 
}