#403 Add date/time handling to HibernateDomainExporter

querydsl · May 10, 2013 · 831fa07 · 831fa07
1 parent 34f09f9
commit 831fa07
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/querydsl-jpa-codegen/src/main/java/com/mysema/query/jpa/codegen/HibernateDomainExporter.java b/querydsl-jpa-codegen/src/main/java/com/mysema/query/jpa/codegen/HibernateDomainExporter.java
@@ -365,6 +365,20 @@ private void handleProperty(EntityType entityType, Class<?> cl, org.hibernate.ma
             // ignore
         }
         Type propertyType = getType(cl, clazz, p.getName());
+        switch (propertyType.getCategory()) {
+        case DATE:            
+        case TIME:
+        case DATETIME:
+            String type = p.getType().getName();
+            if ("time".equals(type)) {
+                propertyType = propertyType.as(TypeCategory.TIME);
+            } else if ("date".equals(type)) {
+                propertyType = propertyType.as(TypeCategory.DATE);
+            } else if ("timestamp".equals(type)) {
+                propertyType = propertyType.as(TypeCategory.DATETIME);
+            }
+        }
+
         if (p.isComposite()) {
             EntityType embeddedType = createEmbeddableType(propertyType);
             Iterator<?> properties = ((Component)p.getValue()).getPropertyIterator();