dbpool!

1 files changed, 16 insertions, 11 deletions

diff --git a/db.go b/db.go
index 8ded5ec..9434273 100644
--- a/db.go
+++ b/db.go
@@ -6,20 +6,25 @@ import (
 	"log"
 )
 
-var db *mysql.Client
+var dbPool chan *mysql.Client
 
 func initDb() {
 	log.SetFlags(log.Ltime | log.Lshortfile)
 
-	var err os.Error
-	db, err = mysql.DialTCP("173.228.31.111", "audio", "audio", "audio")
-	if err != nil {
-		log.Panicln(err)
+	const dbPoolSize = 4
+	dbPool = make(chan *mysql.Client, 4)
+
+	for i := 0; i < dbPoolSize; i++ {
+		db, err := mysql.DialTCP("173.228.31.111", "audio", "audio", "audio")
+		if err != nil {
+			log.Panicln(err)
+		}
+		db.Reconnect = true
+		dbPool <- db
 	}
-	db.Reconnect = true
 }
 
-func prepare(sql string, params ...interface{}) (*mysql.Statement, os.Error) {
+func prepare(db *mysql.Client, sql string, params ...interface{}) (*mysql.Statement, os.Error) {
 	query, err := db.Prepare(sql)
 	if err != nil {
 		log.Println(err)
@@ -38,8 +43,8 @@ func prepare(sql string, params ...interface{}) (*mysql.Statement, os.Error) {
 	return query, err
 }
 
-func queryInt(sql string, params ...interface{}) (int, os.Error) {
-	query, err := prepare(sql, params...)
+func queryInt(db *mysql.Client, sql string, params ...interface{}) (int, os.Error) {
+	query, err := prepare(db, sql, params...)
 	if err != nil {
 		return 0, err
 	}
@@ -61,8 +66,8 @@ func queryInt(sql string, params ...interface{}) (int, os.Error) {
 }
 
 // given an id ('abcd1234'), return the pid (1)
-func getpid(id string) int {
-	pid, err := queryInt("SELECT `pid` FROM `playlist` WHERE `id` = ?", id)
+func getpid(db *mysql.Client, id string) int {
+	pid, err := queryInt(db, "SELECT `pid` FROM `playlist` WHERE `id` = ?", id)
 	if err != nil {
 		return -1
 	}