summaryrefslogtreecommitdiff
path: root/src/helsinki.at/rhimport
diff options
context:
space:
mode:
authorChristian Pointner <equinox@helsinki.at>2015-12-07 16:16:40 (GMT)
committerChristian Pointner <equinox@helsinki.at>2015-12-08 00:34:53 (GMT)
commita1b290f026011276f9c16f6af390a2aceb2f9ccc (patch)
tree2f7be9167b918b16c82d3f71f057a610f6ca749e /src/helsinki.at/rhimport
parent6e5d14f4a2189dbc0f0404c398d11878e356ecd7 (diff)
moved DB into seperate module
Diffstat (limited to 'src/helsinki.at/rhimport')
-rw-r--r--src/helsinki.at/rhimport/conf.go108
-rw-r--r--src/helsinki.at/rhimport/fetcher.go4
-rw-r--r--src/helsinki.at/rhimport/importer.go10
-rw-r--r--src/helsinki.at/rhimport/rddb.go131
4 files changed, 146 insertions, 107 deletions
diff --git a/src/helsinki.at/rhimport/conf.go b/src/helsinki.at/rhimport/conf.go
index 8e373c0..7288a89 100644
--- a/src/helsinki.at/rhimport/conf.go
+++ b/src/helsinki.at/rhimport/conf.go
@@ -25,37 +25,17 @@
package rhimport
import (
- "database/sql"
- "fmt"
- _ "github.com/go-sql-driver/mysql"
"github.com/vaughan0/go-ini"
)
-type getPasswordResult struct {
- password string
- err error
-}
-
-type getPasswordRequest struct {
- username string
- cached bool
- response chan getPasswordResult
-}
-
type Config struct {
- configfile string
- RDXportEndpoint string
- TempDir string
- db_host string
- db_user string
- db_passwd string
- db_db string
- dbh *sql.DB
- password_cache map[string]string
- getPasswordChan chan getPasswordRequest
- dbGetPasswordStmt *sql.Stmt
- quit chan bool
- done chan bool
+ configfile string
+ RDXportEndpoint string
+ TempDir string
+ db_host string
+ db_user string
+ db_passwd string
+ db_db string
}
func get_ini_value(file ini.File, section string, key string, dflt string) string {
@@ -79,87 +59,13 @@ func (self *Config) read_config_file() error {
return nil
}
-func (self *Config) init_database() (err error) {
- dsn := fmt.Sprintf("%s:%s@tcp(%s:3306)/%s", self.db_user, self.db_passwd, self.db_host, self.db_db)
- if self.dbh, err = sql.Open("mysql", dsn); err != nil {
- return
- }
- if self.dbGetPasswordStmt, err = self.dbh.Prepare("select PASSWORD from USERS where LOGIN_NAME = ?"); err != nil {
- return
- }
-
- return
-}
-
-func (self *Config) getPassword(username string, cached bool) (pwd string, err error) {
-
- if cached {
- pwd = self.password_cache[username]
- }
-
- if pwd == "" {
- err = self.dbGetPasswordStmt.QueryRow(username).Scan(&pwd)
- if err != nil {
- if err == sql.ErrNoRows {
- err = fmt.Errorf("user '%s' not known by rivendell", username)
- }
- return
- }
- self.password_cache[username] = pwd
- }
-
- return
-}
-
-func (self *Config) dispatchRequests() {
- defer func() { self.done <- true }()
- for {
- select {
- case <-self.quit:
- return
- case req := <-self.getPasswordChan:
- if req.cached {
- rhdl.Println("Config: got getPassword request for", req.username, "(cached)")
- } else {
- rhdl.Println("Config: got getPassword request for", req.username, "(not cached)")
- }
- pwd, err := self.getPassword(req.username, req.cached)
- req.response <- getPasswordResult{pwd, err}
- }
- }
-}
-
-func (self *Config) Cleanup() {
- self.quit <- true
- <-self.done
- close(self.quit)
- close(self.done)
- close(self.getPasswordChan)
- if self.dbh != nil {
- self.dbh.Close()
- }
- if self.dbGetPasswordStmt != nil {
- self.dbGetPasswordStmt.Close()
- }
-}
-
func NewConfig(configfile, rdxport_endpoint, temp_dir *string) (conf *Config, err error) {
conf = new(Config)
conf.configfile = *configfile
if err = conf.read_config_file(); err != nil {
return
}
- conf.quit = make(chan bool)
- conf.done = make(chan bool)
conf.RDXportEndpoint = *rdxport_endpoint
conf.TempDir = *temp_dir
- conf.password_cache = make(map[string]string)
- conf.getPasswordChan = make(chan getPasswordRequest)
-
- if err = conf.init_database(); err != nil {
- return
- }
-
- go conf.dispatchRequests()
return
}
diff --git a/src/helsinki.at/rhimport/fetcher.go b/src/helsinki.at/rhimport/fetcher.go
index ead9f5f..1a5801b 100644
--- a/src/helsinki.at/rhimport/fetcher.go
+++ b/src/helsinki.at/rhimport/fetcher.go
@@ -26,9 +26,9 @@ package rhimport
func FetchFile(ctx *ImportContext) (err error) {
- // TODO: fetch file from ctx.SourceUri and put it into ctx.Conf.TempDir
+ // TODO: fetch file from ctx.SourceUri and put it into ctx.Config.TempDir
- ctx.SourceFile = ctx.Conf.TempDir + "/source-file.ogg"
+ ctx.SourceFile = ctx.Config.TempDir + "/source-file.ogg"
ctx.DeleteSourceFile = true
return
}
diff --git a/src/helsinki.at/rhimport/importer.go b/src/helsinki.at/rhimport/importer.go
index 95e3e73..5eea054 100644
--- a/src/helsinki.at/rhimport/importer.go
+++ b/src/helsinki.at/rhimport/importer.go
@@ -34,7 +34,8 @@ import (
)
type ImportContext struct {
- Conf *Config
+ *Config
+ *RDDB
UserName string
Password string
Trusted bool
@@ -50,9 +51,10 @@ type ImportContext struct {
DeleteSourceFile bool
}
-func NewImportContext(conf *Config, user string, group string) *ImportContext {
+func NewImportContext(conf *Config, rddb *RDDB, user string, group string) *ImportContext {
ctx := new(ImportContext)
- ctx.Conf = conf
+ ctx.Config = conf
+ ctx.RDDB = rddb
ctx.UserName = user
ctx.Password = ""
ctx.Trusted = false
@@ -74,7 +76,7 @@ func (ctx *ImportContext) getPassword(cached bool) (err error) {
req.username = ctx.UserName
req.cached = cached
req.response = make(chan getPasswordResult)
- ctx.Conf.getPasswordChan <- req
+ ctx.RDDB.getPasswordChan <- req
res := <-req.response
if res.err != nil {
diff --git a/src/helsinki.at/rhimport/rddb.go b/src/helsinki.at/rhimport/rddb.go
new file mode 100644
index 0000000..2a87388
--- /dev/null
+++ b/src/helsinki.at/rhimport/rddb.go
@@ -0,0 +1,131 @@
+//
+// rhimportd
+//
+// The Radio Helsinki Rivendell Import Daemon
+//
+//
+// Copyright (C) 2015 Christian Pointner <equinox@helsinki.at>
+//
+// This file is part of rhimportd.
+//
+// rhimportd is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// any later version.
+//
+// rhimportd is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with rhimportd. If not, see <http://www.gnu.org/licenses/>.
+//
+
+package rhimport
+
+import (
+ "database/sql"
+ "fmt"
+ _ "github.com/go-sql-driver/mysql"
+)
+
+type getPasswordResult struct {
+ password string
+ err error
+}
+
+type getPasswordRequest struct {
+ username string
+ cached bool
+ response chan getPasswordResult
+}
+
+type RDDB struct {
+ dbh *sql.DB
+ password_cache map[string]string
+ getPasswordChan chan getPasswordRequest
+ dbGetPasswordStmt *sql.Stmt
+ quit chan bool
+ done chan bool
+}
+
+func (self *RDDB) init(conf *Config) (err error) {
+ dsn := fmt.Sprintf("%s:%s@tcp(%s:3306)/%s", conf.db_user, conf.db_passwd, conf.db_host, conf.db_db)
+ if self.dbh, err = sql.Open("mysql", dsn); err != nil {
+ return
+ }
+ if self.dbGetPasswordStmt, err = self.dbh.Prepare("select PASSWORD from USERS where LOGIN_NAME = ?"); err != nil {
+ return
+ }
+
+ return
+}
+
+func (self *RDDB) getPassword(username string, cached bool) (pwd string, err error) {
+
+ if cached {
+ pwd = self.password_cache[username]
+ }
+
+ if pwd == "" {
+ err = self.dbGetPasswordStmt.QueryRow(username).Scan(&pwd)
+ if err != nil {
+ if err == sql.ErrNoRows {
+ err = fmt.Errorf("user '%s' not known by rivendell", username)
+ }
+ return
+ }
+ self.password_cache[username] = pwd
+ }
+
+ return
+}
+
+func (self *RDDB) dispatchRequests() {
+ defer func() { self.done <- true }()
+ for {
+ select {
+ case <-self.quit:
+ return
+ case req := <-self.getPasswordChan:
+ if req.cached {
+ rhdl.Println("RDDB: got getPassword request for", req.username, "(cached)")
+ } else {
+ rhdl.Println("RDDB: got getPassword request for", req.username, "(not cached)")
+ }
+ pwd, err := self.getPassword(req.username, req.cached)
+ req.response <- getPasswordResult{pwd, err}
+ }
+ }
+}
+
+func (self *RDDB) Cleanup() {
+ self.quit <- true
+ <-self.done
+ close(self.quit)
+ close(self.done)
+ close(self.getPasswordChan)
+ if self.dbh != nil {
+ self.dbh.Close()
+ }
+ if self.dbGetPasswordStmt != nil {
+ self.dbGetPasswordStmt.Close()
+ }
+}
+
+func NewRDDB(conf *Config) (db *RDDB, err error) {
+ db = new(RDDB)
+
+ db.quit = make(chan bool)
+ db.done = make(chan bool)
+ db.password_cache = make(map[string]string)
+ db.getPasswordChan = make(chan getPasswordRequest)
+
+ if err = db.init(conf); err != nil {
+ return
+ }
+
+ go db.dispatchRequests()
+ return
+}