2b76578994
* Move tracing exporter registration This doesn't belong in the library and should be configured by the consumer of the opencensus package. * Rename `vkublet` package to `node` `vkubelet` does not convey any information to the consumers of the package. Really it would be nice to move this package to the root of the repo, but then you wind up with... interesting... import semantics due to the repo name... and after thinking about it some, a subpackage is really not so bad as long as it has a name that convey's some information. `node` was chosen since this package deals with all the semantics of operating a node in Kubernetes.
45 lines
1.4 KiB
Go
45 lines
1.4 KiB
Go
package root
|
|
|
|
import (
|
|
"github.com/virtual-kubelet/virtual-kubelet/errdefs"
|
|
"go.opencensus.io/trace"
|
|
)
|
|
|
|
type TracingExporterOptions struct {
|
|
Tags map[string]string
|
|
ServiceName string
|
|
}
|
|
|
|
var (
|
|
tracingExporters = make(map[string]TracingExporterInitFunc)
|
|
)
|
|
|
|
// TracingExporterInitFunc is the function that is called to initialize an exporter.
|
|
// This is used when registering an exporter and called when a user specifed they want to use the exporter.
|
|
type TracingExporterInitFunc func(TracingExporterOptions) (trace.Exporter, error)
|
|
|
|
// RegisterTracingExporter registers a tracing exporter.
|
|
// For a user to select an exporter, it must be registered here.
|
|
func RegisterTracingExporter(name string, f TracingExporterInitFunc) {
|
|
tracingExporters[name] = f
|
|
}
|
|
|
|
// GetTracingExporter gets the specified tracing exporter passing in the options to the exporter init function.
|
|
// For an exporter to be availbale here it must be registered with `RegisterTracingExporter`.
|
|
func GetTracingExporter(name string, opts TracingExporterOptions) (trace.Exporter, error) {
|
|
f, ok := tracingExporters[name]
|
|
if !ok {
|
|
return nil, errdefs.NotFoundf("tracing exporter %q not found", name)
|
|
}
|
|
return f(opts)
|
|
}
|
|
|
|
// AvailableTraceExporters gets the list of registered exporters
|
|
func AvailableTraceExporters() []string {
|
|
out := make([]string, 0, len(tracingExporters))
|
|
for k := range tracingExporters {
|
|
out = append(out, k)
|
|
}
|
|
return out
|
|
}
|