Skip to content

Commit

Permalink
feat(scaffold): provide mybatis-plus meta object handler configuration
Browse files Browse the repository at this point in the history
  • Loading branch information
taccisum committed Jul 9, 2019
1 parent d44e61c commit 220e994
Show file tree
Hide file tree
Showing 3 changed files with 63 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -4,4 +4,24 @@

@Configuration
public class ApplicationConfiguration {
<%
if(['mybatis-plus']){
print(`
@Bean
public ApplicationMetaObjectHandler.RuntimeData runtimeData() {
return new ApplicationMetaObjectHandler.RuntimeData() {
@Override
public String getUserId() {
return "1";
}

@Override
public String getTenantId() {
return "1";
}
};
}
`)
}
%>
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,36 @@
package ${basePackage}.config;

import ${basePackage}.util.AuthUtils;
import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.reflection.MetaObject;
import org.springframework.stereotype.Component;

import java.util.Date;

@Component
@Slf4j
public class ApplicationMetaObjectHandler implements MetaObjectHandler {
@Autowired
private RuntimeData runtimeData;

@Override
public void insertFill(MetaObject metaObject) {
this.setInsertFieldValByName("dr", 0, metaObject);
this.setInsertFieldValByName("createdAt", new Date(), metaObject);
this.setInsertFieldValByName("createdBy", runtimeData.getUserId(), metaObject);
this.setInsertFieldValByName("tenantId", runtimeData.getTenantId(), metaObject);
}

@Override
public void updateFill(MetaObject metaObject) {
this.setUpdateFieldValByName("updatedAt", new Date(), metaObject);
this.setUpdateFieldValByName("updatedBy", runtimeData.getUserId(), metaObject);
}

public interface RuntimeData {
String getUserId();

String getTenantId();
}
}
7 changes: 7 additions & 0 deletions generators/test/app/index.test.js
Original file line number Diff line number Diff line change
Expand Up @@ -295,6 +295,13 @@ describe('optional dependencies', () => {

it('should exist files', () => {
assert.file('foo-service-provider/src/main/resources/mapper/.gitkeep')
assert.file('foo-service-provider/src/main/java/com/deepexi/foo/config/ApplicationMetaObjectHandler.java')
});

it('should exist contents', () => {
assert.fileContent([
['foo-service-provider/src/main/java/com/deepexi/foo/config/ApplicationConfiguration.java', /ApplicationMetaObjectHandler.RuntimeData/]
])
});

it('should exist demo files', () => {
Expand Down

0 comments on commit 220e994

Please sign in to comment.