实现单个服务的带宽限制（商业版）

pkg/serverconfigs/shared/size_capacity.go

@@ -9,6 +9,8 @@ const (
 	SizeCapacityUnitKB   SizeCapacityUnit = "kb"
 	SizeCapacityUnitMB   SizeCapacityUnit = "mb"
 	SizeCapacityUnitGB   SizeCapacityUnit = "gb"
+	SizeCapacityUnitTB   SizeCapacityUnit = "tb"
+	SizeCapacityUnitPB   SizeCapacityUnit = "pb"
 )
 
 type SizeCapacity struct {
@@ -17,6 +19,9 @@ type SizeCapacity struct {
 }
 
 func (this *SizeCapacity) Bytes() int64 {
+	if this.Count < 0 {
+		return -1
+	}
 	switch this.Unit {
 	case SizeCapacityUnitByte:
 		return this.Count
@@ -26,6 +31,10 @@ func (this *SizeCapacity) Bytes() int64 {
 		return this.Count * 1024 * 1024
 	case SizeCapacityUnitGB:
 		return this.Count * 1024 * 1024 * 1024
+	case SizeCapacityUnitTB:
+		return this.Count * 1024 * 1024 * 1024 * 1024
+	case SizeCapacityUnitPB:
+		return this.Count * 1024 * 1024 * 1024 * 1024 * 1024
 	default:
 		return this.Count
 	}