bloeys/cogo
1
0
Fork 0
mirror of https://github.com/bloeys/cogo.git synced 2025-12-29 08:58:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

More stuff

Browse Source
This commit is contained in:
bloeys
2022-10-27 03:41:40 +04:00
parent ea05c65c93
commit be2a6f72ba
3 changed files with 21 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
cogo/cogo.go
View File

@ -23,6 +23,3 @@ func (c *Coroutine[InT, OutT]) Tick() (out OutT, done bool) {
func (c *Coroutine[InT, OutT]) Yield(out OutT) { func (c *Coroutine[InT, OutT]) Yield(out OutT) {
} }
func (c *Coroutine[InT, OutT]) End() {
}

50
inliner/main.go
View File

@ -165,40 +165,32 @@ func (p *processor) processDeclNode(c *astutil.Cursor) bool {
lastCaseEndBodyListIndex = i lastCaseEndBodyListIndex = i
} else if cogoFuncSelExpr.Sel.Name == "End" {
// Add everything from the last begin/yield until this yield into a case
stmtsSinceLastCogo := funcDecl.Body.List[lastCaseEndBodyListIndex+1 : i]
stmtsAfterEnd := funcDecl.Body.List[i+1:]
caseStmts := make([]ast.Stmt, 0, len(stmtsSinceLastCogo)+len(stmtsAfterEnd)+1)
caseStmts = append(caseStmts, stmtsSinceLastCogo...)
caseStmts = append(caseStmts,
&ast.AssignStmt{
Lhs: []ast.Expr{ast.NewIdent(coroutineParamName + ".State")},
Tok: token.ASSIGN,
Rhs: []ast.Expr{ast.NewIdent("-1")},
},
)
caseStmts = append(caseStmts, stmtsAfterEnd...)
switchStmt.Body.List = append(switchStmt.Body.List,
getCaseWithStmts(
[]ast.Expr{ast.NewIdent(fmt.Sprint(len(switchStmt.Body.List)))},
caseStmts,
),
)
lastCaseEndBodyListIndex = i
} }
} }
// Add everything after the last yield in a separate case
stmtsToEndOfFunc := funcDecl.Body.List[lastCaseEndBodyListIndex+1:]
caseStmts := make([]ast.Stmt, 0, len(stmtsToEndOfFunc)+1)
caseStmts = append(caseStmts,
&ast.AssignStmt{
Lhs: []ast.Expr{ast.NewIdent(coroutineParamName + ".State")},
Tok: token.ASSIGN,
Rhs: []ast.Expr{ast.NewIdent("-1")},
},
)
caseStmts = append(caseStmts, stmtsToEndOfFunc...)
switchStmt.Body.List = append(switchStmt.Body.List,
getCaseWithStmts(
[]ast.Expr{ast.NewIdent(fmt.Sprint(len(switchStmt.Body.List)))},
caseStmts,
),
)
// Apply changes
funcDecl.Body.List = funcDecl.Body.List[:beginBodyListIndex] funcDecl.Body.List = funcDecl.Body.List[:beginBodyListIndex]
funcDecl.Body.List = append(funcDecl.Body.List, funcDecl.Body.List = append(funcDecl.Body.List,
// &ast.LabeledStmt{
// Label: ast.NewIdent(cogoSwitchLbl),
// Stmt: switchStmt,
// },
switchStmt, switchStmt,
) )
return true return true

22
main.go
View File

@ -30,28 +30,6 @@ func test(c *cogo.Coroutine[int, int]) (out int) {
c.Yield(4) c.Yield(4)
println("Tick before end") println("Tick before end")
c.End()
// switch c.State {
// case 0:
// println("Tick 0")
// c.State++
// return 1, false
// case 1:
// println("Tick 1")
// c.State++
// return 2, false
// case 2:
// println("Tick 2")
// c.State++
// return 3, false
// case 3:
// println("Tick 3")
// c.State++
// return 4, false
// default:
// return out, true
// }
return out return out
} }