org.mobicents.ussdgateway.UssdManagement Maven / Gradle / Ivy

Go to download
/*
 * TeleStax, Open Source Cloud Communications
 * Copyright 2011-2017, Telestax Inc and individual contributors
 * by the @authors tag.
 *
 * This program is free software: you can redistribute it and/or modify
 * under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation; either version 3 of
 * the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see 
 */

package org.mobicents.ussdgateway;

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

import org.apache.log4j.Logger;
import org.jboss.mx.util.MBeanServerLocator;

/**
 * @author amit bhayani
 * 
 */
public class UssdManagement {
	private static final Logger logger = Logger.getLogger(UssdManagement.class);

	public static final String JMX_DOMAIN = "org.mobicents.ussdgateway";

	protected static final String USSD_PERSIST_DIR_KEY = "ussd.persist.dir";
	protected static final String USER_DIR_KEY = "user.dir";

	private String persistDir = null;
	private final String name;

	private UssdPropertiesManagement ussdPropertiesManagement = null;
	private ShortCodeRoutingRuleManagement shortCodeRoutingRuleManagement = null;

	private MBeanServer mbeanServer = null;

	public UssdManagement(String name) {
		this.name = name;
	}

	public String getName() {
		return name;
	}

	public String getPersistDir() {
		return persistDir;
	}

	public void setPersistDir(String persistDir) {
		this.persistDir = persistDir;
	}

	public void start() throws Exception {
        UssdStatAggregator.getInstance().clearDialogsInProcess();

	    this.ussdPropertiesManagement = UssdPropertiesManagement.getInstance(this.name);
		this.ussdPropertiesManagement.setPersistDir(this.persistDir);
		this.ussdPropertiesManagement.start();

		this.shortCodeRoutingRuleManagement = ShortCodeRoutingRuleManagement.getInstance(this.name);
		this.shortCodeRoutingRuleManagement.setPersistDir(this.persistDir);
		this.shortCodeRoutingRuleManagement.start();

		// Register the MBeans
		this.mbeanServer = MBeanServerLocator.locateJBoss();

		ObjectName ussdPropObjNname = new ObjectName(UssdManagement.JMX_DOMAIN + ":name=UssdPropertiesManagement");
		StandardMBean ussdPropMxBean = new StandardMBean(this.ussdPropertiesManagement,
				UssdPropertiesManagementMBean.class, true);
		this.mbeanServer.registerMBean(ussdPropMxBean, ussdPropObjNname);

		ObjectName ussdScRuleObjNname = new ObjectName(UssdManagement.JMX_DOMAIN
				+ ":name=ShortCodeRoutingRuleManagement");
		StandardMBean ussdScRuleMxBean = new StandardMBean(this.shortCodeRoutingRuleManagement,
				ShortCodeRoutingRuleManagementMBean.class, true);
		this.mbeanServer.registerMBean(ussdScRuleMxBean, ussdScRuleObjNname);

		logger.info("Started UssdManagement");
	}

	public void stop() throws Exception {
		this.ussdPropertiesManagement.stop();

		if (this.mbeanServer != null) {

			ObjectName ussdPropObjNname = new ObjectName(UssdManagement.JMX_DOMAIN + ":name=UssdPropertiesManagement");
			this.mbeanServer.unregisterMBean(ussdPropObjNname);

			ObjectName ussdScRuleObjNname = new ObjectName(UssdManagement.JMX_DOMAIN
					+ ":name=ShortCodeRoutingRuleManagement");
			this.mbeanServer.unregisterMBean(ussdScRuleObjNname);
		}
	}
}