diff options
author | r <r@freesoftwareextremist.com> | 2019-12-17 20:17:25 +0000 |
---|---|---|
committer | r <r@freesoftwareextremist.com> | 2019-12-17 20:17:25 +0000 |
commit | 59aad78f66cf58be7f88f2c0675f94a858163560 (patch) | |
tree | 95f00d18a8c847cdf4ee587e412de26960770a28 /repository/sessionRepository.go | |
parent | 3b50f40c081c499c2ecb8913b54186c622561d76 (diff) | |
download | bloat-59aad78f66cf58be7f88f2c0675f94a858163560.tar.gz bloat-59aad78f66cf58be7f88f2c0675f94a858163560.zip |
Use filesystem based kv store instead of sqlite
Diffstat (limited to 'repository/sessionRepository.go')
-rw-r--r-- | repository/sessionRepository.go | 44 |
1 files changed, 20 insertions, 24 deletions
diff --git a/repository/sessionRepository.go b/repository/sessionRepository.go index 2a88b40..6c26313 100644 --- a/repository/sessionRepository.go +++ b/repository/sessionRepository.go @@ -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() { - err = model.ErrSessionNotFound + var s model.Session + err = s.Unmarshal(id, data) + if err != nil { return } - err = rows.Scan(&s.ID, &s.InstanceURL, &s.AccessToken) + 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 = s.Unmarshal(id, data) + return } |