Go 文件写入使用 bytes.Buffer

拼搏现实的明天。 2023-10-12 21:28 32阅读 0赞

#### 阅读文件 ####

*  使用 bytes.Buffer 缓冲区构建一个字符串
 *  resolv.conf

## 使用 bytes.Buffer 缓冲区构建一个字符串 ##

package main
    
    import (
    	"bytes"
    	"fmt"
    	"io"
    )
    
    func main() {
        
    	buf := new(bytes.Buffer) // 创建一个新的缓冲区
    
    	// 通过连续写入字符串构建内容
    	io.WriteString(buf, "Hello, ")
    	io.WriteString(buf, "World!")
    
    	// 从缓冲区中获取完整的字符串
    	result := buf.String()
    
    	fmt.Println(result) // 输出：Hello, World!
    }

PS E:\TEXT\test> go run .\main.go
    Hello, World!
    PS E:\TEXT\test>

## resolv.conf ##

# resolv.conf(5) file generated by
    # For more info, see
    # DO NOT EDIT THIS FILE BY HAND -- CHANGES WILL BE OVERWRITTEN
    
    nameserver 114.114.114.114
    nameserver 8.8.8.8
    nameserver 8.8.4.4
    search example.com example.org

package main
    
    import (
    	"bytes"
    	"fmt"
    	"io"
    	"log"
    
    	// net包为网络I/O提供了一个可移植的接口，包括TCP/IP、UDP、域名解析和Unix域套接字。
    	"net"
    	"os"
    )
    
    type Config struct {
        
    	Nameservers   []net.IP
    	SearchDomains []string
    }
    
    func (c *Config) Write(w io.Writer) error {
        
    	/*
    		bytes.Buffer 是 Go 语言标准库中的一种类型，
    		它实现了 io.Writer 和 io.Reader 接口，用于缓冲数据的读取和写入。
    	*/
    	buf := new(bytes.Buffer)
    	io.WriteString(buf, "# resolv.conf(5) file generated by\n")
    	io.WriteString(buf, "# For more info, seee\n")
    	io.WriteString(buf, "# DO NOT EDIT THIS FILE BY HAND -- CHANGES WILL BE OVERWRITTEN\n\n")
    	io.WriteString(buf, "nameserver 114.114.114.114\n")
    
    	// 遍历 Nameservers 切片，将每个 DNS 服务器写入配置文件
    	for _, ns := range c.Nameservers {
        
    		io.WriteString(buf, "nameserver ")
    		io.WriteString(buf, ns.String())
    		io.WriteString(buf, "\n")
    	}
    
    	// 如果存在 SearchDomains，则将其写入配置文件
    	if len(c.SearchDomains) > 0 {
        
    		io.WriteString(buf, "search")
    		for _, domain := range c.SearchDomains {
        
    			io.WriteString(buf, " ")
    			io.WriteString(buf, domain)
    		}
    		io.WriteString(buf, "\n")
    	}
    	_, err := w.Write(buf.Bytes())
    	return err
    }
    
    func main() {
        
    	// 创建一个 Config 实例
    	config := &Config{
        
    		Nameservers: []net.IP{
        
    			net.ParseIP("8.8.8.8"),
    			net.ParseIP("8.8.4.4"),
    		},
    		SearchDomains: []string{
        
    			"example.com",
    			"example.org",
    		},
    	}
    
    	// 使用文件作为输出
    	file, err := createFile("resolv.conf")
    	if err != nil {
        
    		log.Fatal(err)
    	}
    	defer file.Close()
    
    	// 将配置写入文件
    	err = config.Write(file)
    	if err != nil {
        
    		log.Fatal(err)
    	}
    
    	fmt.Println("resolv.conf 文件已成功生成.")
    }
    
    func createFile(filename string) (*os.File, error) {
        
    	return os.Create(filename)
    }