mirror of https://github.com/k3s-io/k3s
82 lines
2.2 KiB
Go
82 lines
2.2 KiB
Go
|
/*
|
||
|
|
Copyright 2015 The Kubernetes 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 storage
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"sync/atomic"
|
||
|
|
|
||
|
|
"k8s.io/apimachinery/pkg/api/meta"
|
||
|
|
"k8s.io/apimachinery/pkg/api/validation/path"
|
||
|
|
"k8s.io/apimachinery/pkg/runtime"
|
||
|
|
)
|
||
|
|
|
||
|
|
type SimpleUpdateFunc func(runtime.Object) (runtime.Object, error)
|
||
|
|
|
||
|
|
// SimpleUpdateFunc converts SimpleUpdateFunc into UpdateFunc
|
||
|
|
func SimpleUpdate(fn SimpleUpdateFunc) UpdateFunc {
|
||
|
|
return func(input runtime.Object, _ ResponseMeta) (runtime.Object, *uint64, error) {
|
||
|
|
out, err := fn(input)
|
||
|
|
return out, nil, err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func EverythingFunc(runtime.Object) bool {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
|
||
|
|
func NamespaceKeyFunc(prefix string, obj runtime.Object) (string, error) {
|
||
|
|
meta, err := meta.Accessor(obj)
|
||
|
|
if err != nil {
|
||
|
|
return "", err
|
||
|
|
}
|
||
|
|
name := meta.GetName()
|
||
|
|
if msgs := path.IsValidPathSegmentName(name); len(msgs) != 0 {
|
||
|
|
return "", fmt.Errorf("invalid name: %v", msgs)
|
||
|
|
}
|
||
|
|
return prefix + "/" + meta.GetNamespace() + "/" + name, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func NoNamespaceKeyFunc(prefix string, obj runtime.Object) (string, error) {
|
||
|
|
meta, err := meta.Accessor(obj)
|
||
|
|
if err != nil {
|
||
|
|
return "", err
|
||
|
|
}
|
||
|
|
name := meta.GetName()
|
||
|
|
if msgs := path.IsValidPathSegmentName(name); len(msgs) != 0 {
|
||
|
|
return "", fmt.Errorf("invalid name: %v", msgs)
|
||
|
|
}
|
||
|
|
return prefix + "/" + name, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// HighWaterMark is a thread-safe object for tracking the maximum value seen
|
||
|
|
// for some quantity.
|
||
|
|
type HighWaterMark int64
|
||
|
|
|
||
|
|
// Update returns true if and only if 'current' is the highest value ever seen.
|
||
|
|
func (hwm *HighWaterMark) Update(current int64) bool {
|
||
|
|
for {
|
||
|
|
old := atomic.LoadInt64((*int64)(hwm))
|
||
|
|
if current <= old {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
if atomic.CompareAndSwapInt64((*int64)(hwm), old, current) {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|