golang 函数式编程库samber/mo使用: Future

golang 函数式编程库samber/mo使用: Future

如果您对samber/mo库不了解, 请先阅读第一篇 Option

本节讲述Future的使用,它可以帮助我们处理异步编程问题。

示例

我们先来看看下面代码的示例, 注释解释了每一步的操作。

go 复制代码
package main

import (
	"fmt"

	"github.com/samber/mo"
)

func main() {
	// resolve 在这里只是一个定义, NewFuture会以一个 goroutine 的方式执行 cb, 并且传递Future 的 resolve 和 reject
	value, err := mo.NewFuture(func(resolve func(string), reject func(error)) {
		// do something here
		if true { // 这里假定 do something 成功
			// 如果 do something 成功, 执行 resolve, 并传递一个值, 然后会执行 Then
			resolve("foobar")
		} else {
			// 告诉 do something 失败, 执行 reject, 并传递一个错误, 然后会执行 Catch
			reject(fmt.Errorf("failure"))
		}
	}).
		Then(func(s string) (string, error) {
			// 这里 s 就是 resolve 传递的值
			return s, nil
		}).
		Catch(func(err error) (string, error) {
			// 这里 err 就是 reject 传递的错误
			return "foobar", nil
		}).
		Finally(func(value string, err error) (string, error) {
			// 不管发生什么都会执行, value 是 resolve 传递的值, err 是 reject 传递的错误
			return value, nil
		}).
		Collect() // 等待 future 执行完毕, 并返回最终的值

	fmt.Println(value)
	fmt.Println(err)
	// Output:
	// foobar
	// <nil>
}

源码解析

根据mo.NewFuture的实现, 可以看出该函数做的事情就是构造一个Future, 然后执行activate函数, activate实际就是用 goroutine 执行cb函数, 并且将 Future 的 resolve 和 reject函数作为参数传递给cb

go 复制代码
func NewFuture[T any](cb func(resolve func(T), reject func(error))) *Future[T] {
	future := Future[T]{
		cb:       cb,
		cancelCb: func() {},
		done:     make(chan struct{}),
	}

	future.active()

	return &future
}

func (f *Future[T]) active() {
	go f.cb(f.resolve, f.reject)
}

resolve的实现如下, 可以看到resolve做的事情就是用mo.OK包装value, 记录到result中,并且关闭f.done, 表明future已经完成。

resolve加锁的目的是为了确保后续Then或Finally不会同时进行。

go 复制代码
func (f *Future[T]) resolve(value T) {
	f.mu.Lock()
	defer f.mu.Unlock()

	f.result = Ok(value)
	if f.next != nil { 
	    // 这里如果不为空,表明next先于something注册,需要执行
		f.next.activeSync()
	}
	close(f.done)
}

我们来看看Then的实现,这个函数先对f执行加锁, 然后构造一个新的Future,这个新的Future的cb函数就是为了判断f的执行结果, 如果f的result不是error, 就执行Then注册的回调 cb。 所以如果f.cb函数执行resolve后返回, f.result.IsError()为false, 会执行Then中的回调。

最后的select表示如果f已完成,用goroutine 执行Then中的回调。如果f还没有完成,则留待f.cb的resolve或reject执行Then的回调。两种情况都会直接返回f.next,不会阻塞。这样就实现了Future的串联。

go 复制代码
func (f *Future[T]) Then(cb func(T) (T, error)) *Future[T] {
	f.mu.Lock()
	defer f.mu.Unlock()

	f.next = &Future[T]{
		cb: func(resolve func(T), reject func(error)) {
			if f.result.IsError() {
				reject(f.result.Error())
				return
			}
			newValue, err := cb(f.result.MustGet())
			if err != nil {
				reject(err)
				return
			}
			resolve(newValue)
		},
		cancelCb: func() {
			f.Cancel()
		},
		done: make(chan struct{}),
	}

	select {
	case <-f.done:
		f.next.active()
	default:
	}
	return f.next
}

CatchThen的区别是如果f.result是error, 就执行Catch中的回调。Finally是不管f.result是什么, 都会执行Finally中的回调。

最后的Collect是用于等待Future执行完毕, 并返回最终的值。

go 复制代码
func (f *Future[T]) Collect() (T, error) {
	<-f.done
	return f.result.Get()
}

还有 ResultEither方法, 用于获取Future的执行结果, 会阻塞直到Future执行完毕(也就是先执行Collect)

相关推荐
wainyz几秒前
Java NIO操作
java·开发语言·nio
喵叔哟9 分钟前
重构代码之用委托替代继承
开发语言·重构
lzb_kkk14 分钟前
【JavaEE】JUC的常见类
java·开发语言·java-ee
SEEONTIME14 分钟前
python-24-一篇文章彻底掌握Python HTTP库Requests
开发语言·python·http·http库requests
起名字真南33 分钟前
【OJ题解】C++实现字符串大数相乘:无BigInteger库的字符串乘积解决方案
开发语言·c++·leetcode
tyler_download44 分钟前
golang 实现比特币内核:实现基于椭圆曲线的数字签名和验证
开发语言·数据库·golang
小小小~1 小时前
qt5将程序打包并使用
开发语言·qt
hlsd#1 小时前
go mod 依赖管理
开发语言·后端·golang
小春学渗透1 小时前
Day107:代码审计-PHP模型开发篇&MVC层&RCE执行&文件对比法&1day分析&0day验证
开发语言·安全·web安全·php·mvc
杜杜的man1 小时前
【go从零单排】迭代器(Iterators)
开发语言·算法·golang