basic storage package

This commit is contained in:
Nazar Kanaev
2020-06-21 21:58:51 +01:00
committed by nkanaev
parent 7a76027961
commit 6621f49770
7 changed files with 140 additions and 0 deletions

42
storage/folder.go Normal file
View File

@@ -0,0 +1,42 @@
package storage
import "fmt"
type Folder struct {
Id int64
Title string
IsExpanded bool
}
func (s *Storage) CreateFolder(title string) *Folder {
expanded := true
result, err := s.db.Exec(`
insert into folders (title, is_expanded) values (?, ?)`,
title, expanded,
)
if err != nil {
fmt.Println(err)
return nil
}
id, idErr := result.LastInsertId()
if idErr != nil {
return nil
}
return &Folder{Id: id, Title: title, IsExpanded: expanded}
}
func (s *Storage) DeleteFolder(folderId int64) bool {
_, err1 := s.db.Exec(`update feeds set folder_id = null where folder_id = ?`, folderId)
_, err2 := s.db.Exec(`delete from folders where id = ?`, folderId)
return err1 == nil && err2 == nil
}
func (s *Storage) RenameFolder(folderId int64, newTitle string) bool {
_, err := s.db.Exec(`update folders set title = ? where id = ?`, newTitle, folderId)
return err == nil
}
func (s *Storage) ToggleFolderExpanded(folderId int64, isExpanded bool) bool {
_, err := s.db.Exec(`update folders set is_expanded = ? where id = ?`, isExpanded, folderId)
return err == nil
}

60
storage/storage.go Normal file
View File

@@ -0,0 +1,60 @@
package storage
import (
"database/sql"
_ "github.com/mattn/go-sqlite3"
)
var initQuery string = `
create table if not exists folders (
id integer primary key autoincrement,
title text not null,
is_expanded boolean
);
create table if not exists feeds (
id integer primary key autoincrement,
folder_id references folders(id),
title text not null,
description text,
link text,
feed_link text,
icon text
);
create index if not exists idx_feed_folder_id on feeds(folder_id);
create table if not exists items (
id string primary key,
feed_id references feeds(id),
title text,
link text,
description text,
content text,
author text,
date integer,
date_updated integer,
status integer,
image text
);
create index if not exists idx_item_feed_id on items(feed_id);
create index if not exists idx_item_status on items(status);
`
type Storage struct {
db *sql.DB
}
func New() (*Storage, error) {
path := "./storage.db"
db, err := sql.Open("sqlite3", path)
if err != nil {
return nil, err
}
_, err = db.Exec(initQuery)
if err != nil {
return nil, err
}
return &Storage{db: db}, nil
}