-
Notifications
You must be signed in to change notification settings - Fork 1k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
…4397) * Add gotmpl * Add shared retry pkg * Use template retry pkg in otlpconfig * Update license-check to look at first 4 lines * Template out all otlptrace internal * Add envconfig pkg to otlptrace/internal * Generate otlptrace/internal/otlpconfig * Revert templatizing otlptracegrpc * Add changes to changelog * Fix lint --------- Co-authored-by: Robert Pająk <pellared@hotmail.com>
- Loading branch information
Showing
59 changed files
with
5,075 additions
and
95 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
202 changes: 202 additions & 0 deletions
202
exporters/otlp/otlptrace/internal/envconfig/envconfig.go
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,202 @@ | ||
// Code created by gotmpl. DO NOT MODIFY. | ||
// source: internal/shared/otlp/envconfig/envconfig.go.tmpl | ||
|
||
// 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 envconfig // import "go.opentelemetry.io/otel/exporters/otlp/otlptrace/internal/envconfig" | ||
|
||
import ( | ||
"crypto/tls" | ||
"crypto/x509" | ||
"errors" | ||
"fmt" | ||
"net/url" | ||
"strconv" | ||
"strings" | ||
"time" | ||
|
||
"go.opentelemetry.io/otel/internal/global" | ||
) | ||
|
||
// ConfigFn is the generic function used to set a config. | ||
type ConfigFn func(*EnvOptionsReader) | ||
|
||
// EnvOptionsReader reads the required environment variables. | ||
type EnvOptionsReader struct { | ||
GetEnv func(string) string | ||
ReadFile func(string) ([]byte, error) | ||
Namespace string | ||
} | ||
|
||
// Apply runs every ConfigFn. | ||
func (e *EnvOptionsReader) Apply(opts ...ConfigFn) { | ||
for _, o := range opts { | ||
o(e) | ||
} | ||
} | ||
|
||
// GetEnvValue gets an OTLP environment variable value of the specified key | ||
// using the GetEnv function. | ||
// This function prepends the OTLP specified namespace to all key lookups. | ||
func (e *EnvOptionsReader) GetEnvValue(key string) (string, bool) { | ||
v := strings.TrimSpace(e.GetEnv(keyWithNamespace(e.Namespace, key))) | ||
return v, v != "" | ||
} | ||
|
||
// WithString retrieves the specified config and passes it to ConfigFn as a string. | ||
func WithString(n string, fn func(string)) func(e *EnvOptionsReader) { | ||
return func(e *EnvOptionsReader) { | ||
if v, ok := e.GetEnvValue(n); ok { | ||
fn(v) | ||
} | ||
} | ||
} | ||
|
||
// WithBool returns a ConfigFn that reads the environment variable n and if it exists passes its parsed bool value to fn. | ||
func WithBool(n string, fn func(bool)) ConfigFn { | ||
return func(e *EnvOptionsReader) { | ||
if v, ok := e.GetEnvValue(n); ok { | ||
b := strings.ToLower(v) == "true" | ||
fn(b) | ||
} | ||
} | ||
} | ||
|
||
// WithDuration retrieves the specified config and passes it to ConfigFn as a duration. | ||
func WithDuration(n string, fn func(time.Duration)) func(e *EnvOptionsReader) { | ||
return func(e *EnvOptionsReader) { | ||
if v, ok := e.GetEnvValue(n); ok { | ||
d, err := strconv.Atoi(v) | ||
if err != nil { | ||
global.Error(err, "parse duration", "input", v) | ||
return | ||
} | ||
fn(time.Duration(d) * time.Millisecond) | ||
} | ||
} | ||
} | ||
|
||
// WithHeaders retrieves the specified config and passes it to ConfigFn as a map of HTTP headers. | ||
func WithHeaders(n string, fn func(map[string]string)) func(e *EnvOptionsReader) { | ||
return func(e *EnvOptionsReader) { | ||
if v, ok := e.GetEnvValue(n); ok { | ||
fn(stringToHeader(v)) | ||
} | ||
} | ||
} | ||
|
||
// WithURL retrieves the specified config and passes it to ConfigFn as a net/url.URL. | ||
func WithURL(n string, fn func(*url.URL)) func(e *EnvOptionsReader) { | ||
return func(e *EnvOptionsReader) { | ||
if v, ok := e.GetEnvValue(n); ok { | ||
u, err := url.Parse(v) | ||
if err != nil { | ||
global.Error(err, "parse url", "input", v) | ||
return | ||
} | ||
fn(u) | ||
} | ||
} | ||
} | ||
|
||
// WithCertPool returns a ConfigFn that reads the environment variable n as a filepath to a TLS certificate pool. If it exists, it is parsed as a crypto/x509.CertPool and it is passed to fn. | ||
func WithCertPool(n string, fn func(*x509.CertPool)) ConfigFn { | ||
return func(e *EnvOptionsReader) { | ||
if v, ok := e.GetEnvValue(n); ok { | ||
b, err := e.ReadFile(v) | ||
if err != nil { | ||
global.Error(err, "read tls ca cert file", "file", v) | ||
return | ||
} | ||
c, err := createCertPool(b) | ||
if err != nil { | ||
global.Error(err, "create tls cert pool") | ||
return | ||
} | ||
fn(c) | ||
} | ||
} | ||
} | ||
|
||
// WithClientCert returns a ConfigFn that reads the environment variable nc and nk as filepaths to a client certificate and key pair. If they exists, they are parsed as a crypto/tls.Certificate and it is passed to fn. | ||
func WithClientCert(nc, nk string, fn func(tls.Certificate)) ConfigFn { | ||
return func(e *EnvOptionsReader) { | ||
vc, okc := e.GetEnvValue(nc) | ||
vk, okk := e.GetEnvValue(nk) | ||
if !okc || !okk { | ||
return | ||
} | ||
cert, err := e.ReadFile(vc) | ||
if err != nil { | ||
global.Error(err, "read tls client cert", "file", vc) | ||
return | ||
} | ||
key, err := e.ReadFile(vk) | ||
if err != nil { | ||
global.Error(err, "read tls client key", "file", vk) | ||
return | ||
} | ||
crt, err := tls.X509KeyPair(cert, key) | ||
if err != nil { | ||
global.Error(err, "create tls client key pair") | ||
return | ||
} | ||
fn(crt) | ||
} | ||
} | ||
|
||
func keyWithNamespace(ns, key string) string { | ||
if ns == "" { | ||
return key | ||
} | ||
return fmt.Sprintf("%s_%s", ns, key) | ||
} | ||
|
||
func stringToHeader(value string) map[string]string { | ||
headersPairs := strings.Split(value, ",") | ||
headers := make(map[string]string) | ||
|
||
for _, header := range headersPairs { | ||
n, v, found := strings.Cut(header, "=") | ||
if !found { | ||
global.Error(errors.New("missing '="), "parse headers", "input", header) | ||
continue | ||
} | ||
name, err := url.QueryUnescape(n) | ||
if err != nil { | ||
global.Error(err, "escape header key", "key", n) | ||
continue | ||
} | ||
trimmedName := strings.TrimSpace(name) | ||
value, err := url.QueryUnescape(v) | ||
if err != nil { | ||
global.Error(err, "escape header value", "value", v) | ||
continue | ||
} | ||
trimmedValue := strings.TrimSpace(value) | ||
|
||
headers[trimmedName] = trimmedValue | ||
} | ||
|
||
return headers | ||
} | ||
|
||
func createCertPool(certBytes []byte) (*x509.CertPool, error) { | ||
cp := x509.NewCertPool() | ||
if ok := cp.AppendCertsFromPEM(certBytes); !ok { | ||
return nil, errors.New("failed to append certificate to the cert pool") | ||
} | ||
return cp, nil | ||
} |
Oops, something went wrong.