Esempio n. 1
0
func Start(httpAddr string, rep *representative.Representative) {
	http.HandleFunc("/total_resources", func(w http.ResponseWriter, r *http.Request) {
		json.NewEncoder(w).Encode(rep.TotalResources())
	})

	http.HandleFunc("/instances", func(w http.ResponseWriter, r *http.Request) {
		json.NewEncoder(w).Encode(rep.Instances())
	})

	http.HandleFunc("/reset", func(w http.ResponseWriter, r *http.Request) {
		rep.Reset()
	})

	http.HandleFunc("/set_instances", func(w http.ResponseWriter, r *http.Request) {
		var instances []instance.Instance

		err := json.NewDecoder(r.Body).Decode(&instances)
		if err != nil {
			w.WriteHeader(http.StatusBadRequest)
			return
		}

		rep.SetInstances(instances)
	})

	http.HandleFunc("/vote", func(w http.ResponseWriter, r *http.Request) {
		var inst instance.Instance

		err := json.NewDecoder(r.Body).Decode(&inst)
		if err != nil {
			w.WriteHeader(http.StatusBadRequest)
			return
		}

		score, err := rep.Vote(inst)
		if err != nil {
			w.WriteHeader(http.StatusServiceUnavailable)
			return
		}

		json.NewEncoder(w).Encode(score)
	})

	http.HandleFunc("/reserve_and_recast_vote", func(w http.ResponseWriter, r *http.Request) {
		var inst instance.Instance

		err := json.NewDecoder(r.Body).Decode(&inst)
		if err != nil {
			w.WriteHeader(http.StatusBadRequest)
			return
		}

		score, err := rep.ReserveAndRecastVote(inst)
		if err != nil {
			w.WriteHeader(http.StatusServiceUnavailable)
			return
		}

		json.NewEncoder(w).Encode(score)
	})

	http.HandleFunc("/release", func(w http.ResponseWriter, r *http.Request) {
		var inst instance.Instance

		err := json.NewDecoder(r.Body).Decode(&inst)
		if err != nil {
			w.WriteHeader(http.StatusBadRequest)
			return
		}

		rep.Release(inst)

		w.WriteHeader(http.StatusOK)
	})

	http.HandleFunc("/claim", func(w http.ResponseWriter, r *http.Request) {
		var inst instance.Instance

		err := json.NewDecoder(r.Body).Decode(&inst)
		if err != nil {
			w.WriteHeader(http.StatusBadRequest)
			return
		}

		rep.Claim(inst)

		w.WriteHeader(http.StatusOK)
	})

	fmt.Printf("[%s] serving http on %s\n", rep.Guid(), httpAddr)

	panic(http.ListenAndServe(httpAddr, nil))
}
Esempio n. 2
0
func Start(natsAddrs []string, rep *representative.Representative) {
	client := yagnats.NewClient()

	clusterInfo := &yagnats.ConnectionCluster{}

	for _, addr := range natsAddrs {
		clusterInfo.Members = append(clusterInfo.Members, &yagnats.ConnectionInfo{
			Addr: addr,
		})
	}

	err := client.Connect(clusterInfo)
	if err != nil {
		log.Fatalln("no nats:", err)
	}

	guid := rep.Guid()

	client.Subscribe(guid+".guid", func(msg *yagnats.Message) {
		jguid, _ := json.Marshal(rep.Guid())
		client.Publish(msg.ReplyTo, jguid)
	})

	client.Subscribe(guid+".total_resources", func(msg *yagnats.Message) {
		jresources, _ := json.Marshal(rep.TotalResources())
		client.Publish(msg.ReplyTo, jresources)
	})

	client.Subscribe(guid+".reset", func(msg *yagnats.Message) {
		rep.Reset()
		client.Publish(msg.ReplyTo, successResponse)
	})

	client.Subscribe(guid+".set_instances", func(msg *yagnats.Message) {
		var instances []instance.Instance

		err := json.Unmarshal(msg.Payload, &instances)
		if err != nil {
			client.Publish(msg.ReplyTo, errorResponse)
		}

		rep.SetInstances(instances)
		client.Publish(msg.ReplyTo, successResponse)
	})

	client.Subscribe(guid+".instances", func(msg *yagnats.Message) {
		jinstances, _ := json.Marshal(rep.Instances())
		client.Publish(msg.ReplyTo, jinstances)
	})

	client.Subscribe(guid+".vote", func(msg *yagnats.Message) {
		var inst instance.Instance

		err := json.Unmarshal(msg.Payload, &inst)
		if err != nil {
			panic(err)
		}

		response := types.VoteResult{
			Rep: guid,
		}

		defer func() {
			payload, _ := json.Marshal(response)
			client.Publish(msg.ReplyTo, payload)
		}()

		score, err := rep.Vote(inst)
		if err != nil {
			// log.Println(guid, "failed to vote:", err)
			response.Error = err.Error()
			return
		}

		response.Score = score
	})

	client.Subscribe(guid+".reserve_and_recast_vote", func(msg *yagnats.Message) {
		var inst instance.Instance

		responsePayload := errorResponse
		defer func() {
			client.Publish(msg.ReplyTo, responsePayload)
		}()

		err := json.Unmarshal(msg.Payload, &inst)
		if err != nil {
			// log.Println(guid, "invalid reserve_and_recast_vote request:", err)
			return
		}

		score, err := rep.ReserveAndRecastVote(inst)
		if err != nil {
			// log.Println(guid, "failed to reserve_and_recast_vote:", err)
			return
		}

		responsePayload, _ = json.Marshal(score)
	})

	client.Subscribe(guid+".release", func(msg *yagnats.Message) {
		var inst instance.Instance

		responsePayload := errorResponse
		defer func() {
			client.Publish(msg.ReplyTo, responsePayload)
		}()

		err := json.Unmarshal(msg.Payload, &inst)
		if err != nil {
			log.Println(guid, "invalid reserve_and_recast_vote request:", err)
			return
		}

		rep.Release(inst)

		responsePayload = successResponse
	})

	client.Subscribe(guid+".claim", func(msg *yagnats.Message) {
		var inst instance.Instance

		responsePayload := errorResponse
		defer func() {
			client.Publish(msg.ReplyTo, responsePayload)
		}()

		err := json.Unmarshal(msg.Payload, &inst)
		if err != nil {
			log.Println(guid, "invalid reserve_and_recast_vote request:", err)
			return
		}

		rep.Claim(inst)

		responsePayload = successResponse
	})

	fmt.Printf("[%s] listening for nats\n", guid)

	select {}
}