setup.go

package main

import (
	"context"
	"fmt"

	"github.com/lovi-cloud/go-os-brick/osbrick"

	dspb "github.com/lovi-cloud/satelit/api/satelit_datastore"
	pb "github.com/lovi-cloud/teleskop/protoc/agent"
)

func (a *agent) setup(ctx context.Context, hostname, endpoint, parentInterfaceName string) error {
	resp, err := a.datastoreClient.ListBridge(ctx, &dspb.ListBridgeRequest{})
	if err != nil {
		return err
	}

	for _, bridge := range resp.Bridges {
		_, err = a.AddBridge(ctx, &pb.AddBridgeRequest{
			Name:         bridge.Name,
			MetadataCidr: bridge.MetadataCidr,
			InternalOnly: bridge.InternalOnly,
		})
		if err != nil {
			return err
		}
		if bridge.InternalOnly {
			continue
		}
		_, err = a.AddVLANInterface(ctx, &pb.AddVLANInterfaceRequest{
			VlanId:          bridge.VlanId,
			ParentInterface: parentInterfaceName,
		})
		if err != nil {
			return err
		}
		_, err = a.AddInterfaceToBridge(ctx, &pb.AddInterfaceToBridgeRequest{
			Bridge:    bridge.Name,
			Interface: fmt.Sprintf("%s.%d", parentInterfaceName, bridge.VlanId),
		})
		if err != nil {
			return err
		}
	}

	numaNodes, err := GetLocalNUMANodes()
	if err != nil {
		return err
	}

	iqn, err := osbrick.GetIQN(ctx)
	if err != nil {
		return err
	}

	_, err = a.datastoreClient.RegisterTeleskopAgent(ctx, &dspb.RegisterTeleskopAgentRequest{
		Hostname: hostname,
		Endpoint: endpoint,
		Iqn:      iqn,
		Nodes:    numaNodes,
	})
	if err != nil {
		return err
	}

	return nil
}