package/mq
1
0
Fork 0
mirror of https://github.com/oarkflow/mq.git synced 2025-10-17 04:41:54 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Oarkflow
2025-09-07 18:59:10 +05:45
parent b831cb09bd
commit 06a935ce2f
1 changed files with 28 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
services/setup.go
View File

@@ -431,23 +431,43 @@ func prepareHeader(ctx *fiber.Ctx, route *Route) (map[string]any, map[string]any
func customHandler(flow *dag.DAG) fiber.Handler { func customHandler(flow *dag.DAG) fiber.Handler {
return func(ctx *fiber.Ctx) error { return func(ctx *fiber.Ctx) error {
// Step 1: always parse query params
utils.ParseQueryParams(ctx) utils.ParseQueryParams(ctx)
result := flow.Process(ctx.UserContext(), ctx.BodyRaw())
// Step 2: build user context
userCtx := ctx.UserContext()
contentType := ctx.Get("Content-Type")
if strings.HasPrefix(contentType, "multipart/form-data") {
// attach Fiber ctx so downstream can access files later
userCtx = context.WithValue(userCtx, "fiberCtx", ctx)
}
// Step 3: run DAG flow with the enriched context
result := flow.Process(userCtx, ctx.BodyRaw())
if result.Error != nil { if result.Error != nil {
return result.Error return result.Error
} }
contentType := result.Ctx.Value(consts.ContentType)
if contentType == nil { // Step 4: handle response content type
return ctx.JSON(result) contentType = ""
} if ct := result.Ctx.Value(consts.ContentType); ct != nil {
if contentType == fiber.MIMEApplicationJSON || contentType == fiber.MIMEApplicationJSONCharsetUTF8 { if s, ok := ct.(string); ok {
contentType = s
}
}
if contentType == "" ||
contentType == fiber.MIMEApplicationJSON ||
contentType == fiber.MIMEApplicationJSONCharsetUTF8 {
return ctx.JSON(result) return ctx.JSON(result)
} }
var resultData map[string]any var resultData map[string]any
if err := json.Unmarshal(result.Payload, &resultData); err != nil { if err := json.Unmarshal(result.Payload, &resultData); err != nil {
return ctx.JSON(fiber.Map{"success": false, "error": "Invalid response payload"}) return ctx.JSON(fiber.Map{"success": false, "error": "Invalid response payload"})
} }
ctx.Set(consts.ContentType, contentType.(string))
ctx.Set(consts.ContentType, contentType)
html, _ := resultData["html_content"].(string) html, _ := resultData["html_content"].(string)
return ctx.SendString(html) return ctx.SendString(html)
} }
@@ -515,7 +535,7 @@ func setupRender(_ string, clientRoutes fiber.Router) error {
// requestMiddleware validates the request body in the original form of byte array // requestMiddleware validates the request body in the original form of byte array
// against the provided request JSON schema to ensure that the request body is valid. // against the provided request JSON schema to ensure that the request body is valid.
func requestMiddleware(prefix string, route *Route) fiber.Handler { func requestMiddleware(_ string, route *Route) fiber.Handler {
return func(ctx *fiber.Ctx) error { return func(ctx *fiber.Ctx) error {
schema := route.GetSchema() schema := route.GetSchema()
if schema == nil { if schema == nil {