package com.sun.enterprise.shoal.jointest;

import com.sun.enterprise.ee.cms.core.*;
import com.sun.enterprise.ee.cms.impl.client.JoinNotificationActionFactoryImpl;

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

public class SimpleJoinTest {

    private final static Logger logger = Logger.getLogger( "SimpleJoinTest" );

    private final String group = "TestGroup";

    public static void main( String[] args ) {
        SimpleJoinTest check = new SimpleJoinTest();
        try {
            check.runSimpleSample();
        } catch( GMSException e ) {
            logger.log( Level.SEVERE, "Exception occured while joining group:" + e );
        }
    }

    private void runSimpleSample() throws GMSException {
        logger.log( Level.INFO, "Starting SimpleJoinTest...." );

        String serverName = UUID.randomUUID().toString();;

        //initialize Group Management Service
        GroupManagementService gms = initializeGMS( serverName, group );

        //register for Group Events
        logger.log( Level.INFO, "Registering for group event notifications" );
        gms.addActionFactory( new JoinNotificationActionFactoryImpl( new JoinNotificationCallBack( serverName ) ) );

        //join group
        logger.log( Level.INFO, "Joining Group " + group );
        gms.join();

        //leaveGroupAndShutdown( serverName, gms );
    }

    private GroupManagementService initializeGMS( String serverName, String groupName ) {
        logger.log( Level.INFO, "Initializing Shoal for member: " + serverName + " group:" + groupName );
        return (GroupManagementService)GMSFactory.startGMSModule( serverName,
                                                                  groupName,
                                                                  GroupManagementService.MemberType.CORE,
                                                                  //null ); // Now if properties is null, NPE occurred.
                                                                  new Properties() );
    }

    private void leaveGroupAndShutdown( String serverName, GroupManagementService gms ) {
        logger.log( Level.INFO, "Shutting down gms " + gms + "for server " + serverName );
        gms.shutdown( GMSConstants.shutdownType.INSTANCE_SHUTDOWN );
    }

    private class JoinNotificationCallBack implements CallBack {

        private String serverName;

        public JoinNotificationCallBack( String serverName ) {
            this.serverName = serverName;
        }

        public void processNotification( Signal notification ) {
            if ( !( notification instanceof JoinNotificationSignal ) )
                logger.log( Level.SEVERE, "received unkown notification type:" + notification );
            logger.log( Level.INFO, "***JoinNotification received: ServerName = " + serverName + ", Signal.getMemberToken() = " + notification.getMemberToken() );
        }
    }
}