Go系列教程之反射的用法

反射是 Go 語(yǔ)言的高級(jí)主題之一。我會(huì)盡可能讓它變得簡(jiǎn)單易懂。

本教程分為如下小節(jié)。

• 什么是反射？
• 為何需要檢查變量，確定變量的類(lèi)型？
• reflect 包
  • reflect.Type 和 reflect.Value
  • reflect.Kind
  • NumField() 和 Field() 方法
  • Int() 和 String() 方法
• 完整的程序
• 我們應(yīng)該使用反射嗎？

讓我們來(lái)逐個(gè)討論這些章節(jié)。

什么是反射？

反射就是程序能夠在運(yùn)行時(shí)檢查變量和值，求出它們的類(lèi)型。你可能還不太懂，這沒(méi)關(guān)系。在本教程結(jié)束后，你就會(huì)清楚地理解反射，所以跟著我們的教程學(xué)習(xí)吧。

為何需要檢查變量，確定變量的類(lèi)型？

在學(xué)習(xí)反射時(shí)，所有人首先面臨的疑惑就是：如果程序中每個(gè)變量都是我們自己定義的，那么在編譯時(shí)就可以知道變量類(lèi)型了，為什么我們還需要在運(yùn)行時(shí)檢查變量，求出它的類(lèi)型呢？沒(méi)錯(cuò)，在大多數(shù)時(shí)候都是這樣，但并非總是如此。

我來(lái)解釋一下吧。下面我們編寫(xiě)一個(gè)簡(jiǎn)單的程序。

package main

import (
 "fmt"
)

func main() {
 i := 10
 fmt.Printf("%d %T", i, i)
}

在 playground 上運(yùn)行

在上面的程序中，i 的類(lèi)型在編譯時(shí)就知道了，然后我們?cè)谙乱恍写蛴〕?i。這里沒(méi)什么特別之處。

現(xiàn)在了解一下，需要在運(yùn)行時(shí)求得變量類(lèi)型的情況。假如我們要編寫(xiě)一個(gè)簡(jiǎn)單的函數(shù)，它接收結(jié)構(gòu)體作為參數(shù)，并用它來(lái)創(chuàng)建一個(gè) SQL 插入查詢(xún)。

考慮下面的程序：

package main

import (
 "fmt"
)

type order struct {
 ordId  int
 customerId int
}

func main() {
 o := order{
  ordId:  1234,
  customerId: 567,
 }
 fmt.Println(o)
}

在 playground 上運(yùn)行

在上面的程序中，我們需要編寫(xiě)一個(gè)函數(shù)，接收結(jié)構(gòu)體變量 o 作為參數(shù)，返回下面的 SQL 插入查詢(xún)。

insert into order values(1234, 567)

這個(gè)函數(shù)寫(xiě)起來(lái)很簡(jiǎn)單。我們現(xiàn)在編寫(xiě)這個(gè)函數(shù)。

package main

import (
 "fmt"
)

type order struct {
 ordId  int
 customerId int
}

func createQuery(o order) string {
 i := fmt.Sprintf("insert into order values(%d, %d)", o.ordId, o.customerId)
 return i
}

func main() {
 o := order{
  ordId:  1234,
  customerId: 567,
 }
 fmt.Println(createQuery(o))
}

在 playground 上運(yùn)行

在第 12 行，createQuery 函數(shù)用 o 的兩個(gè)字段（ordId 和 customerId），創(chuàng)建了插入查詢(xún)。該程序會(huì)輸出：

insert into order values(1234, 567)

現(xiàn)在我們來(lái)升級(jí)這個(gè)查詢(xún)生成器。如果我們想讓它變得通用，可以適用于任何結(jié)構(gòu)體類(lèi)型，該怎么辦呢？我們用程序來(lái)理解一下。

package main

type order struct {
 ordId  int
 customerId int
}

type employee struct {
 name string
 id int
 address string
 salary int
 country string
}

func createQuery(q interface{}) string {
}

func main() {

}

我們的目標(biāo)就是完成 createQuery 函數(shù)（上述程序中的第 16 行），它可以接收任何結(jié)構(gòu)體作為參數(shù)，根據(jù)結(jié)構(gòu)體的字段創(chuàng)建插入查詢(xún)。

例如，如果我們傳入下面的結(jié)構(gòu)體：

o := order {
 ordId: 1234,
 customerId: 567
}

createQuery 函數(shù)應(yīng)該返回：

insert into order values (1234, 567)

類(lèi)似地，如果我們傳入：

 e := employee {
  name: "Naveen",
  id: 565,
  address: "Science Park Road, Singapore",
  salary: 90000,
  country: "Singapore",
 }

該函數(shù)會(huì)返回：

insert into employee values("Naveen", 565, "Science Park Road, Singapore", 90000, "Singapore")

由于 createQuery 函數(shù)應(yīng)該適用于任何結(jié)構(gòu)體，因此它接收 interface{} 作為參數(shù)。為了簡(jiǎn)單起見(jiàn)，我們只處理包含 string 和 int 類(lèi)型字段的結(jié)構(gòu)體，但可以擴(kuò)展為包含任何類(lèi)型的字段。

