跳转到主内容
新派编程网 - 新潮编程理念,全新技术课堂!

golang怎么实现文件监控

在golang中,可以利用fsnotify来实现文件监控。fsnotify是go语言跨平台文件系统监控工具,实现了一个基于channel的、跨平台的实时监听接口;golang通过fsnotify可监控文件,并通过文件变化重启程序。 本教程操作环境:windows10系统、GO 1.18版本、Dell G3电脑。 在golang中,可以利用fsnotify来实现文件监控。 golang 通过fsnotify监控文件,并通过文件变化重启程序。 go语言跨平台文件系统监控工具 — fsnotify 立即学习 “ go语言免费学习笔记(深入) ”; 在 linux 内核中,Inotify 是一种用于通知用户空间程序文件系统变化的机制。它监控文件系统的变化,如文件新建、修改、删除等,并可以将相应的事件通知给应用程序。 Inotify 既可以监控文件,也可以监控目录。当监控目录时,它可以同时监控目录及目录中的各子目录及文件。Golang 的标准库 syscall 实现了该机制。 为了进一步扩展和抽象, github.com/fsnotify/fsnotify 包实现了一个基于 channel 的、跨平台的实时监听接口。 fsnotify工具的使用 一、下载我们需要的包
go get github.com/fsnotify/fsnotify
二、使用fsnotify监控文件 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
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有一个问题,它无法递归的帮我们捕捉子目录、孙子目录的操作事件,这需要我们自已来实现。 go语言参考手册 中文CHM版 Go 是一个开源的编程语言,它能让构造简单、可靠且高效的软件变得容易。本文给大家带来Go参考手册,需要的可以来下载! Go是从2007年末由Robert Griesemer, Rob Pike, Ken Thompson主持开发,后来还加入了Ian Lance Taylor, Russ Cox等人,并最终于2009年11月开源,在2012年早些时候发布了Go 1稳定版本。现在Go的开发已经是完全开放的,并且拥有一个活跃的社区。 Go 语言特色 简洁、快速、安全 并行、有趣、开源 内存管理、v数组安全、编译 下载 还有一个问题就是当们修改文件夹名称时,fsnotify中event.Name仍然是原来的文件名,这就需要我们在重命名事件中,先移除之前的监控,然后添加新的监控。 修改如下: 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788
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代码如下: 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758
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文件 1
> go build server.go
监控文件fsnotify3.go代码如下: 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111
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掉进程,然后再启动一个进程。 推荐学习: Golang教程

相关文章