Merge pull request #6 from myuon/issue-5

fix: returns nil if value is nil
myuon · Dec 15, 2023 · 9955c96 · 9955c96
2 parents 64a5418 + a8664c5
commit 9955c96
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/gallon/input_sql.go b/gallon/input_sql.go
@@ -153,6 +153,11 @@ type InputPluginSqlConfigSchemaColumn struct {
 }
 
 func (c InputPluginSqlConfigSchemaColumn) getValue(value interface{}) (interface{}, error) {
+	// if value is nil, returns nil anyway
+	if value == nil {
+		return nil, nil
+	}
+
 	switch c.Type {
 	case "string":
 		v, ok := value.(string)

diff --git a/test/mysql/mysql_to_file_test.go b/test/mysql/mysql_to_file_test.go
@@ -81,6 +81,12 @@ func Migrate(db *sql.DB) error {
 			return err
 		}
 
+		// 50% chance to have partner
+		// NOTE: gofakeit does not support nullable bool
+		if gofakeit.Bool() {
+			v.HasPartner = nil
+		}
+
 		if _, err := query.Exec(v.ID, v.Name, v.Age, v.CreatedAt, v.Birthday, v.HasPartner); err != nil {
 			return err
 		}