namespace_cache.go

/*
 * Tencent is pleased to support the open source community by making TKEStack
 * available.
 *
 * Copyright (C) 2012-2019 Tencent. All Rights Reserved.
 *
 * 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
 *
 * https://opensource.org/licenses/Apache-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 OF ANY KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations under the License.
 */

package namespace

import (
	"sync"

	v1 "tkestack.io/tke/api/business/v1"
)

type cachedNamespace struct {
	// The cached state of the value
	state *v1.Namespace
}

type namespaceCache struct {
	mu sync.Mutex
	m  map[string]*cachedNamespace
}

func (s *namespaceCache) getOrCreate(name string, self *v1.Namespace) *cachedNamespace {
	s.mu.Lock()
	defer s.mu.Unlock()
	namespace, ok := s.m[name]
	if !ok {
		namespace = &cachedNamespace{}
		if self.Status.Phase == v1.NamespaceAvailable {
			namespace.state = self.DeepCopy()
			if self.Status.CachedSpecHard != nil {
				namespace.state.Spec.Hard = self.Status.CachedSpecHard
			} else { // For historic data that has no CachedSpecHard
				namespace.state.Spec.Hard = self.Spec.Hard
			}
		}
		s.m[name] = namespace
	}
	return namespace
}

func (s *namespaceCache) get(name string) (*cachedNamespace, bool) {
	s.mu.Lock()
	defer s.mu.Unlock()
	namespace, ok := s.m[name]
	return namespace, ok
}

func (s *namespaceCache) Exist(name string) bool {
	s.mu.Lock()
	defer s.mu.Unlock()
	_, ok := s.m[name]
	return ok
}

func (s *namespaceCache) set(name string, namespace *cachedNamespace) {
	s.mu.Lock()
	defer s.mu.Unlock()
	s.m[name] = namespace
}

func (s *namespaceCache) delete(name string) {
	s.mu.Lock()
	defer s.mu.Unlock()
	delete(s.m, name)
}