db.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

package main

import (
	"github.com/jinzhu/gorm"
	"log"
	"os"
)

var Db, dberr = gorm.Open(os.Getenv("FREEMAIL_DB_DRIVER"), os.Getenv("FREEMAIL_DB_CREDENTIALS"))

func InitDB() {
	if dberr != nil {
		log.Panic(dberr)
	}
	logMode := os.Getenv("FREEMAIL_DB_LOG")
	if logMode == "true" {
		Db.LogMode(true)
	}
	if err := Db.DB().Ping(); err != nil {
		log.Panic(err)
	}

	vD := VirtualDomain{}
	Db.Debug().AutoMigrate(&vD)

	vU := VirtualUser{}
	Db.Debug().AutoMigrate(&vU)

	vA := VirtualAliase{}
	Db.Debug().AutoMigrate(&vA)

	Db.Model(&vU).AddUniqueIndex("idx_virtuser_email", "email")
	Db.Model(&vU).AddForeignKey("domain_id", "virtual_domains(id)", "CASCADE", "RESTRICT")
	/*
		Db.Model(&u).AddUniqueIndex("idx_user_name", "name")
		Db.Model(&u).AddUniqueIndex("idx_user_email", "email")
		Db.Model(&h).AddUniqueIndex("idx_host_url", "url")
		Db.Model(&c).AddUniqueIndex("idx_config_sticky", "sticky")
	*/
}