struct tag默认值
1. 需求背景
有时候gopher在marshal一个struct到json的时候,想要struct某些属性在没有值的情况下有默认值,但是按照现在marshar的作用下不会给struct的属性赋默认值,所以为了解决这个特殊需求,我们应该按照以下方案来解决。
2. 解决方案
利用反射,因为通过反射,我们可以拿到tag属性,进而对默认字段做赋值操作
先看一段代码:
type test struct { Name string `json:"name"` Addr string `json:"addr" default:"localhost"` Port uint `json:"port" default:"8080"` } func MarshalJson(i interface{}) ([]byte, error) { typeof := reflect.TypeOf(i) valueof := reflect.ValueOf(i) for i := 0; i < typeof.Elem().NumField(); i++ { if valueof.Elem().Field(i).IsZero() { def := typeof.Elem().Field(i).Tag.Get("default") if def != "" { switch typeof.Elem().Field(i).Type.String() { case "int": result, _ := strconv.Atoi(def) valueof.Elem().Field(i).SetInt(int64(result)) case "uint": result, _ := strconv.ParseUint(def, 10, 64) valueof.Elem().Field(i).SetUint(result) case "string": valueof.Elem().Field(i).SetString(def) } } } } return json.Marshal(i) } func main() { t := &test{ Name: "test server", } data, err := MarshalJon(t) if err != nil { panic(err) } fmt.Println(string(data)) } 结果: {"name":"test server","addr":"localhost","port":8080}
3. 小结
其实在解析struct的时候golang本身已经为struct的属性按照类型赋默认值了,但是这个默认值在没有办法满足我们的需求的时候,这个时候就需要造轮子,别怕麻烦,解决问题才是王道。大家想想unmarshal的时候如何赋默认值呢?