From fd250025781159f5ca92534472bfcff9565ce5f0 Mon Sep 17 00:00:00 2001 From: magenoxx Date: Wed, 25 Jan 2012 14:26:37 +0400 Subject: [PATCH] SQLite EntityManager --- Mage.Common/src/log4j.properties | 8 ++ Mage.Common/src/mage/db/EntityManager.java | 153 +++++++++++++++++++-- 2 files changed, 146 insertions(+), 15 deletions(-) create mode 100644 Mage.Common/src/log4j.properties diff --git a/Mage.Common/src/log4j.properties b/Mage.Common/src/log4j.properties new file mode 100644 index 0000000000..c1cd7b3d3b --- /dev/null +++ b/Mage.Common/src/log4j.properties @@ -0,0 +1,8 @@ +#default levels +log4j.rootLogger=debug, console + +#console log +log4j.appender.console=org.apache.log4j.ConsoleAppender +log4j.appender.console.layout=org.apache.log4j.PatternLayout +log4j.appender.console.layout.ConversionPattern=%-5p [%d{yyyy-MM-dd HH:mm [ss:SSS]}] %C{1}[%t]: %m%n +log4j.appender.console.Threshold=debug diff --git a/Mage.Common/src/mage/db/EntityManager.java b/Mage.Common/src/mage/db/EntityManager.java index 567bdc6213..44c6016133 100644 --- a/Mage.Common/src/mage/db/EntityManager.java +++ b/Mage.Common/src/mage/db/EntityManager.java @@ -1,5 +1,8 @@ package mage.db; +import org.apache.log4j.Logger; + +import java.io.File; import java.sql.*; /** @@ -7,29 +10,149 @@ import java.sql.*; */ public class EntityManager { - public static void main(String[] args) throws Exception { + private static final Logger log = Logger.getLogger(EntityManager.class); + + private static final EntityManager instance; + + private static final String MAGE_JDBC_URL = "jdbc:sqlite:db/mage.db"; + + static { + instance = new EntityManager(); + try { + init(); + } catch (Exception e) { + log.fatal(e); + e.printStackTrace(); + } + } + + public static EntityManager getInstance() { + return instance; + } + + /** + * Inits database. Creates tables if they don't exist. + * + * @throws Exception + */ + protected static void init() throws Exception { Class.forName("org.sqlite.JDBC"); - Connection conn = DriverManager.getConnection("jdbc:sqlite:mage.db"); - Statement stat = conn.createStatement(); - stat.executeUpdate("drop table if exists users;"); - stat.executeUpdate("create table users (login, password);"); + checkDBFolderExistance(); + Connection conn = DriverManager.getConnection(MAGE_JDBC_URL); + try { + Statement stat = conn.createStatement(); + stat.executeUpdate("create table if not exists users (login, password, status);"); + } finally { + try { + conn.close(); + } catch (Exception e) { + // swallow + } + } + } - PreparedStatement prep = conn.prepareStatement("insert into users values (?, ?);"); + /** + * Reinits database. Drops all tables and then creates them from scratch. + * BE CAREFUL! THIS METHOD WILL DESTROY ALL DATA. + * + * @throws Exception + */ + public static void reinit() throws Exception { + Class.forName("org.sqlite.JDBC"); + checkDBFolderExistance(); + Connection conn = DriverManager.getConnection(MAGE_JDBC_URL); + try { + Statement stat = conn.createStatement(); + stat.executeUpdate("drop table users;"); + stat.executeUpdate("create table users (login, password, status);"); + } finally { + try { + conn.close(); + } catch (Exception e) { + // swallow + } + } + } - prep.setString(1, "TestUser"); - prep.setString(2, "123"); - prep.execute(); + /** + * Creates folders for Mage sqlite db. + */ + protected static void checkDBFolderExistance() { + File file = new File("db"); + if (!file.exists()) { + file.mkdirs(); + } + } - prep.setString(1, "TestUser2"); - prep.setString(2, "12345"); - prep.execute(); + /** + * Tests DB with base operations. + * + * @throws Exception + */ + public void testDB() throws Exception { + + Connection conn = DriverManager.getConnection(MAGE_JDBC_URL); + + try { + Statement stat = conn.createStatement(); + + ResultSet rs = stat.executeQuery("select * from users where login = 'testtest';"); + + if (rs.next()) { + checkTestUser(conn, stat); + } else { + log.debug("[DBTest] creating test user..."); + PreparedStatement prep = conn.prepareStatement("insert into users values (?, ?, ?);"); + + prep.setString(1, "testtest"); + prep.setString(2, "12345"); + prep.setString(3, "disabled"); + + prep.execute(); + + log.debug("[DBTest] creating test user [OK]"); + + checkTestUser(conn, stat); + } + + rs.close(); + } finally { + try { + conn.close(); + } catch (Exception e) { + // swallow + } + } + } + + /** + * Checks test user existence and its parameters. + * + * @param conn + * @param stat + * @throws Exception + */ + private void checkTestUser(Connection conn, Statement stat) throws Exception { + ResultSet rs = stat.executeQuery("select * from users where login = 'testtest';"); + if (rs.next()) { + log.debug("[DBTest] checking test user [OK]"); + if (rs.getString("login").equals("testtest") && rs.getString("password").equals("12345") && rs.getString("status").equals("disabled")) { + log.debug("[DBTest] checking test user parameters [OK]"); + } else { + log.debug("[DBTest] checking test user parameters [ERROR]"); + } + } else { + log.debug("[DBTest] checking test user [ERROR]"); + log.debug("[DBTest] couldn't find test user"); + } - ResultSet rs = stat.executeQuery("select * from users;"); while (rs.next()) { System.out.println("user = " + rs.getString("login")); System.out.println("password = " + rs.getString("password")); } - rs.close(); - conn.close(); + } + + public static void main(String[] args) throws Exception { + EntityManager.getInstance().testDB(); } }