summaryrefslogtreecommitdiff
path: root/orkbasej/java/net/sf/oreka/HibernateManager.java
blob: 6fa5b7fddf23d9aaa2b6e438212cb39d318acd1c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
/*
 * Oreka -- A media capture and retrieval platform
 * 
 * Copyright (C) 2005, orecx LLC
 *
 * http://www.orecx.com
 *
 * This program is free software, distributed under the terms of
 * the GNU General Public License.
 * Please refer to http://www.gnu.org/copyleft/gpl.html
 *
 */

/**
 * 
 */
package net.sf.oreka;

import java.io.File;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Properties;

import net.sf.oreka.persistent.Domain;
import net.sf.oreka.persistent.LoginString;
import net.sf.oreka.persistent.RecPort;
import net.sf.oreka.persistent.RecPortFace;
import net.sf.oreka.persistent.RecProgram;
import net.sf.oreka.persistent.RecSegment;
import net.sf.oreka.persistent.RecSession;
import net.sf.oreka.persistent.RecTape;
import net.sf.oreka.persistent.Service;
import net.sf.oreka.persistent.User;

import org.apache.log4j.Logger;
import org.hibernate.HibernateException;
import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.hibernate.Transaction;
import org.hibernate.cfg.AnnotationConfiguration;
import org.logicalcobwebs.proxool.ProxoolFacade;


public class HibernateManager {
	
	private SessionFactory sessionFactory = null;
	static Logger logger = Logger.getLogger(HibernateManager.class);
	
	public void configure(String filename) throws Exception {
		
		File configFile = new File(filename);
		
		AnnotationConfiguration config = new AnnotationConfiguration();
		config.configure(configFile);
		
		// Configure the proxool connection pool
		Class.forName("org.logicalcobwebs.proxool.ProxoolDriver");
		Properties info = new Properties();
		info.setProperty("proxool.maximum-connection-count", "10");
		info.setProperty("proxool.house-keeping-test-sql", "select CURRENT_DATE");
		info.setProperty("user", config.getProperty("hibernate.connection.username"));
		info.setProperty("password", config.getProperty("hibernate.connection.password"));
		SimpleDateFormat sdf = new SimpleDateFormat("HHmmss");
		// Each time a pool is configured, it will have a different alias.
		// This is so that we don't get a "pool already registered" error.
		String alias = "oreka" + sdf.format(new Date());
		String driverClass = config.getProperty("hibernate.connection.driver_class");
		String driverUrl = config.getProperty("hibernate.connection.url");
		String url = "proxool." + alias + ":" + driverClass + ":" + driverUrl;
		ProxoolFacade.registerConnectionPool(url, info);
		
		// Let hibernate know we want to use proxool
		config.setProperty("hibernate.proxool.pool_alias", alias);
		config.setProperty("hibernate.proxool.existing_pool", "true");
		
		config.addAnnotatedClass(RecProgram.class);
		config.addAnnotatedClass(RecSession.class);
		config.addAnnotatedClass(RecSegment.class);
		config.addAnnotatedClass(RecTape.class);
		config.addAnnotatedClass(User.class);
		config.addAnnotatedClass(LoginString.class);
		config.addAnnotatedClass(Domain.class);		
		config.addAnnotatedClass(Service.class);
		config.addAnnotatedClass(RecPort.class);
		config.addAnnotatedClass(RecPortFace.class);
		sessionFactory = config.buildSessionFactory();
		
		// Add admin user if does not exist yet
		Session hbnSession = null;
		Transaction tx = null;
		try
		{
			hbnSession = getSession();
			tx = hbnSession.beginTransaction();
			User admin = (User)hbnSession.get(User.class, 1);
			if(admin == null) {
				admin = new User();
				admin.setPassword("admin");
				LoginString ls = new LoginString();
				ls.setUser(admin);
				ls.setLoginString("admin");
				hbnSession.save(admin);
				hbnSession.save(ls);				
			}
			tx.commit();
		}
		catch ( HibernateException he ) {
			logger.error("Hibernate error:" + he.getClass().getName());
		}
		catch (Exception e)
		{
			logger.error("Exception:", e);
		}
		finally {
			if(hbnSession != null) {hbnSession.close();}
		}
	}
	
	public Session getSession() throws Exception {
		if (sessionFactory == null) {
			throw new OrkException("HibernateManager.getSession: application must configure hibernate before using it.");
		}
		return sessionFactory.openSession();
	}
}