// Copyright 2022 PingCAP, Inc. // // 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 util import ( "encoding/json" "io/ioutil" "net/http" "github.com/pingcap/errors" ) // SliceToMap converts slice to map // nolint:unused func SliceToMap(slice []string) map[string]interface{} { sMap := make(map[string]interface{}) for _, str := range slice { sMap[str] = struct{}{} } return sMap } // StringsToInterfaces converts string slice to interface slice func StringsToInterfaces(strs []string) []interface{} { is := make([]interface{}, 0, len(strs)) for _, str := range strs { is = append(is, str) } return is } // GetJSON fetches a page and parses it as JSON. The parsed result will be // stored into the `v`. The variable `v` must be a pointer to a type that can be // unmarshalled from JSON. // // Example: // // client := &http.Client{} // var resp struct { IP string } // if err := util.GetJSON(client, "http://api.ipify.org/?format=json", &resp); err != nil { // return errors.Trace(err) // } // fmt.Println(resp.IP) func GetJSON(client *http.Client, url string, v interface{}) error { resp, err := client.Get(url) if err != nil { return errors.Trace(err) } defer resp.Body.Close() if resp.StatusCode != http.StatusOK { body, err := ioutil.ReadAll(resp.Body) if err != nil { return errors.Trace(err) } return errors.Errorf("get %s http status code != 200, message %s", url, string(body)) } return errors.Trace(json.NewDecoder(resp.Body).Decode(v)) }