createQuery 函數(shù)應(yīng)該適用于所有的結(jié)構(gòu)體。因此，要編寫(xiě)這個(gè)函數(shù)，就必須在運(yùn)行時(shí)檢查傳遞過(guò)來(lái)的結(jié)構(gòu)體參數(shù)的類(lèi)型，找到結(jié)構(gòu)體字段，接著創(chuàng)建查詢(xún)。這時(shí)就需要用到反射了。在本教程的下一步，我們將會(huì)學(xué)習(xí)如何使用 reflect 包來(lái)實(shí)現(xiàn)它。

reflect 包

在 Go 語(yǔ)言中，reflect 實(shí)現(xiàn)了運(yùn)行時(shí)反射。reflect 包會(huì)幫助識(shí)別 interface{} 變量的底層具體類(lèi)型和具體值。這正是我們所需要的。createQuery 函數(shù)接收 interface{} 參數(shù)，根據(jù)它的具體類(lèi)型和具體值，創(chuàng)建 SQL 查詢(xún)。這正是 reflect 包能夠幫助我們的地方。

在編寫(xiě)我們通用的查詢(xún)生成器之前，我們首先需要了解 reflect 包中的幾種類(lèi)型和方法。讓我們來(lái)逐個(gè)了解。

reflect.Type 和 reflect.Value

reflect.Type 表示 interface{} 的具體類(lèi)型，而 reflect.Value 表示它的具體值。reflect.TypeOf() 和 reflect.ValueOf() 兩個(gè)函數(shù)可以分別返回 reflect.Type 和 reflect.Value。這兩種類(lèi)型是我們創(chuàng)建查詢(xún)生成器的基礎(chǔ)。我們現(xiàn)在用一個(gè)簡(jiǎn)單的例子來(lái)理解這兩種類(lèi)型。

package main

import (
 "fmt"
 "reflect"
)

type order struct {
 ordId  int
 customerId int
}

func createQuery(q interface{}) {
 t := reflect.TypeOf(q)
 v := reflect.ValueOf(q)
 fmt.Println("Type ", t)
 fmt.Println("Value ", v)


}
func main() {
 o := order{
  ordId:  456,
  customerId: 56,
 }
 createQuery(o)

}

在 playground 上運(yùn)行

在上面的程序中，第 13 行的 createQuery 函數(shù)接收 interface{} 作為參數(shù)。在第 14 行，reflect.TypeOf 接收了參數(shù) interface{}，返回了reflect.Type，它包含了傳入的 interface{} 參數(shù)的具體類(lèi)型。同樣地，在第 15 行，reflect.ValueOf 函數(shù)接收參數(shù) interface{}，并返回了 reflect.Value，它包含了傳來(lái)的 interface{} 的具體值。

上述程序會(huì)打?。?/p>

Type  main.order
Value  {456 56}

從輸出我們可以看到，程序打印了接口的具體類(lèi)型和具體值。

relfect.Kind

reflect 包中還有一個(gè)重要的類(lèi)型：Kind。

在反射包中，Kind 和 Type 的類(lèi)型可能看起來(lái)很相似，但在下面程序中，可以很清楚地看出它們的不同之處。

package main

import (
 "fmt"
 "reflect"
)

type order struct {
 ordId  int
 customerId int
}

func createQuery(q interface{}) {
 t := reflect.TypeOf(q)
 k := t.Kind()
 fmt.Println("Type ", t)
 fmt.Println("Kind ", k)


}
func main() {
 o := order{
  ordId:  456,
  customerId: 56,
 }
 createQuery(o)

}

在 playground 上運(yùn)行

上述程序會(huì)輸出：

Type  main.order
Kind  struct

我想你應(yīng)該很清楚兩者的區(qū)別了。Type 表示 interface{} 的實(shí)際類(lèi)型（在這里是 main.Order)，而 Kind 表示該類(lèi)型的特定類(lèi)別（在這里是 struct）。

NumField() 和 Field() 方法

NumField() 方法返回結(jié)構(gòu)體中字段的數(shù)量，而 Field(i int) 方法返回字段 i 的 reflect.Value。

package main

import (
 "fmt"
 "reflect"
)

type order struct {
 ordId  int
 customerId int
}

func createQuery(q interface{}) {
 if reflect.ValueOf(q).Kind() == reflect.Struct {
  v := reflect.ValueOf(q)
  fmt.Println("Number of fields", v.NumField())
  for i := 0; i < v.NumField(); i++ {
   fmt.Printf("Field:%d type:%T value:%v\n", i, v.Field(i), v.Field(i))
  }
 }

}
func main() {
 o := order{
  ordId:  456,
  customerId: 56,
 }
 createQuery(o)
}

在 playground 上運(yùn)行

在上面的程序中，因?yàn)?NumField 方法只能在結(jié)構(gòu)體上使用，我們?cè)诘?14 行首先檢查了 q 的類(lèi)別是 struct。程序的其他代碼很容易看懂，不作解釋。該程序會(huì)輸出：

