gopl-zh.github.com/ch7/ch7-12.md

81 lines
4.7 KiB
Markdown
Raw Permalink Normal View History

2016-02-15 03:06:34 +00:00
## 7.12. 通过类型断言询问行为
2018-06-09 16:27:25 +00:00
下面这段逻辑和net/http包中web服务器负责写入HTTP头字段例如"Content-type:text/html"的部分相似。io.Writer接口类型的变量w代表HTTP响应写入它的字节最终被发送到某个人的web浏览器上。
2016-01-18 15:27:55 +00:00
```go
func writeHeader(w io.Writer, contentType string) error {
2016-01-21 02:22:10 +00:00
if _, err := w.Write([]byte("Content-Type: ")); err != nil {
return err
}
if _, err := w.Write([]byte(contentType)); err != nil {
return err
}
// ...
2016-01-18 15:27:55 +00:00
}
```
2016-02-15 03:06:34 +00:00
因为Write方法需要传入一个byte切片而我们希望写入的值是一个字符串所以我们需要使用[]byte(...)进行转换。这个转换分配内存并且做一个拷贝但是这个拷贝在转换后几乎立马就被丢弃掉。让我们假装这是一个web服务器的核心部分并且我们的性能分析表示这个内存分配使服务器的速度变慢。这里我们可以避免掉内存分配么
2015-12-09 07:45:11 +00:00
2016-10-18 05:20:38 +00:00
这个io.Writer接口告诉我们关于w持有的具体类型的唯一东西就是可以向它写入字节切片。如果我们回顾net/http包中的内幕我们知道在这个程序中的w变量持有的动态类型也有一个允许字符串高效写入的WriteString方法这个方法会避免去分配一个临时的拷贝。这可能像在黑夜中射击一样但是许多满足io.Writer接口的重要类型同时也有WriteString方法包括`*bytes.Buffer``*os.File`和`*bufio.Writer`。)
2016-01-18 15:27:55 +00:00
2016-02-15 03:06:34 +00:00
我们不能对任意io.Writer类型的变量w假设它也拥有WriteString方法。但是我们可以定义一个只有这个方法的新接口并且使用类型断言来检测是否w的动态类型满足这个新接口。
2016-01-18 15:27:55 +00:00
```go
// writeString writes s to w.
// If w has a WriteString method, it is invoked instead of w.Write.
func writeString(w io.Writer, s string) (n int, err error) {
2016-01-21 02:22:10 +00:00
type stringWriter interface {
WriteString(string) (n int, err error)
}
if sw, ok := w.(stringWriter); ok {
return sw.WriteString(s) // avoid a copy
}
return w.Write([]byte(s)) // allocate temporary copy
2016-01-18 15:27:55 +00:00
}
func writeHeader(w io.Writer, contentType string) error {
2016-01-21 02:22:10 +00:00
if _, err := writeString(w, "Content-Type: "); err != nil {
return err
}
if _, err := writeString(w, contentType); err != nil {
return err
}
// ...
2016-01-18 15:27:55 +00:00
}
```
为了避免重复定义我们将这个检查移入到一个实用工具函数writeString中但是它太有用了以致于标准库将它作为io.WriteString函数提供。这是向一个io.Writer接口写入字符串的推荐方法。
2016-01-18 15:27:55 +00:00
这个例子的神奇之处在于没有定义了WriteString方法的标准接口也没有指定它是一个所需行为的标准接口。一个具体类型只会通过它的方法决定它是否满足stringWriter接口而不是任何它和这个接口类型所表达的关系。它的意思就是上面的技术依赖于一个假设这个假设就是如果一个类型满足下面的这个接口然后WriteString(s)方法就必须和Write([]byte(s))有相同的效果。
2016-01-18 15:27:55 +00:00
```go
interface {
2016-01-21 02:22:10 +00:00
io.Writer
WriteString(s string) (n int, err error)
2016-01-18 15:27:55 +00:00
}
```
尽管io.WriteString实施了这个假设但是调用它的函数极少可能会去实施类似的假设。定义一个特定类型的方法隐式地获取了对特定行为的协约。对于Go语言的新手特别是那些来自有强类型语言使用背景的新手可能会发现它缺乏显式的意图令人感到混乱但是在实战的过程中这几乎不是一个问题。除了空接口interface{},接口类型很少意外巧合地被实现。
2016-01-18 15:27:55 +00:00
上面的writeString函数使用一个类型断言来获知一个普遍接口类型的值是否满足一个更加具体的接口类型并且如果满足它会使用这个更具体接口的行为。这个技术可以被很好的使用不论这个被询问的接口是一个标准如io.ReadWriter或者用户定义的如stringWriter接口。
2016-01-18 15:27:55 +00:00
2016-02-15 03:06:34 +00:00
这也是fmt.Fprintf函数怎么从其它所有值中区分满足error或者fmt.Stringer接口的值。在fmt.Fprintf内部有一个将单个操作对象转换成一个字符串的步骤像下面这样
2016-01-18 15:27:55 +00:00
```go
package fmt
func formatOneValue(x interface{}) string {
2016-01-21 02:22:10 +00:00
if err, ok := x.(error); ok {
return err.Error()
}
if str, ok := x.(Stringer); ok {
return str.String()
}
// ...all other types...
2016-01-18 15:27:55 +00:00
}
```
如果x满足这两个接口类型中的一个具体满足的接口决定对值的格式化方式。如果都不满足默认的case或多或少会统一地使用反射来处理所有的其它类型我们可以在第12章知道具体是怎么实现的。
2016-01-18 15:27:55 +00:00
再一次的它假设任何有String方法的类型都满足fmt.Stringer中约定的行为这个行为会返回一个适合打印的字符串。