sqliteProvider.go

package datastore

import (
	"database/sql"
	"errors"
	"os"

	_ "github.com/mattn/go-sqlite3"
	gorp "gopkg.in/gorp.v2"
)

type sqliteProvider struct {
	sqlitePath string
}

func (p *sqliteProvider) Connect() error {
	rs := RdbStoreInstance()
	if rs.master() == nil {
		if p.sqlitePath == "" {
			return errors.New("not key sqlitePath")
		} else {
			db, err := sql.Open("sqlite3", p.sqlitePath)
			if err != nil {
				fatal(err)
			}
			var master *gorp.DbMap
			master = &gorp.DbMap{Db: db, Dialect: gorp.SqliteDialect{}}
			rs.setMaster(master)
		}
	}
	return nil
}

func (p *sqliteProvider) Init() {
	p.CreateApiStore()
	p.CreateUserStore()
	p.CreateBlockUserStore()
	p.CreateRoomStore()
	p.CreateRoomUserStore()
	p.CreateMessageStore()
	p.CreateDeviceStore()
	p.CreateSubscriptionStore()
}

func (p *sqliteProvider) DropDatabase() error {
	if err := os.Remove(p.sqlitePath); err != nil {
		return err
	}
	return nil
}