content: /*
   * 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.screen;
  
  import java.util.ArrayList;
  import java.util.HashMap;
  import java.util.List;
  import java.util.Map;
  
  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.cobarnode.ConnectionStatus;
  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;
  import com.alibaba.cobar.manager.util.FluenceHashMap;
  import com.alibaba.cobar.manager.util.FormatUtil;
  import com.alibaba.cobar.manager.util.ListSortUtil;
  
  /**
   * @author haiqing.zhuhq 2011-6-27
   */
  public class ConnectionControlScreen extends AbstractController implements InitializingBean {
      private XmlAccesser xmlAccesser;
      private CobarAccesser cobarAccesser;
  
      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)
              throws Exception {
          UserDO user = (UserDO) request.getSession().getAttribute("user");
          String id = request.getParameter("clusterId");
          long clusterId = -1;
          if (null != id) {
              clusterId = Long.parseLong(id);
          }
  
          String cobarNodeId = request.getParameter("cobarNodeId");
          long cobarId = -1;
          if (null != cobarNodeId) {
              cobarId = Long.parseLong(cobarNodeId);
          }
  
          List<ClusterDO> cList = xmlAccesser.getClusterDAO().listAllCluster();
          List<Map<String, Object>> clusterList = new ArrayList<Map<String, Object>>();
  
          ListSortUtil.sortClusterByName(cList);
          for (ClusterDO e : cList) {
              Map<String, Object> map = new HashMap<String, Object>();
              map.put("id", e.getId());
              map.put("name", CobarStringUtil.htmlEscapedString(e.getName()));
              clusterList.add(map);
          }
  
          List<CobarDO> cobarList = null;
  
          if (null != cList && cList.size() > 0) {
              if (-1 == clusterId) {
                  clusterId = cList.get(0).getId();
                  cobarList = xmlAccesser.getCobarDAO().getCobarList(clusterId, ConstantDefine.ACTIVE);
              } else {
                  cobarList = xmlAccesser.getCobarDAO().getCobarList(clusterId, ConstantDefine.ACTIVE);
              }
          }
  
          List<Map<String, Object>> cobarViewList = null;
          if (null != cobarList && cobarList.size() > 0) {
              ListSortUtil.sortCobarByName(cobarList);
              cobarViewList = new ArrayList<Map<String, Object>>();
              for (CobarDO c : cobarList) {
                  CobarAdapterDAO perf = cobarAccesser.getAccesser(c.getId());
                  if (perf.checkConnection()) {
                      Map<String, Object> cobarMap = new HashMap<String, Object>();
                      cobarMap.put("id", c.getId());
                      cobarMap.put("name", CobarStringUtil.htmlEscapedString(c.getName()));
                      cobarViewList.add(cobarMap);
                  }
              }
          }
  
          /* cobarId=-2, cobar InActive; cobarId=-3, connection error */
          List<ConnectionStatus> connecList = null;
          if (cobarId > 0) {
              CobarDO cobar = xmlAccesser.getCobarDAO().getCobarById(cobarId);
              if (cobar.getStatus().equals(ConstantDefine.ACTIVE)) {
                  CobarAdapterDAO perf = cobarAccesser.getAccesser(cobarId);
                  if (!perf.checkConnection()) {
                      cobarId = -3;
                  } else {
                      connecList = perf.listConnectionStatus();
                  }
              } else {
                  cobarId = -2;
              }
          }
  
          List<Map<String, Object>> returnList = new ArrayList<Map<String, Object>>();
  
          if (null != connecList) {
              ListSortUtil.sortConnections(connecList);
              for (ConnectionStatus c : connecList) {
                  Map<String, Object> map = new HashMap<String, Object>();
                  map.put("processor", c.getProcessor());
                  map.put("id", c.getId());
                  map.put("host", c.getHost());
                  map.put("port", c.getPort());
                  map.put("local_port", c.getLocal_port());
                  map.put("schema", c.getSchema());
                  map.put("charset", c.getCharset());
                  map.put("netIn", FormatUtil.formatStore(c.getNetIn()));
                  map.put("netOut", FormatUtil.formatStore(c.getNetOut()));
                  map.put("aliveTime", FormatUtil.formatTime(c.getAliveTime() * 1000, 2));
                  map.put("attempsCount", FormatUtil.formatNumber(c.getAttempsCount()));
                  map.put("recvBuffer", FormatUtil.formatStore(c.getRecvBuffer()));
                  map.put("sendQueue", c.getSendQueue());
                  map.put("channel", c.getChannel());
                  returnList.add(map);
              }
          }
  
          return new ModelAndView(
                  "c_connection",
                  new FluenceHashMap<String, Object>().putKeyValue("cList", clusterList)
                                                      .putKeyValue("cobarList", cobarViewList)
                                                      .putKeyValue("clusterId", clusterId)
                                                      .putKeyValue("cobarId", cobarId)
                                                      .putKeyValue("connecList", returnList)
                                                      .putKeyValue("user", user));
  
      }
  
  }
  