remove duplicate code

jreisinger · Jun 7, 2024 · c989f2f · c989f2f
1 parent 25193ea
commit c989f2f
Showing 1 changed file with 3 additions and 10 deletions.
diff --git a/checkip.go b/checkip.go
@@ -3,6 +3,7 @@ package checkip
 
 import (
  "encoding/json"
+ "fmt"
  "net"
 )
 
@@ -18,19 +19,11 @@ type Type int
 
 // String returns the name of the Check type.
 func (t Type) String() string {
- return [...]string{"Info", "Sec", "InfoSec"}[t]
+ return [...]string{"info", "sec", "infosec"}[t]
 }
 
 func (t Type) MarshalJSON() ([]byte, error) {
- var s string
- switch t {
- case TypeInfo:
- s = "info"
- case TypeSec:
- s = "security"
- case TypeInfoSec:
- s = "infoAndSecurity"
- }
+ s := fmt.Sprint(t)
  return json.Marshal(s)
 }