feat: mysql support

Author: tandemdude

internal/codegen/enums.go

@@ -0,0 +1,62 @@
+package codegen
+
+import (
+	"fmt"
+	"github.com/iancoleman/strcase"
+	"github.com/tandemdude/sqlc-gen-java/internal/core"
+	"regexp"
+	"strings"
+	"unicode"
+	"unicode/utf8"
+)
+
+var javaInvalidIdentChars = regexp.MustCompile("[^$\\w]")
+
+func EnumClassName(qualifiedName, defaultSchema string) string {
+	return strcase.ToCamel(strings.TrimPrefix(qualifiedName, defaultSchema+"."))
+}
+
+func enumValueName(value string) string {
+	rep := strings.NewReplacer("-", "_", ":", "_", "/", "_", ".", "_")
+	name := rep.Replace(value)
+	name = strings.ToUpper(name)
+	name = javaInvalidIdentChars.ReplaceAllString(name, "")
+
+	r, _ := utf8.DecodeRuneInString(name)
+	if unicode.IsDigit(r) {
+		name = "_" + name
+	}
+	return name
+}
+
+func BuildEnumFile(engine string, conf core.Config, qualName string, enum core.Enum, defaultSchema string) (string, []byte, error) {
+	className := EnumClassName(qualName, defaultSchema)
+
+	sb := IndentStringBuilder{indentChar: conf.IndentChar, charsPerIndentLevel: conf.CharsPerIndentLevel}
+	sb.writeSqlcHeader()
+	sb.WriteString("\n")
+	sb.WriteString("package " + conf.Package + ".enums;\n")
+	sb.WriteString("\n")
+	sb.WriteString("public enum " + className + " {\n")
+
+	if engine == "mysql" {
+		sb.WriteIndentedString(1, "BLANK(\"\"),\n")
+	}
+
+	// write other values
+	for i, value := range enum.Values {
+		name := enumValueName(value)
+		sb.WriteIndentedString(1, fmt.Sprintf("%s(\"%s\")", name, value))
+
+		if i < len(enum.Values)-1 {
+			sb.WriteString(",\n")
+		}
+	}
+	sb.WriteString(";\n\n")
+	sb.WriteIndentedString(1, "private final String value;\n")
+	sb.WriteIndentedString(1, className+"(final String value) {this.value = value;}\n")
+	sb.WriteIndentedString(1, "public String getValue() {return this.value;}\n")
+	sb.WriteString("}\n")
+
+	return fmt.Sprintf("enums/%s.java", className), []byte(sb.String()), nil
+}

internal/core/models.go

@@ -41,6 +41,7 @@ type JavaType struct {
 	Type       string
 	IsList     bool
 	IsNullable bool
+	IsEnum     bool
 }
 
 type QueryArg struct {
@@ -94,6 +95,10 @@ func (q QueryArg) BindStmt() string {
 		return fmt.Sprintf("%s == null ? stmt.setNull(%d, java.sql.Types.%s) : %s", q.Name, q.Number, javaSqlType, rawSet)
 	}
 
+	if q.JavaType.IsEnum {
+		return fmt.Sprintf("stmt.setString(%d, %s.getValue());", q.Number, q.Name)
+	}
+
 	return fmt.Sprintf("stmt.setObject(%d, %s);", q.Number, q.Name)
 }
 
@@ -107,7 +112,6 @@ func (q QueryReturn) ResultStmt(number int) string {
 	typeOnly := q.JavaType.Type[strings.LastIndex(q.JavaType.Type, ".")+1:]
 
 	if q.JavaType.IsList {
-		// TODO - check for nullable array support
 		if q.JavaType.IsNullable {
 			return fmt.Sprintf("getList(results, %d, %s[].class)", number, typeOnly)
 		}
@@ -127,6 +131,13 @@ func (q QueryReturn) ResultStmt(number int) string {
 		return fmt.Sprintf("results.get%s(%d)", typeOnly, number)
 	}
 
+	if q.JavaType.IsEnum {
+		if q.JavaType.IsNullable {
+			return fmt.Sprintf("Optional.ofNullable(results.getSting(%d)).map(%s::valueOf).orElse(null)", number, typeOnly)
+		}
+		return fmt.Sprintf("%s.valueOf(results.getString(%d))", typeOnly, number)
+	}
+
 	return fmt.Sprintf("results.getObject(%d, %s.class)", number, typeOnly)
 }
 
@@ -149,7 +160,15 @@ type NullableHelpers struct {
 	List    bool
 }
 
+type Enum struct {
+	Schema string
+	Name   string
+	Values []string
+}
+
 type (
 	Queries        map[string][]Query
 	EmbeddedModels map[string][]QueryReturn
+	// Enums is a map of "schema_name.enum_name" to enum value.
+	Enums map[string]Enum
 )