.gitignore

@@ -2,9 +2,9 @@ config.yaml
 backend/files
 .history/
 .idea/**
-conf.yaml
 docs/**
 **.exe
 files/**
 **.log
-**.sql
+**.sql
\ No newline at end of file
+mysql
\ No newline at end of file

backend/app/controller/foo.go

@@ -11,6 +11,7 @@ import (
 	"github.com/spf13/viper"
 )
 
+// =======authenticate==========
 var letters = []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ")
 
 /*
@@ -26,17 +27,33 @@ func IdGen(n int) string {
 	return string(b)
 }
 
+/*
+func IdGen(n int ) string{
+	id:=newId(n)
+	var check bool
+	// 数据库里查找有无sid与id相同
+	while (!check){
+
+	}
+}
+*/
+
 /*
  * 新建一个sid和url的关联，并返回一个bool表示成功与否
  * 若无sessionId，生成一个八位sid并返回
  */
-func newAuthenticate(sid string, url string, passwd string, time time.Time) (rsid string, stat uint) {
+func newAuthenticate(sid string, url string, passwd string, time time.Time, isFirst bool) (rsid string, stat uint) {
 	if sid == "" {
 		rsid = IdGen(8)
 	} else {
 		rsid = sid
 	}
-	stat = model.Createlink(rsid, passwd, url, time)
+	if isFirst {
+		model.CreatelinkFirstTime(rsid, url, time)
+		stat = 1
+	} else {
+		stat = model.Createlink(rsid, passwd, url, time)
+	}
 	return rsid, stat
 }
 
@@ -50,32 +67,12 @@ func Autheticate(cookie *http.Cookie, url string, passwd string, time time.Time)
 		return 1 // 鉴权通过
 	} else {
 		var stat uint
-		sid, stat = newAuthenticate(sid, url, passwd, time)
+		sid, stat = newAuthenticate(sid, url, passwd, time, false)
 		cookie.Value = sid
 		return stat
 	}
 }
 
