mirror of
https://github.com/NVIDIA/nvidia-container-toolkit
synced 2024-11-28 06:52:34 +00:00
583793b7ae
This change adds a Requirements abstraction that can be used to check an images' NVIDIA_REQUIRE_* envvars against the host properties such as CUDA version or architecture. Signed-off-by: Evan Lezar <elezar@nvidia.com>
92 lines
2.0 KiB
Go
92 lines
2.0 KiB
Go
/**
|
|
# Copyright (c) 2022, NVIDIA CORPORATION. All rights reserved.
|
|
#
|
|
# 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 constraints
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
// or represents an OR operation on a set of constraints
|
|
type or []Constraint
|
|
|
|
// and represents an AND (ALL) operation on a set of contraints
|
|
type and []Constraint
|
|
|
|
// AND constructs a new constraint that is the logical AND of the supplied constraints
|
|
func AND(constraints []Constraint) Constraint {
|
|
if len(constraints) == 0 {
|
|
return &always{}
|
|
}
|
|
if len(constraints) == 1 {
|
|
return constraints[0]
|
|
}
|
|
return and(constraints)
|
|
}
|
|
|
|
// OR constructs a new constrant that is the logical OR of the supplied constraints
|
|
func OR(constraints []Constraint) Constraint {
|
|
if len(constraints) == 0 {
|
|
return nil
|
|
}
|
|
if len(constraints) == 1 {
|
|
return constraints[0]
|
|
}
|
|
|
|
return or(constraints)
|
|
}
|
|
|
|
func (operands or) Assert() error {
|
|
for _, o := range operands {
|
|
// We stop on the first nil
|
|
if err := o.Assert(); err == nil {
|
|
return nil
|
|
}
|
|
}
|
|
return fmt.Errorf("%v not met", operands)
|
|
}
|
|
|
|
func (operands or) String() string {
|
|
var terms []string
|
|
|
|
for _, o := range operands {
|
|
terms = append(terms, o.String())
|
|
}
|
|
|
|
return strings.Join(terms, "||")
|
|
}
|
|
|
|
func (operands and) Assert() error {
|
|
for _, o := range operands {
|
|
// We stop on the first Assert
|
|
if err := o.Assert(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (operands and) String() string {
|
|
var terms []string
|
|
|
|
for _, o := range operands {
|
|
terms = append(terms, o.String())
|
|
}
|
|
|
|
return strings.Join(terms, "&&")
|
|
}
|