fix: time repr wrapper (#2255)
This commit is contained in:
@@ -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())
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user