swirl/dao/bolt/registry.go

package bolt

import (
	"encoding/json"
	"time"

	"github.com/boltdb/bolt"
	"github.com/cuigh/auxo/errors"
	"github.com/cuigh/swirl/model"
)

func (d *Dao) RegistryCreate(registry *model.Registry) (err error) {
	return d.update("registry", registry.ID, registry)
}

func (d *Dao) RegistryUpdate(registry *model.Registry) (err error) {
	return d.batch("registry", func(b *bolt.Bucket) error {
		data := b.Get([]byte(registry.ID))
		if data == nil {
			return errors.New("registry not found: " + registry.ID)
		}

		r := &model.Registry{}
		err = json.Unmarshal(data, r)
		if err != nil {
			return err
		}

		r.Name = registry.Name
		r.URL = registry.URL
		r.Username = registry.Username
		if registry.Password != "" {
			r.Password = registry.Password
		}
		r.UpdatedAt = time.Now()
		data, err = json.Marshal(r)
		if err != nil {
			return err
		}

		return b.Put([]byte(registry.ID), data)
	})
}

func (d *Dao) RegistryList() (registries []*model.Registry, err error) {
	err = d.each("registry", func(v Value) error {
		r := &model.Registry{}
		err = v.Unmarshal(r)
		if err != nil {
			return err
		}
		registries = append(registries, r)
		return nil
	})
	return
}

func (d *Dao) RegistryGet(id string) (registry *model.Registry, err error) {
	var v Value
	v, err = d.get("registry", id)
	if err == nil {
		if v != nil {
			registry = &model.Registry{}
			err = v.Unmarshal(registry)
		}
	}
	return
}

func (d *Dao) RegistryDelete(id string) (err error) {
	return d.delete("registry", id)
}