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; }
/** * 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); }