Merge branch 'sprint-july-4' into refactor/db-txn

Hitenjain14 · web-flow · commit b7213aa0d631 · 2023-08-13T03:39:53.000+05:30
diff --git a/code/go/0chain.net/validatorcore/storage/challenge_handler.go b/code/go/0chain.net/validatorcore/storage/challenge_handler.go
@@ -49,9 +49,16 @@ func challengeHandler(ctx context.Context, r *http.Request) (interface{}, error)
 
 	err = challengeRequest.VerifyChallenge(challengeObj, allocationObj)
 	if err != nil {
+		statsMutex.Lock()
+		updateStats(false)
+		statsMutex.Unlock()
 		return InvalidValidationTicket(challengeObj, err)
 	}
 
+	statsMutex.Lock()
+	updateStats(false)
+	statsMutex.Unlock()
+
 	return ValidValidationTicket(challengeObj, challengeRequest.ChallengeID, challengeHash)
 }
 
diff --git a/code/go/0chain.net/validatorcore/storage/handler_main.go b/code/go/0chain.net/validatorcore/storage/handler_main.go
@@ -18,4 +18,6 @@ func SetupHandlers(r *mux.Router) {
 		RateLimit(common.ToJSONResponse(SetupContext(ChallengeHandler))))
 
 	r.HandleFunc("/debug", common.ToJSONResponse(DumpGoRoutines))
+
+	r.HandleFunc("/stats", statsHandler)
 }
diff --git a/code/go/0chain.net/validatorcore/storage/stats_handler.go b/code/go/0chain.net/validatorcore/storage/stats_handler.go
@@ -0,0 +1,100 @@
+package storage
+
+import (
+	"fmt"
+	"net/http"
+	"sync"
+)
+
+type Stats struct {
+	TotalChallenges      int
+	SuccessfulChallenges int
+	FailedChallenges     int
+}
+
+var (
+	appStats   Stats
+	statsMutex sync.Mutex
+)
+
+func init() {
+	appStats = Stats{}
+
+}
+
+func updateStats(success bool) {
+	statsMutex.Lock()
+	defer statsMutex.Unlock()
+
+	appStats.TotalChallenges++
+
+	if success {
+		appStats.SuccessfulChallenges++
+	} else {
+		appStats.FailedChallenges++
+	}
+}
+
+func getStats() Stats {
+	statsMutex.Lock()
+	defer statsMutex.Unlock()
+	return appStats
+}
+
+func statsHandler(w http.ResponseWriter, r *http.Request) {
+	result := getStats()
+
+	statsHTML := `
+	<!DOCTYPE html>
+	<html>
+	<head>
+		<style>
+			table {
+				font-family: Arial, sans-serif;
+				border-collapse: collapse;
+				width: 50%;
+				margin: auto;
+				margin-top: 50px;
+			}
+
+			th, td {
+				border: 1px solid #dddddd;
+				text-align: left;
+				padding: 8px;
+			}
+
+			th {
+				background-color: #f2f2f2;
+			}
+		</style>
+	</head>
+	<body>
+		<h1>Challenges Statistics</h1>
+		<table>
+			<tr>
+				<th>Statistic</th>
+				<th>Count</th>
+			</tr>
+			<tr>
+				<td>Total Challenges</td>
+				<td>` + fmt.Sprintf("%d", result.TotalChallenges) + `</td>
+			</tr>
+			<tr>
+				<td>Successful Challenges</td>
+				<td>` + fmt.Sprintf("%d", result.SuccessfulChallenges) + `</td>
+			</tr>
+			<tr>
+				<td>Failed Challenges</td>
+				<td>` + fmt.Sprintf("%d", result.FailedChallenges) + `</td>
+			</tr>
+		</table>
+	</body>
+	</html>
+	`
+
+	w.Header().Set("Content-Type", "text/html")
+	_, err := w.Write([]byte(statsHTML))
+	if err != nil {
+		return
+	}
+}

Original file line number	Diff line number	Diff line change
`@@ -49,9 +49,16 @@ func challengeHandler(ctx context.Context, r *http.Request) (interface{}, error)`
`49`	`49`
`50`	`50`	`err = challengeRequest.VerifyChallenge(challengeObj, allocationObj)`
`51`	`51`	`if err != nil {`
	`52`	`+ statsMutex.Lock()`
	`53`	`+ updateStats(false)`
	`54`	`+ statsMutex.Unlock()`
`52`	`55`	`return InvalidValidationTicket(challengeObj, err)`
`53`	`56`	`}`
`54`	`57`
	`58`	`+ statsMutex.Lock()`
	`59`	`+ updateStats(false)`
	`60`	`+ statsMutex.Unlock()`
	`61`	`+`
`55`	`62`	`return ValidValidationTicket(challengeObj, challengeRequest.ChallengeID, challengeHash)`
`56`	`63`	`}`
`57`	`64`
Original file line number	Diff line number	Diff line change
`@@ -18,4 +18,6 @@ func SetupHandlers(r *mux.Router) {`
`18`	`18`	`RateLimit(common.ToJSONResponse(SetupContext(ChallengeHandler))))`
`19`	`19`
`20`	`20`	`r.HandleFunc("/debug", common.ToJSONResponse(DumpGoRoutines))`
	`21`	`+`
	`22`	`+ r.HandleFunc("/stats", statsHandler)`
`21`	`23`	`}`