golang多线程之future

灰太狼 2022-12-02 13:42 194阅读 0赞

package main
    
    import (
    	"fmt"
    	"sync"
    	"time"
    )
    
    // Completer用于获取未来的结果。仅仅用于接受
    type Completer <-chan interface{
        }
    
    // Future用于构建异步任务。 等这个任务完成之后，所以关心这个结果的任务[其他任务可能在干别的事情]都可以接受到消息。
    type Future struct {
        
    	triggered bool
    	item      interface{
        }
    	err       error
    	lock      sync.Mutex
    	wg        sync.WaitGroup
    }
    
    // 阻塞等待，知道结果返回
    func (f *Future) GetResult()(interface{
        }, error)  {
        
    	f.lock.Lock()
    	if(f.triggered){
          // 不需要阻塞就可以获取结果
    		f.lock.Unlock()
    		return f.item, f.err
    	}
    	f.lock.Unlock()
    
    	f.wg.Wait()  // 阻塞等待返回结果
    	return f.item, f.err
    }
    
    // 任务是否过期： 是否有结果可以返回
    func (f *Future)HasResult() bool {
        
    	f.lock.Lock()
    	hasResult := f.triggered
    	f.lock.Unlock()
    	return hasResult
    }
    
    func (f *Future) setItem(item interface{
        }, err error) {
        
    	f.lock.Lock()
    	f.triggered = true // 超时或者已经获取到结果
    	f.item = item
    	f.err = err
    	f.lock.Unlock()
    	f.wg.Done()
    }
    
    func listenForResult(f *Future, ch Completer, timeout time.Duration, wg *sync.WaitGroup) {
        
    	wg.Done() // 创建监听器任务已经完成
    	t := time.NewTimer(timeout)
    	select {
        
    	case item := <-ch:
    		f.setItem(item, nil)
    		t.Stop() //我们想在不再需要这个计时器时立即触发它的GC
    	case <- t.C:
    		f.setItem(nil, fmt.Errorf(`timeout after %f seconds`, timeout.Seconds()))
    	}
    }
    
    //New是生成future的构造函数。将完成的项目传递到toComplete频道，所有侦听器都将收到通知。
    //如果在调用toComplete之前超时，则任何侦听器都将被传递一个错误。
    func New(completer Completer, timeout time.Duration) *Future {
        
    	f := &Future{
        }
    	f.wg.Add(1)
    	var wg sync.WaitGroup
    	wg.Add(1)
    	go listenForResult(f, completer, timeout, &wg)
    	wg.Wait() // 阻塞等待未来任务完成
    	return f
    }