[fix] Get cpuNum in container & image build

cmd/pixiu/pixiu.go

@@ -26,7 +26,7 @@ import (
 )
 
 import (
-	gxruntime "github.com/dubbogo/gost/runtime"
+	pxruntime "github.com/apache/dubbo-go-pixiu/pkg/common/runtime"
 	"github.com/spf13/cobra"
 )
 
@@ -141,14 +141,15 @@ func initApiConfig() (*model.Bootstrap, bool, error) {
 }
 
 func initLimitCpus() error {
-	limitCpuNumber, err := strconv.ParseInt(limitCpus, 10, 64)
+	limitCpuNumberFromEnv, err := strconv.ParseInt(limitCpus, 10, 64)
 	if err != nil {
 		return err
 	}
+	limitCpuNumber := int(limitCpuNumberFromEnv)
 	if limitCpuNumber <= 0 {
-		runtime.GOMAXPROCS(gxruntime.GetCPUNum())
-	} else {
-		runtime.GOMAXPROCS(int(limitCpuNumber))
+		limitCpuNumber = pxruntime.GetCPUNum()
 	}
+	runtime.GOMAXPROCS(limitCpuNumber)
+	logger.Infof("GOMAXPROCS set to %v", limitCpuNumber)
 	return nil
 }

pkg/common/runtime/runtime.go

@@ -0,0 +1,128 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You 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 runtime
+
+import (
+	"bufio"
+	"io/ioutil"
+	"os"
+	"runtime"
+	"strconv"
+	"strings"
+)
+
+const (
+	_cgroupPath    = "/proc/self/cgroup"
+	_dockerPath    = "/docker"
+	_kubepodsPath  = "/kubepods"
+	_cpuPeriodPath = "/sys/fs/cgroup/cpu/cpu.cfs_period_us"
+	_cpuQuotaPath  = "/sys/fs/cgroup/cpu/cpu.cfs_quota_us"
+)
+
+// GetCPUNum gets current os's cpu number
+func GetCPUNum() int {
+	if isContainer() {
+		cpus, _ := numCPU()
+		return cpus
+	}
+	return runtime.NumCPU()
+}
+
+// readLinesFromFile reads the lines from a file.
+func readLinesFromFile(filepath string) []string {
+	res := make([]string, 0)
+	f, err := os.Open(filepath)
+	if err != nil {
+		return res
+	}
+	defer f.Close()
+	buff := bufio.NewReader(f)
+	for {
+		line, _, err := buff.ReadLine()
+		if err != nil {
+			return res
+		}
+		res = append(res, string(line))
+	}
+}
+
+// isContainer returns true if the process is running in a container.
+func isContainer() bool {
+	lines := readLinesFromFile(_cgroupPath)
+	for _, line := range lines {
+		if strings.Contains(line, _dockerPath) ||
+			strings.Contains(line, _kubepodsPath) {
+			return true
+		}
+	}
+	return false
+}
+
+// copied from https://github.com/containerd/cgroups/blob/318312a373405e5e91134d8063d04d59768a1bff/utils.go#L243
+func readUint(path string) (uint64, error) {
+	v, err := ioutil.ReadFile(path)
+	if err != nil {
+		return 0, err
+	}
+	return parseUint(strings.TrimSpace(string(v)), 10, 64)
+}
+
+// copied from https://github.com/containerd/cgroups/blob/318312a373405e5e91134d8063d04d59768a1bff/utils.go#L251
+func parseUint(s string, base, bitSize int) (uint64, error) {
+	v, err := strconv.ParseUint(s, base, bitSize)
+	if err != nil {
+		intValue, intErr := strconv.ParseInt(s, base, bitSize)
+		// 1. Handle negative values greater than MinInt64 (and)
+		// 2. Handle negative values lesser than MinInt64
+		if intErr == nil && intValue < 0 {
+			return 0, nil
+		} else if intErr != nil &&
+			intErr.(*strconv.NumError).Err == strconv.ErrRange &&
+			intValue < 0 {
+			return 0, nil
+		}
+		return 0, err
+	}
+	return v, nil
+}
+
+// numCPU returns the CPU quota
+func numCPU() (num int, err error) {
+	if !isContainer() {
+		return runtime.NumCPU(), nil
+	}
+
+	// If the container is running in a cgroup, we can use the cgroup cpu
+	// quota to limit the number of CPUs.
+	period, err := readUint(_cpuPeriodPath)
+	if err != nil {
+		return runtime.NumCPU(), err
+	}
+	quota, err := readUint(_cpuQuotaPath)
+	if err != nil {
+		return runtime.NumCPU(), err
+	}
+
+	// The number of CPUs is the quota divided by the period.
+	// See https://www.kernel.org/doc/Documentation/scheduler/sched-bwc.txt
+	if quota <= 0 || period <= 0 {
+		return runtime.NumCPU(), err
+	}
+
+	return int(quota) / int(period), nil
+}