com.alibaba.tac.engine.ms.service.redis.RedisMsPublisher Maven / Gradle / Ivy

Go to download
/*
 *   MIT License
 *
 *   Copyright (c) 2016 Alibaba Group
 *
 *   Permission is hereby granted, free of charge, to any person obtaining a copy
 *   of this software and associated documentation files (the "Software"), to deal
 *   in the Software without restriction, including without limitation the rights
 *   to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 *   copies of the Software, and to permit persons to whom the Software is
 *   furnished to do so, subject to the following conditions:
 *
 *   The above copyright notice and this permission notice shall be included in all
 *   copies or substantial portions of the Software.
 *
 *   THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 *   IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 *   FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 *   AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 *   LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 *   OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 *   SOFTWARE.
 */

package com.alibaba.tac.engine.ms.service.redis;

import com.alibaba.fastjson.JSONObject;
import com.alibaba.tac.engine.code.CodeCompileService;
import com.alibaba.tac.engine.git.GitRepoService;
import com.alibaba.tac.engine.inst.domain.TacInst;
import com.alibaba.tac.engine.inst.service.DevMsInstFileService;
import com.alibaba.tac.engine.inst.service.IMsInstService;
import com.alibaba.tac.engine.inst.service.redis.RedisMsInstFileService;
import com.alibaba.tac.engine.ms.domain.TacMsDO;
import com.alibaba.tac.engine.ms.domain.TacMsPublishMeta;
import com.alibaba.tac.engine.ms.service.AbstractDefaultMsPublisher;
import com.alibaba.tac.engine.ms.service.IMsService;
import com.alibaba.tac.engine.properties.TacMsConstants;
import com.alibaba.tac.engine.properties.TacRedisConfigProperties;
import com.alibaba.tac.engine.service.TacFileService;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.data.redis.core.HashOperations;
import org.springframework.data.redis.core.RedisTemplate;

import javax.annotation.Resource;

/**
 * @author jinshuan.li 2018/3/1 07:58 the redis ms publisher
 */
@Slf4j
public class RedisMsPublisher extends AbstractDefaultMsPublisher {

    @Resource
    private DevMsInstFileService devMsInstFileService;

    @Resource(name = "remoteMsInstFileService")
    private RedisMsInstFileService redisMsInstFileService;

    @Resource(name = "prePublishMsInstFileService")
    private RedisMsInstFileService prePublishMsInstFileService;

    @Resource
    private TacRedisConfigProperties tacRedisConfigProperties;

    @Resource(name = "redisTemplate")
    HashOperations hashOperations;

    @Resource
    private GitRepoService gitRepoService;

    @Resource
    private IMsInstService msInstService;

    @Resource
    private CodeCompileService codeCompileService;

    @Resource
    private IMsService msService;

    /**
     * the subsciber is serializ with string ,then the publisher should serialize with string too;
     */
    @Resource(name = "counterRedisTemplate")
    private RedisTemplate redisTopicTemplate;

    @Override
    public Boolean publish(TacInst tacInst, byte[] data) {

        byte[] instanceFile = data;

        tacInst.setJarVersion(TacFileService.getMd5(data));
        tacInst.setStatus(TacInst.STATUS_PUBLISH);
        // 1 save data
        redisMsInstFileService.saveInstanceFile(tacInst, instanceFile);

        // 2. save meta data

        String msCode = tacInst.getMsCode();

        TacMsPublishMeta publishMeta = new TacMsPublishMeta(tacInst);

        hashOperations.put(getMainKey(), msCode, publishMeta);

        // 3. send message
        redisTopicTemplate.convertAndSend(getPublishChannel(), JSONObject.toJSONString(publishMeta));

        TacMsDO ms = msService.getMs(tacInst.getMsCode());

        // 4 update ms

        Long oldPublishId = ms.getPublishedInstId();

        ms.setPublishedInstId(tacInst.getId());
        msService.updateMs(tacInst.getMsCode(), ms);

        // 5 update instance

        msInstService.updateTacMsInst(tacInst.getId(), tacInst);

        if (oldPublishId != null && !oldPublishId.equals(tacInst.getId())) {
            TacInst tacMsInst = msInstService.getTacMsInst(oldPublishId);

            if (tacMsInst != null) {
                tacMsInst.setStatus(TacInst.STATUS_PRE_PUBLISH);
                msInstService.updateTacMsInst(oldPublishId, tacMsInst);
            }
        }

        return true;
    }

    @Override
    public Boolean offline(TacInst tacInst) {

        String msCode = tacInst.getMsCode();

        // 1. update data
        TacMsPublishMeta publishMeta = new TacMsPublishMeta(tacInst, TacMsConstants.INST_STATUS_OFFLINE);

        hashOperations.put(getMainKey(), msCode, publishMeta);

        // 2. send message
        redisTopicTemplate.convertAndSend(getPublishChannel(), JSONObject.toJSONString(publishMeta));

        return true;
    }

    private String getMainKey() {

        return tacRedisConfigProperties.getMsListPath();
    }

    private String getPublishChannel() {

        return tacRedisConfigProperties.getPublishEventChannel();
    }

    @Override
    public Boolean prePublish(TacInst tacInst, byte[] data) {

        byte[] instanceFile = data;
        tacInst.setStatus(TacInst.STATUS_PRE_PUBLISH);
        tacInst.setPrePublishJarVersion(TacFileService.getMd5(data));

        // 1 save data
        prePublishMsInstFileService.saveInstanceFile(tacInst, instanceFile);

        // 2 update meta data

        msInstService.updateTacMsInst(tacInst.getId(), tacInst);

        return true;

    }

    @Override
    public TacInst gitPrePublish(TacMsDO tacMsDO, TacInst tacInst) {

        if (StringUtils.isEmpty(tacMsDO.getGitRepo())) {
            throw new IllegalArgumentException("no git repo address");
        }

        String gitBranch = tacInst.getGitBranch();

        if (StringUtils.isEmpty(gitBranch)) {
            throw new IllegalArgumentException("git branch is emplty");
        }
        TacMsDO ms = tacMsDO;

        String sourcePath = gitRepoService.pullInstanceCode(ms.getGitRepo(), ms.getCode(), tacInst.getGitBranch());

        try {
            codeCompileService.compile(tacInst.getId(), sourcePath);

            byte[] jarFile = codeCompileService.getJarFile(tacInst.getId());

            this.prePublish(tacInst, jarFile);

        } catch (Exception e) {
            throw new IllegalStateException(e);
        }

        return msInstService.getTacMsInst(tacInst.getId());
    }

}