/*
 * Copyright 2008-2009 LinkedIn, 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 voldemort.store.readonly;

import java.io.File;
import java.lang.management.ManagementFactory;
import java.util.Collections;
import java.util.HashSet;
import java.util.Set;

import javax.management.MBeanServer;
import javax.management.ObjectName;

import voldemort.VoldemortException;
import voldemort.routing.RoutingStrategy;
import voldemort.server.VoldemortConfig;
import voldemort.store.StorageConfiguration;
import voldemort.store.StorageEngine;
import voldemort.store.StoreDefinition;
import voldemort.utils.ByteArray;
import voldemort.utils.JmxUtils;
import voldemort.utils.ReflectUtils;

public class ReadOnlyStorageConfiguration implements StorageConfiguration {

    public static final String TYPE_NAME = "read-only";

    private final int numBackups;
    private final File storageDir;
    private final Set<ObjectName> registeredBeans;
    private final SearchStrategy searcher;
    private final int nodeId;
    private RoutingStrategy routingStrategy = null;
    private final int deleteBackupMs;
    private final int maxValueBufferAllocationSize;
    private final VoldemortConfig voldConfig;

    public ReadOnlyStorageConfiguration(VoldemortConfig config) {
        this.storageDir = new File(config.getReadOnlyDataStorageDirectory());
        this.numBackups = config.getNumReadOnlyVersions();
        this.registeredBeans = Collections.synchronizedSet(new HashSet<ObjectName>());
        this.searcher = (SearchStrategy) ReflectUtils.callConstructor(ReflectUtils.loadClass(config.getReadOnlySearchStrategy()
                                                                                                   .trim()));
        this.nodeId = config.getNodeId();
        this.deleteBackupMs = config.getReadOnlyDeleteBackupMs();
        this.maxValueBufferAllocationSize = config.getReadOnlyMaxValueBufferAllocationSize();
        this.voldConfig = config;
    }

    public void close() {
        MBeanServer server = ManagementFactory.getPlatformMBeanServer();
        for(ObjectName name: registeredBeans)
            JmxUtils.unregisterMbean(server, name);
    }

    public void setRoutingStrategy(RoutingStrategy routingStrategy) {
        this.routingStrategy = routingStrategy;
    }

    public StorageEngine<ByteArray, byte[], byte[]> getStore(StoreDefinition storeDef,
                                                             RoutingStrategy strategy) {
        this.setRoutingStrategy(strategy);
        ReadOnlyStorageEngine store = new ReadOnlyStorageEngine(storeDef.getName(),
                                                                this.searcher,
                                                                this.routingStrategy,
                                                                this.nodeId,
                                                                new File(storageDir,
                                                                         storeDef.getName()),
                                                                numBackups,
                                                                deleteBackupMs,
                                                                maxValueBufferAllocationSize);
        ObjectName objName = JmxUtils.createObjectName(JmxUtils.getPackageName(store.getClass()),
                                                       storeDef.getName() );
        JmxUtils.registerMbean(ManagementFactory.getPlatformMBeanServer(),
                               JmxUtils.createModelMBean(store),
                               objName);
        registeredBeans.add(objName);

        return store;
    }

    public String getType() {
        return TYPE_NAME;
    }

    public void update(StoreDefinition storeDef) {
        throw new VoldemortException("Storage config updates not permitted for "
                                     + this.getClass().getCanonicalName());
    }

    /**
     * Cleanup the Jmx bean registered previously
     */
    @Override
    public void removeStorageEngine(StorageEngine<ByteArray, byte[], byte[]> engine) {
        ReadOnlyStorageEngine store = (ReadOnlyStorageEngine) engine;

        if(this.voldConfig.isJmxEnabled()) {
            ObjectName objName = JmxUtils.createObjectName(JmxUtils.getPackageName(store.getClass()),
                                                           store.getName() + nodeId);
            JmxUtils.unregisterMbean(ManagementFactory.getPlatformMBeanServer(), objName);
        }
    }
}

@@ -654,207 +654,8 @@
 ile;
-%0Aimport java.lang.management.ManagementFactory;%0Aimport java.util.Collections;%0Aimport java.util.HashSet;%0Aimport java.util.Set;%0A%0Aimport javax.management.MBeanServer;%0Aimport javax.management.ObjectName;
 %0A%0Aim
@@ -933,41 +933,8 @@
 ay;%0A
-import voldemort.utils.JmxUtils;%0A
 impo
@@ -1174,59 +1174,8 @@
 ir;%0A
-    private final Set%3CObjectName%3E registeredBeans;%0A
     
@@ -1639,95 +1639,8 @@
 ();%0A
-        this.registeredBeans = Collections.synchronizedSet(new HashSet%3CObjectName%3E());%0A
     
@@ -2146,178 +2146,8 @@
 ) %7B%0A
-        MBeanServer server = ManagementFactory.getPlatformMBeanServer();%0A        for(ObjectName name: registeredBeans)%0A            JmxUtils.unregisterMbean(server, name);
 %0A   
@@ -3254,403 +3254,8 @@
 e);%0A
-        ObjectName objName = JmxUtils.createObjectName(JmxUtils.getPackageName(store.getClass()),%0A                                                       storeDef.getName() );%0A        JmxUtils.registerMbean(ManagementFactory.getPlatformMBeanServer(),%0A                               JmxUtils.createModelMBean(store),%0A                               objName);%0A        registeredBeans.add(objName);%0A%0A
     
@@ -3728,413 +3728,8 @@
 ) %7B%0A
-        ReadOnlyStorageEngine store = (ReadOnlyStorageEngine) engine;%0A%0A        if(this.voldConfig.isJmxEnabled()) %7B%0A            ObjectName objName = JmxUtils.createObjectName(JmxUtils.getPackageName(store.getClass()),%0A                                                           store.getName() + nodeId);%0A            JmxUtils.unregisterMbean(ManagementFactory.getPlatformMBeanServer(), objName);%0A        %7D%0A
     

Fix RO JMX register/Unregister