码迷,mamicode.com
首页 > 其他好文 > 详细

读写操作(补充)

时间:2019-05-07 23:03:30      阅读:205      评论:0      收藏:0      [点我收藏+]

标签:扫描   信息   功能   return   内存   bytes   com   动态   类型转换   

读写操作(补充)

之前的一篇有关终端读写的内容比较详细:
https://blog.51cto.com/steed/2315597

这篇,先补充了两个从终端逐行扫描并处理的示例。然后再补充了有关两个写入操作的内容。

bufio包,逐行读取

关于bufio包,使用它可以简便和高效地处理输入和输出。其中一个最有用的特性是称为扫描器(Scanner)的类型,它可以读取输入,以行或者单词为单位断开,这是处理以行为单位输入内容的最简单方式。
先声明一个 bufio.Scanner 类型的变量:

input := bufio.NewScanner(os.Stdin)

扫描器从程序的标准输入进行读取。每一次调用 inout.Scan() 读取下一行,并且将结尾的换行符去掉;通过调用 input.Text() 来获取读到的内容。Scan 方法在读到新行的时候返回true,在没有更多内容的时候返回flase。
下面的例子,从标准输入获取到内容后,转成全大写再打印出来。Windows下使用 Ctrl+Z 后回车可以退出:

package main

import (
    "bufio"
    "fmt"
    "os"
    "strings"
)

func main() {
    input := bufio.NewScanner(os.Stdin)
    for input.Scan() {
        fmt.Println(strings.ToUpper(input.Text()))
    }
}

上面的示例是每从标准输入获取到一行的内容,就进行处理。还有一种做法是,从标准输入获取全部的内容后(Ctrl+Z 后回车表示输入完成),最后再一次全部输出。下面的示例是从标准输入获取到全部内容,然后全部输出,每一行的内容之间插入一个空格。最后也会有一个空格,输出时把最后一个空格截断:

package main

import (
    "bufio"
    "bytes"
    "fmt"
    "os"
)

func main() {
    input := bufio.NewScanner(os.Stdin)
    buff := bytes.NewBuffer(nil)
    for input.Scan() {
        buff.Write(input.Bytes()) // 忽略错误
        buff.WriteByte(‘ ‘)
    }
    fmt.Printf("%q\n", buff.String()[:buff.Len()-1])  // 截掉最后一个空格
}

读取内容并丢弃

使用io.Copy函数读取响应的内容,比如直接复制内容到标准输出,这样就不需要把数据流装到缓冲区:

n, err := io.Copy(os.Stdout, resp.Body)

还可以通过写入 ioutil.Discard 输出流进行丢弃,这样做应该是为了要有一个读取的过程:

n, err := io.Copy(ioutil.Discard, resp.Body)

向 io.Writer 写入字符串

这里要讲的是通过接口类型断言来查询特性,下面是一个标准库中使用的示例。并且,这个是向 io.Writer 写入字符串的推荐方法。
下面定义一个方法,往 io.Writer 接口接入字符串信息:

func writeMsg(w io.Writer, msg string) error {
    if _, err := w.Write([]byte(msg)); err != nil {
        return err
    }
    return nil
}

因为 Write 方法需要一个字节切片([]byte),而需要写入的是一个字符串,所以要做类型转换。这种转换需要进行内存分配和内存复制,但复制后内存又会被马上抛弃。这里就会有性能问题,这个内存分配会导致性能下降,需要避开这个内存分配。
在很多包中,实现了 io.Writer 的重要类,都会提供一个对应的高效写入字符串的 WriteString 方法,比如:

  • *http.response : 这个类型不可导出,我们一般使用的是 http.ResponseWriter 接口
  • *bytes.Buffer
  • *os.File
  • *bufio.Write

由于 Writer 接口并不包括 WriteString 方法,不能直接调用。这里可以先定义一个新的接口,这个接口只包含 WriteString 方法,然后使用类型断言来判断 w 的动态类型是否满足这个新接口:

// 将s写入w,如果w有WriteString方法,就直接调用该方法
func writeString(w io.Writer, s string) (n int, err error) {
    type stringWriter interface {
        WriteString(string) (n int, err error)
    }
    if sw, ok := w.(stringWriter); ok {
        return sw.WriteString(s) // 避免了内存复制
    }
    return w.Write([]byte(s)) // 分配了临时内存
}

func writeMsg(w io.Writer, msg string) error {
    if _, err := writeString(w, msg); err != nil {
        return err
    }
    return nil
}

由于上面的操作太常见了,io 包已经提供了一个函数 io.WriteString 可以直接使用。上面的代码可以简化为如下的方式:

func writeMsg2(w io.Writer, msg string) error {
    if _, err := io.WriteString(w, msg); err != nil {
        return err
    }
    return nil
}

这里本质上还是一样的,接口的定义、类型检查都封装到了io包的内部,并且内部实现的逻辑和上面是一样的。
之前想要向某个具体的类型写入字符串的时候,会查看该类型是否有 WriteString 方法。但是如果要操作的类型是 io.Writer 接口,虽然实际背后的动态类型还是那个类型,但是就无法调用 WriteString 方法了。这时可以直接使用 io 包提供的工具函数完成同样的操作。

通过 fmt.Fprint 写入

fmt.Fprint 系列的函数,接收的第一个参数是 io.Writer,也可以进行写入操作。而且 fmt 包提供的函数,接收空接口,可以是任何的类型,自带格式化成字符串的功能,很多时候更方便。
在向 os.Stdout 打印输出错误,以及 Web 服务器向客户端发送数据的时候使用的比较多,不过同样的也是可以向任何的 io.Writer 写入内容的,比如写入到文件。

读写操作(补充)

标签:扫描   信息   功能   return   内存   bytes   com   动态   类型转换   

原文地址:https://blog.51cto.com/steed/2390679

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!