1 /*** 2 * Redistribution and use of this software and associated documentation 3 * ("Software"), with or without modification, are permitted provided 4 * that the following conditions are met: 5 * 6 * 1. Redistributions of source code must retain copyright 7 * statements and notices. Redistributions must also contain a 8 * copy of this document. 9 * 10 * 2. Redistributions in binary form must reproduce the 11 * above copyright notice, this list of conditions and the 12 * following disclaimer in the documentation and/or other 13 * materials provided with the distribution. 14 * 15 * 3. The name "Exolab" must not be used to endorse or promote 16 * products derived from this Software without prior written 17 * permission of Exoffice Technologies. For written permission, 18 * please contact info@exolab.org. 19 * 20 * 4. Products derived from this Software may not be called "Exolab" 21 * nor may "Exolab" appear in their names without prior written 22 * permission of Exoffice Technologies. Exolab is a registered 23 * trademark of Exoffice Technologies. 24 * 25 * 5. Due credit should be given to the Exolab Project 26 * (http://www.exolab.org/). 27 * 28 * THIS SOFTWARE IS PROVIDED BY EXOFFICE TECHNOLOGIES AND CONTRIBUTORS 29 * ``AS IS'' AND ANY EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT 30 * NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND 31 * FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL 32 * EXOFFICE TECHNOLOGIES OR ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, 33 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 34 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR 35 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 36 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, 37 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 38 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED 39 * OF THE POSSIBILITY OF SUCH DAMAGE. 40 * 41 * Copyright 2005 (C) Exoffice Technologies Inc. All Rights Reserved. 42 */ 43 package org.exolab.jms.tools.migration.proxy; 44 45 import java.sql.Connection; 46 import java.sql.PreparedStatement; 47 import java.sql.ResultSet; 48 import java.sql.SQLException; 49 import java.util.ArrayList; 50 import java.util.List; 51 import javax.jms.JMSException; 52 53 import org.exolab.jms.authentication.User; 54 import org.exolab.jms.persistence.PersistenceException; 55 import org.exolab.jms.persistence.SQLHelper; 56 import org.exolab.jms.tools.migration.IteratorAdapter; 57 import org.exolab.jms.tools.migration.Store; 58 import org.exolab.jms.tools.migration.StoreIterator; 59 60 61 /*** 62 * Provides persistency for {@link User} instances. 63 * 64 * @author <a href="mailto:tma@netspace.net.au">Tim Anderson</a> 65 * @version $Revision: 1.1 $ $Date: 2005/10/20 14:07:03 $ 66 */ 67 public class UserStore implements Store, DBConstants { 68 69 /*** 70 * The database connection. 71 */ 72 private final Connection _connection; 73 74 75 /*** 76 * Construct a new <code>UserStore</code>. 77 * 78 * @param connection the database connection 79 */ 80 public UserStore(Connection connection) { 81 _connection = connection; 82 } 83 84 /*** 85 * Export the users. 86 * 87 * @return an iterator over the collection 88 * @throws JMSException for any JMS error 89 * @throws PersistenceException for any persistence error 90 */ 91 public StoreIterator exportCollection() throws JMSException, 92 PersistenceException { 93 List users = getUsers(); 94 return new IteratorAdapter(users.iterator()); 95 } 96 97 /*** 98 * Import users into the store. 99 * 100 * @param iterator an iterator over the collection 101 * @throws JMSException for any JMS error 102 * @throws PersistenceException for any persistence error 103 */ 104 public void importCollection(StoreIterator iterator) throws JMSException, 105 PersistenceException { 106 while (iterator.hasNext()) { 107 User user = (User) iterator.next(); 108 add(user); 109 } 110 } 111 112 /*** 113 * Returns the number of elements in the collection. 114 * 115 * @return the number of elements in the collection 116 * @throws PersistenceException for any persistence error 117 */ 118 public int size() throws PersistenceException { 119 return getUsers().size(); 120 } 121 122 /*** 123 * Add a new user. 124 * 125 * @param user the user to add 126 * @throws PersistenceException for any persistence error 127 */ 128 private void add(User user) throws PersistenceException { 129 PreparedStatement insert = null; 130 try { 131 insert = _connection.prepareStatement( 132 "insert into " + USER_TABLE + " values (?, ?)"); 133 134 insert.setString(1, user.getUsername()); 135 insert.setString(2, user.getPassword()); 136 insert.executeUpdate(); 137 } catch (SQLException exception) { 138 throw new PersistenceException("Failed to add consumer", 139 exception); 140 } finally { 141 SQLHelper.close(insert); 142 } 143 } 144 145 /*** 146 * Returns the users. 147 * 148 * @return a list of {@link User} instances 149 * @throws PersistenceException for any persistence error 150 */ 151 private List getUsers() throws PersistenceException { 152 ArrayList result = new ArrayList(); 153 154 PreparedStatement select = null; 155 ResultSet set = null; 156 try { 157 select = _connection.prepareStatement( 158 "select * from " + USER_TABLE); 159 160 set = select.executeQuery(); 161 while (set.next()) { 162 String user = set.getString("username"); 163 String password = set.getString("password"); 164 result.add(new User(user, password)); 165 } 166 } catch (SQLException exception) { 167 throw new PersistenceException("Failed to retrieve users", 168 exception); 169 } finally { 170 SQLHelper.close(set); 171 SQLHelper.close(select); 172 } 173 return result; 174 } 175 176 } 177 178 179