69 lines
1.5 KiB
Go
69 lines
1.5 KiB
Go
package pool
|
|
|
|
import (
|
|
"context"
|
|
"github.com/anytypeio/any-sync/app"
|
|
"github.com/anytypeio/any-sync/app/logger"
|
|
"github.com/anytypeio/any-sync/app/ocache"
|
|
"github.com/anytypeio/any-sync/metric"
|
|
"github.com/anytypeio/any-sync/net/dialer"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"time"
|
|
)
|
|
|
|
const (
|
|
CName = "common.net.pool"
|
|
)
|
|
|
|
var log = logger.NewNamed(CName)
|
|
|
|
func New() Service {
|
|
return &poolService{}
|
|
}
|
|
|
|
type Service interface {
|
|
Pool
|
|
NewPool(name string) Pool
|
|
app.ComponentRunnable
|
|
}
|
|
|
|
type poolService struct {
|
|
// default pool
|
|
*pool
|
|
dialer dialer.Dialer
|
|
metricReg *prometheus.Registry
|
|
}
|
|
|
|
func (p *poolService) Init(a *app.App) (err error) {
|
|
p.dialer = a.MustComponent(dialer.CName).(dialer.Dialer)
|
|
p.pool = &pool{dialer: p.dialer}
|
|
if m := a.Component(metric.CName); m != nil {
|
|
p.metricReg = m.(metric.Metric).Registry()
|
|
}
|
|
p.pool.cache = ocache.New(
|
|
func(ctx context.Context, id string) (value ocache.Object, err error) {
|
|
return p.dialer.Dial(ctx, id)
|
|
},
|
|
ocache.WithLogger(log.Sugar()),
|
|
ocache.WithGCPeriod(time.Minute),
|
|
ocache.WithTTL(time.Minute*5),
|
|
ocache.WithPrometheus(p.metricReg, "netpool", "default"),
|
|
)
|
|
return nil
|
|
}
|
|
|
|
func (p *poolService) NewPool(name string) Pool {
|
|
return &pool{
|
|
dialer: p.dialer,
|
|
cache: ocache.New(
|
|
func(ctx context.Context, id string) (value ocache.Object, err error) {
|
|
return p.dialer.Dial(ctx, id)
|
|
},
|
|
ocache.WithLogger(log.Sugar()),
|
|
ocache.WithGCPeriod(time.Minute),
|
|
ocache.WithTTL(time.Minute*5),
|
|
ocache.WithPrometheus(p.metricReg, "netpool", name),
|
|
),
|
|
}
|
|
}
|