初始化

lrj

2 天以前 c61d4fe27c97d2ecc907756aa571a4ef14a7b9b6

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
package com.rongyichuang.judge.service;
 
import com.rongyichuang.judge.entity.Tag;
import com.rongyichuang.judge.repository.TagRepository;
import org.springframework.boot.CommandLineRunner;
import org.springframework.stereotype.Service;
 
import java.util.Arrays;
import java.util.List;
 
@Service
public class DataInitService implements CommandLineRunner {
 
    private final TagRepository tagRepository;
 
    public DataInitService(TagRepository tagRepository) {
        this.tagRepository = tagRepository;
    }
 
    @Override
    public void run(String... args) throws Exception {
        // 检查是否已有数据
        if (tagRepository.count() == 0) {
            initTags();
        }
    }
 
    private void initTags() {
        List<Tag> tags = Arrays.asList(
            createTag("软件工程", "software_engineering", "major"),
            createTag("人工智能", "artificial_intelligence", "major"),
            createTag("机械设计", "mechanical_design", "major"),
            createTag("工业设计", "industrial_design", "major"),
            createTag("创新创业", "innovation_entrepreneurship", "major"),
            createTag("项目管理", "project_management", "major"),
            createTag("数据科学", "data_science", "major"),
            createTag("网络安全", "cyber_security", "major"),
            createTag("物联网", "internet_of_things", "major"),
            createTag("区块链", "blockchain", "major"),
            createTag("云计算", "cloud_computing", "major"),
            createTag("移动开发", "mobile_development", "major")
        );
 
        tagRepository.saveAll(tags);
        System.out.println("初始化专业标签数据完成，共创建 " + tags.size() + " 个标签");
    }
 
    private Tag createTag(String name, String code, String category) {
        Tag tag = new Tag();
        tag.setName(name);
        tag.setCode(code);
        tag.setCategory(category);
        return tag;
    }
}