golang 通过fsnotify监控文件，并通过文件变化重启程序

分类：Go作者：编程之家用户

一、下载我们需要的包

> go get github.com/fsnotify/fsnotify

二、使用fsnotify监控文件

package main;

import (
	"github.com/fsnotify/fsnotify"
	"log"
	"fmt"
)

func main() {
	//创建一个监控对象
	watch,err := fsnotify.NewWatcher();
	if err != nil {
		log.Fatal(err);
	}
	defer watch.Close();
	//添加要监控的对象，文件或文件夹
	err = watch.Add("./tmp");
	if err != nil {
		log.Fatal(err);
	}
	//我们另启一个goroutine来处理监控对象的事件
	go func() {
		for {
			select {
			case ev := <-watch.Events:
				{
					//判断事件发生的类型，如下5种
					// Create 创建
					// Write 写入
					// Remove 删除
					// Rename 重命名
					// Chmod 修改权限
					if ev.Op&fsnotify.Create == fsnotify.Create {
						log.Println("创建文件 : ",ev.Name);
					}
					if ev.Op&fsnotify.Write == fsnotify.Write {
						log.Println("写入文件 : ",ev.Name);
					}
					if ev.Op&fsnotify.Remove == fsnotify.Remove {
						log.Println("删除文件 : ",ev.Name);
					}
					if ev.Op&fsnotify.Rename == fsnotify.Rename {
						log.Println("重命名文件 : ",ev.Name);
					}
					if ev.Op&fsnotify.Chmod == fsnotify.Chmod {
						log.Println("修改权限 : ",ev.Name);
					}
				}
			case err := <-watch.Errors:
				{
					log.Println("error : ",err);
					return;
				}
			}
		}
	}();

	//循环
	select {};
}

测试结果如下：

我们在tmp目录下的操作都被捕捉到了，但是fsnotify有一个问题，它无法递归的帮我们捕捉子目录、孙子目录的操作事件，这需要我们自已来实现。

还有一个问题就是当们修改文件夹名称时，fsnotify中event.Name仍然是原来的文件名，这就需要我们在重命名事件中，先移除之前的监控，然后添加新的监控。

修改如下：

package main;

import (
	"github.com/fsnotify/fsnotify"
	"fmt"
	"path/filepath"
	"os"
)

type Watch struct {
	watch *fsnotify.Watcher;
}

//监控目录
func (w *Watch) watchDir(dir string) {
	//通过Walk来遍历目录下的所有子目录
	filepath.Walk(dir,func(path string,info os.FileInfo,err error) error {
		//这里判断是否为目录，只需监控目录即可
		//目录下的文件也在监控范围内，不需要我们一个一个加
		if info.IsDir() {
			path,err := filepath.Abs(path);
			if err != nil {
				return err;
			}
			err = w.watch.Add(path);
			if err != nil {
				return err;
			}
			fmt.Println("监控 : ",path);
		}
		return nil;
	});
	go func() {
		for {
			select {
			case ev := <-w.watch.Events:
				{
					if ev.Op&fsnotify.Create == fsnotify.Create {
						fmt.Println("创建文件 : ",ev.Name);
						//这里获取新创建文件的信息，如果是目录，则加入监控中
						fi,err := os.Stat(ev.Name);
						if err == nil && fi.IsDir() {
							w.watch.Add(ev.Name);
							fmt.Println("添加监控 : ",ev.Name);
						}
					}
					if ev.Op&fsnotify.Write == fsnotify.Write {
						fmt.Println("写入文件 : ",ev.Name);
					}
					if ev.Op&fsnotify.Remove == fsnotify.Remove {
						fmt.Println("删除文件 : ",ev.Name);
						//如果删除文件是目录，则移除监控
						fi,err := os.Stat(ev.Name);
						if err == nil && fi.IsDir() {
							w.watch.Remove(ev.Name);
							fmt.Println("删除监控 : ",ev.Name);
						}
					}
					if ev.Op&fsnotify.Rename == fsnotify.Rename {
						fmt.Println("重命名文件 : ",ev.Name);
						//如果重命名文件是目录，则移除监控
						//注意这里无法使用os.Stat来判断是否是目录了
						//因为重命名后，go已经无法找到原文件来获取信息了
						//所以这里就简单粗爆的直接remove好了
						w.watch.Remove(ev.Name);
					}
					if ev.Op&fsnotify.Chmod == fsnotify.Chmod {
						fmt.Println("修改权限 : ",ev.Name);
					}
				}
			case err := <-w.watch.Errors:
				{
					fmt.Println("error : ",err);
					return;
				}
			}
		}
	}();
}

