start api call method

Author: gunlee01

scouterx/common/netdata/apicallstep.go

@@ -0,0 +1,39 @@
+package netdata
+
+type ApiCallStep struct {
+	SingleStep
+	Txid int64
+	Hash int32
+	Elapsed int32
+	CpuTime int32
+	Error int32
+	Opt uint8
+	Address string
+}
+
+func NewApiCallStep() *ApiCallStep {
+	step := new(ApiCallStep)
+	return step
+}
+
+func (s *ApiCallStep) ApiCallStep() byte {
+	return APICALL
+}
+
+func (s *ApiCallStep) Write(out *DataOutputX) error {
+	err := s.SingleStep.Write(out)
+	if err != nil {
+		return err
+	}
+	_, err = out.WriteDecimal(s.Txid)
+	_, err = out.WriteDecimal32(s.Hash)
+	_, err = out.WriteDecimal32(s.Elapsed)
+	_, err = out.WriteDecimal32(s.CpuTime)
+	_, err = out.WriteDecimal32(s.Error)
+	_, err = out.WriteUInt8(s.Opt)
+	if s.Opt == 1 {
+		_, err = out.WriteString(s.Address)
+	}
+
+	return err
+}

scouterx/strace/tracemain.go

@@ -515,3 +515,37 @@ func profileHttpHeaders(r *http.Request, tctx *netio.TraceContext) {
 		tctx.Profile.Add(netdata.NewMessageStep(fmt.Sprintf("query: %s", r.URL.RawQuery), startTime))
 	}
 }
+
+func StartApiCall(ctx context.Context, apiCallName string, address string) *netdata.ApiCallStep {
+	defer common.ReportScouterPanic()
+	if ctx == nil {
+		return nil
+	}
+	tctx := tctxmanager.GetTraceContext(ctx)
+	if tctx == nil {
+		return nil
+	}
+
+	step := netdata.NewApiCallStep()
+	step.Hash = netio.SendApicall(apiCallName)
+	step.StartTime = util.MillisToNow(tctx.StartTime)
+	step.Address = address
+	tctx.Profile.Push(step)
+
+	return step
+}
+
+func EndApiCall(ctx context.Context, step *netdata.MethodStep) {
+	defer common.ReportScouterPanic()
+
+	if ctx == nil || step == nil {
+		return
+	}
+	tctx := tctxmanager.GetTraceContext(ctx)
+	if tctx == nil {
+		return
+	}
+	step.Elapsed = util.MillisToNow(tctx.StartTime) - step.StartTime
+	tctx.Profile.Pop(step)
+}
+