UserRepositoryIbatis.java

/*******************************************************************************
 * This file is part of OpenNMS(R).
 *
 * Copyright (C) 2009-2011 The OpenNMS Group, Inc.
 * OpenNMS(R) is Copyright (C) 1999-2011 The OpenNMS Group, Inc.
 *
 * OpenNMS(R) is a registered trademark of The OpenNMS Group, Inc.
 *
 * OpenNMS(R) is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published
 * by the Free Software Foundation, either version 3 of the License,
 * or (at your option) any later version.
 *
 * OpenNMS(R) 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with OpenNMS(R).  If not, see:
 *      http://www.gnu.org/licenses/
 *
 * For more information contact:
 *     OpenNMS(R) Licensing <license@opennms.org>
 *     http://www.opennms.org/
 *     http://www.opennms.com/
 *
 * From the original copyright headers:
 *
 * Copyright (c) 2009+ desmax74
 * Copyright (c) 2009+ The OpenNMS Group, Inc.
 *
 * This program was developed and is maintained by Rocco RIONERO
 * ("the author") and is subject to dual-copyright according to
 * the terms set in "The OpenNMS Project Contributor Agreement".
 *
 * The author can be contacted at the following email address:
 *
 *     Massimiliano Dessì
 *     desmax74@yahoo.it
 *******************************************************************************/

package org.opennms.acl.repository.ibatis;

import java.sql.SQLException;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import org.opennms.acl.model.Pager;
import org.opennms.acl.model.UserAuthoritiesDTO;
import org.opennms.acl.model.UserDTO;
import org.opennms.acl.model.UserDTOLight;
import org.opennms.acl.model.UserView;
import org.opennms.acl.repository.UserRepository;
import org.opennms.acl.util.Cripto;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.orm.ibatis.SqlMapClientCallback;
import org.springframework.orm.ibatis.SqlMapClientTemplate;
import org.springframework.stereotype.Repository;

import com.ibatis.sqlmap.client.SqlMapClient;
import com.ibatis.sqlmap.client.SqlMapExecutor;

/**
 * <p>UserRepositoryIbatis class.</p>
 *
 * @author Massimiliano Dessì (desmax74@yahoo.it)
 * @since jdk 1.5.0
 * @version $Id: $
 */
@Repository("userRepository")
public class UserRepositoryIbatis extends SqlMapClientTemplate implements UserRepository {

    /** {@inheritDoc} */
    public UserDTO getUserCredentials(String id) {
        return (UserDTO) queryForObject("getUserCredentials", new Long(id));
    }

    /** {@inheritDoc} */
    @Autowired
    @Override
    public void setSqlMapClient(@Qualifier("sqlMapClient") SqlMapClient sqlMapClient) {
        super.setSqlMapClient(sqlMapClient);
    }

    /** {@inheritDoc} */
    public Long insertUser(UserDTO user) {
        user.setPassword(Cripto.stringToSHA(user.getPassword()));
        return (Long) insert("insertUser", user);
    }

    /** {@inheritDoc} */
    public Integer updatePassword(UserDTO user) {
        return update("updateUserPassword", user);
    }

    /** {@inheritDoc} */
    public UserAuthoritiesDTO getUserWithAuthorities(String username) {
        return (UserAuthoritiesDTO) queryForObject("getUserWithAuthorities", username);
    }

    /** {@inheritDoc} */
    public UserAuthoritiesDTO getUserWithAuthoritiesByID(Integer sid) {
        return (UserAuthoritiesDTO) queryForObject("getUserWithAuthoritiesById", sid);
    }

    /** {@inheritDoc} */
    public UserView getUser(String id) {
        return (UserView) queryForObject("getUser", Integer.valueOf(id));
    }

    /**
     * <p>getUsersNumber</p>
     *
     * @return a {@link java.lang.Integer} object.
     */
    public Integer getUsersNumber() {
        return (Integer) queryForObject("getUsersNumber");
    }

    /** {@inheritDoc} */
    public Boolean disableUser(String id) {
        return update("disableUser", new Long(id)) == 1 ? true : false;
    }

    /** {@inheritDoc} */
    public Object getIdUser(String username) {
        return queryForObject("getIdUser", username);
    }

    /** {@inheritDoc} */
    @SuppressWarnings("unchecked")
    public List<UserDTOLight> getDisabledUsers(Pager pager) {
        Map params = new HashMap();
        params.put("limit", pager.getItemsNumberOnPage());
        params.put("offset", pager.getPage() * pager.getItemsNumberOnPage());
        return queryForList("getDisabledUsers", params);
    }

    /** {@inheritDoc} */
    @SuppressWarnings("unchecked")
    public List<UserDTOLight> getEnabledUsers(Pager pager) {
        Map params = new HashMap();
        params.put("limit", pager.getItemsNumberOnPage());
        params.put("offset", pager.getPage() * pager.getItemsNumberOnPage());
        return queryForList("getEnabledUsers", params);
    }

    /** {@inheritDoc} */
    public Boolean save(UserAuthoritiesDTO user) {
        return user.isNew() ? insert(user) : update(user);
    }

    private Boolean update(UserAuthoritiesDTO user) {
        boolean result = deleteUserItems(user.getUsername());
        if (user.getItems() != null && user.getItems().size() > 0) {
            result = saveItems(user.getUsername(), user.getItems());
        }
        return result;
    }

    private Boolean insert(UserAuthoritiesDTO user) {
        return insert("insertUser", user) != null;
    }

    private Boolean deleteUserItems(String username) {
        return delete("deleteUserItems", username) > 0;
    }

    private Boolean saveItems(final String username, final List<?> items) {

        return execute(new SqlMapClientCallback<Boolean>() {
            @SuppressWarnings("unchecked")
            public Boolean doInSqlMapClient(SqlMapExecutor executor) {
                int ris = 0;
                try {

                    executor.startBatch();
                    Iterator<Integer> iter = (Iterator<Integer>) items.iterator();

                    while (iter.hasNext()) {
                        Map params = new HashMap();
                        params.put("id", iter.next());
                        params.put("username", username);
                        executor.insert("insertUserItem", params);
                    }
                    ris = executor.executeBatch();
                } catch (SQLException e) {
                    Logger log = LoggerFactory.getLogger(this.getClass());
                    StringBuffer sb = new StringBuffer("saveItems failed \n").append("num groups batch:").append(items.size()).append("\n").append(" username:").append(username).append("\n").append(
                            e.getNextException());
                    log.error(sb.toString());
                }
                return ris > 0;
            }
        });
    }

}