首页 ›互联网技术资讯›内容详情

golang如何实现文件监控

蜗牛互联网技术资讯2023-02-21560

本篇内容介绍了“golang如何实现文件监控”的有关知识，在实际案例的操作过程中，不少人都会遇到这样的困境，接下来就让小编带领大家学习一下如何处理这些情况吧！希望大家仔细阅读，能够学有所成！

在golang中，可以利用fsnotify来实现文件监控。fsnotify是go语言跨平台文件系统监控工具，实现了一个基于channel的、跨平台的实时监听接口；golang通过fsnotify可监控文件，并通过文件变化重启程序。

在golang中，可以利用fsnotify来实现文件监控。

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

go语言跨平台文件系统监控工具 — fsnotify

在 linux 内核中，Inotify 是一种用于通知用户空间程序文件系统变化的机制。它监控文件系统的变化，如文件新建、修改、删除等，并可以将相应的事件通知给应用程序。

Inotify 既可以监控文件，也可以监控目录。当监控目录时，它可以同时监控目录及目录中的各子目录及文件。Golang 的标准库 syscall 实现了该机制。

为了进一步扩展和抽象， github.com/fsnotify/fsnotify 包实现了一个基于 channel 的、跨平台的实时监听接口。

fsnotify工具的使用

一、下载我们需要的包

go get github.com/fsnotify/fsnotify

二、使用fsnotify监控文件

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

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 {};

}

测试结果如下：

golang如何实现文件监控 golang 第1张

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

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

修改如下：

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

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 {};

}

测试结果如下：

golang如何实现文件监控 golang 第2张

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

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

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

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代码如下：

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

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文件来监控我们的配置文件

golang如何实现文件监控 golang 第3张

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

golang如何实现文件监控 golang 第4张

“golang如何实现文件监控”的内容就介绍到这里了，感谢大家的阅读。如果想了解更多行业相关的知识可以关注蜗牛博客网站，小编将为大家输出更多高质量的实用文章！

免责声明：本站发布的内容（图片、视频和文字）以原创、转载和分享为主，文章观点不代表本网站立场，如果涉及侵权请联系站长邮箱：niceseo99@gmail.com进行举报，并提供相关证据，一经查实，将立刻删除涉嫌侵权内容。
#golang
本文地址：http://www.vps90.com/vps/31444.html
版权声明：如无特殊标注，文章均为本站原创，转载时请以链接形式注明文章出处。

相关文章

评论

关于蜗牛120vps博客

蜗牛120博客

感谢大家的支持，欢迎交流讨论！
『蜗牛120博客』

  • 11583699阅读数
  • 143268评论数

最近发表

  • Python集合是什么及怎么创建
  • SpringBoot表单提交全局日期格式转换器如何实现
  • Go反向代理ReverseProxy解析的方法是什么
  • 日本服务器租用 站群服务器 onevps：新增（支付宝+中文网站）
  • 怎样才能挑选到适合的vps服务器
  • NLTK安装错误怎么解决
  • React与TypeScript集成工作流
  • Vue中代码传送怎么实现
  • golang中怎么保证精度
  • 日本服务器租用 站群服务器 kvmla：日本/香港/新加坡，VPS直接8折优惠（+充值多送20%），独立服务器一律7.5折优惠（首月反300元）

    • 标签列表

    有免费节点资源，我们会通知你！加入纸飞机订阅群

    ×
    天气预报查看日历分享网页手机扫码留言评论Telegram