Give the rebuild its own call limit

[onechat] / memory.go

diff --git a/memory.go b/memory.go
index bcc10951921751aad707a32f7180799e023e9a18..7cbcce68c40f4b06382e3c8f3e71dfcbb3388d26 100644 (file)
--- a/memory.go
+++ b/memory.go
@@ -125,6 +125,7 @@ func (m *Memory) Close() error {
        defer m.mutex.Unlock()
 
        var err error
+
        if m.syncer != nil {
                err = m.syncer.Close()
        }
@@ -174,6 +175,7 @@ func (m *Memory) AppendDated(kind, text, date string) (Message, error) {
 // children reads the two children of the node key.
 func (m *Memory) children(key NodeKey) (Node, Node, error) {
        var children [2]Node
+
        for n := range children {
                child, built, err := m.tree.Node(key.level-1, 2*key.index+n)
                if err != nil {
@@ -194,6 +196,7 @@ func (m *Memory) children(key NodeKey) (Node, Node, error) {
 func (m *Memory) Wait(ctx context.Context) error {
        for {
                m.mutex.Lock()
+
                running := m.queue.Running()
                changed := m.changed
                m.mutex.Unlock()
@@ -217,6 +220,7 @@ func (m *Memory) Wait(ctx context.Context) error {
 func (m *Memory) Settle(ctx context.Context) error {
        for {
                m.mutex.Lock()
+
                pending := m.chat.Pending()
                running := m.queue.Running()
                changed := m.changed
@@ -295,6 +299,7 @@ func (m *Memory) load() error {
        }
 
        var file MemoryViewFile
+
        err = json.Unmarshal(data, &file)
        if err != nil {
                return fmt.Errorf("decode view: %w", err)
@@ -342,6 +347,7 @@ func (m *Memory) restoreQueue() error {
                leaf, free := freeLeaf(message)
                if !free {
                        m.queue.AddLeaf(id)
+
                        continue
                }
 
@@ -410,6 +416,7 @@ func (m *Memory) dispatch() {
                }
 
                m.workers.Add(1)
+
                go m.run(key, task)
        }
 }
@@ -508,6 +515,7 @@ func (m *Memory) settle(key NodeKey) error {
                node, free := freeMerge(left, right)
                if !free {
                        m.queue.AddMerge(parent)
+
                        return nil
                }
 
@@ -583,6 +591,7 @@ func (m *Memory) run(key NodeKey, task Task) {
        }
 
        close(m.changed)
+
        m.changed = make(chan struct{})
        m.dispatch()
 }
@@ -659,6 +668,7 @@ func (m *Memory) buildModelLines() ([]int, error) {
        }
 
        var sizes []int
+
        for _, key := range keys {
                var node Node