function schema_mysql_inspect in Schema 6

Same name and namespace in other branches

1. 5 engines/schema_mysql.inc \schema_mysql_inspect()

1 call to schema_mysql_inspect()

schema_mysqli_inspect in engines/schema_mysqli.inc

File

engines/schema_mysql.inc, line 81

Code

function schema_mysql_inspect($name = NULL) {
  global $db_url;

  // Switch to the active database connection.
  // The only way to get the active connection's name is as a return value from
  // db_set_active(). However, calling this function will automatically switch
  // the active connection to 'default', which might not be what we want.
  // Therefore, we must immediately call db_set_active() again with the desired
  // connection name in order to proceed.
  $active_db_connection = db_set_active();
  db_set_active($active_db_connection);
  $tables = array();
  $url = parse_url(is_array($db_url) ? $db_url[$active_db_connection] : $db_url);
  $database = substr($url['path'], 1);
  $sql = 'SELECT TABLE_NAME, TABLE_COMMENT
          FROM information_schema.TABLES
          WHERE TABLE_SCHEMA="%s" ';
  if (isset($name)) {
    $sql .= 'AND TABLE_NAME = "%s" ';
  }
  $res = db_query($sql, $database, $name);
  while ($r = db_fetch_array($res)) {
    $tables[$r['TABLE_NAME']]['description'] = $r['TABLE_COMMENT'];
  }
  $sql = 'SELECT TABLE_NAME, COLUMN_TYPE, COLUMN_NAME, COLUMN_DEFAULT,
                 EXTRA, IS_NULLABLE, NUMERIC_SCALE, COLUMN_COMMENT
          FROM information_schema.COLUMNS
          WHERE TABLE_SCHEMA="%s" ';
  if (isset($name)) {
    $sql .= 'AND TABLE_NAME = "%s" ';
  }
  $sql .= 'ORDER BY TABLE_NAME, ORDINAL_POSITION';
  $res = db_query($sql, $database, $name);
  while ($r = db_fetch_array($res)) {
    $r['NEW_TABLE_NAME'] = schema_unprefix_table($r['TABLE_NAME']);
    $numeric = !is_null($r['NUMERIC_SCALE']);
    $col = array();
    $col['type'] = $r['COLUMN_TYPE'];
    if (preg_match('@([a-z]+)(?:\\((\\d+)(?:,(\\d+))?\\))?\\s*(unsigned)?@', $col['type'], $matches)) {
      list($col['type'], $col['size']) = schema_schema_type($matches[1], $r['TABLE_NAME'], $r['COLUMN_NAME'], 'mysql');
      if (isset($matches[2])) {
        if ($col['type'] == 'numeric' || $col['type'] == 'float' || $col['type'] == 'double') {
          $col['precision'] = $matches[2];
          $col['scale'] = $matches[3];
        }
        else {
          if (!$numeric) {
            $col['length'] = $matches[2];
          }
        }
      }
      if (isset($matches[4])) {
        $col['unsigned'] = TRUE;
      }
    }
    if ($col['type'] == 'int' && isset($r['EXTRA']) && $r['EXTRA'] == 'auto_increment') {
      $col['type'] = 'serial';
    }
    $col['not null'] = $r['IS_NULLABLE'] == 'YES' ? FALSE : TRUE;
    if (!is_null($r['COLUMN_DEFAULT'])) {
      if ($numeric) {

        // XXX floats!
        $col['default'] = intval($r['COLUMN_DEFAULT']);
      }
      else {
        $col['default'] = $r['COLUMN_DEFAULT'];
      }
    }
    $col['description'] = $r['COLUMN_COMMENT'];
    $tables[$r['TABLE_NAME']]['fields'][$r['COLUMN_NAME']] = $col;

    // At this point, $tables is indexed by the raw db table name - save the unprefixed
    // name for later use
    $tables[$r['TABLE_NAME']]['name'] = $r['NEW_TABLE_NAME'];
  }
  $sql = 'SELECT TABLE_NAME, COLUMN_NAME, INDEX_NAME, SUB_PART, NON_UNIQUE
          FROM information_schema.STATISTICS
          WHERE TABLE_SCHEMA="%s" ';
  if (isset($name)) {
    $sql .= 'AND TABLE_NAME = "%s" ';
  }
  $sql .= 'ORDER BY TABLE_NAME, INDEX_NAME, SEQ_IN_INDEX';
  $res = db_query($sql, $database, $name);
  while ($r = db_fetch_array($res)) {
    if (isset($r['SUB_PART']) && !is_null($r['SUB_PART'])) {
      $col = array(
        $r['COLUMN_NAME'],
        intval($r['SUB_PART']),
      );
    }
    else {
      $col = $r['COLUMN_NAME'];
    }
    if ($r['INDEX_NAME'] == 'PRIMARY') {
      $type = 'primary key';
      $tables[$r['TABLE_NAME']][$type][] = $col;
      continue;
    }
    else {
      if ($r['NON_UNIQUE'] == 0) {
        $type = 'unique keys';
      }
      else {
        $type = 'indexes';
      }
    }
    $tables[$r['TABLE_NAME']][$type][$r['INDEX_NAME']][] = $col;
  }

  // Now, for tables which we have unprefixed, index $tables by the unprefixed name
  foreach ($tables as $tablename => $table) {
    $newname = $tables[$tablename]['name'];
    if ($tablename != $newname) {
      $tables[$newname] = $table;
      unset($tables[$tablename]);
    }
  }
  return $tables;
}