async: work both go and llgo
This commit is contained in:
@@ -5,7 +5,6 @@ import (
|
|||||||
"os"
|
"os"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/goplus/llgo/c"
|
|
||||||
"github.com/goplus/llgo/x/async"
|
"github.com/goplus/llgo/x/async"
|
||||||
"github.com/goplus/llgo/x/async/timeout"
|
"github.com/goplus/llgo/x/async/timeout"
|
||||||
"github.com/goplus/llgo/x/tuple"
|
"github.com/goplus/llgo/x/tuple"
|
||||||
@@ -31,10 +30,9 @@ func WriteFile(fileName string, content []byte) async.IO[error] {
|
|||||||
|
|
||||||
func sleep(i int, d time.Duration) async.IO[int] {
|
func sleep(i int, d time.Duration) async.IO[int] {
|
||||||
return async.Async(func(resolve func(int)) {
|
return async.Async(func(resolve func(int)) {
|
||||||
go func() {
|
async.BindIO(timeout.Timeout(d), func(async.Void) {
|
||||||
c.Usleep(c.Uint(d.Microseconds()))
|
|
||||||
resolve(i)
|
resolve(i)
|
||||||
}()
|
})
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -46,6 +44,8 @@ func main() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func RunIO() {
|
func RunIO() {
|
||||||
|
println("RunIO with Await")
|
||||||
|
|
||||||
async.Run(func() {
|
async.Run(func() {
|
||||||
content, err := async.Await(ReadFile("1.txt")).Get()
|
content, err := async.Await(ReadFile("1.txt")).Get()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -62,6 +62,7 @@ func RunIO() {
|
|||||||
})
|
})
|
||||||
|
|
||||||
// Translated to in Go+:
|
// Translated to in Go+:
|
||||||
|
println("RunIO with BindIO")
|
||||||
|
|
||||||
async.Run(func() {
|
async.Run(func() {
|
||||||
async.BindIO(ReadFile("1.txt"), func(v tuple.Tuple2[[]byte, error]) {
|
async.BindIO(ReadFile("1.txt"), func(v tuple.Tuple2[[]byte, error]) {
|
||||||
@@ -84,30 +85,42 @@ func RunIO() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func RunAllAndRace() {
|
func RunAllAndRace() {
|
||||||
|
ms100 := 100 * time.Millisecond
|
||||||
|
ms200 := 200 * time.Millisecond
|
||||||
|
ms300 := 300 * time.Millisecond
|
||||||
|
|
||||||
|
println("Run All with Await")
|
||||||
|
|
||||||
async.Run(func() {
|
async.Run(func() {
|
||||||
all := async.All(sleep(1, time.Second), sleep(2, time.Second*2), sleep(3, time.Second*3))
|
all := async.All(sleep(1, ms200), sleep(2, ms100), sleep(3, ms300))
|
||||||
async.BindIO(all, func(v []int) {
|
async.BindIO(all, func(v []int) {
|
||||||
fmt.Printf("All: %v\n", v)
|
fmt.Printf("All: %v\n", v)
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
|
println("Run Race with Await")
|
||||||
|
|
||||||
async.Run(func() {
|
async.Run(func() {
|
||||||
first := async.Race(sleep(1, time.Second), sleep(2, time.Second*2), sleep(3, time.Second*3))
|
first := async.Race(sleep(1, ms200), sleep(2, ms100), sleep(3, ms300))
|
||||||
v := async.Await(first)
|
v := async.Await(first)
|
||||||
fmt.Printf("Race: %v\n", v)
|
fmt.Printf("Race: %v\n", v)
|
||||||
})
|
})
|
||||||
|
|
||||||
// Translated to in Go+:
|
// Translated to in Go+:
|
||||||
|
|
||||||
|
println("Run All with BindIO")
|
||||||
|
|
||||||
async.Run(func() {
|
async.Run(func() {
|
||||||
all := async.All(sleep(1, time.Second), sleep(2, time.Second*2), sleep(3, time.Second*3))
|
all := async.All(sleep(1, ms200), sleep(2, ms100), sleep(3, ms300))
|
||||||
async.BindIO(all, func(v []int) {
|
async.BindIO(all, func(v []int) {
|
||||||
fmt.Printf("All: %v\n", v)
|
fmt.Printf("All: %v\n", v)
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
|
println("Run Race with BindIO")
|
||||||
|
|
||||||
async.Run(func() {
|
async.Run(func() {
|
||||||
first := async.Race(sleep(1, time.Second), sleep(2, time.Second*2), sleep(3, time.Second*3))
|
first := async.Race(sleep(1, ms200), sleep(2, ms100), sleep(3, ms300))
|
||||||
async.BindIO(first, func(v int) {
|
async.BindIO(first, func(v int) {
|
||||||
fmt.Printf("Race: %v\n", v)
|
fmt.Printf("Race: %v\n", v)
|
||||||
})
|
})
|
||||||
@@ -115,17 +128,21 @@ func RunAllAndRace() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func RunTimeout() {
|
func RunTimeout() {
|
||||||
|
println("Run Timeout with Await")
|
||||||
|
|
||||||
async.Run(func() {
|
async.Run(func() {
|
||||||
fmt.Printf("Start 1 second timeout\n")
|
fmt.Printf("Start 100 ms timeout\n")
|
||||||
async.Await(timeout.Timeout(1 * time.Second))
|
async.Await(timeout.Timeout(100 * time.Millisecond))
|
||||||
fmt.Printf("timeout\n")
|
fmt.Printf("timeout\n")
|
||||||
})
|
})
|
||||||
|
|
||||||
// Translated to in Go+:
|
// Translated to in Go+:
|
||||||
|
|
||||||
|
println("Run Timeout with BindIO")
|
||||||
|
|
||||||
async.Run(func() {
|
async.Run(func() {
|
||||||
fmt.Printf("Start 1 second timeout\n")
|
fmt.Printf("Start 100 ms timeout\n")
|
||||||
async.BindIO(timeout.Timeout(1*time.Second), func(async.Void) {
|
async.BindIO(timeout.Timeout(100*time.Millisecond), func(async.Void) {
|
||||||
fmt.Printf("timeout\n")
|
fmt.Printf("timeout\n")
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|||||||
103
x/async/async.go
103
x/async/async.go
@@ -17,11 +17,7 @@
|
|||||||
package async
|
package async
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
|
||||||
"unsafe"
|
|
||||||
_ "unsafe"
|
_ "unsafe"
|
||||||
|
|
||||||
"github.com/goplus/llgo/c/libuv"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type Void = [0]byte
|
type Void = [0]byte
|
||||||
@@ -30,16 +26,13 @@ type Future[T any] func() T
|
|||||||
|
|
||||||
type IO[T any] func(e *AsyncContext) Future[T]
|
type IO[T any] func(e *AsyncContext) Future[T]
|
||||||
|
|
||||||
type Chain[T any] func(callback func(T))
|
type AsyncContext struct {
|
||||||
|
*Executor
|
||||||
func (f Future[T]) Do(callback func(T)) {
|
complete func()
|
||||||
callback(f())
|
|
||||||
}
|
}
|
||||||
|
|
||||||
type AsyncContext struct {
|
func (ctx *AsyncContext) Complete() {
|
||||||
context.Context
|
ctx.complete()
|
||||||
*Executor
|
|
||||||
Complete func()
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func Async[T any](fn func(resolve func(T))) IO[T] {
|
func Async[T any](fn func(resolve func(T))) IO[T] {
|
||||||
@@ -53,93 +46,9 @@ func Async[T any](fn func(resolve func(T))) IO[T] {
|
|||||||
})
|
})
|
||||||
return func() T {
|
return func() T {
|
||||||
if !done {
|
if !done {
|
||||||
panic("AsyncIO: Future accessed before completion")
|
panic("async.Async: Future accessed before completion")
|
||||||
}
|
}
|
||||||
return result
|
return result
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
type bindAsync struct {
|
|
||||||
libuv.Async
|
|
||||||
cb func()
|
|
||||||
}
|
|
||||||
|
|
||||||
func BindIO[T any](call IO[T], callback func(T)) {
|
|
||||||
loop := Exec().L
|
|
||||||
a := &bindAsync{}
|
|
||||||
loop.Async(&a.Async, func(p *libuv.Async) {
|
|
||||||
(*bindAsync)(unsafe.Pointer(p)).cb()
|
|
||||||
})
|
|
||||||
ctx := &AsyncContext{
|
|
||||||
Context: context.Background(),
|
|
||||||
Executor: Exec(),
|
|
||||||
Complete: func() {
|
|
||||||
a.Async.Send()
|
|
||||||
},
|
|
||||||
}
|
|
||||||
f := call(ctx)
|
|
||||||
a.cb = func() {
|
|
||||||
a.Async.Close(nil)
|
|
||||||
result := f()
|
|
||||||
callback(result)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// -----------------------------------------------------------------------------
|
|
||||||
|
|
||||||
func Await[T1 any](call IO[T1]) (ret T1) {
|
|
||||||
ch := make(chan struct{})
|
|
||||||
f := call(&AsyncContext{
|
|
||||||
Context: context.Background(),
|
|
||||||
Executor: Exec(),
|
|
||||||
Complete: func() {
|
|
||||||
close(ch)
|
|
||||||
},
|
|
||||||
})
|
|
||||||
<-ch
|
|
||||||
return f()
|
|
||||||
}
|
|
||||||
|
|
||||||
func Race[T1 any](calls ...IO[T1]) IO[T1] {
|
|
||||||
return Async(func(resolve func(T1)) {
|
|
||||||
done := false
|
|
||||||
for _, call := range calls {
|
|
||||||
var f Future[T1]
|
|
||||||
f = call(&AsyncContext{
|
|
||||||
Context: context.Background(),
|
|
||||||
Executor: Exec(),
|
|
||||||
Complete: func() {
|
|
||||||
if done {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
done = true
|
|
||||||
resolve(f())
|
|
||||||
},
|
|
||||||
})
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
func All[T1 any](calls ...IO[T1]) IO[[]T1] {
|
|
||||||
return Async(func(resolve func([]T1)) {
|
|
||||||
n := len(calls)
|
|
||||||
results := make([]T1, n)
|
|
||||||
done := 0
|
|
||||||
for i, call := range calls {
|
|
||||||
i := i
|
|
||||||
var f Future[T1]
|
|
||||||
f = call(&AsyncContext{
|
|
||||||
Context: context.Background(),
|
|
||||||
Executor: Exec(),
|
|
||||||
Complete: func() {
|
|
||||||
results[i] = f()
|
|
||||||
done++
|
|
||||||
if done == n {
|
|
||||||
resolve(results)
|
|
||||||
}
|
|
||||||
},
|
|
||||||
})
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|||||||
91
x/async/async_go.go
Normal file
91
x/async/async_go.go
Normal file
@@ -0,0 +1,91 @@
|
|||||||
|
//go:build !llgo
|
||||||
|
// +build !llgo
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Copyright (c) 2024 The GoPlus Authors (goplus.org). All rights reserved.
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package async
|
||||||
|
|
||||||
|
import "sync"
|
||||||
|
|
||||||
|
func BindIO[T any](call IO[T], callback func(T)) {
|
||||||
|
callback(Await(call))
|
||||||
|
}
|
||||||
|
|
||||||
|
func Await[T1 any](call IO[T1]) (ret T1) {
|
||||||
|
ch := make(chan struct{})
|
||||||
|
f := call(&AsyncContext{
|
||||||
|
Executor: Exec(),
|
||||||
|
complete: func() {
|
||||||
|
close(ch)
|
||||||
|
},
|
||||||
|
})
|
||||||
|
<-ch
|
||||||
|
return f()
|
||||||
|
}
|
||||||
|
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
func Race[T1 any](calls ...IO[T1]) IO[T1] {
|
||||||
|
return Async(func(resolve func(T1)) {
|
||||||
|
ch := make(chan int, len(calls))
|
||||||
|
futures := make([]Future[T1], len(calls))
|
||||||
|
for i, call := range calls {
|
||||||
|
i := i
|
||||||
|
call := call
|
||||||
|
go func() {
|
||||||
|
f := call(&AsyncContext{
|
||||||
|
Executor: Exec(),
|
||||||
|
complete: func() {
|
||||||
|
defer func() {
|
||||||
|
_ = recover()
|
||||||
|
}()
|
||||||
|
ch <- i
|
||||||
|
},
|
||||||
|
})
|
||||||
|
futures[i] = f
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
i := <-ch
|
||||||
|
close(ch)
|
||||||
|
resolve(futures[i]())
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func All[T1 any](calls ...IO[T1]) IO[[]T1] {
|
||||||
|
return Async(func(resolve func([]T1)) {
|
||||||
|
n := len(calls)
|
||||||
|
results := make([]T1, n)
|
||||||
|
futures := make([]Future[T1], n)
|
||||||
|
wg := sync.WaitGroup{}
|
||||||
|
wg.Add(n)
|
||||||
|
for i, call := range calls {
|
||||||
|
i := i
|
||||||
|
f := call(&AsyncContext{
|
||||||
|
Executor: Exec(),
|
||||||
|
complete: func() {
|
||||||
|
wg.Done()
|
||||||
|
},
|
||||||
|
})
|
||||||
|
futures[i] = f
|
||||||
|
}
|
||||||
|
wg.Wait()
|
||||||
|
for i, f := range futures {
|
||||||
|
results[i] = f()
|
||||||
|
}
|
||||||
|
resolve(results)
|
||||||
|
})
|
||||||
|
}
|
||||||
113
x/async/async_llgo.go
Normal file
113
x/async/async_llgo.go
Normal file
@@ -0,0 +1,113 @@
|
|||||||
|
//go:build llgo
|
||||||
|
// +build llgo
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Copyright (c) 2024 The GoPlus Authors (goplus.org). All rights reserved.
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package async
|
||||||
|
|
||||||
|
import (
|
||||||
|
"sync/atomic"
|
||||||
|
"unsafe"
|
||||||
|
|
||||||
|
"github.com/goplus/llgo/c/libuv"
|
||||||
|
)
|
||||||
|
|
||||||
|
type bindAsync struct {
|
||||||
|
libuv.Async
|
||||||
|
cb func()
|
||||||
|
}
|
||||||
|
|
||||||
|
func BindIO[T any](call IO[T], callback func(T)) {
|
||||||
|
loop := Exec().L
|
||||||
|
a := &bindAsync{}
|
||||||
|
loop.Async(&a.Async, func(p *libuv.Async) {
|
||||||
|
(*bindAsync)(unsafe.Pointer(p)).cb()
|
||||||
|
})
|
||||||
|
done := atomic.Bool{}
|
||||||
|
ctx := &AsyncContext{
|
||||||
|
Executor: Exec(),
|
||||||
|
complete: func() {
|
||||||
|
done.Store(true)
|
||||||
|
a.Async.Send()
|
||||||
|
},
|
||||||
|
}
|
||||||
|
f := call(ctx)
|
||||||
|
called := false
|
||||||
|
a.cb = func() {
|
||||||
|
if called {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
a.Async.Close(nil)
|
||||||
|
result := f()
|
||||||
|
callback(result)
|
||||||
|
}
|
||||||
|
// don't delay the callback if the future is already done
|
||||||
|
if done.Load() {
|
||||||
|
called = true
|
||||||
|
a.cb()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func Await[T1 any](call IO[T1]) (ret T1) {
|
||||||
|
BindIO(call, func(v T1) {
|
||||||
|
ret = v
|
||||||
|
})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
func Race[T1 any](calls ...IO[T1]) IO[T1] {
|
||||||
|
return Async(func(resolve func(T1)) {
|
||||||
|
done := false
|
||||||
|
for _, call := range calls {
|
||||||
|
var f Future[T1]
|
||||||
|
f = call(&AsyncContext{
|
||||||
|
Executor: Exec(),
|
||||||
|
complete: func() {
|
||||||
|
if done {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
done = true
|
||||||
|
resolve(f())
|
||||||
|
},
|
||||||
|
})
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func All[T1 any](calls ...IO[T1]) IO[[]T1] {
|
||||||
|
return Async(func(resolve func([]T1)) {
|
||||||
|
n := len(calls)
|
||||||
|
results := make([]T1, n)
|
||||||
|
done := 0
|
||||||
|
for i, call := range calls {
|
||||||
|
i := i
|
||||||
|
var f Future[T1]
|
||||||
|
f = call(&AsyncContext{
|
||||||
|
Executor: Exec(),
|
||||||
|
complete: func() {
|
||||||
|
results[i] = f()
|
||||||
|
done++
|
||||||
|
if done == n {
|
||||||
|
resolve(results)
|
||||||
|
}
|
||||||
|
},
|
||||||
|
})
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
33
x/async/executor_go.go
Normal file
33
x/async/executor_go.go
Normal file
@@ -0,0 +1,33 @@
|
|||||||
|
//go:build !llgo
|
||||||
|
// +build !llgo
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Copyright (c) 2024 The GoPlus Authors (goplus.org). All rights reserved.
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package async
|
||||||
|
|
||||||
|
var exec = &Executor{}
|
||||||
|
|
||||||
|
type Executor struct {
|
||||||
|
}
|
||||||
|
|
||||||
|
func Exec() *Executor {
|
||||||
|
return exec
|
||||||
|
}
|
||||||
|
|
||||||
|
func Run(fn func()) {
|
||||||
|
fn()
|
||||||
|
}
|
||||||
@@ -1,3 +1,6 @@
|
|||||||
|
//go:build llgo
|
||||||
|
// +build llgo
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* Copyright (c) 2024 The GoPlus Authors (goplus.org). All rights reserved.
|
* Copyright (c) 2024 The GoPlus Authors (goplus.org). All rights reserved.
|
||||||
*
|
*
|
||||||
@@ -56,4 +59,5 @@ func Run(fn func()) {
|
|||||||
fn()
|
fn()
|
||||||
exec.Run()
|
exec.Run()
|
||||||
loop.Close()
|
loop.Close()
|
||||||
|
setExec(nil)
|
||||||
}
|
}
|
||||||
35
x/async/timeout/timeout_go.go
Normal file
35
x/async/timeout/timeout_go.go
Normal file
@@ -0,0 +1,35 @@
|
|||||||
|
//go:build !llgo
|
||||||
|
// +build !llgo
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Copyright (c) 2024 The GoPlus Authors (goplus.org). All rights reserved.
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package timeout
|
||||||
|
|
||||||
|
import (
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/goplus/llgo/x/async"
|
||||||
|
)
|
||||||
|
|
||||||
|
func Timeout(d time.Duration) async.IO[async.Void] {
|
||||||
|
return async.Async(func(resolve func(async.Void)) {
|
||||||
|
go func() {
|
||||||
|
time.Sleep(d)
|
||||||
|
resolve(async.Void{})
|
||||||
|
}()
|
||||||
|
})
|
||||||
|
}
|
||||||
44
x/async/timeout/timeout_llgo.go
Normal file
44
x/async/timeout/timeout_llgo.go
Normal file
@@ -0,0 +1,44 @@
|
|||||||
|
//go:build llgo
|
||||||
|
// +build llgo
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Copyright (c) 2024 The GoPlus Authors (goplus.org). All rights reserved.
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package timeout
|
||||||
|
|
||||||
|
import (
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/goplus/llgo/c/libuv"
|
||||||
|
"github.com/goplus/llgo/x/async"
|
||||||
|
"github.com/goplus/llgo/x/cbind"
|
||||||
|
)
|
||||||
|
|
||||||
|
func Timeout(d time.Duration) async.IO[async.Void] {
|
||||||
|
return async.Async(func(resolve func(async.Void)) {
|
||||||
|
t, _ := cbind.Bind[libuv.Timer](func() {
|
||||||
|
resolve(async.Void{})
|
||||||
|
})
|
||||||
|
r := libuv.InitTimer(async.Exec().L, t)
|
||||||
|
if r != 0 {
|
||||||
|
panic("InitTimer failed")
|
||||||
|
}
|
||||||
|
r = t.Start(cbind.Callback[libuv.Timer], uint64(d/time.Millisecond), 0)
|
||||||
|
if r != 0 {
|
||||||
|
panic("Start failed")
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
12
x/cbind/buf.go
Normal file
12
x/cbind/buf.go
Normal file
@@ -0,0 +1,12 @@
|
|||||||
|
package cbind
|
||||||
|
|
||||||
|
import "unsafe"
|
||||||
|
|
||||||
|
type slice struct {
|
||||||
|
data unsafe.Pointer
|
||||||
|
len int
|
||||||
|
}
|
||||||
|
|
||||||
|
func GoBytes(buf *int8, n int) []byte {
|
||||||
|
return *(*[]byte)(unsafe.Pointer(&slice{unsafe.Pointer(buf), n}))
|
||||||
|
}
|
||||||
@@ -16,7 +16,21 @@
|
|||||||
|
|
||||||
package cbind
|
package cbind
|
||||||
|
|
||||||
import "unsafe"
|
import (
|
||||||
|
"unsafe"
|
||||||
|
)
|
||||||
|
|
||||||
|
// llgo:type C
|
||||||
|
type Cb[T any] func(*T)
|
||||||
|
|
||||||
|
// llgo:type C
|
||||||
|
type Cb1[T any, A any] func(*T, A)
|
||||||
|
|
||||||
|
// llgo:type C
|
||||||
|
type Cb2[T any, A any, B any] func(*T, A, B)
|
||||||
|
|
||||||
|
// llgo:type C
|
||||||
|
type Cb3[T any, A any, B any, C any] func(*T, A, B, C)
|
||||||
|
|
||||||
type bind[Base any] struct {
|
type bind[Base any] struct {
|
||||||
b Base
|
b Base
|
||||||
@@ -38,50 +52,75 @@ type bind3[Base any, A any, B any, C any] struct {
|
|||||||
fn func(A, B, C)
|
fn func(A, B, C)
|
||||||
}
|
}
|
||||||
|
|
||||||
func callback[Base any](b *Base) {
|
func Callback[Base any](b *Base) {
|
||||||
bind := (*bind[Base])(unsafe.Pointer(b))
|
bind := (*bind[Base])(unsafe.Pointer(b))
|
||||||
bind.fn()
|
bind.fn()
|
||||||
}
|
}
|
||||||
|
|
||||||
func callback1[Base any, A any](b *Base, a A) {
|
func Callback1[Base any, A any](b *Base, a A) {
|
||||||
bind := (*bind1[Base, A])(unsafe.Pointer(b))
|
bind := (*bind1[Base, A])(unsafe.Pointer(b))
|
||||||
bind.fn(a)
|
bind.fn(a)
|
||||||
}
|
}
|
||||||
|
|
||||||
func callback2[Base any, A any, B any](b *Base, a A, c B) {
|
func Callback2[Base any, A any, B any](b *Base, a A, c B) {
|
||||||
bind := (*bind2[Base, A, B])(unsafe.Pointer(b))
|
bind := (*bind2[Base, A, B])(unsafe.Pointer(b))
|
||||||
bind.fn(a, c)
|
bind.fn(a, c)
|
||||||
}
|
}
|
||||||
|
|
||||||
func callback3[Base any, A any, B any, C any](b *Base, a A, c B, d C) {
|
func Callback3[Base any, A any, B any, C any](b *Base, a A, c B, d C) {
|
||||||
bind := (*bind3[Base, A, B, C])(unsafe.Pointer(b))
|
bind := (*bind3[Base, A, B, C])(unsafe.Pointer(b))
|
||||||
bind.fn(a, c, d)
|
bind.fn(a, c, d)
|
||||||
}
|
}
|
||||||
|
|
||||||
func Bind[T any](call func()) (p *T, fn func(*T)) {
|
/**
|
||||||
|
* Bind[N] binds a Go function to a C callback function.
|
||||||
|
*
|
||||||
|
* Example:
|
||||||
|
*
|
||||||
|
* timer, cb := cbind.Bind[libuv.Timer](func() {
|
||||||
|
* println("hello")
|
||||||
|
* })
|
||||||
|
* libuv.InitTimer(async.Exec().L, timer)
|
||||||
|
* timer.Start(cb, 1000, 0)
|
||||||
|
*
|
||||||
|
* TODO(lijie): fn isn't a C func-ptr, it's closure, should fix the LLGo compiler.
|
||||||
|
* See: https://github.com/goplus/llgo/issues/766
|
||||||
|
*
|
||||||
|
* Workaround:
|
||||||
|
*
|
||||||
|
* timer, _ := cbind.Bind[libuv.Timer](func() {
|
||||||
|
* println("hello")
|
||||||
|
* })
|
||||||
|
* libuv.InitTimer(async.Exec().L, timer)
|
||||||
|
* timer.Start(cbind.Callback[libuv.Timer], 1000, 0)
|
||||||
|
*
|
||||||
|
* @param call The Go function to bind.
|
||||||
|
* @return The data pointer and the C callback function.
|
||||||
|
*/
|
||||||
|
func Bind[T any](call func()) (p *T, fn Cb[T]) {
|
||||||
bb := &bind[T]{fn: func() { call() }}
|
bb := &bind[T]{fn: func() { call() }}
|
||||||
p = (*T)(unsafe.Pointer(bb))
|
p = (*T)(unsafe.Pointer(bb))
|
||||||
fn = callback[T]
|
fn = Callback[T]
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func Bind1[T any, A any](call func(A)) (p *T, fn func(*T, A)) {
|
func Bind1[T any, A any](call func(A)) (p *T, fn Cb1[T, A]) {
|
||||||
bb := &bind1[T, A]{fn: func(a A) { call(a) }}
|
bb := &bind1[T, A]{fn: func(a A) { call(a) }}
|
||||||
p = (*T)(unsafe.Pointer(bb))
|
p = (*T)(unsafe.Pointer(bb))
|
||||||
fn = callback1[T, A]
|
fn = Callback1[T, A]
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func Bind2[T any, A any, B any](call func(A, B)) (p *T, fn func(*T, A, B)) {
|
func Bind2[T any, A any, B any](call func(A, B)) (p *T, fn Cb2[T, A, B]) {
|
||||||
bb := &bind2[T, A, B]{fn: func(a A, b B) { call(a, b) }}
|
bb := &bind2[T, A, B]{fn: func(a A, b B) { call(a, b) }}
|
||||||
p = (*T)(unsafe.Pointer(bb))
|
p = (*T)(unsafe.Pointer(bb))
|
||||||
fn = callback2[T, A, B]
|
fn = Callback2[T, A, B]
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func Bind3[T any, A any, B any, C any](call func(A, B, C), a A, b B, c C) (p *T, fn func(*T, A, B, C)) {
|
func Bind3[T any, A any, B any, C any](call func(A, B, C), a A, b B, c C) (p *T, fn Cb3[T, A, B, C]) {
|
||||||
bb := &bind3[T, A, B, C]{fn: func(a A, b B, c C) { call(a, b, c) }}
|
bb := &bind3[T, A, B, C]{fn: func(a A, b B, c C) { call(a, b, c) }}
|
||||||
p = (*T)(unsafe.Pointer(bb))
|
p = (*T)(unsafe.Pointer(bb))
|
||||||
fn = callback3[T, A, B, C]
|
fn = Callback3[T, A, B, C]
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|||||||
51
x/io/io.go
51
x/io/io.go
@@ -1,3 +1,6 @@
|
|||||||
|
//go:build llgo
|
||||||
|
// +build llgo
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* Copyright (c) 2024 The GoPlus Authors (goplus.org). All rights reserved.
|
* Copyright (c) 2024 The GoPlus Authors (goplus.org). All rights reserved.
|
||||||
*
|
*
|
||||||
@@ -29,7 +32,7 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type Tcp struct {
|
type Tcp struct {
|
||||||
tcp *libuv.Tcp
|
tcp libuv.Tcp
|
||||||
}
|
}
|
||||||
|
|
||||||
type libuvError libuv.Errno
|
type libuvError libuv.Errno
|
||||||
@@ -40,8 +43,8 @@ func (e libuvError) Error() string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func NewTcp() *Tcp {
|
func NewTcp() *Tcp {
|
||||||
t := &Tcp{&libuv.Tcp{}}
|
t := &Tcp{}
|
||||||
libuv.InitTcp(async.Exec().L, t.tcp)
|
libuv.InitTcp(async.Exec().L, &t.tcp)
|
||||||
return t
|
return t
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -52,41 +55,40 @@ func (t *Tcp) Bind(addr *net.SockAddr, flags uint) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (t *Tcp) Listen(backlog int, cb libuv.ConnectionCb) error {
|
func (t *Tcp) Listen(backlog int, cb func(server *libuv.Stream, status c.Int)) error {
|
||||||
if res := (*libuv.Stream)(t.tcp).Listen(c.Int(backlog), cb); res != 0 {
|
if res := (*libuv.Stream)(&t.tcp).Listen(c.Int(backlog), cb); res != 0 {
|
||||||
return libuvError(res)
|
return libuvError(res)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (t *Tcp) Accept() (client *Tcp, err error) {
|
func (t *Tcp) Accept() (client *Tcp, err error) {
|
||||||
tcp := &libuv.Tcp{}
|
tcp := &Tcp{}
|
||||||
if res := libuv.InitTcp(async.Exec().L, tcp); res != 0 {
|
if res := libuv.InitTcp(async.Exec().L, &tcp.tcp); res != 0 {
|
||||||
return nil, libuvError(res)
|
return nil, libuvError(res)
|
||||||
}
|
}
|
||||||
if res := (*libuv.Stream)(t.tcp).Accept((*libuv.Stream)(client.tcp)); res != 0 {
|
if res := (*libuv.Stream)(&t.tcp).Accept((*libuv.Stream)(&tcp.tcp)); res != 0 {
|
||||||
return nil, libuvError(res)
|
return nil, libuvError(res)
|
||||||
}
|
}
|
||||||
return &Tcp{tcp}, nil
|
return tcp, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func Connect(addr *net.SockAddr) async.IO[tuple.Tuple2[*Tcp, error]] {
|
func Connect(addr *net.SockAddr) async.IO[tuple.Tuple2[*Tcp, error]] {
|
||||||
return async.Async(func(resolve func(tuple.Tuple2[*Tcp, error])) {
|
return async.Async(func(resolve func(tuple.Tuple2[*Tcp, error])) {
|
||||||
tcp := &libuv.Tcp{}
|
tcp := &Tcp{}
|
||||||
if res := libuv.InitTcp(async.Exec().L, tcp); res != 0 {
|
if res := libuv.InitTcp(async.Exec().L, &tcp.tcp); res != 0 {
|
||||||
resolve(tuple.T2[*Tcp, error]((*Tcp)(nil), libuvError(res)))
|
resolve(tuple.T2[*Tcp, error]((*Tcp)(nil), libuvError(res)))
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
req, cb := cbind.Bind1[libuv.Connect](func(status c.Int) {
|
req, _ := cbind.Bind1[libuv.Connect](func(status c.Int) {
|
||||||
if status != 0 {
|
if status != 0 {
|
||||||
resolve(tuple.T2[*Tcp, error]((*Tcp)(nil), libuvError(libuv.Errno(status))))
|
resolve(tuple.T2[*Tcp, error]((*Tcp)(nil), libuvError(libuv.Errno(status))))
|
||||||
} else {
|
|
||||||
resolve(tuple.T2[*Tcp, error](&Tcp{tcp}, nil))
|
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
if res := libuv.TcpConnect(req, tcp, addr, cb); res != 0 {
|
if res := libuv.TcpConnect(req, &tcp.tcp, addr, cbind.Callback1[libuv.Connect, c.Int]); res != 0 {
|
||||||
resolve(tuple.T2[*Tcp, error]((*Tcp)(nil), libuvError(res)))
|
resolve(tuple.T2[*Tcp, error]((*Tcp)(nil), libuvError(res)))
|
||||||
}
|
}
|
||||||
|
resolve(tuple.T2[*Tcp, error](tcp, nil))
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -95,23 +97,14 @@ func allocBuffer(handle *libuv.Handle, suggestedSize uintptr, buf *libuv.Buf) {
|
|||||||
buf.Len = suggestedSize
|
buf.Len = suggestedSize
|
||||||
}
|
}
|
||||||
|
|
||||||
type slice struct {
|
|
||||||
data unsafe.Pointer
|
|
||||||
len int
|
|
||||||
}
|
|
||||||
|
|
||||||
func goBytes(buf *int8, n int) []byte {
|
|
||||||
return *(*[]byte)(unsafe.Pointer(&slice{unsafe.Pointer(buf), n}))
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t *Tcp) Read() async.IO[tuple.Tuple2[[]byte, error]] {
|
func (t *Tcp) Read() async.IO[tuple.Tuple2[[]byte, error]] {
|
||||||
return func(ctx *async.AsyncContext) async.Future[tuple.Tuple2[[]byte, error]] {
|
return func(ctx *async.AsyncContext) async.Future[tuple.Tuple2[[]byte, error]] {
|
||||||
var result tuple.Tuple2[[]byte, error]
|
var result tuple.Tuple2[[]byte, error]
|
||||||
var done bool
|
var done bool
|
||||||
tcp := (*libuv.Stream)(t.tcp)
|
tcp := (*libuv.Stream)(&t.tcp)
|
||||||
libuv.ReadStart(tcp, allocBuffer, func(client *libuv.Stream, nread c.Long, buf *libuv.Buf) {
|
tcp.StartRead(allocBuffer, func(client *libuv.Stream, nread c.Long, buf *libuv.Buf) {
|
||||||
if nread > 0 {
|
if nread > 0 {
|
||||||
result = tuple.T2[[]byte, error](goBytes(buf.Base, int(nread)), nil)
|
result = tuple.T2[[]byte, error](cbind.GoBytes(buf.Base, int(nread)), nil)
|
||||||
} else if nread < 0 {
|
} else if nread < 0 {
|
||||||
result = tuple.T2[[]byte, error](nil, libuvError(libuv.Errno(nread)))
|
result = tuple.T2[[]byte, error](nil, libuvError(libuv.Errno(nread)))
|
||||||
} else {
|
} else {
|
||||||
@@ -129,6 +122,6 @@ func (t *Tcp) Read() async.IO[tuple.Tuple2[[]byte, error]] {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (t *Tcp) Close(cb libuv.CloseCb) {
|
func (t *Tcp) Close() {
|
||||||
(*libuv.Handle)(unsafe.Pointer(t.tcp)).Close(cb)
|
(*libuv.Handle)(unsafe.Pointer(&t.tcp)).Close(nil)
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user