File: pool.go

package info (click to toggle)
golang-github-icza-gox 0.0~git20210726.cd40a3f-4
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, forky, sid, trixie
  • size: 332 kB
  • sloc: makefile: 2
file content (33 lines) | stat: -rw-r--r-- 659 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
package stringsx

import "sync"

// Pool implements a string pool safe for concurrent use.
//
// For details, see https://stackoverflow.com/a/51983331/1705598
type Pool struct {
	mu    sync.Mutex
	cache map[string]string
}

// NewPool returns a new, empty string pool.
func NewPool() *Pool {
	return &Pool{
		cache: map[string]string{},
	}
}

// Interned returns the string instance from the pool that is equal to s.
// If s is not yet in the pool, it is put into it and returned.
func (p *Pool) Interned(s string) string {
	p.mu.Lock()
	defer p.mu.Unlock()

	if s2, ok := p.cache[s]; ok {
		return s2
	}

	// New string, store it
	p.cache[s] = s
	return s
}