store

Documentation

Overview

store implements a model store which allows downloading models from a remote server

Index

• type Store
  • func NewStore(path, ext, modelUrl string) (*Store, error)
  • func (s *Store) ById(id string) *schema.Model
  • func (s *Store) ByPath(path string) *schema.Model
  • func (s *Store) Delete(id string) error
  • func (s *Store) Download(ctx context.Context, path string, fn func(curBytes, totalBytes uint64)) (*schema.Model, error)
  • func (s *Store) List() []*schema.Model
  • func (s *Store) MarshalJSON() ([]byte, error)
  • func (s *Store) Rescan() error
  • func (s *Store) String() string

Types

type Store

type Store struct {
	sync.RWMutex
	// contains filtered or unexported fields
}

func NewStore

func NewStore(path, ext, modelUrl string) (*Store, error)

Create a new model store

func (*Store) ById

func (s *Store) ById(id string) *schema.Model

Return a model by its Id

func (*Store) ByPath

func (s *Store) ByPath(path string) *schema.Model

Return a model by path

func (*Store) Delete

func (s *Store) Delete(id string) error

Delete a model by its Id

func (*Store) Download

func (s *Store) Download(ctx context.Context, path string, fn func(curBytes, totalBytes uint64)) (*schema.Model, error)

Download a model to the models directory. If the model already exists, it will be returned without downloading. The destination directory is relative to the models directory.

A function can be provided to track the progress of the download. If no Content-Length is provided by the server, the total bytes will be unknown and is set to zero.

func (*Store) List

func (s *Store) List() []*schema.Model

Return the models

func (*Store) MarshalJSON

func (s *Store) MarshalJSON() ([]byte, error)

func (*Store) Rescan

func (s *Store) Rescan() error

Rescan models directory

func (*Store) String

func (s *Store) String() string