Align Scanner to match what SQLite prefers to print

Author: pan93412

lib/scanner.go

@@ -2,15 +2,40 @@ package sqlrunner
 
 import (
 	"database/sql"
+	"encoding/hex"
 	"fmt"
+	"strconv"
+	"time"
 )
 
 type StringScanner struct {
 	value string
 }
 
 func (s *StringScanner) Scan(value any) error {
-	s.value = fmt.Sprintf("%v", value)
+	switch v := value.(type) {
+	case int64:
+		s.value = strconv.FormatInt(v, 10)
+	case float64:
+		s.value = strconv.FormatFloat(v, 'f', -1, 64)
+	case bool:
+		if v {
+			s.value = "1"
+		} else {
+			s.value = "0"
+		}
+	case []byte:
+		s.value = hex.EncodeToString(v)
+	case string:
+		s.value = v
+	case time.Time:
+		s.value = v.Format("2006-01-02 15:04:05")
+	case nil:
+		s.value = "NULL"
+	default:
+		s.value = fmt.Sprintf("%v", value)
+	}
+
 	return nil
 }
 

lib/scanner_test.go

@@ -0,0 +1,85 @@
+package sqlrunner
+
+import (
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestScanner(t *testing.T) {
+	t.Parallel()
+
+	t.Run("int64", func(t *testing.T) {
+		t.Parallel()
+
+		s := &StringScanner{}
+		require.NoError(t, s.Scan(int64(42)))
+		assert.Equal(t, "42", s.Value())
+	})
+
+	t.Run("float64 expless", func(t *testing.T) {
+		t.Parallel()
+
+		s := &StringScanner{}
+		require.NoError(t, s.Scan(float64(42.42)))
+		assert.Equal(t, "42.42", s.Value())
+	})
+
+	t.Run("float64 exp", func(t *testing.T) {
+		t.Parallel()
+
+		s := &StringScanner{}
+		require.NoError(t, s.Scan(float64(42.42424242424242e8)))
+		assert.Equal(t, "4242424242.424242", s.Value())
+	})
+
+	t.Run("bool true", func(t *testing.T) {
+		t.Parallel()
+
+		s := &StringScanner{}
+		require.NoError(t, s.Scan(true))
+		assert.Equal(t, "1", s.Value())
+	})
+
+	t.Run("bool false", func(t *testing.T) {
+		t.Parallel()
+
+		s := &StringScanner{}
+		require.NoError(t, s.Scan(false))
+		assert.Equal(t, "0", s.Value())
+	})
+
+	t.Run("[]byte", func(t *testing.T) {
+		t.Parallel()
+
+		s := &StringScanner{}
+		require.NoError(t, s.Scan([]byte("hello")))
+		assert.Equal(t, "68656c6c6f", s.Value())
+	})
+
+	t.Run("string", func(t *testing.T) {
+		t.Parallel()
+
+		s := &StringScanner{}
+		require.NoError(t, s.Scan("hello"))
+		assert.Equal(t, "hello", s.Value())
+	})
+
+	t.Run("time.Time", func(t *testing.T) {
+		t.Parallel()
+
+		s := &StringScanner{}
+		require.NoError(t, s.Scan(time.Date(2021, 1, 2, 3, 4, 5, 6, time.UTC)))
+		assert.Equal(t, "2021-01-02 03:04:05", s.Value())
+	})
+
+	t.Run("nil", func(t *testing.T) {
+		t.Parallel()
+
+		s := &StringScanner{}
+		require.NoError(t, s.Scan(nil))
+		assert.Equal(t, "NULL", s.Value())
+	})
+}

lib/sqlrunner_test.go

@@ -240,6 +240,29 @@ func TestDbRunnerReadonly(t *testing.T) {
 	require.ErrorAs(t, err, &sqlrunner.QueryError{})
 }
 
+func TestDbRunnerNoScientificNotation(t *testing.T) {
+	t.Parallel()
+
+	runner, err := sqlrunner.NewSQLRunner(`
+		CREATE TABLE notationtest (
+			value REAL
+		);
+
+		INSERT INTO notationtest (value) VALUES (1.0);
+		INSERT INTO notationtest (value) VALUES (1145141919.810)
+	`)
+
+	require.NoError(t, err)
+
+	result, err := runner.Query(context.TODO(), "SELECT value FROM notationtest")
+	require.NoError(t, err)
+
+	assert.Len(t, result.Rows, 2)
+	assert.Equal(t, []string{"value"}, result.Columns)
+	assert.Equal(t, "1", result.Rows[0][0])
+	assert.Equal(t, "1145141919.81", result.Rows[1][0])
+}
+
 func BenchmarkDbrunner(b *testing.B) {
 	b.ReportAllocs()