You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
544 B
Go
35 lines
544 B
Go
package namespace
|
|
|
|
type FileSizeQuota struct {
|
|
AllowanceKB int64
|
|
CurrentUsage int64
|
|
}
|
|
|
|
func (f *FileSizeQuota) Allows(size int64) bool {
|
|
return f.Remaining() >= size
|
|
}
|
|
|
|
func (f *FileSizeQuota) Remaining() int64 {
|
|
return f.AllowanceKB - f.CurrentUsage
|
|
}
|
|
|
|
func (f *FileSizeQuota) Add(size int64) error {
|
|
if !f.Allows(size) {
|
|
return ErrExceedQuota
|
|
}
|
|
|
|
f.CurrentUsage += size
|
|
|
|
return nil
|
|
}
|
|
|
|
func (f *FileSizeQuota) Remove(size int64) error {
|
|
if size > f.CurrentUsage {
|
|
return ErrQuotaInvalid
|
|
}
|
|
|
|
f.CurrentUsage -= size
|
|
|
|
return nil
|
|
}
|