mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2024-11-22 11:46:40 +00:00
[bugfix] fix possible domain blockcache nil ptr + add debug String() func (#1755)
This commit is contained in:
parent
8275d70e38
commit
11e843a273
45
internal/cache/domain/domain.go
vendored
45
internal/cache/domain/domain.go
vendored
|
@ -80,6 +80,14 @@ func (b *BlockCache) Clear() {
|
|||
atomic.StorePointer(&b.rootptr, nil)
|
||||
}
|
||||
|
||||
// String returns a string representation of stored domains in block cache.
|
||||
func (b *BlockCache) String() string {
|
||||
if ptr := atomic.LoadPointer(&b.rootptr); ptr != nil {
|
||||
return (*root)(ptr).String()
|
||||
}
|
||||
return "<empty>"
|
||||
}
|
||||
|
||||
// root is the root node in the domain
|
||||
// block cache radix trie. this is the
|
||||
// singular access point to the trie.
|
||||
|
@ -104,6 +112,13 @@ func (r *root) Sort() {
|
|||
r.root.sort()
|
||||
}
|
||||
|
||||
// String returns a string representation of node (and its descendants).
|
||||
func (r *root) String() string {
|
||||
buf := new(strings.Builder)
|
||||
r.root.writestr(buf, "")
|
||||
return buf.String()
|
||||
}
|
||||
|
||||
type node struct {
|
||||
part string
|
||||
child []*node
|
||||
|
@ -152,12 +167,7 @@ func (n *node) add(parts []string) {
|
|||
}
|
||||
|
||||
func (n *node) match(parts []string) bool {
|
||||
if len(parts) == 0 {
|
||||
// Invalid domain.
|
||||
return false
|
||||
}
|
||||
|
||||
for {
|
||||
for len(parts) > 0 {
|
||||
// Pop next domain part.
|
||||
i := len(parts) - 1
|
||||
part := parts[i]
|
||||
|
@ -181,6 +191,10 @@ func (n *node) match(parts []string) bool {
|
|||
// child node.
|
||||
n = nn
|
||||
}
|
||||
|
||||
// Ran out of parts
|
||||
// without a match.
|
||||
return false
|
||||
}
|
||||
|
||||
// getChild fetches child node with given domain part string
|
||||
|
@ -222,3 +236,22 @@ func (n *node) sort() {
|
|||
child.sort()
|
||||
}
|
||||
}
|
||||
|
||||
func (n *node) writestr(buf *strings.Builder, prefix string) {
|
||||
if prefix != "" {
|
||||
// Suffix joining '.'
|
||||
prefix += "."
|
||||
}
|
||||
|
||||
// Append current part.
|
||||
prefix += n.part
|
||||
|
||||
// Dump current prefix state.
|
||||
buf.WriteString(prefix)
|
||||
buf.WriteByte('\n')
|
||||
|
||||
// Iterate through node children.
|
||||
for _, child := range n.child {
|
||||
child.writestr(buf, prefix)
|
||||
}
|
||||
}
|
||||
|
|
2
internal/cache/domain/domain_test.go
vendored
2
internal/cache/domain/domain_test.go
vendored
|
@ -69,7 +69,9 @@ func TestBlockCache(t *testing.T) {
|
|||
}
|
||||
|
||||
// Clear the cache
|
||||
t.Logf("%+v\n", c)
|
||||
c.Clear()
|
||||
t.Logf("%+v\n", c)
|
||||
|
||||
knownErr := errors.New("known error")
|
||||
|
||||
|
|
Loading…
Reference in a new issue