chain_name added

provalidator · Jul 6, 2023 · b1204ef · b1204ef
1 parent aa95eb8
commit b1204ef
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 8 deletions.
diff --git a/models/DB.go b/models/DB.go
@@ -1,10 +1,11 @@
 package models
 
 type ApiUsage struct {
-	Sub    string `json:"sub"`
-	Token  string `json:"token"`
-	Ip     string `json:"ip"`
-	Method string `json:"method"`
-	Date   string `json:"date"`
-	Cnt    int    `json:"cnt"`
+	Sub       string `json:"sub"`
+	Token     string `json:"token"`
+	Ip        string `json:"ip"`
+	Method    string `json:"method"`
+	Date      string `json:"date"`
+	Cnt       int    `json:"cnt"`
+	ChainName string `json:"chain_name"`
 }
diff --git a/models/database.go b/models/database.go
@@ -35,10 +35,10 @@ func ConnectDatabase() {
 }
 
 func WriteLog(sub string, token string, ip string, method string) {
-	log.Logger.Info.Println("WriteLog. Token:", token, ", IP:", ip, ", Method:", method)
+	//log.Logger.Info.Println("WriteLog. Token:", token, ", IP:", ip, ", Method:", method)
 	now := time.Now()
 	date := now.Format("2006-01-02 15:04:05")
-	input := ApiUsage{Sub: sub, Token: token, Ip: ip, Method: method, Date: date}
+	input := ApiUsage{Sub: sub, Token: token, Ip: ip, Method: method, Date: date, ChainName: os.Getenv("CHAIN_NAME")}
 	DB.Create(&input)
 }