aboutsummaryrefslogtreecommitdiffhomepage
path: root/storage
diff options
context:
space:
mode:
authorGravatar Frédéric Guillot <fred@miniflux.net>2017-12-02 17:04:01 -0800
committerGravatar Frédéric Guillot <fred@miniflux.net>2017-12-02 17:04:01 -0800
commit2f1367a8d4c33e7c6ba459cfc6756e079c7a1af4 (patch)
tree04a2c15d41f79c05b74b67c9bbab4bfb483be6e1 /storage
parent453ff64f29c5ca330b20fb9c3dd41bbf7e7b7396 (diff)
Make entries sorting configurable
Diffstat (limited to 'storage')
-rw-r--r--storage/entry_query_builder.go16
-rw-r--r--storage/migration.go2
-rw-r--r--storage/user.go108
3 files changed, 95 insertions, 31 deletions
diff --git a/storage/entry_query_builder.go b/storage/entry_query_builder.go
index 2c2f270..6f7f4bd 100644
--- a/storage/entry_query_builder.go
+++ b/storage/entry_query_builder.go
@@ -27,15 +27,6 @@ type EntryQueryBuilder struct {
limit int
offset int
entryID int64
- conditions []string
- args []interface{}
-}
-
-// WithCondition defines a new condition.
-func (e *EntryQueryBuilder) WithCondition(column, operator string, value interface{}) *EntryQueryBuilder {
- e.args = append(e.args, value)
- e.conditions = append(e.conditions, fmt.Sprintf("%s %s $%d", column, operator, len(e.args)+1))
- return e
}
// WithEntryID set the entryID.
@@ -187,7 +178,7 @@ func (e *EntryQueryBuilder) GetEntries() (model.Entries, error) {
)
if err != nil {
- return nil, fmt.Errorf("Unable to fetch entry row: %v", err)
+ return nil, fmt.Errorf("unable to fetch entry row: %v", err)
}
if iconID == nil {
@@ -208,11 +199,6 @@ func (e *EntryQueryBuilder) buildCondition() ([]interface{}, string) {
args := []interface{}{e.userID}
conditions := []string{"e.user_id = $1"}
- if len(e.conditions) > 0 {
- conditions = append(conditions, e.conditions...)
- args = append(args, e.args...)
- }
-
if e.categoryID != 0 {
conditions = append(conditions, fmt.Sprintf("f.category_id=$%d", len(args)+1))
args = append(args, e.categoryID)
diff --git a/storage/migration.go b/storage/migration.go
index 5060a34..9cfde8b 100644
--- a/storage/migration.go
+++ b/storage/migration.go
@@ -12,7 +12,7 @@ import (
"github.com/miniflux/miniflux2/sql"
)
-const schemaVersion = 3
+const schemaVersion = 4
// Migrate run database migrations.
func (s *Storage) Migrate() {
diff --git a/storage/user.go b/storage/user.go
index fdbbfda..7d296a9 100644
--- a/storage/user.go
+++ b/storage/user.go
@@ -8,6 +8,7 @@ import (
"database/sql"
"errors"
"fmt"
+ "log"
"strings"
"time"
@@ -110,23 +111,59 @@ func (s *Storage) RemoveExtraField(userID int64, field string) error {
// UpdateUser updates a user.
func (s *Storage) UpdateUser(user *model.User) error {
- defer helper.ExecutionTime(time.Now(), fmt.Sprintf("[Storage:UpdateUser] username=%s", user.Username))
- user.Username = strings.ToLower(user.Username)
-
+ defer helper.ExecutionTime(time.Now(), fmt.Sprintf("[Storage:UpdateUser] userID=%d", user.ID))
+ log.Println(user.EntryDirection)
if user.Password != "" {
hashedPassword, err := hashPassword(user.Password)
if err != nil {
return err
}
- query := "UPDATE users SET username=$1, password=$2, is_admin=$3, theme=$4, language=$5, timezone=$6 WHERE id=$7"
- _, err = s.db.Exec(query, user.Username, hashedPassword, user.IsAdmin, user.Theme, user.Language, user.Timezone, user.ID)
+ query := `UPDATE users SET
+ username=LOWER($1),
+ password=$2,
+ is_admin=$3,
+ theme=$4,
+ language=$5,
+ timezone=$6,
+ entry_direction=$7
+ WHERE id=$8`
+
+ _, err = s.db.Exec(
+ query,
+ user.Username,
+ hashedPassword,
+ user.IsAdmin,
+ user.Theme,
+ user.Language,
+ user.Timezone,
+ user.EntryDirection,
+ user.ID,
+ )
if err != nil {
return fmt.Errorf("unable to update user: %v", err)
}
} else {
- query := "UPDATE users SET username=$1, is_admin=$2, theme=$3, language=$4, timezone=$5 WHERE id=$6"
- _, err := s.db.Exec(query, user.Username, user.IsAdmin, user.Theme, user.Language, user.Timezone, user.ID)
+ query := `UPDATE users SET
+ username=$1,
+ is_admin=$2,
+ theme=$3,
+ language=$4,
+ timezone=$5,
+ entry_direction=$6
+ WHERE id=$7`
+
+ _, err := s.db.Exec(
+ query,
+ user.Username,
+ user.IsAdmin,
+ user.Theme,
+ user.Language,
+ user.Timezone,
+ user.EntryDirection,
+ user.ID,
+ )
+
if err != nil {
return fmt.Errorf("unable to update user: %v", err)
}
@@ -138,11 +175,24 @@ func (s *Storage) UpdateUser(user *model.User) error {
// UserByID finds a user by the ID.
func (s *Storage) UserByID(userID int64) (*model.User, error) {
defer helper.ExecutionTime(time.Now(), fmt.Sprintf("[Storage:UserByID] userID=%d", userID))
+ query := `SELECT
+ id, username, is_admin, theme, language, timezone, entry_direction, extra
+ FROM users
+ WHERE id = $1`
var user model.User
var extra hstore.Hstore
- row := s.db.QueryRow("SELECT id, username, is_admin, theme, language, timezone, extra FROM users WHERE id = $1", userID)
- err := row.Scan(&user.ID, &user.Username, &user.IsAdmin, &user.Theme, &user.Language, &user.Timezone, &extra)
+ row := s.db.QueryRow(query, userID)
+ err := row.Scan(
+ &user.ID,
+ &user.Username,
+ &user.IsAdmin,
+ &user.Theme,
+ &user.Language,
+ &user.Timezone,
+ &user.EntryDirection,
+ &extra,
+ )
if err == sql.ErrNoRows {
return nil, nil
} else if err != nil {
@@ -162,10 +212,22 @@ func (s *Storage) UserByID(userID int64) (*model.User, error) {
// UserByUsername finds a user by the username.
func (s *Storage) UserByUsername(username string) (*model.User, error) {
defer helper.ExecutionTime(time.Now(), fmt.Sprintf("[Storage:UserByUsername] username=%s", username))
+ query := `SELECT
+ id, username, is_admin, theme, language, timezone, entry_direction
+ FROM users
+ WHERE username=LOWER($1)`
var user model.User
- row := s.db.QueryRow("SELECT id, username, is_admin, theme, language, timezone FROM users WHERE username=$1", username)
- err := row.Scan(&user.ID, &user.Username, &user.IsAdmin, &user.Theme, &user.Language, &user.Timezone)
+ row := s.db.QueryRow(query, username)
+ err := row.Scan(
+ &user.ID,
+ &user.Username,
+ &user.IsAdmin,
+ &user.Theme,
+ &user.Language,
+ &user.Timezone,
+ &user.EntryDirection,
+ )
if err == sql.ErrNoRows {
return nil, nil
} else if err != nil {
@@ -178,10 +240,22 @@ func (s *Storage) UserByUsername(username string) (*model.User, error) {
// UserByExtraField finds a user by an extra field value.
func (s *Storage) UserByExtraField(field, value string) (*model.User, error) {
defer helper.ExecutionTime(time.Now(), fmt.Sprintf("[Storage:UserByExtraField] field=%s", field))
+ query := `SELECT
+ id, username, is_admin, theme, language, timezone, entry_direction
+ FROM users
+ WHERE extra->$1=$2`
+
var user model.User
- query := `SELECT id, username, is_admin, theme, language, timezone FROM users WHERE extra->$1=$2`
row := s.db.QueryRow(query, field, value)
- err := row.Scan(&user.ID, &user.Username, &user.IsAdmin, &user.Theme, &user.Language, &user.Timezone)
+ err := row.Scan(
+ &user.ID,
+ &user.Username,
+ &user.IsAdmin,
+ &user.Theme,
+ &user.Language,
+ &user.Timezone,
+ &user.EntryDirection,
+ )
if err == sql.ErrNoRows {
return nil, nil
} else if err != nil {
@@ -215,14 +289,18 @@ func (s *Storage) RemoveUser(userID int64) error {
// Users returns all users.
func (s *Storage) Users() (model.Users, error) {
defer helper.ExecutionTime(time.Now(), "[Storage:Users]")
+ query := `SELECT
+ id, username, is_admin, theme, language, timezone, last_login_at
+ FROM users
+ ORDER BY username ASC`
- var users model.Users
- rows, err := s.db.Query("SELECT id, username, is_admin, theme, language, timezone, last_login_at FROM users ORDER BY username ASC")
+ rows, err := s.db.Query(query)
if err != nil {
return nil, fmt.Errorf("unable to fetch users: %v", err)
}
defer rows.Close()
+ var users model.Users
for rows.Next() {
var user model.User
err := rows.Scan(