Goroutine-Local-Storage 功能使用

Introduction

GLS 用于存储 goroutine 内的上下文信息，作用类似于 context。相比 context 有如下优势：

1. 不需要显式传递，在任意函数位置都可调用 CurSession() 获取上下文（如果有）
2. 可以在父子协程间传递（需要开启选项）

框架目前主要使用 GLS 进行 context 备份，以避免用户误传 context 导致链路透传信息（如 logid、metainfo 等）丢失

Server 端开启请求 context 备份

• 选项开启
  • 使用 server option WithContextBackup；
  • 第一个参数 enable 表示开启备份功能；
  • 第二个选项 async 表示开启异步隐式透传（表示对 go func() 异步调用中的 context 也进行透明兜底）

svr := xxx.NewServer(new(XXXImpl), server.WithContextBackup(true, true))

• 环境变量调整 localsession 管理选项
  1. 首先在【Server 端开启 WithContextBackup】
  2. 环境变量中配置 CLOUDWEGO_SESSION_CONFIG_KEY``=[{是否开启异步透传}][,{全局分片数}][,{GC间隔}]，三个选项都为可选，空表示使用默认值
     1. ex: true,10,1h 表示 开启异步 + 分片 10+1 小时 GC 间隔

Client 端开启请求 context 兜底

• 选项开启
  • 使用 client option client.``WithContextBackup``(handler)；
  • 参数 handler 表示业务自定义的备份逻辑 BackupHandler，其签名为

func(prev, cur context.Context) (ctx context.Context, backup bool)

• prev 参数表示备份的 context
• cur 参数表示当前 client 拿到的 context
• ctx 返回值表示用户处理完成的最终 context
• backup 返回值表示是否继续进行 localsession 内置的兜底备份，当前主要是 metainfo Persistent KVS 的透传

var expectedKey interface{}
cli := client.New(serverName, client.WithContextBackup(func(prev, cur context.Context) (ctx context.Context, backup bool) {
    if v := cur.Value(expectedKey); v != nil {
    // expectedKey exists, no need for recover context
        return cur, false
    }
    // expectedKey doesn't exists, need recover context from prev
    ctx = context.WithValue(cur, expectedKey, prev.Value(expectedKey))
    return ctx, true
})