mirror of
				https://github.com/superseriousbusiness/gotosocial.git
				synced 2025-11-03 23:52:26 -06:00 
			
		
		
		
	
		
			
	
	
		
			68 lines
		
	
	
	
		
			2.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
		
		
			
		
	
	
			68 lines
		
	
	
	
		
			2.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| 
								 | 
							
								package cache
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								import "time"
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								// Cache represents a TTL cache with customizable callbacks, it
							 | 
						||
| 
								 | 
							
								// exists here to abstract away the "unsafe" methods in the case that
							 | 
						||
| 
								 | 
							
								// you do not want your own implementation atop TTLCache{}.
							 | 
						||
| 
								 | 
							
								type Cache[Key comparable, Value any] interface {
							 | 
						||
| 
								 | 
							
									// Start will start the cache background eviction routine with given sweep frequency.
							 | 
						||
| 
								 | 
							
									// If already running or a freq <= 0 provided, this is a no-op. This will block until
							 | 
						||
| 
								 | 
							
									// the eviction routine has started
							 | 
						||
| 
								 | 
							
									Start(freq time.Duration) bool
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									// Stop will stop cache background eviction routine. If not running this is a no-op. This
							 | 
						||
| 
								 | 
							
									// will block until the eviction routine has stopped
							 | 
						||
| 
								 | 
							
									Stop() bool
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									// SetEvictionCallback sets the eviction callback to the provided hook
							 | 
						||
| 
								 | 
							
									SetEvictionCallback(hook Hook[Key, Value])
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									// SetInvalidateCallback sets the invalidate callback to the provided hook
							 | 
						||
| 
								 | 
							
									SetInvalidateCallback(hook Hook[Key, Value])
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									// SetTTL sets the cache item TTL. Update can be specified to force updates of existing items in
							 | 
						||
| 
								 | 
							
									// the cache, this will simply add the change in TTL to their current expiry time
							 | 
						||
| 
								 | 
							
									SetTTL(ttl time.Duration, update bool)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									// Get fetches the value with key from the cache, extending its TTL
							 | 
						||
| 
								 | 
							
									Get(key Key) (value Value, ok bool)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									// Put attempts to place the value at key in the cache, doing nothing if
							 | 
						||
| 
								 | 
							
									// a value with this key already exists. Returned bool is success state
							 | 
						||
| 
								 | 
							
									Put(key Key, value Value) bool
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									// Set places the value at key in the cache. This will overwrite any
							 | 
						||
| 
								 | 
							
									// existing value, and call the update callback so. Existing values
							 | 
						||
| 
								 | 
							
									// will have their TTL extended upon update
							 | 
						||
| 
								 | 
							
									Set(key Key, value Value)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									// CAS will attempt to perform a CAS operation on 'key', using provided
							 | 
						||
| 
								 | 
							
									// comparison and swap values. Returned bool is success.
							 | 
						||
| 
								 | 
							
									CAS(key Key, cmp, swp Value) bool
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									// Swap will attempt to perform a swap on 'key', replacing the value there
							 | 
						||
| 
								 | 
							
									// and returning the existing value. If no value exists for key, this will
							 | 
						||
| 
								 | 
							
									// set the value and return the zero value for V.
							 | 
						||
| 
								 | 
							
									Swap(key Key, swp Value) Value
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									// Has checks the cache for a value with key, this will not update TTL
							 | 
						||
| 
								 | 
							
									Has(key Key) bool
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									// Invalidate deletes a value from the cache, calling the invalidate callback
							 | 
						||
| 
								 | 
							
									Invalidate(key Key) bool
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									// Clear empties the cache, calling the invalidate callback
							 | 
						||
| 
								 | 
							
									Clear()
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									// Size returns the current size of the cache
							 | 
						||
| 
								 | 
							
									Size() int
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								// New returns a new initialized Cache.
							 | 
						||
| 
								 | 
							
								func New[K comparable, V any]() Cache[K, V] {
							 | 
						||
| 
								 | 
							
									c := TTLCache[K, V]{}
							 | 
						||
| 
								 | 
							
									c.Init()
							 | 
						||
| 
								 | 
							
									return &c
							 | 
						||
| 
								 | 
							
								}
							 |