Go1.7做一了一个简单的远程目录比较工具

package main

import (
"bufio"
"bytes"
"flag"
"fmt"
"io"
"log"
"net/http"
"os"
"path/filepath"
"strings"

"github.com/czxichen/AutoWork/tools/md5"
"github.com/czxichen/AutoWork/tools/split"
)

var (
raddr, laddr        string
sdir, ddir, exclude string
passName            []string
result              string = "result/"
)

func init() {
flag.StringVar(&ddir, "d", "", "-d 指定要匹配的目录")
flag.StringVar(&sdir, "s", "", "-s 指定要读取的目录")
flag.StringVar(&raddr, "p", "", "-p 指定原始目录的IP和端口")
flag.StringVar(&laddr, "l", ":1789", "-l 127.0.0.1:1789 指定监听的端口")
flag.StringVar(&exclude, "v", "", "-v log,txt 指定排除的后缀文件")
flag.Parse()
}

func main() {
if sdir != "" && laddr != "" {
Server()
return
}
if ddir != "" && raddr != "" {
Client(raddr)
return
}
flag.Usage()
}

func Server() {
sdir = filepath.ToSlash(sdir)
if !strings.HasSuffix(sdir, "/") {
sdir += "/"
}
if exclude != "" {
passName = strings.Split(exclude, ",")
}

os.MkdirAll(result, 0666)
Walk(sdir)
http.HandleFunc("/", Router)
http.ListenAndServe(laddr, nil)
}

func Router(w http.ResponseWriter, r *http.Request) {
log.Printf("远端地址:%s\t访问的路径:%s\n", r.RemoteAddr, r.URL.Path)
defer r.Body.Close()
switch r.URL.Path {
case "/":
File, err := os.Open("md5_list.txt")
if err != nil {
http.NotFound(w, r)
return
}
io.Copy(w, File)
File.Close()
case "/result":
file, err := os.Create(result + strings.Split(r.RemoteAddr, ":")[0] + ".txt")
if err != nil {
http.Error(w, err.Error(), 501)
return
}
io.Copy(file, r.Body)
file.Close()
case "/flushmd5":
Walk(sdir)
fmt.Fprintln(w, "flush md5 list ok")
}
}

func Walk(dir string) {
File, err := os.Create("md5_list.txt")
if err != nil {
log.Println("创建md5列表文件失败:", err.Error())
return
}
defer File.Close()

err = filepath.Walk(dir, func(root string, info os.FileInfo, err error) error {
if err != nil {
return err
}
if info.IsDir() {
return nil
}
if Exclude(root) {
return nil
}
m5, err := md5.Md5(root)
if err != nil {
log.Printf("读取文件:%s的md5失败,错误信息:\n", root, err.Error())
return err
}
root = strings.TrimPrefix(filepath.ToSlash(root), dir)
fmt.Fprintln(File, root, m5)
return nil
})
if err != nil {
log.Println("遍历文件夹出错:", err.Error())
}
log.Println("遍历获取md5完成")
}

func Exclude(Suffix string) bool {
for _, name := range passName {
if strings.HasSuffix(Suffix, name) {
return true
}
}
return false
}

func Client(ip string) {
ddir = filepath.ToSlash(ddir)
if !strings.HasSuffix(ddir, "/") {
ddir += "/"
}

resp, err := http.Get("http://" + ip)
if err != nil {
fmt.Println("连接远端出错:", err.Error())
return
}
buf := bytes.NewBuffer(nil)
io.Copy(buf, resp.Body)
resp.Body.Close()
File, err := os.Create("cmd5_list.txt")
if err != nil {
fmt.Println("创建结果文件失败:", err.Error())
return
}
defer File.Close()
err = Compare(buf, File, ddir)
if err != nil {
fmt.Println(err)
}
File.Sync()
File.Seek(0, 0)
err = client(File, ip)
if err != nil {
fmt.Println("上传结果出错:", err.Error())
}
os.Remove("cmd5_list.txt")
}

func Compare(r io.Reader, w io.Writer, dst string) error {
rd := bufio.NewReader(r)
for {
line, _, err := rd.ReadLine()
if err != nil {
if err == io.EOF {
break
}
return err
}
path_md5 := split.Split(string(line))
if len(path_md5) != 2 {
continue
}
m5, err := md5.Md5(dst + path_md5[0])
if err != nil {
if os.IsNotExist(err) {
fmt.Fprintln(w, path_md5[0], path_md5[1], "'File_not_is_exist'")
} else {
fmt.Fprintln(w, path_md5[0], path_md5[1], err.Error())
}
continue
}
if path_md5[1] != m5 {
fmt.Fprintln(w, path_md5[0], path_md5[1], m5)
}
}
return nil
}

func client(r io.Reader, ip string) error {
resp, err := http.Post("http://"+ip+"/result", "application/octet-stream", r)
if err != nil {
return err
}
resp.Body.Close()
return nil
}