NSClient.go

package main

import (
	"context"
	"fmt"
	"log"
	"strconv"
	"strings"
)
import "github.com/go-resty/resty/v2"

type NSClient struct {
	nsUri   string
	nsToken string
}

type nsDeviceStatusResult struct {
	Status  int       `json:"status"`
	Records []NsEntry `json:"result"`
}
type nsTreatmentsResult struct {
	Status  int           `json:"status"`
	Records []NsTreatment `json:"result"`
}

func NewNSClient(uri string, token string) *NSClient {
	return &NSClient{
		nsUri:   strings.TrimRight(uri, "/"),
		nsToken: token,
	}
}

func (c NSClient) LoadDeviceStatuses(queue chan NsEntry, limit int64, skip int64, _ context.Context) {
	defer wg.Done()

	fmt.Println("LoadDeviceStatuses from NS, limit: ", limit, ", skip: ", skip)

	client := resty.New()

	entries := &nsDeviceStatusResult{}
	_, err := client.R().
		SetQueryParams(map[string]string{
			"skip":      strconv.FormatInt(skip, 10),
			"limit":     strconv.FormatInt(limit, 10),
			"sort$desc": "created_at",
			"token":     c.nsToken,
		}).
		SetResult(entries).
		SetHeader("Accept", "application/json").
		Get(c.nsUri + "/api/v3/devicestatus")

	if err != nil {
		log.Fatal(err)
	}

	for _, entry := range entries.Records {
		queue <- entry
	}
}

func (c NSClient) LoadTreatments(queue chan NsTreatment, limit int64, skip int64, _ context.Context) {
	defer wg.Done()

	fmt.Println("LoadTreatments from NS, limit: ", limit, ", skip: ", skip)

	client := resty.New()

	entries := &nsTreatmentsResult{}
	_, err := client.R().
		SetQueryParams(map[string]string{
			"skip":      strconv.FormatInt(skip, 10),
			"limit":     strconv.FormatInt(limit, 10),
			"sort$desc": "created_at",
			"token":     c.nsToken,
		}).
		SetResult(entries).
		SetHeader("Accept", "application/json").
		Get(c.nsUri + "/api/v3/treatments")

	if err != nil {
		log.Fatal(err)
	}
	for _, entry := range entries.Records {
		queue <- entry
	}
}

func (c NSClient) Close(_ context.Context) {}