r3/request/request_pgFunction.go

package request

import (
	"context"
	"encoding/json"
	"fmt"
	"r3/cache"
	"r3/handler"
	"r3/schema/pgFunction"
	"r3/types"
	"strings"

	"github.com/gofrs/uuid"
	"github.com/jackc/pgx/v5"
)

func PgFunctionDel_tx(ctx context.Context, tx pgx.Tx, reqJson json.RawMessage) (interface{}, error) {

	var req struct {
		Id uuid.UUID `json:"id"`
	}

	if err := json.Unmarshal(reqJson, &req); err != nil {
		return nil, err
	}
	return nil, pgFunction.Del_tx(ctx, tx, req.Id)
}

func PgFunctionExec_tx(ctx context.Context, tx pgx.Tx, reqJson json.RawMessage, onlyFrontendFnc bool) (interface{}, error) {
	cache.Schema_mx.RLock()
	defer cache.Schema_mx.RUnlock()

	var req struct {
		Id   uuid.UUID     `json:"id"`
		Args []interface{} `json:"args"`
	}

	if err := json.Unmarshal(reqJson, &req); err != nil {
		return nil, err
	}

	fnc, exists := cache.PgFunctionIdMap[req.Id]
	if !exists {
		return nil, handler.ErrSchemaUnknownPgFunction(req.Id)
	}
	if fnc.IsTrigger {
		return nil, handler.ErrSchemaTriggerPgFunctionCall(req.Id)
	}
	if onlyFrontendFnc && !fnc.IsFrontendExec {
		return nil, handler.ErrSchemaBadFrontendExecPgFunctionCall(req.Id)
	}

	mod := cache.ModuleIdMap[fnc.ModuleId]

	placeholders := make([]string, 0)
	for i, _ := range req.Args {
		placeholders = append(placeholders, fmt.Sprintf("$%d", i+1))
	}

	var returnIf interface{}
	if err := tx.QueryRow(ctx, fmt.Sprintf(`
		SELECT "%s"."%s"(%s)
	`, mod.Name, fnc.Name, strings.Join(placeholders, ",")),
		req.Args...).Scan(&returnIf); err != nil {

		return nil, err
	}
	return returnIf, nil
}

func PgFunctionSet_tx(ctx context.Context, tx pgx.Tx, reqJson json.RawMessage) (interface{}, error) {

	var req types.PgFunction

	if err := json.Unmarshal(reqJson, &req); err != nil {
		return nil, err
	}
	return nil, pgFunction.Set_tx(ctx, tx, req)
}