gosuki/db.go

package main

import (
	"fmt"
	"gomark/database"
	"gomark/tools"
	"path/filepath"
)

type DB = database.DB

// Global cache database
var (
	CacheDB *DB // Main in memory db, is synced with disc
)

func initDB() {
	// Initialize memory db with schema
	cachePath := fmt.Sprintf(database.DBMemcacheFmt, database.DBCacheName)
	CacheDB = database.New(database.DBCacheName, cachePath)
	CacheDB.Init()

	// Check and initialize local db as last step
	// browser bookmarks should already be in cache

	dbdir := tools.GetDefaultDBPath()
	dbpath := filepath.Join(dbdir, database.DBFileName)
	// Verifiy that local db directory path is writeable
	err := tools.CheckWriteable(dbdir)
	if err != nil {
		log.Critical(err)
	}

	// If local db exists load it to cacheDB
	var exists bool
	if exists, err = tools.CheckFileExists(dbpath); exists {
		if err != nil {
			log.Warning(err)
		}
		log.Infof("<%s> exists, preloading to cache", dbpath)
		er := CacheDB.SyncFromDisk(dbpath)
		if er != nil {
			log.Critical(er)
		}
	} else {
		if err != nil {
			log.Error(err)
		}

		// Else initialize it
		initLocalDB(CacheDB, dbpath)
	}

}

//Initialize the local database file
func initLocalDB(db *DB, dbpath string) {

	log.Infof("Initializing local db at '%s'", dbpath)
	err := db.SyncToDisk(dbpath)
	if err != nil {
		log.Critical(err)
	}

}