forked from mirrors/gotosocial
7cc40302a5
* add miekg/dns dependency * set/validate accountDomain * move finger to dereferencer * totally break GetRemoteAccount * start reworking finger func a bit * start reworking getRemoteAccount a bit * move mention parts to namestring * rework webfingerget * use util function to extract webfinger parts * use accountDomain * rework finger again, final form * just a real nasty commit, the worst * remove refresh from account * use new ASRepToAccount signature * fix incorrect debug call * fix for new getRemoteAccount * rework GetRemoteAccount * start updating tests to remove repetition * break a lot of tests Move shared test logic into the testrig, rather than having it scattered all over the place. This allows us to just mock the transport controller once, and have all tests use it (unless they need not to for some other reason). * fix up tests to use main mock httpclient * webfinger only if necessary * cheeky linting with the lads * update mentionName regex recognize instance accounts * don't finger instance accounts * test webfinger part extraction * increase default worker count to 4 per cpu * don't repeat regex parsing * final search for discovered accountDomain * be more permissive in namestring lookup * add more extraction tests * simplify GetParseMentionFunc * skip long search if local account * fix broken test * consolidate to all use same caching libraries Signed-off-by: kim <grufwub@gmail.com> * perform more caching in the database layer Signed-off-by: kim <grufwub@gmail.com> * remove ASNote cache Signed-off-by: kim <grufwub@gmail.com> * update cache library, improve db tracing hooks Signed-off-by: kim <grufwub@gmail.com> * return ErrNoEntries if no account status IDs found, small formatting changes Signed-off-by: kim <grufwub@gmail.com> * fix tests, thanks tobi! Signed-off-by: kim <grufwub@gmail.com> Co-authored-by: tsmethurst <tobi.smethurst@protonmail.com>
58 lines
1.4 KiB
Go
58 lines
1.4 KiB
Go
package atomics
|
|
|
|
import "sync"
|
|
|
|
// State provides user-friendly means of performing atomic-like
|
|
// operations on a uint32 state, and allowing callbacks on successful
|
|
// state change. This is a bit of a misnomer being where it is, as it
|
|
// actually uses a mutex under-the-hood.
|
|
type State struct {
|
|
mutex sync.Mutex
|
|
state uint32
|
|
}
|
|
|
|
// Store will update State value safely within mutex lock.
|
|
func (st *State) Store(val uint32) {
|
|
st.mutex.Lock()
|
|
st.state = val
|
|
st.mutex.Unlock()
|
|
}
|
|
|
|
// Load will get value of State safely within mutex lock.
|
|
func (st *State) Load() uint32 {
|
|
st.mutex.Lock()
|
|
state := st.state
|
|
st.mutex.Unlock()
|
|
return state
|
|
}
|
|
|
|
// WithLock performs fn within State mutex lock, useful if you want
|
|
// to just use State's mutex for locking instead of creating another.
|
|
func (st *State) WithLock(fn func()) {
|
|
st.mutex.Lock()
|
|
defer st.mutex.Unlock()
|
|
fn()
|
|
}
|
|
|
|
// Update performs fn within State mutex lock, with the current state
|
|
// value provided as an argument, and return value used to update state.
|
|
func (st *State) Update(fn func(state uint32) uint32) {
|
|
st.mutex.Lock()
|
|
defer st.mutex.Unlock()
|
|
st.state = fn(st.state)
|
|
}
|
|
|
|
// CAS performs a compare-and-swap on State, calling fn on success. Success value is also returned.
|
|
func (st *State) CAS(cmp, swp uint32, fn func()) (ok bool) {
|
|
// Acquire lock
|
|
st.mutex.Lock()
|
|
defer st.mutex.Unlock()
|
|
|
|
// Perform CAS operation, fn() on success
|
|
if ok = (st.state == cmp); ok {
|
|
st.state = swp
|
|
fn()
|
|
}
|
|
|
|
return
|
|
}
|