package storage

import (
	"fmt"
	"sort"

	"github.com/chrislusf/seaweedfs/weed/pb/master_pb"
)

type VolumeInfo struct {
	Id               VolumeId
	Size             uint64
	ReplicaPlacement *ReplicaPlacement
	Ttl              *TTL
	Collection       string
	Version          Version
	FileCount        int
	DeleteCount      int
	DeletedByteCount uint64
	ReadOnly         bool

	MemoryTotal uint64
	MemoryFree  uint64

	DiskWatermark uint64
	DiskTotal     uint64
	DiskFree      uint64

	DiskDevice      string
	DiskMountPoint  string
	Load1           float64
	Load5           float64
	Load15          float64
	ProcessCpuUsage float64
	ProcessRss      uint64
}

func NewVolumeInfo(m *master_pb.VolumeInformationMessage) (vi VolumeInfo, err error) {
	vi = VolumeInfo{
		Id:               VolumeId(m.Id),
		Size:             m.Size,
		Collection:       m.Collection,
		FileCount:        int(m.FileCount),
		DeleteCount:      int(m.DeleteCount),
		DeletedByteCount: m.DeletedByteCount,
		ReadOnly:         m.ReadOnly,
		Version:          Version(m.Version),

		MemoryTotal: m.MemoryTotal,
		MemoryFree:  m.MemoryFree,

		DiskWatermark:   m.DiskWatermark,
		DiskTotal:       m.DiskTotal,
		DiskFree:        m.DiskFree,
		DiskDevice:      m.DiskDevice,
		DiskMountPoint:  m.DiskMountPoint,
		Load1:           m.Load1,
		Load5:           m.Load5,
		Load15:          m.Load15,
		ProcessCpuUsage: m.ProcessCpuUsage,
		ProcessRss:      m.ProcessRss,
	}
	rp, e := NewReplicaPlacementFromByte(byte(m.ReplicaPlacement))
	if e != nil {
		return vi, e
	}
	vi.ReplicaPlacement = rp
	vi.Ttl = LoadTTLFromUint32(m.Ttl)
	return vi, nil
}

func (vi VolumeInfo) String() string {
	return fmt.Sprintf("Id:%d, Size:%d, ReplicaPlacement:%s, Collection:%s, Version:%v, FileCount:%d, DeleteCount:%d, DeletedByteCount:%d, ReadOnly:%v",
		vi.Id, vi.Size, vi.ReplicaPlacement, vi.Collection, vi.Version, vi.FileCount, vi.DeleteCount, vi.DeletedByteCount, vi.IsReadOnly())
}

// IsReadOnly tells the current vi is readonly or not, including the following conditions:
// 1. the data file is readonly;
// 2. the disk free space is under the watermark;
func (vi VolumeInfo) IsReadOnly() bool {
	return vi.ReadOnly || vi.DiskWatermark > 0 && vi.DiskFree <= vi.DiskWatermark
}

/*VolumesInfo sorting*/

type volumeInfos []*VolumeInfo

func (vis volumeInfos) Len() int {
	return len(vis)
}

func (vis volumeInfos) Less(i, j int) bool {
	return vis[i].Id < vis[j].Id
}

func (vis volumeInfos) Swap(i, j int) {
	vis[i], vis[j] = vis[j], vis[i]
}

func sortVolumeInfos(vis volumeInfos) {
	sort.Sort(vis)
}