hooker.go

Rama master · réplica de lectura del repositorio original.

package main

import (
	"sync"

	"github.com/gofrs/uuid/v3"
)

type (
	hooker struct {
		hooks map[string]func(any)
		lock  sync.RWMutex
	}
)

func newHooker() *hooker {
	return &hooker{
		hooks: make(map[string]func(any)),
	}
}

func (h *hooker) Ping(payload any) {
	h.lock.RLock()
	defer h.lock.RUnlock()

	for _, hf := range h.hooks {
		hf(payload)
	}
}

func (h *hooker) Register(hook func(any)) func() {
	h.lock.Lock()
	defer h.lock.Unlock()

	id := uuid.Must(uuid.NewV4()).String()
	h.hooks[id] = hook

	return func() { h.unregister(id) }
}

func (h *hooker) unregister(id string) {
	h.lock.Lock()
	defer h.lock.Unlock()

	delete(h.hooks, id)
}