fix: time repr wrapper (#2255)

This commit is contained in:
Kevin Wan
2022-08-11 22:39:54 +08:00
committed by GitHub
parent 27e773fa1f
commit 1568c3be0e
2 changed files with 21 additions and 0 deletions

View File

@@ -6,6 +6,7 @@ import (
"fmt" "fmt"
"strconv" "strconv"
"strings" "strings"
"time"
"github.com/zeromicro/go-zero/core/logx" "github.com/zeromicro/go-zero/core/logx"
"github.com/zeromicro/go-zero/core/mapping" "github.com/zeromicro/go-zero/core/mapping"
@@ -152,6 +153,14 @@ func writeValue(buf *strings.Builder, arg interface{}) {
buf.WriteByte('\'') buf.WriteByte('\'')
buf.WriteString(escape(v)) buf.WriteString(escape(v))
buf.WriteByte('\'') buf.WriteByte('\'')
case time.Time:
buf.WriteByte('\'')
buf.WriteString(v.String())
buf.WriteByte('\'')
case *time.Time:
buf.WriteByte('\'')
buf.WriteString(v.String())
buf.WriteByte('\'')
default: default:
buf.WriteString(mapping.Repr(v)) buf.WriteString(mapping.Repr(v))
} }

View File

@@ -3,6 +3,7 @@ package sqlx
import ( import (
"strings" "strings"
"testing" "testing"
"time"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
) )
@@ -138,3 +139,14 @@ func TestFormat(t *testing.T) {
}) })
} }
} }
func TestWriteValue(t *testing.T) {
var buf strings.Builder
tm := time.Now()
writeValue(&buf, &tm)
assert.Equal(t, "'"+tm.String()+"'", buf.String())
buf.Reset()
writeValue(&buf, tm)
assert.Equal(t, "'"+tm.String()+"'", buf.String())
}