📚 gohive - Awesome Go Library for Goroutines
A highly performant and easy to use Goroutine pool for Go.
🏷️ Goroutines
📂 Tools for managing and working with Goroutines.
⭐ 52 stars
Detailed Description of gohive
gohive
Package gohive implements a simple and easy to use goroutine pool for Go
Features
- Pool can be created with a specific size as per the requirement
- Accepts tasks which implements Runner interface
- Uses channels to accepts tasks and gets them executed via workers
- Uses synchronization among workers to avoid race conditions
Installation
Use go get
to install and update:
$ go get -u github.com/loveleshsharma/gohive
Usage
- Create an instance of Pool type first
hive := gohive.NewFixedPool(5)
- Invoke the Submit() function and pass the task to execute
hive.Submit(object Runner)
Submit function accepts a Runner object as an argument, which it passes to the pool if a worker is available, otherwise it will wait for the worker to be available
- To close the pool we can invoke the Close() function
hive.Close()
Once the pool is closed, we cannot assign any task to it
Example
Let's get into a full program where we can see how to use the gohive package in order to execute many goroutines simultaneously
package main
import (
"fmt"
"github.com/loveleshsharma/gohive"
"sync"
)
func main() {
var wg sync.WaitGroup
pool := gohive.NewFixedPool(5)
for i := 1; i <= 20; i++ {
if err := pool.Submit(NewMyStruct(i, &wg)); err != nil {
fmt.Println("error: ", err)
break
}
}
wg.Wait()
}
type MyStruct struct {
num int
wg *sync.WaitGroup
}
func NewMyStruct(num int, wg *sync.WaitGroup) MyStruct {
myStruct := MyStruct{
num: num,
wg: wg,
}
wg.Add(1)
return myStruct
}
func (s MyStruct) Run() {
defer s.wg.Done()
val := s.num
fact := s.num
for i := s.num - 1; i > 0; i-- {
fact *= i
}
fmt.Printf("Factorial of %d: %d\n", val, fact)
}
Important : Always keep sync.WaitGroup in your struct and put defer wg.Done()
as the first statement of your Run() function. It will wait for your task to complete.
TODO
- Maintain a waiting queue to stop blocking submit method when all goroutines are busy.
- Submitting priority tasks which takes priority over other tasks.
- Handling panics inside goroutines to prevent them from crashing.
- Implement dynamic pool which will scale the number of goroutines as per requirement and scales down when they are idle.
- Submitting multiple tasks together.