health: Replace header check if with switch

This commit is contained in:
crypto-services 2024-02-21 09:27:05 +08:00
parent b1597736d7
commit 08dc03e2c6

View file

@ -46,26 +46,24 @@ func processFromHeaders(ec ethClient, headers []string, w http.ResponseWriter, r
for _, header := range headers { for _, header := range headers {
lHeader := strings.ToLower(header) lHeader := strings.ToLower(header)
if lHeader == synced { switch {
case lHeader == synced:
errCheckSynced = checkSynced(ec, r) errCheckSynced = checkSynced(ec, r)
} case strings.HasPrefix(lHeader, minPeerCount):
if strings.HasPrefix(lHeader, minPeerCount) {
peers, err := strconv.Atoi(strings.TrimPrefix(lHeader, minPeerCount)) peers, err := strconv.Atoi(strings.TrimPrefix(lHeader, minPeerCount))
if err != nil { if err != nil {
errCheckPeer = err errCheckPeer = err
break break
} }
errCheckPeer = checkMinPeers(ec, uint(peers)) errCheckPeer = checkMinPeers(ec, uint(peers))
} case strings.HasPrefix(lHeader, checkBlock):
if strings.HasPrefix(lHeader, checkBlock) {
block, err := strconv.Atoi(strings.TrimPrefix(lHeader, checkBlock)) block, err := strconv.Atoi(strings.TrimPrefix(lHeader, checkBlock))
if err != nil { if err != nil {
errCheckBlock = err errCheckBlock = err
break break
} }
errCheckBlock = checkBlockNumber(ec, big.NewInt(int64(block))) errCheckBlock = checkBlockNumber(ec, big.NewInt(int64(block)))
} case strings.HasPrefix(lHeader, maxSecondsBehind):
if strings.HasPrefix(lHeader, maxSecondsBehind) {
seconds, err := strconv.Atoi(strings.TrimPrefix(lHeader, maxSecondsBehind)) seconds, err := strconv.Atoi(strings.TrimPrefix(lHeader, maxSecondsBehind))
if err != nil { if err != nil {
errCheckSeconds = err errCheckSeconds = err