AdminClient.java

/*
 * JBoss, Home of Professional Open Source
 * Copyright 2014, Red Hat, Inc. and/or its affiliates, and individual
 * contributors by the @authors tag. See the copyright.txt in the
 * distribution for a full listing of 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.as.quickstarts.datagrid.eap.app;

import java.util.Properties;
import java.util.logging.Level;
import java.util.logging.Logger;

import javax.naming.Context;
import javax.naming.InitialContext;

import org.jboss.ejb.client.ContextSelector;
import org.jboss.ejb.client.EJBClientConfiguration;
import org.jboss.ejb.client.EJBClientContext;
import org.jboss.ejb.client.PropertiesBasedEJBClientConfiguration;
import org.jboss.ejb.client.remoting.ConfigBasedEJBClientContextSelector;

/**
 * <p>
 * A simple standalone application which uses the JBoss API to invoke the MainApp demonstration
 * Bean.
 * </p>
 * <p>
 * With the boolean property <i>UseScopedContext</i> the basic example or the example with the
 * scoped-environment will be called.
 * </p>
 * 
 * @author <a href="mailto:wfink@redhat.com">Wolf-Dieter Fink</a>
 */
public class AdminClient {

   /**
    * @param args
    *           optional to override the connect parameter
    *           <ul>
    *           <li>hostname for AdminApp (default localhost)</li>
    *           <li>port for AdminApp (default 4447)</li>
    *           <li>hostname for AppOne (default localhost)</li>
    *           <li>port for AppOne (default 4547)</li>
    *           </ul>
    * @throws Exception
    */
   public static void main(String[] args) throws Exception {
      String hostAdmin = "localhost";
      String portAdmin = "8080";
      String hostAppOne = "localhost";
      String portAppOne = "8180";
      
      // suppress output of client messages
      Logger.getLogger("org.jboss").setLevel(Level.OFF);
      Logger.getLogger("org.xnio").setLevel(Level.OFF);

      if(args.length > 0) {
         hostAdmin = args[0];
      }
      if(args.length > 1) {
         portAdmin = args[1];
      }
      if(args.length > 2) {
         hostAppOne = args[2];
      }
      if(args.length > 3) {
         portAppOne = args[3];
      }
      
      Properties p = new Properties();
      p.put("remote.connectionprovider.create.options.org.xnio.Options.SSL_ENABLED", "false");
      p.put("remote.connections", "admin,appOne");
      p.put("remote.connection.admin.port", portAdmin);
      p.put("remote.connection.admin.host", hostAdmin);
      p.put("remote.connection.admin.username", "quickuser");
      p.put("remote.connection.admin.password", "quick-123");
      p.put("remote.connection.appOne.port", portAppOne);
      p.put("remote.connection.appOne.host", hostAppOne);
      p.put("remote.connection.appOne.username", "quickuser");
      p.put("remote.connection.appOne.password", "quick-123");

      EJBClientConfiguration cc = new PropertiesBasedEJBClientConfiguration(p);
      ContextSelector<EJBClientContext> selector = new ConfigBasedEJBClientContextSelector(cc);
      EJBClientContext.setSelector(selector);

      Properties props = new Properties();
      props.put(Context.URL_PKG_PREFIXES, "org.jboss.ejb.client.naming");
      InitialContext context = new InitialContext(props);

      final String adminLookup = "ejb:jboss-eap-application-adminApp/ejb//CacheAdminBean!" + CacheAdmin.class.getName();
      final CacheAdmin admin = (CacheAdmin) context.lookup(adminLookup);

      System.out.println("        Add a value to App1Cache with the AdminApp and check on the same instance that the value is correct added");
      admin.addToApp1Cache("App1One", "The App1 One entry");
      // check that the Admin has the correct key entry local
      admin.verifyApp1Cache("App1One", "The App1 One entry");
      System.out.println("          success");

      final String appOneLookup = "ejb:jboss-eap-application-AppOne/ejb//AppOneCacheAccessBean!"
            + AppOneCacheAccess.class.getName();
      final AppOneCacheAccess appOne = (AppOneCacheAccess) context.lookup(appOneLookup);
      
      System.out.println("        Check the previous added value of App1Cache by accessing the AppOne Server");
      // check that the App1 cache has the correct replicated key entry
      appOne.verifyApp1Cache("App1One", "The App1 One entry");
      System.out.println("          success");

      System.out.println("        Add a value to App2Cache and check on the same instance that the value is correct added");
      admin.addToApp2Cache("One", "The App2 One entry");
      admin.verifyApp2Cache("One", "The App2 One entry");
      System.out.println("          success");

      // check that the cache is transactional
      System.out.println("        Check whether changes to a cache are rollbacked if the transaction fail");
      admin.removeFromApp2Cache("ShouldNeverExist");
      admin.addToApp2Cache("ShouldNeverExist", "This value should not available as the transaction fail", true);
      if (admin.containsApp2Key("ShouldNeverExist")) {
         throw new RuntimeException("Unexpected result, the key exists after transaction rollback!");
      } else {
         System.out.println("        The cache App2 work as expected on rollback");
      }
   }
}