mrb · December 17, 2015 15:19 · May 22, 2013 · Aug 28, 2012 · Aug 28, 2012
diff --git a/gistfile1.txt → pool.go b/gistfile1.txt → pool.go
diff --git a/gistfile1.txt b/gistfile1.txt
@@ -5,14 +5,22 @@ type ConnectionPoolWrapper struct {
 	conn chan interface{}
 }
 
+/**
+ Call the init function size times. If the init function fails during any call, then
+ the creation of the pool is considered a failure.
+ We call the same function size times to make sure each connection shares the same
+ state.
+*/
 func (p *ConnectionPoolWrapper) InitPool(size int, initfn InitFunction) error {
+	// Create a buffered channel allowing size senders
 	p.conn = make(chan interface{}, size)
 	for x := 0; x < size; x++ {
-		fmt.Println("Calling init function ", x)
 		conn, err := initfn()
 		if err != nil {
 			return err
 		}
+
+		// If the init function succeeded, add the connection to the channel
 		p.conn <- conn
 	}
 	p.size = size

diff --git a/gistfile1.txt b/gistfile1.txt
@@ -0,0 +1,30 @@
+type InitFunction func() (interface{}, error)
+
+type ConnectionPoolWrapper struct {
+	size int
+	conn chan interface{}
+}
+
+func (p *ConnectionPoolWrapper) InitPool(size int, initfn InitFunction) error {
+	p.conn = make(chan interface{}, size)
+	for x := 0; x < size; x++ {
+		fmt.Println("Calling init function ", x)
+		conn, err := initfn()
+		if err != nil {
+			return err
+		}
+		p.conn <- conn
+	}
+	p.size = size
+	return nil
+}
+
+func (p *ConnectionPoolWrapper) GetConnection() interface{} {
+	return <-p.conn
+}
+
+func (p *ConnectionPoolWrapper) ReleaseConnection(conn interface{}) {
+	p.conn <- conn
+}
+
+var dbPool = &ConnectionPoolWrapper{}
No results found