hongtao3_watch/watch/watch.go

package watch

import (
	"fmt"
	"github.com/fsnotify/fsnotify"
	"hongze/hongtao3_watch/cache"
	"hongze/hongtao3_watch/services"
	"os"
	"path/filepath"
	"strings"
	"time"
)

/*
CREATE动作即临时文件的创建
WRITE写文件动作
CHMOD修改文件属性
REMOVE删除临时文件。
*/

// excel存放的根目录
var rootDir = `E:\hz\hongtao3_data\excel`

//var rootDir = `D:\data`	// 本地开发环境的目录

func ListenFolderNew() {
	fmt.Println("-----文件夹监听-------")
	watcher, err := fsnotify.NewWatcher()
	if err != nil {
		fmt.Println("fsnotify.NewWatcher err:" + err.Error())
		//log.Fatal(err)
	}
	defer watcher.Close()

	done2 := make(chan bool)
	go func() {
		for {
			select {
			case event, ok := <-watcher.Events:
				//fmt.Println("event.Name", event.Name)
				//fmt.Println(event.Op)
				//if ok && event.Op == fsnotify.Create &&
				//	!strings.Contains(event.Name, "tmp") &&
				//	!strings.Contains(event.Name, ".TMP") &&
				//	!strings.Contains(event.Name, "~") &&
				//	(strings.Contains(event.Name, "xlsx") || strings.Contains(event.Name, "xls")) {
				//	WatchIndexFileRelease(event.Name)
				//	fmt.Println("op:event.Name", event.Name)
				//}

				// 如果是临时文件,那么就忽略
				if strings.Contains(event.Name, "tmp") ||
					strings.Contains(event.Name, ".TMP") ||
					strings.Contains(event.Name, "~") {
					continue
				}

				if ok && event.Op&fsnotify.Create == fsnotify.Create {
					//fmt.Println("新增文件 : ", event.Name)

					// 判断是否属于文件夹,如果是文件夹,那么就加上该文件夹的监听权限
					fi, err := os.Stat(event.Name)
					if err == nil {
						if fi.IsDir() {
							watcherDir(watcher, event.Name)
							fmt.Println("添加文件夹,添加监控 : ", event.Name)
						} else {
							// 读取文件
							cache.AddIndexHandleExcel(event.Name)
						}
					}

				}
				if ok && event.Op&fsnotify.Write == fsnotify.Write {
					//fmt.Println("写入文件 : ", event.Name)
					// 读取文件

					cache.AddIndexHandleExcel(event.Name)
				}
				if ok && event.Op&fsnotify.Remove == fsnotify.Remove {
					//fmt.Println("删除文件 : ", event.Name)
					//如果删除文件是目录，则移除监控
					fi, err := os.Stat(event.Name)
					if err == nil && fi.IsDir() {
						_ = watcher.Remove(event.Name)
						fmt.Println("删除文件夹,删除监控 : ", event.Name)
					}
				}
				if ok && event.Op&fsnotify.Rename == fsnotify.Rename {
					//fmt.Println("重命名文件 : ", event.Name)
					//如果重命名文件是目录，则移除监控
					//注意这里无法使用os.Stat来判断是否是目录了
					//因为重命名后，go已经无法找到原文件来获取信息了
					//所以这里就简单粗爆的直接remove好了
					_ = watcher.Remove(event.Name)
					//fmt.Println("更改文件夹名称,删除监控 : ", event.Name)
				}
				if ok && event.Op&fsnotify.Chmod == fsnotify.Chmod {
					fmt.Println("修改权限 : ", event.Name)
				}
			case err := <-watcher.Errors:
				if err != nil {
					fmt.Println("watcher.Errors:", err)
					//log.Println("error:", err)
				}
			case <-time.After(60 * time.Second):
				continue
			}
		}
	}()

	// 开始监听
	watcherDir(watcher, rootDir)

	<-done2
}

var watcherDirMap map[string]string

// 添加监听目录
func watcherDir(watcher *fsnotify.Watcher, dirPath string) {
	err := filepath.Walk(dirPath, func(path string, info os.FileInfo, err error) error {
		//这里判断是否为目录，只需监控目录即可
		//目录下的文件也在监控范围内，不需要我们一个一个加
		if info.IsDir() {
			subpath, e := filepath.Abs(path)
			if e != nil {
				return e
			}
			err = watcher.Add(subpath)
			if err != nil {
				return err
			}
			// 入库
			if watcherDirMap == nil {
				watcherDirMap = make(map[string]string)
			}
			watcherDirMap[subpath] = subpath
			fmt.Println("监控 : ", subpath)
		}
		return nil
	})
	if err != nil {
		fmt.Println("watcher.Add:" + err.Error())
		//log.Fatal(err)
	}
}

// HandleFileUpdate 处理文件变化
func HandleFileUpdate(updateFilePath string) {
	fileExt := filepath.Ext(updateFilePath)

	if fileExt != ".xlsx" && fileExt != ".xls" {
		fmt.Println("不是excel文件")
		return
	}

	// 读取excel内容并操作入库
	services.ReadExcel(updateFilePath)
}