context.go

package common

import (
	"github.com/jinzhu/gorm"
	"github.com/op/go-logging"
)

type Context interface {
	GetAppRoot() string
	GetLogger() *logging.Logger
	GetCoreDB() *gorm.DB
	GetPriceDB() *gorm.DB
	GetUser() UserContext
	SetUser(user UserContext)
	GetDebug() bool
	GetSSL() bool
	GetIPC() string
	GetKeystore() string
	GetEthereumMode() string
	Close()
}

type Ctx struct {
	AppRoot      string
	Logger       *logging.Logger
	CoreDB       *gorm.DB
	PriceDB      *gorm.DB
	User         UserContext
	Debug        bool
	SSL          bool
	IPC          string
	Keystore     string
	EthereumMode string
	Context
}

func (c *Ctx) GetAppRoot() string {
	return c.AppRoot
}

func (c *Ctx) GetLogger() *logging.Logger {
	return c.Logger
}

func (c *Ctx) GetCoreDB() *gorm.DB {
	return c.CoreDB
}

func (c *Ctx) GetPriceDB() *gorm.DB {
	return c.PriceDB
}

func (c *Ctx) GetUser() UserContext {
	return c.User
}

func (c *Ctx) SetUser(user UserContext) {
	c.User = user
}

func (c *Ctx) GetDebug() bool {
	return c.Debug
}

func (c *Ctx) GetSSL() bool {
	return c.SSL
}

func (c *Ctx) GetIPC() string {
	return c.IPC
}

func (c *Ctx) GetKeystore() string {
	return c.Keystore
}

func (c *Ctx) GetEthereumMode() string {
	return c.EthereumMode
}

func (c *Ctx) Close() {
	c.GetLogger().Debugf("[common.Context] Closing context")
	c.GetCoreDB().Close()
	c.GetPriceDB().Close()
}