diff --git a/pkg/initialize/init.go b/pkg/initialize/init.go index b817de434d4..2e820672e48 100644 --- a/pkg/initialize/init.go +++ b/pkg/initialize/init.go @@ -29,6 +29,7 @@ import ( "code.vikunja.io/api/pkg/models" "code.vikunja.io/api/pkg/modules/keyvalue" migrator "code.vikunja.io/api/pkg/modules/migration" + "code.vikunja.io/api/pkg/notifications" "code.vikunja.io/api/pkg/red" "code.vikunja.io/api/pkg/user" ) @@ -66,6 +67,10 @@ func InitEngines() { if err != nil { log.Fatal(err.Error()) } + err = notifications.InitDB() + if err != nil { + log.Fatal(err.Error()) + } } // FullInit initializes all kinds of things in the right order diff --git a/pkg/migration/migration.go b/pkg/migration/migration.go index 8102d1f5000..7f77f45360d 100644 --- a/pkg/migration/migration.go +++ b/pkg/migration/migration.go @@ -26,6 +26,7 @@ import ( "code.vikunja.io/api/pkg/log" "code.vikunja.io/api/pkg/models" "code.vikunja.io/api/pkg/modules/migration" + "code.vikunja.io/api/pkg/notifications" "code.vikunja.io/api/pkg/user" "github.com/olekukonko/tablewriter" "src.techknowlogick.com/xormigrate" @@ -163,5 +164,6 @@ func initSchema(tx *xorm.Engine) error { schemeBeans = append(schemeBeans, files.GetTables()...) schemeBeans = append(schemeBeans, migration.GetTables()...) schemeBeans = append(schemeBeans, user.GetTables()...) + schemeBeans = append(schemeBeans, notifications.GetTables()...) return tx.Sync2(schemeBeans...) } diff --git a/pkg/notifications/db.go b/pkg/notifications/db.go new file mode 100644 index 00000000000..2e7f6b0ce79 --- /dev/null +++ b/pkg/notifications/db.go @@ -0,0 +1,39 @@ +// Vikunja is a to-do list application to facilitate your life. +// Copyright 2018-2021 Vikunja and contributors. All rights reserved. +// +// This program is free software: you can redistribute it and/or modify +// it under the terms of the GNU Affero General Public Licensee as published by +// the Free Software Foundation, either version 3 of the License, or +// (at your option) any later version. +// +// This program is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +// GNU Affero General Public Licensee for more details. +// +// You should have received a copy of the GNU Affero General Public Licensee +// along with this program. If not, see . + +package notifications + +import ( + "code.vikunja.io/api/pkg/config" + "code.vikunja.io/api/pkg/db" +) + +// InitDB sets up the database connection to use in this module +func InitDB() (err error) { + // Cache + if config.CacheEnabled.GetBool() && config.CacheType.GetString() == "redis" { + db.RegisterTableStructsForCache(GetTables()) + } + + return nil +} + +// GetTables returns all structs which are also a table. +func GetTables() []interface{} { + return []interface{}{ + &DatabaseNotification{}, + } +}