wip

Signed-off-by: Kevin Su <pingsutw@apache.org>

go/tasks/pluginmachinery/internal/webapi/core.go

@@ -70,7 +70,6 @@ func (c CorePlugin) GetProperties() core.PluginProperties {
 
 func (c CorePlugin) Handle(ctx context.Context, tCtx core.TaskExecutionContext) (core.Transition, error) {
 	c.metrics.NumberOfTasks.Inc(ctx)
-	start := time.Now()
 	incomingState, err := c.unmarshalState(ctx, tCtx.PluginStateReader())
 	if err != nil {
 		return core.UnknownTransition, err
@@ -100,7 +99,7 @@ func (c CorePlugin) Handle(ctx context.Context, tCtx core.TaskExecutionContext)
 	}
 	c.metrics.NumberOfTasks.Dec(ctx)
 	logger.Infof(ctx, "number of requests [%v]", c.metrics.NumberOfTasks)
-	logger.Infof(ctx, "request latency [%v]", time.Since(start).Round(time.Microsecond).String())
+	// logger.Infof(ctx, "request latency [%v]", time.Since(start).Round(time.Microsecond).String())
 	logger.Infof(ctx, "phaseInfo [%v]", phaseInfo)
 	return core.DoTransitionType(core.TransitionTypeBarrier, phaseInfo), nil
 }

go/tasks/plugins/webapi/fastapi/plugin.go

@@ -12,6 +12,7 @@ import (
 	"github.com/flyteorg/flytestdlib/logger"
 	"io/ioutil"
 	"net/http"
+	"time"
 
 	"github.com/flyteorg/flytestdlib/promutils"
 
@@ -84,7 +85,9 @@ func (p Plugin) Create(ctx context.Context, taskCtx webapi.TaskExecutionContextR
 		return nil, nil, err
 	}
 
+	start := time.Now()
 	resp, err := p.client.Do(req)
+	logger.Infof(ctx, "fastapi create request latency [%v]", time.Since(start).Round(time.Microsecond).String())
 	if err != nil {
 		return nil, nil, err
 	}