localhost_frssoft/bloat
1
0
Fork 0
mirror of https://gitea.phreedom.club/localhost_frssoft/bloat.git synced 2025-05-01 09:34:21 +00:00
Code Issues Projects Releases Wiki Activity

Use filesystem based kv store instead of sqlite

Browse source
This commit is contained in:
r 2019-12-17 20:17:25 +00:00
parent 3b50f40c08
commit 59aad78f66
11 changed files with 217 additions and 88 deletions
Download patch file Download diff file Expand all files Collapse all files

37
repository/appRepository.go
View file

@ -1,54 +1,33 @@
package repository
import (
"database/sql"
"web/kv"
"web/model"
)
type appRepository struct {
db *sql.DB
db *kv.Database
}
func NewAppRepository(db *sql.DB) (*appRepository, error) {
_, err := db.Exec(`CREATE TABLE IF NOT EXISTS app
(instance_url varchar, client_id varchar, client_secret varchar)`,
)
if err != nil {
return nil, err
}
func NewAppRepository(db *kv.Database) *appRepository {
return &appRepository{
db: db,
}, nil
}
}
func (repo *appRepository) Add(a model.App) (err error) {
_, err = repo.db.Exec("INSERT INTO app VALUES (?, ?, ?)", a.InstanceURL, a.ClientID, a.ClientSecret)
err = repo.db.Set(a.InstanceDomain, a.Marshal())
return
}
func (repo *appRepository) Update(instanceURL string, clientID string, clientSecret string) (err error) {
_, err = repo.db.Exec("UPDATE app SET client_id = ?, client_secret = ? where instance_url = ?", clientID, clientSecret, instanceURL)
return
}
func (repo *appRepository) Get(instanceURL string) (a model.App, err error) {
rows, err := repo.db.Query("SELECT * FROM app WHERE instance_url = ?", instanceURL)
func (repo *appRepository) Get(instanceDomain string) (a model.App, err error) {
data, err := repo.db.Get(instanceDomain)
if err != nil {
return
}
defer rows.Close()
if !rows.Next() {
err = model.ErrAppNotFound
return
}
err = rows.Scan(&a.InstanceURL, &a.ClientID, &a.ClientSecret)
if err != nil {
return
}
err = a.Unmarshal(instanceDomain, data)
return
}

48
repository/sessionRepository.go
View file

@ -1,54 +1,50 @@
package repository
import (
"database/sql"
"web/kv"
"web/model"
)
type sessionRepository struct {
db *sql.DB
db *kv.Database
}
func NewSessionRepository(db *sql.DB) (*sessionRepository, error) {
_, err := db.Exec(`CREATE TABLE IF NOT EXISTS session
(id varchar, instance_url varchar, access_token varchar)`,
)
if err != nil {
return nil, err
}
func NewSessionRepository(db *kv.Database) *sessionRepository {
return &sessionRepository{
db: db,
}, nil
}
}
func (repo *sessionRepository) Add(s model.Session) (err error) {
_, err = repo.db.Exec("INSERT INTO session VALUES (?, ?, ?)", s.ID, s.InstanceURL, s.AccessToken)
err = repo.db.Set(s.ID, s.Marshal())
return
}
func (repo *sessionRepository) Update(sessionID string, accessToken string) (err error) {
_, err = repo.db.Exec("UPDATE session SET access_token = ? where id = ?", accessToken, sessionID)
return
}
func (repo *sessionRepository) Get(id string) (s model.Session, err error) {
rows, err := repo.db.Query("SELECT * FROM session WHERE id = ?", id)
func (repo *sessionRepository) Update(id string, accessToken string) (err error) {
data, err := repo.db.Get(id)
if err != nil {
return
}
defer rows.Close()
if !rows.Next() {
var s model.Session
err = s.Unmarshal(id, data)
if err != nil {
return
}
s.AccessToken = accessToken
return repo.db.Set(id, s.Marshal())
}
func (repo *sessionRepository) Get(id string) (s model.Session, err error) {
data, err := repo.db.Get(id)
if err != nil {
err = model.ErrSessionNotFound
return
}
err = rows.Scan(&s.ID, &s.InstanceURL, &s.AccessToken)
if err != nil {
return
}
err = s.Unmarshal(id, data)
return
}