diff options
Diffstat (limited to 'internal')
| -rw-r--r-- | internal/dbmigrations.go | 2 | ||||
| -rw-r--r-- | internal/models/feed.go | 14 |
2 files changed, 11 insertions, 5 deletions
diff --git a/internal/dbmigrations.go b/internal/dbmigrations.go index 7d520ba..fcbaaaf 100644 --- a/internal/dbmigrations.go +++ b/internal/dbmigrations.go @@ -42,7 +42,7 @@ func Migrate(db *sql.DB, webmaster string, passOne string, passTwo string) { _, table_check = db.Query("SELECT * FROM feeds;") if table_check != nil { - _, err := db.Exec("CREATE TABLE feeds (id SERIAL PRIMARY KEY, name VARCHAR(50) NOT NULL, user_id INTEGER REFERENCES logins(id), url VARCHAR(255) NOT NULL);") + _, err := db.Exec("CREATE TABLE feeds (id SERIAL PRIMARY KEY, name VARCHAR(50) NOT NULL, user_id INTEGER REFERENCES logins(id), url VARCHAR(255) NOT NULL, category INTEGER);") if err != nil { fmt.Fprintf(os.Stderr, "Unable to create feeds table: %v\n", err) os.Exit(1) diff --git a/internal/models/feed.go b/internal/models/feed.go index 9c68adb..289b1e6 100644 --- a/internal/models/feed.go +++ b/internal/models/feed.go @@ -1,8 +1,14 @@ package models +const ( + TextFeed = iota + YoutubeFeed +) + type Feed struct { - Id int - Name string - User string - Url string + Id int + Name string + User string + Url string + Category int } |