func main() {
	watch,_ := fsnotify.NewWatcher()
	w := Watch{
		watch: watch,}
	w.watchDir("./tmp");
	select {};
}

测试结果如下：

经过上面的例子，我们通过fsnotify来写一个监控配置文件，如果配置文件有修改，就重新启动服务。

我们先写一个可以运行的exe程序，server.go代码如下：

package main;

import (
	"io/ioutil"
	"log"
	"encoding/json"
	"net"
	"fmt"
	"os"
	"os/signal"
)

const (
	confFilePath = "./conf/conf.json";
)

//我们这里只是演示，配置项只设置一个
type Conf struct {
	Port int `json:port`;
}

func main() {
	//读取文件内容
	data,err := ioutil.ReadFile(confFilePath);
	if err != nil {
		log.Fatal(err);
	}
	var c Conf;
	//解析配置文件
	err = json.Unmarshal(data,&c);
	if err != nil {
		log.Fatal(err);
	}
	//根据配置项来监听端口
	lis,err := net.Listen("tcp",fmt.Sprintf(":%d",c.Port));
	if err != nil {
		log.Fatal(err);
	}
	log.Println("server start");
	go func() {
		ch := make(chan os.Signal);
		//获取程序退出信号
		signal.Notify(ch,os.Interrupt,os.Kill);
		<-ch;
		log.Println("server exit");
		os.Exit(1);
	}();
	for {
		conn,err := lis.Accept();
		if err != nil {
			continue;
		}
		go func(conn net.Conn) {
			defer conn.Close();
			conn.Write([]byte("hello\n"));
		}(conn);
	}
}

使用如下命令，编译成exe文件

> go build server.go

监控文件fsnotify3.go代码如下：

package main;

import (
	"github.com/fsnotify/fsnotify"
	"log"
	"fmt"
	"os/exec"
	"regexp"
	"strconv"
	"bytes"
	"errors"
	"os"
	"path/filepath"
)

const (
	confFilePath = "./conf";
)

//获取进程ID
func getPid(processName string) (int,error) {
	//通过wmic process get name,processid | findstr server.exe获取进程ID
	buf := bytes.Buffer{};
	cmd := exec.Command("wmic","process","get","name,processid");
	cmd.Stdout = &buf;
	cmd.Run();
	cmd2 := exec.Command("findstr",processName);
	cmd2.Stdin = &buf;
	data,_ := cmd2.CombinedOutput();
	if len(data) == 0 {
		return -1,errors.New("not find");
	}
	info := string(data);
	//这里通过正则把进程id提取出来
	reg := regexp.MustCompile(`[0-9]+`);
	pid := reg.FindString(info);
	return strconv.Atoi(pid);
}

//启动进程
func startProcess(exePath string,args []string) error {
	attr := &os.ProcAttr{
		//files指定新进程继承的活动文件对象
		//前三个分别为，标准输入、标准输出、标准错误输出
		Files: []*os.File{os.Stdin,os.Stdout,os.Stderr},//新进程的环境变量
		Env: os.Environ(),}

	p,err := os.StartProcess(exePath,args,attr);
	if err != nil {
		return err;
	}
	fmt.Println(exePath,"进程启动");
	p.Wait();
	return nil;
}

func main() {
	//创建一个监控对象
	watch,err := fsnotify.NewWatcher();
	if err != nil {
		log.Fatal(err);
	}
	defer watch.Close();
	//添加要监控的文件
	err = watch.Add(confFilePath);
	if err != nil {
		log.Fatal(err);
	}
	//我们另启一个goroutine来处理监控对象的事件
	go func() {
		for {
			select {
			case ev := <-watch.Events:
				{
					//我们只需关心文件的修改
					if ev.Op&fsnotify.Write == fsnotify.Write {
						fmt.Println(ev.Name,"文件写入");
						//查找进程
						pid,err := getPid("server.exe");
						//获取运行文件的绝对路径
						exePath,_ := filepath.Abs("./server.exe")
						if err != nil {
							//启动进程
							go startProcess(exePath,[]string{});
						} else {
							//找到进程，并退出
							process,err := os.FindProcess(pid);
							if err == nil {
								//让进程退出
								process.Kill();
								fmt.Println(exePath,"进程退出");
							}
							//启动进程
							go startProcess(exePath,[]string{});
						}
					}
				}
			case err := <-watch.Errors:
				{
					fmt.Println("error : ",err);
					return;
				}
			}
		}
	}();

	//循环
	select {};
}

