trace/runtime.go

// Package trace ...
//
// Description : trace ...
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 2022-10-14 23:25
package trace

import (
	"sync"
	"time"

	"git.zhangdeman.cn/zhangdeman/util"
)

// NewRuntime 获取runtime实例
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 23:26 2022/10/14
func NewRuntime(traceID string, stackOffset int) *Runtime {
	if len(traceID) == 0 {
		// 若不指定 trace id , 随机生成
		traceID = util.String.Md5(util.String.GenRandomMd5())
	}

	if stackOffset < 0 {
		stackOffset = 0
	}
	return &Runtime{
		lock:         &sync.RWMutex{},
		traceID:      traceID,
		behaviorList: make([]Behavior, 0),
		stackOffset:  stackOffset,
	}
}

// Runtime ...
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 23:32 2022/10/14
type Runtime struct {
	lock         *sync.RWMutex // 锁
	traceID      string        // 日志追踪ID
	behaviorList []Behavior    // 行为列表
	stackOffset  int           // 堆栈回溯层级
}

// StartBehavior 开始一个行为
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 23:41 2022/10/14
func (r *Runtime) StartBehavior(action string, data map[string]interface{}) int {
	if nil == data {
		data = make(map[string]interface{})
	}
	r.lock.Lock()
	defer r.lock.Unlock()
	b := Behavior{
		ID:        len(r.behaviorList),
		Stack:     GetTraceFileInfo(r.stackOffset),
		Action:    action,
		Type:      BehaviorActionTypeStart,
		Timestamp: time.Now().UnixNano(),
		Data:      data,
	}
	r.behaviorList = append(r.behaviorList, b)
	return b.ID
}

// FinishBehavior 结束某一个行为
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 23:53 2022/10/14
func (r *Runtime) FinishBehavior(behaviorID int, data map[string]interface{}) {
	if nil == data {
		data = make(map[string]interface{})
	}
	r.lock.Lock()
	defer r.lock.Unlock()
	b := Behavior{
		ID:              len(r.behaviorList),
		StartBehaviorID: behaviorID,
		Stack:           GetTraceFileInfo(r.stackOffset),
		Action:          r.behaviorList[behaviorID].Action,
		Type:            BehaviorActionTypeFinish,
		Timestamp:       time.Now().UnixNano(),
		Data:            data,
	}
	r.behaviorList = append(r.behaviorList, b)
}

// WrapRun ...
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 00:17 2022/10/15
func (r *Runtime) WrapRun(action string, startData map[string]interface{}, logic func()) {
	if nil == logic {
		return
	}
	behaviorID := r.StartBehavior(action, startData)
	defer r.FinishBehavior(behaviorID, nil)
}

// GetTraceID 获取 trace id
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 00:28 2022/10/15
func (r *Runtime) GetTraceID() string {
	return r.traceID
}

// GetBehaviorList 获取行为列表
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 00:29 2022/10/15
func (r *Runtime) GetBehaviorList() []Behavior {
	r.lock.RLock()
	defer r.lock.RUnlock()
	behaviorList := make([]Behavior, 0)
	for _, item := range r.behaviorList {
		behaviorList = append(behaviorList, item)
	}
	return behaviorList
}