golang xorm及time.Time自定義解決json日期格式的問題
golang默認(rèn)的time.Time類型在轉(zhuǎn)為json格式時(shí)不是常用的2019-05-08 10:00:01這種格式，解決辦法是自定義一個(gè)時(shí)間類型，例如
type myTime time.Time ，然后針對(duì)myTime實(shí)現(xiàn)Marshaler接口的MarshalJSON方法，例如：
package models
import (
"database/sql/driver"
"time"
)
const localDateTimeFormat string = "2006-01-02 15:04:05"
type LocalTime time.Time
func (l LocalTime) MarshalJSON() ([]byte, error) {
b := make([]byte, 0, len(localDateTimeFormat)+2)
b = append(b, '"')
b = time.Time(l).AppendFormat(b, localDateTimeFormat)
b = append(b, '"')
return b, nil
}
func (l *LocalTime) UnmarshalJSON(b []byte) error {
now, err := time.ParseInLocation(`"`+localDateTimeFormat+`"`, string(b), time.Local)
*l = LocalTime(now)
return err
}
上面的代碼在網(wǎng)上隨手一搜就能找到，沒有什么困難的，接下來的才是本篇文章的重點(diǎn)，這玩意結(jié)合xorm使用時(shí)，特別是字段類型為*LocalTime的時(shí)候才需要折騰一番。
下面是我的對(duì)應(yīng)數(shù)據(jù)庫表結(jié)構(gòu)的struct 定義，
type ServerInfo struct {
ServerInfoId string `xorm:"varchar(32) pk server_info_id"`
CreatedAt LocalTime `xorm:"timestamp created"`
UpdatedAt LocalTime `xorm:"timestamp updated"`
DeletedAt *LocalTime `xorm:"timestamp deleted index"`
OrgId string `xorm:"varchar(100) org_id" json:"orgId"`
ServerIp string `xorm:"varchar(128) server_ip" json:"serverIp"`
ServerNameDesc string `xorm:"varchar(500) server_name_desc" json:"serverNameDesc"`
ServerTimeNow LocalTime `xorm:"timestamp server_time" json:"serverTime"`
DataReceiveTime LocalTime `xorm:"timestamp data_receive_time" sql:"DEFAULT:current_timestamp" json:"dataRecvTime"`
LastUploadDataTime *LocalTime `xorm:"timestamp last_upload_data_time" json:"lastUploadDataTime"`
LastCheckTime *LocalTime `xorm:"timestamp last_check_time" json:"lastCheckTime"`
LastErrorTime *LocalTime `xorm:"timestamp last_error_time" json:"lastErrorTime"`
}
注意上面的字段類型，既有LocalTime類型的，又有*LocalTime類型的，*LocalTime是考慮到有時(shí)候數(shù)據(jù)值可能為NULL，即字段值可能為空的情況。
xorm不知道如何為L(zhǎng)ocalTime這個(gè)自定義類型進(jìn)行賦值或者取值，因此需要實(shí)現(xiàn)xorm的core包中的Conversion接口，這個(gè)接口的定義如下：
注意，坑已經(jīng)隱藏在上面的接口定義中了，過一會(huì)說。
整個(gè)完整的自定義時(shí)間類型的代碼變成了下面的這樣：
package models
import (
"database/sql/driver"
"time"
)
const localDateTimeFormat string = "2006-01-02 15:04:05"
type LocalTime time.Time
func (l LocalTime) MarshalJSON() ([]byte, error) {
b := make([]byte, 0, len(localDateTimeFormat)+2)
b = append(b, '"')
b = time.Time(l).AppendFormat(b, localDateTimeFormat)
b = append(b, '"')
return b, nil
}
func (l *LocalTime) UnmarshalJSON(b []byte) error {
now, err := time.ParseInLocation(`"`+localDateTimeFormat+`"`, string(b), time.Local)
*l = LocalTime(now)
return err
}
func (l LocalTime) String() string {
return time.Time(l).Format(localDateTimeFormat)
}
func (l LocalTime)Now()(LocalTime){
return LocalTime(time.Now())
}
func (l LocalTime)ParseTime(t time.Time)(LocalTime){
return LocalTime(t)
}
func (j LocalTime) format() string {
return time.Time(j).Format(localDateTimeFormat)
}
func (j LocalTime) MarshalText() ([]byte, error) {
return []byte(j.format()), nil
}
func (l *LocalTime) FromDB(b []byte) error {
if nil == b || len(b) == 0 {
l = nil
return nil
}
var now time.Time
var err error
now, err = time.ParseInLocation(localDateTimeFormat, string(b), time.Local)
if nil == err {
*l = LocalTime(now)
return nil
}
now, err = time.ParseInLocation("2006-01-02T15:04:05Z", string(b), time.Local)
if nil == err {
*l = LocalTime(now)
return nil
}
panic("自己定義個(gè)layout日期格式處理一下數(shù)據(jù)庫里面的日期型數(shù)據(jù)解析!")
return err
}
//func (t *LocalTime) Scan(v interface{}) error {
// // Should be more strictly to check this type.
// vt, err := time.Parse("2006-01-02 15:04:05", string(v.([]byte)))
// if err != nil {
// return err
// }
// *t = LocalTime(vt)
// return nil
//}
func (l *LocalTime) ToDB() ([]byte, error) {
if nil == l {
return nil,nil
}
return []byte(time.Time(*l).Format(localDateTimeFormat)), nil
}
func (l *LocalTime) Value() (driver.Value, error) {
if nil==l {
return nil, nil
}
return time.Time(*l).Format(localDateTimeFormat), nil
}
此時(shí)，要是數(shù)據(jù)庫的字段內(nèi)容都有值的話插入和更新應(yīng)該是沒有什么問題，但是*LocalTime字段的值為nil的話問題就開始出現(xiàn)了，上面說了，ToDB()方法的返回值類型為[]byte，當(dāng)字段值為nil時(shí)，返回nil看上去一切正常，但是xorm打印出來的sql語句數(shù)據(jù)值是下面這個(gè)樣子的：
這個(gè)[]uint8(nil)就是*LocalTime值為nil時(shí)的情況，數(shù)據(jù)庫驅(qū)動(dòng)是不認(rèn)可[]uint8(nil)這種數(shù)據(jù)去寫給timestamp類型字段的，問題的根源就是ToDB方法的返回值類型為[]byte，既然是這樣，就需要我們?nèi)藶榈陌裑]uint8(nil)這種類型改為interface(nil)類型，數(shù)據(jù)庫驅(qū)動(dòng)會(huì)識(shí)別interface(nil)為NULL值，修改代碼xorm\statement.go第322行，把原來的val=data改成下面的樣子：
就是把val=data改為 if nil==data { val=nil } else {val=data} ，看上去邏輯沒有什么變化，但是給val=nil賦值的時(shí)候，val的類型就從[]uint8(nil)變成了interface(nil)了，這樣數(shù)據(jù)庫驅(qū)動(dòng)就可以正確處理空值了。
除了需要修改xorm\statement.go文件的內(nèi)容，還需要修改xorm\session_convert.go的第558行，增加以下代碼：
主要是增加下面的代碼
//fix when pointer type value is null,added by peihexian,2019-05-07
if nil==data {
return nil,nil
}
之所以加這個(gè)代碼是因?yàn)閤orm作者沒有考慮指針類型字段值為nil的情況，xorm對(duì)有轉(zhuǎn)換的字段要么當(dāng)成數(shù)字，要么當(dāng)成了字符串，這兩種對(duì)于NULL類型的值都不適用，所以需要增加if nil==data return nil,nil這樣的代碼，還是把數(shù)據(jù)值組織成interface(nil)去給數(shù)據(jù)庫驅(qū)動(dòng)去處理。
另外還有一個(gè)地方，是session_convert.go 第556行，同樣需要增加
if nil==data { //edit by peihexian 2019.06.19
return nil,nil
}
下面是加完以后的樣子
到這里，對(duì)xorm做了幾處小的修改，自定義日期的問題及json格式化問題完美解決。
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教。
相關(guān)文章
詳解golang函數(shù)多返回值錯(cuò)誤處理與error類型
這篇文章主要為大家詳細(xì)介紹了golang中函數(shù)多返回值錯(cuò)誤處理與error類型的相關(guān)知識(shí),文中的示例代碼簡(jiǎn)潔易懂,感興趣的小伙伴快跟隨小編一起學(xué)習(xí)吧2023-10-10
基于Golang實(shí)現(xiàn)內(nèi)存數(shù)據(jù)庫的示例詳解
這篇文章主要為大家詳細(xì)介紹了如何基于Golang實(shí)現(xiàn)內(nèi)存數(shù)據(jù)庫，文中的示例代碼講解詳細(xì)，具有一定的借鑒價(jià)值，需要的小伙伴可以參考一下2023-03-03
Golang設(shè)計(jì)模式工廠模式實(shí)戰(zhàn)寫法示例詳解
這篇文章主要為大家介紹了Golang 工廠模式實(shí)戰(zhàn)寫法示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2022-08-08
這篇文章主要介紹了Go語言五種變量創(chuàng)建的方法,本文給大家提到了匿名變量的優(yōu)點(diǎn)，通過實(shí)例代碼給大家介紹的非常詳細(xì)，需要的朋友可以參考下2020-02-02
在很多的開源項(xiàng)目里,經(jīng)?？吹介]包的運(yùn)用,這篇文章主要為大家簡(jiǎn)要記錄一下閉包的概念和一些常用的場(chǎng)景,感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下2023-11-11
用Go+Redis實(shí)現(xiàn)分布式鎖的示例代碼
在分布式的業(yè)務(wù)中 , 如果有的共享資源需要安全的被訪問和處理 , 那就需要分布式鎖，本文主要介紹了用Go+Redis實(shí)現(xiàn)分布式鎖的示例代碼，感興趣的可以了解一下2021-12-12
Go語言Elasticsearch數(shù)據(jù)清理工具思路詳解
這篇文章主要介紹了Go語言Elasticsearch數(shù)據(jù)清理工具思路詳解,本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2021-10-10
本文主要介紹了go?map搬遷的實(shí)現(xiàn)，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-04-04
備忘錄是一種行為設(shè)計(jì)模式, 允許生成對(duì)象狀態(tài)的快照并在以后將其還原,本文就通過代碼示例給大家講講Go備忘錄模式,感興趣的小伙伴跟著小編一起來看看吧2023-08-08
今天小編就為大家分享一篇關(guān)于go語言編譯器的介紹，小編覺得內(nèi)容挺不錯(cuò)的，現(xiàn)在分享給大家，具有很好的參考價(jià)值，需要的朋友一起跟隨小編來看看吧2018-09-09