-/*
- * 通过文件扩展名获取ContentType
- */
-func GetFileContentType(fileType string) string {
-	var StrRet string = ""
-	switch fileType {
-	case ".txt":
-		StrRet = "text/plain"
-	case ".csv":
-		StrRet = "text/csv"
-	case ".tex":
-		StrRet = "application/x-tex"
-	case ".md":
-		StrRet = "text/x-markdown"
-	default:
-		StrRet = "text/plain"
-	}
-	return StrRet
-}
-
 // 设置cookie name sid, value link
 func SetCookie(c echo.Context, cookie *http.Cookie, sid string, maxAge int, time_ time.Time) error {
 	cookie.Name = "User" // 标识为user
@@ -114,6 +111,7 @@ func overflow(info *File, threshold int) bool {
 
 }
 
+// / ========== DB related==================
 func DBupdate(c echo.Context, info *File) (string, string) {
 	url := "http://pastebin/" + IdGen(8)
 	cookie, _ := c.Cookie("User")
@@ -123,8 +121,7 @@ func DBupdate(c echo.Context, info *File) (string, string) {
 	} else {
 		sid = cookie.Value
 	}
-	sid, _ = newAuthenticate(sid, url, info.Url, info.Time)
-	model.CreatelinkFirstTime(sid, url, info.Time)
+	sid, _ = newAuthenticate(sid, url, info.Url, info.Time, true)
 	if info.MaxAccess == 0 {
 		//num,_=strconv.ParseUint(GetSetting("maxDefaultAccess"), 10, 64)
 		info.MaxAccess = 30 // 设置最大默认可访问次数
@@ -133,17 +130,6 @@ func DBupdate(c echo.Context, info *File) (string, string) {
 	return sid, url
 }
 
-// 格式化后缀，仿止出错
-func TypeComplement(typ string) string {
-	// 类型判断
-	if typ == "" {
-		typ = ".txt"
-	} else if typ[0] != '.' {
-		typ = "." + typ
-	}
-	return typ
-}
-
 /*
  * 没有设定过期时间，oriTime 为 0
  * 此函数将默认过期时间设为当前时间后半小时
@@ -156,6 +142,8 @@ func timeAssign(oriTime time.Time) time.Time {
 	}
 }
 
+// ===========sumdry=======================
+// 从config读取数据
 func GetSetting(key string) (value string) {
 	// config
 	viper.SetConfigName("config")
@@ -169,3 +157,34 @@ func GetSetting(key string) (value string) {
 	logInfo := viper.GetStringMapString("setting")
 	return logInfo[key]
 }
+
+/*
+ * 通过文件扩展名获取ContentType
+ */
+func GetFileContentType(fileType string) string {
+	var StrRet string = ""
+	switch fileType {
+	case ".txt":
+		StrRet = "text/plain"
+	case ".csv":
+		StrRet = "text/csv"
+	case ".tex":
+		StrRet = "application/x-tex"
+	case ".md":
+		StrRet = "text/x-markdown"
+	default:
+		StrRet = "text/plain"
+	}
+	return StrRet
+}
+
+// 格式化后缀，仿止出错
+func TypeComplement(typ string) string {
+	// 类型判断
+	if typ == "" {
+		typ = ".txt"
+	} else if typ[0] != '.' {
+		typ = "." + typ
+	}
+	return typ
+}

backend/app/controller/interface.go

@@ -74,6 +74,7 @@ func SendFile(c echo.Context) error {
 	stat := Autheticate(cookie, info.Url, info.Passwd, info.Time) // 包含创建链接Createlink
 	// response
 	switch stat {
+
 	case 0:
 		return response.SendResponse(c, http.StatusForbidden, cookieMsg+"error:密码错误", "") //403
 	case 1: // 鉴权通过

backend/model/init.go

@@ -25,7 +25,7 @@ func Init() {
 	if err != nil {
 		logrus.Fatal(err)
 	}
-	err = DB.AutoMigrate(&Sid{}, &Url{})
+	err = DB.AutoMigrate(&Rel{})
 	if err != nil {
 		logrus.Fatal(err)
 	}
@@ -97,36 +97,23 @@ func Checkt(p Content) bool {
 }
 
 // 这个函数那边有可能需要
-// 检查sid是否超
-func Checkt1(sid string) bool {
-	var s Sid
-	DB.First(&s, "S = ?", sid)
+// 检查sid_url是否超时
+// 同时也可以查询sid,_url是否关联
+func Find(sid string, _url string) bool {
+	var s Rel
+	err := DB.Where(&Rel{Sid: sid, Url: _url}).First(&s).Error
+	if err != nil {
+		return false
+	}
 	t := time.Now()
 	if t.After(s.Time) {
+		DB.Delete(&s)
 		return false
 	} else {
 		return true
 	}
 }
 
-// 查找sessionid和url是否关联
-func Find(sid string, _url string) bool {
-	var s Sid
-	err1 := DB.First(&s, "S = ?", sid).Error
-	//fmt.Println(url.Url1,url.ID,"\n")
-	if err1 != nil {
-		return false
-	}
-	var Urls []Url
-	DB.Model(&s).Association("Url1").Find(&Urls)
-	for _, value := range Urls {
-		if value.Url1 == _url {
-			return true
-		}
-	}
-	return false
-}
-
 // 通过url查询文件类型
 func Find1(_url string, key string) string {
 	var p Content
@@ -142,18 +129,6 @@ func Find1(_url string, key string) string {
 	return ""
 }
 
-//var letters = []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ123456789")
-
-/*//*随机生成字符串
-func randStr(n int) string {
-	rand.Seed(time.Now().Unix())
-    b := make([]rune, n)
-    for i := range b {
-        b[i] = letters[rand.Intn(len(letters))]
-    }
-    return string(b)
-}*/
-
 // 新建用户
 func Createuser(_User Users) {
 	//_User.SessionId=randStr(8)
@@ -162,7 +137,7 @@ func Createuser(_User Users) {
 
 //新建链接
 /* 返回值：
-   3 表示sid已经过期，需要重新分配
+   3 表示sid_url已经过期，需要重新分配
    2 表示内容过期
    1 表示密码正确
    0 表示密码不正确
@@ -175,23 +150,18 @@ func Createlink(sid string, _passwd string, _url string, _Time1 time.Time) uint
 			//fmt.Println(sid,_passwd,_url,_Time1)
 			return 2
 		}
-		var p1 Sid
-		err := DB.First(&p1, "S = ?", sid).Error
+		var p1 Rel
+		err := DB.Where(&Rel{Sid: sid, Url: _url}).First(&p1).Error
 		if err != nil {
-			sid1 := Sid{
-				S:    sid,
+			rel1 := Rel{
+				Sid:  sid,
+				Url:  _url,
 				Time: _Time1,
-				Url1: []Url{
-					{Url1: _url},
-				},
 			}
-			DB.Create(&sid1)
+			DB.Create(&rel1)
 		} else {
-			t := time.Now()
-			if t.After(p1.Time) {
-				return 3
-			}
-			DB.Model(&p1).Association("Url1").Append(&Url{Url1: _url})
+			p1.Time = _Time1
+			DB.Save(&p1)
 		}
 		return 1
 	}
@@ -200,20 +170,15 @@ func Createlink(sid string, _passwd string, _url string, _Time1 time.Time) uint
 
 // 第一次上传
 func CreatelinkFirstTime(sid string, _url string, _Time1 time.Time) {
-	sid1 := Sid{
-		S:    sid,
+	sid1 := Rel{
+		Sid:  sid,
+		Url:  _url,
 		Time: _Time1,
-		Url1: []Url{
-			{Url1: _url},
-		},
 	}
 	DB.Create(&sid1)
 }
 
 /*
 func Test() {
-	var p Content
-	var Url2 string = "1234"
-	DB.First(&p, "url1 = ?", Url2)
-	fmt.Println(Checkt(p))
+	fmt.Println(Find("ab","12345"))
 }*/

backend/model/model.go

@@ -19,17 +19,11 @@ type Content struct {
     Filename string
 }
 
-type Sid struct {
+type Rel struct {
     gorm.Model
-    Time time.Time
-    S string 
-    Url1 []Url  `gorm:"many2many:sid_urls;"` 
-}
-
-type Url struct
-{
-    gorm.Model
-    Url1 string
+    Sid string
+    Url string
+    Time time.Time 
 }
 
 type Users struct {

backend/readme.txt

@@ -13,11 +13,6 @@
     model：
         init：  数据库相关函数，包含着有关和数据库交互的各种操作，比如连接数据库，通过数据库反查内容，保存文本和文件信息等操作的函数
         model： 存储着各表的结构，一共有五个结构体
-    mysql:
-        pastebinDatabase: 建立数据库脚本
-        pastebinTable: 建立数据库的数据表脚本
-    files：（修改）
-        sys.log: 日志文件
     config.yaml: 修改设置
     go.mod
     go.sum