/*
* Document-method: search
*
* call-seq:
* column.search(query, options={}) -> Groonga::Hash
*
* _object_から_query_に対応するオブジェクトを検索し、見つかっ
* たオブジェクトのIDがキーになっているGroonga::Hashを返す。
*
* 利用可能なオプションは以下の通り。
*
* [_:result_]
* 結果を格納するGroonga::Hash。指定しない場合は新しく
* Groonga::Hashを生成し、それに結果を格納して返す。
* [_:operator_]
* 以下のどれかの値を指定する。+nil+, <tt>"or"</tt>, <tt>"||"</tt>,
* <tt>"and"</tt>, <tt>"+"</tt>, <tt>"&&"</tt>, <tt>"but"</tt>,
* <tt>"not"</tt>, <tt>"-"</tt>, <tt>"adjust"</tt>, <tt>">"</tt>。
* それぞれ以下のようになる。（FIXME: 「以下」）
* [_:exact_]
* +true+を指定すると完全一致で検索する
* [_:longest_common_prefix_]
* +true+を指定すると_query_と同じ接頭辞をもつエントリのう
* ち、もっとも長いエントリを検索する
* [_:suffix_]
* +true+を指定すると_query_が後方一致するエントリを検索す
* る
* [_:prefix_]
* +true+を指定すると_query_が前方一致するレコードを検索す
* る
* [_:near_]
* +true+を指定すると_query_に指定した複数の語が近傍に含ま
* れるレコードを検索する
* [...]
* ...
*/
static VALUE
rb_grn_index_column_search (int argc, VALUE *argv, VALUE self)
{
grn_ctx *context;
grn_obj *column;
grn_obj *range;
grn_obj *query = NULL, *id_query = NULL, *string_query = NULL;
grn_obj *result;
grn_operator operator;
grn_rc rc;
VALUE rb_query, options, rb_result, rb_operator;
rb_grn_index_column_deconstruct(SELF(self), &column, &context,
NULL, NULL,
NULL, NULL, NULL, &range,
&id_query, &string_query);
rb_scan_args(argc, argv, "11", &rb_query, &options);
if (CBOOL2RVAL(rb_obj_is_kind_of(rb_query, rb_cGrnQuery))) {
grn_query *_query;
_query = RVAL2GRNQUERY(rb_query);
query = (grn_obj *)_query;
} else if (CBOOL2RVAL(rb_obj_is_kind_of(rb_query, rb_cInteger))) {
grn_id id;
id = NUM2UINT(rb_query);
GRN_TEXT_SET(context, id_query, &id, sizeof(grn_id));
query = id_query;
} else {
const char *_query;
_query = StringValuePtr(rb_query);
GRN_TEXT_SET(context, string_query, _query, RSTRING_LEN(rb_query));
query = string_query;
}
rb_grn_scan_options(options,
"result", &rb_result,
"operator", &rb_operator,
NULL);
if (NIL_P(rb_result)) {
result = grn_table_create(context, NULL, 0, NULL,
GRN_OBJ_TABLE_HASH_KEY | GRN_OBJ_WITH_SUBREC,
range, 0);
rb_grn_context_check(context, self);
rb_result = GRNOBJECT2RVAL(Qnil, context, result, RB_GRN_TRUE);
} else {
result = RVAL2GRNOBJECT(rb_result, &context);
}
operator = RVAL2GRNOPERATOR(rb_operator);
rc = grn_obj_search(context, column, query, result, operator, NULL);
rb_grn_rc_check(rc, self);
return rb_result;
}
grn_obj *
rb_grn_type_from_ruby_object (VALUE object)
{
if (!RVAL2CBOOL(rb_obj_is_kind_of(object, rb_cGrnType))) {
rb_raise(rb_eTypeError, "not a groonga type");
}
return RVAL2GRNOBJECT(object, NULL);
}
grn_obj *
rb_grn_column_from_ruby_object (VALUE object, grn_ctx **context)
{
if (!RVAL2CBOOL(rb_obj_is_kind_of(object, rb_cGrnColumn))) {
rb_raise(rb_eTypeError, "not a groonga column");
}
return RVAL2GRNOBJECT(object, context);
}
static VALUE
rb_grn_type_geo_point_p (VALUE self)
{
grn_obj *type;
grn_obj_flags key_type;
type = RVAL2GRNOBJECT(self, NULL);
key_type = type->header.flags & GRN_OBJ_KEY_MASK;
return CBOOL2RVAL(key_type == GRN_OBJ_KEY_GEO_POINT);
}
static VALUE
rb_grn_type_unsigned_integer_p (VALUE self)
{
grn_obj *type;
grn_obj_flags key_type;
type = RVAL2GRNOBJECT(self, NULL);
key_type = type->header.flags & GRN_OBJ_KEY_MASK;
return CBOOL2RVAL(key_type == GRN_OBJ_KEY_UINT);
}
static VALUE
rb_grn_type_variable_size_p (VALUE self)
{
grn_ctx *context;
grn_obj *type;
type = RVAL2GRNOBJECT(self, &context);
return CBOOL2RVAL(type->header.flags & GRN_OBJ_KEY_VAR_SIZE);
}
static VALUE
rb_grn_type_flags (VALUE self)
{
grn_ctx *context;
grn_obj *type;
type = RVAL2GRNOBJECT(self, &context);
return UINT2NUM(type->header.flags);
}
static VALUE
rb_grn_type_size (VALUE self)
{
grn_ctx *context;
grn_obj *type;
type = RVAL2GRNOBJECT(self, &context);
return UINT2NUM(grn_obj_get_range(context, type));
}
grn_obj *
rb_grn_obj_from_ruby_object (VALUE rb_object, grn_ctx *context, grn_obj **_obj)
{
if (RVAL2CBOOL(rb_obj_is_kind_of(rb_object, rb_cGrnObject))) {
if (*_obj) {
grn_obj_unlink(context, *_obj); /* TODO: reduce memory allocation */
}
*_obj = RVAL2GRNOBJECT(rb_object, &context);
} else {
*_obj = RVAL2GRNBULK(rb_object, context, *_obj);
}
return *_obj;
}
/*
* _key_ にマッチするレコードのIDがキーに入っている
* {Groonga::Hash} を返す。マッチするレコードがない場合は空の
* {fGroonga::Hash} が返る。
* _options_ で +:result+ を指定することにより、そのテーブルにマッ
* チしたレコードIDがキーのレコードを追加することができる。
* +:result+ にテーブルを指定した場合は、そのテーブルが返る。
*
* @example 複数のキーで検索し、結果を1つのテーブルに集める。
* result = nil
* keys = ["morita", "gunyara-kun", "yu"]
* keys.each do |key|
* result = users.search(key, :result => result)
* end
* result.each do |record|
* user = record.key
* p user.key # -> "morita"または"gunyara-kun"または"yu"
* end
*
* @overload search(key, options=nil)
* @return [Groonga::Hash]
* @param key [String] レコードにマッチさせる値
* @param options [::Hash] The name and value
* pairs. Omitted names are initialized as the default value.
* @option options :result The result
* 結果を格納するテーブル。
* @option options :operator (Groonga::Operator::OR)
* マッチしたレコードをどのように扱うか。指定可能な値は以
* 下の通り。
*
* - Groonga::Operator::OR :=
* マッチしたレコードを追加。すでにレコードが追加され
* ている場合は何もしない。 =:
* - Groonga::Operator::AND :=
* マッチしたレコードのスコアを増加。マッチしなかった
* レコードを削除。 =:
* - Groonga::Operator::AND_NOT :=
* マッチしたレコードを削除。 =:
* - Groonga::Operator::ADJUST :=
* マッチしたレコードのスコアを増加。 =:
* - +:type+ :=
* ????? =:
*
*/
static VALUE
rb_grn_double_array_trie_search (int argc, VALUE *argv, VALUE self)
{
grn_rc rc;
grn_ctx *context;
grn_obj *table;
grn_id domain_id;
grn_obj *key, *domain, *result;
grn_operator operator;
grn_search_optarg search_options;
grn_bool search_options_is_set = GRN_FALSE;
VALUE rb_key, options, rb_result, rb_operator, rb_type;
rb_grn_double_array_trie_deconstruct(SELF(self), &table, &context,
&key, NULL, &domain_id, &domain,
NULL, NULL, NULL,
NULL);
rb_scan_args(argc, argv, "11", &rb_key, &options);
RVAL2GRNKEY(rb_key, context, key, domain_id, domain, self);
rb_grn_scan_options(options,
"result", &rb_result,
"operator", &rb_operator,
"type", &rb_type,
NULL);
if (NIL_P(rb_result)) {
result = grn_table_create(context, NULL, 0, NULL,
GRN_OBJ_TABLE_HASH_KEY | GRN_OBJ_WITH_SUBREC,
table, 0);
rb_grn_context_check(context, self);
rb_result = GRNOBJECT2RVAL(Qnil, context, result, GRN_TRUE);
} else {
result = RVAL2GRNOBJECT(rb_result, &context);
}
operator = RVAL2GRNOPERATOR(rb_operator);
rc = grn_obj_search(context, table, key,
result, operator,
search_options_is_set ? &search_options : NULL);
rb_grn_rc_check(rc, self);
return rb_result;
}
static grn_id
resolve_source_id (grn_ctx *context, grn_obj *column, VALUE rb_source)
{
grn_id source_id;
if (CBOOL2RVAL(rb_obj_is_kind_of(rb_source, rb_cInteger))) {
source_id = NUM2UINT(rb_source);
} else {
grn_obj *source;
if (TYPE(rb_source) == T_STRING) {
grn_obj *table;
const char *name;
const char *dot_point;
int length;
table = grn_ctx_at(context, grn_obj_get_range(context, column));
name = StringValueCStr(rb_source);
length = RSTRING_LEN(rb_source);
dot_point = strstr(name, ".");
if (dot_point) {
char table_name[4096];
int table_name_length;
table_name_length = grn_obj_name(context, table,
table_name, sizeof(table_name));
table_name[table_name_length] = '\0';
if (strncmp(table_name, name, dot_point - name) != 0) {
rb_raise(rb_eArgError,
"wrong table's column: <%s>: "
"expected table: <%s>",
name, table_name);
}
length -= (dot_point - name) + 1;
name = dot_point + 1;
}
source = grn_obj_column(context, table, name, length);
} else {
source = RVAL2GRNOBJECT(rb_source, &context);
}
rb_grn_context_check(context, rb_source);
source_id = grn_obj_id(context, source);
}
return source_id;
}
/*
* _object_ を追加し、 _n_arguments_ 個の引数を取る _operation_ を追加する。
*
* @overload append_object(object, operation=Groonga::Operator::PUSH, n_arguments=1)
* @param [Object] object 追加するオブジェクト
* @param [Groonga::Operator::XXX] operation 追加する _operation_
* @param [Integer] n_arguments _operation_ の取る引数
* @return [Self] self
*
*/
static VALUE
rb_grn_expression_append_object (int argc, VALUE *argv, VALUE self)
{
VALUE rb_object, rb_operation, rb_n_arguments;
grn_ctx *context = NULL;
grn_obj *expression, *object;
grn_operator operation = GRN_OP_PUSH;
int n_arguments = 1;
rb_scan_args(argc, argv, "12", &rb_object, &rb_operation, &rb_n_arguments);
if (!NIL_P(rb_operation))
operation = RVAL2GRNOPERATOR(rb_operation);
if (!NIL_P(rb_n_arguments))
n_arguments = NUM2INT(rb_n_arguments);
rb_grn_expression_deconstruct(SELF(self), &expression, &context,
NULL, NULL,
NULL, NULL, NULL);
if (RB_TYPE_P(rb_object, RUBY_T_HASH)) {
RbGrnHashFromRubyHashData data;
data.context = context;
data.hash = grn_hash_create(context, NULL,
GRN_TABLE_MAX_KEY_SIZE,
sizeof(grn_obj),
GRN_OBJ_KEY_VAR_SIZE |
GRN_OBJ_TEMPORARY |
GRN_HASH_TINY);
grn_expr_take_obj(context, expression, (grn_obj *)(data.hash));
data.rb_hash = rb_object;
rb_hash_foreach(rb_object,
rb_grn_hash_from_ruby_hash_body,
(VALUE)&data);
grn_expr_append_obj(context, expression, (grn_obj *)(data.hash),
operation, n_arguments);
} else {
object = RVAL2GRNOBJECT(rb_object, &context);
grn_expr_append_obj(context, expression, object,
operation, n_arguments);
}
rb_grn_context_check(context, self);
rb_ary_push(rb_iv_get(self, "@objects"), rb_object);
return self;
}
/*
* Specifies the normalizer used by {Groonga::IndexColumn}.
*
* @example
* # Uses NFKC normalizer.
* table.normalizer = "NormalizerNFKC51"
* # Specifies normalizer object.
* table.normalizer = Groonga::Context["NormalizerNFKC51"]
* # Uses auto normalizer that is a normalizer for backward compatibility.
* table.normalizer = "TNormalizerAuto"
*
* @overload normalizer=(name)
* @param [String] name Set a nomalizer named @name@.
*
* @overload normalizer=(normalizer)
* @param [Groonga::Procedure] normalizer Set the normalizer object.
*
* @overload normalizer=(normalizer)
* @param [nil] normalizer Unset normalizer.
*/
static VALUE
rb_grn_table_key_support_set_normalizer (VALUE self, VALUE rb_normalizer)
{
grn_ctx *context;
grn_obj *table;
grn_obj *normalizer;
grn_rc rc;
rb_grn_table_key_support_deconstruct(SELF(self), &table, &context,
NULL, NULL, NULL,
NULL, NULL, NULL,
NULL);
normalizer = RVAL2GRNOBJECT(rb_normalizer, &context);
rc = grn_obj_set_info(context, table, GRN_INFO_NORMALIZER, normalizer);
rb_grn_context_check(context, self);
rb_grn_rc_check(rc, self);
return Qnil;
}
/*
* call-seq:
* view.add_table(table)
*
* _table_をビューからアクセスできるようにする。
*/
static VALUE
rb_grn_view_add_table (VALUE self, VALUE rb_table)
{
#ifdef WIN32
rb_raise(rb_eNotImpError, "grn_view_add() isn't available on Windows.");
#else
grn_ctx *context = NULL;
grn_obj *view, *table;
rb_grn_table_deconstruct(SELF(self), &view, &context,
NULL, NULL,
NULL, NULL, NULL,
NULL);
table = RVAL2GRNOBJECT(rb_table, &context);
grn_view_add(context, view, table);
rb_grn_context_check(context, self);
#endif
return Qnil;
}
/*
* {Groonga::IndexColumn} で使用するトークナイザを設定する。
*
* @example
* # 2-gramを使用。
* table.default_tokenizer = "TokenBigram"
* # オブジェクトで指定
* table.default_tokenizer = Groonga::Context.default["TokenBigram"]
* # オブジェクトIDで指定
* table.default_tokenizer = Groonga::Type::BIGRAM
* # N-gram用のトークナイザを使うときはGroonga::IndexColumn
* # には自動的に:with_section => trueが指定される。
* index = table.define_index_column("blog_content", "Blogs",
* :source => "content")
* p index # -> #<Groonga::IndexColumn ... flags: <WITH_POSITION|...>>
*
* # MeCabを使用
* table.default_tokenizer = "TokenMecab"
*
* @overload default_tokenizer=(tokenizer)
*/
static VALUE
rb_grn_table_key_support_set_default_tokenizer (VALUE self, VALUE rb_tokenizer)
{
grn_ctx *context;
grn_obj *table;
grn_obj *tokenizer;
grn_rc rc;
rb_grn_table_key_support_deconstruct(SELF(self), &table, &context,
NULL, NULL, NULL,
NULL, NULL, NULL,
NULL);
tokenizer = RVAL2GRNOBJECT(rb_tokenizer, &context);
rc = grn_obj_set_info(context, table,
GRN_INFO_DEFAULT_TOKENIZER, tokenizer);
rb_grn_context_check(context, self);
rb_grn_rc_check(rc, self);
return Qnil;
}
/*
* _key_ にマッチするレコードのIDがキーに入っている
* {Groonga::Hash} を返す。マッチするレコードがない場合は空の
* {Groonga::Hash} が返る。
*
* _options_ で +:result+ を指定することにより、そのテーブルにマッ
* チしたレコードIDがキーのレコードを追加することができる。
* +:result+ にテーブルを指定した場合は、そのテーブルが返る。
*
* @example 複数のキーで検索し、結果を1つのテーブルに集める。
* result = nil
* keys = ["morita", "gunyara-kun", "yu"]
* keys.each do |key|
* result = users.search(key, :result => result)
* end
* result.each do |record|
* user = record.key
* p user.key # -> "morita"または"gunyara-kun"または"yu"
* end
*
* @overload search(key, options=nil)
* @param [::Hash] options The name and value
* pairs. Omitted names are initialized as the default value
* @option options :result
* 結果を格納するテーブル。
* @return [Groonga::Hash]
*/
static VALUE
rb_grn_hash_search (int argc, VALUE *argv, VALUE self)
{
grn_rc rc;
grn_ctx *context;
grn_obj *table;
grn_id domain_id;
grn_obj *key, *domain, *result;
VALUE rb_key, options, rb_result;
rb_grn_table_key_support_deconstruct(SELF(self), &table, &context,
&key, &domain_id, &domain,
NULL, NULL, NULL,
NULL);
rb_scan_args(argc, argv, "11", &rb_key, &options);
RVAL2GRNKEY(rb_key, context, key, domain_id, domain, self);
rb_grn_scan_options(options,
"result", &rb_result,
NULL);
if (NIL_P(rb_result)) {
result = grn_table_create(context, NULL, 0, NULL,
GRN_OBJ_TABLE_HASH_KEY | GRN_OBJ_WITH_SUBREC,
table, 0);
rb_grn_context_check(context, self);
rb_result = GRNOBJECT2RVAL(Qnil, context, result, GRN_TRUE);
} else {
result = RVAL2GRNOBJECT(rb_result, &context);
}
rc = grn_obj_search(context, table, key,
result, GRN_OP_OR, NULL);
rb_grn_rc_check(rc, self);
return rb_result;
}
/* FIXME: DON'T WORK!!! */
static VALUE
rb_grn_view_sort (int argc, VALUE *argv, VALUE self)
{
VALUE rb_result = Qnil;
#ifdef WIN32
rb_raise(rb_eNotImpError, "grn_view_add() isn't available on Windows.");
#else
grn_ctx *context = NULL;
grn_obj *view;
grn_obj *result;
grn_table_sort_key *keys;
int i, n_keys;
int n_records, offset = 0, limit = -1;
VALUE rb_keys, options;
VALUE rb_offset, rb_limit;
VALUE *rb_sort_keys;
grn_table_cursor *cursor;
VALUE exception;
grn_obj id;
rb_grn_table_deconstruct(SELF(self), &view, &context,
NULL, NULL,
NULL, NULL, NULL,
NULL);
rb_scan_args(argc, argv, "11", &rb_keys, &options);
if (!RVAL2CBOOL(rb_obj_is_kind_of(rb_keys, rb_cArray)))
rb_raise(rb_eArgError, "keys should be an array of key: <%s>",
rb_grn_inspect(rb_keys));
n_keys = RARRAY_LEN(rb_keys);
rb_sort_keys = RARRAY_PTR(rb_keys);
keys = ALLOCA_N(grn_table_sort_key, n_keys);
for (i = 0; i < n_keys; i++) {
VALUE rb_sort_options, rb_key, rb_resolved_key, rb_order;
if (RVAL2CBOOL(rb_obj_is_kind_of(rb_sort_keys[i], rb_cHash))) {
rb_sort_options = rb_sort_keys[i];
} else if (RVAL2CBOOL(rb_obj_is_kind_of(rb_sort_keys[i], rb_cArray))) {
rb_sort_options = rb_hash_new();
rb_hash_aset(rb_sort_options,
RB_GRN_INTERN("key"),
rb_ary_entry(rb_sort_keys[i], 0));
rb_hash_aset(rb_sort_options,
RB_GRN_INTERN("order"),
rb_ary_entry(rb_sort_keys[i], 1));
} else {
rb_sort_options = rb_hash_new();
rb_hash_aset(rb_sort_options,
RB_GRN_INTERN("key"),
rb_sort_keys[i]);
}
rb_grn_scan_options(rb_sort_options,
"key", &rb_key,
"order", &rb_order,
NULL);
if (RVAL2CBOOL(rb_obj_is_kind_of(rb_key, rb_cString))) {
rb_resolved_key = rb_grn_table_get_column(self, rb_key);
} else {
rb_resolved_key = rb_key;
}
keys[i].key = RVAL2GRNOBJECT(rb_resolved_key, &context);
if (!keys[i].key) {
rb_raise(rb_eGrnNoSuchColumn,
"no such column: <%s>: <%s>",
rb_grn_inspect(rb_key), rb_grn_inspect(self));
}
if (NIL_P(rb_order)) {
keys[i].flags = 0;
} else if (rb_grn_equal_option(rb_order, "desc") ||
rb_grn_equal_option(rb_order, "descending")) {
keys[i].flags = GRN_TABLE_SORT_DESC;
} else if (rb_grn_equal_option(rb_order, "asc") ||
rb_grn_equal_option(rb_order, "ascending")) {
keys[i].flags = GRN_TABLE_SORT_ASC;
} else {
rb_raise(rb_eArgError,
"order should be one of "
"[nil, :desc, :descending, :asc, :ascending]: %s",
rb_grn_inspect(rb_order));
}
}
rb_grn_scan_options(options,
"offset", &rb_offset,
"limit", &rb_limit,
NULL);
if (!NIL_P(rb_offset))
offset = NUM2INT(rb_offset);
if (!NIL_P(rb_limit))
limit = NUM2INT(rb_limit);
result = grn_table_create(context, NULL, 0, NULL, GRN_TABLE_VIEW,
NULL, NULL);
grn_view_add(context, result,
grn_table_create(context, NULL, 0, NULL, GRN_TABLE_NO_KEY,
NULL, grn_ctx_get(context, "People", strlen("People"))));
grn_view_add(context, result,
grn_table_create(context, NULL, 0, NULL, GRN_TABLE_NO_KEY,
NULL, grn_ctx_get(context, "People", strlen("People"))));
n_records = grn_table_sort(context, view, offset, limit,
result, keys, n_keys);
exception = rb_grn_context_to_exception(context, self);
if (!NIL_P(exception)) {
grn_obj_unlink(context, result);
rb_exc_raise(exception);
}
rb_result = rb_ary_new();
cursor = grn_table_cursor_open(context, result, NULL, 0, NULL, 0,
0, -1, GRN_CURSOR_ASCENDING);
GRN_TEXT_INIT(&id, 0);
while (grn_table_cursor_next_o(context, cursor, &id) == GRN_SUCCESS) {
rb_ary_push(rb_result, rb_grn_view_record_new(self, &id));
}
GRN_OBJ_FIN(context, &id);
grn_table_cursor_close(context, cursor);
grn_obj_unlink(context, result);
#endif
return rb_result;
}
/*
* 文字列 _query_ をパースする。
* @overload parse(query, options={})
* @param [String] query パースする文字列
* @param [::Hash] options The name and value
* pairs. Omitted names are initialized as the default value.
* @option options :default_column
* "column_name:hoge"ではなく"hoge"のようにcolumn_nameが指
* 定されない条件の検索対象となるカラムを指定する。
* @option options :default_operator (Groonga::Operator::AND)
* "+"や"OR"で繋がれず、ただ列挙された複数の条件があった時、
* _expression_ 全体として各レコードをヒットとみなすかの論理
* 条件を指定する。省略した場合はGroonga::Operator::AND。
*
* - Groonga::Operator::OR :=
* レコードはいずれかの条件にマッチすればいい。 =:
* - Groonga::Operator::AND :=
* レコードは全ての条件にマッチしなければならない。 =:
* - Groonga::Operator::AND_NOT :=
* 最初の条件にレコードはマッチし、残りの条件にレコードは
* マッチしてはならない。 =:
*
* @option options :default_mode (Groonga::Operator::MATCH)
* 検索時のモードを指定する。省略した場合はGroonga::Operator::MATCH。
* （FIXME: モードによってどういう動作になるかを書く。）
* @option options :syntax (:query)
* _query_ の構文を指定する。指定可能な値は以下の通り。省略
* した場合は +:query+ 。
*
* - +:query+ :=
* 「文字列1 OR 文字列2」で「"文字列1"あるいは"文字列2"
* にマッチという検索エンジンで利用できるような構文を使
* う。
* 参考: "Groongaのクエリ構文のドキュメント":http://groonga.org/ja/docs/reference/grn_expr/query_syntax.html =:
* - +nil+ :=
* +:query+と同様 =:
* - +:script+ :=
* 「[カラム名] == [値]」というようにECMAScript風の構文を使う。
* 参考: "Groongaのscript構文のドキュメント":http://groonga.org/ja/docs/reference/grn_expr/script_syntax.html =:
* @option options :allow_pragma
* _query_ の構文に query を用いているとき（ +:syntax+
* オプション参照）、「*E-1」というようにクエリの先頭で
* pragmaを利用できるようにする。script構文を用いている
* ときはこのオプションを利用できない。
*
* デフォルトではプラグマを利用できる。
*
* 参考: "Groongaのクエリ構文のドキュメント":http://groonga.org/ja/docs/reference/grn_expr/query_syntax.html
* @option options :allow_column
* _query_ の構文にqueryを用いているとき（ +:syntax+ オプショ
* ン参照）、「カラム名:値」というようにカラム名を指定した
* 条件式を利用できるようにする。script構文を用いていると
* きはこのオプションを利用できない。
*
* デフォルトではカラム名を指定した条件式を利用できる。
*
* 参考: "Groongaのクエリ構文のドキュメント":http://groonga.org/ja/docs/reference/grn_expr/query_syntax.html
* @option options :allow_update
* _query_ の構文にscriptを用いているとき（ +:syntax+ オプショ
* ン参照）、「カラム名 = 値」というように更新操作を利用で
* きるようにする。query構文を用いているときはこのオプショ
* ンを利用できない。
*
* デフォルトでは更新操作を利用できる。
*
* 参考: "Groongaのクエリ構文のドキュメント":http://groonga.org/ja/docs/reference/grn_expr/query_syntax.html
*/
static VALUE
rb_grn_expression_parse (int argc, VALUE *argv, VALUE self)
{
grn_ctx *context = NULL;
grn_obj *expression, *default_column;
grn_bool default_column_is_created = GRN_FALSE;
grn_operator default_operator = GRN_OP_AND;
grn_operator default_mode = GRN_OP_MATCH;
grn_rc rc;
char *query = NULL;
unsigned query_size = 0;
grn_expr_flags flags = 0;
VALUE options, rb_query, rb_default_column, rb_default_operator;
VALUE rb_default_mode, rb_syntax;
VALUE rb_allow_pragma, rb_allow_column, rb_allow_update, rb_allow_leading_not;
VALUE exception = Qnil;
rb_scan_args(argc, argv, "11", &rb_query, &options);
rb_grn_scan_options(options,
"default_column", &rb_default_column,
"default_operator", &rb_default_operator,
"default_mode", &rb_default_mode,
"syntax", &rb_syntax,
"allow_pragma", &rb_allow_pragma,
"allow_column", &rb_allow_column,
"allow_update", &rb_allow_update,
"allow_leading_not", &rb_allow_leading_not,
NULL);
query = StringValuePtr(rb_query);
query_size = RSTRING_LEN(rb_query);
rb_grn_expression_deconstruct(SELF(self), &expression, &context,
NULL, NULL,
NULL, NULL, NULL);
if (NIL_P(rb_default_column)) {
default_column = NULL;
} else if (RVAL2CBOOL(rb_obj_is_kind_of(rb_default_column, rb_cGrnObject))) {
default_column = RVAL2GRNOBJECT(rb_default_column, &context);
} else {
default_column = RVAL2GRNBULK(rb_default_column, context, NULL);
default_column_is_created = GRN_TRUE;
}
if (!NIL_P(rb_default_mode))
default_mode = RVAL2GRNOPERATOR(rb_default_mode);
if (!NIL_P(rb_default_operator))
default_operator = RVAL2GRNSETOPERATOR(rb_default_operator);
if (NIL_P(rb_syntax) ||
rb_grn_equal_option(rb_syntax, "query")) {
flags = GRN_EXPR_SYNTAX_QUERY;
} else if (rb_grn_equal_option(rb_syntax, "script")) {
flags = GRN_EXPR_SYNTAX_SCRIPT;
} else {
rb_raise(rb_eArgError,
"syntax should be one of "
"[nil, :query, :script]: %s",
rb_grn_inspect(rb_syntax));
}
if (NIL_P(rb_allow_pragma)) {
if (!(flags & GRN_EXPR_SYNTAX_SCRIPT))
flags |= GRN_EXPR_ALLOW_PRAGMA;
} else {
if ((flags & GRN_EXPR_SYNTAX_SCRIPT))
rb_raise(rb_eArgError,
":allow_pragma isn't allowed in script syntax");
if (RVAL2CBOOL(rb_allow_pragma))
flags |= GRN_EXPR_ALLOW_PRAGMA;
}
if (NIL_P(rb_allow_column)) {
if (!(flags & GRN_EXPR_SYNTAX_SCRIPT))
flags |= GRN_EXPR_ALLOW_COLUMN;
} else {
if ((flags & GRN_EXPR_SYNTAX_SCRIPT))
rb_raise(rb_eArgError,
":allow_column isn't allowed in script syntax");
if (RVAL2CBOOL(rb_allow_column))
flags |= GRN_EXPR_ALLOW_COLUMN;
}
if (NIL_P(rb_allow_update)) {
flags |= GRN_EXPR_ALLOW_UPDATE;
} else {
if (RVAL2CBOOL(rb_allow_update))
flags |= GRN_EXPR_ALLOW_UPDATE;
}
if (!NIL_P(rb_allow_leading_not)) {
if (RVAL2CBOOL(rb_allow_leading_not))
flags |= GRN_EXPR_ALLOW_LEADING_NOT;
}
rc = grn_expr_parse(context, expression, query, query_size,
default_column, default_mode, default_operator,
flags);
if (rc != GRN_SUCCESS) {
VALUE related_object;
related_object =
rb_ary_new_from_args(2,
self,
rb_ary_new_from_values(argc, argv));
exception = rb_grn_context_to_exception(context, related_object);
}
if (default_column_is_created)
grn_obj_unlink(context, default_column);
if (!NIL_P(exception))
rb_exc_raise(exception);
return Qnil;
}
grn_obj *
rb_grn_bulk_from_ruby_object (VALUE object, grn_ctx *context, grn_obj *bulk)
{
if (bulk && bulk->header.domain == GRN_DB_TIME)
return RVAL2GRNBULK_WITH_TYPE(object, context, bulk,
bulk->header.domain,
grn_ctx_at(context, bulk->header.domain));
if (!bulk) {
bulk = grn_obj_open(context, GRN_BULK, 0, GRN_ID_NIL);
rb_grn_context_check(context, object);
}
switch (TYPE(object)) {
case T_NIL:
grn_obj_reinit(context, bulk, GRN_DB_VOID, 0);
break;
case T_SYMBOL:
object = rb_funcall(object, rb_intern("to_s"), 0);
case T_STRING:
grn_obj_reinit(context, bulk, GRN_DB_TEXT, 0);
rb_grn_context_text_set(context, bulk, object);
break;
case T_FIXNUM:
case T_BIGNUM: {
int64_t int64_value;
int64_value = NUM2LL(object);
if (int64_value <= INT32_MAX) {
grn_obj_reinit(context, bulk, GRN_DB_INT32, 0);
GRN_INT32_SET(context, bulk, int64_value);
} else {
grn_obj_reinit(context, bulk, GRN_DB_INT64, 0);
GRN_INT64_SET(context, bulk, int64_value);
}
break;
}
case T_FLOAT:
grn_obj_reinit(context, bulk, GRN_DB_FLOAT, 0);
GRN_FLOAT_SET(context, bulk, NUM2DBL(object));
break;
case T_TRUE:
grn_obj_reinit(context, bulk, GRN_DB_BOOL, 0);
GRN_BOOL_SET(context, bulk, GRN_TRUE);
break;
case T_FALSE:
grn_obj_reinit(context, bulk, GRN_DB_BOOL, 0);
GRN_BOOL_SET(context, bulk, GRN_FALSE);
break;
default:
if (RVAL2CBOOL(rb_obj_is_kind_of(object, rb_cTime))) {
VALUE sec, usec;
int64_t time_value;
sec = rb_funcall(object, rb_intern("to_i"), 0);
usec = rb_funcall(object, rb_intern("usec"), 0);
time_value = GRN_TIME_PACK(NUM2LL(sec), NUM2LL(usec));
grn_obj_reinit(context, bulk, GRN_DB_TIME, 0);
GRN_TIME_SET(context, bulk, time_value);
} else if (RVAL2CBOOL(rb_obj_is_kind_of(object, rb_cGrnObject))) {
grn_obj *grn_object;
grn_id id_value;
grn_object = RVAL2GRNOBJECT(object, &context);
grn_obj_reinit(context, bulk, grn_object->header.domain, 0);
id_value = grn_obj_id(context, grn_object);
GRN_RECORD_SET(context, bulk, id_value);
} else if (RVAL2CBOOL(rb_obj_is_kind_of(object, rb_cGrnRecord))) {
grn_obj *table;
grn_id id_value;
table = RVAL2GRNOBJECT(rb_funcall(object, rb_intern("table"), 0),
&context);
id_value = NUM2UINT(rb_funcall(object, rb_intern("id"), 0));
grn_obj_reinit(context, bulk, grn_obj_id(context, table), 0);
GRN_RECORD_SET(context, bulk, id_value);
} else {
rb_raise(rb_eTypeError,
"bulked object should be one of "
"[nil, true, false, String, Symbol, Integer, Float, Time, "
"Groonga::Object, Groonga::Record]: %s",
rb_grn_inspect(object));
}
break;
}
return bulk;
}
static grn_id
resolve_source_id (grn_ctx *context, grn_obj *column, grn_id range_id,
VALUE rb_source)
{
grn_id source_id;
if (CBOOL2RVAL(rb_obj_is_kind_of(rb_source, rb_cInteger))) {
source_id = NUM2UINT(rb_source);
} else {
grn_obj *source;
grn_bool need_source_unlink = GRN_FALSE;
if (TYPE(rb_source) == T_STRING) {
grn_obj *table;
const char *name;
const char *dot_point;
int length;
table = grn_ctx_at(context, grn_obj_get_range(context, column));
name = StringValueCStr(rb_source);
length = RSTRING_LEN(rb_source);
dot_point = strstr(name, ".");
if (dot_point) {
char table_name[4096];
int table_name_length;
table_name_length = grn_obj_name(context, table,
table_name, sizeof(table_name));
table_name[table_name_length] = '\0';
if (strncmp(table_name, name, dot_point - name) != 0) {
rb_raise(rb_eArgError,
"wrong table's column: <%s>: "
"expected table: <%s>",
name, table_name);
}
length -= (dot_point - name) + 1;
name = dot_point + 1;
}
source = grn_obj_column(context, table, name, length);
need_source_unlink = GRN_TRUE;
} else {
source = RVAL2GRNOBJECT(rb_source, &context);
}
rb_grn_context_check(context, rb_source);
if (source->header.type == GRN_ACCESSOR) {
char name[256];
int length;
length = grn_column_name(context, source, name, sizeof(name));
name[length] = '\0';
if (strcmp(name, "_key") != 0) {
rb_raise(rb_eArgError,
"source accessor must be '_key': <%s>", name);
}
source_id = range_id;
} else {
source_id = grn_obj_id(context, source);
}
if (need_source_unlink) {
grn_obj_unlink(context, source);
}
}
return source_id;
}
void
rb_grn_table_sort_keys_fill (grn_ctx *context,
grn_table_sort_key *sort_keys,
size_t n_sort_keys,
VALUE rb_sort_keys,
VALUE rb_table)
{
size_t i;
for (i = 0; i < n_sort_keys; i++) {
VALUE rb_sort_key;
VALUE rb_sort_options;
VALUE rb_key;
VALUE rb_resolved_key;
VALUE rb_order;
rb_sort_key = RARRAY_PTR(rb_sort_keys)[i];
if (RVAL2CBOOL(rb_obj_is_kind_of(rb_sort_key, rb_cHash))) {
rb_sort_options = rb_sort_key;
} else if (RVAL2CBOOL(rb_obj_is_kind_of(rb_sort_key, rb_cArray))) {
rb_sort_options = rb_hash_new();
rb_hash_aset(rb_sort_options,
RB_GRN_INTERN("key"),
rb_ary_entry(rb_sort_key, 0));
rb_hash_aset(rb_sort_options,
RB_GRN_INTERN("order"),
rb_ary_entry(rb_sort_key, 1));
} else {
rb_sort_options = rb_hash_new();
rb_hash_aset(rb_sort_options,
RB_GRN_INTERN("key"),
rb_sort_key);
}
rb_grn_scan_options(rb_sort_options,
"key", &rb_key,
"order", &rb_order,
NULL);
if (RVAL2CBOOL(rb_obj_is_kind_of(rb_key, rb_cString))) {
rb_resolved_key = rb_grn_table_get_column(rb_table, rb_key);
} else {
rb_resolved_key = rb_key;
}
sort_keys[i].key = RVAL2GRNOBJECT(rb_resolved_key, &context);
if (!sort_keys[i].key) {
rb_raise(rb_eGrnNoSuchColumn,
"no such column: <%s>: <%s>",
rb_grn_inspect(rb_key),
rb_grn_inspect(rb_table));
}
if (NIL_P(rb_order) ||
rb_grn_equal_option(rb_order, "asc") ||
rb_grn_equal_option(rb_order, "ascending")) {
sort_keys[i].flags = GRN_TABLE_SORT_ASC;
} else if (rb_grn_equal_option(rb_order, "desc") ||
rb_grn_equal_option(rb_order, "descending")) {
sort_keys[i].flags = GRN_TABLE_SORT_DESC;
} else {
rb_raise(rb_eArgError,
"order should be one of "
"[nil, :desc, :descending, :asc, :ascending]: %s",
rb_grn_inspect(rb_order));
}
}
}
/*
* It creates a table that manages records by double array trie.
* ブロックを指定すると、そのブロックに生成したテーブルが渡さ
* れ、ブロックを抜けると自動的にテーブルが破棄される。
*
* @example
* #無名一時テーブルを生成する。
* Groonga::DoubleArrayTrie.create
*
* #無名永続テーブルを生成する。
* Groonga::DoubleArrayTrie.create(:path => "/tmp/hash.grn")
*
* #名前付き永続テーブルを生成する。ただし、ファイル名は気に
* #しない。
* Groonga::DoubleArrayTrie.create(:name => "Bookmarks",
* :persistent => true)
*
* #それぞれのレコードに512バイトの値を格納できる無名一時テー
* #ブルを生成する。
* Groonga::DoubleArrayTrie.create(:value => 512)
*
* #キーとして文字列を使用する無名一時テーブルを生成する。
* Groonga::DoubleArrayTrie.create(:key_type => Groonga::Type::SHORT_TEXT)
*
* #キーとして文字列を使用する無名一時テーブルを生成する。
* #（キーの種類を表すオブジェクトは文字列で指定。）
* Groonga::DoubleArrayTrie.create(:key_type => "ShortText")
*
* #キーとしてBookmarksテーブルのレコードを使用す
* #る無名一時テーブルを生成する。
* bookmarks = Groonga::DoubleArrayTrie.create(:name => "Bookmarks")
* Groonga::DoubleArrayTrie.create(:key_type => bookmarks)
*
* #キーとしてBookmarksテーブルのレコードを使用す
* #る無名一時テーブルを生成する。
* #（テーブルは文字列で指定。）
* Groonga::DoubleArrayTrie.create(:name => "Bookmarks")
* Groonga::DoubleArrayTrie.create(:key_type => "Bookmarks")
*
* #全文検索用のトークンをバイグラムで切り出す無名一時テーブ
* #ルを生成する。
* bookmarks = Groonga::DoubleArrayTrie.create(:name => "Bookmarks")
* bookmarks.define_column("comment", "Text")
* terms = Groonga::DoubleArrayTrie.create(:name => "Terms",
* :default_tokenizer => "TokenBigram")
* terms.define_index_column("content", bookmarks,
* :source => "Bookmarks.comment")
*
* @overload create(options={})
* @return [Groonga::DoubleArrayTrie]
* @!macro [new] double-array-trie.create.options
* @param [::Hash] options The name and value
* pairs. Omitted names are initialized as the default value.
* @option options [Groonga::Context] :context (Groonga::Context.default)
* テーブルが利用する {Groonga::Context} 。
* @option options :name
* テーブルの名前。名前をつけると、 {Groonga::Context#[]} に名
* 前を指定してテーブルを取得することができる。省略すると
* 無名テーブルになり、テーブルIDでのみ取得できる。
* @option options :path
* テーブルを保存するパス。パスを指定すると永続テーブルとな
* り、プロセス終了後もレコードは保持される。次回起動時に
* {Groonga::Context#[]} で保存されたレコードを利用する
* ことができる。省略すると一時テーブルになり、プロセスが終
* 了するとレコードは破棄される。
* @option options :persistent
* +true+ を指定すると永続テーブルとなる。 +path+ を省略した
* 場合は自動的にパスが付加される。 +:context+ で指定した
* {Groonga::Context} に結びついているデータベースが一時デー
* タベースの場合は例外が発生する。
*
* @option options :key_normalize (false) Keys are normalized
* if this value is @true@.
*
* @deprecated Use @:normalizer => "NormalizerAuto"@ instead.
*
* @option options :key_with_sis
* +true+ を指定するとキーの文字列の全suffixが自動的に登
* 録される。
* @option options :key_type
* キーの種類を示すオブジェクトを指定する。キーの種類には型
* 名（"Int32"や"ShortText"など）または {Groonga::Type} または
* テーブル（ {Groonga::Array} 、{Groonga::Hash} 、
* {Groonga::DoubleArrayTrie} のどれか）を指定する。
* {Groonga::Type} を指定した場合は、その型が示す範囲の値をキー
* として使用する。ただし、キーの最大サイズは4096バイトで
* あるため、 {Groonga::Type::TEXT} や {Groonga::Type::LONG_TEXT}
* は使用できない。
*
* テーブルを指定した場合はレコードIDをキーとして使用する。
* 指定したテーブルの {Groonga::Record} をキーとして使用するこ
* ともでき、その場合は自動的に {Groonga::Record} からレコード
* IDを取得する。
*
* 省略した場合はShortText型をキーとして使用する。この場合、
* 4096バイトまで使用可能である。
* @option options :value_type
* 値の型を指定する。省略すると値のための領域を確保しない。
* 値を保存したい場合は必ず指定すること。
*
* 参考: {Groonga::Type.new}
*
* @option options :default_tokenizer
* {Groonga::IndexColumn} で使用するトークナイザを指定する。
* デフォルトでは何も設定されていないので、テーブルに
* {Groonga::IndexColumn} を定義する場合は
* @"TokenBigram"@ などを指定する必要がある。
*
* @option options [::Array<String, Groonga::Procedure>, nil]
* :token_filters (nil) The token filters to be used in the
* table.
*
* @option options :sub_records
* +true+ を指定すると {#group} でグループ化したときに、
* {Groonga::Record#n_sub_records} でグループに含まれるレコー
* ドの件数を取得できる。
*
* @option options [String, Groonga::Procedure, nil] :normalizer
* The normalizer that is used by {Groonga::IndexColumn}. You
* can specify this by normalizer name as String such as
* @"NormalizerAuto"@ or normalizer object.
*
* @!macro double-array-trie.create.options
* @overload create(options={})
* @yield [table]
* @!macro double-array-trie.create.options
*/
static VALUE
rb_grn_double_array_trie_s_create (int argc, VALUE *argv, VALUE klass)
{
grn_ctx *context;
grn_obj *key_type = NULL, *value_type = NULL, *table;
const char *name = NULL, *path = NULL;
unsigned name_size = 0;
grn_obj_flags flags = GRN_OBJ_TABLE_DAT_KEY;
VALUE rb_table;
VALUE options, rb_context, rb_name, rb_path, rb_persistent;
VALUE rb_key_normalize, rb_key_with_sis, rb_key_type;
VALUE rb_value_type;
VALUE rb_default_tokenizer;
VALUE rb_token_filters;
VALUE rb_sub_records;
VALUE rb_normalizer;
rb_scan_args(argc, argv, "01", &options);
rb_grn_scan_options(options,
"context", &rb_context,
"name", &rb_name,
"path", &rb_path,
"persistent", &rb_persistent,
"key_normalize", &rb_key_normalize,
"key_with_sis", &rb_key_with_sis,
"key_type", &rb_key_type,
"value_type", &rb_value_type,
"default_tokenizer", &rb_default_tokenizer,
"token_filters", &rb_token_filters,
"sub_records", &rb_sub_records,
"normalizer", &rb_normalizer,
NULL);
context = rb_grn_context_ensure(&rb_context);
if (!NIL_P(rb_name)) {
name = StringValuePtr(rb_name);
name_size = RSTRING_LEN(rb_name);
flags |= GRN_OBJ_PERSISTENT;
}
if (!NIL_P(rb_path)) {
path = StringValueCStr(rb_path);
flags |= GRN_OBJ_PERSISTENT;
}
if (RVAL2CBOOL(rb_persistent))
flags |= GRN_OBJ_PERSISTENT;
if (RVAL2CBOOL(rb_key_normalize))
flags |= GRN_OBJ_KEY_NORMALIZE;
if (RVAL2CBOOL(rb_key_with_sis))
flags |= GRN_OBJ_KEY_WITH_SIS;
if (NIL_P(rb_key_type)) {
key_type = grn_ctx_at(context, GRN_DB_SHORT_TEXT);
} else {
key_type = RVAL2GRNOBJECT(rb_key_type, &context);
}
if (!NIL_P(rb_value_type))
value_type = RVAL2GRNOBJECT(rb_value_type, &context);
if (RVAL2CBOOL(rb_sub_records))
flags |= GRN_OBJ_WITH_SUBREC;
table = grn_table_create(context, name, name_size, path,
flags, key_type, value_type);
if (!table)
rb_grn_context_check(context, rb_ary_new4(argc, argv));
rb_table = GRNOBJECT2RVAL(klass, context, table, GRN_TRUE);
if (!NIL_P(rb_default_tokenizer))
rb_funcall(rb_table, rb_intern("default_tokenizer="), 1,
rb_default_tokenizer);
if (!NIL_P(rb_token_filters))
rb_funcall(rb_table, rb_intern("token_filters="), 1,
rb_token_filters);
if (!NIL_P(rb_normalizer))
rb_funcall(rb_table, rb_intern("normalizer="), 1,
rb_normalizer);
if (rb_block_given_p())
return rb_ensure(rb_yield, rb_table, rb_grn_object_close, rb_table);
else
return rb_table;
}
