Go 安全使用goroutine

Go 安全使用goroutine

go 正常使用goroutine开启一个携程很简单

go 复制代码
var a int
go func(){
	a=1+1
}()

这么用在日常工具什么的开发中肯定没问题,如果携程内有问题崩掉了,使用工具的人可以马上获得堆栈信息将其反应给开发人员。但是你如果在web服务器或者后台程序中使用就有大问题。因为golang无法捕获携程中的panic,也就是说你携程崩掉了,你携程中又没有recover,你整个程序都会被其带崩,并且崩溃是父携程不可捕获的。

上案例

go 复制代码
package main

import (
	"net/http"

	"github.com/gin-gonic/gin"
)

func main() {
	app := gin.New()
	app.Handle(http.MethodGet, "/panic", func(ctx *gin.Context) {
		go func() { panic("goroutine panic") }()
	})
	app.Handle(http.MethodPost, "/panic", func(ctx *gin.Context) { panic("panic") })
	http.ListenAndServe("0.0.0.0:7999", app)
}

此处使用gin框架做示范,模拟了两种崩溃情况.此时分别以get和post两种方法请求/panic接口,get请求程序,程序会因为无法捕获携程中的panic会崩溃,而post请求会恢复,因为golang 处理http请求时,在携程中使用了recover,你只要用http.ListenAndServe就能保你。但是你如果再自己的handlefunc 中再开子携程且没有上保护措施那么就保不住你啰

编写一个安全的goroutine 构建器

go 复制代码
type Trace struct {
	info  any
	stack string
}
type RoutineBuilder struct {
	out       io.Writer
	traceChan chan Trace
}

func (rb *RoutineBuilder) Go(fun func()) {
	go func(f func()) {
		defer func() {
		//记录panic原因,以及堆栈信息用于排查
			if r := recover(); r != nil {
				rb.traceChan <- Trace{
					info:  r,
					stack: string(debug.Stack()),
				}
			}
		}()
		f()
	}(fun)
}
func NewRoutineBuilder(out io.Writer) (*RoutineBuilder, <-chan Trace) {
	var rb = RoutineBuilder{
		out:       out,
		traceChan: make(chan Trace, 20),
	}
	return &rb, rb.traceChan
}

应用该携程构建器到案例中

这里的策略为将意外崩溃的携程信息直接写到标准输出,日志处理那步可以根据自身需求对堆栈信息做处理

go 复制代码
package main

import (
	"fmt"
	"io"
	"net/http"
	"os"
	"runtime/debug"
	"time"

	"github.com/gin-gonic/gin"
)

type Trace struct {
	info  any
	stack string
}
type RoutineBuilder struct {
	out       io.Writer
	traceChan chan Trace
}

func (rb *RoutineBuilder) Go(fun func()) {
	go func(f func()) {
		defer func() {
			if r := recover(); r != nil {
				rb.traceChan <- Trace{
					info:  r,
					stack: string(debug.Stack()),
				}
			}
		}()
		f()
	}(fun)
}
func NewRoutineBuilder(out io.Writer) (*RoutineBuilder, <-chan Trace) {
	var rb = RoutineBuilder{
		out:       out,
		traceChan: make(chan Trace, 20),
	}
	return &rb, rb.traceChan
}

var routineBuilder, stackChan = NewRoutineBuilder(os.Stderr)
//记录意外崩溃的堆栈信息和原因到日志中
func traceLogCheck() {
	defer func() {
		if r := recover(); r != nil {
			fmt.Fprintln(routineBuilder.out, time.Now(), " trace log check panic", r, string(debug.Stack()))
			traceLogCheck()
		}
	}()
	for info := range stackChan {
		fmt.Fprintln(routineBuilder.out, time.Now(), " trace log check panic", info.info, info.stack)
	}
}
func init() {
	go traceLogCheck()
}
func main() {
	app := gin.New()
	app.Handle(http.MethodGet, "/panic", func(ctx *gin.Context) {
		routineBuilder.Go(func() { panic("goroutine panic") })
	})
	app.Handle(http.MethodPost, "panic", func(ctx *gin.Context) { panic("panic") })
	http.ListenAndServe("0.0.0.0:7999", app)
}

看看效果

成功拦截到handfunc中子携程的panic

相关推荐
wei_shuo36 分钟前
飞算 JavaAI 开发助手:深度学习驱动下的 Java 全链路智能开发新范式
java·开发语言·飞算javaai
熊猫钓鱼>_>36 分钟前
用Python解锁图像处理之力:从基础到智能应用的深度探索
开发语言·图像处理·python
GO兔1 小时前
开篇:GORM入门——Go语言的ORM王者
开发语言·后端·golang·go
好开心啊没烦恼1 小时前
Python 数据分析:numpy,抽提,整数数组索引与基本索引扩展(元组传参)。听故事学知识点怎么这么容易?
开发语言·人工智能·python·数据挖掘·数据分析·numpy·pandas
future14122 小时前
C#学习日记
开发语言·学习·c#
king_harry2 小时前
Java程序-OceanBase Connector/J 示例
开发语言
傻啦嘿哟3 小时前
Python 办公实战:用 python-docx 自动生成 Word 文档
开发语言·c#
翻滚吧键盘3 小时前
js代码09
开发语言·javascript·ecmascript
q567315233 小时前
R语言初学者爬虫简单模板
开发语言·爬虫·r语言·iphone
rzl024 小时前
java web5(黑马)
java·开发语言·前端