main.go

package main

import (
	"fmt"
	"os"
	"strconv"

	"github.com/pysel/dkvs/balancer"
	"github.com/pysel/dkvs/partition"
	"github.com/pysel/dkvs/shared"
	"github.com/pysel/dkvs/types"
)

/*
If node is a partition, arguments should be (in mentioned order):
  - mode (partition)
  - port
  - db path
*/
func main() {
	args := os.Args

	if len(args) < 2 {
		panic("Provide a mode of this node")
	}
	mode := args[1]
	switch mode {
	case types.PARTITION_MODE:
		if len(args) < 3 {
			panic(fmt.Sprintf("Need a port, db path, but got: %T", args))
		}
		port, err := strconv.Atoi(args[2])
		if err != nil {
			panic(err)
		}

		dbPath := args[3]

		p := partition.NewPartition(dbPath)
		server := partition.RegisterPartitionServer(p)

		wg, addr := shared.StartListeningOnPort(server, uint64(port))
		fmt.Println("Address: ", addr)

		wg.Wait() // wait while server is still running

	case types.BALANCER_MODE:
		if len(args) < 3 {
			panic(fmt.Sprintf("Need a port, amount of partitions, but got: %T", args))
		}

		port, err := strconv.Atoi(args[2])
		if err != nil {
			panic(err)
		}

		goalReplicas, err := strconv.Atoi(args[3])
		if err != nil {
			panic("invalid parameter for desired amount of partitions")
		}

		b := balancer.NewBalancer(balancer.BalancerDBPath, goalReplicas)
		server := balancer.RegisterBalancerServer(b)

		wg, addr := shared.StartListeningOnPort(server, uint64(port))
		fmt.Println("Address: ", addr)

		wg.Wait()
	}
}