Skip to content

Commit 0d5200c

Browse files
author
caryliao(廖飞强)
committed
add admin getDataset by datasetId
1 parent b1dcca7 commit 0d5200c

File tree

4 files changed

+21
-10
lines changed

4 files changed

+21
-10
lines changed

wedpr-components/admin/src/main/java/com/webank/wedpr/components/admin/controller/WedprDatasetController.java

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
import com.webank.wedpr.components.admin.common.Utils;
44
import com.webank.wedpr.components.admin.request.GetWedprDatasetListRequest;
55
import com.webank.wedpr.components.admin.service.WedprDatasetService;
6+
import com.webank.wedpr.components.dataset.dao.Dataset;
67
import com.webank.wedpr.components.dataset.message.ListDatasetResponse;
78
import com.webank.wedpr.components.token.auth.model.UserToken;
89
import com.webank.wedpr.core.utils.Constant;
@@ -12,6 +13,7 @@
1213
import org.springframework.beans.factory.annotation.Autowired;
1314
import org.springframework.web.bind.annotation.GetMapping;
1415
import org.springframework.web.bind.annotation.RequestMapping;
16+
import org.springframework.web.bind.annotation.RequestParam;
1517
import org.springframework.web.bind.annotation.RestController;
1618

1719
/**
@@ -43,4 +45,18 @@ public WeDPRResponse listDataset(
4345
return new WeDPRResponse(Constant.WEDPR_FAILED, e.getMessage());
4446
}
4547
}
48+
49+
@GetMapping("/queryDataset")
50+
public WeDPRResponse queryDataset(
51+
@RequestParam("datasetId") String datasetId, HttpServletRequest request) {
52+
try {
53+
// check user permission
54+
UserToken userToken = Utils.checkPermission(request);
55+
Dataset dataset = wedprDatasetService.getById(datasetId);
56+
return new WeDPRResponse(Constant.WEDPR_SUCCESS, Constant.WEDPR_SUCCESS_MSG, dataset);
57+
} catch (Exception e) {
58+
log.error("queryDataset error", e);
59+
return new WeDPRResponse(Constant.WEDPR_FAILED, e.getMessage());
60+
}
61+
}
4662
}

wedpr-components/dataset/src/main/java/com/webank/wedpr/components/dataset/dao/Dataset.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
package com.webank.wedpr.components.dataset.dao;
22

33
import com.baomidou.mybatisplus.annotation.TableField;
4+
import com.baomidou.mybatisplus.annotation.TableId;
45
import com.baomidou.mybatisplus.annotation.TableName;
56
import com.fasterxml.jackson.annotation.JsonProperty;
67
import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
@@ -14,6 +15,7 @@
1415
@Data
1516
public class Dataset {
1617

18+
@TableId("dataset_id")
1719
private String datasetId;
1820

1921
private String datasetLabel;

wedpr-components/mybatis/src/main/java/com/webank/wedpr/components/mybatis/tool/GenerateCodeMain.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,10 +23,10 @@ public static void main(String[] args) {
2323
// 数据源配置
2424
com.baomidou.mybatisplus.generator.config.DataSourceConfig dsc = new DataSourceConfig();
2525
dsc.setUrl(
26-
"jdbc:mysql://139.159.202.235:3306/cary?characterEncoding=utf8&useSSL=false&serverTimezone=UTC");
26+
"jdbc:mysql://127.0.0.1:3306/wedpr?characterEncoding=utf8&useSSL=false&serverTimezone=UTC");
2727
dsc.setDriverName("com.mysql.cj.jdbc.Driver");
28-
dsc.setUsername("root");
29-
dsc.setPassword("Wedpr2023");
28+
dsc.setUsername("");
29+
dsc.setPassword("");
3030
mpg.setDataSource(dsc);
3131

3232
// 全局配置
Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,23 +1,17 @@
11
/** Copyright (C) @2014-2022 Webank */
22
package com.webank.wedpr.components.report;
33

4-
import com.webank.wedpr.components.dataset.dao.Dataset;
5-
import com.webank.wedpr.components.report.service.WedprDatasetService;
64
import java.time.LocalDateTime;
7-
import java.util.List;
85
import lombok.extern.slf4j.Slf4j;
96
import org.quartz.DisallowConcurrentExecution;
107
import org.quartz.Job;
118
import org.quartz.JobExecutionContext;
12-
import org.springframework.beans.factory.annotation.Autowired;
139
import org.springframework.transaction.annotation.Transactional;
1410

1511
@DisallowConcurrentExecution
1612
@Slf4j
1713
public class ReportQuartzJob implements Job {
1814

19-
@Autowired private WedprDatasetService wedprDatasetService;
20-
2115
@Override
2216
public void execute(JobExecutionContext context) {
2317
log.info("ReportQuartzJob run at:{}", LocalDateTime.now());
@@ -30,7 +24,6 @@ public void execute(JobExecutionContext context) {
3024

3125
@Transactional(rollbackFor = Exception.class)
3226
private void doReport() {
33-
List<Dataset> wedprDatasetList = wedprDatasetService.list();
3427
log.info("do report...");
3528
}
3629
}

0 commit comments

Comments
 (0)