JCS-pub/common/pkgs/storage/local/agent.go

package local

import (
	stgmod "gitlink.org.cn/cloudream/storage/common/models"
	"gitlink.org.cn/cloudream/storage/common/pkgs/storage/types"
)

type agent struct {
	Detail      stgmod.StorageDetail
	ShardStore  *ShardStore
	SharedStore *SharedStore
}

func (s *agent) Start(ch *types.StorageEventChan) {
	if s.ShardStore != nil {
		s.ShardStore.Start(ch)
	}

	if s.SharedStore != nil {
		s.SharedStore.Start(ch)
	}
}

func (s *agent) Stop() {
	if s.ShardStore != nil {
		s.ShardStore.Stop()
	}

	if s.SharedStore != nil {
		s.SharedStore.Stop()
	}
}

func (s *agent) Info() stgmod.StorageDetail {
	return s.Detail
}

func (a *agent) GetShardStore() (types.ShardStore, error) {
	if a.ShardStore == nil {
		return nil, types.ErrUnsupported
	}

	return a.ShardStore, nil
}

func (a *agent) GetSharedStore() (types.SharedStore, error) {
	if a.SharedStore == nil {
		return nil, types.ErrUnsupported
	}

	return a.SharedStore, nil
}