mirror of
https://github.com/superseriousbusiness/gotosocial
synced 2024-12-23 19:23:12 +00:00
108 lines
2.9 KiB
Go
108 lines
2.9 KiB
Go
// Copyright The OpenTelemetry Authors
|
|
//
|
|
// 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 resource // import "go.opentelemetry.io/otel/sdk/resource"
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/url"
|
|
"os"
|
|
"strings"
|
|
|
|
"go.opentelemetry.io/otel"
|
|
"go.opentelemetry.io/otel/attribute"
|
|
semconv "go.opentelemetry.io/otel/semconv/v1.17.0"
|
|
)
|
|
|
|
const (
|
|
// resourceAttrKey is the environment variable name OpenTelemetry Resource information will be read from.
|
|
resourceAttrKey = "OTEL_RESOURCE_ATTRIBUTES"
|
|
|
|
// svcNameKey is the environment variable name that Service Name information will be read from.
|
|
svcNameKey = "OTEL_SERVICE_NAME"
|
|
)
|
|
|
|
var (
|
|
// errMissingValue is returned when a resource value is missing.
|
|
errMissingValue = fmt.Errorf("%w: missing value", ErrPartialResource)
|
|
)
|
|
|
|
// fromEnv is a Detector that implements the Detector and collects
|
|
// resources from environment. This Detector is included as a
|
|
// builtin.
|
|
type fromEnv struct{}
|
|
|
|
// compile time assertion that FromEnv implements Detector interface.
|
|
var _ Detector = fromEnv{}
|
|
|
|
// Detect collects resources from environment.
|
|
func (fromEnv) Detect(context.Context) (*Resource, error) {
|
|
attrs := strings.TrimSpace(os.Getenv(resourceAttrKey))
|
|
svcName := strings.TrimSpace(os.Getenv(svcNameKey))
|
|
|
|
if attrs == "" && svcName == "" {
|
|
return Empty(), nil
|
|
}
|
|
|
|
var res *Resource
|
|
|
|
if svcName != "" {
|
|
res = NewSchemaless(semconv.ServiceName(svcName))
|
|
}
|
|
|
|
r2, err := constructOTResources(attrs)
|
|
|
|
// Ensure that the resource with the service name from OTEL_SERVICE_NAME
|
|
// takes precedence, if it was defined.
|
|
res, err2 := Merge(r2, res)
|
|
|
|
if err == nil {
|
|
err = err2
|
|
} else if err2 != nil {
|
|
err = fmt.Errorf("detecting resources: %s", []string{err.Error(), err2.Error()})
|
|
}
|
|
|
|
return res, err
|
|
}
|
|
|
|
func constructOTResources(s string) (*Resource, error) {
|
|
if s == "" {
|
|
return Empty(), nil
|
|
}
|
|
pairs := strings.Split(s, ",")
|
|
attrs := []attribute.KeyValue{}
|
|
var invalid []string
|
|
for _, p := range pairs {
|
|
field := strings.SplitN(p, "=", 2)
|
|
if len(field) != 2 {
|
|
invalid = append(invalid, p)
|
|
continue
|
|
}
|
|
k := strings.TrimSpace(field[0])
|
|
v, err := url.QueryUnescape(strings.TrimSpace(field[1]))
|
|
if err != nil {
|
|
// Retain original value if decoding fails, otherwise it will be
|
|
// an empty string.
|
|
v = field[1]
|
|
otel.Handle(err)
|
|
}
|
|
attrs = append(attrs, attribute.String(k, v))
|
|
}
|
|
var err error
|
|
if len(invalid) > 0 {
|
|
err = fmt.Errorf("%w: %v", errMissingValue, invalid)
|
|
}
|
|
return NewSchemaless(attrs...), err
|
|
}
|