update file

pkg/db/db.go

@@ -1,6 +1,7 @@
 package db
 
 import (
+	"os"
 	"time"
 
 	"gorm.io/driver/mysql"
@@ -13,10 +14,11 @@ var (
 
 type TinyUrl struct {
 	gorm.Model
-	LongUrl    string `gorm:"size:200"`
-	ShortUrl   string `gorm:"size:50"`
+	LongUrl    string `gorm:"size:7713"`
+	ShortUrl   string `gorm:"size:20"`
 	ExpireTime time.Time
 	Counter    float64
+	AddIP      string
 }
 
 type URLDetail struct {
@@ -27,7 +29,12 @@ type URLDetail struct {
 
 func InitDb() {
 	// 参考 https://github.com/go-sql-driver/mysql#dsn-data-source-name 获取详情
-	dsn := "tinyurl:tinyurl@tcp(42.192.36.14:3306)/tinyurl?charset=utf8mb4&parseTime=True&loc=Local"
+	var dsn string
+	if len(os.Getenv("DSN")) > 1 {
+		dsn = os.Getenv("DSN")
+	} else {
+		dsn = "tinyurl:tinyurl@tcp(42.192.36.14:3306)/tinyurl?charset=utf8mb4&parseTime=True&loc=Local"
+	}
 	var err error
 	DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
 	if err != nil {