2016-12-25 20:34:02 +00:00
|
|
|
package bolt
|
|
|
|
|
|
|
|
import (
|
|
|
|
"github.com/portainer/portainer"
|
|
|
|
"github.com/portainer/portainer/bolt/internal"
|
|
|
|
|
|
|
|
"github.com/boltdb/bolt"
|
|
|
|
)
|
|
|
|
|
2017-06-20 11:00:32 +00:00
|
|
|
// EndpointService represents a service for managing endpoints.
|
2016-12-25 20:34:02 +00:00
|
|
|
type EndpointService struct {
|
|
|
|
store *Store
|
|
|
|
}
|
|
|
|
|
|
|
|
// Endpoint returns an endpoint by ID.
|
|
|
|
func (service *EndpointService) Endpoint(ID portainer.EndpointID) (*portainer.Endpoint, error) {
|
|
|
|
var data []byte
|
|
|
|
err := service.store.db.View(func(tx *bolt.Tx) error {
|
|
|
|
bucket := tx.Bucket([]byte(endpointBucketName))
|
|
|
|
value := bucket.Get(internal.Itob(int(ID)))
|
|
|
|
if value == nil {
|
|
|
|
return portainer.ErrEndpointNotFound
|
|
|
|
}
|
|
|
|
|
|
|
|
data = make([]byte, len(value))
|
|
|
|
copy(data, value)
|
|
|
|
return nil
|
|
|
|
})
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
var endpoint portainer.Endpoint
|
2018-06-17 16:57:22 +00:00
|
|
|
err = internal.UnmarshalObject(data, &endpoint)
|
2016-12-25 20:34:02 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return &endpoint, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// Endpoints return an array containing all the endpoints.
|
|
|
|
func (service *EndpointService) Endpoints() ([]portainer.Endpoint, error) {
|
2017-01-23 03:29:49 +00:00
|
|
|
var endpoints = make([]portainer.Endpoint, 0)
|
2016-12-25 20:34:02 +00:00
|
|
|
err := service.store.db.View(func(tx *bolt.Tx) error {
|
|
|
|
bucket := tx.Bucket([]byte(endpointBucketName))
|
|
|
|
|
|
|
|
cursor := bucket.Cursor()
|
|
|
|
for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
|
|
|
|
var endpoint portainer.Endpoint
|
2018-06-17 16:57:22 +00:00
|
|
|
err := internal.UnmarshalObject(v, &endpoint)
|
2016-12-25 20:34:02 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
endpoints = append(endpoints, endpoint)
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
})
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return endpoints, nil
|
|
|
|
}
|
|
|
|
|
2017-02-06 05:29:34 +00:00
|
|
|
// Synchronize creates, updates and deletes endpoints inside a single transaction.
|
|
|
|
func (service *EndpointService) Synchronize(toCreate, toUpdate, toDelete []*portainer.Endpoint) error {
|
|
|
|
return service.store.db.Update(func(tx *bolt.Tx) error {
|
|
|
|
bucket := tx.Bucket([]byte(endpointBucketName))
|
|
|
|
|
|
|
|
for _, endpoint := range toCreate {
|
2017-02-07 03:26:12 +00:00
|
|
|
err := storeNewEndpoint(endpoint, bucket)
|
2017-02-06 05:29:34 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, endpoint := range toUpdate {
|
2017-02-07 03:26:12 +00:00
|
|
|
err := marshalAndStoreEndpoint(endpoint, bucket)
|
2017-02-06 05:29:34 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, endpoint := range toDelete {
|
|
|
|
err := bucket.Delete(internal.Itob(int(endpoint.ID)))
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2016-12-25 20:34:02 +00:00
|
|
|
// CreateEndpoint assign an ID to a new endpoint and saves it.
|
|
|
|
func (service *EndpointService) CreateEndpoint(endpoint *portainer.Endpoint) error {
|
|
|
|
return service.store.db.Update(func(tx *bolt.Tx) error {
|
|
|
|
bucket := tx.Bucket([]byte(endpointBucketName))
|
2017-02-07 03:26:12 +00:00
|
|
|
err := storeNewEndpoint(endpoint, bucket)
|
2016-12-25 20:34:02 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
// UpdateEndpoint updates an endpoint.
|
|
|
|
func (service *EndpointService) UpdateEndpoint(ID portainer.EndpointID, endpoint *portainer.Endpoint) error {
|
2018-06-17 16:57:22 +00:00
|
|
|
data, err := internal.MarshalObject(endpoint)
|
2016-12-25 20:34:02 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
return service.store.db.Update(func(tx *bolt.Tx) error {
|
|
|
|
bucket := tx.Bucket([]byte(endpointBucketName))
|
|
|
|
err = bucket.Put(internal.Itob(int(ID)), data)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
// DeleteEndpoint deletes an endpoint.
|
|
|
|
func (service *EndpointService) DeleteEndpoint(ID portainer.EndpointID) error {
|
|
|
|
return service.store.db.Update(func(tx *bolt.Tx) error {
|
|
|
|
bucket := tx.Bucket([]byte(endpointBucketName))
|
|
|
|
err := bucket.Delete(internal.Itob(int(ID)))
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2017-02-07 03:26:12 +00:00
|
|
|
func marshalAndStoreEndpoint(endpoint *portainer.Endpoint, bucket *bolt.Bucket) error {
|
2018-06-17 16:57:22 +00:00
|
|
|
data, err := internal.MarshalObject(endpoint)
|
2017-02-07 03:26:12 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
err = bucket.Put(internal.Itob(int(endpoint.ID)), data)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func storeNewEndpoint(endpoint *portainer.Endpoint, bucket *bolt.Bucket) error {
|
|
|
|
id, _ := bucket.NextSequence()
|
|
|
|
endpoint.ID = portainer.EndpointID(id)
|
|
|
|
return marshalAndStoreEndpoint(endpoint, bucket)
|
|
|
|
}
|