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)

相关推荐
2401_882727573 分钟前
BY组态-低代码web可视化组件
前端·后端·物联网·低代码·数学建模·前端框架
baivfhpwxf20234 分钟前
C# 5000 转16进制 字节(激光器串口通讯生成指定格式命令)
开发语言·c#
许嵩667 分钟前
IC脚本之perl
开发语言·perl
长亭外的少年18 分钟前
Kotlin 编译失败问题及解决方案:从守护进程到 Gradle 配置
android·开发语言·kotlin
直裾18 分钟前
Scala全文单词统计
开发语言·c#·scala
心仪悦悦18 分钟前
Scala中的集合复习(1)
开发语言·后端·scala
JIAY_WX21 分钟前
kotlin
开发语言·kotlin
代码小鑫1 小时前
A043-基于Spring Boot的秒杀系统设计与实现
java·开发语言·数据库·spring boot·后端·spring·毕业设计
真心喜欢你吖1 小时前
SpringBoot与MongoDB深度整合及应用案例
java·spring boot·后端·mongodb·spring