init.go

package quacktors

import (
	"fmt"
	"github.com/Azer0s/qpmd"
	"github.com/Azer0s/quacktors/config"
	"github.com/Azer0s/quacktors/logging"
	"github.com/Azer0s/quacktors/typeregister"
	"github.com/vmihailenco/msgpack/v5"
	"net"
)

var messageGatewayPort = uint16(0)
var gpGatewayPort = uint16(0)

var logger logging.Logger
var qpmdPort uint16

func initQuacktorSystems() {
	logger = config.GetLogger()
	qpmdPort = config.GetQpmdPort()

	initializeGateways()
	initializeQpmdConnection()
	initializeBuiltInMessages()
}

func initializeGateways() {
	var err error

	messageGatewayPort, err = startMessageGateway()
	if err != nil {
		logger.Fatal("there was an error while starting the message gateway",
			"error", err)
	}

	gpGatewayPort, err = startGeneralPurposeGateway()
	if err != nil {
		logger.Fatal("there was an error while starting the general purpose gateway",
			"error", err)
	}
}

func initializeQpmdConnection() {
	failIfConnectionError := func(err error) {
		if err != nil {
			panic("Couldn't connect to qpmd! Is qpmd running?")
		}
	}

	conn, err := net.Dial("tcp", fmt.Sprintf("localhost:%d", qpmdPort))
	failIfConnectionError(err)

	b, err := msgpack.Marshal(qpmd.Request{
		RequestType: qpmd.REQUEST_HELLO,
		Data: map[string]interface{}{
			qpmd.MACHINE_ID:           machineId,
			qpmd.MESSAGE_GATEWAY_PORT: messageGatewayPort,
			qpmd.GP_GATEWAY_PORT:      gpGatewayPort,
		},
	})
	try(err)

	_, err = conn.Write(b)
	failIfConnectionError(err)

	buf := make([]byte, 4096)
	_, err = conn.Read(buf)
	failIfConnectionError(err)

	res := qpmd.Response{}
	err = msgpack.Unmarshal(buf, &res)
	try(err)
}

func initializeBuiltInMessages() {
	typeregister.Store(Pid{}.Type(), Pid{})
	typeregister.Store(DownMessage{}.Type(), DownMessage{})
	typeregister.Store(PoisonPill{}.Type(), PoisonPill{})
	typeregister.Store(GenericMessage{}.Type(), GenericMessage{})
	typeregister.Store(DisconnectMessage{}.Type(), DisconnectMessage{})
	typeregister.Store(KillMessage{}.Type(), KillMessage{})
}