handler.go

// Copyright © 2017 The Things Network
// Use of this source code is governed by the MIT license that can be found in the LICENSE file.

package ttnsdk

import (
	"strings"

	"github.com/TheThingsNetwork/go-utils/log"
	"google.golang.org/grpc"
	"google.golang.org/grpc/credentials"
)

func (c *client) connectHandler() (err error) {
	c.handler.Lock()
	defer c.handler.Unlock()
	if c.handler.conn != nil {
		return nil
	}
	if c.handler.announcement == nil {
		if err := c.discover(); err != nil {
			return err
		}
	}
	tlsConfig, err := c.handler.announcement.GetTLSConfig()
	if err != nil {
		return err
	}
	logger := c.Logger.WithFields(log.Fields{
		"ID":      c.handler.announcement.ID,
		"Address": c.handler.announcement.NetAddress,
	})
	logger.Debug("ttn-sdk: Connecting to handler...")
	c.handler.conn, err = grpc.Dial(
		strings.Split(c.handler.announcement.NetAddress, ",")[0],
		append(DialOptions, grpc.WithTransportCredentials(credentials.NewTLS(tlsConfig)))...)
	if err != nil {
		logger.WithError(err).Debug("ttn-sdk: Could not connect to handler")
		return err
	}
	logger.Debug("ttn-sdk: Connected to handler")
	return nil
}

func (c *client) closeHandler() error {
	c.handler.Lock()
	defer c.handler.Unlock()
	if c.handler.conn != nil {
		c.handler.conn.Close()
	}
	c.handler.conn = nil
	return nil
}