コード例 #1
0
ファイル: dbManager.go プロジェクト: postfix/wavepipe
// dbManager manages database connections, and communicates back and forth with the manager goroutine
func dbManager(conf config.Config, dbLaunchChan chan struct{}, dbKillChan chan struct{}) {
	log.Println("db: starting...")

	// Attempt to open database connection, depending on configuration
	// sqlite
	if conf.Sqlite != nil {
		log.Println("db: sqlite:", conf.Sqlite.File)

		// Replace the home character to set path
		path := common.ExpandHomeDir(conf.Sqlite.File)

		// Set DSN
		data.DB = new(data.SqliteBackend)
		data.DB.DSN(path)

		// Set up the database
		if err := data.DB.Setup(); err != nil {
			log.Fatalf("db: could not set up database: %s", err.Error())
		}

		// Verify database file exists and is ready
		if _, err := os.Stat(path); err != nil {
			log.Fatalf("db: database file does not exist: %s", conf.Sqlite.File)
		}

		// Open the database connection
		if err := data.DB.Open(); err != nil {
			log.Fatalf("db: could not open database: %s", err)
		}

		// TODO: temporary, create a test user
		data.NewUser("test", "test", data.RoleAdmin)
	} else {
		// Invalid config
		log.Fatalf("db: invalid database selected")
	}

	// Database set up, trigger manager that it's ready
	close(dbLaunchChan)

	// Trigger events via channel
	for {
		select {
		// Stop database manager
		case <-dbKillChan:
			// Close the database connection pool
			if err := data.DB.Close(); err != nil {
				log.Fatalf("db: could not close connection")
			}

			// Inform manager that shutdown is complete
			log.Println("db: stopped!")
			dbKillChan <- struct{}{}
			return
		}
	}
}
コード例 #2
0
ファイル: db_sqlite.go プロジェクト: postfix/wavepipe
// DSN sets the Path for use with sqlite3
func (s *SqliteBackend) DSN(path string) {
	// Replace the home character to set path
	s.Path = common.ExpandHomeDir(path)
}
コード例 #3
0
ファイル: config.go プロジェクト: postfix/wavepipe
// Media returns the media folder from config, but with special
// characters such as '~' replaced, and any trailing slashes trimmed.
func (c Config) Media() string {
	// Return path with strings replaced, trailing slash removed,
	// tilde replaced with current user's home directory
	return path.Clean(common.ExpandHomeDir(c.MediaFolder))
}