Go 文件操作

package main

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

func main() {
	// === bufio.Scanner 按行读取 ===
	text := "第一行\n第二行\n第三行\n第四行"
	scanner := bufio.NewScanner(strings.NewReader(text))

	lineNum := 0
	for scanner.Scan() {
		lineNum++
		fmt.Printf("第 %d 行: %s\n", lineNum, scanner.Text())
	}
	if err := scanner.Err(); err != nil {
		fmt.Println("扫描错误:", err)
	}

	// === bufio.Reader 读取文件 ===
	// 先创建一个示例文件
	os.WriteFile("/tmp/go-demo/bufio.txt", []byte("line1\nline2\nline3\n"), 0644)

	file, err := os.Open("/tmp/go-demo/bufio.txt")
	if err != nil {
		fmt.Println("打开失败:", err)
		return
	}
	defer file.Close()

	reader := bufio.NewReader(file)
	for {
		line, err := reader.ReadString('\n')
		fmt.Printf("读取: %s", line)
		if err != nil {
			break
		}
	}

	// === bufio.Writer 缓冲写入 ===
	var buf strings.Builder
	writer := bufio.NewWriter(&buf)

	writer.WriteString("缓冲写入第一行\n")
	writer.WriteString("缓冲写入第二行\n")
	writer.Flush() // 必须 Flush 才能将缓冲内容写出

	fmt.Println("\n--- 缓冲写入结果 ---")
	fmt.Print(buf.String())
}

package main

import (
	"bufio"
	"fmt"
	"io"
	"os"
)

// copyFile 使用 io.Copy 实现文件复制
func copyFile(src, dst string) error {
	sourceFile, err := os.Open(src)
	if err != nil {
		return fmt.Errorf("打开源文件失败: %w", err)
	}
	defer sourceFile.Close()

	destFile, err := os.Create(dst)
	if err != nil {
		return fmt.Errorf("创建目标文件失败: %w", err)
	}
	defer destFile.Close()

	n, err := io.Copy(destFile, sourceFile)
	if err != nil {
		return fmt.Errorf("复制失败: %w", err)
	}

	fmt.Printf("复制完成，共 %d 字节\n", n)
	return nil
}

// copyFileBuffered 使用 bufio 缓冲复制大文件
func copyFileBuffered(src, dst string) error {
	sourceFile, err := os.Open(src)
	if err != nil {
		return err
	}
	defer sourceFile.Close()

	destFile, err := os.Create(dst)
	if err != nil {
		return err
	}
	defer destFile.Close()

	reader := bufio.NewReader(sourceFile)
	writer := bufio.NewWriter(destFile)

	if _, err := io.Copy(writer, reader); err != nil {
		return err
	}

	return writer.Flush() // 确保缓冲数据写入磁盘
}

func main() {
	// 创建示例文件
	os.MkdirAll("/tmp/go-demo", 0755)
	os.WriteFile("/tmp/go-demo/original.txt", []byte("这是原始文件的内容。\n包含多行文本。\n"), 0644)

	// 复制文件
	err := copyFile("/tmp/go-demo/original.txt", "/tmp/go-demo/copy.txt")
	if err != nil {
		fmt.Println("错误:", err)
		return
	}

	// 验证复制结果
	data, _ := os.ReadFile("/tmp/go-demo/copy.txt")
	fmt.Printf("复制的文件内容:\n%s", string(data))
}

package main

import (
	"fmt"
	"os"
	"path/filepath"
)

func main() {
	baseDir := "/tmp/go-demo/dirs"

	// 创建多级目录
	err := os.MkdirAll(baseDir+"/a/b/c", 0755)
	if err != nil {
		fmt.Println("创建目录失败:", err)
		return
	}
	fmt.Println("目录创建成功")

	// 在子目录中创建文件
	os.WriteFile(baseDir+"/a/file1.txt", []byte("file1"), 0644)
	os.WriteFile(baseDir+"/a/b/file2.txt", []byte("file2"), 0644)

	// 读取目录内容
	entries, err := os.ReadDir(baseDir + "/a")
	if err != nil {
		fmt.Println("读取目录失败:", err)
		return
	}

	fmt.Println("\n--- 目录内容 ---")
	for _, entry := range entries {
		info, _ := entry.Info()
		typeStr := "文件"
		if entry.IsDir() {
			typeStr = "目录"
		}
		fmt.Printf("  %s [%s] %d 字节\n", entry.Name(), typeStr, info.Size())
	}

	// 递归遍历目录树
	fmt.Println("\n--- 递归遍历 ---")
	filepath.Walk(baseDir, func(path string, info os.FileInfo, err error) error {
		if err != nil {
			return err
		}
		indent := ""
		for i := 0; i < len(path)-len(baseDir); i++ {
			if path[i] == '/' {
				indent += "  "
			}
		}
		fmt.Printf("%s%s\n", indent, info.Name())
		return nil
	})

	// 删除目录（递归删除）
	err = os.RemoveAll(baseDir)
	if err != nil {
		fmt.Println("删除失败:", err)
		return
	}
	fmt.Println("\n目录已删除")
}

package main

import (
	"fmt"
	"os"
)

// 错误示例：忘记关闭文件
func badExample() {
	file, _ := os.Open("somefile.txt")
	// 忘记调用 file.Close()
	// 文件描述符泄漏！
	_ = file
}

// 正确示例：使用 defer 关闭文件
func goodExample() {
	file, err := os.Open("somefile.txt")
	if err != nil {
		fmt.Println("打开失败:", err)
		return
	}
	defer file.Close() // 函数返回时自动关闭

	// 使用 file...
}

// 最佳实践：封装文件操作函数
func readFirstLine(filename string) (string, error) {
	file, err := os.Open(filename)
	if err != nil {
		return "", err
	}
	defer file.Close()

	buf := make([]byte, 1024)
	n, err := file.Read(buf)
	if err != nil && err.Error() != "EOF" {
		return "", err
	}
	return string(buf[:n]), nil
}

func main() {
	// 创建测试文件
	os.WriteFile("/tmp/go-demo/somefile.txt", []byte("第一行内容"), 0644)

	line, err := readFirstLine("/tmp/go-demo/somefile.txt")
	if err != nil {
		fmt.Println("读取失败:", err)
		return
	}
	fmt.Println("内容:", line)
}