service_user_stream.go

package binance

import (
	"encoding/json"
	"io/ioutil"
	"log"

	"github.com/pkg/errors"
)

func (as *apiService) StartUserDataStream() (*Stream, error) {
	params := make(map[string]string)

	res, err := as.request("POST", "api/v1/userDataStream", params, true, false)
	if err != nil {
		return nil, err
	}
	textRes, err := ioutil.ReadAll(res.Body)
	if err != nil {
		return nil, errors.Wrap(err, "unable to read response from userDataStream.post")
	}
	defer res.Body.Close()

	log.Println(string(textRes))
	if res.StatusCode != 200 {
		return nil, as.handleError(textRes)
	}

	var s Stream
	if err := json.Unmarshal(textRes, &s); err != nil {
		return nil, errors.Wrap(err, "stream unmarshal failed")
	}
	return &s, nil
}
func (as *apiService) KeepAliveUserDataStream(s *Stream) error {
	params := make(map[string]string)
	params["listenKey"] = s.ListenKey

	res, err := as.request("PUT", "api/v1/userDataStream", params, true, false)
	if err != nil {
		return err
	}
	textRes, err := ioutil.ReadAll(res.Body)
	if err != nil {
		return errors.Wrap(err, "unable to read response from userDataStream.put")
	}
	defer res.Body.Close()

	if res.StatusCode != 200 {
		return as.handleError(textRes)
	}
	return nil
}
func (as *apiService) CloseUserDataStream(s *Stream) error {
	params := make(map[string]string)
	params["listenKey"] = s.ListenKey

	res, err := as.request("DELETE", "api/v1/userDataStream", params, true, false)
	if err != nil {
		return err
	}
	textRes, err := ioutil.ReadAll(res.Body)
	if err != nil {
		return errors.Wrap(err, "unable to read response from userDataStream.delete")
	}
	defer res.Body.Close()

	if res.StatusCode != 200 {
		return as.handleError(textRes)
	}
	return nil
}