| /* |
| Copyright 2015 The Kubernetes Authors. |
| |
| Licensed under the Apache License, Version 2.0 (the "License"); |
| you may not use this file except in compliance with the License. |
| You may obtain a copy of the License at |
| |
| http://www.apache.org/licenses/LICENSE-2.0 |
| |
| Unless required by applicable law or agreed to in writing, software |
| distributed under the License is distributed on an "AS IS" BASIS, |
| WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| See the License for the specific language governing permissions and |
| limitations under the License. |
| */ |
| |
| package bandwidth |
| |
| import ( |
| "fmt" |
| |
| "k8s.io/apimachinery/pkg/api/resource" |
| ) |
| |
| var minRsrc = resource.MustParse("1k") |
| var maxRsrc = resource.MustParse("1P") |
| |
| func validateBandwidthIsReasonable(rsrc *resource.Quantity) error { |
| if rsrc.Value() < minRsrc.Value() { |
| return fmt.Errorf("resource is unreasonably small (< 1kbit)") |
| } |
| if rsrc.Value() > maxRsrc.Value() { |
| return fmt.Errorf("resoruce is unreasonably large (> 1Pbit)") |
| } |
| return nil |
| } |
| |
| func ExtractPodBandwidthResources(podAnnotations map[string]string) (ingress, egress *resource.Quantity, err error) { |
| if podAnnotations == nil { |
| return nil, nil, nil |
| } |
| str, found := podAnnotations["kubernetes.io/ingress-bandwidth"] |
| if found { |
| ingressValue, err := resource.ParseQuantity(str) |
| if err != nil { |
| return nil, nil, err |
| } |
| ingress = &ingressValue |
| if err := validateBandwidthIsReasonable(ingress); err != nil { |
| return nil, nil, err |
| } |
| } |
| str, found = podAnnotations["kubernetes.io/egress-bandwidth"] |
| if found { |
| egressValue, err := resource.ParseQuantity(str) |
| if err != nil { |
| return nil, nil, err |
| } |
| egress = &egressValue |
| if err := validateBandwidthIsReasonable(egress); err != nil { |
| return nil, nil, err |
| } |
| } |
| return ingress, egress, nil |
| } |