go-service-template/pool/pool.go

package pool

import (
	"context"
	"git.ouxuan.net/3136352472/go-service-template/jsruntime"
	pool "github.com/jolestar/go-commons-pool"
	"log"
	"time"
)

type JsRuntimePool struct {
	ctx           context.Context
	jsRuntimePool *pool.ObjectPool
}
type Config struct {
	MaxIdle  int
	MinIdle  int
	MaxTotal int
}

var DefaultConfig Config

func NewJsRuntimePool(mainScript string, useFileName string, staticPath string, relys jsruntime.Relys, mode jsruntime.RunMode) (jrp *JsRuntimePool) {
	jrp = &JsRuntimePool{}

	factory := pool.NewPooledObjectFactorySimple(
		func(context.Context) (interface{}, error) {
			render, err := jsruntime.NewJsRuntime(mainScript, useFileName, staticPath, relys, mode)

			return render, err
		})

	jrp.ctx = context.Background()

	if DefaultConfig.MaxIdle == 0 {
		DefaultConfig.MaxIdle = 3000
	}
	if DefaultConfig.MinIdle == 0 {
		DefaultConfig.MinIdle = 20
	}

	if DefaultConfig.MaxTotal == 0 {
		DefaultConfig.MaxTotal = 200
	}

	jrp.jsRuntimePool = pool.NewObjectPoolWithDefaultConfig(jrp.ctx, factory)
	jrp.jsRuntimePool.Config.LIFO = false

	jrp.jsRuntimePool.Config.MaxIdle = DefaultConfig.MaxIdle
	jrp.jsRuntimePool.Config.MinIdle = DefaultConfig.MinIdle
	jrp.jsRuntimePool.Config.MaxTotal = DefaultConfig.MaxTotal
	jrp.jsRuntimePool.Config.BlockWhenExhausted = true

	go func() {
		for ; ; {

			jrp.jsRuntimePool.PreparePool(jrp.ctx)
			time.Sleep(time.Second / 5)
		}
	}()

	return
}

func (jrp *JsRuntimePool) JsRuntimeCall(call func(jr *jsruntime.JsRuntime)) (err error) {
	obj, err := jrp.jsRuntimePool.BorrowObject(jrp.ctx)
	if err != nil {
		return
	}

	o := obj.(*jsruntime.JsRuntime)
	call(o)

	err = jrp.jsRuntimePool.ReturnObject(jrp.ctx, obj)
	return
}

func (jrp *JsRuntimePool) Log() {
	idle := jrp.jsRuntimePool.GetNumIdle()
	active := jrp.jsRuntimePool.GetNumActive()
	all := idle + active
	log.Println("所有:", all, "空闲:", idle, "活动:", active)
}

func (jrp *JsRuntimePool) NumInfo() (int, int, int) {
	idle := jrp.jsRuntimePool.GetNumIdle()
	active := jrp.jsRuntimePool.GetNumActive()
	all := idle + active
	return all, idle, active
}