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