com.avaje.ebeaninternal.server.ldap.DefaultLdapPersister Maven / Gradle / Ivy

Go to download
/**
 * Copyright (C) 2009 Authors
 * 
 * This file is part of Ebean.
 * 
 * Ebean 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.
 *  
 * Ebean 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 Ebean; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA  
 */
package com.avaje.ebeaninternal.server.ldap;

import java.util.Iterator;
import java.util.Set;
import java.util.logging.Level;
import java.util.logging.Logger;

import javax.naming.Name;
import javax.naming.NamingException;
import javax.naming.directory.Attribute;
import javax.naming.directory.Attributes;
import javax.naming.directory.BasicAttributes;
import javax.naming.directory.DirContext;

import com.avaje.ebean.config.ldap.LdapContextFactory;
import com.avaje.ebeaninternal.server.deploy.BeanDescriptor;
import com.avaje.ebeaninternal.server.deploy.BeanProperty;

public class DefaultLdapPersister {

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

    private final LdapContextFactory contextFactory;

    public DefaultLdapPersister(LdapContextFactory dirContextFactory) {
        this.contextFactory = dirContextFactory;
    }

    public int persist(LdapPersistBeanRequest request) {

        switch (request.getType()) {
        case INSERT:
            return insert(request);
        case UPDATE:
            return update(request);
        case DELETE:
            return delete(request);

        default:
            throw new LdapPersistenceException("Invalid type " + request.getType());
        }
    }

    private int insert(LdapPersistBeanRequest request) {

        DirContext dc = contextFactory.createContext();

        Name name = request.createLdapName();
        Attributes attrs = createAttributes(request, false, request.getLoadedProperties());

        if (logger.isLoggable(Level.FINE)) {
            logger.fine("Ldap Insert Name:" + name + " Attrs:" + attrs);
        }
        try {
            dc.bind(name, null, attrs);
            return 1;

        } catch (NamingException e) {
            throw new LdapPersistenceException(e);
        }
    }

    private int delete(LdapPersistBeanRequest request) {

        DirContext dc = contextFactory.createContext();
        Name name = request.createLdapName();

        if (logger.isLoggable(Level.FINE)) {
            logger.fine("Ldap Delete Name:" + name);
        }

        try {
            dc.unbind(name);
            return 1;

        } catch (NamingException e) {
            throw new LdapPersistenceException(e);
        }
    }

    private int update(LdapPersistBeanRequest request) {

        Name name = request.createLdapName();

        Set updatedProperties = request.getUpdatedProperties();
        if (updatedProperties == null || updatedProperties.isEmpty()) {
            logger.info("Ldap Update has no changed properties?  Name:" + name);
            return 0;
        }

        DirContext dc = contextFactory.createContext();
        Attributes attrs = createAttributes(request, true, updatedProperties);

        if (logger.isLoggable(Level.FINE)) {
            logger.fine("Ldap Update Name:" + name + " Attrs:" + attrs);
        }

        try {
            dc.modifyAttributes(name, DirContext.REPLACE_ATTRIBUTE, attrs);
            return 1;

        } catch (NamingException e) {
            throw new LdapPersistenceException(e);
        }
    }

    private Attributes createAttributes(LdapPersistBeanRequest request, boolean update, Set props) {

        BeanDescriptor desc = request.getBeanDescriptor();

        Attributes attrs = desc.createAttributes();
        if (update) {
            attrs = new BasicAttributes(true);
        } else {
            attrs = desc.createAttributes();
        }

        Object bean = request.getBean();

        if (props != null) {
            for (String propName : props) {
                BeanProperty p = desc.getBeanPropertyFromPath(propName);
                Attribute attr = p.createAttribute(bean);
                if (attr != null) {
                    attrs.put(attr);
                }
            }
        } else {
            Iterator it = desc.propertiesAll();
            while (it.hasNext()) {
                BeanProperty p = it.next();
                Attribute attr = p.createAttribute(bean);
                if (attr != null) {
                    attrs.put(attr);
                }
            }
        }

        return attrs;
    }
}