Kyle/mysql bool tinyint one

internal/codegen/golang/mysql_type.go

@@ -21,7 +21,20 @@ func mysqlType(r *compiler.Result, col *compiler.Column, settings config.Combine
 		}
 		return "sql.NullString"
 
-	case "int", "integer", "tinyint", "smallint", "mediumint", "year":
+	case "tinyint":
+		if col.Length != nil && *col.Length == 1 {
+			if notNull {
+				return "bool"
+			}
+			return "sql.NullBool"
+		} else {
+			if notNull {
+				return "int32"
+			}
+			return "sql.NullInt32"
+		}
+
+	case "int", "integer", "smallint", "mediumint", "year":
 		if notNull {
 			return "int32"
 		}

internal/compiler/query.go

@@ -1,6 +1,8 @@
 package compiler
 
-import "github.com/kyleconroy/sqlc/internal/sql/ast"
+import (
+	"github.com/kyleconroy/sqlc/internal/sql/ast"
+)
 
 type Table struct {
 	Rel     *ast.TableName
@@ -13,6 +15,7 @@ type Column struct {
 	NotNull  bool
 	IsArray  bool
 	Comment  string
+	Length   *int
 
 	// XXX: Figure out what PostgreSQL calls `foo.id`
 	Scope string

internal/compiler/query_catalog.go

@@ -52,6 +52,7 @@ func ConvertColumn(rel *ast.TableName, c *catalog.Column) *Column {
 		NotNull:  c.IsNotNull,
 		IsArray:  c.IsArray,
 		Type:     &c.Type,
+		Length:   c.Length,
 	}
 }
 

internal/endtoend/testdata/data_type_boolean/mysql/query.sql

@@ -0,0 +1,19 @@
+CREATE TABLE foo
+(
+    col_a BOOL             NOT NULL,
+    col_b BOOLEAN          NOT NULL,
+    col_c TINYINT(1)       NOT NULL
+);
+
+-- name: ListFoo :many
+SELECT * FROM foo;
+
+CREATE TABLE bar
+(
+    col_a BOOL,
+    col_b BOOLEAN,
+    col_c TINYINT(1)
+);
+
+-- name: ListBar :many
+SELECT * FROM bar;

internal/endtoend/testdata/data_type_boolean/mysql/sqlc.json

@@ -0,0 +1,12 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "path": "db",
+      "engine": "mysql",
+      "schema": "query.sql",
+      "queries": "query.sql"
+    }
+  ]
+}
+

internal/endtoend/testdata/data_type_boolean/postgresql/query.sql

@@ -0,0 +1,17 @@
+CREATE TABLE foo
+(
+    col_a BOOL             NOT NULL,
+    col_b BOOLEAN          NOT NULL
+);
+
+-- name: ListFoo :many
+SELECT * FROM foo;
+
+CREATE TABLE bar
+(
+    col_a BOOL,
+    col_b BOOLEAN
+);
+
+-- name: ListBar :many
+SELECT * FROM bar;

internal/endtoend/testdata/data_type_boolean/postgresql/sqlc.json

@@ -0,0 +1,12 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "path": "db",
+      "engine": "postgresql",
+      "schema": "query.sql",
+      "queries": "query.sql"
+    }
+  ]
+}
+