init: 1.0.0

pkg/syncx/managedresource.go

@@ -0,0 +1,48 @@
+package syncx
+
+import "sync"
+
+// A ManagedResource is used to manage a resource that might be broken and refetched, like a connection.
+type ManagedResource struct {
+	resource any
+	lock     sync.RWMutex
+	generate func() any
+	equals   func(a, b any) bool
+}
+
+// NewManagedResource returns a ManagedResource.
+func NewManagedResource(generate func() any, equals func(a, b any) bool) *ManagedResource {
+	return &ManagedResource{
+		generate: generate,
+		equals:   equals,
+	}
+}
+
+// MarkBroken marks the resource broken.
+func (mr *ManagedResource) MarkBroken(resource any) {
+	mr.lock.Lock()
+	defer mr.lock.Unlock()
+
+	if mr.equals(mr.resource, resource) {
+		mr.resource = nil
+	}
+}
+
+// Take takes the resource, if not loaded, generates it.
+func (mr *ManagedResource) Take() any {
+	mr.lock.RLock()
+	resource := mr.resource
+	mr.lock.RUnlock()
+
+	if resource != nil {
+		return resource
+	}
+
+	mr.lock.Lock()
+	defer mr.lock.Unlock()
+	// maybe another Take() call already generated the resource.
+	if mr.resource == nil {
+		mr.resource = mr.generate()
+	}
+	return mr.resource
+}