Consistent hashing server that will determine a key path to write information to across a key space

main.go

package main

import (
	"flag"
	"fmt"
	"log"
	"net/http"

	"github.com/gorilla/mux"
	"github.com/lafikl/consistent"
)

var (
	key        string
	partitions int
)

func init() {
	flag.StringVar(&key, "key-path", "kv/test/path", "the path for storage")
	flag.IntVar(&partitions, "partitions", 3, "numbers of partitions in keyspace")
	flag.Parse()
}

func main() {
	// set up consistent hashing
	c := consistent.New()

	// add keys spaces for hashing
	for i := 0; i < partitions; i++ {
		c.Add(fmt.Sprintf("%s-%v", key, i))
	}

	// setup endpoint
	r := mux.NewRouter()
	r.HandleFunc("/key/{namespace}/{service}", func(w http.ResponseWriter, r *http.Request) {
		vars := mux.Vars(r)

		// get partition and increment load
		p, err := c.GetLeast(vars["namespace"])
		if err != nil {
			log.Println(err)
		}
		c.Inc(p)

		// return
		w.WriteHeader(http.StatusOK)
		fmt.Fprintf(w, "%v/%v/%v\n", p, vars["namespace"], vars["service"])
	})
	http.ListenAndServe(":8080", r)
}

results.txt

(base) ➜  ~ curl localhost:8080/key/cat-production/service-1
kv/test/path-2/cat-production/service-1
(base) ➜  ~ curl localhost:8080/key/cat-production/service-2
kv/test/path-2/cat-production/service-2
(base) ➜  ~ curl localhost:8080/key/cat-production/service-3
kv/test/path-1/cat-production/service-3