GithubProxy/coai
1
0
Fork 0
mirror of https://github.com/coaidev/coai.git synced 2025-05-19 04:50:14 +09:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat alpha: support sqlite engine

Browse Source
This commit is contained in:
Zhang Minghan 2024-03-08 23:46:49 +08:00
parent 52419e1b2b
commit 4f0390815c
5 changed files with 16 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
addition/article/utils.go
View File

@ -31,7 +31,7 @@ func GenerateDocxFile(target, title, content string) error {
func CreateArticleFile(hash, title, content string) string {
target := fmt.Sprintf("addition/article/data/%s/%s.docx", hash, title)
utils.CreateFolderOnFile(target)
utils.FileDirSafe(target)
if err := GenerateDocxFile(target, title, content); err != nil {
globals.Debug(fmt.Sprintf("[article] error during generate article %s: %s", title, err.Error()))
}

2
app/src/conf/env.ts
View File

@ -116,7 +116,7 @@ export function setAnnouncement(announcement: string): void {
*/
if (!announcement || announcement.trim() === "") return;
const firstReceived = getMemory("announcement") !== announcement;
const firstReceived = getMemory("announcement").trim() !== announcement.trim();
setMemory("announcement", announcement);
announcementEvent.emit({

4
connection/database.go
View File

@ -23,8 +23,8 @@ func InitMySQLSafe() *sql.DB {
func getConn() *sql.DB {
if viper.GetString("mysql.host") == "" {
globals.SqliteEngine = true
globals.Warn("[connection] mysql host is not set, using sqlite (chatnio.db)")
db, err := sql.Open("sqlite3", "chatnio.db")
globals.Warn("[connection] mysql host is not set, using sqlite (~/db/chatnio.db)")
db, err := sql.Open("sqlite3", utils.FileSafe("./db/chatnio.db"))
if err != nil {
panic(err)
}

4
utils/compress.go
View File

@ -57,7 +57,7 @@ func handlePath(path string) string {
}
func CreateZipObject(output string, files []string, replacer string) error {
CreateFolderOnFile(output)
FileDirSafe(output)
file, err := os.Create(output)
if err != nil {
return err
@ -107,7 +107,7 @@ func addFileToZip(zipWriter *zip.Writer, path string, replacer string) error {
}
func CreateGzipObject(output string, files []string, replacer string) error {
CreateFolderOnFile(output)
FileDirSafe(output)
tarFile, err := os.Create(output)
if err != nil {
return err

15
utils/fs.go
View File

@ -23,22 +23,27 @@ func Exists(path string) bool {
return err != nil && os.IsExist(err)
}
func CreateFolderNotExists(path string) string {
func DirSafe(path string) string {
CreateFolder(path)
return path
}
func CreateFolderOnFile(file string) string {
func FileDirSafe(file string) string {
if strings.LastIndex(file, "/") == -1 {
return file
}
return CreateFolderNotExists(file[:strings.LastIndex(file, "/")])
return DirSafe(file[:strings.LastIndex(file, "/")])
}
func FileSafe(file string) string {
FileDirSafe(file)
return file
}
func WriteFile(path string, data string, folderSafe bool) error {
if folderSafe {
CreateFolderOnFile(path)
FileDirSafe(path)
}
file, err := os.Create(path)
@ -152,7 +157,7 @@ func CopyFile(src string, dst string) error {
}
}(in)
CreateFolderOnFile(dst)
FileDirSafe(dst)
out, err := os.Create(dst)
if err != nil {
return err