persistent

type Build struct {
	// Git version.
	GitVersion string

	// Git commit.
	GitCommit string

	// Golang version.
	GoVersion string

	// Rust version.
	RustVersion string

	// Build platform.
	Platform string
}

type CPU struct {
	// Number of logical cores in the system.
	LogicalCount uint32

	// Number of physical cores in the system.
	PhysicalCount uint32

	// Percent calculates the percentage of cpu used.
	Percent float64

	// Calculates the percentage of cpu used by process.
	ProcessPercent float64

	// Times contains the amounts of time the CPU has spent performing different kinds of work.
	Times CPUTimes
}

type CPUTimes struct {
	// CPU time of user.
	User float64

	// CPU time of system.
	System float64

	// CPU time of idle.
	Idle float64

	// CPU time of nice.
	Nice float64

	// CPU time of iowait.
	Iowait float64

	// CPU time of irq.
	Irq float64

	// CPU time of softirq.
	Softirq float64

	// CPU time of steal.
	Steal float64

	// CPU time of guest.
	Guest float64

	// CPU time of guest nice.
	GuestNice float64
}

type Disk struct {
	// Total amount of disk on the data path of dragonfly.
	Total uint64

	// Free amount of disk on the data path of dragonfly.
	Free uint64

	// Used amount of disk on the data path of dragonfly.
	Used uint64

	// Used percent of disk on the data path of dragonfly directory.
	UsedPercent float64

	// Total amount of indoes on the data path of dragonfly directory.
	InodesTotal uint64

	// Used amount of indoes on the data path of dragonfly directory.
	InodesUsed uint64

	// Free amount of indoes on the data path of dragonfly directory.
	InodesFree uint64

	// Used percent of indoes on the data path of dragonfly directory.
	InodesUsedPercent float64

	// Disk write bandwidth, unit is byte/s.
	WriteBandwidth uint64

	// Disk read bandwidth, unit is byte/s.
	ReadBandwidth uint64
}

type Host struct {
	// ID is host id.
	ID string

	// Type is host type.
	Type types.HostType

	// Name is the instance name.
	Name string

	// Hostname is host name.
	Hostname string

	// IP is host ip.
	IP string

	// Port is grpc service port.
	Port int32

	// DownloadPort is piece downloading port.
	DownloadPort int32

	// ProxyPort is the proxy server port.
	ProxyPort int32

	// DisableShared is whether the host is disabled for
	// shared with other peers.
	DisableShared bool

	// Host OS.
	OS string

	// Host platform.
	Platform string

	// Host platform family.
	PlatformFamily string

	// Host platform version.
	PlatformVersion string

	// Host kernel version.
	KernelVersion string

	// CPU Stat.
	CPU CPU

	// Memory Stat.
	Memory Memory

	// Network Stat.
	Network Network

	// Dist Stat.
	Disk Disk

	// Build information.
	Build Build

	// SchedulerClusterID is the scheduler cluster id matched by scopes.
	SchedulerClusterID uint64

	// AnnounceInterval is the interval between host announces to scheduler.
	AnnounceInterval time.Duration

	// CreatedAt is host create time.
	CreatedAt time.Time

	// UpdatedAt is host update time.
	UpdatedAt time.Time

	// Host log.
	Log *logger.SugaredLoggerOnWith
}

type HostManager interface {
	// Load returns host by a key.
	Load(context.Context, string) (*Host, bool)

	// Store sets host.
	Store(context.Context, *Host) error

	// Delete deletes host by a key.
	Delete(context.Context, string) error

	// LoadAll returns all hosts.
	LoadAll(context.Context) ([]*Host, error)

	// LoadRandom loads host randomly through the set of redis.
	LoadRandom(context.Context, int, set.SafeSet[string]) ([]*Host, error)

	// RunGC runs garbage collection.
	RunGC(context.Context) error
}

type Memory struct {
	// Total amount of RAM on this system.
	Total uint64

	// RAM available for programs to allocate.
	Available uint64

	// RAM used by programs.
	Used uint64

	// Percentage of RAM used by programs.
	UsedPercent float64

	// Calculates the percentage of memory used by process.
	ProcessUsedPercent float64

	// This is the kernel's notion of free memory.
	Free uint64
}

type MockHostManager struct {
	// contains filtered or unexported fields
}

type MockHostManagerMockRecorder struct {
	// contains filtered or unexported fields
}

type MockPeerManager struct {
	// contains filtered or unexported fields
}

type MockPeerManagerMockRecorder struct {
	// contains filtered or unexported fields
}

type MockResource struct {
	// contains filtered or unexported fields
}

type MockResourceMockRecorder struct {
	// contains filtered or unexported fields
}

type MockTaskManager struct {
	// contains filtered or unexported fields
}

