validate canons without yatest_common

1 files changed, 46 insertions, 0 deletions

diff --git a/library/go/maxprocs/helpers.go b/library/go/maxprocs/helpers.go
new file mode 100644
index 0000000000..70263e6eb3
--- /dev/null
+++ b/library/go/maxprocs/helpers.go
@@ -0,0 +1,46 @@
+package maxprocs
+
+import (
+	"bytes"
+	"io/ioutil"
+	"math"
+	"os"
+	"strconv"
+)
+
+func getEnv(envName string) (string, bool) {
+	val, ok := os.LookupEnv(envName)
+	return val, ok && val != ""
+}
+
+func applyIntStringLimit(val string) int {
+	maxProc, err := strconv.Atoi(val)
+	if err == nil {
+		return Adjust(maxProc)
+	}
+
+	return Adjust(SafeProc)
+}
+
+func applyFloatStringLimit(val string) int {
+	maxProc, err := strconv.ParseFloat(val, 64)
+	if err != nil {
+		return Adjust(SafeProc)
+	}
+
+	return applyFloatLimit(maxProc)
+}
+
+func applyFloatLimit(val float64) int {
+	maxProc := int(math.Floor(val))
+	return Adjust(maxProc)
+}
+
+func readFileInt(filename string) (int, error) {
+	raw, err := ioutil.ReadFile(filename)
+	if err != nil {
+		return 0, err
+	}
+
+	return strconv.Atoi(string(bytes.TrimSpace(raw)))
+}