package com.ctrip.framework.apollo.biz.message;

 import com.ctrip.framework.apollo.biz.entity.ReleaseMessage;

 import com.ctrip.framework.apollo.biz.repository.ReleaseMessageRepository;

 import com.dianping.cat.Cat;

 import com.dianping.cat.message.Message;

 import com.dianping.cat.message.Transaction;

 import org.slf4j.Logger;

 import org.slf4j.LoggerFactory;

 import org.springframework.beans.factory.annotation.Autowired;

 import org.springframework.stereotype.Component;

 import java.util.Objects;

 /**

  * @author Jason Song(song_s@ctrip.com)

  */

 @Component

 public class DatabaseMessageSender implements MessageSender {

   private static final Logger logger = LoggerFactory.getLogger(DatabaseMessageSender.class);

   @Autowired

   private ReleaseMessageRepository releaseMessageRepository;

   @Override

   public void sendMessage(String message, String channel) {

     logger.info("Sending message {} to channel {}", message, channel);

     if (!Objects.equals(channel, Topics.APOLLO_RELEASE_TOPIC)) {

       logger.warn("Channel {} not supported by DatabaseMessageSender!");

       return;

     }

     Cat.logEvent("Apollo.AdminService.ReleaseMessage", message);

     Transaction transaction = Cat.newTransaction("Apollo.AdminService", "sendMessage");

     try {

       releaseMessageRepository.save(new ReleaseMessage(message));

       transaction.setStatus(Message.SUCCESS);

     } catch (Throwable ex) {

       logger.error("Sending message to database failed", ex);

       transaction.setStatus(ex);

     } finally {

       transaction.complete();

     }

   }

 }