Golang优雅保持main函数不退出的办法

作者:MiKogy_ 时间:2023-07-12 21:15:54 

高能预警

本文包含演示部分,请读者自行copy代码编译体验。

参考资料:sync.WaitGroup / signal.Notify / context.CancelFunc

我们有时会希望我们的程序保持执行,但是有一种情况是:我们的代码全部塞入go routine时,主函数会立刻退出,本文将和大家分享如何让main函数优雅地保持执行。

问题演示:

func main() {
   go func() {
       for i := 0; i<10000;i ++ {
           fmt.Println(i)
       }
   }()
}

此时我们可以看到,控制台几乎不会输出任何内容。究其原因,是主函数在go routine执行前就已经结束,也就是说go routine不会阻塞主函数。

可能有些读者会想到,我直接加个死循环在下面,让主函数不退出不就行啦?博主表示十分赞同,因为博主就是采用这个方法,导致服务器跑满CPU从而不停的告警。

那么解决办法是:让死循环慢一点执行,即添加以下内容:

for {
       time.Sleep(time.Second)
   }

但是在博主的完美主义光环加持下,还是希望我们的代码能更加优雅,下面将介绍另外三种比较优雅的保持main函数的办法。

解决办法演示

操作系统信号阻塞

先上代码:

func main() {
   c := make(chan os.Signal)
   signal.Notify(c)
   go func() {
       fmt.Println("Go routine running")
       time.Sleep(3*time.Second)
       fmt.Println("Go routine done")
   }()
   <-c
   fmt.Println("bye")
}

官网机翻:signal.Notify()方法使信号将传入c。如果没有提供信号,所有传入的信号将被中继到c。

  • 这里我们创建了一个os.Signal类型的管道。当管道为空的时候,读管道操作&ldquo;<-&rdquo;会阻塞住,直到我们向进程发送一个信号(例如 Ctrl+C),才会继续执行该操作后面的代码。

上下文操作阻塞

再上代码:

func main() {
   ctx, cancel := context.WithCancel(context.Background())
   defer cancel()
   go func() {
       fmt.Println("Go routine running")
       time.Sleep(3 * time.Second)
       fmt.Println("Go routine done")
       cancel()
   }()
   <-ctx.Done()
   fmt.Println("bye")
}

官网机翻:CancelFunc() 通知操作放弃其(当前的)工作。CancelFunc() 不会等待工作停止。

  • 这也是一个十分优雅的办法,我们创建一个可以终止的上下文&mdash;&mdash;context.WithCancel(),并在go routine执行完毕时调用其返回的CancelFunc() 方法,即表示该上下文已经结束了。而在这之前,我们会使用<-ctx.Done()来一直等待上下文的结束,也就是说main函数被成功阻塞,并等待go routine执行完毕并执行了cancel()方法后优雅退出。

WaitGroup阻塞

然后上代码:

func main() {
   wg := &sync.WaitGroup{}
   wg.Add(2)
   go func() {
       time.Sleep(3*time.Second)
       fmt.Println("3 second passed")
       wg.Done()
   }()
   go func() {
       time.Sleep(5*time.Second)
       fmt.Println("5 second passed")
       wg.Done()
   }()
   wg.Wait()
   fmt.Println("bye")
}

官网机翻:WaitGroup 等待一组 go routine 完成。主 go routine 调用 Add() 来设置要等待的 go routine 的数量。

  • 我们首先创建一个WaitGroup{}对象,然后调用Add()方法,在里面传入我们接下来会创建的go routine的数量,每当我们执行完一个go routine时,调用一次Done()方法,使得正执行的go routine的数量减一,当减到0时,Wait()方法将不再等待(阻塞),使main函数继续向下执行

来源:https://blog.csdn.net/Liing0/article/details/122481451

标签:golang,main,函数
0
投稿

猜你喜欢

  • OpenCV图像变换之傅里叶变换的一些应用

    2023-12-01 22:11:34
  • sql分类汇总及Select的自增长脚本

    2024-01-22 20:43:34
  • matplotlib 使用 plt.savefig() 输出图片去除旁边的空白区域

    2023-06-24 02:55:42
  • ActiveMQ:使用Python访问ActiveMQ的方法

    2022-04-19 01:48:30
  • Pandas常用累计、同比、环比等统计方法实践过程

    2021-03-13 01:48:36
  • laravel yajra插件 datatable的使用详解

    2023-11-18 20:05:06
  • Django web框架使用url path name详解

    2022-01-06 19:43:59
  • DOM和XMLHttpRequest对象的属性和方法整理

    2024-04-23 09:05:27
  • Golang中slice删除元素的性能对比

    2024-05-05 09:33:03
  • Django中对通过测试的用户进行限制访问的方法

    2021-08-27 16:42:46
  • Python模块zipfile原理及使用方法详解

    2022-03-24 10:33:37
  • IIS服务器中部署PHP案例详解

    2023-06-11 19:17:42
  • Pandas 类型转换astype()的实现

    2022-03-13 17:39:52
  • 详解Python使用tensorflow入门指南

    2023-08-22 13:37:09
  • Python基于ImageAI实现图像识别详解

    2023-06-11 14:04:49
  • PyQt5根据控件Id获取控件对象的方法

    2023-02-13 15:22:50
  • Oracle PL/SQL入门案例实践

    2010-07-18 13:13:00
  • 如何恢复MySQL主从数据一致性

    2024-01-26 23:34:33
  • Go高级特性探究之稳定排序详解

    2023-07-17 16:11:48
  • Python3的urllib.parse常用函数小结(urlencode,quote,quote_plus,unquote,unquote_plus等)

    2021-01-23 04:23:13
  • asp之家 网络编程 m.aspxhome.com