archive

MovieNight/superlock.go

@@ -0,0 +1,78 @@
+package main
+
+import (
+	"log"
+	"sync"
+	"time"
+)
+
+type SuperLock struct {
+	self *sync.Mutex
+	lock *sync.RWMutex
+	sem  chan struct{}
+}
+
+func NewSuperLock() *SuperLock {
+	return &SuperLock{
+		sem:  make(chan struct{}, 100),
+		self: &sync.Mutex{},
+		lock: &sync.RWMutex{},
+	}
+}
+
+func (sl *SuperLock) RLock() {
+	log.Println("sl.rlock...")
+	sl.lock.RLock()
+}
+
+func (sl *SuperLock) RUnlock() {
+	log.Println("sl.runlock")
+	sl.lock.RUnlock()
+}
+
+func (sl *SuperLock) Lock() {
+	log.Println("sl.lock...")
+	sl.lock.Lock()
+}
+
+func (sl *SuperLock) Unlock() {
+	log.Println("sl.unlock")
+	sl.lock.Unlock()
+}
+
+func (sl *SuperLock) StartStream() {
+	log.Println("sl.startstream")
+	sl.self.Lock()
+	defer sl.self.Unlock()
+
+	select {
+	case sl.sem <- struct{}{}:
+	case <-time.After(time.Second * 3):
+		log.Println("timed out getting semaphore to start stream")
+		return
+	}
+
+	log.Println("CONT STREAM", len(sl.sem))
+	startStream()
+}
+
+func (sl *SuperLock) StopStream() {
+	log.Println("sl.stopstream")
+	sl.self.Lock()
+	defer sl.self.Unlock()
+
+	select {
+	case <-sl.sem:
+	case <-time.After(time.Second * 3):
+		log.Println("timed out getting semaphore to stop stream")
+		return
+	}
+
+	log.Println("STOP STREAM", len(sl.sem))
+	if len(sl.sem) > 0 {
+		return
+	}
+
+	log.Println("REALLY DO STOP STREAM", len(sl.sem))
+	stopStream()
+}