提交 #1

developtools/integration_verification/tools/fotff/utils/exec.go

@@ -0,0 +1,90 @@
+/*
+ * Copyright (c) 2022 Huawei Device Co., Ltd.
+ * 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 utils
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"github.com/sirupsen/logrus"
+	"io"
+	"os"
+	"os/exec"
+	"time"
+)
+
+func ExecContext(ctx context.Context, name string, args ...string) error {
+	ctx, fn := context.WithTimeout(ctx, 6*time.Hour)
+	defer fn()
+	if err := execContext(ctx, name, args...); err != nil {
+		if errors.Is(err, context.Canceled) {
+			return err
+		}
+		logrus.Errorf("exec failed: %v, try again...", err)
+		return execContext(ctx, name, args...)
+	}
+	return nil
+}
+
+func execContext(ctx context.Context, name string, args ...string) error {
+	cmdStr := append([]string{name}, args...)
+	logrus.Infof("cmd: %s", cmdStr)
+	cmd := exec.CommandContext(ctx, name, args...)
+	stdout, err := cmd.StdoutPipe()
+	if err != nil {
+		return err
+	}
+	stderr, err := cmd.StderrPipe()
+	if err != nil {
+		return err
+	}
+	if err := cmd.Start(); err != nil {
+		return err
+	}
+	fmt.Printf("[%s] exec %s :\n", time.Now(), cmdStr)
+	go io.Copy(os.Stdout, stdout)
+	go io.Copy(os.Stderr, stderr)
+	return cmd.Wait()
+}
+
+func ExecCombinedOutputContext(ctx context.Context, name string, args ...string) ([]byte, error) {
+	ctx, fn := context.WithTimeout(ctx, 6*time.Hour)
+	defer fn()
+	out, err := execCombinedOutputContext(ctx, name, args...)
+	if err != nil {
+		if errors.Is(err, context.Canceled) {
+			return out, err
+		}
+		logrus.Errorf("exec failed: %v, try again...", err)
+		return execCombinedOutputContext(ctx, name, args...)
+	}
+	return out, nil
+}
+
+func execCombinedOutputContext(ctx context.Context, name string, args ...string) ([]byte, error) {
+	cmdStr := append([]string{name}, args...)
+	logrus.Infof("cmd: %s", cmdStr)
+	out, err := exec.CommandContext(ctx, name, args...).CombinedOutput()
+	fmt.Printf("[%s] exec %s :\n", time.Now(), cmdStr)
+	return out, err
+}
+
+func SleepContext(duration time.Duration, ctx context.Context) {
+	select {
+	case <-time.NewTimer(duration).C:
+	case <-ctx.Done():
+	}
+}