com.github.developframework.mock.MockTask Maven / Gradle / Ivy

Go to download
package com.github.developframework.mock;

import com.github.developframework.mock.random.RandomGenerator;

import java.util.LinkedList;
import java.util.List;
import java.util.Optional;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * 一个任务
 *
 * @author qiuzhenhao
 * @since 0.1
 */
public class MockTask {

    private static final String REGEX = "(?=\\$\\{)(.*?)(?<=\\})";

    private RandomGeneratorRegistry randomGeneratorRegistry;

    private String template;

    private MockCache cache;

    public MockTask(RandomGeneratorRegistry randomGeneratorRegistry, MockCache cache, String template) {
        this.randomGeneratorRegistry = randomGeneratorRegistry;
        this.cache = cache;
        this.template = template;
    }

    /**
     * 运行任务
     *
     * @return 随机生成结果
     */
    public String run() {
        List mockPlaceholders = extractPlaceholder(template);
        String result = template;
        for (MockPlaceholder mockPlaceholder : mockPlaceholders) {
            RandomGenerator randomGenerator = randomGeneratorRegistry.getRandomGenerator(mockPlaceholder.getName());
            Optional idOptional = mockPlaceholder.getId();
            Object value = randomGenerator.randomValue(randomGeneratorRegistry, mockPlaceholder, cache);
            // 加入缓存
            if (idOptional.isPresent()) {
                cache.put(idOptional.get(), new MockCache.Cache(value, mockPlaceholder));
            }
            final String stringValue = randomGenerator.forString(mockPlaceholder, value);
            result = result.replace(mockPlaceholder.getPlaceholder(), stringValue);
        }
        return result;
    }

    private List extractPlaceholder(String template) {
        List list = new LinkedList<>();
        Pattern pattern = Pattern.compile(REGEX);
        Matcher m = pattern.matcher(template);
        while(m.find()) {
            list.add(new MockPlaceholder(m.group()));
        }
        return list;
    }
}