Skip to content

Commit 546d03a

Browse files
committed
update registrar client to expect ZosVersion instead of encoded string
1 parent 6d689b3 commit 546d03a

File tree

1 file changed

+5
-30
lines changed

1 file changed

+5
-30
lines changed

node-registrar/client/zos_version.go

Lines changed: 5 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -2,11 +2,9 @@ package client
22

33
import (
44
"bytes"
5-
"encoding/base64"
65
"encoding/json"
76
"net/http"
87
"net/url"
9-
"strings"
108
"time"
119

1210
"github.com/pkg/errors"
@@ -40,20 +38,7 @@ func (c *RegistrarClient) getZosVersion() (version ZosVersion, err error) {
4038

4139
defer resp.Body.Close()
4240

43-
var versionString string
44-
err = json.NewDecoder(resp.Body).Decode(&versionString)
45-
if err != nil {
46-
return version, err
47-
}
48-
49-
versionBytes, err := base64.StdEncoding.DecodeString(versionString)
50-
if err != nil {
51-
return version, err
52-
}
53-
54-
correctedJSON := strings.ReplaceAll(string(versionBytes), "'", "\"")
55-
56-
err = json.NewDecoder(strings.NewReader(correctedJSON)).Decode(&version)
41+
err = json.NewDecoder(resp.Body).Decode(&version)
5742
if err != nil {
5843
return version, err
5944
}
@@ -77,23 +62,13 @@ func (c *RegistrarClient) setZosVersion(v string, safeToUpgrade bool) (err error
7762
SafeToUpgrade: safeToUpgrade,
7863
}
7964

80-
jsonData, err := json.Marshal(version)
81-
if err != nil {
82-
return errors.Wrap(err, "failed to marshal zos version")
83-
}
84-
85-
encodedVersion := struct {
86-
Version string `json:"version"`
87-
}{
88-
Version: base64.StdEncoding.EncodeToString(jsonData),
89-
}
90-
91-
jsonData, err = json.Marshal(encodedVersion)
65+
var body bytes.Buffer
66+
err = json.NewEncoder(&body).Encode(version)
9267
if err != nil {
93-
return errors.Wrap(err, "failed to marshal zos version in hex format")
68+
return errors.Wrap(err, "failed to encode request body")
9469
}
9570

96-
req, err := http.NewRequest("PUT", url, bytes.NewReader(jsonData))
71+
req, err := http.NewRequest("PUT", url, &body)
9772
if err != nil {
9873
return errors.Wrap(err, "failed to construct http request to the registrar")
9974
}

0 commit comments

Comments
 (0)