ETRI_SMARTFARM_POC/model/interface.go

package model

import (
	"io"
)

// func newDBHandler(dbtype, path string) (*gorm.DB, error) {
// 	if dbtype == "sqlite" {
// 		return gorm.Open(sqlite.Open("./test.db"), &gorm.Config{})
// 	} else {
// 		dsn := "host=localhost user=user password=user_password dbname=godopudb port=5432 sslmode=disable TimeZone=Asia/Seoul"
// 		return gorm.Open(postgres.Open(dsn), &gorm.Config{})
// 	}
// }

type DBHandler interface {
	GetDevices() ([]*Device, int, error)
	AddDevice(d *Device) error
	AddController(r io.Reader) (*Controller, error)
	GetControllers() ([]*Controller, error)
	IsExistController(cid string) bool
	GetServices() ([]*Service, error)
	AddService(name string) error
	UpdateService(name, addr string) (*Service, error)
	GetAddr(sid string) (string, error)
}

func NewDBHandler(dbtype, path string) (DBHandler, error) {
	return newSqliteHandler(path)
	// return newPostgresqlHandler(path)
}