在使用Go開(kāi)發(fā)web項(xiàng)目的過(guò)程中, 數(shù)據(jù)庫(kù)讀寫操作與JSON格式的輸入輸出是兩塊最基礎(chǔ)的模塊, Go的標(biāo)準(zhǔn)庫(kù)已經(jīng)幫我們做了很多, 熟悉database/sql與encoding/json這兩個(gè)庫(kù)能幫我們更自在地開(kāi)發(fā)web應(yīng)用.
但此篇文章拋開(kāi)基礎(chǔ)不說(shuō), 只說(shuō)一些在開(kāi)發(fā)中遇到一些真實(shí)存在的痛點(diǎn).
Go的一大特色就是zero value, 比如int類型的zero value是0, string為"", struct為每個(gè)field里各自類型的zero value. 因此在Go的很多ORM處理NULL值時(shí), 都是通過(guò)zero value機(jī)制入庫(kù)或出庫(kù)的, 因此, 使用ORM操作的數(shù)據(jù)庫(kù), 如何沒(méi)有明確指明, 基本上看不到NULL值. 一個(gè)可能為NULL的varchar字段, 存入了""(空字符串).
這里不討論關(guān)于空字符串與NULL的優(yōu)劣, 而是說(shuō)明如何處理NULL值.
sql標(biāo)準(zhǔn)庫(kù)里帶了這么幾個(gè)類型:sql.NullString, sql.NullInt64, sql.NullBool, sql.NullFloat64, 如果在定義struct里field類型的時(shí)候, 使用sql.NullString代替string的話, 就可以在數(shù)據(jù)庫(kù)里存入NULL值.
通過(guò)源碼看到, NullString的結(jié)構(gòu)與兩個(gè)方法:
type NullString struct {
? String string
? Valid bool
}
Scan(value interface{}) error
Value()(driver.Value, error)
Scan實(shí)現(xiàn)了sql.Scanner接口, Value實(shí)現(xiàn)了driver.Valuer接口. 這兩個(gè)接口表示數(shù)據(jù)從Go端與DB端的轉(zhuǎn)換. Scan用于數(shù)據(jù)從DB轉(zhuǎn)到Go時(shí)被調(diào)用,也就是說(shuō)select時(shí)用(查看源碼); 而Value則由Go轉(zhuǎn)到DB時(shí)被調(diào)用. 也就是說(shuō)insert/update時(shí)候用(查看源碼).
在兩個(gè)不同領(lǐng)域做數(shù)據(jù)轉(zhuǎn)換時(shí), 通常用接口來(lái)定義, 這是接口的作用, JSON也是同樣如此.
有了接口的幫助, 任何實(shí)現(xiàn)了這兩個(gè)接口的類型, 都能在DB與Go之間做轉(zhuǎn)換, 接口建立起了橋梁.
由上一節(jié)可以想到, 如果在想要保存Go自定義類型到DB里, 只需要實(shí)現(xiàn)Scanner與Valuer接口, 就像sql.NullString一樣.
典型如NullTime查看源碼:
type NullTime struct {
? ? Time? time.Time
? ? Valid bool
}
func (nt *NullTime) Scan(value interface{}) error {
? ? nt.Time, nt.Valid = value.(time.Time)
? ? return nil
}
func (nt NullTime) Value() (driver.Value, error) {
? ? if !nt.Valid {
? ? ? ? return nil, nil
}
? ? return nt.Time, nil
}
為了讓這個(gè)類型更好用, 添加一個(gè)幫助函數(shù):
func ToNullTime(t time.Time) NullTime {
? ? return NullTime{Time: t, Valid: !t.IsZero()}
}
如果拿一個(gè)sql.NullString的類型, 做json.Marshal, 那么得到輸出是這樣的:
{? ?
"column_name": {"String":"Hello?","Valid":true}
}
而我們想要的是:
{? ?
"column_name":"Hello?"} ?//or{ "column_name": null
}
我們知道, json庫(kù)有兩個(gè)標(biāo)準(zhǔn)的接口,json.Marshaler查看源碼,json.Unmarshaler查看源碼, 如果一個(gè)類型實(shí)現(xiàn)此兩個(gè)接口, 則在使用json.Marshal/Unmarshal調(diào)用時(shí), 調(diào)用此類型的自定義方法:
type NullString struct {
? ? sql.NullString
}
func (v *NullString) MarshalJSON() ([]byte, error) {
? ? if v.Valid {
? ? ? ? return json.Marshal(v.String)
? ? } else {
? ? ? ? return json.Marshal(nil)
? ? }
}
func (v NullString) UnmarshalJSON(data []byte) error {
? ? var s *string
? ? if err := json.Unmarshal(data, &s); err != nil {
? ? ? ? return err
? ? }
? ? if s != nil {
? ? ? ? v.Valid = true
? ? ? ? v.String = *s
? ? } else {
? ? ? ? v.Valid = false
? ? }
? ? return nil
}
Go語(yǔ)言的接口, 扮演了橋梁的角色, 連接起了Go與其它領(lǐng)域的數(shù)據(jù)轉(zhuǎn)換, 通過(guò)實(shí)現(xiàn)標(biāo)準(zhǔn)庫(kù)的接口, 我們可以讓Go的數(shù)據(jù)類型輕松適應(yīng)不同數(shù)據(jù)領(lǐng)域.
[參考]
http://dennissuratna.com/marshalling-nullable-string-db-value-to-json-in-go
/http://blog.carbonfive.com/2015/07/09/there-will-be-sql/
http://marcesher.com/2014/10/13/go-working-effectively-with-database-nulls/
