dadata/client/client_test.go

package client

import (
	"bytes"
	"context"
	"encoding/json"
	"fmt"
	"gitea.24example.ru/spavelit/dadata/api/suggest"
	"gitea.24example.ru/spavelit/dadata/client/transport"
	"gitea.24example.ru/spavelit/dadata/client/transport/encoder"
	"io"
	"net/url"
)

func ExampleNewClient() {
	var err error
	endpointUrl, err := url.Parse("https://suggestions.dadata.ru/suggestions/api/4_1/rs/")
	if err != nil {
		return
	}

	api := suggest.Api{
		Client: NewClient(endpointUrl),
	}

	params := suggest.RequestParams{
		Query: "ул Свободы",
	}

	suggestions, err := api.Address(context.Background(), &params)
	if err != nil {
		return
	}

	for _, s := range suggestions {
		fmt.Printf("%s", s.Value)
	}
}

func ExampleCredentials() {
	var err error
	endpointUrl, err := url.Parse("https://suggestions.dadata.ru/suggestions/api/4_1/rs/")
	if err != nil {
		return
	}

	creds := Credentials{
		ApiKeyValue:    "<YOUR_API_KEY>",
		SecretKeyValue: "<YOUR_SECRET_KEY>",
	}

	api := suggest.Api{
		Client: NewClient(endpointUrl, WithCredentialProvider(&creds)),
	}

	params := suggest.RequestParams{
		Query: "ул Свободы",
	}

	suggestions, err := api.Address(context.Background(), &params)
	if err != nil {
		return
	}

	for _, s := range suggestions {
		fmt.Printf("%s", s.Value)
	}
}

func ExampleWithEncoderFactory() {
	var err error
	endpointUrl, err := url.Parse("https://suggestions.dadata.ru/suggestions/api/4_1/rs/")
	if err != nil {
		return
	}

	// Customize json encoding
	encoderFactory := func(w io.Writer) transport.Encoder {
		e := json.NewEncoder(w)
		e.SetIndent("", "  ")
		return func(v interface{}) error {
			return e.Encode(v)
		}
	}

	// Customize json decoding
	decoderFactory := func(r io.Reader) transport.Decoder {
		d := json.NewDecoder(r)
		d.DisallowUnknownFields()
		return func(v interface{}) error {
			return d.Decode(v)
		}
	}

	api := suggest.Api{
		Client: NewClient(endpointUrl, WithEncoderFactory(encoderFactory), WithDecoderFactory(decoderFactory)),
	}

	params := suggest.RequestParams{
		Query: "ул Свободы",
	}

	suggestions, err := api.Address(context.Background(), &params)
	if err != nil {
		return
	}

	for _, s := range suggestions {
		fmt.Printf("%s", s.Value)
	}
}

func ExampleClient_low_level() {
	var err error
	endpointUrl, err := url.Parse("https://suggestions.dadata.ru/suggestions/api/4_1/rs/")
	if err != nil {
		return
	}

	cli := NewClient(endpointUrl,
		WithEncoderFactory(encoder.RawEncoderFactory()),
		WithDecoderFactory(encoder.RawDecoderFactory()),
	)

	request := bytes.NewBufferString("{ \"query\": \"москва хабар\" }")
	response := &bytes.Buffer{}

	err = cli.Post(context.Background(), "suggest/address", request, response)
	if err != nil {
		return
	}

	fmt.Print(response.String())
}