engine/hook.go

package engine

import (
	"context"
	"reflect"
	"sync"
)

type TransCodeReq struct {
	*Subscriber
	RequestCodec string
}

const (
	HOOK_SUBSCRIBE          = "Subscribe"
	HOOK_UNSUBSCRIBE        = "UnSubscibe"
	HOOK_STREAMCLOSE        = "StreamClose"
	HOOK_PUBLISH            = "Publish"
	HOOK_REQUEST_TRANSAUDIO = "RequestTransAudio"
)

var Hooks = make(map[string]*RingBuffer)
var hookLocker sync.Mutex

func AddHooks(hooks map[string]interface{}) {
	hookLocker.Lock()
	for name, hook := range hooks {
		rl, ok := Hooks[name]
		if !ok {
			rl = &RingBuffer{}
			rl.Init(4)
			Hooks[name] = rl
		}
		go func(hooks *RingBuffer, callback interface{}) {
			vf := reflect.ValueOf(callback)
			if vf.Kind() != reflect.Func {
				panic("callback is not a function")
			}
			for {
				vf.Call(hooks.Read().([]reflect.Value))
				hooks.MoveNext()
			}
		}(rl.Clone(), hook)
	}
	hookLocker.Unlock()
}

func AddHook(name string, callback interface{}) {
	hookLocker.Lock()
	rl, ok := Hooks[name]
	if !ok {
		rl = &RingBuffer{}
		rl.Init(4)
		Hooks[name] = rl
	}
	hookLocker.Unlock()
	vf := reflect.ValueOf(callback)
	if vf.Kind() != reflect.Func {
		panic("callback is not a function")
	}
	for hooks := rl.Clone(); ; hooks.MoveNext() {
		vf.Call(hooks.Read().([]reflect.Value))
	}
}

func AddHookWithContext(ctx context.Context, name string, callback interface{}) {
	hookLocker.Lock()
	rl, ok := Hooks[name]
	if !ok {
		rl = NewRingBuffer(4)
		Hooks[name] = rl
	}
	hookLocker.Unlock()
	vf := reflect.ValueOf(callback)
	if vf.Kind() != reflect.Func {
		panic("callback is not a function")
	}
	for hooks := rl.Clone(); ctx.Err() == nil; hooks.MoveNext() {
		vf.Call(hooks.Read().([]reflect.Value))
	}
}

func TriggerHook(name string, payload ...interface{}) {
	args := make([]reflect.Value, len(payload))
	for i, arg := range payload {
		args[i] = reflect.ValueOf(&arg).Elem()
	}
	hookLocker.Lock()
	defer hookLocker.Unlock()
	if rl, ok := Hooks[name]; ok {
		rl.Write(args)
	} else {
		rl = NewRingBuffer(4)
		Hooks[name] = rl
		rl.Write(args)
	}
}