写在前面
在上一篇文章《Golang中的读写数据（上）》当中，我们介绍了Golang中一些简单的文件读写、拷贝操作，接下来，我们将继续学习Golang中的读写数据的相关知识点。
从命令行读取参数
在一些场景当中，我们在执行一个软件，有时候需要传入一些初始化的信息，例如连接数据库的username和password等属性。那么，在Golang中是如何读取参数的呢？请看下面的例子:
使用os.Args 获取参数
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
| package main
import (
"fmt"
"os"
)
func main() {
if len(os.Args) > 1 {
for i:=1;i<len(os.Args);i++ {
sayHello(os.Args[i])
}
}else {
fmt.Println("参数为空")
}
}
func sayHello(name... string) {
fmt.Println("hello!",name)
}
|
输出：
注意：
这个命令行参数会放置在切片 os.Args[] 中（以空格分隔），从索引1开始（os.Args[0] 放的是程序本身的名字）。函数 strings.Join 以空格为间隔连接这些参数。
使用flag.Args获取参数
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
| package main
import (
"flag"
"fmt"
)
func main() {
flag.Parse()
for _, arg := range flag.Args() {
sayHello(arg)
}
}
func sayHello(name... string) {
fmt.Println("hello!",name)
}
|
输出：
注意：
flag.Arg(0) 就是第一个真实的 flag，而不是像 os.Args(0) 放置程序的名字。
一个例子：使用缓存读取文件与flag.Args的综合应用
接触过Linux系统的小伙伴们应该经常会cat filename这个命令，接下来，我们将使用Golang实现这个小工具.
需求描述
我们需要一个小工具来打印文件的内容，当文件不存在时给用户一个友好的提示。
功能设计
- 根据用户输入的文件名读取文件到缓存并分批次打印
- 当文件不存在时输出”不存在文件 {文件名}”
代码：
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
| package main
import (
"bufio"
"flag"
"fmt"
"io"
"os"
)
func main() {
flag.Parse()
filename :=flag.Arg(0)
if filename=="" {
fmt.Println("Command Usage Format: mycat filename")
return
}
open, err := os.Open(filename)
defer open.Close()
if err!=nil {
fmt.Println(err)
return
}
myCat(bufio.NewReader(open))
}
func myCat(reader *bufio.Reader) {
for {
buf, err := reader.ReadBytes('\n')
fmt.Fprintf(os.Stdout, "%s\n", buf)
if err == io.EOF {
break
}
}
return
}
|
测试：
- 当文件存在时打印文件内容：
- 当文件不存在时给出提示：
提个问题：这个程序能满足需求吗？
大家可以思考一下，如果被打印的文件是一个只有一行数据的大文件，会出现什么情况？
因此，为了避免出现这个情况，我们将对mycat进行改造。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
| package main
import (
"bufio"
"flag"
"fmt"
"io"
"os"
)
func main() {
flag.Parse()
filename :=flag.Arg(0)
if filename=="" {
fmt.Println("Command Usage Format: mycat filename")
return
}
file, err := os.Open(filename)
defer file.Close()
if err!=nil {
fmt.Println(err)
return
}
myCatV2(bufio.NewReader(file))
}
func myCatV2(reader *bufio.Reader) {
buf := make([]byte,512)
for {
n, err := reader.Read(buf)
fmt.Fprintf(os.Stdout,"%s",buf[0:n])
if err ==io.EOF {
break
}
}
return
}
|
在这一版当中，我们指定了缓存区的大小，它是一个512位的byte数组。
输出：
我们指定了缓冲区的大小，这样就避免了将整个文件都加载到内存当中。
当然，大家可以继续对程序进行改进，例如引入协程等技术，这便不在本文的讨论当中了。
重要的一点：使用defer关闭文件
在前面的文章当中，我们介绍了defer关键字的作用：他将在函数退出时（return之后）执行其修饰的语句。在这里，我们使用其来在main函数退出前关闭文件。
写在最后
关于Golang中读取命令行参数以及与文件读写操作的综合应用我们就介绍到这里。本文当中涉及到的例子可以点击此处下载。如果我的学习笔记能够给你带来帮助，还请多多点赞鼓励。文章如有错漏之处还请各位小伙伴帮忙斧正。
在下一篇文章当中我们将一起来学习Golang中的数据格式化以及数据网络传输等内容。