-
Notifications
You must be signed in to change notification settings - Fork 103
/
enhancement_helper.go
78 lines (64 loc) · 2.61 KB
/
enhancement_helper.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
/*
Copyright 2022 The Katalyst 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 helper
import (
"encoding/json"
"sync"
"github.com/kubewharf/katalyst-core/pkg/util/general"
)
// EnhancementUpdateFunc provides a mechanism for user-specified enhancement kvs for specific key
// since we may need to set qos-level for some customized cases
type EnhancementUpdateFunc func(enhancementKVs, podAnnotations map[string]string) map[string]string
var enhancementUpdaters sync.Map
func RegisterEnhancementUpdateFunc(enhancementKey string, f EnhancementUpdateFunc) {
enhancementUpdaters.Store(enhancementKey, f)
}
func GetRegisteredEnhancementUpdateFuncs() map[string]EnhancementUpdateFunc {
updaters := make(map[string]EnhancementUpdateFunc)
enhancementUpdaters.Range(func(key, value interface{}) bool {
updaters[key.(string)] = value.(EnhancementUpdateFunc)
return true
})
return updaters
}
func GetRegisteredEnhancementUpdateFunc(enhancementKey string) EnhancementUpdateFunc {
value, ok := enhancementUpdaters.Load(enhancementKey)
if !ok {
return nil
}
return value.(EnhancementUpdateFunc)
}
// ParseKatalystQOSEnhancement parses enhancements from annotations by given key,
// since enhancement values are stored as k-v, so we should unmarshal it into maps.
func ParseKatalystQOSEnhancement(enhancements, podAnnotations map[string]string, enhancementKey string) (enhancementConfig map[string]string) {
defer func() {
updatedEnhancementConfig := enhancementConfig
enhancementUpdateFunc := GetRegisteredEnhancementUpdateFunc(enhancementKey)
if enhancementUpdateFunc != nil {
updatedEnhancementConfig = enhancementUpdateFunc(enhancementConfig, podAnnotations)
general.Infof("update enhancementConfig from %+v to %+v", enhancementConfig, updatedEnhancementConfig)
}
enhancementConfig = updatedEnhancementConfig
}()
enhancementValue, ok := enhancements[enhancementKey]
if !ok {
return nil
}
enhancementConfig = map[string]string{}
err := json.Unmarshal([]byte(enhancementValue), &enhancementConfig)
if err != nil {
general.Errorf("parse enhancement %s failed: %v", enhancementKey, err)
return nil
}
return enhancementConfig
}