78 lines
2.3 KiB
Go
Executable File
78 lines
2.3 KiB
Go
Executable File
package utils
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"net/textproto"
|
||
"os"
|
||
"strings"
|
||
|
||
"github.com/google/uuid"
|
||
"github.com/pkg/errors"
|
||
)
|
||
|
||
const (
|
||
UnifiedPattern = "unified_pattern_" // 为了只注册一个handler到asynq，所有的task type都要加上这个前缀来做模糊匹配
|
||
)
|
||
|
||
// ExtendUnifiedPattern 给任务type追加通用前缀。为了让每个队列固定挂载一个统一的handler
|
||
func ExtendUnifiedPattern(queue, taskType string) (string, error) {
|
||
if strings.Contains(queue, ".") || strings.Contains(taskType, ".") {
|
||
return "", errors.Errorf("cannot contain '.': %s/%s", queue, taskType)
|
||
}
|
||
return UnifiedPattern + queue + "." + taskType, nil
|
||
}
|
||
|
||
// TrimUnifiedPattern 将任务type去除通用前缀，返回AddTask时真正传入的type
|
||
func TrimUnifiedPattern(taskType string) string {
|
||
subs := strings.SplitN(taskType, ".", 2)
|
||
if len(subs) == 2 {
|
||
return subs[1]
|
||
} else {
|
||
// 走到这里其实是出问题了，会造成任务找不到handler
|
||
// 在extendUnifiedPattern兜底了，理论上不应该走到这里
|
||
panic(fmt.Sprintf("found no '.' in task type: %s", taskType))
|
||
}
|
||
}
|
||
|
||
// GenDefaultQueuePrefix 默认队列前缀：环境、服务名、灰度，用于灰度能力支持、环境安全、服务间隔离（连同一redis时）
|
||
func GenDefaultQueuePrefix() string {
|
||
var prefixes []string
|
||
|
||
if env := os.Getenv("PROJECT_ENV"); env != "" {
|
||
prefixes = append(prefixes, env)
|
||
} else {
|
||
prefixes = append(prefixes, "unkonwn_env")
|
||
}
|
||
|
||
if service := os.Getenv("PROJECT_NAME"); service != "" {
|
||
prefixes = append(prefixes, service)
|
||
} else {
|
||
prefixes = append(prefixes, "unkonwn_service")
|
||
}
|
||
|
||
if gray := os.Getenv("GRAY"); gray == "1" {
|
||
prefixes = append(prefixes, "gray")
|
||
}
|
||
|
||
if os.Getenv("LOCAL_DEBUG") == "1" {
|
||
// 开发本地debug模式，更严格的隔离，加入hostname
|
||
hostname, _ := os.Hostname()
|
||
prefixes = append(prefixes, "local_debug", hostname)
|
||
}
|
||
return strings.Join(prefixes, ".") + "."
|
||
}
|
||
|
||
// BaseContext 向context中注入logger包WithCtx需要的请求ID、请求者，方便串联同一次任务执行的日志
|
||
func BaseContext() context.Context {
|
||
ctx := context.Background()
|
||
|
||
key := textproto.CanonicalMIMEHeaderKey("Xc-Req-Id")
|
||
ctx = context.WithValue(ctx, key, uuid.NewString())
|
||
|
||
key = "caller"
|
||
ctx = context.WithValue(ctx, key, "me-xcasynq")
|
||
|
||
return ctx
|
||
}
|