122

backend/model/init.go

@@ -63,7 +63,7 @@ func Findlink(_url string) string {
 
 // save text information
 // 保存文本和文件信息
-func Savetext(_S string, _Time uint, _Passwd string, _Time1 time.Time, _url string, _Filetype string, _Filename string) {
+func Savetext(_S string, _Time uint, _Passwd string, _Time1 time.Time, _url string, _Filetype string, _Filename string,_Hlight bool) {
 	var id1 int64
 	DB.Model(&Content{}).Count(&id1)
 	id1++
@@ -78,6 +78,7 @@ func Savetext(_S string, _Time uint, _Passwd string, _Time1 time.Time, _url stri
 		Url1:     _url,
 		Filetype: _Filetype,
 		Filename: _Filename,
+		Hlight: _Hlight,
 	}
 	DB.Create(&p)
 }
@@ -98,12 +99,12 @@ func Checkt(p Content) bool {
 
 // 这个函数那边有可能需要
 // 检查sid_url是否超时
-// 同时也可以查询sid,_url是否关联
-func Find(sid string, _url string) bool {
+//同时也可以查询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
+	err := DB.Where(&Rel{Sid: sid,Url: _url}).First(&s).Error
+	if err != nil  {
+		return false 
 	}
 	t := time.Now()
 	if t.After(s.Time) {
@@ -129,6 +130,25 @@ func Find1(_url string, key string) string {
 	return ""
 }
 
+// 通过url查询文件是否高亮
+func Find2(_url string) bool {
+	var p Content
+	DB.First(&p, "Url1 = ?", _url)
+	return p.Hlight
+}
+
+//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)
@@ -144,19 +164,19 @@ func Createuser(_User Users) {
 */
 func Createlink(sid string, _passwd string, _url string, _Time1 time.Time) uint {
 	var p Content
-	DB.First(&p, "Url1 = ?", _url)
+	DB.First(&p,"Url1 = ?", _url)
 	if p.Passwd == _passwd {
 		if Checkt(p) {
 			//fmt.Println(sid,_passwd,_url,_Time1)
 			return 2
 		}
 		var p1 Rel
-		err := DB.Where(&Rel{Sid: sid, Url: _url}).First(&p1).Error
+		err := DB.Where(&Rel{Sid: sid,Url: _url}).First(&p1).Error
 		if err != nil {
 			rel1 := Rel{
-				Sid:  sid,
-				Url:  _url,
-				Time: _Time1,
+				Sid:    sid,
+				Url:    _url,
+				Time:   _Time1,
 			}
 			DB.Create(&rel1)
 		} else {
@@ -170,11 +190,11 @@ func Createlink(sid string, _passwd string, _url string, _Time1 time.Time) uint
 
 // 第一次上传
 func CreatelinkFirstTime(sid string, _url string, _Time1 time.Time) {
-	sid1 := Rel{
+    sid1 := Rel{
 		Sid:  sid,
 		Url:  _url,
 		Time: _Time1,
-	}
+		}
 	DB.Create(&sid1)
 }
 

backend/model/model.go

@@ -17,6 +17,7 @@ type Content struct {
     Url1 string
     Filetype string
     Filename string
+    Hlight bool
 }
 
 type Rel struct {