读取用户的输入
从键盘和标准输入 os.Stdin 读取输入,最简单的办法是使用 fmt 包提供的 Scan 和 Sscan 开头的函数
package mainimport "fmt"var (firstName, lastName, s stringi intf float32input = "56.12 / 5212 / Go"format = "%f / %d / %s")func main() {fmt.Println("Please enter your full name: ")fmt.Scanln(&firstName, &lastName)// fmt.Scanf("%s %s",&firstName, &lastName)fmt.Printf("Hi %s %s!\n", firstName, lastName)fmt.Sscanf(input, format, &f, &i, &s)fmt.Println("From the string we read: ", f, i, s)}
Scanln 扫描来自标准输入的文本,将空格分隔的值依次存放到后续的参数内,直到碰到换行。Scanf 与其类似,除了 Scanf 的第一个参数用作格式字符串,用来决定如何读取。Sscan 和以 Sscan 开头的函数则是从字符串读取,除此之外,与 Scanf 相同。
也可以使用bufio包提供的缓冲读取(buffered reader)来读取函数:
package mainimport ("bufio""fmt""os")// var inputReader *bufio.Reader// var input string// var err errorfunc main() {inputReader := bufio.NewReader(os.Stdin)fmt.Println("Please enter some input: ")input, err := inputReader.ReadString('\n')if err == nil {fmt.Printf("The input was: %s\n", input)}}
inputReader是一个指向bufio.Reader的指针。inputReader = bufio.NewReader(os.Stdin)这行代码,将会创建一个读取器,并将其与标准输入绑定。
bufio.NewReader()构建函数的签名为:func NewReader(rd io.Reader) *Reader,该函数的实参可以是满足io.Reader接口的任意对象,函数返回一个新的带缓冲的io.Reader的对象,它将从指定读取器(如 os.Stdin)读取内容。
返回的读取器对象提供一个方法ReadString(delim byte),该方法从输入中读取内容,直到碰到delim指定的字符,然后将读取到的内容连同delim字符一起放到缓冲区。
ReadString返回读取到的字符串,如果碰到错误则返回nil。如果它一直读取到文件结束,则返回读取到的字符串和io.EOF。如果读取过程中没有碰到delim字符,将返回错误err != nil。
文件读写
文件使用指向os.File类型的指针来表示的,也叫做文件句柄。前面使用到过标准输入os.Stdin和标准输出os.Stdout,它们的类型都是*os.File。
fileinput.go:
package mainimport ("bufio""fmt""io""os")func main() {path, _ := os.Getwd()fmt.Println("Path: ", path)inputFile, inputError := os.Open("MaGedu-Go/local_code/io/input.txt")if inputError != nil {fmt.Printf("An error occurred on opening the inputfile\n" + "Dose the file exist?\n" + "Have you got access to it?\n")return}defer inputFile.Close()inputReader := bufio.NewReader(inputFile)for {inputString, readerError := inputReader.ReadString('\n')fmt.Printf("The input was: %s", inputString)if readerError == io.EOF {return}}}
变量inputFile是*os.File类型的。该类型是一个结构,表示一个打开文件的描述符(文件句柄)。使用os.Open打开文件,该函数的参数是文件名,类型为string,上边的示例是以只读模式打开文件。
如果文件不存在或者没有足够的权限打开,Open函数会返回一个错误。如果文件打开正常,我们就使用defer inputFile.Close()语句确保在程序退出前关闭该文件。然后使用bufio.NewReader来获的一个读取器变量。
注意: 在使用ReadString和ReadBytes方法的时候,不需要关心操作系统的类型,直接使用\n就可以了,另外,也可以使用ReadLine()方法来实现相同的功能。
一旦读取到文件末尾,变量readerError的值将变成非空(事实上,其值为常量io.EOF),会执行return语句从而退出循环。
1.将整个文件的内容读到一个字符串里:
可以使用io/ioutil包里的ioutil.ReadFile()方法,该方法第一个返回值的类型是[]byte,里面存放读取到的内容,第二个返回值是错误,如果没有错误发生,第二个返回值为nil。
read_write_file1.go:
package mainimport ("fmt""io/ioutil""os")func main() {inputFile := "MaGedu-Go/local_code/io/input.txt"outputFile := "MaGedu-Go/local_code/io/output.txt"buf, err := ioutil.ReadFile(inputFile)if err != nil {fmt.Fprintf(os.Stderr, "File Error: %s\n", err)// panic(err.Error())}fmt.Printf("%s\n", string(buf))err = ioutil.WriteFile(outputFile, buf, 0644)if err != nil {panic(err.Error())}}
2.带缓冲的读取
在很多情况下,文件的内容是不安行划分的,或者干脆就是一个二进制文件。在这种情况下,ReadString()就无法使用了,可以使用bufio.Reader的Read(),它只接收一个参数:
buf := make([]byte, 1024)...n, err := inputReader.Read(buf)if (n == 0) { break }
3.按列读取文件中的数据
如果数据是按列排列并用空格分隔的,可以使用fmt包提供的以FScan开头的一系列函数来读取他们。以下示例,将3列的数据分别读入变量v1、v2和v3内,然后分别添加到切片的尾部。
package mainimport ("fmt""os")func main() {file, err := os.Open("MaGedu-Go/local_code/io/input.txt")if err != nil {panic(err)}defer file.Close()var col1, col2, col3 []stringfor {var v1, v2, v3 string_, err := fmt.Fscanln(file, &v1, &v2, &v3)// scans until newlineif err != nil {break}col1 = append(col1, v1)col2 = append(col2, v2)col3 = append(col3, v3)}fmt.Println(col1)fmt.Println(col2)fmt.Println(col3)}
注意: path包里包含一个子包叫filepath,其提供了跨平台的函数,用于处理文件名和路径。例如Base()函数用于获得路径中的最后一个元素(不包含后面的分隔符):
import "path/filepath"filename := filepath.Base(path)
compress包:读取压缩文件
支持的压缩文件格式为:bzip2、flate、gzip、lzw和zlib
gzipped.go:
package mainimport ("bufio""compress/gzip""fmt""os")func main() {fName := "test.gz"var r *bufio.Readerfi, err := os.Open(fName)if err != nil {fmt.Fprintf(os.Stderr, "%v, Can't open %s: error: %s\n", os.Args[0], fName, err)os.Exit(1)}defer fi.Close()fz, err := gzip.NewReader(fi)if err != nil {r = bufio.NewReader(fi)} else {r = bufio.NewReader(fz)}for {line, err := r.ReadString('\n')if err != nil {fmt.Println("Done reading file")os.Exit(0)}fmt.Println(line)}}
写文件
package mainimport ("bufio""fmt""os")func main() {outputFile, outputError := os.OpenFile("MaGedu-Go/local_code/io/output.txt", os.O_WRONLY|os.O_CREATE, 0666)if outputError != nil {fmt.Printf("An error\n")return}defer outputFile.Close()outputWriter := bufio.NewWriter(outputFile)outputString := "hello world\n"for i := 0; i < 10; i++ {outputWriter.WriteString(outputString)}outputWriter.Flush()}
OpenFile函数有三个参数:文件名、一个或多个标志(使用逻辑运算符”|”连接),使用的文件权限。
- os.O_RDONLY: 只读
- os.O_WRONLY: 只写
- os.O_CREATE: 创建,如果指定文件不存在,就创建该文件
- os.O_TRUNC: 截断,如果指定文件已存在,就将该文件的长度截为0
在读文件的时候,文件的权限是被忽略的,所以在使用OpenFile时传入的第三个参数可以用0。而在写文件时,不管是Unix还是Windows,都需要使用0666。
然后,创建一个写入器(缓冲区)对象:outputWriter := bufio.NewWriter(outputFile),将字符串写入缓冲区,outputWriter.WriteString(outputString),缓冲区的内容紧接着被完全写入文件:outputWriter.Flush()
如果写入的东西很简单,可以使用fmt.Fprintf(outputFile, “Some test data.\n”)直接将内容写入文件。fmt包里的F开头的Print函数可以直接写入任何io.Writer,包括文件。
示例:不适用fmt.Fprintf函数
filewrite.go:
package mainimport "os"func main() {os.Stdout.WriteString("hello, world\n")f, _ := os.OpenFile("test", os.O_CREATE|os.O_WRONLY, 0666)defer f.Close()f.WriteString("hello world\n")}
文件拷贝
package mainimport ("fmt""io""os")func CopyFile(dstName, srcName string) (writen int64, err error) {src, err := os.Open(srcName)if err != nil {return}defer src.Close()dst, err := os.Create(dstName)if err != nil {return}defer dst.Close()return io.Copy(dst, src)}func main() {CopyFile("MaGedu-Go/local_code/io/des.txt", "MaGedu-Go/local_code/io/src.txt")fmt.Println("Copy done!")}
从命令行读取参数
os包
os包中有一个string类型的切片变量os.Args,用来处理一些基本的命令行参数,它在程序启动后读取命令行输入的参数。
os_args.go:
package mainimport ("fmt""os""strings")func main() {who := "Alice"if len(os.Args) > 1 {who += strings.Join(os.Args[1:], " ")}fmt.Println("Good Morning", who)}
flag包
flag包有一个扩展功能用来解析命令行选项。但是通常被用来替换基本常量。
在flag包中有一个Flag被定义成一个含有如下字段的结构体:
type Flag struct {Name stringUsage stringValue ValueDefValue string}
echo.go:
package mainimport ("flag""os")var NewLine = flag.Bool("n", false, "print newline") // echo -n flag, of type *boolconst (Space = " "Newline = "\n")func main() {flag.PrintDefaults()// Scans the arg list and sets up flagsflag.Parse()s := ""for i := 0; i < flag.NArg(); i++ {if i > 0 {s += " "if *NewLine {s += Newline}}s += flag.Arg(i)}os.Stdout.WriteString(s)}
flag.Parse()扫描参数列表(或常量列表)并设置flag,flag.Arg(i)表示第i个参数。Parse()之后flag.Arg(i)全部可用,flag.Arg(0)就是第一个真实的flag,而不是像os.Args(0)放置程序的名字。
flag.Narg()返回参数的数量。解析后flag或常量就可用了。flag.Bool()定义了一个默认值是false的flag:当在命令行出现了第一个参数(这里是’n’),flag被设置成true(NewLine是_bool类型)。flag被解引用到 _NewLine,所以当值是true时将添加一个Newline
flag.PrintDefaults()打印flag的使用帮助信息,本例中打印的是:
-n=false: print newline
flag.VisitAll(fn func(*Flag))是另一个有用的功能:按照字典顺序遍历flag,并且对每个标签调用fn
对于flag.Bool可以设置布尔型flag来测试你的代码,例如定义一个flag processedFlag:
var processedFlag = flag.Bool("proc", false, "nothing processed yet")
在后面用如下代码来测试:
if *processedFlag { // fount flag -procr = process()}
用buffer读取文件
在下面的例子中,结合使用了缓冲读取文件和命令行 flag 解析这两项技术。如果不加参数,那么你输入什么屏幕就打印什么。
参数被认为是文件名,如果文件存在的话就打印文件内容到屏幕。命令行执行 cat test 测试输出。
cat.go:
package mainimport ("bufio""flag""fmt""io""os")func cat(r *bufio.Reader) {for {buf, err := r.ReadBytes('\n')fmt.Fprintf(os.Stdout, "%s", buf)if err == io.EOF {break}}return}func main() {flag.Parse()if flag.NArg() == 0 {cat(bufio.NewReader(os.Stdin))}for i := 0; i < flag.NArg(); i++ {f, err := os.Open(flag.Arg(i))if err != nil {fmt.Fprintf(os.Stderr, "%s: error reading from %s: %s\n", os.Args[0], flag.Arg(i), err.Error())continue}cat(bufio.NewReader(f))f.Close()}}
用切片读写文件
切片提供了 Go 中处理 I/O 缓冲的标准方式
cat2.go:
package mainimport ("flag""fmt""os")func cat(f *os.File) {const NBUF = 512var buf [NBUF]bytefor {switch nr, err := f.Read(buf[:]); true {case nr < 0:fmt.Fprintf(os.Stderr, "cat: error reading: %s\n", err.Error())os.Exit(1)case nr == 0: //EOFreturncase nr > 0:if nw, ew := os.Stdout.Write(buf[0:nr]); nw != nr {fmt.Fprintf(os.Stderr, "cat: error writing: %s\n", ew.Error())}}}}func main() {flag.Parse()if flag.NArg() == 0 {cat(os.Stdin)}for i := 0; i < flag.NArg(); i++ {f, err := os.Open(flag.Arg(i))if err != nil {fmt.Fprintf(os.Stderr, "%s: error reading from %s: %s\n", os.Args[0], flag.Arg(i), err.Error())os.Exit(1)}cat(f)f.Close()}}
使用接口的实际例子:fmt.Fprintf
io_interfaces.go:
package mainimport ("bufio""fmt""os")func main() {// unbufferedfmt.Fprintf(os.Stdout, "%s\n", "hello world! - unbuffered")// buffered: os.Stdout implements io.Writerbuf := bufio.NewWriter(os.Stdout)fmt.Fprintf(buf, "%s\n", "hello world! - buffered")buf.Flush()}
fmt.Fprintf(),func Fprintf(w io.Writer, format string, a ...interface{}) (n int, err error)
不是写入一个文件,而是写入一个io.Writer接口类型的变量,下面是Writer接口在io包中的定义:
type Writer interface {Write(p []byte) (n int, err error)}
fmt.Fprintf() 依据指定的格式向第一个参数内写入字符串,第一个参数必须实现了 io.Writer 接口。Fprintf() 能够写入任何类型,只要其实现了 Write 方法,包括 os.Stdout,文件(例如 os.File),管道,网络连接,通道等等,同样的也可以使用 bufio 包中缓冲写入。bufio 包中定义了 type Writer struct{…} 。
bufio.Writer 实现了 Write 方法:
func (b *Writer) Write(p []byte) (nn int, err error)
它还有一个工厂函数:传给它一个 io.Writer 类型的参数,它会返回一个带缓冲的 bufio.Writer 类型的 io.Writer :
func NewWriter(wr io.Writer) (b *Writer)
适合任何形式的缓冲写入。
在缓冲写入的最后千万不要忘了使用 Flush(),否则最后的输出不会被写入。
JSON数据格式
通过把数据转换成纯文本,使用命名的字段来标注,让其具有可读性。这样的数据格式可以通过网络传输,而且是与平台无关的,任何类型的应用都能够读取和输出,不与操作系统和编程语言的类型相关。
术语说明:
- 数据结构 —> 指定格式 = 序列化或编码(传输之前)
- 指定格式 —> 数据结构 = 反序列化或解码(传输之后)
序列化是在内存中把数据转换成指定格式(data -> string),反之亦然(string -> data)。
编码也是一样的,只是输出一个数据流(实现了 io.Writer 接口);解码是从一个数据流(实现了 io.Reader)输出到一个数据结构。
json.go:
package mainimport ("encoding/json""fmt""log""os")type Address struct {Type stringCity stringCountry string}type VCard struct {FirstName stringLastName stringAddresses []*AddressRemark string}func main() {pa := &Address{"private", "Aartselaar", "Belgiunm"}wa := &Address{"work", "Boom", "Belgium"}vc := VCard{"Jan", "Kersschot", []*Address{pa, wa}, "none"}js, _ := json.Marshal(vc)fmt.Printf("JSON format: %s\n", js)file, _ := os.OpenFile("vcard.json", os.O_CREATE|os.O_WRONLY, 0666)defer file.Close()enc := json.NewEncoder(file)err := enc.Encode(vc)if err != nil {fmt.Println(err)log.Println("Error in encoding json")}}
出于安全考虑,在web应用中最好使用json.MarshalforHTML(),其对数据执行HTML转码,所以文本可以被安全地嵌在HTMLscript标签中。
不是所有的数据都可以编码为JSON类型,只有验证通过的数据结构才能被编码:
- JSON对象只支持字符串类型的key;要编码一个Go map类型,map必须是map[string]T(T是json包中支持的任何类型)
- Channel,复杂类型和函数类型不能被编码
- 不支持循环数据结构;它将引起序列化进入一个无限循环
- 指针可以被编码,实际上是对指针指向的值进行编码(或者指针是nil)
反序列化
json.Unmarshal()的函数签名是func Unmarshal(data []byte, v interface{}) error 把JSON解码为数据结构。
上边示例中对vc编码后的数据为js,对其解码时,首先创建结构VCard用来保存解码的数据:var v VCard并调用json.Unmarshal(js,&v),解析[]byte中的JSON数据并将结果存入指针&v指向的值
反射能够让JSON字段去尝试匹配目标结构字段;但是只有真正匹配上的字段才会填充数据。字段没有匹配不会报错,而是直接忽略掉。
package mainimport ("bytes""encoding/json""fmt""strings")type Person struct {Name string `json: "name"`Age int `json: "age"`}func main() {// 1.编码person1 := Person{"张三", 24}bytes1, err := json.Marshal(&person1)if err == nil {// 返回是的字节数组 []bytefmt.Println("json.Marshal 编码结果:", string(bytes1))}// 2.解码str := `{"name":"李四", "age":25}`// json.Unmarshal 需要字节数组参数,需要把字符串转为 []byte 类型bytes2 := []byte(str)var person2 Personif json.Unmarshal(bytes2, &person2) == nil {fmt.Println("json.Unmarshal 解码结果:", person2.Name, person2.Age)}// 3.使用json.NewEncoder 编码person3 := Person{"王五", 26}// 编码结果暂存到bufferbytes3 := new(bytes.Buffer)_ = json.NewEncoder(bytes3).Encode(person3)if err == nil {fmt.Print("json.NewEncoder 编码结果:", string(bytes3.Bytes()))}// 4.使用json.NewDecoder 解码str4 := `{"name":"赵六", "age": 27}`var person4 Personerr = json.NewDecoder(strings.NewReader(str4)).Decode(&person4)if err == nil {fmt.Println("json.NewDecoder 解码结果:", person4.Name, person4.Age)}}
解码任意的数据:
json包使用map[string]interface()和[]interface()存储任意的JSON对象和数组;其可以被反序列化为任何的JSON blob存储到接口值中。
b := []byte({"Name": "Wednesday", "Age": 6, "Parents": ["Gomez", "Morticia"]})
不用理解这个数据的结构,可以直接使用Unmarshal把这个数据编码并保存在接口值中:
var f interface{}err := json.Unmarshal(b, &f)
f指向的值是一个map,key是一个字符串,value是自身存储作为空接口类型的值:
map[string]interface{} {"Name": "Wednesday","Age": 6,"Parenets": []interface{} {"Gomez","Morticia",},}
要访问这个数据,可以使用类型断言:m := f.(map[string]interface{})
可以通过for range语法和type switch来访问其实际类型:
for k, v := range m {switch vv := v.(type) {case string:fmt.Println(k, "is string", vv)case int:fmt.Println(k, "is int", vv)case []interface{}:fmt.Println(k, "is an array:")for i, u := range vv {fmt.Println(i, u)}default:fmt.Println(k, "is of a type")}}
解码数据到结构体
如果事先知道JSON数据,可以定义一个适当的结构并对JSON数据反序列化:
type FamilyMember struct {Name stringAge intParents []string}var m FamilyMembererr := json.Unmarshal(b,&m)
编码和解码流
json包提供Decoder和Encoder类型来支持常用JSON数据流读写。NewDecoder和NewEncoder函数分别封装了io.Reader和io.Writer接口。
func NewDecoder(r io.Reader) *Decoderfunc NewEncoder(w io.Writer) *Encoder
要想把JSON直接写入文件,可以使用json.NewEncoder初始化文件(或者任何实现 io.Writer 的类型),并调用 Encode();反过来与其对应的是使用 json.NewDecoder 和 Decode() 函数:
func NewDecoder(r io.Reader) *Decoderfunc (dec *Decoder) Decode(v interface{}) error
接口是如何对实现进行抽象的:数据结构可以是任何类型,只要其实现了某种接口,目标或源数据要能够被编码就必须实现 io.Writer 或 io.Reader 接口。由于 Go 语言中到处都实现了 Reader 和 Writer,因此 Encoder 和 Decoder 可被应用的场景非常广泛,例如读取或写入 HTTP 连接、websockets 或文件。
XML数据格式
// xml.gopackage mainimport ("encoding/xml""fmt""strings")var t, token xml.Tokenvar err errorfunc main() {input := "<Person><FirstName>Laura</FirstName><LastName>Lynn</LastName></Person>"inputReader := strings.NewReader(input)p := xml.NewDecoder(inputReader)for t, err = p.Token(); err == nil; t, err = p.Token() {switch token := t.(type) {case xml.StartElement:name := token.Name.Localfmt.Printf("Token name: %s\n", name)for _, attr := range token.Attr {attrName := attr.Name.LocalattrValue := attr.Valuefmt.Printf("An attribute is: %s %s\n", attrName, attrValue)// ...}case xml.EndElement:fmt.Println("End of token")case xml.CharData:content := string([]byte(token))fmt.Printf("This is the content: %v\n", content)// ...default:// ...}}}
加密
通过网络传输的数据必须加密,以防止被 hacker(黑客)读取或篡改,并且保证发出的数据和收到的数据检验和一致。
- hash 包:实现了 adler32、crc32、crc64 和 fnv 校验。
- crypto 包:实现了其它的 hash 算法,比如 md4、md5、sha1 等。以及完整地实现了 aes、blowfish、rc4、rsa、xtea 等加密算法。
hash_sha1.go:
package mainimport ("crypto/sha1""fmt""io""log")func main() {hasher := sha1.New()io.WriteString(hasher, "test")b := []byte{}fmt.Printf("Result: %x\n", hasher.Sum(b))fmt.Printf("Result: %d\n", hasher.Sum(b))hasher.Reset()data := []byte("We shall overcome!")n, err := hasher.Write(data)if n != len(data) || err != nil {log.Printf("Hash write error: %v / %v", n, err)}checksum := hasher.Sum(b)fmt.Printf("Result: %x\n", checksum)}
