From 26b01b500065634eb3133dc354a0ba71b13bff56 Mon Sep 17 00:00:00 2001 From: Trygve Laugstøl Date: Wed, 7 Aug 2013 23:53:53 +0200 Subject: wip o Start of JPA implementation. --- .../io/trygvis/container/myapp/AddressBookJpa.java | 242 +++++++++++++++++++++ 1 file changed, 242 insertions(+) create mode 100644 myapp/src/main/java/io/trygvis/container/myapp/AddressBookJpa.java (limited to 'myapp/src/main/java/io/trygvis/container/myapp/AddressBookJpa.java') diff --git a/myapp/src/main/java/io/trygvis/container/myapp/AddressBookJpa.java b/myapp/src/main/java/io/trygvis/container/myapp/AddressBookJpa.java new file mode 100644 index 0000000..3981805 --- /dev/null +++ b/myapp/src/main/java/io/trygvis/container/myapp/AddressBookJpa.java @@ -0,0 +1,242 @@ +package io.trygvis.container.myapp; + +import io.trygvis.persistence.sql.SqlEntityMeta; +import io.trygvis.persistence.sql.SqlUnit; + +import javax.persistence.EntityManager; +import javax.persistence.EntityManagerFactory; +import javax.persistence.EntityTransaction; +import javax.persistence.TypedQuery; +import javax.sql.DataSource; +import java.io.BufferedReader; +import java.io.EOFException; +import java.io.InputStreamReader; +import java.sql.SQLException; +import java.util.List; + +import static io.trygvis.container.myapp.Contact.Gender.FEMALE; +import static io.trygvis.container.myapp.Contact.Gender.MALE; + +public class AddressBookJpa { + + private BufferedReader reader = new BufferedReader(new InputStreamReader(System.in)); + + private EntityManager entityManager; + + public static void main(String[] args) throws Exception { + try { + new AddressBookJpa().work(); + } catch (EOFException ignore) { + } + } + + private void work() throws Exception { + DataSource ds = new DriverManagerDataSource("jdbc:h2:mem:address-book;DB_CLOSE_DELAY=-1"); + EntityManagerFactory emf = new MyAppEntityManagerFactory(ds); + boolean done = false; + while (!done) { + try { + entityManager = emf.createEntityManager(); + EntityTransaction transaction = entityManager.getTransaction(); + done = main(); + System.out.println("OK"); + transaction.commit(); + } finally { + entityManager.close(); + } + } + } + + private boolean main() throws Exception { + System.out.println("Menu:"); + System.out.println("c Create"); + System.out.println("d Drop"); + System.out.println("m Company menu"); + System.out.println("n Contact menu"); + System.out.println("q Quit"); + String cmd = cmd(); + switch (cmd) { + case "c": + create(); + break; + case "d": + drop(); + break; + case "m": + company(); + break; + case "n": + contact(); + break; + case "q": + return true; + default: + System.out.println("Unknown command: " + cmd); + } + return false; + } + + private String cmd() throws Exception { + String cmd = null; + do { + String read = line(); + if (read.length() != 0) { + cmd = read.trim(); + } + } while (cmd == null); + return cmd; + } + + private String line() throws Exception { + String line = reader.readLine(); + if (line == null) { + throw new EOFException(); + } + return line.trim(); + } + + public void create() throws SQLException { + for (SqlEntityMeta entityMeta : entityManager.unwrap(SqlUnit.class).getEntities()) { + entityManager.createNativeQuery(entityMeta.createTableSql).executeUpdate(); + } + for (String sql : Sequences.createSequences) { + entityManager.createNativeQuery(sql).executeUpdate(); + } + } + + public void drop() throws SQLException { + for (String sql : Sequences.dropSequences) { + entityManager.createNativeQuery(sql); + } + for (SqlEntityMeta entityMeta : entityManager.unwrap(SqlUnit.class).getEntities()) { + entityManager.createNativeQuery(entityMeta.dropTableSql).executeUpdate(); + } + } + + // ----------------------------------------------------------------------- + // Company + // ----------------------------------------------------------------------- + + private void company() throws Exception { + while (true) { + System.out.println("Company menu:"); + System.out.println("c Create"); + System.out.println("d Drop"); + System.out.println("d List"); + System.out.println("q Back"); + String cmd = cmd(); + switch (cmd) { + case "c": + addCompany(); + break; + case "d": + deleteCompany(); + break; + case "l": + listCompanies(); + break; + case "q": + return; + default: + System.out.println("Unknown command: " + cmd); + } + } + } + + public void addCompany() throws Exception { + System.out.print("Name: "); + String name = line(); + + Company company = new Company(name); + entityManager.persist(company); + } + + public void deleteCompany() { + } + + public void listCompanies() { + TypedQuery p = entityManager.createQuery("", Company.class); + + List resultList = p.getResultList(); + for (Company company : resultList) { + System.out.println("====================="); + System.out.println("Id: " + company.getId()); + System.out.println("Name: " + company.name); + } + System.out.println(); + } + + // ----------------------------------------------------------------------- + // Contact + // ----------------------------------------------------------------------- + + private void contact() throws Exception { + while (true) { + System.out.println("Contact menu:"); + System.out.println("c Create"); + System.out.println("d Drop"); + System.out.println("q Back"); + String cmd = cmd(); + switch (cmd) { + case "c": + addContact(); + break; + case "d": + deleteContact(); + break; + case "l": + listContacts(); + break; + case "q": + return; + default: + System.out.println("Unknown command: " + cmd); + } + } + } + + public void addContact() throws Exception { + System.out.print("Name: "); + String name = line(); + Contact.Gender g = null; + while (g == null) { + System.out.print("Gender (m/f): "); + try { + String line = line(); + switch (line) { + case "m": + g = MALE; + break; + case "f": + g = FEMALE; + break; + default: + g = Contact.Gender.valueOf(line); + break; + } + } catch (IllegalArgumentException ignore) { + } + } + + Company company = null; + Contact o = new Contact(name, g, company); + entityManager.persist(company); + } + + public void deleteContact() { + + } + + public void listContacts() { + TypedQuery p = entityManager.createQuery("", Contact.class); + + List resultList = p.getResultList(); + for (Contact contact : resultList) { + System.out.println("====================="); + System.out.println("Id: " + contact.getId()); + System.out.println("Name: " + contact.name); + System.out.println("Gender: " + contact.gender); + } + System.out.println(); + } +} -- cgit v1.2.3