func (b *ArgsMethodBuilder) Build() ast.Decl {
mutexLockBuilder := NewMutexActionBuilder()
mutexLockBuilder.SetMutexFieldSelector(b.mutexFieldSelector)
mutexLockBuilder.SetAction("RLock")
mutexUnlockBuilder := NewMutexActionBuilder()
mutexUnlockBuilder.SetMutexFieldSelector(b.mutexFieldSelector)
mutexUnlockBuilder.SetAction("RUnlock")
mutexUnlockBuilder.SetDeferred(true)
b.methodBuilder.SetType(&ast.FuncType{
Params: &ast.FieldList{
List: []*ast.Field{
util.CreateField("index", ast.NewIdent("int")),
},
},
Results: &ast.FieldList{
List: util.FieldsWithoutEllipsis(util.FieldsAsAnonymous(b.params)),
},
})
b.methodBuilder.AddStatementBuilder(mutexLockBuilder)
b.methodBuilder.AddStatementBuilder(mutexUnlockBuilder)
results := []ast.Expr{}
for _, param := range b.params {
results = append(results, &ast.SelectorExpr{
X: &ast.IndexExpr{
X: b.argsFieldSelector,
Index: ast.NewIdent("index"),
},
Sel: ast.NewIdent(param.Names[0].String()),
})
}
b.methodBuilder.AddStatementBuilder(StatementToBuilder(&ast.ReturnStmt{
Results: results,
}))
return b.methodBuilder.Build()
}
func (b *StubMethodBuilder) Build() ast.Decl {
mutexLockBuilder := NewMutexActionBuilder()
mutexLockBuilder.SetMutexFieldSelector(b.mutexFieldSelector)
mutexLockBuilder.SetAction("Lock")
mutexUnlockBuilder := NewMutexActionBuilder()
mutexUnlockBuilder.SetMutexFieldSelector(b.mutexFieldSelector)
mutexUnlockBuilder.SetAction("Unlock")
mutexUnlockBuilder.SetDeferred(true)
b.methodBuilder.SetType(&ast.FuncType{
Params: &ast.FieldList{
List: b.params,
},
Results: &ast.FieldList{
List: util.FieldsAsAnonymous(b.results),
},
})
b.methodBuilder.AddStatementBuilder(mutexLockBuilder)
b.methodBuilder.AddStatementBuilder(mutexUnlockBuilder)
paramSelectors := []ast.Expr{}
for _, param := range b.params {
paramSelectors = append(paramSelectors, ast.NewIdent(param.Names[0].String()))
}
b.methodBuilder.AddStatementBuilder(StatementToBuilder(&ast.AssignStmt{
Lhs: []ast.Expr{
b.argsFieldSelector,
},
Tok: token.ASSIGN,
Rhs: []ast.Expr{
&ast.CallExpr{
Fun: ast.NewIdent("append"),
Args: []ast.Expr{
b.argsFieldSelector,
&ast.CompositeLit{
Type: &ast.StructType{
Fields: &ast.FieldList{
List: util.FieldsWithoutEllipsis(b.params),
},
},
Elts: paramSelectors,
},
},
},
},
}))
hasEllipsis := false
if parCount := len(b.params); parCount > 0 {
if _, ok := b.params[parCount-1].Type.(*ast.Ellipsis); ok {
hasEllipsis = true
}
}
b.methodBuilder.AddStatementBuilder(StatementToBuilder(&ast.IfStmt{
Cond: &ast.BinaryExpr{
X: b.stubFieldSelector,
Op: token.NEQ,
Y: ast.NewIdent("nil"),
},
Body: b.buildCallStubMethodCode(paramSelectors, hasEllipsis),
Else: b.buildReturnReturnsCode(),
}))
return b.methodBuilder.Build()
}