hibernate: Setup initial schema and db connection

Author: nvb

java/hibernate/src/main/java/com/cockroachlabs/Application.java

@@ -1,9 +1,48 @@
 package com.cockroachlabs;
 
+import com.cockroachlabs.model.Customer;
+import com.cockroachlabs.model.Order;
+import com.cockroachlabs.model.Product;
+import org.hibernate.Session;
+import org.hibernate.SessionFactory;
+import org.hibernate.boot.registry.StandardServiceRegistryBuilder;
+import org.hibernate.cfg.Configuration;
+import org.hibernate.service.ServiceRegistry;
+
+import java.math.BigDecimal;
+
 public class Application {
 
     public static void main(String[] args) {
+        try (SessionFactory sf = buildSessionFactory()) {
+            Session session = sf.getCurrentSession();
+
+            Customer c = new Customer();
+            c.setName("joe");
+
+            Order o = new Order();
+            o.setCustomer(c);
+            o.setSubtotal(new BigDecimal(100));
+
+            session.beginTransaction();
+            session.save(c);
+            session.save(o);
+            session.getTransaction().commit();
+        }
+    }
+
+    private static SessionFactory buildSessionFactory() {
+        Configuration configuration = new Configuration();
+        configuration.configure("hibernate.cfg.xml");
+        configuration.addAnnotatedClass(Customer.class);
+        configuration.addAnnotatedClass(Order.class);
+        configuration.addAnnotatedClass(Product.class);
+
+        ServiceRegistry serviceRegistry = new StandardServiceRegistryBuilder()
+                .applySettings(configuration.getProperties())
+                .build();
 
+        return configuration.buildSessionFactory(serviceRegistry);
     }
 
 }

java/hibernate/src/main/java/com/cockroachlabs/dialect/CockroachDialect.java

@@ -0,0 +1,17 @@
+package com.cockroachlabs.dialect;
+
+import org.hibernate.dialect.PostgreSQL94Dialect;
+
+public class CockroachDialect extends PostgreSQL94Dialect {
+
+    /**
+     * Override needed to avoid referencing "pg_class".
+     * <p/>
+     * {@inheritDoc}
+     */
+    @Override
+    public String getQuerySequencesString() {
+        return null;
+    }
+
+}

java/hibernate/src/main/java/com/cockroachlabs/model/Customer.java

@@ -0,0 +1,33 @@
+package com.cockroachlabs.model;
+
+import javax.persistence.*;
+
+@Entity
+@Table(name="Customers")
+public class Customer {
+
+    @Id
+    @GeneratedValue(strategy=GenerationType.IDENTITY)
+    @Column(name="ID", nullable=false, unique=true)
+    private long id;
+
+    @Column(name="NAME")
+    private String name;
+
+    public long getId() {
+        return id;
+    }
+
+    public void setId(long id) {
+        this.id = id;
+    }
+
+    public String getName() {
+        return name;
+    }
+
+    public void setName(String name) {
+        this.name = name;
+    }
+
+}

java/hibernate/src/main/java/com/cockroachlabs/model/Order.java

@@ -0,0 +1,61 @@
+package com.cockroachlabs.model;
+
+import javax.persistence.*;
+import java.math.BigDecimal;
+import java.util.Set;
+
+@Entity
+@Table(name="Orders")
+public class Order {
+
+    @Id
+    @GeneratedValue(strategy=GenerationType.IDENTITY)
+    @Column(name="ID", nullable=false, unique=true)
+    private long id;
+
+    @Column(name="SUBTOTAL", precision=18, scale=2)
+    private BigDecimal subtotal;
+
+    @ManyToOne
+    @JoinColumn(name="CUSTOMER_ID")
+    private Customer customer;
+
+    @ManyToMany()
+    @JoinTable(name="PRODUCT_ORDERS",
+               joinColumns=@JoinColumn(name="ORDER_ID"),
+               inverseJoinColumns=@JoinColumn(name="PRODUCT_ID"))
+    private Set<Product> products;
+
+    public long getId() {
+        return id;
+    }
+
+    public void setId(long id) {
+        this.id = id;
+    }
+
+    public BigDecimal getSubtotal() {
+        return subtotal;
+    }
+
+    public void setSubtotal(BigDecimal subtotal) {
+        this.subtotal = subtotal;
+    }
+
+    public Customer getCustomer() {
+        return customer;
+    }
+
+    public void setCustomer(Customer customer) {
+        this.customer = customer;
+    }
+
+    public Set<Product> getProducts() {
+        return products;
+    }
+
+    public void setProducts(Set<Product> products) {
+        this.products = products;
+    }
+
+}

java/hibernate/src/main/java/com/cockroachlabs/model/Product.java

@@ -0,0 +1,57 @@
+package com.cockroachlabs.model;
+
+import javax.persistence.*;
+import java.math.BigDecimal;
+import java.util.Set;
+
+@Entity
+@Table(name="Products")
+public class Product {
+
+    @Id
+    @GeneratedValue(strategy=GenerationType.IDENTITY)
+    @Column(name="ID", nullable=false, unique=true)
+    private long id;
+
+    @Column(name="NAME")
+    private String name;
+
+    @Column(name="PRODUCT", precision=18, scale=2)
+    private BigDecimal product;
+
+    @ManyToMany(cascade=CascadeType.ALL, mappedBy="products")
+    private Set<Order> orders;
+
+    public long getId() {
+        return id;
+    }
+
+    public void setId(long id) {
+        this.id = id;
+    }
+
+    public String getName() {
+        return name;
+    }
+
+    public void setName(String name) {
+        this.name = name;
+    }
+
+    public BigDecimal getProduct() {
+        return product;
+    }
+
+    public void setProduct(BigDecimal product) {
+        this.product = product;
+    }
+
+    public Set<Order> getOrders() {
+        return orders;
+    }
+
+    public void setOrders(Set<Order> orders) {
+        this.orders = orders;
+    }
+
+}

java/hibernate/src/main/resources/hibernate.cfg.xml

@@ -17,13 +17,15 @@
         <property name="connection.pool_size">1</property>
 
         <!-- SQL dialect -->
-        <property name="dialect">org.hibernate.dialect.PostgreSQL94Dialect</property>
+        <property name="dialect">com.cockroachlabs.dialect.CockroachDialect</property>
+        <!--<property name="dialect">org.hibernate.dialect.PostgreSQL94Dialect</property>-->
 
         <!-- Echo all executed SQL to stdout -->
         <property name="show_sql">true</property>
 
         <!-- Drop and re-create the database schema on startup -->
-        <property name="hbm2ddl.auto">validate</property>
+        <!--<property name="hbm2ddl.auto">validate</property>-->
+        <property name="hbm2ddl.auto">update</property>
 
         <!-- org.hibernate.HibernateException: No CurrentSessionContext configured! -->
         <property name="hibernate.current_session_context_class">thread</property>