ServiceContainer

container.go

// Copyright 2015 The Nanoninja Authors. All rights reserved.

package container

// ServiceFunc registers the service function.
type ServiceFunc func(c *container) interface{}

type container struct {
    name     string
    services map[string]interface{}
}

// New allocates a new service container with the given name.
func New(name string) *container {
    return &container{name, make(map[string]interface{})}
}

// Name returns the name of the container.
func (c container) Name() string {
    return c.name
}

func (c *container) Set(key string, val ServiceFunc) {
    c.services[key] = val
}

func (c container) Get(key string) interface{} {
    if service, ok := c.services[key]; ok {
        return service.(ServiceFunc)(&c)
    }
    return nil
}

func (c container) Share(s ServiceFunc) ServiceFunc {
    var service interface{}
    return func(c *container) interface{} {
        if service == nil {
            service = s(c)
        }
        return service
    }
}