host.go

package deployment

import (
	"fmt"

	"github.com/go-logr/logr"

	aero "github.com/aerospike/aerospike-client-go/v7"
	"github.com/tanmayja/aerospike-management-lib/info"
)

// host is a system on which the aerospike server is running. It provides aerospike
// specific capabilities on the system.
type host struct {
	log        logr.Logger
	asConnInfo *asConnInfo
	id         string // host UUID string
}

type asConnInfo struct {
	// aerospike specific details
	aerospikePolicy   *aero.ClientPolicy
	asInfo            *info.AsInfo
	aerospikeHostName string
	aerospikePort     int
}

// newHost creates an aerospike host.
func newHost(
	id string, aerospikePolicy *aero.ClientPolicy, asConn *ASConn,
) (*host, error) {
	nd := host{
		id: id,
	}

	if asConn != nil {
		nd.log = asConn.Log
		nd.asConnInfo = newASConnInfo(aerospikePolicy, asConn)
	}

	return &nd, nil
}

func newASConnInfo(aerospikePolicy *aero.ClientPolicy, asConn *ASConn) *asConnInfo {
	h := aero.Host{
		Name:    asConn.AerospikeHostName,
		Port:    asConn.AerospikePort,
		TLSName: asConn.AerospikeTLSName,
	}
	asInfo := info.NewAsInfo(asConn.Log, &h, aerospikePolicy)

	return &asConnInfo{
		aerospikeHostName: asConn.AerospikeHostName,
		aerospikePort:     asConn.AerospikePort,
		aerospikePolicy:   aerospikePolicy,
		asInfo:            asInfo,
	}
}

func (n *host) String() string {
	return n.id
}

// Close closes all the open connections of the host.
func (n *host) Close() error {
	if n.asConnInfo.asInfo != nil {
		if err := n.asConnInfo.asInfo.Close(); err != nil {
			return fmt.Errorf(
				"failed to close asinfo/system connection for host %s: %v",
				n.asConnInfo.aerospikeHostName, err,
			)
		}
	}

	return nil
}