support sql nullable types

cmd/gen/conditions/condition.go

@@ -108,20 +108,25 @@ func (condition *Condition) generateForNamedType(objectType Type, field Field) {
 	_, err := field.Type.BadaasModelStruct()
 
 	if err == nil {
+		// field is a badaas model
 		condition.generateForBadaasModel(objectType, field)
+	} else if field.Type.IsSQLNullableType() {
+		// field is a sql nullable type (sql.NullBool, sql.NullInt, etc.)
+		condition.param.SQLToBasicType(field.Type)
+		condition.generateWhere(
+			objectType,
+			field,
+		)
+	} else if field.Type.IsGormCustomType() || field.TypeString() == "time.Time" {
+		// field is a Gorm Custom type (implements Scanner and Valuer interfaces)
+		// or a named type supported by gorm (time.Time)
+		condition.param.ToCustomType(condition.destPkg, field.Type)
+		condition.generateWhere(
+			objectType,
+			field,
+		)
 	} else {
-		// field is not a Badaas Model
-		if field.Type.IsGormCustomType() || field.TypeString() == "time.Time" {
-			// field is a Gorm Custom type (implements Scanner and Valuer interfaces)
-			// or a named type supported by gorm (time.Time, gorm.DeletedAt)
-			condition.param.ToCustomType(condition.destPkg, field.Type)
-			condition.generateWhere(
-				objectType,
-				field,
-			)
-		} else {
-			log.Logger.Debugf("struct field type not handled: %s", field.TypeString())
-		}
+		log.Logger.Debugf("struct field type not handled: %s", field.TypeString())
 	}
 }
 

cmd/gen/conditions/jenParam.go

@@ -84,3 +84,27 @@ func (param JenParam) ToCustomType(destPkg string, typeV Type) {
 		typeV.Name(),
 	)
 }
+
+func (param JenParam) SQLToBasicType(typeV Type) {
+	switch typeV.String() {
+	case nullString:
+		param.internalType.String()
+	case nullInt64:
+		param.internalType.Int64()
+	case nullInt32:
+		param.internalType.Int32()
+	case nullInt16:
+		param.internalType.Int16()
+	case nullByte:
+		param.internalType.Int8()
+	case nullFloat64:
+		param.internalType.Float64()
+	case nullBool:
+		param.internalType.Bool()
+	case nullTime, deletedAt:
+		param.internalType.Qual(
+			"time",
+			"Time",
+		)
+	}
+}

cmd/gen/conditions/main_test.go

@@ -68,6 +68,12 @@ func TestCustomType(t *testing.T) {
 	})
 }
 
+func TestNullableTypes(t *testing.T) {
+	doTest(t, "./tests/nullabletypes", []Comparison{
+		{Have: "nullable_types_conditions.go", Expected: "./tests/results/nullabletypes.go"},
+	})
+}
+
 func TestBelongsTo(t *testing.T) {
 	doTest(t, "./tests/belongsto", []Comparison{
 		{Have: "owner_conditions.go", Expected: "./tests/results/belongsto_owner.go"},

cmd/gen/conditions/tests/nullabletypes/nullabletypes.go

@@ -0,0 +1,20 @@
+package nullabletypes
+
+import (
+	"database/sql"
+
+	"github.com/ditrit/badaas/orm"
+)
+
+type NullableTypes struct {
+	orm.UUIDModel
+
+	String  sql.NullString
+	Int64   sql.NullInt64
+	Int32   sql.NullInt32
+	Int16   sql.NullInt16
+	Byte    sql.NullByte
+	Float64 sql.NullFloat64
+	Bool    sql.NullBool
+	Time    sql.NullTime
+}