xing/wp-go
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

优化代码

Browse Source
This commit is contained in:
xing 2023-05-08 22:24:39 +08:00
parent 4058ff9b7b
commit 68c345f928
2 changed files with 19 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
app/plugins/wphandle/apply/apply.go
View File

@ -1,11 +1,13 @@
package apply package apply
var fn any import "github.com/fthvgb1/wp-go/safety"
var fn safety.Var[any]
func SetFn(f any) { func SetFn(f any) {
fn = f fn.Store(f)
} }
func UsePlugins() any { func UsePlugins() any {
return fn return fn.Load()
} }

27
app/theme/wp/pipe.go
View File

@ -65,11 +65,12 @@ func (h *Handle) PushDataHandler(scene string, fns ...HandleCall) {
h.PushHandler(constraints.PipeData, scene, fns...) h.PushHandler(constraints.PipeData, scene, fns...)
} }
func PipeHandle(pipeScene string, keyFn func(*Handle, string) string, fn func(*Handle, map[string][]HandleCall) []HandleCall) func(HandleFn[*Handle], *Handle) { func PipeHandle(pipeScene string, keyFn func(*Handle, string) string, fn func(*Handle, map[string][]HandleCall, string) []HandleCall) func(HandleFn[*Handle], *Handle) {
return func(next HandleFn[*Handle], h *Handle) { return func(next HandleFn[*Handle], h *Handle) {
handlers := reload.GetAnyValMapBy("pipeHandlers", keyFn(h, pipeScene), h, func(h *Handle) []HandleCall { key := keyFn(h, pipeScene)
handlers := reload.GetAnyValMapBy("pipeHandlers", key, h, func(h *Handle) []HandleCall {
conf := h.handleHook[pipeScene] conf := h.handleHook[pipeScene]
calls := fn(h, h.handlers[pipeScene]) calls := fn(h, h.handlers[pipeScene], key)
calls = slice.FilterAndMap(calls, func(call HandleCall) (HandleCall, bool) { calls = slice.FilterAndMap(calls, func(call HandleCall) (HandleCall, bool) {
ok := true ok := true
for _, hook := range conf { for _, hook := range conf {
@ -135,26 +136,26 @@ func MiddlewareKey(h *Handle, pipScene string) string {
return h.ComponentFilterFnHook("middleware", "middleware", pipScene) return h.ComponentFilterFnHook("middleware", "middleware", pipScene)
} }
func PipeMiddlewareHandle(h *Handle, middlewares map[string][]HandleCall) (handlers []HandleCall) { func PipeMiddlewareHandle(h *Handle, middlewares map[string][]HandleCall, key string) (handlers []HandleCall) {
handlers = append(handlers, middlewares[h.scene]...) handlers = append(handlers, middlewares[h.scene]...)
handlers = append(handlers, middlewares[constraints.AllScene]...) handlers = append(handlers, middlewares[constraints.AllScene]...)
handlers = h.PipeHandleHook("PipeMiddlewareHandle", handlers) handlers = h.PipeHandleHook("PipeMiddlewareHandle", handlers, key)
return return
} }
func PipeDataHandle(h *Handle, dataHandlers map[string][]HandleCall) (handlers []HandleCall) { func PipeDataHandle(h *Handle, dataHandlers map[string][]HandleCall, key string) (handlers []HandleCall) {
handlers = append(handlers, dataHandlers[h.scene]...) handlers = append(handlers, dataHandlers[h.scene]...)
handlers = append(handlers, dataHandlers[constraints.AllScene]...) handlers = append(handlers, dataHandlers[constraints.AllScene]...)
handlers = h.PipeHandleHook("PipeDataHandle", handlers) handlers = h.PipeHandleHook("PipeDataHandle", handlers, key)
return return
} }
func PipeRender(h *Handle, renders map[string][]HandleCall) (handlers []HandleCall) { func PipeRender(h *Handle, renders map[string][]HandleCall, key string) (handlers []HandleCall) {
handlers = append(handlers, renders[h.Stats]...) handlers = append(handlers, renders[h.Stats]...)
handlers = append(handlers, renders[h.scene]...) handlers = append(handlers, renders[h.scene]...)
handlers = append(handlers, renders[constraints.AllStats]...) handlers = append(handlers, renders[constraints.AllStats]...)
handlers = append(handlers, renders[constraints.AllScene]...) handlers = append(handlers, renders[constraints.AllScene]...)
handlers = h.PipeHandleHook("PipeRender", handlers) handlers = h.PipeHandleHook("PipeRender", handlers, key)
return return
} }
@ -184,10 +185,10 @@ func (h *Handle) PushPipeHandleHook(name string, fn ...func([]HandleCall) []Hand
return PushFnHook("pipeHandleHook", name, fn...) return PushFnHook("pipeHandleHook", name, fn...)
} }
func (h *Handle) PipeHandleHook(name string, calls []HandleCall) []HandleCall { func (h *Handle) PipeHandleHook(name string, calls []HandleCall, key string) []HandleCall {
fn := GetFnHook[func(*Handle, []HandleCall) []HandleCall]("pipeHandleHook", name) fn := GetFnHook[func(*Handle, []HandleCall, string) []HandleCall]("pipeHandleHook", name)
return slice.Reduce(fn, func(t func(*Handle, []HandleCall) []HandleCall, r []HandleCall) []HandleCall { return slice.Reduce(fn, func(t func(*Handle, []HandleCall, string) []HandleCall, r []HandleCall) []HandleCall {
return t(h, r) return t(h, r, key)
}, calls) }, calls)
} }