1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
// vespa-deploy command
// Author: arnej
package deploy
import (
"bytes"
"fmt"
"io"
"os"
"os/exec"
"strings"
"github.com/vespa-engine/vespa/client/go/vespa"
)
func curlPut(url string, cfgSrc string) (string, error) {
args := append(curlPutArgs(), url)
return runCurl(args, new(strings.Reader), cfgSrc)
}
func curlPost(url string, input io.Reader, cfgSrc string) (string, error) {
args := append(curlPostArgs(), url)
return runCurl(args, input, cfgSrc)
}
func curlPostZip(url string, input io.Reader, cfgSrc string) (string, error) {
args := append(curlPostZipArgs(), url)
return runCurl(args, input, cfgSrc)
}
func curlGet(url string, output io.Writer) error {
args := append(curlGetArgs(), url)
cmd := exec.Command(curlWrapper(), args...)
cmd.Stdout = output
cmd.Stderr = os.Stderr
// fmt.Printf("running command: %v\n", cmd)
err := cmd.Run()
return err
}
func urlWithoutQuery(url string) string {
before, _, _ := strings.Cut(url, "?")
return before
}
func getOutputFromCmd(program string, args ...string) (string, error) {
cmd := exec.Command(program, args...)
var out bytes.Buffer
cmd.Stdout = &out
cmd.Stderr = os.Stderr
err := cmd.Run()
return out.String(), err
}
func runCurl(args []string, input io.Reader, cfgSrc string) (string, error) {
cmd := exec.Command(curlWrapper(), args...)
cmd.Stdin = input
var out bytes.Buffer
cmd.Stdout = &out
cmd.Stderr = os.Stderr
// fmt.Printf("running command: %v\n", cmd)
err := cmd.Run()
// fmt.Printf("output: %s\n", out.String())
if err != nil {
if cmd.ProcessState.ExitCode() == 7 {
return "", fmt.Errorf("HTTP request failed. Could not connect to %s", cfgSrc)
}
return "", fmt.Errorf("HTTP request failed with curl %s", err.Error())
}
return out.String(), err
}
func curlWrapper() string {
return vespa.FindHome() + "/libexec/vespa/vespa-curl-wrapper"
}
func commonCurlArgs() []string {
return []string{
"-A", "vespa-deploy",
"--silent",
"--show-error",
"--connect-timeout", "30",
"--max-time", "1200",
}
}
func curlPutArgs() []string {
return append(commonCurlArgs(),
"--write-out", "%{http_code}",
"--request", "PUT")
}
func curlGetArgs() []string {
return append(commonCurlArgs(),
"--request", "GET")
}
func curlPostArgs() []string {
return append(commonCurlArgs(),
"--write-out", "%{http_code}",
"--request", "POST",
"--header", "Content-Type: application/x-gzip",
"--data-binary", "@-")
}
func curlPostZipArgs() []string {
return append(commonCurlArgs(),
"--write-out", "%{http_code}",
"--request", "POST",
"--header", "Content-Type: application/zip",
"--data-binary", "@-")
}
|
