| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693 |
- package tracing
- import (
- "context"
- "fmt"
- klog "github.com/sirupsen/logrus"
- "os"
- "strconv"
- "strings"
- "sync"
- "sync/atomic"
- "time"
- "github.com/coroot/coroot-node-agent/ebpftracer"
- "github.com/coroot/coroot-node-agent/ebpftracer/l7"
- "github.com/coroot/coroot-node-agent/utils"
- . "github.com/coroot/coroot-node-agent/utils/modelse"
- "go.opentelemetry.io/otel/attribute"
- "go.opentelemetry.io/otel/codes"
- semconv "go.opentelemetry.io/otel/semconv/v1.18.0"
- "go.opentelemetry.io/otel/trace"
- "inet.af/netaddr"
- )
- /**
- * Trace
- */
- func (t *Trace) setContext(ctx context.Context) {
- t.lock.Lock()
- defer t.lock.Unlock()
- t.ctx = ctx
- }
- func (t *Trace) setSpan(span trace.Span) {
- t.lock.Lock()
- defer t.lock.Unlock()
- t.span = span
- }
- type TimeMap struct {
- Time uint64
- Type int
- Map *ebpftracer.StackFunEvent
- }
- //func (t *Trace) buildFun() {
- // mapSlice := []TimeMap{}
- // for i, v := range t.stack {
- // timeStartMap := TimeMap{}
- // if v.StackEvent.Location == 0 {
- // timeStartMap = TimeMap{
- // Time: v.StackEvent.TimeNsStart,
- // Type: 0,
- // Map: &t.stack[i],
- // }
- // } else {
- // timeStartMap = TimeMap{
- // Time: v.StackEvent.TimeNsEnd,
- // Type: 1,
- // Map: &t.stack[i],
- // }
- // }
- // mapSlice = append(mapSlice, timeStartMap)
- // }
- // sort.Slice(mapSlice, func(i, j int) bool {
- // return mapSlice[i].Time < mapSlice[j].Time
- // })
- //
- // funStack := []TimeMap{}
- //
- // currentfunNum := 1
- //
- // // for k, v := range mapSlice {
- // // fmt.Println("---SliceSliceindex", k, "value", v.Time, v.Type, v.Map.Uprobe.Funcname, v.Map.StackEvent.Nid)
- // // }
- //
- // mapSliceLen := len(mapSlice)
- // for k, v := range mapSlice {
- // // fmt.Println("SliceSliceindex", k, "value", v.Time, v.Type, v.Map.Uprobe.Funcname, v.Map.StackEvent.Nid)
- // if v.Type == 0 {
- // // 函数入口
- // funStack = append(funStack, v)
- // } else if v.Type == 1 {
- // // 函数出口
- // len := len(funStack)
- // if len < 1 {
- // fmt.Printf("buildFun ErrorError return before enter: %v\n", v)
- // continue
- // }
- // currnt := funStack[len-1]
- // if currnt.Map.StackEvent.Location != 0 {
- // fmt.Printf("currnt StackEvent ErrorError is not enter: %v\n", v)
- // continue
- // }
- // if k < mapSliceLen-1 && len >= 2 {
- // nextfun := mapSlice[k+1]
- // preCurrnt := funStack[len-2]
- // // // 处理 .netcore 多次 returun
- // // 下一个事件是 return 并且函数名跟当前事件是一样的,且上一个函数不是当前函数
- // if nextfun.Map.StackEvent.Location == 1 && nextfun.Map.Uprobe.Funcname == currnt.Map.Uprobe.Funcname && preCurrnt.Map.Uprobe.Funcname != currnt.Map.Uprobe.Funcname {
- // currentfunNum++
- // continue
- // }
- // }
- // funStack = funStack[:len-1]
- // duration := v.Map.StackEvent.TimeNsEnd - currnt.Map.StackEvent.TimeNsStart
- // t.FuncTraceQuery(currnt.Map.Uprobe.Funcname, time.Duration(duration), currnt.Map.StackEvent.TimeNsStart, v.Map.StackEvent.TimeNsEnd, currentfunNum)
- // currentfunNum = 1
- // }
- // }
- //}
- func (t *Trace) startReady() {
- t.lock.Lock()
- defer t.lock.Unlock()
- t.startEventReady = true
- }
- func (t *Trace) endReadyEvent(needCount uint32) {
- t.lock.Lock()
- defer t.lock.Unlock()
- t.endEventReady = true
- t.needEventCount = needCount
- }
- func (t *Trace) AllEventReady(traceID uint64) bool {
- klog.Debugf("[AllEventReady (current/need)|start|end|traceid](%d/%d)%v|%v|%d", *t.currenEventCount, t.needEventCount, t.startEventReady, t.endEventReady, traceID)
- return t.startEventReady && t.endEventReady && *t.currenEventCount >= t.needEventCount
- }
- func (t *Trace) TraceStartEvent(method, path, sn string, sport uint16, status l7.Status, addr netaddr.IPPort, pid uint32, appInfo AppInfo, container_id string) {
- t.span.SetAttributes(semconv.HTTPURL(fmt.Sprintf("http://%s:%d%s", sn, sport, path)),
- semconv.HTTPMethod(method),
- attribute.String("http.uri", path))
- if status > 399 {
- t.span.SetStatus(codes.Error, "")
- }
- //host := fmt.Sprintf("%s:%s", sn, sport)
- t.destination = addr
- t.commonAttrs = []attribute.KeyValue{
- semconv.NetPeerName(sn),
- semconv.NetPeerPort(int(sport)),
- // buildAppMapFromEvent
- attribute.Int("server.code_type", appInfo.CodeType.Int()),
- attribute.String("server.app_name", appInfo.AppName),
- attribute.String("server.service_name", appInfo.ServiceType),
- attribute.Int64("server.app_id", appInfo.AppIdHash.IntVal),
- attribute.Int64("server.agent_id", appInfo.AgentId),
- attribute.Int64("server.instance_id", appInfo.InstanceIdHash.IntVal),
- attribute.Int("server.pid", int(pid)),
- attribute.String("server.container_id", container_id),
- }
- t.span.SetAttributes(t.commonAttrs...)
- t.startReady()
- }
- // set context span
- func (t *Trace) CreateRootSpan(traceId uint64) {
- traceIdStr := strconv.Itoa(int(traceId))
- ctx, span := tracer(t.containerId).Start(context.Background(), traceIdStr, trace.WithSpanKind(trace.SpanKindClient))
- t.setContext(ctx)
- t.setSpan(span)
- }
- func (t *Trace) TraceStart(method, path string, status l7.Status, duration time.Duration) {
- if t == nil || method == "" {
- return
- }
- t.createParentSpan("APPLICATION", duration, status >= 400,
- semconv.HTTPURL(fmt.Sprintf("http://%s%s", t.destination.String(), path)),
- semconv.HTTPMethod(method),
- //semconv.HTTPStatusCode(int(status)),
- attribute.String("http.uri", path),
- )
- }
- func (t *Trace) TraceEnd(r *l7.RequestData) {
- if t == nil {
- return
- }
- t.span.SetAttributes(
- semconv.HTTPStatusCode(int(r.Status)),
- attribute.String("server.trace_id_from", r.ParentSpanContext.TraceIdFrom),
- )
- CalledId, err := strconv.ParseInt(r.ParentSpanContext.CalledId, 10, 64)
- if err == nil && CalledId != 0 {
- t.span.SetAttributes(attribute.Int64("server.called_id", CalledId))
- }
- InstanceIdFrom, err := strconv.ParseInt(r.ParentSpanContext.InstanceIdFrom, 10, 64)
- if err == nil && InstanceIdFrom != 0 {
- t.span.SetAttributes(attribute.Int64("server.instance_id_from", InstanceIdFrom))
- }
- AppIdFrom, err := strconv.ParseInt(r.ParentSpanContext.AppIdFrom, 10, 64)
- if err == nil && AppIdFrom != 0 {
- t.span.SetAttributes(attribute.Int64("server.app_id_from", AppIdFrom))
- }
- if r.ParentSpanContext.SpanIdFrom != "0000000000000000" {
- t.span.SetAttributes(attribute.String("server.span_id_from", r.ParentSpanContext.SpanIdFrom))
- }
- // klog.Debugf("r.ParentSpanContext.TypeFrom tpp %s", r.ParentSpanContext.TypeFrom)
- if r.ParentSpanContext.TypeFrom == "10" {
- t.span.SetAttributes(attribute.String("server.type_from", "SDK"))
- t.span.SetAttributes(attribute.String("server.app_id_from", "-1"))
- t.span.SetAttributes(attribute.String("server.span_id_from", r.ParentSpanContext.TraceIdFrom))
- t.span.SetAttributes(attribute.String("server.instance_id_from", "-1"))
- t.span.SetAttributes(attribute.String("server.call_id", "-1"))
- }
- // for _, v := range t.stack {
- // fmt.Printf("TraceEndTraceEndTraceEnd%s\n", v)
- // }
- //t.buildFun()
- t.span.End(trace.WithTimestamp(time.Now()))
- }
- // 新增结束事件
- func (t *Trace) TraceEndEvent(r *l7.RequestData) {
- if t == nil {
- return
- }
- var attr []attribute.KeyValue
- attr = append(attr,
- semconv.HTTPStatusCode(int(r.Status)),
- attribute.String("server.trace_id_from", r.ParentSpanContext.TraceIdFrom),
- )
- //t.span.SetAttributes(
- // semconv.HTTPStatusCode(int(r.Status)),
- // attribute.String("server.trace_id_from", r.ParentSpanContext.TraceIdFrom),
- //)
- calledId, err := strconv.ParseInt(r.ParentSpanContext.CalledId, 10, 64)
- if err == nil && calledId != 0 {
- attr = append(attr, attribute.Int64("server.called_id", calledId))
- //t.span.SetAttributes(attribute.Int64("server.called_id", CalledId))
- }
- instanceIdFrom, err := strconv.ParseInt(r.ParentSpanContext.InstanceIdFrom, 10, 64)
- if err == nil && instanceIdFrom != 0 {
- attr = append(attr, attribute.Int64("server.instance_id_from", instanceIdFrom))
- //t.span.SetAttributes(attribute.Int64("server.instance_id_from", InstanceIdFrom))
- }
- appIdFrom, err := strconv.ParseInt(r.ParentSpanContext.AppIdFrom, 10, 64)
- if err == nil && appIdFrom != 0 {
- attr = append(attr, attribute.Int64("server.app_id_from", appIdFrom))
- //t.span.SetAttributes(attribute.Int64("server.app_id_from", AppIdFrom))
- }
- if r.ParentSpanContext.SpanIdFrom != "0000000000000000" {
- attr = append(attr, attribute.String("server.span_id_from", r.ParentSpanContext.SpanIdFrom))
- //t.span.SetAttributes(attribute.String("server.span_id_from", r.ParentSpanContext.SpanIdFrom))
- }
- // klog.Debugf("r.ParentSpanContext.TypeFrom event tpp %s", r.ParentSpanContext.TypeFrom)
- if r.ParentSpanContext.TypeFrom == "10" {
- attr = append(attr, attribute.String("server.type_from", "SDK"))
- attr = append(attr, attribute.String("server.app_id_from", "-1"))
- attr = append(attr, attribute.String("server.span_id_from", r.ParentSpanContext.TraceIdFrom))
- attr = append(attr, attribute.String("server.instance_id_from", "-1"))
- attr = append(attr, attribute.String("server.call_id", "-1"))
- }
- attr = append(attr, attribute.String("server.src_addr", r.DAddr.String()))
- attr = append(attr, attribute.String("server.dst_addr", r.SAddr.String()))
- klog.Debugf("TraceEndEvent SAddr.String %s", r.SAddr.String())
- klog.Debugf("TraceEndEvent DAddr.String %s", r.DAddr.String())
- t.appendTimestamp(&attr, r.StartAt, r.EndAt, r.Duration.Nanoseconds())
- t.span.SetAttributes(attr...)
- t.endReadyEvent(r.EventCount)
- }
- func (t *Trace) appendTimestamp(attr *[]attribute.KeyValue, s, e uint64, d int64) {
- *attr = append(*attr,
- attribute.Int64("time.start_at", utils.KtimeToTimestamp(s)),
- attribute.Int64("time.end_at", utils.KtimeToTimestamp(e)),
- attribute.Int64("time.duration", d),
- )
- }
- func (t *Trace) createParentSpan(name string, duration time.Duration, error bool, attrs ...attribute.KeyValue) {
- end := time.Now()
- start := end.Add(-duration)
- ctx, span := tracer(t.containerId).Start(context.Background(), name, trace.WithTimestamp(start), trace.WithSpanKind(trace.SpanKindClient))
- span.SetAttributes(attrs...)
- span.SetAttributes(t.commonAttrs...)
- if error {
- span.SetStatus(codes.Error, "")
- }
- t.setContext(ctx)
- t.setSpan(span)
- }
- func (t *Trace) SendEvent() {
- t.span.End()
- }
- func (t *Trace) GetSpan() trace.Span {
- return t.span
- }
- func (t *Trace) createTraceEvent(name string, eventType int, l7Type int, attrs ...attribute.KeyValue) {
- t.span.AddEventApm(name, eventType, l7Type, trace.WithAttributes(attrs...))
- //atomic.AddUint32(t.currenEventCount, 1)
- }
- func (t *Trace) addEvent() {
- atomic.AddUint32(t.currenEventCount, 1)
- }
- func (t *Trace) createTraceSpan(name string, duration time.Duration, error bool, attrs ...attribute.KeyValue) {
- end := time.Now()
- start := end.Add(-duration)
- //fmt.Println("createTraceSpan:", t.ctx)
- _, span := tracer(t.containerId).Start(t.ctx, name, trace.WithTimestamp(start), trace.WithSpanKind(trace.SpanKindClient))
- span.SetAttributes(t.commonAttrs...)
- span.SetAttributes(attrs...)
- if error {
- span.SetStatus(codes.Error, "")
- }
- span.End(trace.WithTimestamp(end))
- }
- func (t *Trace) MysqlTraceQuery(query string, error bool, duration time.Duration, destination netaddr.IPPort) {
- if t == nil || query == "" {
- return
- }
- t.createTraceSpan(l7.ProtocolMysql.String(), duration, error,
- semconv.DBSystemMySQL,
- semconv.DBStatement(query),
- semconv.NetPeerName(destination.IP().String()),
- semconv.NetPeerPort(int(destination.Port())),
- )
- }
- // 内置允许操作(始终放行)
- var builtinOps = map[string]struct{}{
- "SELECT": {},
- "INSERT": {},
- "UPDATE": {},
- "DELETE": {},
- "WITH": {},
- }
- var (
- allowedOps map[string]struct{}
- initOnce sync.Once
- )
- // 解析并缓存环境变量中的前缀
- func initAllowedOps() {
- allowedOps = make(map[string]struct{})
- env := os.Getenv("ALLOWED_SQL_PREFIXES")
- for _, s := range strings.Split(env, ",") {
- if trimmed := strings.ToUpper(strings.TrimSpace(s)); trimmed != "" {
- allowedOps[trimmed] = struct{}{}
- }
- }
- }
- func isCURDOperation(q string) bool {
- if len(q) < 6 {
- return false
- }
- // 提取前缀(考虑最长10个字符)
- prefix := strings.ToUpper(strings.TrimSpace(q))
- if len(prefix) > 10 {
- prefix = prefix[:10]
- }
- // 提取第一个单词作为操作关键字
- spaceIdx := strings.IndexAny(prefix, " \t\n")
- if spaceIdx > 0 {
- prefix = prefix[:spaceIdx]
- }
- // 先查固定允许的操作
- if _, ok := builtinOps[prefix]; ok {
- return true
- }
- // 加载环境变量控制的操作(只执行一次)
- initOnce.Do(initAllowedOps)
- // 查额外允许操作
- _, ok := allowedOps[prefix]
- return ok
- }
- func (t *Trace) SQLTraceQueryEvent(l7Type l7.Protocol, semconvVal attribute.KeyValue, query string, r *l7.RequestData, destination netaddr.IPPort) {
- //fmt.Println("query", query)
- if t == nil {
- return
- }
- t.addEvent()
- // 只保留增删改查基本操作
- if !isCURDOperation(query) {
- return
- }
- var attr []attribute.KeyValue
- attr = append(attr,
- semconvVal,
- semconv.DBStatement(query),
- semconv.NetPeerName(destination.IP().String()),
- semconv.NetPeerPort(int(destination.Port())),
- attribute.String("sql.src_addr", r.ComponentSAddr.String()),
- attribute.String("sql.destination_addr", r.ComponentDAddr.String()),
- )
- t.appendTimestamp(&attr, r.StartAt, r.EndAt, r.Duration.Nanoseconds())
- attr = append(attr,
- attribute.Bool("sql.exception", r.Status.Error()),
- )
- if r.Status.Error() {
- attr = append(attr,
- attribute.String("sql.exception_msg", r.ErrorMsg),
- )
- }
- t.createTraceEvent(l7Type.String(), ebpftracer.EventTypeL7Request.Int(), l7Type.Int(), attr...)
- }
- //func (t *Trace) MysqlTraceQueryEvent(query string, r *l7.RequestData, destination netaddr.IPPort) {
- // fmt.Println("query", query)
- // if t == nil {
- // return
- // }
- //
- // var attr []attribute.KeyValue
- // attr = append(attr,
- // semconv.DBSystemMySQL,
- // semconv.DBStatement(query),
- // semconv.NetPeerName(destination.IP().String()),
- // semconv.NetPeerPort(int(destination.Port())),
- // attribute.String("sql.src_addr", r.ComponentSAddr.String()),
- // attribute.String("sql.destination_addr", r.ComponentDAddr.String()),
- // )
- // t.appendTimestamp(&attr, r.StartAt, r.EndAt, r.Duration.Nanoseconds())
- //
- // attr = append(attr,
- // attribute.Bool("sql.exception", r.Status.Error()),
- // )
- // t.createTraceEvent(l7.ProtocolMysql.String(), int(ebpftracer.EventTypeL7Request), int(l7.ProtocolMysql), attr...)
- //}
- //func (t *Trace) PostGreSqlTraceQuery(query string, error bool, duration time.Duration, destination netaddr.IPPort) {
- // if t == nil || query == "" {
- // return
- // }
- // t.createTraceSpan(l7.ProtocolPostgres.String(), duration, error,
- // semconv.DBSystemPostgreSQL,
- // semconv.DBStatement(query),
- // semconv.NetPeerName(destination.IP().String()),
- // semconv.NetPeerPort(int(destination.Port())),
- // )
- //}
- //func (t *Trace) PostGreSqlTraceQueryEvent(query string, r *l7.RequestData, destination netaddr.IPPort) {
- // fmt.Println("query", query)
- // if t == nil {
- // return
- // }
- //
- // var attr []attribute.KeyValue
- // attr = append(attr,
- // semconv.DBSystemPostgreSQL,
- // semconv.DBStatement(query),
- // semconv.NetPeerName(destination.IP().String()),
- // semconv.NetPeerPort(int(destination.Port())),
- // attribute.String("sql.src_addr", r.ComponentSAddr.String()),
- // attribute.String("sql.destination_addr", r.ComponentDAddr.String()),
- // )
- // t.appendTimestamp(&attr, r.StartAt, r.EndAt, r.Duration.Nanoseconds())
- // attr = append(attr,
- // attribute.Bool("sql.exception", r.Status.Error()),
- // )
- // t.createTraceEvent(l7.ProtocolPostgres.String(), int(ebpftracer.EventTypeL7Request), int(l7.ProtocolPostgres), attr...)
- //}
- //func (t *Trace) DmTraceQueryEvent(query string, r *l7.RequestData, destination netaddr.IPPort) {
- // if t == nil || query == "" {
- // return
- // }
- // l7Type := int(l7.ProtocolDM)
- // var attr []attribute.KeyValue
- // attr = append(attr,
- // semconv.DBSystemDaMengDB,
- // semconv.DBStatement(query),
- // semconv.NetPeerName(destination.IP().String()),
- // semconv.NetPeerPort(int(destination.Port())),
- // attribute.String("sql.src_addr", r.ComponentSAddr.String()),
- // attribute.String("sql.destination_addr", r.ComponentDAddr.String()),
- // )
- // t.appendTimestamp(&attr, r.StartAt, r.EndAt, r.Duration.Nanoseconds())
- //
- // attr = append(attr,
- // attribute.Bool("sql.exception", r.Status.Error()),
- // )
- // t.createTraceEvent(l7.ProtocolHTTP.String(), int(ebpftracer.EventTypeL7Request), l7Type, attr...)
- //}
- //func (t *Trace) RedisTraceQuery(cmd, args string, error bool, duration time.Duration) {
- // if t == nil || cmd == "" {
- // return
- // }
- // statement := cmd
- // if args != "" {
- // statement += " " + args
- // }
- // t.createTraceSpan(l7.ProtocolRedis.String(), duration, error,
- // semconv.DBSystemRedis,
- // semconv.DBOperation(cmd),
- // semconv.DBStatement(statement),
- // )
- //}
- func (t *Trace) RedisTraceQueryEvent(cmd, args string, r *l7.RequestData, destination netaddr.IPPort) {
- if t == nil {
- return
- }
- t.addEvent()
- if cmd == "" {
- return
- }
- statement := cmd
- if args != "" {
- statement += " " + args
- }
- var attr []attribute.KeyValue
- attr = append(attr,
- semconv.DBSystemRedis,
- semconv.DBOperation(cmd),
- semconv.DBStatement(statement),
- semconv.NetPeerName(destination.IP().String()),
- semconv.NetPeerPort(int(destination.Port())),
- attribute.String("nosql.src_addr", r.ComponentSAddr.String()),
- attribute.String("nosql.destination_addr", r.ComponentDAddr.String()),
- )
- t.appendTimestamp(&attr, r.StartAt, r.EndAt, r.Duration.Nanoseconds())
- t.createTraceEvent(l7.ProtocolRedis.String(), ebpftracer.EventTypeL7Request.Int(), l7.ProtocolRedis.Int(), attr...)
- }
- func (t *Trace) MongoTraceQueryEvent(query string, r *l7.RequestData, destination netaddr.IPPort) {
- if t == nil {
- return
- }
- t.addEvent()
- if query == "" {
- return
- }
- var attr []attribute.KeyValue
- attr = append(attr,
- semconv.DBSystemRedis,
- semconv.DBStatement(query),
- semconv.NetPeerName(destination.IP().String()),
- semconv.NetPeerPort(int(destination.Port())),
- attribute.String("nosql.src_addr", r.ComponentSAddr.String()),
- attribute.String("nosql.destination_addr", r.ComponentDAddr.String()),
- )
- t.appendTimestamp(&attr, r.StartAt, r.EndAt, r.Duration.Nanoseconds())
- t.createTraceEvent(l7.ProtocolMongo.String(), ebpftracer.EventTypeL7Request.Int(), l7.ProtocolMongo.Int(), attr...)
- }
- func (t *Trace) DNSTraceQueryEvent(r *l7.RequestData, _type string, fqdn string, ttl uint32, ips []netaddr.IP) {
- if t == nil {
- return
- }
- t.addEvent()
- var attr []attribute.KeyValue
- var ipList []string
- for _, ip := range ips {
- ipList = append(ipList, ip.String())
- }
- ip := strings.Join(ipList, ",")
- attr = append(attr,
- attribute.String("dns.type", _type),
- attribute.String("dns.fqdn", fqdn),
- attribute.Int("dns.ttl", int(ttl)),
- attribute.String("dns.ips", ip),
- )
- t.appendTimestamp(&attr, r.StartAt, r.EndAt, r.Duration.Nanoseconds())
- t.createTraceEvent(l7.ProtocolDNS.String(), ebpftracer.EventTypeL7Request.Int(), l7.ProtocolDNS.Int(), attr...)
- }
- func (t *Trace) HttpTraceRequest(method, path, ip string, port uint16, r *l7.RequestData) {
- if t == nil || method == "" {
- return
- }
- assumedAppID, err := strconv.ParseInt(r.AssumedAppId, 10, 64)
- if err != nil {
- assumedAppID = 0
- }
- status := r.Status
- duration := r.Duration
- t.createTraceSpan(l7.ProtocolHTTP.String(), duration, status >= 400,
- semconv.HTTPURL(fmt.Sprintf("http://%s%s", t.destination.String(), path)),
- semconv.HTTPMethod(method),
- semconv.HTTPStatusCode(int(status)),
- attribute.String("http.uri", path),
- attribute.String("http.ip", ip),
- attribute.Int64("http.assumed_app_id", assumedAppID),
- attribute.String("http.span_id", r.SpanId),
- attribute.Int("http.port", int(port)),
- )
- }
- // 新增事件处理
- func (t *Trace) HttpTraceRequestEvent(method, path, ip string, port uint16, r *l7.RequestData) {
- if t == nil {
- return
- }
- t.addEvent()
- if method == "" {
- return
- }
- assumedAppID, err := strconv.ParseInt(r.AssumedAppId, 10, 64)
- if err != nil {
- assumedAppID = 0
- }
- status := r.Status
- var attr []attribute.KeyValue
- attr = append(attr,
- semconv.HTTPURL(fmt.Sprintf("http://%s:%d%s", ip, port, path)),
- semconv.HTTPMethod(method),
- semconv.HTTPStatusCode(int(status)),
- attribute.Bool("http.status_error", status > 399),
- attribute.String("http.uri", path),
- attribute.String("http.ip", ip),
- attribute.Int64("http.assumed_app_id", assumedAppID),
- attribute.String("http.span_id", r.SpanId),
- attribute.Int("http.port", int(port)),
- attribute.String("http.src_addr", r.ComponentSAddr.String()),
- attribute.String("http.destination_addr", r.ComponentDAddr.String()),
- )
- t.appendTimestamp(&attr, r.StartAt, r.EndAt, r.Duration.Nanoseconds())
- t.createTraceEvent(l7.ProtocolHTTP.String(), ebpftracer.EventTypeL7Request.Int(), l7.ProtocolHTTP.Int(), attr...)
- }
- func (t *Trace) FuncTraceQuery(funcname string, duration time.Duration, start uint64, end uint64) {
- if t == nil {
- return
- }
- t.addEvent()
- if funcname == "" {
- return
- }
- t.createTraceSpanNoTime2(funcname, duration, false, start, end)
- //t.createTraceSpanNoTime2(funcname, duration, false, start, end, attribute.Int("num", num))
- }
- func (t *Trace) createTraceSpanNoTime(name string, duration time.Duration, error bool, start uint64, end uint64, attrs ...attribute.KeyValue) {
- // end := time.Now()
- // start := end.Add(-duration)
- startTime := time.Unix(0, int64(start))
- endTime := time.Unix(0, int64(end))
- //fmt.Println("createTraceSpan:", t.ctx)
- _, span := tracer(t.containerId).Start(t.ctx, name, trace.WithTimestamp(startTime), trace.WithSpanKind(trace.SpanKindClient))
- span.SetAttributes(t.commonAttrs...)
- span.SetAttributes(attrs...)
- if error {
- span.SetStatus(codes.Error, "")
- }
- span.End(trace.WithTimestamp(endTime))
- }
- func (t *Trace) createTraceSpanNoTime2(name string, duration time.Duration, error bool, start uint64, end uint64) {
- // end := time.Now()
- // start := end.Add(-duration)
- //startTime := time.Unix(0, int64(start))
- //endTime := time.Unix(0, int64(end))
- ////fmt.Println("createTraceSpan:", t.ctx)
- //_, span := tracer(t.containerId).Start(t.ctx, name, trace.WithTimestamp(startTime), trace.WithSpanKind(trace.SpanKindClient))
- //span.SetAttributes(t.commonAttrs...)
- //span.SetAttributes(attrs...)
- //if error {
- // span.SetStatus(codes.Error, "")
- //}
- //span.End(trace.WithTimestamp(endTime))
- //attrs = append([]attribute.KeyValue{
- // attribute.Int64("startAt", int64(start)),
- // attribute.Int64("endAt", int64(end)),
- //})
- var attr []attribute.KeyValue
- t.appendTimestamp(&attr, start, end, int64(end-start))
- t.createTraceEvent(name, ebpftracer.EventTypeFunEnt.Int(), 0, attr...)
- }
|