1
0
Fork 0

Merge pull request #683 from rqlite/fix_auth_status

Restore request should re-read file every attempt
master
Philip O'Toole 4 years ago committed by GitHub
commit 399e5f346f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -4,7 +4,6 @@ import (
"bytes" "bytes"
"crypto/tls" "crypto/tls"
"fmt" "fmt"
"io"
"io/ioutil" "io/ioutil"
"net/http" "net/http"
"net/url" "net/url"
@ -86,9 +85,9 @@ func dump(ctx *cli.Context, filename string, argv *argT) error {
return nil return nil
} }
func makeRestoreRequest(restoreFile io.Reader) func(string) (*http.Request, error) { func makeRestoreRequest(b []byte) func(string) (*http.Request, error) {
return func(urlStr string) (*http.Request, error) { return func(urlStr string) (*http.Request, error) {
req, err := http.NewRequest("POST", urlStr, restoreFile) req, err := http.NewRequest("POST", urlStr, bytes.NewReader(b))
req.Header["Content-type"] = []string{"text/plain"} req.Header["Content-type"] = []string{"text/plain"}
if err != nil { if err != nil {
return nil, err return nil, err
@ -158,8 +157,7 @@ func restore(ctx *cli.Context, filename string, argv *argT) error {
Path: fmt.Sprintf("%sdb/load", argv.Prefix), Path: fmt.Sprintf("%sdb/load", argv.Prefix),
RawQuery: queryStr.Encode(), RawQuery: queryStr.Encode(),
} }
restoreFileReader := bytes.NewReader(restoreFile) response, err := sendRequest(ctx, makeRestoreRequest(restoreFile), restoreURL.String(), argv)
response, err := sendRequest(ctx, makeRestoreRequest(restoreFileReader), restoreURL.String(), argv)
if err != nil { if err != nil {
return err return err
} }

Loading…
Cancel
Save