From 0d7c58716e7fb232e3449acf945e5895647e2f0c Mon Sep 17 00:00:00 2001
From: zu1k <i@lgf.im>
Date: Sat, 5 Sep 2020 10:40:30 +0800
Subject: [PATCH] fix db

---
 internal/app/task.go       | 4 +++-
 internal/database/proxy.go | 4 ++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/internal/app/task.go b/internal/app/task.go
index 4f02b07..19ace3b 100644
--- a/internal/app/task.go
+++ b/internal/app/task.go
@@ -33,7 +33,6 @@ func CrawlGo() {
 	}
 	// 节点去重
 	proxies = proxies.Deduplication()
-	database.SaveProxyList(proxies)
 	log.Println("CrawlGo node count:", len(proxies))
 	proxies = provider.Clash{
 		provider.Base{
@@ -44,6 +43,9 @@ func CrawlGo() {
 	proxies.NameAddCounrty().Sort().NameAddIndex().NameAddTG()
 	log.Println("Proxy rename DONE!")
 
+	// 全节点存储到数据库
+	database.SaveProxyList(proxies)
+
 	cache.SetProxies("allproxies", proxies)
 	cache.AllProxiesCount = proxies.Len()
 	log.Println("AllProxiesCount:", cache.AllProxiesCount)
diff --git a/internal/database/proxy.go b/internal/database/proxy.go
index 095c069..7c64835 100644
--- a/internal/database/proxy.go
+++ b/internal/database/proxy.go
@@ -7,9 +7,9 @@ import (
 
 type Proxy struct {
 	gorm.Model
-	proxy.Base
+	proxy.Base `gorm:"index"`
 	Link       string
-	Identifier string
+	Identifier string `gorm:"primaryKey"`
 }
 
 func InitTables() {