From 45dfb7134a78e9a241b8132119341dfc29debd37 Mon Sep 17 00:00:00 2001
From: Matthew Kim <38759997+friendlymatthew@users.noreply.github.com>
Date: Wed, 10 Jul 2024 10:44:01 -0400
Subject: [PATCH] feat: adjacency manager

---
 pkg/vectorpage/manager.go      | 53 ++++++++++++++++++++++++++++++++++
 pkg/vectorpage/manager_test.go | 22 ++++++++++++++
 2 files changed, 75 insertions(+)
 create mode 100644 pkg/vectorpage/manager.go
 create mode 100644 pkg/vectorpage/manager_test.go

diff --git a/pkg/vectorpage/manager.go b/pkg/vectorpage/manager.go
new file mode 100644
index 0000000..2b434e3
--- /dev/null
+++ b/pkg/vectorpage/manager.go
@@ -0,0 +1,53 @@
+package vectorpage
+
+import (
+	"fmt"
+	"github.com/kevmo314/appendable/pkg/bptree"
+	"github.com/kevmo314/appendable/pkg/btree"
+	"github.com/kevmo314/appendable/pkg/hnsw"
+	"github.com/kevmo314/appendable/pkg/pointer"
+)
+
+type HNSWAdjacencyPage [16][8]uint32
+
+type VectorPageManager struct {
+	btree   *btree.BTree
+	vectors []*hnsw.Point
+
+	bptree       *bptree.BPTree
+	neighborhood map[hnsw.Id]*hnsw.Friends
+}
+
+func NewVectorPageManager(btree *btree.BTree, bptree *bptree.BPTree, vectors []*hnsw.Point, neighborhood map[hnsw.Id]*hnsw.Friends) *VectorPageManager {
+	if btree == nil || bptree == nil {
+		panic("btree and bptree must not be nil")
+	}
+
+	return &VectorPageManager{btree, vectors, bptree, neighborhood}
+}
+
+func (vp *VectorPageManager) AddNode(x hnsw.Id) error {
+	// we'll assume that this node id is the freshly inserted vector
+	xvector := *vp.vectors[x]
+
+	if err := vp.btree.Insert(pointer.ReferencedId{Value: x}, xvector); err != nil {
+		return err
+	}
+
+	xfriends, ok := vp.neighborhood[x]
+
+	if !ok {
+		return fmt.Errorf("vector id %v not found in hnsw neighborhood", x)
+	}
+
+	xfriendsBuf, err := xfriends.Flush(8)
+	if err != nil {
+		return err
+	}
+
+	if err := vp.bptree.Insert(pointer.ReferencedValue{Value: xfriendsBuf}, pointer.MemoryPointer{}); err != nil {
+		return fmt.Errorf("failed to insert buf: %v", err)
+	}
+
+	return nil
+}
diff --git a/pkg/vectorpage/manager_test.go b/pkg/vectorpage/manager_test.go
new file mode 100644
index 0000000..3c33bd8
--- /dev/null
+++ b/pkg/vectorpage/manager_test.go
@@ -0,0 +1,22 @@
+package vectorpage
+
+import (
+	"github.com/kevmo314/appendable/pkg/hnsw"
+	"testing"
+)
+
+func TestNewVectorPageManager(t *testing.T) {
+
+	t.Run("", func(t *testing.T) {
+		p0 := hnsw.Point{3, 3}
+
+		h := hnsw.NewHnsw(2, 10, 8, p0)
+
+		for i := 0; i < 100; i++ {
+			if err := h.InsertVector(hnsw.Point{float32(i), float32(i)}); err != nil {
+				t.Fatal(err)
+			}
+		}
+
+	})
+}