Add harvester + farmer healthchecks (#83)

* Add harvester healthcheck

* Update to have an endpoint for harvester with and without plots

* Add farmer healthcheck

Author: cmmarslender

internal/healthcheck/farmer.go

@@ -0,0 +1,32 @@
+package healthcheck
+
+import (
+	"encoding/json"
+	"net/http"
+	"time"
+
+	"github.com/chia-network/go-chia-libs/pkg/types"
+	log "github.com/sirupsen/logrus"
+)
+
+func (h *Healthcheck) farmerReceive(resp *types.WebsocketResponse) {
+	if resp.Command != "new_signage_point" {
+		return
+	}
+
+	newSP := &types.EventFarmerNewSignagePoint{}
+	err := json.Unmarshal(resp.Data, newSP)
+	if err != nil {
+		log.Errorf("Error unmarshalling: %s\n", err.Error())
+		return
+	}
+
+	h.lastFarmerTime = time.Now()
+}
+
+// farmerHealthcheck endpoint for the farmer service as a whole
+func (h *Healthcheck) farmerHealthcheck() func(http.ResponseWriter, *http.Request) {
+	return func(w http.ResponseWriter, r *http.Request) {
+		timeMetricHealthcheckHelper(h.lastFarmerTime, w, r)
+	}
+}

internal/healthcheck/harvester.go

@@ -0,0 +1,46 @@
+package healthcheck
+
+import (
+	"encoding/json"
+	"net/http"
+	"time"
+
+	"github.com/chia-network/go-chia-libs/pkg/types"
+	log "github.com/sirupsen/logrus"
+)
+
+func (h *Healthcheck) harvesterReceive(resp *types.WebsocketResponse) {
+	if resp.Command != "farming_info" {
+		return
+	}
+
+	farmingInfo := &types.EventHarvesterFarmingInfo{}
+	err := json.Unmarshal(resp.Data, farmingInfo)
+	if err != nil {
+		log.Errorf("Error unmarshalling: %s\n", err.Error())
+		return
+	}
+
+	h.lastHarvesterTime = time.Now()
+
+	if farmingInfo.TotalPlots == 0 {
+		log.Errorf("No plots found. Not Ready!")
+		return
+	}
+
+	h.lastHarvesterTimeWithPlots = time.Now()
+}
+
+// harvesterHealthcheckWithPlots endpoint for the harvester service requiring that at least one plot is found
+func (h *Healthcheck) harvesterHealthcheckWithPlots() func(http.ResponseWriter, *http.Request) {
+	return func(w http.ResponseWriter, r *http.Request) {
+		timeMetricHealthcheckHelper(h.lastHarvesterTimeWithPlots, w, r)
+	}
+}
+
+// harvesterHealthcheck endpoint for the harvester service as a whole
+func (h *Healthcheck) harvesterHealthcheck() func(http.ResponseWriter, *http.Request) {
+	return func(w http.ResponseWriter, r *http.Request) {
+		timeMetricHealthcheckHelper(h.lastHarvesterTime, w, r)
+	}
+}

internal/healthcheck/healthcheck.go

@@ -40,6 +40,15 @@ type Healthcheck struct {
 
 	// Time we got a good response from the timelord
 	lastTimelordTime time.Time
+
+	// Last time the farmer received new SP info from the full node
+	lastFarmerTime time.Time
+
+	// Last time harvester looked for plots eligible to farm
+	lastHarvesterTime time.Time
+
+	// Last time harvester looked for plots eligible to farm AND had > 0 plots
+	lastHarvesterTimeWithPlots time.Time
 }
 
 // NewHealthcheck returns a new instance of healthcheck
@@ -99,6 +108,9 @@ func (h *Healthcheck) StartServer() error {
 	http.HandleFunc("/seeder/readiness", h.seederReadiness())
 	http.HandleFunc("/timelord", h.timelordHealthcheck())
 	http.HandleFunc("/timelord/readiness", h.timelordHealthcheck())
+	http.HandleFunc("/farmer", h.farmerHealthcheck())
+	http.HandleFunc("/harvester", h.harvesterHealthcheck())
+	http.HandleFunc("/harvester/with-plots", h.harvesterHealthcheckWithPlots())
 	return http.ListenAndServe(fmt.Sprintf(":%d", h.healthcheckPort), nil)
 }
 
@@ -131,10 +143,6 @@ func (h *Healthcheck) walletReceive(resp *types.WebsocketResponse) {}
 
 func (h *Healthcheck) crawlerReceive(resp *types.WebsocketResponse) {}
 
-func (h *Healthcheck) harvesterReceive(resp *types.WebsocketResponse) {}
-
-func (h *Healthcheck) farmerReceive(resp *types.WebsocketResponse) {}
-
 func (h *Healthcheck) disconnectHandler() {
 	log.Debug("Calling disconnect handlers")
 	// @TODO should we mark unhealthy immediately?