diff --git a/README.md b/README.md index e2e3ef4..0cb54ff 100644 --- a/README.md +++ b/README.md @@ -43,8 +43,8 @@ DeleteStmtFromFuncBody(df *dst.File, funcName string, stmt dst.Stmt) (modified b AddStmtToFuncBody(df *dst.File, funcName string, stmt dst.Stmt, pos int) (modified bool) AddStmtToFuncBodyStart(df *dst.File, funcName string, stmt dst.Stmt) (modified bool) AddStmtToFuncBodyEnd(df *dst.File, funcName string, stmt dst.Stmt) (modified bool) -AddNodeToFuncBodyBefore(df *dst.File, funcName string, stmt, refStmt dst.Stmt) (modified bool) -AddNodeToFuncBodyAfter(df *dst.File, funcName string, stmt, refStmt dst.Stmt) (modified bool) +AddStmtToFuncBodyBefore(df *dst.File, funcName string, stmt, refStmt dst.Stmt) (modified bool) +AddStmtToFuncBodyAfter(df *dst.File, funcName string, stmt, refStmt dst.Stmt) (modified bool) ``` ### function call utilities diff --git a/func_body.go b/func_body.go index 737b847..5c9adb2 100644 --- a/func_body.go +++ b/func_body.go @@ -134,14 +134,14 @@ func addStmtToFuncBodyRelativeTo(df *dst.File, funcName string, stmt, refStmt ds return } -// AddNodeToFuncBodyBefore adds given statement, to the function body, before the position of refStmt. +// AddStmtToFuncBodyBefore adds given statement, to the function body, before the position of refStmt. // if refStmt not found, nothing will happen -func AddNodeToFuncBodyBefore(df *dst.File, funcName string, stmt, refStmt dst.Stmt) (modified bool) { +func AddStmtToFuncBodyBefore(df *dst.File, funcName string, stmt, refStmt dst.Stmt) (modified bool) { return addStmtToFuncBodyRelativeTo(df, funcName, stmt, refStmt, relativeDirectionBefore) } -// AddNodeToFuncBodyAfter adds given statement, to the function body, after the position of refStmt, +// AddStmtToFuncBodyAfter adds given statement, to the function body, after the position of refStmt, // if refStmt not found, nothing will happen -func AddNodeToFuncBodyAfter(df *dst.File, funcName string, stmt, refStmt dst.Stmt) (modified bool) { +func AddStmtToFuncBodyAfter(df *dst.File, funcName string, stmt, refStmt dst.Stmt) (modified bool) { return addStmtToFuncBodyRelativeTo(df, funcName, stmt, refStmt, relativeDirectionAfter) }