mirror of
https://github.com/crazybber/awesome-patterns.git
synced 2025-02-16 13:43:13 +03:00
added comments and docs
This commit is contained in:
parent
f57dd8dad4
commit
3251699069
@ -10,19 +10,39 @@ import (
|
|||||||
// Pool manages a set of resources that can be shared safely by
|
// Pool manages a set of resources that can be shared safely by
|
||||||
// multiply goroutines. The resource been managed must implement
|
// multiply goroutines. The resource been managed must implement
|
||||||
// the io.Closer interface.
|
// the io.Closer interface.
|
||||||
|
// declares a struct named Pool that allows
|
||||||
|
// the caller to create as many different pools as needed. Each pool can manage any type
|
||||||
|
// of resource as long as the type implements the io.Closer interface.
|
||||||
type Pool struct {
|
type Pool struct {
|
||||||
m sync.Mutex
|
// This mutex is used to keep all the operations against a Pool value-safe for multigoroutine access.
|
||||||
|
m sync.Mutex
|
||||||
|
// The second field is named resources and is declared as a channel of interface type io.Closer.
|
||||||
|
// This channel will be created as a buffered channel and will contain the resources being shared.
|
||||||
|
// Because an interface type is being used, the pool can manage any type of resource that
|
||||||
|
// implements the io.Closer interface
|
||||||
resources chan io.Closer
|
resources chan io.Closer
|
||||||
factory func() (io.Closer, error)
|
// The factory field is of a function type. Any function that takes no parameters and
|
||||||
closed bool
|
// returns an io.Closer and an error interface value can be assigned to this field. The
|
||||||
|
// purpose of this function is to create a new resource when the pool requires one. This
|
||||||
|
// functionality is an implementation detail beyond the scope of the pool package and
|
||||||
|
// needs to be implemented and supplied by the user using this package.
|
||||||
|
factory func() (io.Closer, error)
|
||||||
|
// This field is a flag that indicates the Pool is being shut down or is already shut down.
|
||||||
|
closed bool
|
||||||
}
|
}
|
||||||
|
|
||||||
// ErrPoolClosed is returned when an Acquire returns on a closed pool.
|
// ErrPoolClosed is returned when an Acquire returns on a closed pool.
|
||||||
|
// Creating error interface variables is a common practice in Go. This allows the caller
|
||||||
|
// to identify specific returned error values from any function or method within the package.
|
||||||
var ErrPoolClosed = errors.New("Pool has been closed")
|
var ErrPoolClosed = errors.New("Pool has been closed")
|
||||||
|
|
||||||
// New creates a pool that manage resources. A pool requires a function
|
// New creates a pool that manage resources. A pool requires a function
|
||||||
// that can allocate a new resource and the size of the pool.
|
// that can allocate a new resource and the size of the pool.
|
||||||
|
// The function parameter represents a factory function that creates values of the resource being managed by the pool.
|
||||||
|
// The second parameter, size, represents the size of the buffered channel created to hold the resources.
|
||||||
func New(fn func() (io.Closer, error), size uint32) (*Pool, error) {
|
func New(fn func() (io.Closer, error), size uint32) (*Pool, error) {
|
||||||
|
// The first parameter, fn, is declared as a function type that accepts no parameters and
|
||||||
|
// returns an io.Closer and an error interface value
|
||||||
if size <= 0 {
|
if size <= 0 {
|
||||||
return nil, errors.New("Size value too small")
|
return nil, errors.New("Size value too small")
|
||||||
}
|
}
|
||||||
@ -33,9 +53,11 @@ func New(fn func() (io.Closer, error), size uint32) (*Pool, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Acquire retrieves a resource from the pool.
|
// Acquire retrieves a resource from the pool.
|
||||||
|
// This method returns a resource from the pool if one is available, or creates a new one for the call.
|
||||||
func (p *Pool) Acquire() (io.Closer, error) {
|
func (p *Pool) Acquire() (io.Closer, error) {
|
||||||
select {
|
select {
|
||||||
// Check for free resources
|
// Check for free resources
|
||||||
|
// using a select / case statement to check if there’s a resource in the buffered channel.
|
||||||
case r, ok := <-p.resources:
|
case r, ok := <-p.resources:
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, ErrPoolClosed
|
return nil, ErrPoolClosed
|
||||||
@ -49,11 +71,20 @@ func (p *Pool) Acquire() (io.Closer, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Release place a new resource into the pool.
|
// Release place a new resource into the pool.
|
||||||
|
// After a resource is acquired and no longer needed, it must be released back into
|
||||||
|
// the pool. This is where the Release method comes in.
|
||||||
func (p *Pool) Release(r io.Closer) {
|
func (p *Pool) Release(r io.Closer) {
|
||||||
// Secure this operation with the Close operation.
|
// Secure this operation with the Close operation.
|
||||||
|
// The use of the mutex serves two purposes. First, it protects the
|
||||||
|
// read on the closed flag on line 65 from happening at the same time as a write on this
|
||||||
|
// flag in the Close method. Second, we don’t want to attempt to send on a closed channel because this will cause a panic. When the closed
|
||||||
|
// field is false, we know the resources channel has been closed
|
||||||
p.m.Lock()
|
p.m.Lock()
|
||||||
defer p.m.Unlock()
|
defer p.m.Unlock()
|
||||||
// If the pool is closed, discard the resource
|
// If the pool is closed, discard the resource
|
||||||
|
// the Close method on the resource is called directly when the pool is
|
||||||
|
// closed. This is because there’s no way to release the resource back into the pool. At
|
||||||
|
// this point the pool has been both closed and flushed.
|
||||||
if p.closed {
|
if p.closed {
|
||||||
r.Close()
|
r.Close()
|
||||||
return
|
return
|
||||||
@ -70,6 +101,11 @@ func (p *Pool) Release(r io.Closer) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Close will shutdown the pool and close all existing resources.
|
// Close will shutdown the pool and close all existing resources.
|
||||||
|
// Once the program is finished with the pool, it should call the Close method.
|
||||||
|
// The method closes and flushes the buffered channel on lines 98 and 101, closing any resources that exist until the channel is
|
||||||
|
// empty. All the code in this method must be executed by only one goroutine at a time.
|
||||||
|
// In fact, when this code is being executed, goroutines must also be
|
||||||
|
// prevented from executing code in the Release method. You’ll understand why this is important soon
|
||||||
func (p *Pool) Close() {
|
func (p *Pool) Close() {
|
||||||
// Secure this operation with release operation.
|
// Secure this operation with release operation.
|
||||||
p.m.Lock()
|
p.m.Lock()
|
||||||
|
Loading…
x
Reference in New Issue
Block a user