mirror of
https://github.com/containerd/containerd.git
synced 2026-08-04 15:10:45 +00:00
2c102c6cbe
Avoid using logrus concepts in the API, use slog style log levels with integer values and 0 meaning the default "info" level. Signed-off-by: Derek McGowan <derek@mcg.dev>
92 lines
2.5 KiB
Go
92 lines
2.5 KiB
Go
/*
|
|
Copyright The containerd 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 bootstrap
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
|
|
"google.golang.org/protobuf/proto"
|
|
"google.golang.org/protobuf/types/known/anypb"
|
|
)
|
|
|
|
// LogLevelFromString converts a log level string (e.g. "debug", "info") to a
|
|
// LogLevel enum value. The accepted strings are compatible with logrus level names.
|
|
func LogLevelFromString(s string) LogLevel {
|
|
switch s {
|
|
case "trace":
|
|
return LogLevel_LOG_LEVEL_TRACE
|
|
case "debug":
|
|
return LogLevel_LOG_LEVEL_DEBUG
|
|
case "info":
|
|
return LogLevel_LOG_LEVEL_INFO
|
|
case "warn", "warning":
|
|
return LogLevel_LOG_LEVEL_WARN
|
|
case "error":
|
|
return LogLevel_LOG_LEVEL_ERROR
|
|
case "fatal":
|
|
return LogLevel_LOG_LEVEL_FATAL
|
|
case "panic":
|
|
return LogLevel_LOG_LEVEL_PANIC
|
|
default:
|
|
if v, err := strconv.ParseInt(s, 10, 32); err == nil {
|
|
return LogLevel(v)
|
|
}
|
|
return LogLevel_LOG_LEVEL_INFO
|
|
}
|
|
}
|
|
|
|
// AddExtension adds a new extension to the BootstrapParams.
|
|
// The message is wrapped in a google.protobuf.Any with its type URL automatically set.
|
|
// If the message is already an *anypb.Any, it is used directly without double-wrapping.
|
|
func (p *BootstrapParams) AddExtension(msg proto.Message) error {
|
|
var anyVal *anypb.Any
|
|
if a, ok := msg.(*anypb.Any); ok {
|
|
// Already an Any, use it directly
|
|
anyVal = a
|
|
} else {
|
|
var err error
|
|
anyVal, err = anypb.New(msg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
p.Extensions = append(p.Extensions, &Extension{Value: anyVal})
|
|
return nil
|
|
}
|
|
|
|
// FindExtension finds an extension matching the type of dst and unmarshals it.
|
|
func (p *BootstrapParams) FindExtension(dst proto.Message) (bool, error) {
|
|
if p == nil {
|
|
return false, nil
|
|
}
|
|
|
|
name := dst.ProtoReflect().Descriptor().FullName()
|
|
|
|
for _, ext := range p.Extensions {
|
|
if ext.GetValue().MessageIs(dst) {
|
|
if err := ext.GetValue().UnmarshalTo(dst); err != nil {
|
|
return false, fmt.Errorf("failed to unmarshal extension %q: %w", name, err)
|
|
}
|
|
return true, nil
|
|
}
|
|
}
|
|
|
|
return false, nil
|
|
}
|