content: /*
   * Copyright (C) 2016-2023 ActionTech.
   * based on code by MyCATCopyrightHolder Copyright (c) 2013, OpenCloudDB/MyCAT.
   * License: http://www.gnu.org/licenses/gpl.html GPL version 2 or higher.
   */
  package com.actiontech.dble.services.manager.response;
  
  import com.actiontech.dble.DbleServer;
  import com.actiontech.dble.backend.datasource.PhysicalDbGroup;
  import com.actiontech.dble.backend.datasource.PhysicalDbInstance;
  import com.actiontech.dble.backend.datasource.ShardingNode;
  import com.actiontech.dble.backend.mysql.PacketUtil;
  import com.actiontech.dble.config.Fields;
  import com.actiontech.dble.config.ServerConfig;
  import com.actiontech.dble.net.mysql.*;
  import com.actiontech.dble.services.manager.ManagerService;
  import com.actiontech.dble.util.IntegerUtil;
  import com.actiontech.dble.util.LongUtil;
  import com.actiontech.dble.util.StringUtil;
  
  import java.nio.ByteBuffer;
  import java.util.Map;
  
  public final class ShowDbInstance {
      private ShowDbInstance() {
      }
  
      private static final int FIELD_COUNT = 11;
      private static final ResultSetHeaderPacket HEADER = PacketUtil.getHeader(FIELD_COUNT);
      private static final FieldPacket[] FIELDS = new FieldPacket[FIELD_COUNT];
      private static final EOFPacket EOF = new EOFPacket();
  
      static {
          int i = 0;
          byte packetId = 0;
          HEADER.setPacketId(++packetId);
  
          FIELDS[i] = PacketUtil.getField("DB_GROUP", Fields.FIELD_TYPE_VAR_STRING);
          FIELDS[i++].setPacketId(++packetId);
  
          FIELDS[i] = PacketUtil.getField("NAME", Fields.FIELD_TYPE_VAR_STRING);
          FIELDS[i++].setPacketId(++packetId);
  
          FIELDS[i] = PacketUtil.getField("HOST", Fields.FIELD_TYPE_VAR_STRING);
          FIELDS[i++].setPacketId(++packetId);
  
          FIELDS[i] = PacketUtil.getField("PORT", Fields.FIELD_TYPE_LONG);
          FIELDS[i++].setPacketId(++packetId);
  
          FIELDS[i] = PacketUtil.getField("W/R", Fields.FIELD_TYPE_VAR_STRING);
          FIELDS[i++].setPacketId(++packetId);
  
          FIELDS[i] = PacketUtil.getField("ACTIVE", Fields.FIELD_TYPE_LONG);
          FIELDS[i++].setPacketId(++packetId);
  
          FIELDS[i] = PacketUtil.getField("IDLE", Fields.FIELD_TYPE_LONG);
          FIELDS[i++].setPacketId(++packetId);
  
          FIELDS[i] = PacketUtil.getField("SIZE", Fields.FIELD_TYPE_LONG);
          FIELDS[i++].setPacketId(++packetId);
  
          FIELDS[i] = PacketUtil.getField("READ_LOAD", Fields.FIELD_TYPE_LONG);
          FIELDS[i++].setPacketId(++packetId);
  
          FIELDS[i] = PacketUtil.getField("WRITE_LOAD", Fields.FIELD_TYPE_LONG);
          FIELDS[i++].setPacketId(++packetId);
  
          FIELDS[i] = PacketUtil.getField("DISABLED", Fields.FIELD_TYPE_VAR_STRING);
          FIELDS[i].setPacketId(++packetId);
  
          EOF.setPacketId(++packetId);
      }
  
      public static void execute(ManagerService service, String name) {
          ByteBuffer buffer = service.allocate();
  
          // write header
          buffer = HEADER.write(buffer, service, true);
  
          // write fields
          for (FieldPacket field : FIELDS) {
              buffer = field.write(buffer, service, true);
          }
  
          // write eof
          buffer = EOF.write(buffer, service, true);
  
          // write rows
          byte packetId = EOF.getPacketId();
          ServerConfig conf = DbleServer.getInstance().getConfig();
  
          if (null != name) {
              ShardingNode dn = conf.getShardingNodes().get(StringUtil.removeAllApostrophe(name));
              if (dn != null) {
                  for (PhysicalDbInstance w : dn.getDbGroup().getDbInstances(true)) {
                      RowDataPacket row = getRow(w.getDbGroupConfig().getName(), w, service.getCharset().getResults());
                      row.setPacketId(++packetId);
                      buffer = row.write(buffer, service, true);
                  }
              }
          } else {
              // add all
              for (Map.Entry<String, PhysicalDbGroup> entry : conf.getDbGroups().entrySet()) {
                  PhysicalDbGroup dbGroup = entry.getValue();
                  String dbGroupName = entry.getKey();
                  for (PhysicalDbInstance source : dbGroup.getDbInstances(true)) {
                      RowDataPacket sRow = getRow(dbGroupName, source, service.getCharset().getResults());
                      sRow.setPacketId(++packetId);
                      buffer = sRow.write(buffer, service, true);
                  }
              }
          }
          // write last eof
          EOFRowPacket lastEof = new EOFRowPacket();
          lastEof.setPacketId(++packetId);
  
  
          lastEof.write(buffer, service);
      }
  
      private static RowDataPacket getRow(String dbGroup, PhysicalDbInstance ds,
                                          String charset) {
          RowDataPacket row = new RowDataPacket(FIELD_COUNT);
          row.add(StringUtil.encode(dbGroup, charset));
          row.add(StringUtil.encode(ds.getName(), charset));
          row.add(StringUtil.encode(ds.getConfig().getIp(), charset));
          row.add(IntegerUtil.toBytes(ds.getConfig().getPort()));
          row.add(StringUtil.encode(ds.isReadInstance() ? "R" : "W", charset));
          row.add(IntegerUtil.toBytes(ds.getActiveConnections()));
          row.add(IntegerUtil.toBytes(ds.getIdleConnections()));
          row.add(LongUtil.toBytes(ds.getConfig().getMaxCon()));
          row.add(LongUtil.toBytes(ds.getCount(true)));
          row.add(LongUtil.toBytes(ds.getCount(false)));
          row.add(StringUtil.encode(ds.isDisabled() ? "true" : "false", charset));
          return row;
      }
  
  }
  