Free bitmap in case of error

ydb-platform · Jan 9, 2025 · cce62ff · cce62ff
1 parent e14fa40
commit cce62ff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ydb/library/yql/udfs/common/roaring/roaring.cpp b/ydb/library/yql/udfs/common/roaring/roaring.cpp
@@ -285,9 +285,8 @@ namespace {
         TUnboxedValue Run(const IValueBuilder* valueBuilder,
                           const TUnboxedValuePod* args) const override {
             Y_UNUSED(valueBuilder);
+            auto* bitmap = roaring_bitmap_create();
             try {
-                auto* bitmap = roaring_bitmap_create();
-
                 const auto vector = args[0];
                 const auto* elements = vector.GetElements();
                 if (elements) {
@@ -304,6 +303,7 @@ namespace {
 
                 return TUnboxedValuePod(new TRoaringWrapper(bitmap));
             } catch (const std::exception& e) {
+                roaring_bitmap_free(bitmap);
                 UdfTerminate((TStringBuilder() << Pos_ << " " << e.what()).data());
             }
         }