OptimisticLockerPlugin nextVersion方法增加返回

Author: hewei

src/main/java/com/itfsw/mybatis/generator/plugins/OptimisticLockerPlugin.java

@@ -239,13 +239,14 @@ public boolean modelSetterMethodGenerated(Method method, TopLevelClass topLevelC
             Method nextVersion = JavaElementGeneratorTools.generateMethod(
                     METHOD_NEXT_VERSION,
                     JavaVisibility.PUBLIC,
-                    null,
+                    topLevelClass.getType(),
                     new Parameter(this.versionColumn.getFullyQualifiedJavaType(), "version")
             );
             commentGenerator.addSetterComment(nextVersion, introspectedTable, this.versionColumn);
             JavaElementGeneratorTools.generateMethodBody(
                     nextVersion,
-                    "this." + this.versionColumn.getJavaProperty() + " = version;"
+                    "this." + this.versionColumn.getJavaProperty() + " = version;",
+                    "return this;"
             );
 
             FormatTools.addMethodWithBestPosition(topLevelClass, nextVersion);
@@ -273,6 +274,7 @@ public boolean modelBuilderSetterMethodGenerated(Method method, TopLevelClass to
                     builderClass.getType(),
                     new Parameter(this.versionColumn.getFullyQualifiedJavaType(), "version")
             );
+            nextVersion.addAnnotation("@Deprecated");
             commentGenerator.addSetterComment(nextVersion, introspectedTable, this.versionColumn);
 
             Method setterMethod = JavaBeansUtil.getJavaBeansSetter(this.versionColumn, context, introspectedTable);