使用Go开发一个文件同步小工具(附源码)

package main

import (
	"fmt"
	"io"
	"os"
	"path/filepath"
	"strings"
	"sync"
	"time"
)

var exit = false           // 退出状态控制
var wg sync.WaitGroup      // 保证正常退出

func main() {
	fmt.Println("FileSync soft run")
	if len(os.Args) < 3 {
		fmt.Println("Usage: FileSync <source_dir> <target_dir>")
		return
	}
	fromDir := os.Args[1]
	toDir := os.Args[2]
	wg.Add(1)
	go mainRun(fromDir, toDir) // 启动工作主线程

	for {
		var cmd string
		fmt.Scanln(&cmd)
		fmt.Println("cmd：", cmd)
		if cmd == "exit" {
			exit = true
			break
		} else {
			fmt.Println("unknow command")
			fmt.Println("exit exit soft")
		}
	}
	wg.Wait()
	fmt.Println("FileSync soft exit")
}

func mainRun(fromDir, toDir string) {
	defer wg.Done()
	for !exit {
		filepath.Walk(fromDir, func(path string, info os.FileInfo, err error) error {
			if err != nil {
				return nil
			}
			fmt.Println(path)
			p := strings.ToLower(path)
			// 跳过常见日志目录（注意：Windows风格路径）
			if strings.Contains(p, "\\runlog\\") || 
			   strings.Contains(p, "\\logs\\") || 
			   strings.Contains(p, "\\log\\") {
				return nil
			}
			if info.IsDir() {
				syncDir(strings.Replace(path, fromDir, toDir, 1))
			} else {
				syncFile(path, strings.Replace(path, fromDir, toDir, 1))
			}
			return nil
		})
		// 每小时同步一次
		for i := 0; i < 60*60; i++ {
			time.Sleep(time.Second)
			if exit {
				break
			}
		}
	}
}

func syncDir(dirname string) {
	if err := os.MkdirAll(dirname, 0777); err != nil {
		fmt.Println(err)
	}
}

func syncFile(f, t string) {
	CopyFile(f, t)
}

func CopyFile(f, t string) (written int64, err error) {
	src, err := os.Open(f)
	if err != nil {
		return
	}
	defer src.Close()
	st, _ := src.Stat()
	mt := st.ModTime()

	var dst *os.File
	if ft, exist := checkFileIsExist(t); exist {
		// 修改时间相同，跳过复制
		if st.ModTime().UnixNano() == ft.UnixNano() {
			fmt.Println("修改时间相同，放弃")
			return
		}
		dst, err = os.OpenFile(t, os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0777)
		if err != nil {
			fmt.Println(err)
			return
		}
	} else {
		dst, err = os.Create(t)
		if err != nil {
			fmt.Println(err)
			return
		}
	}
	l, err := io.Copy(dst, src)
	dst.Close()
	if err == nil {
		// 保留原始修改时间
		err := os.Chtimes(t, mt, mt)
		if err != nil {
			fmt.Println(err)
		}
	}
	return l, err
}

func checkFileIsExist(filename string) (time.Time, bool) {
	fi, err := os.Stat(filename)
	if os.IsNotExist(err) {
		return time.Now(), false
	} else if err != nil {
		return time.Now(), false
	}
	return fi.ModTime(), true
}