Skip to content

Commit d4ea373

Browse files
biaozybiao.zhoub
andauthored
update hologres client, use hybrid index (#594)
Co-authored-by: biao.zhoub <[email protected]>
1 parent 04e16c3 commit d4ea373

File tree

2 files changed

+21
-2
lines changed

2 files changed

+21
-2
lines changed

vectordb_bench/backend/clients/hologres/config.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -104,15 +104,14 @@ def builder_params(self) -> dict:
104104
self.base_quantization_type = "fp32"
105105

106106
return {
107-
"min_flush_proxima_row_count": self.min_flush_proxima_row_count,
108-
"min_compaction_proxima_row_count": self.min_compaction_proxima_row_count,
109107
"max_total_size_to_merge_mb": self.max_total_size_to_merge_mb,
110108
"build_thread_count": self.build_thread_count,
111109
"base_quantization_type": self.base_quantization_type,
112110
"max_degree": self.max_degree,
113111
"ef_construction": self.ef_construction,
114112
"precise_quantization_type": self.precise_quantization_type,
115113
"use_reorder": self.use_reorder,
114+
"precise_io_type": "reader_io",
116115
}
117116

118117
def searcher_params(self) -> dict:

vectordb_bench/backend/clients/hologres/hologres.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -128,6 +128,18 @@ def _drop_table(self):
128128
)
129129
self.conn.commit()
130130

131+
try:
132+
log.info(f"{self.name} client purge table recycle bin: {self.table_name}")
133+
self.cursor.execute(
134+
sql.SQL("purge TABLE {table_name};").format(
135+
table_name=sql.Identifier(self.table_name),
136+
),
137+
)
138+
except Exception as e:
139+
log.info(f"{self.name} client purge table {self.table_name} recycle bin failed, error: {e}, ignore.")
140+
finally:
141+
self.conn.commit()
142+
131143
try:
132144
log.info(f"{self.name} client drop table group : {self._tg_name}")
133145
self.cursor.execute(sql.SQL(f"CALL HG_DROP_TABLE_GROUP('{self._tg_name}');"))
@@ -136,6 +148,14 @@ def _drop_table(self):
136148
finally:
137149
self.conn.commit()
138150

151+
try:
152+
log.info(f"{self.name} client free cache")
153+
self.cursor.execute("select hg_admin_command('freecache');")
154+
except Exception as e:
155+
log.info(f"{self.name} client free cache failed, error: {e}, ignore.")
156+
finally:
157+
self.conn.commit()
158+
139159
def optimize(self, data_size: int | None = None):
140160
if self.case_config.create_index_after_load:
141161
self._create_index()

0 commit comments

Comments
 (0)