我们运行fsnotify3.go文件来监控我们的配置文件

通过上面的图可以看到，当我们修改配置文件中的端口号时，会先kill掉进程，然后再启动一个进程。

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 dio@foxmail.com 举报，一经查实，本站将立刻删除。

相关推荐

Golang类型转换
类型转换 1、int转string 2、string转int 3、string转float 4、用户结构类型转换
golang 字符串操作实例
package main import s &quot;strings&quot; import &quot;fmt&quot; var p = fmt.Println func main() { p(&quot;Contains: &quot;, s.Contains(&quot;test&quo
golang中的类和接口的使用
类使用：实现一个people中有一个sayhi的方法调用功能，代码如下： 接口使用：实现上面功能，代码如下：
beego上传文件
html代码： beego代码：
golang操作文件
1、读取文件信息： 2、读取文件夹下的所有文件： 3、写入文件信息 4、删除文件，成功返回true，失败返回false
golang开发环境配置及Beego框架安装
配置环境：Windows7+推荐IDE：LiteIDEGO下载地址：http://www.golangtc.com/downloadBeego开发文档地址：http://beego.me/docs/intro/ 安装步骤： 一、GO环境安装 二、配置系统变量 三、Beego安装 一、GO环境安装 根
golang获取程序运行路径
golang获取程序运行路径：
Golang的文档和社区资源：为什么它可以帮助开发人员快速上手？
Golang的文档和社区资源：为什么它可以帮助开发人员快速上手？
Golang：AI 开发者的实用工具
Golang：AI 开发者的实用工具
Golang的标准库：为什么它可以大幅度提高开发效率？
Golang的标准库：为什么它可以大幅度提高开发效率？
Golang的部署和运维：如何将应用程序部署到生产环境中？
Golang的部署和运维：如何将应用程序部署到生产环境中？
高性能AI开发：Golang的优势所在
高性能AI开发：Golang的优势所在
go语言开发优雅得关闭协程的方法
本篇文章和大家了解一下go语言开发优雅得关闭协程的方法。有一定的参考价值，有需要的朋友可以参考一下，希望对大家有所帮助。1.简介本文将介绍首先为什么需要主...
Go关闭goroutine协程的方法
这篇文章主要介绍了Go关闭goroutine协程的方法，具有一定借鉴价值，需要的朋友可以参考下。下面就和我一起来看看吧。1.简介本文将介绍首先为什么需要主动关闭gor...
go关闭Graceful Shutdown服务的几种方法
本篇文章和大家了解一下go关闭GracefulShutdown服务的几种方法。有一定的参考价值，有需要的朋友可以参考一下，希望对大家有所帮助。目录Shutdown方法Regi...
Go语言如何实现LRU算法的核心思想和实现过程
这篇文章主要介绍了Go语言如何实现LRU算法的核心思想和实现过程，具有一定借鉴价值，需要的朋友可以参考下。下面就和我一起来看看吧。GO实现Redis的LRU例子常
Go 简单实现多租户数据库隔离的方法
今天小编给大家分享的是Go简单实现多租户数据库隔离的方法，相信很多人都不太了解，为了让大家更加了解，所以给大家总结了以下内容，一起往下看吧。一定会...
Linux系统中怎么安装NSQ的Go语言客户端
这篇“Linux系统中怎么安装NSQ的Go语言客户端”文章的知识点大部分人都不太理解，所以小编给大家总结了以下内容，内容详细，步骤清晰，具有一定的借鉴价值，希
怎么在Go语言中实现锁机制
本文小编为大家详细介绍“怎么在Go语言中实现锁机制”，内容详细，步骤清晰，细节处理妥当，希望这篇“怎么在Go语言中实现锁机制”文章能帮助大家解决疑惑，下面...
Go语言中interface类型怎么使用
今天小编给大家分享一下Go语言中interface类型怎么使用的相关知识点，内容详细，逻辑清晰，相信大部分人都还太了解这方面的知识，所以分享这篇文章给大家参考
pythonJavaScriptjavaHTMLPHPreactjsC#AndroidCSSNode.jssqlrpython-3.xMysqLjQueryc++pandasFlutterangularIOSdjangolinuxswifttypescript路由器JSON路由器设置无线路由器h3c华三华三路由器设置华三路由器电脑软件教程arraysdocker软件图文教程Cvue.jslaravelspring-boot