DatabaseSmsRoutingRule.java

/*
 * TeleStax, Open Source Cloud Communications  
 * Copyright 2012, Telestax Inc and individual contributors
 * by the @authors tag. See the copyright.txt in the distribution for a
 * full listing of individual contributors.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */

package org.mobicents.smsc.domain;

import java.util.List;

import org.apache.log4j.Logger;
import org.mobicents.smsc.cassandra.DBOperations;
import org.mobicents.smsc.cassandra.PersistenceException;
import org.mobicents.smsc.cassandra.SmsRoutingRuleType;
import org.mobicents.smsc.library.DbSmsRoutingRule;
import org.restcomm.smpp.EsmeManagement;

/**
 * @author Amit Bhayani
 * 
 */
public class DatabaseSmsRoutingRule implements DatabaseSmsRoutingRuleMBean {

    private static final Logger logger = Logger.getLogger(DatabaseSmsRoutingRule.class);

    private SmscPropertiesManagement smscPropertiesManagement;
    private EsmeManagement esmeManagement;
    private SipManagement sipManagement;

    // private DBOperations_C1 dbOperations_C1 = null;
    private DBOperations dbOperations_C2 = null;

    /**
	 * 
	 */
    public DatabaseSmsRoutingRule() {
        this.init();
    }

    @Override
    public void setEsmeManagement(EsmeManagement em) {
        this.esmeManagement = em;
    }

    @Override
    public void setSipManagement(SipManagement sm) {
        this.sipManagement = sm;
    }

    @Override
    public void setSmscPropertiesManagement(SmscPropertiesManagement sm) {
        this.smscPropertiesManagement = sm;
    }

    private void init() {
        try {
            dbOperations_C2 = DBOperations.getInstance();
        } catch (Exception e) {
            logger.error("Error initializing cassandra database for DatabaseSmsRoutingRule", e);
        }
    }

    /*
     * (non-Javadoc)
     * 
     * @see org.mobicents.smsc.smpp.SmsRoutingRule#getSystemId(byte, byte, java.lang.String)
     */
    @Override
    public String getEsmeClusterName(int ton, int npi, String address, String name, int networkId) {
        String clusterName = null;

        if (!dbOperations_C2.isDatabaseAvailable())
            return null;

        try {
            DbSmsRoutingRule rr;
            // if (smscPropertiesManagement.getDatabaseType() == DatabaseType.Cassandra_1) {
            // rr = dbOperations_C1.getSmsRoutingRule(address, networkId);
            // } else {
            rr = dbOperations_C2.c2_getSmppSmsRoutingRule(address, networkId);
            // }
            if (rr != null) {
                clusterName = rr.getClusterName();
            } else {
                if (smscPropertiesManagement == null)
                    smscPropertiesManagement = SmscPropertiesManagement.getInstance();
                if (smscPropertiesManagement != null) {
                    String dcn = smscPropertiesManagement.getEsmeDefaultClusterName();
                    if (dcn != null) {
                        if (esmeManagement.getEsmeByClusterName(dcn) != null) {
                            clusterName = dcn;
                        }
                    }
                }
            }
        } catch (PersistenceException e) {
            logger.error("PersistenceException while selecting from table SmsRoutingRule", e);
        }

        return clusterName;
    }

    @Override
    public String getSipClusterName(int ton, int npi, String address, int networkId) {
        String clusterName = null;

        if (!dbOperations_C2.isDatabaseAvailable())
            return null;

        try {
            DbSmsRoutingRule rr;
//            if (smscPropertiesManagement.getDatabaseType() == DatabaseType.Cassandra_1) {
//                return null;
//            }

            rr = dbOperations_C2.c2_getSipSmsRoutingRule(address, networkId);

            if (rr != null) {
                clusterName = rr.getClusterName();
            }
        } catch (PersistenceException e) {
            logger.error("PersistenceException while selecting from table SmsRoutingRule", e);
        }

        return clusterName;
    }