Number of fields 2
Field:0 type:reflect.Value value:456
Field:1 type:reflect.Value value:56

Int() 和 String() 方法

Int 和 String 可以幫助我們分別取出 reflect.Value 作為 int64 和 string。

package main

import (
 "fmt"
 "reflect"
)

func main() {
 a := 56
 x := reflect.ValueOf(a).Int()
 fmt.Printf("type:%T value:%v\n", x, x)
 b := "Naveen"
 y := reflect.ValueOf(b).String()
 fmt.Printf("type:%T value:%v\n", y, y)

}

在 playground 上運(yùn)行

在上面程序中的第 10 行，我們?nèi)〕?reflect.Value，并轉(zhuǎn)換為 int64，而在第 13 行，我們?nèi)〕?reflect.Value 并將其轉(zhuǎn)換為 string。該程序會(huì)輸出：

type:int64 value:56
type:string value:Naveen

完整的程序

現(xiàn)在我們已經(jīng)具備足夠多的知識(shí)，來(lái)完成我們的查詢(xún)生成器了，我們來(lái)實(shí)現(xiàn)它把。

package main

import (
 "fmt"
 "reflect"
)

type order struct {
 ordId  int
 customerId int
}

type employee struct {
 name string
 id  int
 address string
 salary int
 country string
}

func createQuery(q interface{}) {
 if reflect.ValueOf(q).Kind() == reflect.Struct {
  t := reflect.TypeOf(q).Name()
  query := fmt.Sprintf("insert into %s values(", t)
  v := reflect.ValueOf(q)
  for i := 0; i < v.NumField(); i++ {
   switch v.Field(i).Kind() {
   case reflect.Int:
    if i == 0 {
     query = fmt.Sprintf("%s%d", query, v.Field(i).Int())
    } else {
     query = fmt.Sprintf("%s, %d", query, v.Field(i).Int())
    }
   case reflect.String:
    if i == 0 {
     query = fmt.Sprintf("%s\"%s\"", query, v.Field(i).String())
    } else {
     query = fmt.Sprintf("%s, \"%s\"", query, v.Field(i).String())
    }
   default:
    fmt.Println("Unsupported type")
    return
   }
  }
  query = fmt.Sprintf("%s)", query)
  fmt.Println(query)
  return

 }
 fmt.Println("unsupported type")
}

func main() {
 o := order{
  ordId:  456,
  customerId: 56,
 }
 createQuery(o)

 e := employee{
  name: "Naveen",
  id:  565,
  address: "Coimbatore",
  salary: 90000,
  country: "India",
 }
 createQuery(e)
 i := 90
 createQuery(i)

}

在 playground 上運(yùn)行

在第 22 行，我們首先檢查了傳來(lái)的參數(shù)是否是一個(gè)結(jié)構(gòu)體。在第 23 行，我們使用了 Name() 方法，從該結(jié)構(gòu)體的 reflect.Type 獲取了結(jié)構(gòu)體的名字。接下來(lái)一行，我們用 t 來(lái)創(chuàng)建查詢(xún)。

在第 28 行，case 語(yǔ)句 檢查了當(dāng)前字段是否為 reflect.Int，如果是的話，我們會(huì)取到該字段的值，并使用 Int() 方法轉(zhuǎn)換為 int64。if else 語(yǔ)句用于處理邊界情況。請(qǐng)?zhí)砑尤罩緛?lái)理解為什么需要它。在第 34 行，我們用來(lái)相同的邏輯來(lái)取到 string。

我們還作了額外的檢查，以防止 createQuery 函數(shù)傳入不支持的類(lèi)型時(shí)，程序發(fā)生崩潰。程序的其他代碼是自解釋性的。我建議你在合適的地方添加日志，檢查輸出，來(lái)更好地理解這個(gè)程序。

該程序會(huì)輸出：

insert into order values(456, 56)
insert into employee values("Naveen", 565, "Coimbatore", 90000, "India")
unsupported type

至于向輸出的查詢(xún)中添加字段名，我們把它留給讀者作為練習(xí)。請(qǐng)嘗試著修改程序，打印出以下格式的查詢(xún)。

insert into order(ordId, customerId) values(456, 56)

我們應(yīng)該使用反射嗎？

我們已經(jīng)展示了反射的實(shí)際應(yīng)用，現(xiàn)在考慮一個(gè)很現(xiàn)實(shí)的問(wèn)題。我們應(yīng)該使用反射嗎？我想引用 Rob Pike 關(guān)于使用反射的格言，來(lái)回答這個(gè)問(wèn)題。

清晰優(yōu)于聰明。而反射并不是一目了然的。

反射是 Go 語(yǔ)言中非常強(qiáng)大和高級(jí)的概念，我們應(yīng)該小心謹(jǐn)慎地使用它。使用反射編寫(xiě)清晰和可維護(hù)的代碼是十分困難的。你應(yīng)該盡可能避免使用它，只在必須用到它時(shí)，才使用反射。

本教程到此結(jié)束。希望你們喜歡。祝你愉快。希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持億速云。