mirror of
https://github.com/fnproject/fn.git
synced 2022-10-28 21:29:17 +03:00
206aa3c203
* update vendor directory, add go.opencensus.io * update imports * oops * s/opentracing/opencensus/ & remove prometheus / zipkin stuff & remove old stats * the dep train rides again * fix gin build * deps from last guy * start in on the agent metrics * she builds * remove tags for now, cardinality error is fussing. subscribe instead of register * update to patched version of opencensus to proceed for now TODO switch to a release * meh fix imports * println debug the bad boys * lace it with the tags * update deps again * fix all inconsistent cardinality errors * add our own logger * fix init * fix oom measure * remove bugged removal code * fix s3 measures * fix prom handler nil
119 lines
3.6 KiB
Go
119 lines
3.6 KiB
Go
package datastoreutil
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
|
|
"go.opencensus.io/trace"
|
|
|
|
"github.com/fnproject/fn/api/models"
|
|
"github.com/jmoiron/sqlx"
|
|
)
|
|
|
|
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) {
|
|
ctx, span := trace.StartSpan(ctx, "ds_get_app")
|
|
defer span.End()
|
|
return m.ds.GetApp(ctx, appName)
|
|
}
|
|
|
|
func (m *metricds) GetApps(ctx context.Context, filter *models.AppFilter) ([]*models.App, error) {
|
|
ctx, span := trace.StartSpan(ctx, "ds_get_apps")
|
|
defer span.End()
|
|
return m.ds.GetApps(ctx, filter)
|
|
}
|
|
|
|
func (m *metricds) InsertApp(ctx context.Context, app *models.App) (*models.App, error) {
|
|
ctx, span := trace.StartSpan(ctx, "ds_insert_app")
|
|
defer span.End()
|
|
return m.ds.InsertApp(ctx, app)
|
|
}
|
|
|
|
func (m *metricds) UpdateApp(ctx context.Context, app *models.App) (*models.App, error) {
|
|
ctx, span := trace.StartSpan(ctx, "ds_update_app")
|
|
defer span.End()
|
|
return m.ds.UpdateApp(ctx, app)
|
|
}
|
|
|
|
func (m *metricds) RemoveApp(ctx context.Context, appName string) error {
|
|
ctx, span := trace.StartSpan(ctx, "ds_remove_app")
|
|
defer span.End()
|
|
return m.ds.RemoveApp(ctx, appName)
|
|
}
|
|
|
|
func (m *metricds) GetRoute(ctx context.Context, appName, routePath string) (*models.Route, error) {
|
|
ctx, span := trace.StartSpan(ctx, "ds_get_route")
|
|
defer span.End()
|
|
return m.ds.GetRoute(ctx, appName, routePath)
|
|
}
|
|
|
|
func (m *metricds) GetRoutesByApp(ctx context.Context, appName string, filter *models.RouteFilter) (routes []*models.Route, err error) {
|
|
ctx, span := trace.StartSpan(ctx, "ds_get_routes_by_app")
|
|
defer span.End()
|
|
return m.ds.GetRoutesByApp(ctx, appName, filter)
|
|
}
|
|
|
|
func (m *metricds) InsertRoute(ctx context.Context, route *models.Route) (*models.Route, error) {
|
|
ctx, span := trace.StartSpan(ctx, "ds_insert_route")
|
|
defer span.End()
|
|
return m.ds.InsertRoute(ctx, route)
|
|
}
|
|
|
|
func (m *metricds) UpdateRoute(ctx context.Context, route *models.Route) (*models.Route, error) {
|
|
ctx, span := trace.StartSpan(ctx, "ds_update_route")
|
|
defer span.End()
|
|
return m.ds.UpdateRoute(ctx, route)
|
|
}
|
|
|
|
func (m *metricds) RemoveRoute(ctx context.Context, appName, routePath string) error {
|
|
ctx, span := trace.StartSpan(ctx, "ds_remove_route")
|
|
defer span.End()
|
|
return m.ds.RemoveRoute(ctx, appName, routePath)
|
|
}
|
|
|
|
func (m *metricds) InsertCall(ctx context.Context, call *models.Call) error {
|
|
ctx, span := trace.StartSpan(ctx, "ds_insert_call")
|
|
defer span.End()
|
|
return m.ds.InsertCall(ctx, call)
|
|
}
|
|
|
|
func (m *metricds) UpdateCall(ctx context.Context, from *models.Call, to *models.Call) error {
|
|
ctx, span := trace.StartSpan(ctx, "ds_update_call")
|
|
defer span.End()
|
|
return m.ds.UpdateCall(ctx, from, to)
|
|
}
|
|
|
|
func (m *metricds) GetCall(ctx context.Context, appName, callID string) (*models.Call, error) {
|
|
ctx, span := trace.StartSpan(ctx, "ds_get_call")
|
|
defer span.End()
|
|
return m.ds.GetCall(ctx, appName, callID)
|
|
}
|
|
|
|
func (m *metricds) GetCalls(ctx context.Context, filter *models.CallFilter) ([]*models.Call, error) {
|
|
ctx, span := trace.StartSpan(ctx, "ds_get_calls")
|
|
defer span.End()
|
|
return m.ds.GetCalls(ctx, filter)
|
|
}
|
|
|
|
func (m *metricds) InsertLog(ctx context.Context, appName, callID string, callLog io.Reader) error {
|
|
ctx, span := trace.StartSpan(ctx, "ds_insert_log")
|
|
defer span.End()
|
|
return m.ds.InsertLog(ctx, appName, callID, callLog)
|
|
}
|
|
|
|
func (m *metricds) GetLog(ctx context.Context, appName, callID string) (io.Reader, error) {
|
|
ctx, span := trace.StartSpan(ctx, "ds_get_log")
|
|
defer span.End()
|
|
return m.ds.GetLog(ctx, appName, callID)
|
|
}
|
|
|
|
// instant & no context ;)
|
|
func (m *metricds) GetDatabase() *sqlx.DB { return m.ds.GetDatabase() }
|