@Override
public RunResult run(GeneratorInstanceRunRequest request) {
RunResult runResult = new RunResult();
Authentication authentication = request.getAuthentication();
Long userId = authentication.getUserId();
Long id = request.getId();
GeneratorInstance generatorInstance = generatorInstanceRepository.selectById(id);
if (generatorInstance == null) {
throw new AppException("实例不存在");
}
if (!userId.equals(generatorInstance.getUser().getId())) {
throw new AppException("权限不足");
}
Long generatorId = generatorInstance.getGenerator().getId();
Generator generator = generatorRepository.selectById(generatorId);
if (generator == null) {
throw new AppException("生成器不存在");
}
if (generatorInstance.getVersion() < generator.getVersion()) {
throw new AppException("当前生成器已升级发布，请刷新数据，重新操作！");
}
if (!generator.getIsOpen() && !generator.getDeveloper().getId().equals(userId)) {
throw new AppException("当前生成器正在维护，请暂停操作，等待发布！");
}
Long templateStrategyId = request.getTemplateStrategyId();
TemplateStrategy templateStrategy = templateStrategyRepository.selectById(templateStrategyId);
DynamicModelQueryRequest dynamicModelQueryRequest = new DynamicModelQueryRequest();
dynamicModelQueryRequest.setGeneratorId(generatorId);
dynamicModelQueryRequest.setAuthentication(authentication);
List<DynamicModel> dynamicModels = dynamicModelService.query(dynamicModelQueryRequest);
Map<Long, DynamicModel> dynamicModelCache = new HashMap<>();
Map<Long, Map<String, Map<String, Set<String>>>> dynamicModelKeysCache = new HashMap<>();
dynamicModels.forEach(
dynamicModel -> {
DynamicModel newDynamicModel = new DynamicModel();
newDynamicModel.setId(dynamicModel.getId());
newDynamicModel.setName(dynamicModel.getName());
newDynamicModel.setIcon(dynamicModel.getIcon());
Map<String, Map<String, Set<String>>> keysCache = new HashMap<>();
Map<String, Set<String>> propertiesKeys = new HashMap<>();
Set<String> propertiesKeys_dateTypeKeys = new HashSet<>();
Set<String> propertiesKeys_dataModelTypeKeys = new HashSet<>();
propertiesKeys.put("dateTypeKeys", propertiesKeys_dateTypeKeys);
propertiesKeys.put("dataModelTypeKeys", propertiesKeys_dataModelTypeKeys);
Map<String, Set<String>> associationKeys = new HashMap<>();
Set<String> associationKeys_dateTypeKeys = new HashSet<>();
Set<String> associationKeys_dataModelTypeKeys = new HashSet<>();
associationKeys.put("dateTypeKeys", associationKeys_dateTypeKeys);
associationKeys.put("dataModelTypeKeys", associationKeys_dataModelTypeKeys);
keysCache.put("propertiesKeys", propertiesKeys);
keysCache.put("associationKeys", associationKeys);
dynamicModelKeysCache.put(dynamicModel.getId(), keysCache);
List<DynamicProperty> properties = dynamicModel.getProperties();
properties.forEach(
property -> {
if ("Date".equals(property.getType())) {
propertiesKeys_dateTypeKeys.add(property.getName());
} else if ("Model".equals(property.getType())) {
propertiesKeys_dataModelTypeKeys.add(property.getName());
}
DynamicProperty newDynamicProperty = new DynamicProperty();
newDynamicProperty.setId(property.getId());
newDynamicProperty.setLabel(property.getLabel());
newDynamicProperty.setName(property.getName());
newDynamicProperty.setViewWidth(property.getViewWidth());
newDynamicProperty.setType(property.getType());
newDynamicProperty.setDefaultValue(property.getDefaultValue());
newDynamicModel.getProperties().add(newDynamicProperty);
});
List<DynamicProperty> association = dynamicModel.getAssociation();
association.forEach(
property -> {
if ("Date".equals(property.getType())) {
associationKeys_dateTypeKeys.add(property.getName());
} else if ("Model".equals(property.getType())) {
associationKeys_dataModelTypeKeys.add(property.getName());
}
DynamicProperty newDynamicProperty = new DynamicProperty();
newDynamicProperty.setId(property.getId());
newDynamicProperty.setLabel(property.getLabel());
newDynamicProperty.setName(property.getName());
newDynamicProperty.setViewWidth(property.getViewWidth());
newDynamicProperty.setType(property.getType());
newDynamicProperty.setDefaultValue(property.getDefaultValue());
newDynamicModel.getAssociation().add(newDynamicProperty);
});
dynamicModelCache.put(newDynamicModel.getId(), newDynamicModel);
});
List<Long> excludeIds = request.getExcludeIds();
Map<Long, Long> excludeIdCache = new HashMap<>();
excludeIds.forEach(excludeId -> excludeIdCache.put(excludeId, excludeId));
Long dataModelId = generatorInstance.getDataModel().getId();
DataModel rootDataModel = dataModelRepository.selectById(dataModelId);
Map<Long, DataModel> dataModelSourceCache = new HashMap<>();
Map<Long, DataModel> dataModelTargetCache = new HashMap<>();
LinkedList<DataModel> stack = new LinkedList<>();
stack.push(rootDataModel);
while (!stack.isEmpty()) {
DataModel dataModel = stack.pop();
dataModelSourceCache.put(dataModel.getId(), dataModel);
dataModelTargetCache.put(dataModel.getId(), new DataModel());
dataModel.getChildren().forEach(stack::push);
}
dataModelSourceCache.forEach(
(dataModelSourceId, dataModelSource) -> {
DataModel dataModelTarget = dataModelTargetCache.get(dataModelSourceId);
dataModelTarget.setId(dataModelSource.getId());
dataModelTarget.setName(dataModelSource.getName());
if (dataModelSource.getParent() != null) {
Long parentId = dataModelSource.getParent().getId();
DataModel parentSource = dataModelSourceCache.get(parentId);
if (parentSource != null) {
DataModel parentTarget = dataModelTargetCache.get(parentId);
dataModelTarget.setParent(parentTarget);
}
}
dataModelSource
.getChildren()
.forEach(
child -> {
Long childId = child.getId();
if (!excludeIdCache.containsKey(childId)) {
dataModelTarget.getChildren().add(dataModelTargetCache.get(childId));
}
});
if (!dataModelSource.equals(rootDataModel)) {
DynamicModel dynamicModel =
dynamicModelCache.get(dataModelSource.getDynamicModel().getId());
dataModelTarget.setDynamicModel(dynamicModel);
Map<String, Map<String, Set<String>>> keysCache =
dynamicModelKeysCache.get(dynamicModel.getId());
Map<String, Set<String>> propertiesKeys = keysCache.get("propertiesKeys");
Map<String, Set<String>> associationKeys = keysCache.get("associationKeys");
Set<String> propertiesKeys_dateTypeKeys = propertiesKeys.get("dateTypeKeys");
Set<String> propertiesKeys_dataModelTypeKeys = propertiesKeys.get("dataModelTypeKeys");
Set<String> associationKeys_dateTypeKeys = associationKeys.get("dateTypeKeys");
Set<String> associationKeys_dataModelTypeKeys =
associationKeys.get("dataModelTypeKeys");
dataModelSource
.getProperties()
.forEach(
(name, value) -> {
try {
if (propertiesKeys_dateTypeKeys.contains(name)) {
dataModelTarget.getProperties().put(name, new Date((Long) value));
} else if (propertiesKeys_dataModelTypeKeys.contains(name)) {
dataModelTarget
.getProperties()
.put(name, dataModelTargetCache.get(value));
} else {
dataModelTarget.getProperties().put(name, value);
}
} catch (Exception e) {
//
}
});
dataModelSource
.getAssociation()
.forEach(
property -> {
Map<String, Object> newProperty = new LinkedHashMap<>();
property.forEach(
(name, value) -> {
try {
if (associationKeys_dateTypeKeys.contains(name)) {
newProperty.put(name, new Date((Long) value));
} else if (associationKeys_dataModelTypeKeys.contains(name)) {
newProperty.put(name, dataModelTargetCache.get(value));
} else {
newProperty.put(name, value);
}
} catch (Exception e) {
//
}
});
dataModelTarget.getAssociation().add(newProperty);
});
}
});
Template templateTemplate = new Template();
templateTemplate.setIsDelete(false);
templateTemplate.setGenerator(generator);
List<Template> templates = templateRepository.selectList(templateTemplate);
Map<Long, Template> templateCache = new HashMap<>();
templates.forEach(
template -> {
Template newTemplate = new Template();
newTemplate.setId(template.getId());
newTemplate.setName(template.getName());
newTemplate.setUrl(template.getUrl());
templateCache.put(template.getId(), newTemplate);
});
Long generateId = idWorker.nextId();
User developer = userRepository.selectById(generator.getDeveloper().getId());
TemplateStrategy templateStrategyClone = templateStrategy.clone();
Global global = new Global();
global.setGenerateId(generateId);
global.setTemplateCache(templateCache);
User userClone = new User();
userClone.setId(userId);
userClone.setUserName(authentication.getUserName());
global.setUser(userClone);
Generator generatorClone = new Generator();
generatorClone.setId(generator.getId());
generatorClone.setName(generator.getName());
global.setGenerator(generatorClone);
GeneratorInstance generatorInstanceClone = new GeneratorInstance();
generatorInstanceClone.setId(generatorInstance.getId());
generatorInstanceClone.setName(generatorInstance.getName());
global.setGeneratorInstance(generatorInstanceClone);
User developerClone = new User();
developerClone.setId(developer.getId());
developerClone.setUserName(developer.getUserName());
global.setDeveloper(developerClone);
global.setTemplateStrategy(templateStrategyClone);
Context context = new Context();
context.setVariable("global", global);
context.setVariable("data", dataModelTargetCache.get(rootDataModel.getId()));
try {
templateStrategyClone.execute(context);
} catch (Throwable e) {
List<String> messages = new ArrayList<>();
LinkedList<Throwable> exceptionStack = new LinkedList<>();
exceptionStack.push(e);
while (!exceptionStack.isEmpty()) {
Throwable exception = exceptionStack.pop();
messages.add(exception.toString());
if (exception.getCause() != null) {
exceptionStack.push(exception.getCause());
}
}
runResult.setMessages(messages);
return runResult;
}
String generatePath =
ConfigProperties.TEMPORARY_PATH
+ ConfigProperties.fileSeparator
+ userId
+ ConfigProperties.fileSeparator
+ generatorInstance.getName()
+ "("
+ generateId
+ ")";
FileUtil.mkdirs(generatePath);
File generateFolder = new File(generatePath);
try {
ZipUtil.compress(generateFolder);
} catch (Exception e) {
throw new AppException(e, "压缩文件失败");
}
FileUtil.deleteFile(generateFolder);
runResult.setUrl(userId + "/" + generatorInstance.getName() + "(" + generateId + ").zip");
runResult.setFileName(generatorInstance.getName() + "(" + generateId + ").zip");
return runResult;
}
@Override
public GeneratorInstance get(GeneratorInstanceGetRequest request) {
Long id = request.getId();
GeneratorInstance generatorInstancePersistence = generatorInstanceRepository.selectById(id);
if (generatorInstancePersistence == null) {
throw new AppException("实例不存在");
}
GeneratorInstance generatorInstance = new GeneratorInstance();
generatorInstance.setId(generatorInstancePersistence.getId());
generatorInstance.setName(generatorInstancePersistence.getName());
generatorInstance.setCreateDate(generatorInstancePersistence.getCreateDate());
generatorInstance.setModifyDate(generatorInstancePersistence.getModifyDate());
generatorInstance.setIsDelete(generatorInstancePersistence.getIsDelete());
User userPersistence =
userRepository.selectById(generatorInstancePersistence.getUser().getId());
generatorInstance.setUser(userPersistence);
Long generatorId = generatorInstancePersistence.getGenerator().getId();
Generator generatorPersistence = generatorRepository.selectById(generatorId);
if (generatorPersistence == null) {
throw new AppException("生成器不存在");
}
generatorInstance.setGenerator(generatorPersistence);
generatorInstance.setDataModel(generatorInstancePersistence.getDataModel());
generatorInstance.setVersion(generatorInstancePersistence.getVersion());
return generatorInstance;
}
@Override
public PagerResult<GeneratorInstance> search(GeneratorInstanceSearchRequest request) {
Long userId = request.getUserId();
String name = StringUtils.hasText(request.getName()) ? request.getName() : null;
List<GeneratorInstance> records =
generatorInstanceRepository.filter(
generatorInstance -> {
if (generatorInstance.getIsDelete()) {
return false;
}
if (name != null) {
if (!generatorInstance.getName().contains(name)) {
return false;
}
}
if (userId != null) {
if (!userId.equals(generatorInstance.getUser().getId())) {
return false;
}
}
return true;
});
Integer page = request.getPage();
Integer pageSize = request.getPageSize();
Integer fromIndex = (page - 1) * pageSize;
Integer toIndex = fromIndex + pageSize > records.size() ? records.size() : fromIndex + pageSize;
List<GeneratorInstance> limitRecords = records.subList(fromIndex, toIndex);
List<GeneratorInstance> result = new ArrayList<>();
for (GeneratorInstance g : limitRecords) {
GeneratorInstance generatorInstance = new GeneratorInstance();
generatorInstance.setId(g.getId());
generatorInstance.setName(g.getName());
generatorInstance.setCreateDate(g.getCreateDate());
generatorInstance.setModifyDate(g.getModifyDate());
generatorInstance.setIsDelete(g.getIsDelete());
User userPersistence = userRepository.selectById(g.getUser().getId());
generatorInstance.setUser(userPersistence);
Generator generatorPersistence = generatorRepository.selectById(g.getGenerator().getId());
if (generatorPersistence == null) {
throw new AppException("生成器不存在");
}
generatorInstance.setGenerator(generatorPersistence);
generatorInstance.setDataModel(g.getDataModel());
generatorInstance.setVersion(g.getVersion());
result.add(generatorInstance);
}
String sortField = request.getSortField();
String sortDirection = request.getSortDirection();
if ("modifyDate".equals(sortField)) {
if ("DESC".equalsIgnoreCase(sortDirection)) {
result.sort(
(g1, g2) -> (int) (g2.getModifyDate().getTime() - g1.getModifyDate().getTime()));
}
}
return new PagerResult<>(result, (long) records.size());
}
@Override
public User updateUser(
long userId, String username, String password, String firstName, String lastName) {
User user = userRepository.findOne(userId);
user.setUsername(username);
user.setFirstName(firstName);
user.setLastName(lastName);
user.setPassword(password);
return this.userRepository.save(user);
}
@Override
public Customer removeCustomer(long userId, long customerId) {
User user = userRepository.findOne(userId);
Customer customer = customerRepository.findOne(customerId);
user.getCustomers().remove(customer);
this.userRepository.save(user);
customer.setUser(null);
this.customerRepository.delete(customer);
return customer;
}
@Override
public void setUp() throws Exception {
super.setUp();
userRepository = new UserRepository(em);
user = new User(TEST_USER_ID, TEST_PASSWORD);
userRepository.create(user);
em.flush();
repository = new AlbumRepository(em);
}
@Override
public void writeUserProfilePhoto(long userId, MediaType ext, byte[] bytesForProfilePhoto) {
User user = findById(userId);
user.setProfilePhotoMediaType(ext.toString());
user.setProfilePhotoImported(true);
userRepository.save(user);
ByteArrayInputStream byteArrayInputStream = null;
FileOutputStream fileOutputStream = null;
try {
fileOutputStream = new FileOutputStream(fileForPhoto(userId));
byteArrayInputStream = new ByteArrayInputStream(bytesForProfilePhoto);
IOUtils.copy(byteArrayInputStream, fileOutputStream);
} catch (IOException e) {
throw new UserProfilePhotoWriteException(userId, e);
} finally {
IOUtils.closeQuietly(fileOutputStream);
IOUtils.closeQuietly(byteArrayInputStream);
}
}
@Override
public User removeUser(long userId) {
User u = userRepository.findOne(userId);
this.userRepository.delete(userId);
return u;
}
@Override
public User findById(long userId) {
User user = userRepository.findOne(userId);
if (null == user) throw new UserNotFoundException(userId);
return user;
}
@Override
public User findUserByUsername(String username) {
return userRepository.findByUsername(username);
}
/** Stores to persistence layer initial data. */
@PostConstruct
public void init() {
if (roleRepository.findByName("ROLE_ADMIN") == null) {
Role roleUser = new Role();
roleUser.setName("ROLE_USER");
roleRepository.save(roleUser);
Role roleAdmin = new Role();
roleAdmin.setName("ROLE_ADMIN");
roleRepository.save(roleAdmin);
User user = new User();
user.setEnabled(true);
user.setEmail("admin@admin");
BCryptPasswordEncoder encoder = new BCryptPasswordEncoder();
user.setPassword(encoder.encode("admin"));
List<Role> roles = new ArrayList<>();
roles.add(roleAdmin);
roles.add(roleUser);
user.setRoles(roles);
userRepository.save(user);
// Create account for wallet
Account walletAccount = new Account();
walletAccount.setName(context.getMessage("Name.default.account", null, Locale.ENGLISH));
walletAccount.setUser(user);
walletAccount.setAmount(new BigDecimal(0));
walletAccount.setCurrency(Currency.getInstance("UAH"));
accountRepository.save(walletAccount);
Account bankAccount = new Account();
bankAccount.setName("Bank");
bankAccount.setUser(user);
bankAccount.setAmount(new BigDecimal(500));
bankAccount.setCurrency(Currency.getInstance("UAH"));
accountRepository.save(bankAccount);
// Create categories for expenses
for (int i = 1; i < 6; i++) {
Category category = new Category();
category.setName(
context.getMessage("Name" + i + ".default.category", null, Locale.ENGLISH));
category.setType(Operation.EXPENSE);
category.setUser(user);
categoryRepository.save(category);
}
// Create categories for incomes
for (int i = 6; i < 8; i++) {
Category category = new Category();
category.setName(
context.getMessage("Name" + i + ".default.category", null, Locale.ENGLISH));
category.setType(Operation.INCOME);
category.setUser(user);
categoryRepository.save(category);
}
Transaction transaction1 = new Transaction();
transaction1.setDate(new Date());
transaction1.setAccount(walletAccount);
transaction1.setAmount(new BigDecimal(50));
transaction1.setCurrency(Currency.getInstance("UAH"));
transaction1.setCategory(categoryRepository.findOne(3));
transaction1.setType(Operation.EXPENSE);
transaction1.setComment("McDonalds");
transaction1.setUser(user);
transactionRepository.save(transaction1);
Transaction transaction2 = new Transaction();
Calendar calendar = new GregorianCalendar();
calendar.set(Calendar.MONTH, calendar.get(Calendar.MONTH) - 1);
transaction2.setDate(calendar.getTime());
transaction2.setAccount(bankAccount);
transaction2.setAmount(new BigDecimal(45));
transaction2.setCurrency(Currency.getInstance("UAH"));
transaction2.setCategory(categoryRepository.findOne(7));
transaction2.setType(Operation.INCOME);
transaction2.setComment("Festo");
transaction2.setUser(user);
transactionRepository.save(transaction2);
List<Transaction> transactions = new ArrayList<>();
transactions.add(transaction1);
user.setTransactions(transactions);
}
}