New structure (#7)

2018-10-31 12:42:38 +00:00
parent 3f9fad0e2a
commit 301a4eedda
104 changed files with 326 additions and 280 deletions
--- a/pkg/models/models.go
+++ b/pkg/models/models.go
@ -0,0 +1,92 @@
+package models
+
+import (
+	"fmt"
+	_ "github.com/go-sql-driver/mysql" // Because.
+	"github.com/go-xorm/core"
+	"github.com/go-xorm/xorm"
+	xrc "github.com/go-xorm/xorm-redis-cache"
+	_ "github.com/mattn/go-sqlite3" // Because.
+
+	"encoding/gob"
+	"github.com/spf13/viper"
+)
+
+var (
+	x *xorm.Engine
+
+	tables []interface{}
+)
+
+func getEngine() (*xorm.Engine, error) {
+	// Use Mysql if set
+	if viper.GetString("database.type") == "mysql" {
+		connStr := fmt.Sprintf(
+			"%s:%s@tcp(%s)/%s?charset=utf8&parseTime=true",
+			viper.GetString("database.user"),
+			viper.GetString("database.password"),
+			viper.GetString("database.host"),
+			viper.GetString("database.database"))
+		e, err := xorm.NewEngine("mysql", connStr)
+		e.SetMaxOpenConns(viper.GetInt("database.openconnections"))
+		return e, err
+	}
+
+	// Otherwise use sqlite
+	path := viper.GetString("database.path")
+	if path == "" {
+		path = "./db.db"
+	}
+	return xorm.NewEngine("sqlite3", path)
+}
+
+func init() {
+	tables = append(tables,
+		new(User),
+		new(List),
+		new(ListTask),
+		new(Team),
+		new(TeamMember),
+		new(TeamList),
+		new(TeamNamespace),
+		new(Namespace),
+		new(ListUser),
+		new(NamespaceUser),
+	)
+}
+
+// SetEngine sets the xorm.Engine
+func SetEngine() (err error) {
+	x, err = getEngine()
+	if err != nil {
+		return fmt.Errorf("Failed to connect to database: %v", err)
+	}
+
+	// Cache
+	if viper.GetBool("cache.enabled") {
+		switch viper.GetString("cache.type") {
+		case "memory":
+			cacher := xorm.NewLRUCacher(xorm.NewMemoryStore(), viper.GetInt("cache.maxelementsize"))
+			x.SetDefaultCacher(cacher)
+			break
+		case "redis":
+			cacher := xrc.NewRedisCacher(viper.GetString("cache.redishost"), viper.GetString("cache.redispassword"), xrc.DEFAULT_EXPIRATION, x.Logger())
+			x.SetDefaultCacher(cacher)
+			gob.Register(tables)
+			break
+		default:
+			fmt.Println("Did not find a valid cache type. Caching disabled. Please refer to the docs for poosible cache types.")
+		}
+	}
+
+	x.SetMapper(core.GonicMapper{})
+
+	// Sync dat shit
+	if err = x.StoreEngine("InnoDB").Sync2(tables...); err != nil {
+		return fmt.Errorf("sync database struct error: %v", err)
+	}
+
+	x.ShowSQL(viper.GetBool("database.showqueries"))
+
+	return nil
+}