How to specify BeamSQL UDF for Numeric Types

Question

I'm trying to add a User Defined Function (UDF) to a SqlTransform in a Beam pipeline, and the SQL parser doesn't seem to understand the function's type. The error i get is:

No match found for function signature IF(<BOOLEAN>, <NUMERIC>, <NUMERIC>)

I've tried defining this for a bunch of different types (Double, Float, Long, Integer), but don't see how i can write this generically in a way that Beam will understand.

Is there some way to give type hints to the UDF, to to write one that handles arbitrary numeric types?

Answer 1

Seems, you should use BigDecimal.

Here is a similar question: What is the equivalent Data type for Numeric in apache.beam.sdk.schemas.Schema.FieldType

Answer 2

I ended up finding out the problem was with my code registering the UDF. The error message you get from not registering the UDF appears to be identical to the message you'd get if you wrote one with incompatible types. Just using 'Double' seemed to work fine.