fix

Signed-off-by: Kosuke Morimoto <ksk@vdaas.org>
vdaas · Aug 20, 2024 · 50d6a8c · 50d6a8c
1 parent e4526c3
commit 50d6a8c
Show file tree

Hide file tree

Showing 2 changed files with 64 additions and 0 deletions.
diff --git a/pkg/agent/core/ngt/handler/grpc/index.go b/pkg/agent/core/ngt/handler/grpc/index.go
@@ -240,3 +240,23 @@ func (s *server) IndexStatisticsDetail(
 		},
 	}, nil
 }
+
+func (s *server) IndexProperty(
+	ctx context.Context, _ *payload.Empty,
+) (res *payload.Info_Index_PropertyDetail, err error) {
+	_, span := trace.StartSpan(ctx, apiName+".IndexStatisticsDetail")
+	defer func() {
+		if span != nil {
+			span.End()
+		}
+	}()
+	prop, err := s.ngt.IndexProperty()
+	if err != nil {
+		return nil, err
+	}
+	return &payload.Info_Index_PropertyDetail{
+		Details: map[string]*payload.Info_Index_Property{
+			s.name: prop,
+		},
+	}, nil
+}
diff --git a/pkg/agent/core/ngt/service/ngt.go b/pkg/agent/core/ngt/service/ngt.go
@@ -92,6 +92,7 @@ type NGT interface {
 	BrokenIndexCount() uint64
 	IndexStatistics() (*payload.Info_Index_Statistics, error)
 	IsStatisticsEnabled() bool
+	IndexProperty() (*payload.Info_Index_Property, error)
 	Close(ctx context.Context) error
 }
 
@@ -2044,6 +2045,49 @@ func (n *ngt) IsStatisticsEnabled() bool {
 	return n.enableStatistics
 }
 
+func (n *ngt) IndexProperty() (*payload.Info_Index_Property, error) {
+	p, err := n.core.GetProperty()
+	if err != nil {
+		return nil, err
+	}
+	return &payload.Info_Index_Property{
+		Dimension:                     p.Dimension,
+		ThreadPoolSize:                p.ThreadPoolSize,
+		ObjectType:                    p.ObjectType.String(),
+		DistanceType:                  p.DistanceType.String(),
+		IndexType:                     p.IndexType.String(),
+		DatabaseType:                  p.DatabaseType.String(),
+		ObjectAlignment:               p.ObjectAlignment.String(),
+		PathAdjustmentInterval:        p.PathAdjustmentInterval,
+		GraphSharedMemorySize:         p.GraphSharedMemorySize,
+		TreeSharedMemorySize:          p.TreeSharedMemorySize,
+		ObjectSharedMemorySize:        p.ObjectSharedMemorySize,
+		PrefetchOffset:                p.PrefetchOffset,
+		PrefetchSize:                  p.PrefetchSize,
+		AccuracyTable:                 p.AccuracyTable,
+		SearchType:                    p.SearchType,
+		MaxMagnitude:                  p.MaxMagnitude,
+		NOfNeighborsForInsertionOrder: p.NOfNeighborsForInsertionOrder,
+		EpsilonForInsertionOrder:      p.EpsilonForInsertionOrder,
+		RefinementObjectType:          p.RefinementObjectType.String(),
+		TruncationThreshold:           p.TruncationThreshold,
+		EdgeSizeForCreation:           p.EdgeSizeForCreation,
+		EdgeSizeForSearch:             p.EdgeSizeForSearch,
+		EdgeSizeLimitForCreation:      p.EdgeSizeLimitForCreation,
+		InsertionRadiusCoefficient:    p.InsertionRadiusCoefficient,
+		SeedSize:                      p.SeedSize,
+		SeedType:                      p.SeedType.String(),
+		TruncationThreadPoolSize:      p.TruncationThreadPoolSize,
+		BatchSizeForCreation:          p.BatchSizeForCreation,
+		GraphType:                     p.GraphType.String(),
+		DynamicEdgeSizeBase:           p.DynamicEdgeSizeBase,
+		DynamicEdgeSizeRate:           p.DynamicEdgeSizeRate,
+		BuildTimeLimit:                p.BuildTimeLimit,
+		OutgoingEdge:                  p.OutgoingEdge,
+		IncomingEdge:                  p.IncomingEdge,
+	}, nil
+}
+
 func (n *ngt) toSearchResponse(
 	sr []algorithm.SearchResult,
 ) (res *payload.Search_Response, err error) {