基于Java的magic-api接口快速开发框架设计源码

package org.ssssssss.magicapi.core.service.impl; import org.apache.commons.lang3.StringUtils; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.boot.context.event.ApplicationStartedEvent; import org.springframework.context.ApplicationEventPublisher; import org.springframework.context.ApplicationListener; import org.ssssssss.magicapi.core.config.Constants; import org.ssssssss.magicapi.core.config.JsonCodeConstants; import org.ssssssss.magicapi.core.resource.Resource; import org.ssssssss.magicapi.core.resource.ZipResource; import org.ssssssss.magicapi.core.event.EventAction; import org.ssssssss.magicapi.core.event.FileEvent; import org.ssssssss.magicapi.core.event.GroupEvent; import org.ssssssss.magicapi.core.event.MagicEvent; import org.ssssssss.magicapi.core.exception.InvalidArgumentException; import org.ssssssss.magicapi.core.model.*; import org.ssssssss.magicapi.core.service.AbstractPathMagicResourceStorage; import org.ssssssss.magicapi.core.service.MagicResourceService; import org.ssssssss.magicapi.core.service.MagicResourceStorage; import org.ssssssss.magicapi.utils.IoUtils; import org.ssssssss.magicapi.utils.JsonUtils; import org.ssssssss.magicapi.utils.WebUtils; import java.io.IOException; import java.io.InputStream; import java.io.OutputStream; import java.util.*; import java.util.concurrent.locks.ReadWriteLock; import java.util.concurrent.locks.ReentrantReadWriteLock; import java.util.function.Supplier; import java.util.stream.Collectors; import java.util.zip.ZipEntry; import java.util.zip.ZipOutputStream; public class DefaultMagicResourceService implements MagicResourceService, JsonCodeConstants, ApplicationListener<ApplicationStartedEvent> { private final Resource root; private final Map<String, Resource> groupMappings = new HashMap<>(16); private final Map<String, Group> groupCache = new HashMap<>(16); private final Map<String, Resource> fileMappings = new HashMap<>(32); private final Map<String, MagicEntity> fileCache = new HashMap<>(32); private final Map<String, Map<String, String>> pathCache = new HashMap<>(16); private final Map<String, MagicResourceStorage<? extends MagicEntity>> storages; private final ReadWriteLock lock = new ReentrantReadWriteLock(); private final ApplicationEventPublisher publisher; private final Logger logger = LoggerFactory.getLogger(DefaultMagicResourceService.class); public DefaultMagicResourceService(Resource resource, List<MagicResourceStorage<? extends MagicEntity>> storages, ApplicationEventPublisher publisher) { this.root = resource; this.storages = storages.stream() .peek(it -> it.setMagicResourceService(this)) .collect(Collectors.toMap(MagicResourceStorage::folder, it -> it)); this.publisher = publisher; } public boolean processNotify(MagicNotify notify) { if (Constants.EVENT_TYPE_FILE.equals(notify.getType())) { return processFileNotify(notify.getId(), notify.getAction()); } if (notify.getAction() == EventAction.CLEAR) { this.read(false); return true; } return processGroupNotify(notify.getId(), notify.getAction()); } private boolean processGroupNotify(String id, EventAction action) { Group group = groupCache.get(id); if (group == null) { // create this.readAll(); group = groupCache.get(id); } TreeNode<Group> treeNode = tree(group.getType()).findTreeNode(it -> it.getId().equals(id)); if (treeNode != null) { GroupEvent event = new GroupEvent(group.getType(), action, group); event.setSource(Constants.EVENT_SOURCE_NOTIFY); if (event.getAction() == EventAction.DELETE) { event.setEntities(deleteGroup(id)); } else if (action != EventAction.CREATE) { Resource folder = groupMappings.get(id); folder.readAll(); if (folder.exists()) { // 刷新分组缓存 refreshGroup(folder, storages.get(group.getType())); } else { this.readAll(); treeNode = tree(group.getType()).findTreeNode(it -> it.getId().equals(id)); } event.setGroup(groupCache.get(id)); event.setEntities(treeNode .flat() .stream() .flatMap(g -> listFiles(g.getId()).stream()) .collect(Collectors.toList())); } publisher.publishEvent(event); return true; } return false; } private boolean processFileNotify(String id, EventAction action) { MagicEntity entity = fileCache.get(id); if (entity == null) { // create this.readAll(); entity = fileCache.get(id); } if (entity != null) { Group group = groupCache.get(entity.getGroupId()); if (group != null) { MagicResourceStorage<? extends MagicEntity> storage = storages.get(group.getType()); Map<String, String> pathCacheMap = storage.requirePath() ? pathCache.get(storage.folder()) : null; if (action == EventAction.DELETE) { fileMappings.remove(id); entity = fileCache.remove(id); if (pathCacheMap != null) { pathCacheMap.remove(id); } } else { Resource resource = fileMappings.get(id); resource.readAll(); if (resource.exists()) { entity = storage.read(resource.read()); putFile(storage, entity, resource); } else { this.readAll(); entity = fileCache.get(id); } } publisher.publishEvent(new FileEvent(group.getType(), action, entity, Constants.EVENT_SOURCE_NOTIFY)); } } return false; } private void init() { groupMappings.clear(); groupCache.clear(); fileMappings.clear(); fileCache.clear(); pathCache.clear(); storages.forEach((key, registry) -> { if (registry.requirePath()) { pathCache.put(registry.folder(), new HashMap<>(32)); } Resource folder = root.getDirectory(key); if (registry.allowRoot()) { String rootId = key + ":0"; Group group = new Group(); group.setId(rootId); group.setType(key); group.setParentId("0"); putGroup(group, folder); } if (!folder.exists()) { folder.mkdir(); } }); } private void read(boolean triggerEvent) { writeLock(() -> { if (triggerEvent) { publisher.publishEvent(new MagicEvent("clear", EventAction.CLEAR)); } this.readAll(); fileCache.values().forEach(entity -> { Group group = groupCache.get(entity.getGroupId()); publisher.publishEvent(new FileEvent(group.getType(), EventAction.LOAD, entity)); }); return null; }); } private void readAll() { writeLock(() -> { this.init(); this.root.readAll(); storages.forEach((key, registry) -> refreshGroup(root.getDirectory(key), registry)); return null; }); } @Override public void refresh() { this.read(true); } @Override public Resource getResource() { return root; } private void refreshGroup(Resource folder, MagicResourceStorage<? extends MagicEntity> storage) { if (storage.allowRoot()) { folder.files(storage.suffix()).forEach(file -> putFile(storage, storage.readResource(file), file)); } else { folder.dirs().forEach(dir -> { Resource meta = dir.getResource(Constants.GROUP_METABASE); if (meta.exists()) { putGroup(JsonUtils.readValue(meta.read(), Group.class), dir); dir.files(storage.suffix()).forEach(file -> putFile(storage, storage.readResource(file), file)); } }); } } @Override public boolean saveGroup(Group group) { isTrue(!root.readonly(), IS_READ_ONLY); // 类型校验 isTrue(storages.containsKey(group.getType()), NOT_SUPPORTED_GROUP_TYPE); // 名字校验 notNull(group.getName(), NAME_REQUIRED); notNull(IoUtils.validateFileName(group.getName()), NAME_INVALID); // 需要填写parentId notNull(group.getParentId(), GROUP_ID_REQUIRED); MagicResourceStorage<? extends MagicEntity> storage = storages.get(group.getType()); return writeLock(() -> { Resource resource; // 判断是否要保存到根节点下 if (Constants.ROOT_ID.equals(group.getParentId())) { resource = root.getDirectory(group.getType()); } else { // 找到上级分组 resource = getGroupResource(group.getParentId()); // 上级分组