subtype.go

// Package subtype contains a Service type that can be used to hold all resources of a certain subtype.
package subtype

import (
	"sync"

	"go.viam.com/rdk/resource"
)

// Service defines an service that holds and replaces resources.
type Service interface {
	Resource(name string) interface{}
	Replace(resources map[resource.Name]interface{}) error
}

type subtypeSvc struct {
	mu        sync.RWMutex
	resources map[string]interface{}
}

// New creates a new subtype service, which holds and replaces resources belonging to that subtype.
func New(r map[resource.Name]interface{}) (Service, error) {
	s := &subtypeSvc{}
	if err := s.Replace(r); err != nil {
		return nil, err
	}
	return s, nil
}

// Resource returns resource by name, if it exists.
func (s *subtypeSvc) Resource(name string) interface{} {
	s.mu.RLock()
	defer s.mu.RUnlock()
	if resource, ok := s.resources[name]; ok {
		return resource
	}
	return nil
}

// Replace replaces all resources with r.
func (s *subtypeSvc) Replace(r map[resource.Name]interface{}) error {
	s.mu.Lock()
	defer s.mu.Unlock()
	resources := make(map[string]interface{}, len(r))
	for n, v := range r {
		switch {
		case n.Name == "":
			resources[n.String()] = v
		default:
			resources[n.ShortName()] = v
		}
	}
	s.resources = resources
	return nil
}