    @Override
    public void updateDbSmsRoutingRule(SmsRoutingRuleType dbSmsRoutingRuleType, String address, int networkId,
            String clusterName) throws PersistenceException {
        DbSmsRoutingRule dbSmsRoutingRule = new DbSmsRoutingRule(dbSmsRoutingRuleType, address, networkId, clusterName);

//        if (smscPropertiesManagement.getDatabaseType() == DatabaseType.Cassandra_1) {
//            dbOperations_C1.updateDbSmsRoutingRule(dbSmsRoutingRule);
//        } else {

        switch (dbSmsRoutingRuleType) {
            case SMPP:
                dbOperations_C2.c2_updateSmppSmsRoutingRule(dbSmsRoutingRule);
                break;
            case SIP:
                dbOperations_C2.c2_updateSipSmsRoutingRule(dbSmsRoutingRule);
                break;
            default:
                throw new PersistenceException("Unknown DbSmsRoutingRuleType=" + dbSmsRoutingRuleType);
        }

//        }
    }

    @Override
    public void deleteDbSmsRoutingRule(SmsRoutingRuleType dbSmsRoutingRuleType, String address, int networkId)
            throws PersistenceException {
//        if (smscPropertiesManagement.getDatabaseType() == DatabaseType.Cassandra_1) {
//            dbOperations_C1.deleteDbSmsRoutingRule(address);
//        } else {

        switch (dbSmsRoutingRuleType) {
            case SMPP:
                dbOperations_C2.c2_deleteSmppSmsRoutingRule(address, networkId);
                break;
            case SIP:
                dbOperations_C2.c2_deleteSipSmsRoutingRule(address, networkId);
                break;
            default:
                throw new PersistenceException("Unknown DbSmsRoutingRuleType=" + dbSmsRoutingRuleType);
        }

//        }
    }

    @Override
    public DbSmsRoutingRule getSmsRoutingRule(SmsRoutingRuleType dbSmsRoutingRuleType, String address, int networkId)
            throws PersistenceException {
//        if (smscPropertiesManagement.getDatabaseType() == DatabaseType.Cassandra_1) {
//            return dbOperations_C1.getSmsRoutingRule(address, networkId);
//        } else {


        switch (dbSmsRoutingRuleType) {
            case SMPP:
                return dbOperations_C2.c2_getSmppSmsRoutingRule(address, networkId);
            case SIP:
                return dbOperations_C2.c2_getSipSmsRoutingRule(address, networkId);
            default:
                throw new PersistenceException("Unknown DbSmsRoutingRuleType=" + dbSmsRoutingRuleType);
        }

//        }
    }

    @Override
    public List<DbSmsRoutingRule> getSmsRoutingRulesRange(SmsRoutingRuleType dbSmsRoutingRuleType) throws PersistenceException {
//        if (smscPropertiesManagement.getDatabaseType() == DatabaseType.Cassandra_1) {
//            return dbOperations_C1.getSmsRoutingRulesRange();
//        } else {



        switch (dbSmsRoutingRuleType) {
            case SMPP:
                return dbOperations_C2.c2_getSmppSmsRoutingRulesRange();
            case SIP:
                return dbOperations_C2.c2_getSipSmsRoutingRulesRange();
            default:
                throw new PersistenceException("Unknown DbSmsRoutingRuleType=" + dbSmsRoutingRuleType);
        }


//        }
    }

    @Override
    public List<DbSmsRoutingRule> getSmsRoutingRulesRange(SmsRoutingRuleType dbSmsRoutingRuleType, String lastAdress)
            throws PersistenceException {
//        if (smscPropertiesManagement.getDatabaseType() == DatabaseType.Cassandra_1) {
//            return dbOperations_C1.getSmsRoutingRulesRange(lastAdress);
//        } else {



        switch (dbSmsRoutingRuleType) {
            case SMPP:
                return dbOperations_C2.c2_getSmppSmsRoutingRulesRange(lastAdress);
            case SIP:
                return dbOperations_C2.c2_getSipSmsRoutingRulesRange(lastAdress);
            default:
                throw new PersistenceException("Unknown DbSmsRoutingRuleType=" + dbSmsRoutingRuleType);
        }            
   

//        }
    }
}