de.mhus.osgi.api.util.DataSourceUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of osgi-api Show documentation
commands OSGi blueprint bundle project.
The newest version!
/**
 * Copyright (C) 2018 Mike Hummel ([email protected])
 *
 * 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 de.mhus.osgi.api.util;

import java.util.Collection;
import java.util.Dictionary;
import java.util.Hashtable;

import javax.sql.DataSource;

import org.osgi.framework.BundleContext;
import org.osgi.framework.InvalidSyntaxException;
import org.osgi.framework.ServiceReference;
import org.osgi.service.jdbc.DataSourceFactory;

import de.mhus.osgi.api.MOsgi;

public class DataSourceUtil {

    public static final String SERVICE_JNDI_NAME_KEY = "osgi.jndi.service.name";

    public static DataSource getDataSource(String name) {
        BundleContext context = MOsgi.getBundleContext();
        return getDataSource(name, context);
    }

    public static DataSource getDataSource(String name, BundleContext context) {
        try {
            // BundleContext context =
            // FrameworkUtil.getBundle(DataSourceUtil.class).getBundleContext();
            if (context == null) return null;
            Collection> refs =
                    context.getServiceReferences(
                            DataSource.class, "(" + SERVICE_JNDI_NAME_KEY + "=" + name + ")");
            if (refs.size() > 0) {
                return context.getService(refs.iterator().next());
            }
        } catch (InvalidSyntaxException e) {
            throw new RuntimeException(e);
        }

        //        ServiceReference[] dataSources = getDataSources();
        //        for (ServiceReference ref : dataSources) {
        //            String jndiName = getServiceJndiName(ref);
        //            if (name.equals(jndiName)) {
        //                DataSource ds = (DataSource)getContext().getService(ref);
        //                return ds;
        //            }
        //        }

        // throw new RuntimeException("No DataSource with name " + name + " found");
        return null;
    }

    @SuppressWarnings("unchecked")
    public static ServiceReference[] getDataSources() {
        try {
            BundleContext context = MOsgi.getBundleContext();
            ServiceReference[] dsRefs =
                    context.getServiceReferences(DataSource.class.getName(), null);
            if (dsRefs == null) {
                dsRefs = new ServiceReference[] {};
            }
            return (ServiceReference[]) dsRefs;
        } catch (InvalidSyntaxException e) {
            throw new RuntimeException(e);
        }
    }

    public static String getServiceJndiName(ServiceReference ref) {
        String jndiName = (String) ref.getProperty(SERVICE_JNDI_NAME_KEY);
        return jndiName;
    }

    public static void registerDataSource(DataSource dataSource, String name) {
        BundleContext context = MOsgi.getBundleContext();
        Dictionary properties = new Hashtable();
        properties.put(SERVICE_JNDI_NAME_KEY, name);
        properties.put(DataSourceFactory.JDBC_DATASOURCE_NAME, name);

        context.registerService(DataSource.class, dataSource, properties);
    }
}