feat(data): 添加数据格式支持 更改数据写入读取

db/local.go

@@ -12,6 +12,7 @@ var localDB []Token
 type Token struct {
 	Token       string `json:"token"`
 	DedupObject string `json:"dedup_object"`
+	DataFormat  string `json:"data_format"`
 }
 
 func InitLocalDB() {
@@ -35,7 +36,7 @@ func ListToken() []Token {
 	return localDB
 }
 
-func CreateToken(token string, dedupObject string) error {
+func CreateToken(token string, dedupObject string, dataFormat string) error {
 	InitLocalDB()
 	for _, t := range localDB {
 		if t.Token == token {
@@ -43,7 +44,7 @@ func CreateToken(token string, dedupObject string) error {
 		}
 	}
 
-	a := append(localDB, Token{token, dedupObject})
+	a := append(localDB, Token{token, dedupObject, dataFormat})
 	marshal, err := json.Marshal(a)
 	if err != nil {
 		return err
@@ -55,12 +56,13 @@ func CreateToken(token string, dedupObject string) error {
 	return nil
 }
 
-func UpdateToken(token string, dedupObject string) error {
+func UpdateToken(token string, dedupObject string, dataFormat string) error {
 	InitLocalDB()
 
 	for i, t := range localDB {
 		if t.Token == token {
 			localDB[i].DedupObject = dedupObject
+			localDB[i].DataFormat = dataFormat
 
 			marshal, err := json.Marshal(localDB)
 			if err != nil {
@@ -72,6 +74,7 @@ func UpdateToken(token string, dedupObject string) error {
 			return nil
 		}
 	}
+
 	InitLocalDB()
 	return errors.New("token not found")
 }
@@ -93,6 +96,7 @@ func DeleteToken(token string) error {
 			return nil
 		}
 	}
+
 	InitLocalDB()
 	return errors.New("token not found")
 }
@@ -105,3 +109,12 @@ func GetDedupObject(token string) (dedupObject string, err error) {
 	}
 	return "", errors.New("未找到Token")
 }
+
+func GetDataFormat(token string) (dataFormat string, err error) {
+	for i, t := range localDB {
+		if t.Token == token {
+			return localDB[i].DataFormat, nil
+		}
+	}
+	return "", errors.New("未找到Token")
+}