DataStoreAdd.java

/**
 * JBoss, Home of Professional Open Source
 * Copyright Red Hat, Inc., and individual contributors
 *
 * 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.jboss.aerogear.simplepush.subsystem;


import java.util.List;

import org.jboss.aerogear.simplepush.server.datastore.DataStore;
import org.jboss.as.controller.AbstractAddStepHandler;
import org.jboss.as.controller.OperationContext;
import org.jboss.as.controller.OperationFailedException;
import org.jboss.as.controller.PathAddress;
import org.jboss.as.controller.ServiceVerificationHandler;
import org.jboss.as.naming.deployment.ContextNames;
import org.jboss.as.naming.deployment.ContextNames.BindInfo;
import org.jboss.dmr.ModelNode;
import org.jboss.logging.Logger;
import org.jboss.msc.service.ServiceBuilder;
import org.jboss.msc.service.ServiceController;
import org.jboss.msc.service.ServiceController.Mode;

class DataStoreAdd extends AbstractAddStepHandler {

    public static final DataStoreAdd INSTANCE = new DataStoreAdd();
    private final Logger logger = Logger.getLogger(DataStoreAdd.class);

    private DataStoreAdd() {
    }

    @Override
    protected void populateModel(final ModelNode operation, final ModelNode model) throws OperationFailedException {
        DataStoreDefinition.DATASOURCE_ATTR.validateAndSet(operation, model);
        DataStoreDefinition.PERSISTENCE_UNIT_ATTR.validateAndSet(operation, model);
        DataStoreDefinition.HOST_ATTR.validateAndSet(operation, model);
        DataStoreDefinition.PORT_ATTR.validateAndSet(operation, model);
        DataStoreDefinition.URL_ATTR.validateAndSet(operation, model);
        DataStoreDefinition.DB_NAME_ATTR.validateAndSet(operation, model);
    }

    @Override
    protected void performRuntime(OperationContext context, ModelNode operation, ModelNode model,
            ServiceVerificationHandler verificationHandler, List<ServiceController<?>> newControllers)
            throws OperationFailedException {

        final PathAddress pathAddress = PathAddress.pathAddress(operation.get("address"));
        final String serverName = pathAddress.getElement(1).getValue();
        final String type = pathAddress.getLastElement().getValue();
        ServiceBuilder<DataStore> sb;
        switch (DataStoreDefinition.Element.of(type)) {
            case JPA:
                final ModelNode datasourceNode = DataStoreDefinition.DATASOURCE_ATTR.resolveModelAttribute(context, model);
                final ModelNode persistenceUnitNode = DataStoreDefinition.PERSISTENCE_UNIT_ATTR.resolveModelAttribute(context, model);
                final BindInfo bindinfo = ContextNames.bindInfoFor(datasourceNode.asString());
                logger.debug("Adding dependency to [" + bindinfo.getAbsoluteJndiName() + "]");
                DataStoreService jpa = new JpaDataStoreService(persistenceUnitNode.asString());
                sb = context.getServiceTarget().addService(DataStoreService.SERVICE_NAME.append(serverName), jpa);
                sb.addDependencies(bindinfo.getBinderServiceName());
                break;
            case REDIS:
                final ModelNode hostNode = DataStoreDefinition.HOST_ATTR.resolveModelAttribute(context, model);
                final ModelNode portNode = DataStoreDefinition.PORT_ATTR.resolveModelAttribute(context, model);
                final DataStoreService redis = new RedisDataStoreService(hostNode.asString(), portNode.asInt());
                sb = context.getServiceTarget().addService(DataStoreService.SERVICE_NAME.append(serverName), redis);
                break;
            case COUCHDB:
                final ModelNode urlNode = DataStoreDefinition.URL_ATTR.resolveModelAttribute(context, model);
                final ModelNode dbNameNode = DataStoreDefinition.DB_NAME_ATTR.resolveModelAttribute(context, model);
                final DataStoreService couchdb = new CouchDBDataStoreService(urlNode.asString(), dbNameNode.asString());
                sb = context.getServiceTarget().addService(DataStoreService.SERVICE_NAME.append(serverName), couchdb);
                break;
            case IN_MEMORY:
                sb = context.getServiceTarget().addService(DataStoreService.SERVICE_NAME.append(serverName), new InMemoryDataStoreService());
                break;
            default:
                throw new IllegalStateException("invalid datastore type");
        }
        sb.addListener(verificationHandler);
        sb.setInitialMode(Mode.ACTIVE);
        newControllers.add(sb.install());
    }



}