summaryrefslogtreecommitdiff
path: root/rddb.go
diff options
context:
space:
mode:
Diffstat (limited to 'rddb.go')
-rw-r--r--rddb.go455
1 files changed, 0 insertions, 455 deletions
diff --git a/rddb.go b/rddb.go
deleted file mode 100644
index 4c9c6c8..0000000
--- a/rddb.go
+++ /dev/null
@@ -1,455 +0,0 @@
-//
-// rhimportd
-//
-// The Radio Helsinki Rivendell Import Daemon
-//
-//
-// Copyright (C) 2015-2016 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/ziutek/mymysql/godrv"
- "regexp"
- "strings"
-)
-
-var (
- showMacroRe = regexp.MustCompile(`^LL 1 ([^ ]+) 0\!$`)
- mysqlTableNameRe = regexp.MustCompile(`^[_0-9a-zA-Z-]+$`)
-)
-
-const (
- DB_VERSION = 245
-)
-
-type getPasswordResult struct {
- password string
- err error
-}
-
-type getPasswordRequest struct {
- username string
- cached bool
- response chan<- getPasswordResult
-}
-
-type getGroupOfCartResult struct {
- group string
- err error
-}
-
-type getGroupOfCartRequest struct {
- cart uint
- response chan<- getGroupOfCartResult
-}
-
-type getShowInfoResult struct {
- title string
- group string
- carts []uint
- normLvl int
- trimLvl int
- err error
-}
-
-type getShowInfoRequest struct {
- showid uint
- response chan<- getShowInfoResult
-}
-
-type checkMusicGroupResult struct {
- ismusic bool
- err error
-}
-
-type checkMusicGroupRequest struct {
- group string
- response chan<- checkMusicGroupResult
-}
-
-type getMusicInfoResult struct {
- normLvl int
- trimLvl int
- err error
-}
-
-type getMusicInfoRequest struct {
- group string
- response chan<- getMusicInfoResult
-}
-
-type RdDb struct {
- dbh *sql.DB
- passwordCache map[string]string
- getPasswordChan chan getPasswordRequest
- getPasswordStmt *sql.Stmt
- getGroupOfCartChan chan getGroupOfCartRequest
- getGroupOfCartStmt *sql.Stmt
- getShowInfoChan chan getShowInfoRequest
- getShowInfoStmt *sql.Stmt
- checkMusicGroupChan chan checkMusicGroupRequest
- checkMusicGroupStmt *sql.Stmt
- getMusicInfoChan chan getMusicInfoRequest
- getMusicInfoStmt *sql.Stmt
- quit chan bool
- done chan bool
-}
-
-func (self *RdDb) init(conf *Config) (err error) {
- godrv.Register("SET CHARACTER SET utf8;")
-
- dsn := fmt.Sprintf("tcp:%s:3306*%s/%s/%s", conf.dbHost, conf.dbDb, conf.dbUser, conf.dbPasswd)
- if self.dbh, err = sql.Open("mymysql", dsn); err != nil {
- return
- }
-
- var dbver int
- err = self.dbh.QueryRow("select DB from VERSION;").Scan(&dbver)
- if err != nil {
- err = fmt.Errorf("fetching version: %s", err)
- return
- }
- if dbver != DB_VERSION {
- err = fmt.Errorf("version mismatch is %d, should be %d", dbver, DB_VERSION)
- return
- }
-
- if self.getPasswordStmt, err = self.dbh.Prepare("select PASSWORD from USERS where LOGIN_NAME = ?;"); err != nil {
- return
- }
- if self.getGroupOfCartStmt, err = self.dbh.Prepare("select NAME,DEFAULT_LOW_CART,DEFAULT_HIGH_CART from GROUPS where DEFAULT_LOW_CART <= ? and DEFAULT_HIGH_CART >= ?;"); err != nil {
- return
- }
- if self.getShowInfoStmt, err = self.dbh.Prepare("select CART.TITLE,CART.MACROS,DROPBOXES.GROUP_NAME,DROPBOXES.NORMALIZATION_LEVEL,DROPBOXES.AUTOTRIM_LEVEL,GROUPS.DEFAULT_LOW_CART,GROUPS.DEFAULT_HIGH_CART from CART, DROPBOXES, GROUPS where CART.NUMBER = DROPBOXES.TO_CART and GROUPS.NAME = DROPBOXES.GROUP_NAME and CART.NUMBER = ?;"); err != nil {
- return
- }
- if self.checkMusicGroupStmt, err = self.dbh.Prepare("select count(*) from DROPBOXES where GROUP_NAME = ? and SET_USER_DEFINED like \"M;%\";"); err != nil {
- return
- }
- if self.getMusicInfoStmt, err = self.dbh.Prepare("select NORMALIZATION_LEVEL,AUTOTRIM_LEVEL from DROPBOXES where DROPBOXES.GROUP_NAME = ?;"); err != nil {
- return
- }
- return
-}
-
-func (self *RdDb) getPassword(username string, cached bool) (result getPasswordResult) {
- if cached {
- result.password = self.passwordCache[username]
- }
-
- if result.password == "" {
- if result.err = self.getPasswordStmt.QueryRow(username).Scan(&result.password); result.err != nil {
- if result.err == sql.ErrNoRows {
- result.err = fmt.Errorf("user '%s' not known by rivendell", username)
- }
- return
- }
- self.passwordCache[username] = result.password
- }
-
- return
-}
-
-func (self *RdDb) getGroupOfCart(cart uint) (result getGroupOfCartResult) {
- var rows *sql.Rows
- if rows, result.err = self.getGroupOfCartStmt.Query(cart, cart); result.err != nil {
- return
- }
- defer rows.Close()
- sizeMin := ^uint(0)
- for rows.Next() {
- var name string
- var lowCart, highCart uint
- if result.err = rows.Scan(&name, &lowCart, &highCart); result.err != nil {
- return
- }
- if highCart >= lowCart {
- size := (highCart - lowCart) + 1
- if sizeMin > size {
- result.group = name
- sizeMin = size
- }
- }
- }
- if result.err = rows.Err(); result.err != nil {
- return
- }
- if result.group == "" {
- result.err = fmt.Errorf("cart is outside of all group cart ranges")
- }
- return
-}
-
-func (self *RdDb) getLogTableName(log string) (logtable string, err error) {
- logtable = strings.Replace(log, " ", "_", -1) + "_LOG"
- if !mysqlTableNameRe.MatchString(logtable) {
- return "", fmt.Errorf("the log table name contains illegal charecters: %s", logtable)
- }
- return
-}
-
-func (self *RdDb) getShowCarts(log string, lowCart, highCart int) (carts []uint, err error) {
- var logtable string
- if logtable, err = self.getLogTableName(log); err != nil {
- return
- }
- q := fmt.Sprintf("select CART_NUMBER from %s where CART_NUMBER >= %d and CART_NUMBER <= %d order by COUNT;", logtable, lowCart, highCart)
- var rows *sql.Rows
- if rows, err = self.dbh.Query(q); err != nil {
- return
- }
- defer rows.Close()
- for rows.Next() {
- var cart uint
- if err = rows.Scan(&cart); err != nil {
- return
- }
- carts = append(carts, cart)
- }
- err = rows.Err()
- return
-}
-
-func (self *RdDb) getShowInfo(showid uint) (result getShowInfoResult) {
- var macros string
- var lowCart, highCart int
- result.err = self.getShowInfoStmt.QueryRow(showid).Scan(&result.title, &macros, &result.group, &result.normLvl, &result.trimLvl, &lowCart, &highCart)
- if result.err != nil {
- if result.err == sql.ErrNoRows {
- result.err = fmt.Errorf("show '%d' not found", showid)
- }
- return
- }
- result.normLvl /= 100
- result.trimLvl /= 100
- result.carts, result.err = self.getShowCarts(showMacroRe.FindStringSubmatch(macros)[1], lowCart, highCart)
- return
-}
-
-func (self *RdDb) checkMusicGroup(group string) (result checkMusicGroupResult) {
- var cnt int
- if result.err = self.checkMusicGroupStmt.QueryRow(group).Scan(&cnt); result.err != nil {
- if result.err == sql.ErrNoRows {
- result.err = nil
- result.ismusic = false
- }
- return
- }
- result.ismusic = cnt > 0
- return
-}
-
-func (self *RdDb) getMusicInfo(group string) (result getMusicInfoResult) {
- result.err = self.getMusicInfoStmt.QueryRow(group).Scan(&result.normLvl, &result.trimLvl)
- if result.err != nil {
- if result.err == sql.ErrNoRows {
- result.err = fmt.Errorf("music pool '%s' not found", group)
- }
- return
- }
- return
-}
-
-func (self *RdDb) dispatchRequests() {
- defer func() { self.done <- true }()
- for {
- select {
- case <-self.quit:
- return
- case req := <-self.getPasswordChan:
- req.response <- self.getPassword(req.username, req.cached)
- case req := <-self.getGroupOfCartChan:
- req.response <- self.getGroupOfCart(req.cart)
- case req := <-self.getShowInfoChan:
- req.response <- self.getShowInfo(req.showid)
- case req := <-self.checkMusicGroupChan:
- req.response <- self.checkMusicGroup(req.group)
- case req := <-self.getMusicInfoChan:
- req.response <- self.getMusicInfo(req.group)
- }
- }
-}
-
-// *********************************************************
-// Public Interface
-
-type RdDbChan struct {
- getPasswordChan chan<- getPasswordRequest
- getGroupOfCartChan chan<- getGroupOfCartRequest
- getShowInfoChan chan<- getShowInfoRequest
- checkMusicGroupChan chan<- checkMusicGroupRequest
- getMusicInfoChan chan<- getMusicInfoRequest
-}
-
-func (self *RdDbChan) GetPassword(username string, cached bool) (string, error) {
- resCh := make(chan getPasswordResult)
- req := getPasswordRequest{}
- req.username = username
- req.cached = cached
- req.response = resCh
- self.getPasswordChan <- req
-
- res := <-resCh
- if res.err != nil {
- return "", res.err
- }
- return res.password, nil
-}
-
-func (self *RdDbChan) CheckPassword(username, password string) (bool, error) {
- cached := true
-
- for {
- resCh := make(chan getPasswordResult)
- req := getPasswordRequest{}
- req.username = username
- req.cached = cached
- req.response = resCh
- self.getPasswordChan <- req
-
- res := <-resCh
- if res.err != nil {
- return false, res.err
- }
- if password == res.password {
- return true, nil
- }
- if cached {
- cached = false
- } else {
- break
- }
- }
- return false, nil
-}
-
-func (self *RdDbChan) GetGroupOfCart(cart uint) (string, error) {
- resCh := make(chan getGroupOfCartResult)
- req := getGroupOfCartRequest{}
- req.cart = cart
- req.response = resCh
- self.getGroupOfCartChan <- req
-
- res := <-resCh
- if res.err != nil {
- return "", res.err
- }
- return res.group, nil
-}
-
-func (self *RdDbChan) GetShowInfo(showid uint) (string, int, int, []uint, error) {
- resCh := make(chan getShowInfoResult)
- req := getShowInfoRequest{}
- req.showid = showid
- req.response = resCh
- self.getShowInfoChan <- req
-
- res := <-resCh
- if res.err != nil {
- return "", 0, 0, nil, res.err
- }
- return res.group, res.normLvl, res.trimLvl, res.carts, nil
-}
-
-func (self *RdDbChan) CheckMusicGroup(groupname string) (bool, error) {
- resCh := make(chan checkMusicGroupResult)
- req := checkMusicGroupRequest{}
- req.group = groupname
- req.response = resCh
- self.checkMusicGroupChan <- req
-
- res := <-resCh
- if res.err != nil {
- return false, res.err
- }
- return res.ismusic, nil
-}
-
-func (self *RdDbChan) GetMusicInfo(groupname string) (int, int, error) {
- resCh := make(chan getMusicInfoResult)
- req := getMusicInfoRequest{}
- req.group = groupname
- req.response = resCh
- self.getMusicInfoChan <- req
-
- res := <-resCh
- if res.err != nil {
- return 0, 0, res.err
- }
- return res.normLvl, res.trimLvl, nil
-}
-
-func (self *RdDb) GetInterface() *RdDbChan {
- ch := &RdDbChan{}
- ch.getPasswordChan = self.getPasswordChan
- ch.getGroupOfCartChan = self.getGroupOfCartChan
- ch.getShowInfoChan = self.getShowInfoChan
- ch.checkMusicGroupChan = self.checkMusicGroupChan
- ch.getMusicInfoChan = self.getMusicInfoChan
- return ch
-}
-
-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.getPasswordStmt != nil {
- self.getPasswordStmt.Close()
- }
- if self.getGroupOfCartStmt != nil {
- self.getGroupOfCartStmt.Close()
- }
- if self.getShowInfoStmt != nil {
- self.getShowInfoStmt.Close()
- }
- if self.checkMusicGroupStmt != nil {
- self.checkMusicGroupStmt.Close()
- }
- if self.getMusicInfoStmt != nil {
- self.getMusicInfoStmt.Close()
- }
-}
-
-func NewRdDb(conf *Config) (db *RdDb, err error) {
- db = new(RdDb)
-
- db.quit = make(chan bool)
- db.done = make(chan bool)
- db.passwordCache = make(map[string]string)
- db.getPasswordChan = make(chan getPasswordRequest, 10)
- db.getGroupOfCartChan = make(chan getGroupOfCartRequest, 10)
- db.getShowInfoChan = make(chan getShowInfoRequest, 10)
- db.checkMusicGroupChan = make(chan checkMusicGroupRequest, 10)
- db.getMusicInfoChan = make(chan getMusicInfoRequest, 10)
-
- if err = db.init(conf); err != nil {
- return
- }
-
- go db.dispatchRequests()
- return
-}