聊聊gost的GoUnterminated

最新推荐文章于 2024-01-15 15:12:14 发布

原创最新推荐文章于 2024-01-15 15:12:14 发布 · 134 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#golang

golang 专栏收录该内容

72 篇文章

订阅专栏

GoUnterminated是gost库中的一个函数，用于处理希望长时间运行的goroutine。它接受一个处理函数、等待组、忽略恢复标志和周期时间作为参数。当period大于0时，发生panic后会睡眠一段时间再递归调用自身，确保在handle函数出错时能持续运行。在测试中模拟了handle函数触发panic和正常运行的情况。GoUnterminated通过递归和延迟处理来实现对panic的控制。

序

本文主要研究一下gost的GoUnterminated

GoUnterminated

gost/runtime/goroutine.go

// GoUnterminated is used for which goroutine wanna long live as its process.
// @period: sleep time duration after panic to defeat @handle panic so frequently. if it is not positive,
//          the @handle will be invoked asap after panic.
func GoUnterminated(handle func(), wg *sync.WaitGroup, ignoreRecover bool, period time.Duration) {
	GoSafely(wg,
		ignoreRecover,
		handle,
		func(r interface{}) {
			if period > 0 {
				time.Sleep(period)
			}
			GoUnterminated(handle, wg, ignoreRecover, period)
		},
	)
}

GoUnterminated方法提供handle、WaitGroup、ignoreRecover、period参数，其内部使用的是GoSafely，只是catchFunc是内置的；catchFunc对于period大于0的会sleep一下，之后还是执行GoUnterminated，这样子在handle出错(panic)的时候会一直递归循环下去

实例

gost/runtime/goroutine_test.go

func TestGoUnterminated(t *testing.T) {
	times := uint64(1)
	var wg sync.WaitGroup
	GoUnterminated(
		func() {
			if atomic.AddUint64(&times, 1) == 2 {
				panic("hello")
			}
		},
		&wg,
		false,
		1e8,
	)
	wg.Wait()
	assert.True(t, atomic.LoadUint64(&times) == 3)

	GoUnterminated(func() {
		atomic.AddUint64(&times, 1)
	},
		nil,
		false,
		1e8,
	)
	time.Sleep(1e9)
	assert.True(t, atomic.LoadUint64(&times) == 4)
}

这里模拟了一下handler在times为1的时候产生panic，以及handler不产生panic就立刻结束的场景

小结

gost提供了GoSafely方法，该方法提供handle、WaitGroup、ignoreRecover、period参数，其内部使用的是GoSafely，只是catchFunc是内置的；catchFunc对于period大于0的会sleep一下，之后还是执行GoUnterminated，这样子在handle出错(panic)的时候会一直递归循环下去。