91 lines
2.1 KiB
Go
91 lines
2.1 KiB
Go
/*
|
|
Copyright 2014 Google Inc. 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
|
|
|
|
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 registrytest
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/api"
|
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/labels"
|
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/watch"
|
|
)
|
|
|
|
type PodRegistry struct {
|
|
Err error
|
|
Pod *api.Pod
|
|
Pods []api.Pod
|
|
sync.Mutex
|
|
|
|
mux *watch.Mux
|
|
}
|
|
|
|
func NewPodRegistry(pods []api.Pod) *PodRegistry {
|
|
return &PodRegistry{
|
|
Pods: pods,
|
|
mux: watch.NewMux(0),
|
|
}
|
|
}
|
|
|
|
func (r *PodRegistry) ListPods(selector labels.Selector) ([]api.Pod, error) {
|
|
r.Lock()
|
|
defer r.Unlock()
|
|
if r.Err != nil {
|
|
return r.Pods, r.Err
|
|
}
|
|
var filtered []api.Pod
|
|
for _, pod := range r.Pods {
|
|
if selector.Matches(labels.Set(pod.Labels)) {
|
|
filtered = append(filtered, pod)
|
|
}
|
|
}
|
|
return filtered, nil
|
|
}
|
|
|
|
func (r *PodRegistry) WatchPods(resourceVersion uint64, filter func(*api.Pod) bool) (watch.Interface, error) {
|
|
// TODO: wire filter down into the mux; it needs access to current and previous state :(
|
|
return r.mux.Watch(), nil
|
|
}
|
|
|
|
func (r *PodRegistry) GetPod(podId string) (*api.Pod, error) {
|
|
r.Lock()
|
|
defer r.Unlock()
|
|
return r.Pod, r.Err
|
|
}
|
|
|
|
func (r *PodRegistry) CreatePod(pod api.Pod) error {
|
|
r.Lock()
|
|
defer r.Unlock()
|
|
r.Pod = &pod
|
|
r.mux.Action(watch.Added, &pod)
|
|
return r.Err
|
|
}
|
|
|
|
func (r *PodRegistry) UpdatePod(pod api.Pod) error {
|
|
r.Lock()
|
|
defer r.Unlock()
|
|
r.Pod = &pod
|
|
r.mux.Action(watch.Modified, &pod)
|
|
return r.Err
|
|
}
|
|
|
|
func (r *PodRegistry) DeletePod(podId string) error {
|
|
r.Lock()
|
|
defer r.Unlock()
|
|
r.mux.Action(watch.Deleted, r.Pod)
|
|
return r.Err
|
|
}
|