142 lines
4.0 KiB
Go
142 lines
4.0 KiB
Go
package main
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"log"
|
||
"path/filepath"
|
||
"runtime"
|
||
|
||
"git.fsdpf.net/go/gobridge"
|
||
)
|
||
|
||
type User struct {
|
||
ID int `json:"id"`
|
||
Name string `json:"name"`
|
||
Score float64 `json:"score"`
|
||
Level string `json:"level,omitempty"`
|
||
}
|
||
|
||
func main() {
|
||
_, file, _, _ := runtime.Caller(0)
|
||
script := filepath.Join(filepath.Dir(file), "worker.py")
|
||
|
||
pool, err := gobridge.NewPool(script,
|
||
gobridge.WithWorkers(2),
|
||
gobridge.WithMaxConns(4),
|
||
)
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
defer pool.Close()
|
||
|
||
ctx := context.Background()
|
||
|
||
// ── 普通调用 ──────────────────────────────────────────────────────────
|
||
sum, err := gobridge.Invoke[int](ctx, pool, "add", 3, 4)
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
fmt.Println("add(3, 4) =", sum) // 7
|
||
|
||
// ── 流式输出：Python yield → Go channel ──────────────────────────────
|
||
ch, err := gobridge.Invoke[chan int](ctx, pool, "range_gen", 1, 6)
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
fmt.Print("range_gen(1, 6) =")
|
||
for v := range ch {
|
||
fmt.Print(" ", v)
|
||
}
|
||
fmt.Println() // 1 2 3 4 5
|
||
|
||
// ── 流式输入：Go channel → Python Iterator ───────────────────────────
|
||
inputCh := make(chan int, 10)
|
||
go func() {
|
||
for i := 1; i <= 5; i++ {
|
||
inputCh <- i
|
||
}
|
||
close(inputCh)
|
||
}()
|
||
total, err := gobridge.Invoke[int](ctx, pool, "sum_stream", inputCh)
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
fmt.Println("sum_stream(1..5) =", total) // 15
|
||
|
||
// ── 双向流：Go channel 输入 + Go channel 输出 ────────────────────────
|
||
inputCh2 := make(chan int, 10)
|
||
go func() {
|
||
for i := 1; i <= 5; i++ {
|
||
inputCh2 <- i
|
||
}
|
||
close(inputCh2)
|
||
}()
|
||
outCh, err := gobridge.Invoke[chan int](ctx, pool, "double_stream", inputCh2)
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
fmt.Print("double_stream(1..5) =")
|
||
for v := range outCh {
|
||
fmt.Print(" ", v)
|
||
}
|
||
fmt.Println() // 1 4 9 16 25
|
||
|
||
// ── struct 普通调用 ───────────────────────────────────────────────────
|
||
user, err := gobridge.Invoke[User](ctx, pool, "get_user", 42)
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
fmt.Printf("get_user(42) = %+v\n", user)
|
||
|
||
// ── slice 输入，返回标量 ───────────────────────────────────────────────
|
||
users := []User{
|
||
{ID: 1, Name: "alice", Score: 5.0},
|
||
{ID: 2, Name: "bob", Score: 8.0},
|
||
{ID: 3, Name: "carol", Score: 12.0},
|
||
}
|
||
scoreSum, err := gobridge.Invoke[float64](ctx, pool, "total_score", users)
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
fmt.Printf("total_score([alice,bob,carol]) = %.1f\n", scoreSum)
|
||
|
||
// ── slice 输入输出 ─────────────────────────────────────────────────────
|
||
enriched, err := gobridge.Invoke[[]User](ctx, pool, "enrich_users", users)
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
fmt.Println("enrich_users:")
|
||
for _, u := range enriched {
|
||
fmt.Printf(" %+v\n", u)
|
||
}
|
||
|
||
// ── 流式输出 struct：Python yield User → Go chan User ─────────────────
|
||
userCh, err := gobridge.Invoke[chan User](ctx, pool, "gen_users", 3)
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
fmt.Print("gen_users(3) =")
|
||
for u := range userCh {
|
||
fmt.Printf(" {%d %s %.0f}", u.ID, u.Name, u.Score)
|
||
}
|
||
fmt.Println()
|
||
|
||
// ── 双向流 struct：Go chan User 输入 → Python 处理 → Go chan User 输出 ─
|
||
inCh := make(chan User, 5)
|
||
go func() {
|
||
for _, u := range users {
|
||
inCh <- u
|
||
}
|
||
close(inCh)
|
||
}()
|
||
procCh, err := gobridge.Invoke[chan User](ctx, pool, "process_users", inCh)
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
fmt.Println("process_users:")
|
||
for u := range procCh {
|
||
fmt.Printf(" %+v\n", u)
|
||
}
|
||
}
|