ydb/library/yql/public/udf/support/udf_support.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63

#define BUILD_UDF
#include <util/system/backtrace.h>

#if defined(_win_) || defined(_darwin_) 
#include <ydb/library/yql/public/udf/udf_registrator.h>

#include <exception>

static NYql::NUdf::TStaticSymbols Symbols; 

extern "C" void* UdfAllocate(ui64 size) {
    return Symbols.UdfAllocateFunc(size);
}

extern "C" void UdfFree(const void* mem) { 
    return Symbols.UdfFreeFunc(mem);
}

extern "C" [[noreturn]] void UdfTerminate(const char* message) {
    Symbols.UdfTerminate(message); 
    std::terminate();
} 
 
extern "C" void UdfRegisterObject(::NYql::NUdf::TBoxedValue* object) { 
    return Symbols.UdfRegisterObject(object); 
} 
 
extern "C" void UdfUnregisterObject(::NYql::NUdf::TBoxedValue* object) { 
    return Symbols.UdfUnregisterObject(object); 
} 
 
#if UDF_ABI_COMPATIBILITY_VERSION_CURRENT >= UDF_ABI_COMPATIBILITY_VERSION(2, 8)
extern "C" void* UdfAllocateWithSize(ui64 size) {
    return Symbols.UdfAllocateWithSizeFunc(size);
}

extern "C" void UdfFreeWithSize(const void* mem, ui64 size) { 
    return Symbols.UdfFreeWithSizeFunc(mem, size);
}
#endif

extern "C" void BindSymbols(const NYql::NUdf::TStaticSymbols& symbols) { 
    Symbols = symbols;
}
#endif

namespace NYql { 
namespace NUdf {

typedef void(*TBackTraceCallback)();
static TBackTraceCallback BackTraceCallback;

static void UdfBackTraceFn(IOutputStream*, void* const*, size_t) {
    BackTraceCallback();
}

void SetBackTraceCallbackImpl(TBackTraceCallback callback) {
    BackTraceCallback = callback;
    SetFormatBackTraceFn(UdfBackTraceFn);
}

}
}