mirror of
https://gitea.com/Lydanne/buildx.git
synced 2025-05-18 00:47:48 +08:00
db4b96e62c
full diff: https://github.com/docker/cli/compare/v28.0.1...v28.0.2 Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
86 lines
2.8 KiB
Go
86 lines
2.8 KiB
Go
package manager
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/docker/cli/cli-plugins/metadata"
|
|
"github.com/spf13/cobra"
|
|
"go.opentelemetry.io/otel"
|
|
"go.opentelemetry.io/otel/attribute"
|
|
"go.opentelemetry.io/otel/baggage"
|
|
)
|
|
|
|
const (
|
|
// resourceAttributesEnvVar is the name of the envvar that includes additional
|
|
// resource attributes for OTEL as defined in the [OpenTelemetry specification].
|
|
//
|
|
// [OpenTelemetry specification]: https://opentelemetry.io/docs/specs/otel/configuration/sdk-environment-variables/#general-sdk-configuration
|
|
resourceAttributesEnvVar = "OTEL_RESOURCE_ATTRIBUTES"
|
|
|
|
// dockerCLIAttributePrefix is the prefix for any docker cli OTEL attributes.
|
|
//
|
|
// It is a copy of the const defined in [command.dockerCLIAttributePrefix].
|
|
dockerCLIAttributePrefix = "docker.cli."
|
|
cobraCommandPath = attribute.Key("cobra.command_path")
|
|
)
|
|
|
|
func getPluginResourceAttributes(cmd *cobra.Command, plugin Plugin) attribute.Set {
|
|
commandPath := cmd.Annotations[metadata.CommandAnnotationPluginCommandPath]
|
|
if commandPath == "" {
|
|
commandPath = fmt.Sprintf("%s %s", cmd.CommandPath(), plugin.Name)
|
|
}
|
|
|
|
attrSet := attribute.NewSet(
|
|
cobraCommandPath.String(commandPath),
|
|
)
|
|
|
|
kvs := make([]attribute.KeyValue, 0, attrSet.Len())
|
|
for iter := attrSet.Iter(); iter.Next(); {
|
|
attr := iter.Attribute()
|
|
kvs = append(kvs, attribute.KeyValue{
|
|
Key: dockerCLIAttributePrefix + attr.Key,
|
|
Value: attr.Value,
|
|
})
|
|
}
|
|
return attribute.NewSet(kvs...)
|
|
}
|
|
|
|
func appendPluginResourceAttributesEnvvar(env []string, cmd *cobra.Command, plugin Plugin) []string {
|
|
if attrs := getPluginResourceAttributes(cmd, plugin); attrs.Len() > 0 {
|
|
// Construct baggage members for each of the attributes.
|
|
// Ignore any failures as these aren't significant and
|
|
// represent an internal issue.
|
|
members := make([]baggage.Member, 0, attrs.Len())
|
|
for iter := attrs.Iter(); iter.Next(); {
|
|
attr := iter.Attribute()
|
|
m, err := baggage.NewMemberRaw(string(attr.Key), attr.Value.AsString())
|
|
if err != nil {
|
|
otel.Handle(err)
|
|
continue
|
|
}
|
|
members = append(members, m)
|
|
}
|
|
|
|
// Combine plugin added resource attributes with ones found in the environment
|
|
// variable. Our own attributes should be namespaced so there shouldn't be a
|
|
// conflict. We do not parse the environment variable because we do not want
|
|
// to handle errors in user configuration.
|
|
attrsSlice := make([]string, 0, 2)
|
|
if v := strings.TrimSpace(os.Getenv(resourceAttributesEnvVar)); v != "" {
|
|
attrsSlice = append(attrsSlice, v)
|
|
}
|
|
if b, err := baggage.New(members...); err != nil {
|
|
otel.Handle(err)
|
|
} else if b.Len() > 0 {
|
|
attrsSlice = append(attrsSlice, b.String())
|
|
}
|
|
|
|
if len(attrsSlice) > 0 {
|
|
env = append(env, resourceAttributesEnvVar+"="+strings.Join(attrsSlice, ","))
|
|
}
|
|
}
|
|
return env
|
|
}
|