Skip to content

Commit 4d7beb8

Browse files
committed
chore: ns/gw/cfgmap constants
For easier code reuse common defaults for namespace, gateway name and tier configmap has been extracted as constants. Signed-off-by: Bartosz Majsak <[email protected]>
1 parent 77ae8d9 commit 4d7beb8

File tree

5 files changed

+19
-14
lines changed

5 files changed

+19
-14
lines changed

maas-api/internal/config/config.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ package config
33
import (
44
"flag"
55

6+
"github.com/opendatahub-io/maas-billing/maas-api/internal/constant"
67
"k8s.io/utils/env"
78
)
89

@@ -40,8 +41,8 @@ func Load() *Config {
4041
defaultTeam, _ := env.GetBool("CREATE_DEFAULT_TEAM", true)
4142

4243
c := &Config{
43-
Name: env.GetString("INSTANCE_NAME", "openshift-ai-inference"),
44-
Namespace: env.GetString("NAMESPACE", "maas-api"),
44+
Name: env.GetString("INSTANCE_NAME", constant.DefaultGatewayName),
45+
Namespace: env.GetString("NAMESPACE", constant.DefaultNamespace),
4546
Port: env.GetString("PORT", "8080"),
4647
Provider: ProviderType(env.GetString("PROVIDER", string(SATokens))),
4748
DebugMode: debugMode,
@@ -54,7 +55,6 @@ func Load() *Config {
5455
CreateDefaultTeam: defaultTeam,
5556
AdminAPIKey: env.GetString("ADMIN_API_KEY", ""),
5657
}
57-
5858
c.bindFlags(flag.CommandLine)
5959

6060
return c
Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
package constant
2+
3+
const (
4+
TierMappingConfigMap = "tier-to-group-mapping"
5+
DefaultNamespace = "maas-api"
6+
DefaultGatewayName = "openshift-ai-inference"
7+
)

maas-api/internal/tier/mapper.go

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ import (
88
"sort"
99
"strings"
1010

11+
"github.com/opendatahub-io/maas-billing/maas-api/internal/constant"
1112
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
1213
"k8s.io/client-go/kubernetes"
1314
corev1typed "k8s.io/client-go/kubernetes/typed/core/v1"
@@ -16,10 +17,6 @@ import (
1617
"k8s.io/apimachinery/pkg/api/errors"
1718
)
1819

19-
const (
20-
MappingConfigMap = "tier-to-group-mapping"
21-
)
22-
2320
var defaultTier = Tier{
2421
Name: "free",
2522
Level: 0,
@@ -71,10 +68,10 @@ func (m *Mapper) GetTierForGroups(ctx context.Context, groups ...string) (string
7168
tiers, err := m.loadTierConfig(ctx)
7269
if err != nil {
7370
if errors.IsNotFound(err) {
74-
log.Printf("tier mapping %s not found, defaulting to 'free' tier", MappingConfigMap)
71+
log.Printf("tier mapping %s not found, defaulting to 'free' tier", constant.TierMappingConfigMap)
7572
return "free", nil
7673
}
77-
log.Printf("Failed to load tier configuration from ConfigMap %s: %v", MappingConfigMap, err)
74+
log.Printf("Failed to load tier configuration from ConfigMap %s: %v", constant.TierMappingConfigMap, err)
7875
return "", fmt.Errorf("failed to load tier configuration: %w", err)
7976
}
8077

@@ -94,14 +91,14 @@ func (m *Mapper) GetTierForGroups(ctx context.Context, groups ...string) (string
9491
}
9592

9693
func (m *Mapper) loadTierConfig(ctx context.Context) ([]Tier, error) {
97-
cm, err := m.configMapClient.Get(ctx, MappingConfigMap, metav1.GetOptions{})
94+
cm, err := m.configMapClient.Get(ctx, constant.TierMappingConfigMap, metav1.GetOptions{})
9895
if err != nil {
9996
return nil, err
10097
}
10198

10299
configData, exists := cm.Data["tiers"]
103100
if !exists {
104-
log.Printf("tiers key not found in ConfigMap %s", MappingConfigMap)
101+
log.Printf("tiers key not found in ConfigMap %s", constant.TierMappingConfigMap)
105102
return nil, fmt.Errorf("tier to group mapping configuration not found")
106103
}
107104

maas-api/internal/tier/mapper_test.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ package tier_test
33
import (
44
"testing"
55

6+
"github.com/opendatahub-io/maas-billing/maas-api/internal/constant"
67
"github.com/opendatahub-io/maas-billing/maas-api/internal/tier"
78
"github.com/opendatahub-io/maas-billing/maas-api/test/fixtures"
89
corev1 "k8s.io/api/core/v1"
@@ -162,7 +163,7 @@ func TestMapper_GetTierForGroups_SameLevels(t *testing.T) {
162163
// Test case where two tiers have the same level
163164
configMap := &corev1.ConfigMap{
164165
ObjectMeta: metav1.ObjectMeta{
165-
Name: tier.MappingConfigMap,
166+
Name: constant.TierMappingConfigMap,
166167
Namespace: testNamespace,
167168
},
168169
Data: map[string]string{

maas-api/test/fixtures/configmaps.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
package fixtures
22

33
import (
4-
"github.com/opendatahub-io/maas-billing/maas-api/internal/tier"
4+
"github.com/opendatahub-io/maas-billing/maas-api/internal/constant"
55
corev1 "k8s.io/api/core/v1"
66
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
77
)
@@ -37,7 +37,7 @@ const TierConfigYAML = `
3737
func CreateTierConfigMap(namespace string) *corev1.ConfigMap {
3838
return &corev1.ConfigMap{
3939
ObjectMeta: metav1.ObjectMeta{
40-
Name: tier.MappingConfigMap,
40+
Name: constant.TierMappingConfigMap,
4141
Namespace: namespace,
4242
},
4343
Data: map[string]string{

0 commit comments

Comments
 (0)