fix: time repr wrapper (#2255)

master
Kevin Wan 2 years ago committed by GitHub
parent 27e773fa1f
commit 1568c3be0e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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))
} }

@ -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())
}

Loading…
Cancel
Save