block_body_download_stats.go

package diagnostics

import (
	"fmt"
	"io"
	"net/http"
	"strconv"

	"github.com/nebojsa94/erigon/dataflow"
)

func SetupBlockBodyDownload(metricsMux *http.ServeMux) {
	metricsMux.HandleFunc("/block_body_download", func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("Access-Control-Allow-Origin", "*")
		writeBlockBodyDownload(w, r)
	})
}

func writeBlockBodyDownload(w io.Writer, r *http.Request) {
	if err := r.ParseForm(); err != nil {
		fmt.Fprintf(w, "ERROR: parsing arguments: %v\n", err)
		return
	}
	sinceTickStr := r.Form.Get("sincetick")
	var tick int64
	if sinceTickStr != "" {
		var err error
		if tick, err = strconv.ParseInt(sinceTickStr, 10, 64); err != nil {
			fmt.Fprintf(w, "ERROR: parsing sincemilli: %v\n", err)
		}
	}
	fmt.Fprintf(w, "SUCCESS\n")
	dataflow.BlockBodyDownloadStates.ChangesSince(int(tick), w)
}