func call(ctx context.Context, method string, req proto.Message, resp proto.Message) error {
payload, err := proto.Marshal(req)
if err != nil {
return err
}
url := baseUrl(ctx) + internal.ProjID(ctx) + "/" + method
r, err := internal.HTTPClient(ctx).Post(url, "application/x-protobuf", bytes.NewReader(payload))
if err != nil {
return err
}
defer r.Body.Close()
all, err := ioutil.ReadAll(r.Body)
if r.StatusCode != http.StatusOK {
e := &errHTTP{
StatusCode: r.StatusCode,
err: err,
}
if err == nil {
e.Body = string(all)
}
return e
}
if err != nil {
return err
}
if err = proto.Unmarshal(all, resp); err != nil {
return err
}
return nil
}
// Encode returns an opaque representation of the key
// suitable for use in HTML and URLs.
// This is compatible with the Python and Java runtimes.
func (k *Key) Encode() string {
pKey := keyToProto(k)
b, err := proto.Marshal(pKey)
if err != nil {
panic(err)
}
// Trailing padding is stripped.
return strings.TrimRight(base64.URLEncoding.EncodeToString(b), "=")
}
func (t *fakeTransport) RoundTrip(req *http.Request) (*http.Response, error) {
body, err := ioutil.ReadAll(req.Body)
var in pb.RunQueryRequest
var resp http.Response
if err = proto.Unmarshal(body, &in); err != nil {
// Get back an error
resp = http.Response{
StatusCode: http.StatusBadRequest,
}
} else {
// Run our fake query and serialize the response
var out pb.RunQueryResponse
err := t.Handler(&in, &out)
if err != nil {
resp = http.Response{
StatusCode: http.StatusBadRequest,
}
} else {
payload, err := proto.Marshal(&out)
if err != nil {
resp = http.Response{
StatusCode: http.StatusBadRequest,
}
} else {
resp = http.Response{
StatusCode: http.StatusOK,
Body: ioutil.NopCloser(bytes.NewBuffer(payload)),
}
}
}
}
// Set common response fields
resp.Proto = "HTTP/1.0"
resp.ProtoMajor = 1
resp.ProtoMinor = 1
return &resp, nil
}