從官方庫學到了寫代碼的新方式（1）

2025年4月15日#go #迭代器 #工具14

AI 翻譯

這篇文章透過AI由簡體中文翻譯成繁體中文。查看原文

AI 生成的摘要

這個系列會比較簡短，主要用作便簽。最近計劃重寫項目，並從其他庫中尋找靈感，特別是如何處理SSE數據流。參考了gemini golang的庫，使用了iter.Seq2來生成迭代器函數。函數的簽名為`type Seq2[K, V any] func(yield func(K, V) bool)`。以下是一個簡單的例子，展示了如何迭代一個字符串數組。當遇到"exit"時，迭代器會返回錯誤並中斷；如果yield返回false，則停止迭代。使用for range遍歷iterateArray返回的序列時，for range循環機制會提供yield函數。

這個系列會比較簡短，當做便簽使用吧

從官方庫偷 xue 到的新的使用方式，最近打算重寫項目，打算從別的庫找找靈感，看看他們是怎麼處理 SSE 的數據流的，看了 gemini golang 的庫，他用到了 iter,Seq2 這個來生成迭代器函數

函數簽名是這樣的

type Seq2[K, V any] func(yield func(K, V) bool)

下面簡單寫一個例子

package main

import (
	"fmt"
	"iter"
)

func iterateArray(a []string) iter.Seq2[string, error] { 
	return func(yield func(string, error) bool) {
		for _, s := range a {
			if s == "exit" {
				yield("", fmt.Errorf("exit")) //返回錯誤並中斷
				return 
			}
			if !yield(s, nil) { // if yield returns false, stop iterating
				return
			}
		}
	}
}

func main() {
	ia := iterateArray([]string{"a", "b", "c", "exit", "d"})
	for k, v := range ia {
		fmt.Println(k, v)
	}
}

當你使用 for range 來遍歷 iterateArray 返回的序列時，for range 循環機制會提供這個 yield 函數。

yield 返回 false 表示 for range 中斷了，停止迭代

當接收到 "exit" 迭代器主動停止