in_memory_container_repository.go

package container_repository

import (
	"sync"

	"code.cloudfoundry.org/garden"
	"code.cloudfoundry.org/garden-linux/linux_backend"
	"code.cloudfoundry.org/lager"
)

type InMemoryContainerRepository struct {
	store map[string]linux_backend.Container
	mutex *sync.RWMutex
}

func New() *InMemoryContainerRepository {
	return &InMemoryContainerRepository{
		store: map[string]linux_backend.Container{},
		mutex: &sync.RWMutex{},
	}
}

func (cr *InMemoryContainerRepository) All() []linux_backend.Container {
	return cr.Query(func(c linux_backend.Container) bool {
		return true
	}, nil)
}

func (cr *InMemoryContainerRepository) Add(container linux_backend.Container) {
	cr.mutex.Lock()
	defer cr.mutex.Unlock()

	cr.store[container.Handle()] = container
}

func (cr *InMemoryContainerRepository) FindByHandle(handle string) (linux_backend.Container, error) {
	cr.mutex.RLock()
	defer cr.mutex.RUnlock()

	container, ok := cr.store[handle]
	if !ok {
		return nil, garden.ContainerNotFoundError{handle}
	}

	return container, nil
}

func (cr *InMemoryContainerRepository) Delete(container linux_backend.Container) {
	cr.mutex.Lock()
	defer cr.mutex.Unlock()

	delete(cr.store, container.Handle())
}

func (cr *InMemoryContainerRepository) Query(filter func(linux_backend.Container) bool, logger lager.Logger) []linux_backend.Container {
	cr.mutex.RLock()
	defer cr.mutex.RUnlock()

	var matches []linux_backend.Container
	for _, c := range cr.store {
		if filter(c) {
			if logger != nil {
				logger.Debug("matched", lager.Data{"handle": c.Handle()})
			}
			matches = append(matches, c)
		} else {
			if logger != nil {
				logger.Debug("did-not-match", lager.Data{"handle": c.Handle()})
			}
		}
	}

	return matches
}