From 0169188d2797837ea369d33fcbd87c43eb6c5fb5 Mon Sep 17 00:00:00 2001
From: Season Shi <seasonshi@elyart.dev>
Date: Thu, 23 Mar 2023 00:22:15 +0800
Subject: [PATCH] Change Token KV func to common func.

Add File based KV func
---
 internal/db/kv.go | 40 +++++++++++++++++++++++++++++++++++++---
 1 file changed, 37 insertions(+), 3 deletions(-)

diff --git a/internal/db/kv.go b/internal/db/kv.go
index 6d0e4cc8..01785109 100644
--- a/internal/db/kv.go
+++ b/internal/db/kv.go
@@ -15,6 +15,40 @@ func fileInit() *badger.DB {
 	return kv
 }
 
+func setFileKV(key string, value string) bool {
+	kv := fileInit()
+	err := kv.Update(func(txn *badger.Txn) error {
+		err := txn.Set([]byte(key), []byte(value))
+		return err
+	})
+	if err != nil {
+		log.Fatal(err)
+		return false
+	}
+	return true
+}
+
+func getFileKV(key string) string {
+	kv := fileInit()
+	var valCopy []byte
+	err := kv.View(func(txn *badger.Txn) error {
+		item, err := txn.Get([]byte(key))
+		if err != nil {
+			return err
+		}
+		err = item.Value(func(val []byte) error {
+			valCopy = append([]byte{}, val...)
+			return nil
+		})
+		return err
+	})
+	if err != nil {
+		log.Fatal(err)
+		return ""
+	}
+	return string(valCopy)
+}
+
 func memInit() *badger.DB {
 	opt := badger.DefaultOptions("").WithInMemory(true)
 	kv, err := badger.Open(opt)
@@ -25,10 +59,10 @@ func memInit() *badger.DB {
 	return kv
 }
 
-func setTokenKV(key string, token string) bool {
+func setMemKV(key string, value string) bool {
 	kv := memInit()
 	err := kv.Update(func(txn *badger.Txn) error {
-		err := txn.Set([]byte(key), []byte(token))
+		err := txn.Set([]byte(key), []byte(value))
 		return err
 	})
 	if err != nil {
@@ -38,7 +72,7 @@ func setTokenKV(key string, token string) bool {
 	return true
 }
 
-func getTokenKV(key string) string {
+func getMemKV(key string) string {
 	kv := memInit()
 	var valCopy []byte
 	err := kv.View(func(txn *badger.Txn) error {