Merge pull request #825 from maniak89/null_int32

Support sql.NullInt32 for write
tealeg · Nov 1, 2024 · faa27f6 · faa27f6
2 parents 7c33d07 + c8f4cd0
commit faa27f6
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/write.go b/write.go
@@ -59,6 +59,11 @@ func (r *Row) WriteSlice(e interface{}, cols int) int {
 			if cell.SetString(``); t.Valid {
 				cell.SetValue(t.Int64)
 			}
+		case sql.NullInt32:
+			cell := r.AddCell()
+			if cell.SetString(``); t.Valid {
+				cell.SetValue(t.Int32)
+			}
 		case sql.NullFloat64:
 			cell := r.AddCell()
 			if cell.SetString(``); t.Valid {
@@ -136,6 +141,11 @@ func (r *Row) WriteStruct(e interface{}, cols int) int {
 			if cell.SetString(``); t.Valid {
 				cell.SetValue(t.Int64)
 			}
+		case sql.NullInt32:
+			cell := r.AddCell()
+			if cell.SetString(``); t.Valid {
+				cell.SetValue(t.Int32)
+			}
 		case sql.NullFloat64:
 			cell := r.AddCell()
 			if cell.SetString(``); t.Valid {

diff --git a/write_test.go b/write_test.go
@@ -42,7 +42,7 @@ func TestWrite(t *testing.T) {
 			Raiting         sql.NullFloat64
 			NullLastName    sql.NullString
 			NullHasPhd      sql.NullBool
-			NullGithubStars sql.NullInt64
+			NullGithubStars sql.NullInt32
 			NullRaiting     sql.NullFloat64
 		}
 		testStruct := e{
@@ -59,7 +59,7 @@ func TestWrite(t *testing.T) {
 			sql.NullFloat64{Float64: 0.123, Valid: true},
 			sql.NullString{String: `What ever`, Valid: false},
 			sql.NullBool{Bool: true, Valid: false},
-			sql.NullInt64{Int64: 100, Valid: false},
+			sql.NullInt32{Int32: 100, Valid: false},
 			sql.NullFloat64{Float64: 0.123, Valid: false},
 		}
 		cnt := row.WriteStruct(&testStruct, -1)