1 /**
2 * Licensed to the Apache Software Foundation (ASF) under one
3 * or more contributor license agreements. See the NOTICE file
4 * distributed with this work for additional information
5 * regarding copyright ownership. The ASF licenses this file
6 * to you under the Apache License, Version 2.0 (the
7 * "License"); you may not use this file except in compliance
8 * with the License. You may obtain a copy of the License at
9 *
10 * http://www.apache.org/licenses/LICENSE-2.0
11 *
12 * Unless required by applicable law or agreed to in writing,
13 * software distributed under the License is distributed on an
14 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15 * KIND, either express or implied. See the License for the
16 * specific language governing permissions and limitations
17 * under the License.
18 */
19
20 #include <ruby.h>
21 #include <bytes.h>
22 #include <struct.h>
23 #include <binary_protocol_accelerated.h>
24 #include <compact_protocol.h>
25 #include <memory_buffer.h>
26
27 // cached classes/modules
28 VALUE rb_cSet;
29 VALUE thrift_module;
30 VALUE thrift_bytes_module;
31 VALUE thrift_types_module;
32
33 // TType constants
34 int TTYPE_STOP;
35 int TTYPE_BOOL;
36 int TTYPE_BYTE;
37 int TTYPE_I16;
38 int TTYPE_I32;
39 int TTYPE_I64;
40 int TTYPE_DOUBLE;
41 int TTYPE_STRING;
42 int TTYPE_MAP;
43 int TTYPE_SET;
44 int TTYPE_LIST;
45 int TTYPE_STRUCT;
46
47 // method ids
48 ID validate_method_id;
49 ID write_struct_begin_method_id;
50 ID write_struct_end_method_id;
51 ID write_field_begin_method_id;
52 ID write_field_end_method_id;
53 ID write_boolean_method_id;
54 ID write_byte_method_id;
55 ID write_i16_method_id;
56 ID write_i32_method_id;
57 ID write_i64_method_id;
58 ID write_double_method_id;
59 ID write_string_method_id;
60 ID write_binary_method_id;
61 ID write_map_begin_method_id;
62 ID write_map_end_method_id;
63 ID write_list_begin_method_id;
64 ID write_list_end_method_id;
65 ID write_set_begin_method_id;
66 ID write_set_end_method_id;
67 ID read_bool_method_id;
68 ID read_byte_method_id;
69 ID read_i16_method_id;
70 ID read_i32_method_id;
71 ID read_i64_method_id;
72 ID read_string_method_id;
73 ID read_binary_method_id;
74 ID read_double_method_id;
75 ID read_map_begin_method_id;
76 ID read_map_end_method_id;
77 ID read_list_begin_method_id;
78 ID read_list_end_method_id;
79 ID read_set_begin_method_id;
80 ID read_set_end_method_id;
81 ID read_struct_begin_method_id;
82 ID read_struct_end_method_id;
83 ID read_field_begin_method_id;
84 ID read_field_end_method_id;
85 ID keys_method_id;
86 ID entries_method_id;
87 ID write_field_stop_method_id;
88 ID skip_method_id;
89 ID write_method_id;
90 ID read_all_method_id;
91 ID read_into_buffer_method_id;
92 ID force_binary_encoding_id;
93 ID convert_to_utf8_byte_buffer_id;
94 ID convert_to_string_id;
95
96 // constant ids
97 ID fields_const_id;
98 ID transport_ivar_id;
99 ID strict_read_ivar_id;
100 ID strict_write_ivar_id;
101
102 // cached symbols
103 VALUE type_sym;
104 VALUE name_sym;
105 VALUE key_sym;
106 VALUE value_sym;
107 VALUE element_sym;
108 VALUE class_sym;
109 VALUE binary_sym;
110 VALUE protocol_exception_class;
111
Init_thrift_native()112 void Init_thrift_native() {
113 // cached classes
114 thrift_module = rb_const_get(rb_cObject, rb_intern("Thrift"));
115 rb_global_variable(&thrift_module);
116
117 thrift_bytes_module = rb_const_get(thrift_module, rb_intern("Bytes"));
118 rb_global_variable(&thrift_bytes_module);
119
120 thrift_types_module = rb_const_get(thrift_module, rb_intern("Types"));
121 rb_global_variable(&thrift_types_module);
122
123 rb_cSet = rb_const_get(rb_cObject, rb_intern("Set"));
124 rb_global_variable(&rb_cSet);
125
126 protocol_exception_class = rb_const_get(thrift_module, rb_intern("ProtocolException"));
127 rb_global_variable(&protocol_exception_class);
128
129 // Init ttype constants
130 TTYPE_BOOL = FIX2INT(rb_const_get(thrift_types_module, rb_intern("BOOL")));
131 TTYPE_BYTE = FIX2INT(rb_const_get(thrift_types_module, rb_intern("BYTE")));
132 TTYPE_I16 = FIX2INT(rb_const_get(thrift_types_module, rb_intern("I16")));
133 TTYPE_I32 = FIX2INT(rb_const_get(thrift_types_module, rb_intern("I32")));
134 TTYPE_I64 = FIX2INT(rb_const_get(thrift_types_module, rb_intern("I64")));
135 TTYPE_DOUBLE = FIX2INT(rb_const_get(thrift_types_module, rb_intern("DOUBLE")));
136 TTYPE_STRING = FIX2INT(rb_const_get(thrift_types_module, rb_intern("STRING")));
137 TTYPE_MAP = FIX2INT(rb_const_get(thrift_types_module, rb_intern("MAP")));
138 TTYPE_SET = FIX2INT(rb_const_get(thrift_types_module, rb_intern("SET")));
139 TTYPE_LIST = FIX2INT(rb_const_get(thrift_types_module, rb_intern("LIST")));
140 TTYPE_STRUCT = FIX2INT(rb_const_get(thrift_types_module, rb_intern("STRUCT")));
141
142 // method ids
143 validate_method_id = rb_intern("validate");
144 write_struct_begin_method_id = rb_intern("write_struct_begin");
145 write_struct_end_method_id = rb_intern("write_struct_end");
146 write_field_begin_method_id = rb_intern("write_field_begin");
147 write_field_end_method_id = rb_intern("write_field_end");
148 write_boolean_method_id = rb_intern("write_bool");
149 write_byte_method_id = rb_intern("write_byte");
150 write_i16_method_id = rb_intern("write_i16");
151 write_i32_method_id = rb_intern("write_i32");
152 write_i64_method_id = rb_intern("write_i64");
153 write_double_method_id = rb_intern("write_double");
154 write_string_method_id = rb_intern("write_string");
155 write_binary_method_id = rb_intern("write_binary");
156 write_map_begin_method_id = rb_intern("write_map_begin");
157 write_map_end_method_id = rb_intern("write_map_end");
158 write_list_begin_method_id = rb_intern("write_list_begin");
159 write_list_end_method_id = rb_intern("write_list_end");
160 write_set_begin_method_id = rb_intern("write_set_begin");
161 write_set_end_method_id = rb_intern("write_set_end");
162 read_bool_method_id = rb_intern("read_bool");
163 read_byte_method_id = rb_intern("read_byte");
164 read_i16_method_id = rb_intern("read_i16");
165 read_i32_method_id = rb_intern("read_i32");
166 read_i64_method_id = rb_intern("read_i64");
167 read_string_method_id = rb_intern("read_string");
168 read_binary_method_id = rb_intern("read_binary");
169 read_double_method_id = rb_intern("read_double");
170 read_map_begin_method_id = rb_intern("read_map_begin");
171 read_map_end_method_id = rb_intern("read_map_end");
172 read_list_begin_method_id = rb_intern("read_list_begin");
173 read_list_end_method_id = rb_intern("read_list_end");
174 read_set_begin_method_id = rb_intern("read_set_begin");
175 read_set_end_method_id = rb_intern("read_set_end");
176 read_struct_begin_method_id = rb_intern("read_struct_begin");
177 read_struct_end_method_id = rb_intern("read_struct_end");
178 read_field_begin_method_id = rb_intern("read_field_begin");
179 read_field_end_method_id = rb_intern("read_field_end");
180 keys_method_id = rb_intern("keys");
181 entries_method_id = rb_intern("entries");
182 write_field_stop_method_id = rb_intern("write_field_stop");
183 skip_method_id = rb_intern("skip");
184 write_method_id = rb_intern("write");
185 read_all_method_id = rb_intern("read_all");
186 read_into_buffer_method_id = rb_intern("read_into_buffer");
187 force_binary_encoding_id = rb_intern("force_binary_encoding");
188 convert_to_utf8_byte_buffer_id = rb_intern("convert_to_utf8_byte_buffer");
189 convert_to_string_id = rb_intern("convert_to_string");
190
191 // constant ids
192 fields_const_id = rb_intern("FIELDS");
193 transport_ivar_id = rb_intern("@trans");
194 strict_read_ivar_id = rb_intern("@strict_read");
195 strict_write_ivar_id = rb_intern("@strict_write");
196
197 // cached symbols
198 type_sym = ID2SYM(rb_intern("type"));
199 name_sym = ID2SYM(rb_intern("name"));
200 key_sym = ID2SYM(rb_intern("key"));
201 value_sym = ID2SYM(rb_intern("value"));
202 element_sym = ID2SYM(rb_intern("element"));
203 class_sym = ID2SYM(rb_intern("class"));
204 binary_sym = ID2SYM(rb_intern("binary"));
205
206 rb_global_variable(&type_sym);
207 rb_global_variable(&name_sym);
208 rb_global_variable(&key_sym);
209 rb_global_variable(&value_sym);
210 rb_global_variable(&element_sym);
211 rb_global_variable(&class_sym);
212 rb_global_variable(&binary_sym);
213
214 Init_struct();
215 Init_binary_protocol_accelerated();
216 Init_compact_protocol();
217 Init_memory_buffer();
218 }
219
