Connection Pool

ConnectionPoolWrapper.go

type InitFunction func() (interface{}, error)

type ConnectionPoolWrapper struct {
	size int
	pool 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.pool = make(chan interface{}, size)
	for x := 0; x < size; x++ {
		conn, err := initfn()
		if err != nil {
			return err
		}

		// If the init function succeeded, add the connection to the channel
		p.pool <- conn
	}
	p.size = size
	return nil
}

func (p *ConnectionPoolWrapper) GetConnection() interface{} {
	return <-p.pool
}

func (p *ConnectionPoolWrapper) ReleaseConnection(conn interface{}) {
	p.pool <- conn
}

var dbPool = &ConnectionPoolWrapper{}