JNDIDataSourceProvider.java 3.49 KB
Newer Older
Matt Tucker's avatar
Matt Tucker committed
1 2 3 4
/**
 * $Revision$
 * $Date$
 *
5
 * Copyright (C) 2004-2005 Jive Software. All rights reserved.
Matt Tucker's avatar
Matt Tucker committed
6 7 8 9 10 11 12
 *
 * This software is published under the terms of the GNU Public License (GPL),
 * a copy of which is included in this distribution.
 */

package org.jivesoftware.database;

13
import org.jivesoftware.util.JiveGlobals;
Matt Tucker's avatar
Matt Tucker committed
14 15 16 17 18
import org.jivesoftware.util.Log;

import javax.naming.Context;
import javax.naming.InitialContext;
import javax.sql.DataSource;
19 20 21
import java.sql.Connection;
import java.sql.SQLException;
import java.util.*;
Matt Tucker's avatar
Matt Tucker committed
22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62

/**
 * An implementation of ConnectionProvider that utilizes a JDBC 2.0 DataSource
 * made available via JNDI. This is useful for application servers where a pooled
 * data connection is already provided so Jive can share the pool with the
 * other applications.<p>
 * <p/>
 * The JNDI location of the DataSource stored as the Jive property
 * <code>database.JNDIProvider.name</code>. This can be overridden by setting
 * the provider's <code>name</code> property if required.
 *
 * @author <a href="mailto:joe@truemesh.com">Joe Walnes</a>
 * @see ConnectionProvider
 */
public class JNDIDataSourceProvider implements ConnectionProvider {

    private String dataSourceName;
    private DataSource dataSource;

    /**
     * Keys of JNDI properties to query PropertyManager for.
     */
    private static final String[] jndiPropertyKeys = {
        Context.APPLET,
        Context.AUTHORITATIVE,
        Context.BATCHSIZE,
        Context.DNS_URL,
        Context.INITIAL_CONTEXT_FACTORY,
        Context.LANGUAGE,
        Context.OBJECT_FACTORIES,
        Context.PROVIDER_URL,
        Context.REFERRAL,
        Context.SECURITY_AUTHENTICATION,
        Context.SECURITY_CREDENTIALS,
        Context.SECURITY_PRINCIPAL,
        Context.SECURITY_PROTOCOL,
        Context.STATE_FACTORIES,
        Context.URL_PKG_PREFIXES
    };

    /**
63
     * Constructs a new JNDI pool.
Matt Tucker's avatar
Matt Tucker committed
64 65
     */
    public JNDIDataSourceProvider() {
66
        dataSourceName = JiveGlobals.getXMLProperty("database.JNDIProvider.name");
Matt Tucker's avatar
Matt Tucker committed
67 68 69 70 71 72 73 74
    }

    public boolean isPooled() {
        return true;
    }

    public void start() {
        if (dataSourceName == null || dataSourceName.equals("")) {
75
            Log.error("No name specified for DataSource. JNDI lookup will fail", null);
Matt Tucker's avatar
Matt Tucker committed
76 77 78 79
            return;
        }
        try {
            Properties contextProperties = new Properties();
80 81 82 83
            for (String key: jndiPropertyKeys) {
                String value = JiveGlobals.getXMLProperty(key);
                if (value != null) {
                    contextProperties.setProperty(key, value);
Matt Tucker's avatar
Matt Tucker committed
84 85
                }
            }
86
            Context context;
Matt Tucker's avatar
Matt Tucker committed
87 88 89 90 91 92 93 94 95
            if (contextProperties.size() > 0) {
                context = new InitialContext(contextProperties);
            }
            else {
                context = new InitialContext();
            }
            dataSource = (DataSource)context.lookup(dataSourceName);
        }
        catch (Exception e) {
96
            Log.error("Could not lookup DataSource at '" + dataSourceName + "'", e);
Matt Tucker's avatar
Matt Tucker committed
97 98 99 100 101 102 103 104 105 106 107 108 109 110
        }
    }

    public void restart() {
        destroy();
        start();
    }

    public void destroy() {

    }

    public Connection getConnection() {
        if (dataSource == null) {
111
            Log.error("DataSource has not been initialized.", null);
Matt Tucker's avatar
Matt Tucker committed
112 113 114 115 116 117
            return null;
        }
        try {
            return dataSource.getConnection();
        }
        catch (SQLException e) {
118
            Log.error("Could not retrieve Connection from DataSource", e);
Matt Tucker's avatar
Matt Tucker committed
119 120 121
            return null;
        }
    }
122
}