add opentracing spans for metrics

api/datastore/internal/datastoreutil/metrics.go

@@ -0,0 +1,122 @@
+package datastoreutil
+
+import (
+	"context"
+
+	"github.com/jmoiron/sqlx"
+	"github.com/opentracing/opentracing-go"
+	"gitlab-odx.oracle.com/odx/functions/api/models"
+)
+
+func MetricDS(ds models.Datastore) models.Datastore {
+	return &metricds{ds}
+}
+
+type metricds struct {
+	ds models.Datastore
+}
+
+func (m *metricds) GetApp(ctx context.Context, appName string) (*models.App, error) {
+	span, ctx := opentracing.StartSpanFromContext(ctx, "ds_get_app")
+	defer span.Finish()
+	return m.ds.GetApp(ctx, appName)
+}
+
+func (m *metricds) GetApps(ctx context.Context, filter *models.AppFilter) ([]*models.App, error) {
+	span, ctx := opentracing.StartSpanFromContext(ctx, "ds_get_apps")
+	defer span.Finish()
+	return m.ds.GetApps(ctx, filter)
+}
+
+func (m *metricds) InsertApp(ctx context.Context, app *models.App) (*models.App, error) {
+	span, ctx := opentracing.StartSpanFromContext(ctx, "ds_insert_app")
+	defer span.Finish()
+	return m.ds.InsertApp(ctx, app)
+}
+
+func (m *metricds) UpdateApp(ctx context.Context, app *models.App) (*models.App, error) {
+	span, ctx := opentracing.StartSpanFromContext(ctx, "ds_update_app")
+	defer span.Finish()
+	return m.ds.UpdateApp(ctx, app)
+}
+
+func (m *metricds) RemoveApp(ctx context.Context, appName string) error {
+	span, ctx := opentracing.StartSpanFromContext(ctx, "ds_remove_app")
+	defer span.Finish()
+	return m.ds.RemoveApp(ctx, appName)
+}
+
+func (m *metricds) GetRoute(ctx context.Context, appName, routePath string) (*models.Route, error) {
+	span, ctx := opentracing.StartSpanFromContext(ctx, "ds_get_route")
+	defer span.Finish()
+	return m.ds.GetRoute(ctx, appName, routePath)
+}
+
+func (m *metricds) GetRoutes(ctx context.Context, filter *models.RouteFilter) (routes []*models.Route, err error) {
+	span, ctx := opentracing.StartSpanFromContext(ctx, "ds_get_routes")
+	defer span.Finish()
+	return m.ds.GetRoutes(ctx, filter)
+}
+
+func (m *metricds) GetRoutesByApp(ctx context.Context, appName string, filter *models.RouteFilter) (routes []*models.Route, err error) {
+	span, ctx := opentracing.StartSpanFromContext(ctx, "ds_get_routes_by_app")
+	defer span.Finish()
+	return m.ds.GetRoutesByApp(ctx, appName, filter)
+}
+
+func (m *metricds) InsertRoute(ctx context.Context, route *models.Route) (*models.Route, error) {
+	span, ctx := opentracing.StartSpanFromContext(ctx, "ds_insert_route")
+	defer span.Finish()
+	return m.ds.InsertRoute(ctx, route)
+}
+
+func (m *metricds) UpdateRoute(ctx context.Context, route *models.Route) (*models.Route, error) {
+	span, ctx := opentracing.StartSpanFromContext(ctx, "ds_update_route")
+	defer span.Finish()
+	return m.ds.UpdateRoute(ctx, route)
+}
+
+func (m *metricds) RemoveRoute(ctx context.Context, appName, routePath string) error {
+	span, ctx := opentracing.StartSpanFromContext(ctx, "ds_remove_route")
+	defer span.Finish()
+	return m.ds.RemoveRoute(ctx, appName, routePath)
+}
+
+func (m *metricds) InsertTask(ctx context.Context, task *models.Task) error {
+	span, ctx := opentracing.StartSpanFromContext(ctx, "ds_insert_task")
+	defer span.Finish()
+	return m.ds.InsertTask(ctx, task)
+}
+
+func (m *metricds) GetTask(ctx context.Context, callID string) (*models.FnCall, error) {
+	span, ctx := opentracing.StartSpanFromContext(ctx, "ds_get_task")
+	defer span.Finish()
+	return m.ds.GetTask(ctx, callID)
+}
+
+func (m *metricds) GetTasks(ctx context.Context, filter *models.CallFilter) (models.FnCalls, error) {
+	span, ctx := opentracing.StartSpanFromContext(ctx, "ds_get_tasks")
+	defer span.Finish()
+	return m.ds.GetTasks(ctx, filter)
+}
+
+func (m *metricds) InsertLog(ctx context.Context, callID string, callLog string) error {
+	span, ctx := opentracing.StartSpanFromContext(ctx, "ds_insert_log")
+	defer span.Finish()
+	return m.ds.InsertLog(ctx, callID, callLog)
+}
+
+func (m *metricds) GetLog(ctx context.Context, callID string) (*models.FnCallLog, error) {
+	span, ctx := opentracing.StartSpanFromContext(ctx, "ds_get_log")
+	defer span.Finish()
+	return m.ds.GetLog(ctx, callID)
+}
+
+func (m *metricds) DeleteLog(ctx context.Context, callID string) error {
+	span, ctx := opentracing.StartSpanFromContext(ctx, "ds_delete_log")
+	defer span.Finish()
+	return m.ds.DeleteLog(ctx, callID)
+}
+
+// instant & no context ;)
+func (m *metricds) GetDatabase() *sqlx.DB { return m.ds.GetDatabase() }