1 files changed, 44 insertions, 0 deletions
diff --git a/container-compiler-plugin/src/main/java/io/trygvis/persistence/TypeHandler.java b/container-compiler-plugin/src/main/java/io/trygvis/persistence/TypeHandler.java
index dfb2be3..f13018a 100644
--- a/container-compiler-plugin/src/main/java/io/trygvis/persistence/TypeHandler.java
+++ b/container-compiler-plugin/src/main/java/io/trygvis/persistence/TypeHandler.java
@@ -1,6 +1,7 @@
 package io.trygvis.persistence;
 
 import io.trygvis.container.compiler.InternalErrorException;
+import io.trygvis.container.compiler.model.TypeRef;
 
 import java.lang.reflect.Field;
 import java.sql.Types;
@@ -69,6 +70,49 @@ public abstract class TypeHandler {
         }
     }
 
+//    public static class OrdinalEnumTypeHandler extends TypeHandler {
+//        private final TypeRef enumType;
+//        public OrdinalEnumTypeHandler(TypeRef enumType) {
+//            super("INTEGER", INTEGER, true);
+//            this.enumType = enumType;
+//        }
+//
+//        @Override
+//        String setter(int i, String expr) {
+//            return "setInt(" + i + ", " + expr + ".ordinal())";
+//        }
+//
+//        @Override
+//        String getter(String rs, int i) {
+//            return "int ordinal = -1; for (Gender v : Gender.values()) if (v.ordinal() == ordinal) return v;";
+//            int ordinal = -1; for (EnumType v : EnumType.values()) if (v.ordinal() == ordinal) return v;
+//            return enumType.plainName + ".ordinal(" + rs + ".getInt(" + i + "))";
+//        }
+//    }
+
+    public static class StringEnumTypeHandler extends TypeHandler {
+        private final TypeRef enumType;
+
+        public StringEnumTypeHandler(TypeRef enumType) {
+            super("VARCHAR", VARCHAR, true);
+            this.enumType = enumType;
+        }
+
+        @Override
+        String setter(int i, String expr) {
+            return "setString(" + i + ", " + expr + ".name())";
+        }
+
+        @Override
+        String getter(String rs, int i) {
+            return enumType.plainName + ".valueOf(" + rs + ".getString(" + i + "))";
+        }
+    }
+
+    // -----------------------------------------------------------------------
+    //
+    // -----------------------------------------------------------------------
+
     public static class StringTypeHandler extends PrimitiveTypeHandler {
         protected StringTypeHandler() {
             super(null, VARCHAR, "String", true);