[44] | 1 | package omq.common.util; |
---|
| 2 | |
---|
| 3 | import java.io.IOException; |
---|
| 4 | import java.security.KeyManagementException; |
---|
| 5 | import java.security.NoSuchAlgorithmException; |
---|
| 6 | import java.util.Properties; |
---|
| 7 | |
---|
| 8 | import com.rabbitmq.client.Channel; |
---|
| 9 | import com.rabbitmq.client.Connection; |
---|
| 10 | import com.rabbitmq.client.ConnectionFactory; |
---|
| 11 | |
---|
| 12 | /** |
---|
| 13 | * |
---|
| 14 | * @author Sergi Toda <sergi.toda@estudiants.urv.cat> |
---|
| 15 | * |
---|
| 16 | */ |
---|
| 17 | public class OmqConnectionFactory { |
---|
| 18 | private static Connection connection; |
---|
| 19 | private static int connectionTimeout = 2 * 1000; |
---|
| 20 | |
---|
| 21 | public static void init(Properties env) throws KeyManagementException, NoSuchAlgorithmException, IOException { |
---|
| 22 | if (connection == null) { |
---|
| 23 | connection = getNewConnection(env); |
---|
| 24 | } |
---|
| 25 | } |
---|
| 26 | |
---|
| 27 | public static Connection getNewWorkingConnection(Properties env) throws Exception { |
---|
| 28 | Connection connection = null; |
---|
| 29 | boolean working = false; |
---|
| 30 | |
---|
| 31 | while (!working) { |
---|
| 32 | try { |
---|
| 33 | connection = getNewConnection(env); |
---|
| 34 | working = true; |
---|
| 35 | } catch (Exception e) { |
---|
| 36 | e.printStackTrace(); |
---|
| 37 | long milis = Long.parseLong(env.getProperty(ParameterQueue.RETRY_TIME_CONNECTION, "2000")); |
---|
| 38 | Thread.sleep(milis); |
---|
| 39 | } |
---|
| 40 | } |
---|
| 41 | |
---|
| 42 | return connection; |
---|
| 43 | } |
---|
| 44 | |
---|
| 45 | public static Connection getNewConnection(Properties env) throws IOException, KeyManagementException, NoSuchAlgorithmException { |
---|
| 46 | // Get login info of rabbitmq |
---|
| 47 | String username = env.getProperty(ParameterQueue.USER_NAME); |
---|
| 48 | String password = env.getProperty(ParameterQueue.USER_PASS); |
---|
| 49 | |
---|
| 50 | // Get host info of rabbimq (where it is) |
---|
| 51 | String host = env.getProperty(ParameterQueue.SERVER_HOST); |
---|
| 52 | int port = Integer.parseInt(env.getProperty(ParameterQueue.SERVER_PORT)); |
---|
| 53 | |
---|
| 54 | boolean ssl = Boolean.parseBoolean(env.getProperty(ParameterQueue.ENABLE_SSL)); |
---|
| 55 | |
---|
| 56 | // Start a new connection and channel |
---|
| 57 | ConnectionFactory factory = new ConnectionFactory(); |
---|
| 58 | factory.setUsername(username); |
---|
| 59 | factory.setPassword(password); |
---|
| 60 | factory.setHost(host); |
---|
| 61 | factory.setPort(port); |
---|
| 62 | factory.setConnectionTimeout(connectionTimeout); |
---|
| 63 | if (ssl) { |
---|
| 64 | factory.useSslProtocol(); |
---|
| 65 | } |
---|
| 66 | return factory.newConnection(); |
---|
| 67 | } |
---|
| 68 | |
---|
| 69 | public static Channel getNewChannel() throws IOException { |
---|
| 70 | return connection.createChannel(); |
---|
| 71 | } |
---|
| 72 | } |
---|