/*
 * Copyright 1999-2012 Alibaba Group.
 *  
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *  
 *      http://www.apache.org/licenses/LICENSE-2.0
 *  
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.alibaba.cobar.manager.web.action;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.locks.ReentrantLock;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.springframework.beans.factory.InitializingBean;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.servlet.mvc.AbstractController;

import com.alibaba.cobar.manager.dao.CobarAdapterDAO;
import com.alibaba.cobar.manager.dataobject.xml.ClusterDO;
import com.alibaba.cobar.manager.dataobject.xml.CobarDO;
import com.alibaba.cobar.manager.dataobject.xml.UserDO;
import com.alibaba.cobar.manager.service.CobarAccesser;
import com.alibaba.cobar.manager.service.XmlAccesser;
import com.alibaba.cobar.manager.util.CobarStringUtil;
import com.alibaba.cobar.manager.util.ConstantDefine;

/**
 * @author haiqing.zhuhq 2011-8-17
 */
public class StopHeartbeat extends AbstractController implements InitializingBean {
    private XmlAccesser xmlAccesser;
    private CobarAccesser cobarAccesser;
    private static final ReentrantLock lock = new ReentrantLock();

    public void setXmlAccesser(XmlAccesser xmlAccesser) {
        this.xmlAccesser = xmlAccesser;
    }

    public void setCobarAccesser(CobarAccesser cobarAccesser) {
        this.cobarAccesser = cobarAccesser;
    }

    @Override
    public void afterPropertiesSet() throws Exception {
        if (xmlAccesser == null) {
            throw new IllegalArgumentException("property 'xmlAccesser' is null!");
        }
        if (null == cobarAccesser) {
            throw new IllegalArgumentException("property 'cobarAccesser' is null!");
        }
    }

    @Override
    protected ModelAndView handleRequestInternal(HttpServletRequest request, HttpServletResponse response) {
        UserDO user = (UserDO) request.getSession().getAttribute("user");
        String datanodes = request.getParameter("datanodes");
        int time = Integer.parseInt(request.getParameter("stoptime"));
        long clusterId = Long.parseLong(request.getParameter("clusterIdK"));
        ClusterDO cluster = xmlAccesser.getClusterDAO().getClusterById(clusterId);
        List<Map<String, Object>> resultList = new ArrayList<Map<String, Object>>();

        if (logger.isWarnEnabled()) {
            StringBuilder log = new StringBuilder();
            log.append(user.getUsername()).append(" | stop heartbeat | cluster:");
            log.append(cluster.getName()).append(" | ");
            log.append(datanodes).append(" | time:");
            log.append(time);
            logger.warn(log.toString());
        }

        List<CobarDO> cobarList = null;

        lock.lock();
        try {
            cobarList = xmlAccesser.getCobarDAO().getCobarList(clusterId, ConstantDefine.ACTIVE);
            for (CobarDO c : cobarList) {
                CobarAdapterDAO control = cobarAccesser.getAccesser(c.getId());
                Map<String, Object> map = new HashMap<String, Object>();
                map.put("name", CobarStringUtil.htmlEscapedString(c.getName()));
                if (control.checkConnection()) {
                    int num = control.stopHeartbeat(datanodes, time);
                    map.put("result", num + " rows");
                } else {
                    map.put("result", "connction error!");
                }
                resultList.add(map);
            }
        } catch (Exception e) {
            logger.error(e.getMessage(), e);
            Map<String, Object> map = new HashMap<String, Object>();
            map.put("name", "UNKNOWN ERROR");
            map.put("result", "unknown exception occurs when stopping heartbeat");
            resultList.clear();
            resultList.add(map);
        } finally {
            lock.unlock();
        }

        return new ModelAndView("c_result", "resultList", resultList);

    }
}