type MockTaskManagerMockRecorder struct {
	// contains filtered or unexported fields
}

type Network struct {
	// Return count of tcp connections opened and status is ESTABLISHED.
	TCPConnectionCount uint32

	// Return count of upload tcp connections opened and status is ESTABLISHED.
	UploadTCPConnectionCount uint32

	// Location path(area|country|province|city|...).
	Location string

	// IDC where the peer host is located
	IDC string

	// RxBandwidth of the host, unit is byte/s.
	RxBandwidth uint64

	// MaxRxBandwidth of the host, unit is byte/s.
	MaxRxBandwidth uint64

	// TxBandwidth of the host, unit is byte/s.
	TxBandwidth uint64

	// MaxTxBandwidth of the host, unit is byte/s.
	MaxTxBandwidth uint64
}

type Peer struct {
	// ID is persistent peer id.
	ID string

	// Persistent is whether the peer is persistent.
	Persistent bool

	// ConcurrentPieceCount is the count of pieces that can be downloaded concurrently.
	ConcurrentPieceCount uint32

	// Pieces is finished pieces bitset.
	FinishedPieces *bitset.BitSet

	// Persistent peer state machine.
	FSM *fsm.FSM

	// Task is persistent task.
	Task *Task

	// Host is the peer host.
	Host *Host

	// BlockParents is bad parents ids.
	BlockParents []string

	// Cost is the cost of downloading.
	Cost time.Duration

	// CreatedAt is persistent peer create time.
	CreatedAt time.Time

	// UpdatedAt is persistent peer update time.
	UpdatedAt time.Time

	// Persistent peer log.
	Log *logger.SugaredLoggerOnWith
}

type PeerManager interface {
	// Load returns peer by a key.
	Load(context.Context, string) (*Peer, bool)

	// Store sets peer.
	Store(context.Context, *Peer) error

	// Delete deletes peer by a key.
	Delete(context.Context, string) error

	// LoadAll returns all peers.
	LoadAll(context.Context) ([]*Peer, error)

	// LoadAllByTaskID returns all peers by task id.
	LoadAllByTaskID(context.Context, string) ([]*Peer, error)

	// LoadAllIDsByTaskID returns all peer ids by task id.
	LoadAllIDsByTaskID(context.Context, string) ([]string, error)

	// LoadPersistentAllByTaskID returns all persistent peers by task id.
	LoadPersistentAllByTaskID(context.Context, string) ([]*Peer, error)

	// DeleteAllByTaskID deletes all peers by task id.
	DeleteAllByTaskID(context.Context, string) error

	// LoadAllByHostID returns all peers by host id.
	LoadAllByHostID(context.Context, string) ([]*Peer, error)

	// LoadAllIDsByHostID returns all peer ids by host id.
	LoadAllIDsByHostID(context.Context, string) ([]string, error)

	// DeleteAllByHostID deletes all peers by host id.
	DeleteAllByHostID(context.Context, string) error
}

type PeerOption func(peer *Peer)

type Resource interface {
	// Host manager interface.
	HostManager() HostManager

	// Peer manager interface.
	PeerManager() PeerManager

	// Task manager interface.
	TaskManager() TaskManager
}

type Task struct {
	// ID is task id.
	ID string

	// URL is download url.
	URL string

	// ObjectStorageRegion is object storage region.
	ObjectStorageRegion string

	// ObjectStorageEndpoint is object storage endpoint.
	ObjectStorageEndpoint string

	// Replica count of the persistent task. The persistent task will
	// not be deleted when dfdamon runs garbage collection. It only be deleted
	// when the task is deleted by the user.
	PersistentReplicaCount uint64

	// ContentLength is persistent task total content length.
	ContentLength uint64

	// TotalPieceCount is total piece count.
	TotalPieceCount uint32

	// Persistent task state machine.
	FSM *fsm.FSM

	// TTL is persistent task time to live.
	TTL time.Duration

	// CreatedAt is persistent task create time.
	CreatedAt time.Time

	// UpdatedAt is persistent task update time.
	UpdatedAt time.Time

	// Persistent task log.
	Log *logger.SugaredLoggerOnWith
}

type TaskManager interface {
	// Load returns persistent task by a key.
	Load(context.Context, string) (*Task, bool)

	// LoadCurrentReplicaCount returns current replica count of the persistent task.
	LoadCurrentReplicaCount(context.Context, string) (uint64, error)

	// LoadCurrentPersistentReplicaCount returns current persistent replica count of the persistent task.
	LoadCurrentPersistentReplicaCount(context.Context, string) (uint64, error)

	// Store sets persistent task.
	Store(context.Context, *Task) error

	// Delete deletes persistent task by a key.
	Delete(context.Context, string) error

	// LoadAll returns all persistent tasks.
	LoadAll(context.Context) ([]*Task, error)
}