fetching pool cart list

1 files changed, 90 insertions, 1 deletions

diff --git a/rddb/rddb.go b/rddb/rddb.go
index c889dff..4edda6a 100644
--- a/rddb/rddb.go
+++ b/rddb/rddb.go
@@ -134,7 +134,9 @@ type CutListEntry struct {
 type CartListEntry struct {
 	Number uint
 	Exists bool
+	Artist string
 	Title  string
+	Album  string
 	Cuts   []CutListEntry
 }
 
@@ -195,6 +197,16 @@ type getPoolListRequest struct {
 	response chan<- getPoolListResult
 }
 
+type getPoolCartListResult struct {
+	carts map[uint]CartListEntry
+	err   error
+}
+
+type getPoolCartListRequest struct {
+	pool     PoolListEntry
+	response chan<- getPoolCartListResult
+}
+
 type DB struct {
 	conf                 *config
 	dbh                  *sql.DB
@@ -219,6 +231,8 @@ type DB struct {
 	getJingleCutListChan chan getJingleCutListRequest
 	getPoolListChan      chan getPoolListRequest
 	getPoolListStmt      *sql.Stmt
+	getPoolCartListChan  chan getPoolCartListRequest
+	getPoolCartsStmt     *sql.Stmt
 	quit                 chan bool
 	done                 chan bool
 }
@@ -263,7 +277,7 @@ func (db *DB) init() (err error) {
 	if db.getCartInfoStmt, err = db.dbh.Prepare("select TITLE from CART where TYPE=1 and NUMBER = ?"); err != nil {
 		return
 	}
-	if db.getCutInfoStmt, err = db.dbh.Prepare("select CUT_NAME, EVERGREEN, DESCRIPTION, LENGTH, ORIGIN_DATETIME, PLAY_COUNTER, LAST_PLAY_DATETIME from CUTS where CART_NUMBER = ?"); err != nil {
+	if db.getCutInfoStmt, err = db.dbh.Prepare("select CUT_NAME,EVERGREEN,DESCRIPTION,LENGTH,ORIGIN_DATETIME,PLAY_COUNTER,LAST_PLAY_DATETIME from CUTS where CART_NUMBER = ?"); err != nil {
 		return
 	}
 	if db.getJingleListStmt, err = db.dbh.Prepare("select GROUPS.NAME,DROPBOXES.SET_USER_DEFINED,GROUPS.DEFAULT_LOW_CART,GROUPS.DEFAULT_HIGH_CART,GROUPS.DESCRIPTION from USER_PERMS, DROPBOXES, GROUPS where USER_PERMS.USER_NAME = ? and DROPBOXES.GROUP_NAME = USER_PERMS.GROUP_NAME and DROPBOXES.GROUP_NAME = GROUPS.NAME and DROPBOXES.STATION_NAME = ?;"); err != nil {
@@ -272,6 +286,9 @@ func (db *DB) init() (err error) {
 	if db.getPoolListStmt, err = db.dbh.Prepare("select GROUPS.NAME,DROPBOXES.SET_USER_DEFINED,GROUPS.DESCRIPTION from USER_PERMS, DROPBOXES, GROUPS where USER_PERMS.USER_NAME = ? and DROPBOXES.GROUP_NAME = USER_PERMS.GROUP_NAME and DROPBOXES.GROUP_NAME = GROUPS.NAME and DROPBOXES.STATION_NAME = ?;"); err != nil {
 		return
 	}
+	if db.getPoolCartsStmt, err = db.dbh.Prepare("select CART.NUMBER,CART.ARTIST,CART.TITLE,CART.ALBUM,CUTS.CUT_NAME,CUTS.EVERGREEN,CUTS.DESCRIPTION,CUTS.LENGTH,CUTS.ORIGIN_DATETIME,CUTS.PLAY_COUNTER,CUTS.LAST_PLAY_DATETIME from CUTS,CART where CUTS.CART_NUMBER = CART.NUMBER and CART.GROUP_NAME = ?;"); err != nil {
+		return
+	}
 
 	return
 }
@@ -551,6 +568,56 @@ func (db *DB) getPoolList(username string) (result getPoolListResult) {
 	return
 }
 
+func (db *DB) getPoolCartList(pool PoolListEntry) (result getPoolCartListResult) {
+	var rows *sql.Rows
+	if rows, result.err = db.getPoolCartsStmt.Query(pool.Group); result.err != nil {
+		return
+	}
+	defer rows.Close()
+
+	result.carts = make(map[uint]CartListEntry)
+	for rows.Next() {
+		var cut CutListEntry
+		var cart, length uint
+		var cutName, evergreen string
+		var artist, title, album sql.NullString
+
+		if result.err = rows.Scan(&cart, &artist, &title, &album, &cutName, &evergreen, &cut.Description, &length, &cut.Imported, &cut.NumPlayed, &cut.LastPlayed); result.err != nil {
+			return
+		}
+
+		parts := strings.Split(cutName, "_")
+		if len(parts) == 2 {
+			if cn, converr := strconv.ParseUint(parts[1], 10, 32); converr != nil {
+				continue
+			} else {
+				cut.Number = uint(cn)
+			}
+		} else {
+			continue
+		}
+		switch evergreen {
+		case "Y":
+			cut.Evergreen = true
+		default:
+			cut.Evergreen = false
+		}
+		cut.Duration = time.Duration(length) * time.Millisecond
+
+		if c, exists := result.carts[cart]; !exists {
+			c = CartListEntry{cart, true, artist.String, title.String, album.String, nil}
+			c.Cuts = append(c.Cuts, cut)
+			result.carts[cart] = c
+		} else {
+			c.Cuts = append(c.Cuts, cut)
+		}
+	}
+	result.err = rows.Err()
+	return
+
+	return
+}
+
 func (db *DB) dispatchRequests() {
 	defer func() { db.done <- true }()
 	for {
@@ -577,6 +644,8 @@ func (db *DB) dispatchRequests() {
 			req.response <- db.getJingleCutList(req.jingle)
 		case req := <-db.getPoolListChan:
 			req.response <- db.getPoolList(req.username)
+		case req := <-db.getPoolCartListChan:
+			req.response <- db.getPoolCartList(req.pool)
 		}
 	}
 }
@@ -595,6 +664,7 @@ type DBChan struct {
 	getJingleListChan    chan<- getJingleListRequest
 	getJingleCutListChan chan<- getJingleCutListRequest
 	getPoolListChan      chan<- getPoolListRequest
+	getPoolCartListChan  chan<- getPoolCartListRequest
 }
 
 func (db *DBChan) GetPassword(username string, cached bool) (string, error) {
@@ -762,6 +832,20 @@ func (db *DBChan) GetPoolList(username string) (PoolList, error) {
 	return res.pools, nil
 }
 
+func (db *DBChan) GetPoolCartList(pool PoolListEntry) (map[uint]CartListEntry, error) {
+	resCh := make(chan getPoolCartListResult)
+	req := getPoolCartListRequest{}
+	req.pool = pool
+	req.response = resCh
+	db.getPoolCartListChan <- req
+
+	res := <-resCh
+	if res.err != nil {
+		return nil, res.err
+	}
+	return res.carts, nil
+}
+
 func (db *DB) GetInterface() *DBChan {
 	ch := &DBChan{}
 	ch.getPasswordChan = db.getPasswordChan
@@ -774,6 +858,7 @@ func (db *DB) GetInterface() *DBChan {
 	ch.getJingleListChan = db.getJingleListChan
 	ch.getJingleCutListChan = db.getJingleCutListChan
 	ch.getPoolListChan = db.getPoolListChan
+	ch.getPoolCartListChan = db.getPoolCartListChan
 	return ch
 }
 
@@ -816,6 +901,9 @@ func (db *DB) Cleanup() {
 	if db.getPoolListStmt != nil {
 		db.getPoolListStmt.Close()
 	}
+	if db.getPoolCartsStmt != nil {
+		db.getPoolCartsStmt.Close()
+	}
 }
 
 func NewDB(configfile string) (db *DB, err error) {
@@ -836,6 +924,7 @@ func NewDB(configfile string) (db *DB, err error) {
 	db.getJingleListChan = make(chan getJingleListRequest, 10)
 	db.getJingleCutListChan = make(chan getJingleCutListRequest, 10)
 	db.getPoolListChan = make(chan getPoolListRequest, 10)
+	db.getPoolCartListChan = make(chan getPoolCartListRequest, 10)
 
 	if err = db.init(); err != nil {
 		return