52 lines
		
	
	
		
			1.1 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			52 lines
		
	
	
		
			1.1 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| package controller
 | |
| 
 | |
| import (
 | |
| 	"bytes"
 | |
| 	"encoding/json"
 | |
| 	"fmt"
 | |
| 	"github.com/loveuer/nf-disk/internal/api"
 | |
| 	"github.com/loveuer/nf-disk/internal/opt"
 | |
| 	"github.com/loveuer/nf-disk/internal/tool"
 | |
| 	"github.com/loveuer/nf-disk/ndh"
 | |
| 	"github.com/loveuer/nf/nft/log"
 | |
| 	"strings"
 | |
| )
 | |
| 
 | |
| func handleError(err error) string {
 | |
| 	bs, _ := json.Marshal(map[string]any{
 | |
| 		"err":    err.Error(),
 | |
| 		"msg":    opt.Msg500,
 | |
| 		"status": 500,
 | |
| 	})
 | |
| 
 | |
| 	return string(bs)
 | |
| }
 | |
| 
 | |
| func handleNotFound(path string) string {
 | |
| 	bs, _ := json.Marshal(map[string]any{
 | |
| 		"err":    fmt.Sprintf("path not found, path: %s", path),
 | |
| 		"msg":    opt.Msg500,
 | |
| 		"status": 404,
 | |
| 	})
 | |
| 
 | |
| 	return string(bs)
 | |
| }
 | |
| 
 | |
| func (a *App) Invoke(path string, req string) (res string) {
 | |
| 	log.Info("app invoke: path = %s, req = %s", path, req)
 | |
| 	handler, ok := api.Resolve(path)
 | |
| 	if !ok {
 | |
| 		log.Warn("app invoke: path not found, path = %s", path)
 | |
| 		return handleNotFound(path)
 | |
| 	}
 | |
| 
 | |
| 	var buf bytes.Buffer
 | |
| 	ctx := ndh.NewCtx(tool.TimeoutCtx(a.ctx), strings.NewReader(req), &buf)
 | |
| 
 | |
| 	if err := handler(ctx); err != nil {
 | |
| 		return handleError(err)
 | |
| 	}
 | |
| 
 | |
| 	return buf.String()
 | |
| }
 |