liuyu/pgvector
1
0
Fork 0
mirror of https://github.com/pgvector/pgvector.git synced 2026-07-13 08:06:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

Updated IvfflatGetType [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2024-04-11 09:25:07 -07:00
parent e8bd7cd2f5
commit e1647473c9
1 changed files with 23 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src/ivfutils.c
View File

@@ -1,9 +1,11 @@
#include "postgres.h" #include "postgres.h"
#include "access/generic_xlog.h" #include "access/generic_xlog.h"
#include "catalog/pg_type.h"
#include "ivfflat.h" #include "ivfflat.h"
#include "storage/bufmgr.h" #include "storage/bufmgr.h"
#include "vector.h" #include "vector.h"
#include "utils/syscache.h"
/* /*
* Allocate a vector array * Allocate a vector array
@@ -72,7 +74,27 @@ IvfflatOptionalProcInfo(Relation index, uint16 procnum)
IvfflatType IvfflatType
IvfflatGetType(Relation index) IvfflatGetType(Relation index)
{ {
return IVFFLAT_TYPE_VECTOR; Oid typid = TupleDescAttr(index->rd_att, 0)->atttypid;
HeapTuple tuple;
Form_pg_type type;
IvfflatType result;
tuple = SearchSysCache1(TYPEOID, ObjectIdGetDatum(typid));
if (!HeapTupleIsValid(tuple))
elog(ERROR, "cache lookup failed for type %u", typid);
type = (Form_pg_type) GETSTRUCT(tuple);
if (strcmp(NameStr(type->typname), "vector") == 0)
result = IVFFLAT_TYPE_VECTOR;
else
{
ReleaseSysCache(tuple);
elog(ERROR, "type not supported for ivfflat index");
}
ReleaseSysCache(tuple);
return result;
} }
/* /*