Browse Source

修改顺序

dev-DXTZYK
fanshuai 1 year ago
parent
commit
aba0396e06
3 changed files with 17 additions and 28 deletions
  1. +4
    -3
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/GitServiceImpl.java
  2. +3
    -1
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java
  3. +10
    -24
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/NewDatasetServiceImpl.java

+ 4
- 3
ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/GitServiceImpl.java View File

@@ -28,7 +28,8 @@ public class GitServiceImpl implements GitService {


@Value("${spring.redis.host}") @Value("${spring.redis.host}")
private String redisHost; private String redisHost;

@Value("${spring.redis.port}")
private Integer redisPort;
private static final Logger log = LoggerFactory.getLogger(GitServiceImpl.class); private static final Logger log = LoggerFactory.getLogger(GitServiceImpl.class);


@Override @Override
@@ -61,7 +62,7 @@ public class GitServiceImpl implements GitService {
LoginUser loginUser = SecurityUtils.getLoginUser(); LoginUser loginUser = SecurityUtils.getLoginUser();
String ci4sUsername = loginUser.getUsername(); String ci4sUsername = loginUser.getUsername();
// 将access_token存入Redis // 将access_token存入Redis
Jedis jedis = new Jedis(redisHost);
Jedis jedis = new Jedis(redisHost,redisPort);
jedis.set(ci4sUsername + "_gitToken", accessToken); jedis.set(ci4sUsername + "_gitToken", accessToken);
jedis.set(ci4sUsername + "_gitUserInfo", userReq); jedis.set(ci4sUsername + "_gitUserInfo", userReq);
return accessToken; return accessToken;
@@ -72,7 +73,7 @@ public class GitServiceImpl implements GitService {
} }


public String checkoutToken() { public String checkoutToken() {
Jedis jedis = new Jedis(redisHost);
Jedis jedis = new Jedis(redisHost,redisPort);
LoginUser loginUser = SecurityUtils.getLoginUser(); LoginUser loginUser = SecurityUtils.getLoginUser();
String ci4sUsername = loginUser.getUsername(); String ci4sUsername = loginUser.getUsername();
String token = jedis.get(ci4sUsername + "_gitToken"); String token = jedis.get(ci4sUsername + "_gitToken");


+ 3
- 1
ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java View File

@@ -86,6 +86,8 @@ public class ModelsServiceImpl implements ModelsService {


@Value("${spring.redis.host}") @Value("${spring.redis.host}")
private String redisHost; private String redisHost;
@Value("${spring.redis.port}")
private Integer redisPort;
@Value("${git.endpoint}") @Value("${git.endpoint}")
String gitendpoint; String gitendpoint;
@Value("${git.localPath}") @Value("${git.localPath}")
@@ -1048,7 +1050,7 @@ public class ModelsServiceImpl implements ModelsService {
} }


Map<String, Object> getUserInfo(String ci4sUsername, String gitLinkUsername, String gitLinkPassword) throws IOException { Map<String, Object> getUserInfo(String ci4sUsername, String gitLinkUsername, String gitLinkPassword) throws IOException {
Jedis jedis = new Jedis(redisHost);
Jedis jedis = new Jedis(redisHost,redisPort);
String userReq = jedis.get(ci4sUsername + "_gitUserInfo"); String userReq = jedis.get(ci4sUsername + "_gitUserInfo");
if (userReq == null) { if (userReq == null) {
gitService.login(gitLinkUsername, gitLinkPassword); gitService.login(gitLinkUsername, gitLinkPassword);


+ 10
- 24
ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/NewDatasetServiceImpl.java View File

@@ -50,6 +50,8 @@ public class NewDatasetServiceImpl implements NewDatasetService {
private DvcService dvcService; private DvcService dvcService;
@Value("${spring.redis.host}") @Value("${spring.redis.host}")
private String redisHost; private String redisHost;
@Value("${spring.redis.port}")
private Integer redisPort;
@Value("${minio.accessKey}") @Value("${minio.accessKey}")
String accessKeyId; String accessKeyId;
@Value("${minio.secretKey}") @Value("${minio.secretKey}")
@@ -65,7 +67,7 @@ public class NewDatasetServiceImpl implements NewDatasetService {
@Override @Override
public String newCreateDataset(NewDatasetVo datasetVo) throws Exception { public String newCreateDataset(NewDatasetVo datasetVo) throws Exception {
String token = gitService.checkoutToken(); String token = gitService.checkoutToken();
Jedis jedis = new Jedis(redisHost);
Jedis jedis = new Jedis(redisHost,redisPort);
LoginUser loginUser = SecurityUtils.getLoginUser(); LoginUser loginUser = SecurityUtils.getLoginUser();
String ci4sUsername = loginUser.getUsername(); String ci4sUsername = loginUser.getUsername();
String gitLinkUsername = loginUser.getSysUser().getGitLinkUsername(); String gitLinkUsername = loginUser.getSysUser().getGitLinkUsername();
@@ -146,11 +148,10 @@ public class NewDatasetServiceImpl implements NewDatasetService {
} }


public String newCreateVersion(NewDatasetVo datasetVo) throws Exception { public String newCreateVersion(NewDatasetVo datasetVo) throws Exception {
gitService.checkoutToken();
Jedis jedis = new Jedis(redisHost);
String token = gitService.checkoutToken();
Jedis jedis = new Jedis(redisHost,redisPort);
LoginUser loginUser = SecurityUtils.getLoginUser(); LoginUser loginUser = SecurityUtils.getLoginUser();
String ci4sUsername = loginUser.getUsername(); String ci4sUsername = loginUser.getUsername();
String token = jedis.get(ci4sUsername + "_gitToken");
String gitLinkUsername = loginUser.getSysUser().getGitLinkUsername(); String gitLinkUsername = loginUser.getSysUser().getGitLinkUsername();
String gitLinkPassword = loginUser.getSysUser().getGitLinkPassword(); String gitLinkPassword = loginUser.getSysUser().getGitLinkPassword();
String userReq = jedis.get(ci4sUsername + "_gitUserInfo"); String userReq = jedis.get(ci4sUsername + "_gitUserInfo");
@@ -217,11 +218,10 @@ public class NewDatasetServiceImpl implements NewDatasetService {


@Override @Override
public Page<NewDatasetVo> newPersonalQueryByPage(Dataset dataset, PageRequest pageRequest) throws Exception { public Page<NewDatasetVo> newPersonalQueryByPage(Dataset dataset, PageRequest pageRequest) throws Exception {
gitService.checkoutToken();
Jedis jedis = new Jedis(redisHost);
String token = gitService.checkoutToken();
Jedis jedis = new Jedis(redisHost,redisPort);
LoginUser loginUser = SecurityUtils.getLoginUser(); LoginUser loginUser = SecurityUtils.getLoginUser();
String ci4sUsername = loginUser.getUsername(); String ci4sUsername = loginUser.getUsername();
String token = jedis.get(ci4sUsername + "_gitToken");
String userReq = jedis.get(ci4sUsername + "_gitUserInfo"); String userReq = jedis.get(ci4sUsername + "_gitUserInfo");
Map<String, Object> userInfo = JsonUtils.jsonToMap(userReq); Map<String, Object> userInfo = JsonUtils.jsonToMap(userReq);
// 拼接查询url // 拼接查询url
@@ -245,11 +245,10 @@ public class NewDatasetServiceImpl implements NewDatasetService {


@Override @Override
public Page<NewDatasetVo> newPubilcQueryByPage(Dataset dataset, PageRequest pageRequest) throws Exception { public Page<NewDatasetVo> newPubilcQueryByPage(Dataset dataset, PageRequest pageRequest) throws Exception {
gitService.checkoutToken();
Jedis jedis = new Jedis(redisHost);
String token = gitService.checkoutToken();
Jedis jedis = new Jedis(redisHost,redisPort);
LoginUser loginUser = SecurityUtils.getLoginUser(); LoginUser loginUser = SecurityUtils.getLoginUser();
String ci4sUsername = loginUser.getUsername(); String ci4sUsername = loginUser.getUsername();
String token = jedis.get(ci4sUsername + "_gitToken");
String userReq = jedis.get(ci4sUsername + "_gitUserInfo"); String userReq = jedis.get(ci4sUsername + "_gitUserInfo");
Map<String, Object> userInfo = JsonUtils.jsonToMap(userReq); Map<String, Object> userInfo = JsonUtils.jsonToMap(userReq);
Integer userId = (Integer) userInfo.get("user_id"); Integer userId = (Integer) userInfo.get("user_id");
@@ -381,24 +380,11 @@ public class NewDatasetServiceImpl implements NewDatasetService {
public ResponseEntity<InputStreamResource> downloadAllDatasetFilesNew(String name,Integer id, String version) throws Exception { public ResponseEntity<InputStreamResource> downloadAllDatasetFilesNew(String name,Integer id, String version) throws Exception {
// 命令行操作 git clone 项目地址 // 命令行操作 git clone 项目地址
LoginUser loginUser = SecurityUtils.getLoginUser(); LoginUser loginUser = SecurityUtils.getLoginUser();
String gitLinkUsername = loginUser.getSysUser().getGitLinkUsername();
String gitLinkPassword = loginUser.getSysUser().getGitLinkPassword();
String ci4sUsername = loginUser.getUsername(); String ci4sUsername = loginUser.getUsername();
Jedis jedis = new Jedis(redisHost);
Jedis jedis = new Jedis(redisHost,redisPort);
String userReq = jedis.get(ci4sUsername + "_gitUserInfo"); String userReq = jedis.get(ci4sUsername + "_gitUserInfo");
Map<String, Object> userInfo = JsonUtils.jsonToMap(userReq); Map<String, Object> userInfo = JsonUtils.jsonToMap(userReq);
String localPath = localPathlocal+ loginUser.getUsername()+"/datasets/"+id+"/"+name; String localPath = localPathlocal+ loginUser.getUsername()+"/datasets/"+id+"/"+name;
//不需要,没意义,我查看详情已经是最新的了
// if(folder.exists() && folder.isDirectory()){
// //切换分支
// DVCUtils.gitCheckoutBranch(localPath, version);
// //pull
// DVCUtils.gitPull(localPath,gitLinkUsername, gitLinkPassword);
// //dvc pull
// DVCUtils.dvcPull(localPath);
// }else {
// DVCUtils.gitClone(localPath, projectUrl, version, gitLinkUsername, gitLinkPassword);
// }


// 打包 data 文件夹 // 打包 data 文件夹
String dataFolderPath = localPath + "/data"; String dataFolderPath = localPath + "/data";


Loading…
Cancel
Save