[#1375] Use more strict type

fivitti · fivitti · commit e01d567c0faa · 2024-06-25T11:49:01.000Z
diff --git a/backend/server/database/model/subnet.go b/backend/server/database/model/subnet.go
@@ -50,7 +50,7 @@ const (
 type SubnetStats map[SubnetStatsName]any
 
 // Returns the value of the statistic with the specified name as a big counter.
-func (s SubnetStats) GetBigCounter(name string) *storkutil.BigCounter {
+func (s SubnetStats) GetBigCounter(name SubnetStatsName) *storkutil.BigCounter {
 	value, ok := s[name]
 	if !ok {
 		return nil
@@ -69,7 +69,7 @@ func (s SubnetStats) GetBigCounter(name string) *storkutil.BigCounter {
 }
 
 // Sets the value of the statistic with the specified name as a big counter.
-func (s SubnetStats) SetBigCounter(name string, counter *storkutil.BigCounter) {
+func (s SubnetStats) SetBigCounter(name SubnetStatsName, counter *storkutil.BigCounter) {
 	s[name] = counter.ConvertToNativeType()
 }
 

Original file line number	Diff line number	Diff line change
`@@ -50,7 +50,7 @@ const (`
`50`	`50`	`type SubnetStats map[SubnetStatsName]any`
`51`	`51`
`52`	`52`	`// Returns the value of the statistic with the specified name as a big counter.`
`53`		`-func (s SubnetStats) GetBigCounter(name string) *storkutil.BigCounter {`
	`53`	`+func (s SubnetStats) GetBigCounter(name SubnetStatsName) *storkutil.BigCounter {`
`54`	`54`	`value, ok := s[name]`
`55`	`55`	`if !ok {`
`56`	`56`	`return nil`
`@@ -69,7 +69,7 @@ func (s SubnetStats) GetBigCounter(name string) *storkutil.BigCounter {`
`69`	`69`	`}`
`70`	`70`
`71`	`71`	`// Sets the value of the statistic with the specified name as a big counter.`
`72`		`-func (s SubnetStats) SetBigCounter(name string, counter *storkutil.BigCounter) {`
	`72`	`+func (s SubnetStats) SetBigCounter(name SubnetStatsName, counter *storkutil.BigCounter) {`
`73`	`73`	`s[name] = counter.ConvertToNativeType()`
`74`	`74`	`}`
`75`	`75`