| 197 | } |
| 198 | |
| 199 | static bool LoadFromVecsFile() { |
| 200 | tools::VecsReader reader; |
| 201 | if (!reader.load(FLAGS_file)) { |
| 202 | LOG_ERROR("Load vecs file failed."); |
| 203 | return false; |
| 204 | } |
| 205 | |
| 206 | for (uint32_t i = 0; i < reader.num_vecs(); i++) { |
| 207 | Record new_record; |
| 208 | uint64_t key = reader.get_key(i); |
| 209 | const char *feature = (const char *)reader.get_vector(i); |
| 210 | |
| 211 | new_record.key = key; |
| 212 | new_record.vector.append(feature, reader.index_meta().element_size()); |
| 213 | new_record.dimension = reader.index_meta().dimension(); |
| 214 | g_record_list.emplace_back(new_record); |
| 215 | |
| 216 | // Limit rows count |
| 217 | if (FLAGS_rows > 0 && g_record_list.size() >= FLAGS_rows) { |
| 218 | break; |
| 219 | } |
| 220 | } |
| 221 | |
| 222 | return true; |
| 223 | } |
| 224 | |
| 225 | static bool LoadRecords() { |
| 226 | bool ret; |
no test coverage detected