使用延迟释放文件句柄
文件的操作需要经过打开文件、获取和操作文件资源、关闭资源几个过程,如果在操作完毕后不关闭文件资源,进程将一直无法释放文件资源,在下面的例子中将实现根据文件名获取文件大小的函数,函数中需要打开文件、获取文件大小和关闭文件等操作,由于每一步系统操作都需要进行错误处理,而每一步处理都会造成一次可能的退出,因此就需要在退出时释放资源,而我们需要密切关注在函数退出处正确地释放文件资源,参考下面的代码:
- // 根据文件名查询其大小
- func fileSize(filename string) int64 {
-
- // 根据文件名打开文件, 返回文件句柄和错误
- f, err := os.Open(filename)
-
- // 如果打开时发生错误, 返回文件大小为0
- if err != nil {
- return 0
- }
-
- // 取文件状态信息
- info, err := f.Stat()
-
- // 如果获取信息时发生错误, 关闭文件并返回文件大小为0
- if err != nil {
- f.Close()
- return 0
- }
-
- // 取文件大小
- size := info.Size()
-
- // 关闭文件
- f.Close()
-
- // 返回文件大小
- return size
- }
代码说明如下:
- 第 2 行,定义获取文件大小的函数,返回值是 64 位的文件大小值。
- 第 5 行,使用 os 包提供的函数 Open(),根据给定的文件名打开一个文件,并返回操作文件用的句柄和操作错误。
- 第 8 行,如果打开的过程中发生错误,如文件没找到、文件被占用等,将返回文件大小为 0。
- 第 13 行,此时文件句柄 f 可以正常使用,使用 f 的方法 Stat() 来获取文件的信息,获取信息时,可能也会发生错误。
- 第 16~19 行对错误进行处理,此时文件是正常打开的,为了释放资源,必须要调用 f 的 Close() 方法来关闭文件,否则会发生资源泄露。
- 第 22 行,获取文件大小。
- 第 25 行,关闭文件、释放资源。
- 第 28 行,返回获取到的文件大小。
在上面的例子中,第 25 行是对文件的关闭操作,下面使用 defer 对代码进行简化,代码如下:
- func fileSize(filename string) int64 {
-
- f, err := os.Open(filename)
-
- if err != nil {
- return 0
- }
-
- // 延迟调用Close, 此时Close不会被调用
- defer f.Close()
-
- info, err := f.Stat()
-
- if err != nil {
- // defer机制触发, 调用Close关闭文件
- return 0
- }
-
- size := info.Size()
-
- // defer机制触发, 调用Close关闭文件
- return size
- }
代码中加粗部分为对比前面代码而修改的部分,代码说明如下:
- 第 10 行,在文件正常打开后,使用 defer,将 f.Close() 延迟调用,注意,不能将这一句代码放在第 4 行空行处,一旦文件打开错误,f 将为空,在延迟语句触发时,将触发宕机错误。
- 第 16 行和第 22 行,defer 后的语句(f.Close())将会在函数返回前被调用,自动释放资源。