Add predicates for bulk and text family bulk

groonga · Feb 3, 2017 · 438d3f3 · 438d3f3
1 parent fea9674
commit 438d3f3
Show file tree

Hide file tree

Showing 3 changed files with 62 additions and 0 deletions.
diff --git a/include/groonga/obj.h b/include/groonga/obj.h
@@ -31,6 +31,8 @@ extern "C" {
 GRN_API grn_bool grn_obj_is_true(grn_ctx *ctx, grn_obj *obj);
 
 GRN_API grn_bool grn_obj_is_builtin(grn_ctx *ctx, grn_obj *obj);
+GRN_API grn_bool grn_obj_is_bulk(grn_ctx *ctx, grn_obj *obj);
+GRN_API grn_bool grn_obj_is_text_family_bulk(grn_ctx *ctx, grn_obj *obj);
 GRN_API grn_bool grn_obj_is_table(grn_ctx *ctx, grn_obj *obj);
 GRN_API grn_bool grn_obj_is_column(grn_ctx *ctx, grn_obj *obj);
 GRN_API grn_bool grn_obj_is_vector_column(grn_ctx *ctx, grn_obj *obj);

diff --git a/lib/obj.c b/lib/obj.c
@@ -75,6 +75,26 @@ grn_obj_is_builtin(grn_ctx *ctx, grn_obj *obj)
   return grn_id_is_builtin(ctx, id);
 }
 
+grn_bool
+grn_obj_is_bulk(grn_ctx *ctx, grn_obj *obj)
+{
+  if (!obj) {
+    return GRN_FALSE;
+  }
+
+  return obj->header.type == GRN_BULK;
+}
+
+grn_bool
+grn_obj_is_text_family_bulk(grn_ctx *ctx, grn_obj *obj)
+{
+  if (!grn_obj_is_bulk(ctx, obj)) {
+    return GRN_FALSE;
+  }
+
+  return GRN_TYPE_IS_TEXT_FAMILY(obj->header.domain);
+}
+
 grn_bool
 grn_obj_is_table(grn_ctx *ctx, grn_obj *obj)
 {

diff --git a/test/unit/core/test-object.c b/test/unit/core/test-object.c
@@ -27,6 +27,8 @@
 
 void data_is_builtin(void);
 void test_is_builtin(gconstpointer data);
+void test_is_bulk(void);
+void test_is_text_family_bulk(void);
 void data_is_table(void);
 void test_is_table(gconstpointer data);
 void data_is_column(void);
@@ -158,6 +160,44 @@ test_is_builtin(gconstpointer data)
   }
 }
 
+void
+test_is_bulk(void)
+{
+  grn_obj bulk;
+  grn_obj *tokenizer;
+
+  GRN_UINT32_INIT(&bulk, 0);
+  cut_assert_true(grn_obj_is_bulk(context, &bulk));
+  GRN_OBJ_FIN(context, &bulk);
+
+  tokenizer = grn_ctx_get(context, "TokenBigram", -1);
+  cut_assert_false(grn_obj_is_bulk(context, tokenizer));
+}
+
+void
+test_is_text_family_bulk(void)
+{
+  grn_obj uint32_bulk;
+  grn_obj short_text_bulk;
+  grn_obj text_bulk;
+  grn_obj long_text_bulk;
+
+  GRN_UINT32_INIT(&uint32_bulk, 0);
+  GRN_SHORT_TEXT_INIT(&short_text_bulk, 0);
+  GRN_TEXT_INIT(&text_bulk, 0);
+  GRN_LONG_TEXT_INIT(&long_text_bulk, 0);
+
+  cut_assert_false(grn_obj_is_text_family_bulk(context, &uint32_bulk));
+  cut_assert_true(grn_obj_is_text_family_bulk(context, &short_text_bulk));
+  cut_assert_true(grn_obj_is_text_family_bulk(context, &text_bulk));
+  cut_assert_true(grn_obj_is_text_family_bulk(context, &long_text_bulk));
+
+  GRN_OBJ_FIN(context, &uint32_bulk);
+  GRN_OBJ_FIN(context, &short_text_bulk);
+  GRN_OBJ_FIN(context, &text_bulk);
+  GRN_OBJ_FIN(context, &long_text_bulk);
+}
+
 void
 data_is_table(void)
 {