Ejemplo n.º 1
0
 /**
  * response method.
  *
  * @param c
  */
 public static void response(ServerConnection c, String stmt, int type) {
   String showSchemal = SchemaUtil.parseShowTableSchema(stmt);
   String cSchema = showSchemal == null ? c.getSchema() : showSchemal;
   SchemaConfig schema = CloudbServer.getInstance().getConfig().getSchemas().get(cSchema);
   if (schema != null) {
     // 不分库的schema,show tables从后端 mysql中查
     String node = schema.getDataNode();
     if (!Strings.isNullOrEmpty(node)) {
       c.execute(stmt, ServerParse.SHOW, null);
       return;
     }
   } else {
     c.writeErrMessage(ErrorCode.ER_NO_DB_ERROR, "No database selected");
   }
   // 分库的schema,直接从SchemaConfig中获取所有表名
   Map<String, String> parm = buildFields(c, stmt);
   java.util.Set<String> tableSet = getTableSet(c, parm);
   int i = 0;
   byte packetId = 0;
   header.packetId = ++packetId;
   fields[i] =
       PacketUtil.getField("Tables in " + parm.get(SCHEMA_KEY), Fields.FIELD_TYPE_VAR_STRING);
   fields[i++].packetId = ++packetId;
   eof.packetId = ++packetId;
   ByteBuffer buffer = c.allocate();
   // write header
   buffer = header.write(buffer, c, true);
   // write fields
   for (FieldPacket field : fields) {
     buffer = field.write(buffer, c, true);
   }
   // write eof
   buffer = eof.write(buffer, c, true);
   // write rows
   packetId = eof.packetId;
   for (String name : tableSet) {
     RowDataPacket row = new RowDataPacket(FIELD_COUNT);
     row.add(StringUtil.encode(name.toLowerCase(), c.getCharset()));
     row.packetId = ++packetId;
     buffer = row.write(buffer, c, true);
   }
   // write last eof
   EOFPacket lastEof = new EOFPacket();
   lastEof.packetId = ++packetId;
   buffer = lastEof.write(buffer, c, true);
   // post write
   c.write(buffer);
 }
Ejemplo n.º 2
0
/**
 * @Package com.tydic.cloudbserver.server.response @ClassName ShowTables.java
 *
 * @author xiangsl
 * @date 2016年2月1日 下午12:25:31 @Description: 显示表信息……
 * @version V1.0
 */
public class ShowTables {
  private static final int FIELD_COUNT = 1;
  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();
  private static final String SCHEMA_KEY = "schemaName";
  private static final String LIKE_KEY = "like";
  private static final Pattern pattern =
      Pattern.compile(
          "^\\s*(SHOW)\\s+(TABLES)(\\s+(FROM)\\s+([a-zA-Z_0-9]+))?(\\s+(LIKE\\s+'(.*)'))?\\s*",
          Pattern.CASE_INSENSITIVE);

  /**
   * response method.
   *
   * @param c
   */
  public static void response(ServerConnection c, String stmt, int type) {
    String showSchemal = SchemaUtil.parseShowTableSchema(stmt);
    String cSchema = showSchemal == null ? c.getSchema() : showSchemal;
    SchemaConfig schema = CloudbServer.getInstance().getConfig().getSchemas().get(cSchema);
    if (schema != null) {
      // 不分库的schema,show tables从后端 mysql中查
      String node = schema.getDataNode();
      if (!Strings.isNullOrEmpty(node)) {
        c.execute(stmt, ServerParse.SHOW, null);
        return;
      }
    } else {
      c.writeErrMessage(ErrorCode.ER_NO_DB_ERROR, "No database selected");
    }
    // 分库的schema,直接从SchemaConfig中获取所有表名
    Map<String, String> parm = buildFields(c, stmt);
    java.util.Set<String> tableSet = getTableSet(c, parm);
    int i = 0;
    byte packetId = 0;
    header.packetId = ++packetId;
    fields[i] =
        PacketUtil.getField("Tables in " + parm.get(SCHEMA_KEY), Fields.FIELD_TYPE_VAR_STRING);
    fields[i++].packetId = ++packetId;
    eof.packetId = ++packetId;
    ByteBuffer buffer = c.allocate();
    // write header
    buffer = header.write(buffer, c, true);
    // write fields
    for (FieldPacket field : fields) {
      buffer = field.write(buffer, c, true);
    }
    // write eof
    buffer = eof.write(buffer, c, true);
    // write rows
    packetId = eof.packetId;
    for (String name : tableSet) {
      RowDataPacket row = new RowDataPacket(FIELD_COUNT);
      row.add(StringUtil.encode(name.toLowerCase(), c.getCharset()));
      row.packetId = ++packetId;
      buffer = row.write(buffer, c, true);
    }
    // write last eof
    EOFPacket lastEof = new EOFPacket();
    lastEof.packetId = ++packetId;
    buffer = lastEof.write(buffer, c, true);
    // post write
    c.write(buffer);
  }

  public static Set<String> getTableSet(ServerConnection c, String stmt) {
    Map<String, String> parm = buildFields(c, stmt);
    return getTableSet(c, parm);
  }

  private static Set<String> getTableSet(ServerConnection c, Map<String, String> parm) {
    TreeSet<String> tableSet = new TreeSet<String>();
    CloudbConfig conf = CloudbServer.getInstance().getConfig();
    Map<String, UserConfig> users = conf.getUsers();
    UserConfig user = users == null ? null : users.get(c.getUser());
    if (user != null) {
      Map<String, SchemaConfig> schemas = conf.getSchemas();
      for (String name : schemas.keySet()) {
        if (null != parm.get(SCHEMA_KEY)
            && parm.get(SCHEMA_KEY).toUpperCase().equals(name.toUpperCase())) {
          if (null == parm.get("LIKE_KEY")) {
            tableSet.addAll(schemas.get(name).getTables().keySet());
          } else {
            String p = "^" + parm.get("LIKE_KEY").replaceAll("%", ".*");
            Pattern pattern = Pattern.compile(p, Pattern.CASE_INSENSITIVE);
            Matcher ma;
            for (String tname : schemas.get(name).getTables().keySet()) {
              ma = pattern.matcher(tname);
              if (ma.matches()) {
                tableSet.add(tname);
              }
            }
          }
        }
      }
      ;
    }
    return tableSet;
  }

  /**
   * build fields
   *
   * @param c
   * @param stmt
   */
  private static Map<String, String> buildFields(ServerConnection c, String stmt) {
    Map<String, String> map = new HashMap<String, String>();
    Matcher ma = pattern.matcher(stmt);
    if (ma.find()) {
      String schemaName = ma.group(5);
      if (null != schemaName && (!"".equals(schemaName)) && (!"null".equals(schemaName))) {
        map.put(SCHEMA_KEY, schemaName);
      }
      String like = ma.group(8);
      if (null != like && (!"".equals(like)) && (!"null".equals(like))) {
        map.put("LIKE_KEY", like);
      }
    }
    if (null == map.get(SCHEMA_KEY)) {
      map.put(SCHEMA_KEY, c.getSchema());
    }
    return map;
  }
}