|
|
|
|
@ -1,11 +1,14 @@
|
|
|
|
|
package com.yj.earth.common.service;
|
|
|
|
|
|
|
|
|
|
import cn.dev33.satoken.stp.StpUtil;
|
|
|
|
|
import cn.hutool.core.lang.UUID;
|
|
|
|
|
import cn.hutool.crypto.digest.BCrypt;
|
|
|
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
|
|
|
|
import com.yj.earth.business.domain.Role;
|
|
|
|
|
import com.yj.earth.business.domain.Source;
|
|
|
|
|
import com.yj.earth.business.domain.User;
|
|
|
|
|
import com.yj.earth.business.service.RoleService;
|
|
|
|
|
import com.yj.earth.business.service.RoleSourceService;
|
|
|
|
|
import com.yj.earth.business.service.SourceService;
|
|
|
|
|
import com.yj.earth.business.service.UserService;
|
|
|
|
|
import lombok.extern.slf4j.Slf4j;
|
|
|
|
|
@ -23,7 +26,8 @@ public class ServerInitService {
|
|
|
|
|
private UserService userService;
|
|
|
|
|
@Resource
|
|
|
|
|
private RoleService roleService;
|
|
|
|
|
|
|
|
|
|
@Resource
|
|
|
|
|
private RoleSourceService roleSourceService;
|
|
|
|
|
public void init() {
|
|
|
|
|
// 查询数据库所有需要加载的资源
|
|
|
|
|
List<Source> list =sourceService.list(new LambdaQueryWrapper<Source>()
|
|
|
|
|
@ -39,9 +43,14 @@ public class ServerInitService {
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void checkDefaultData() {
|
|
|
|
|
checkDefaultUserAndRole();
|
|
|
|
|
checkDefaultSource();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void checkDefaultUserAndRole() {
|
|
|
|
|
// 查询角色表和用户表是否有数据
|
|
|
|
|
if(roleService.count() == 0 && userService.count() == 0) {
|
|
|
|
|
log.info("初始化默认数据");
|
|
|
|
|
log.info("初始化默认用户角色数据");
|
|
|
|
|
// 新增一个管理员角色
|
|
|
|
|
Role adminRole = new Role();
|
|
|
|
|
adminRole.setRoleName("管理员");
|
|
|
|
|
@ -64,4 +73,98 @@ public class ServerInitService {
|
|
|
|
|
userService.save(user);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void checkDefaultSource() {
|
|
|
|
|
if(sourceService.count() == 0) {
|
|
|
|
|
// 查询管理员角色
|
|
|
|
|
Role adminRole = roleService.getOne(new LambdaQueryWrapper<Role>().eq(Role::getRoleName, "管理员"));
|
|
|
|
|
log.info("初始化默认资源数据");
|
|
|
|
|
Source source1 = new Source();
|
|
|
|
|
source1.setId(UUID.fastUUID().toString(true));
|
|
|
|
|
source1.setSourceName("倾斜模型");
|
|
|
|
|
source1.setSourceType("directory");
|
|
|
|
|
source1.setTreeIndex(0);
|
|
|
|
|
source1.setIsShow(1);
|
|
|
|
|
sourceService.save(source1);
|
|
|
|
|
roleSourceService.addRoleSource(adminRole.getId(), source1.getId());
|
|
|
|
|
|
|
|
|
|
Source source2 = new Source();
|
|
|
|
|
source2.setId(UUID.fastUUID().toString(true));
|
|
|
|
|
source2.setSourceName("人工模型");
|
|
|
|
|
source2.setSourceType("directory");
|
|
|
|
|
source2.setTreeIndex(0);
|
|
|
|
|
source2.setIsShow(1);
|
|
|
|
|
sourceService.save(source2);
|
|
|
|
|
roleSourceService.addRoleSource(adminRole.getId(), source2.getId());
|
|
|
|
|
|
|
|
|
|
Source source3 = new Source();
|
|
|
|
|
source3.setId(UUID.fastUUID().toString(true));
|
|
|
|
|
source3.setSourceName("卫星底图");
|
|
|
|
|
source3.setSourceType("directory");
|
|
|
|
|
source3.setTreeIndex(0);
|
|
|
|
|
source3.setIsShow(1);
|
|
|
|
|
sourceService.save(source3);
|
|
|
|
|
roleSourceService.addRoleSource(adminRole.getId(), source3.getId());
|
|
|
|
|
|
|
|
|
|
Source source4 = new Source();
|
|
|
|
|
source4.setId(UUID.fastUUID().toString(true));
|
|
|
|
|
source4.setSourceName("地形");
|
|
|
|
|
source4.setSourceType("directory");
|
|
|
|
|
source4.setTreeIndex(0);
|
|
|
|
|
source4.setIsShow(1);
|
|
|
|
|
sourceService.save(source4);
|
|
|
|
|
roleSourceService.addRoleSource(adminRole.getId(), source4.getId());
|
|
|
|
|
|
|
|
|
|
Source source5 = new Source();
|
|
|
|
|
source5.setId(UUID.fastUUID().toString(true));
|
|
|
|
|
source5.setSourceName("在线图源");
|
|
|
|
|
source5.setSourceType("directory");
|
|
|
|
|
source5.setTreeIndex(0);
|
|
|
|
|
source5.setIsShow(1);
|
|
|
|
|
sourceService.save(source5);
|
|
|
|
|
roleSourceService.addRoleSource(adminRole.getId(), source5.getId());
|
|
|
|
|
|
|
|
|
|
Source source6 = new Source();
|
|
|
|
|
source6.setId(UUID.fastUUID().toString(true));
|
|
|
|
|
source6.setSourceName("卫星图");
|
|
|
|
|
source6.setSourceType("arcgisWximagery");
|
|
|
|
|
source6.setParentId(source5.getId());
|
|
|
|
|
source6.setTreeIndex(0);
|
|
|
|
|
source6.setIsShow(1);
|
|
|
|
|
sourceService.save(source6);
|
|
|
|
|
roleSourceService.addRoleSource(adminRole.getId(), source6.getId());
|
|
|
|
|
|
|
|
|
|
Source source7 = new Source();
|
|
|
|
|
source7.setId(UUID.fastUUID().toString(true));
|
|
|
|
|
source7.setSourceName("暗黑地图");
|
|
|
|
|
source7.setSourceType("arcgisBlueImagery");
|
|
|
|
|
source7.setParentId(source5.getId());
|
|
|
|
|
source7.setTreeIndex(0);
|
|
|
|
|
source7.setIsShow(1);
|
|
|
|
|
sourceService.save(source7);
|
|
|
|
|
roleSourceService.addRoleSource(adminRole.getId(), source7.getId());
|
|
|
|
|
|
|
|
|
|
Source source8 = new Source();
|
|
|
|
|
source8.setId(UUID.fastUUID().toString(true));
|
|
|
|
|
source8.setSourceName("路网图");
|
|
|
|
|
source8.setSourceType("gdlwImagery");
|
|
|
|
|
source8.setParentId(source5.getId());
|
|
|
|
|
source8.setTreeIndex(0);
|
|
|
|
|
source8.setIsShow(1);
|
|
|
|
|
sourceService.save(source8);
|
|
|
|
|
roleSourceService.addRoleSource(adminRole.getId(), source8.getId());
|
|
|
|
|
|
|
|
|
|
Source source9 = new Source();
|
|
|
|
|
source9.setId(UUID.fastUUID().toString(true));
|
|
|
|
|
source9.setSourceName("矢量图");
|
|
|
|
|
source9.setSourceType("gdslImagery");
|
|
|
|
|
source9.setParentId(source5.getId());
|
|
|
|
|
source9.setTreeIndex(0);
|
|
|
|
|
source9.setIsShow(1);
|
|
|
|
|
sourceService.save(source9);
|
|
|
|
|
roleSourceService.addRoleSource(adminRole.getId(), source9.getId());
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|