|
- // Copyright 2017 Google LLC
- //
- // Licensed under the Apache License, Version 2.0 (the "License");
- // you may not use this file except in compliance with the License.
- // You may obtain a copy of the License at
- //
- // http://www.apache.org/licenses/LICENSE-2.0
- //
- // Unless required by applicable law or agreed to in writing, software
- // distributed under the License is distributed on an "AS IS" BASIS,
- // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- // See the License for the specific language governing permissions and
- // limitations under the License.
-
- package spanner
-
- import (
- "context"
- "fmt"
-
- "go.opencensus.io/stats"
- "go.opencensus.io/stats/view"
- "go.opencensus.io/trace"
- )
-
- func startSpan(ctx context.Context, name string) context.Context {
- ctx, _ = trace.StartSpan(ctx, name)
- return ctx
- }
-
- func endSpan(ctx context.Context, err error) {
- span := trace.FromContext(ctx)
- if err != nil {
- // TODO(jba): Add error code to the status.
- span.SetStatus(trace.Status{Message: err.Error()})
- }
- span.End()
- }
-
- func statsPrintf(ctx context.Context, attrMap map[string]interface{}, format string, args ...interface{}) {
- var attrs []trace.Attribute
- for k, v := range attrMap {
- var a trace.Attribute
- switch v := v.(type) {
- case string:
- a = trace.StringAttribute(k, v)
- case bool:
- a = trace.BoolAttribute(k, v)
- case int:
- a = trace.Int64Attribute(k, int64(v))
- case int64:
- a = trace.Int64Attribute(k, v)
- default:
- a = trace.StringAttribute(k, fmt.Sprintf("%#v", v))
- }
- attrs = append(attrs, a)
- }
- trace.FromContext(ctx).Annotatef(attrs, format, args...)
- }
-
- const statsPrefix = "cloud.google.com/go/spanner/"
-
- func recordStat(ctx context.Context, m *stats.Int64Measure, n int64) {
- stats.Record(ctx, m.M(n))
- }
-
- var (
- // OpenSessionCount is a measure of the number of sessions currently opened.
- // It is EXPERIMENTAL and subject to change or removal without notice.
- OpenSessionCount = stats.Int64(statsPrefix+"open_session_count", "Number of sessions currently opened",
- stats.UnitDimensionless)
-
- // OpenSessionCountView is a view of the last value of OpenSessionCount.
- // It is EXPERIMENTAL and subject to change or removal without notice.
- OpenSessionCountView = &view.View{
- Name: OpenSessionCount.Name(),
- Description: OpenSessionCount.Description(),
- Measure: OpenSessionCount,
- Aggregation: view.LastValue(),
- }
- )
|