includes/libs/rdbms/field/PostgresField.php

   1 <?php
   2
   3 namespace Wikimedia\Rdbms;
   4
   5 class PostgresField implements Field {
   6         private $name, $tablename, $type, $nullable, $max_length, $deferred, $deferrable, $conname,
   7                 $has_default, $default;
   8
   9         /**
  10          * @param DatabasePostgres $db
  11          * @param string $table
  12          * @param string $field
  13          * @return null|PostgresField
  14          */
  15         static function fromText( DatabasePostgres $db, $table, $field ) {
  16                 $q = <<<SQL
  17 SELECT
  18  attnotnull, attlen, conname AS conname,
  19  atthasdef,
  20  adsrc,
  21  COALESCE(condeferred, 'f') AS deferred,
  22  COALESCE(condeferrable, 'f') AS deferrable,
  23  CASE WHEN typname = 'int2' THEN 'smallint'
  24   WHEN typname = 'int4' THEN 'integer'
  25   WHEN typname = 'int8' THEN 'bigint'
  26   WHEN typname = 'bpchar' THEN 'char'
  27  ELSE typname END AS typname
  28 FROM pg_class c
  29 JOIN pg_namespace n ON (n.oid = c.relnamespace)
  30 JOIN pg_attribute a ON (a.attrelid = c.oid)
  31 JOIN pg_type t ON (t.oid = a.atttypid)
  32 LEFT JOIN pg_constraint o ON (o.conrelid = c.oid AND a.attnum = ANY(o.conkey) AND o.contype = 'f')
  33 LEFT JOIN pg_attrdef d on c.oid=d.adrelid and a.attnum=d.adnum
  34 WHERE relkind = 'r'
  35 AND nspname=%s
  36 AND relname=%s
  37 AND attname=%s;
  38 SQL;
  39
  40                 $table = $db->remappedTableName( $table );
  41                 foreach ( $db->getCoreSchemas() as $schema ) {
  42                         $res = $db->query(
  43                                 sprintf( $q,
  44                                         $db->addQuotes( $schema ),
  45                                         $db->addQuotes( $table ),
  46                                         $db->addQuotes( $field )
  47                                 )
  48                         );
  49                         $row = $db->fetchObject( $res );
  50                         if ( !$row ) {
  51                                 continue;
  52                         }
  53                         $n = new PostgresField;
  54                         $n->type = $row->typname;
  55                         $n->nullable = ( $row->attnotnull == 'f' );
  56                         $n->name = $field;
  57                         $n->tablename = $table;
  58                         $n->max_length = $row->attlen;
  59                         $n->deferrable = ( $row->deferrable == 't' );
  60                         $n->deferred = ( $row->deferred == 't' );
  61                         $n->conname = $row->conname;
  62                         $n->has_default = ( $row->atthasdef === 't' );
  63                         $n->default = $row->adsrc;
  64
  65                         return $n;
  66                 }
  67
  68                 return null;
  69         }
  70
  71         function name() {
  72                 return $this->name;
  73         }
  74
  75         function tableName() {
  76                 return $this->tablename;
  77         }
  78
  79         function type() {
  80                 return $this->type;
  81         }
  82
  83         function isNullable() {
  84                 return $this->nullable;
  85         }
  86
  87         function maxLength() {
  88                 return $this->max_length;
  89         }
  90
  91         function is_deferrable() {
  92                 return $this->deferrable;
  93         }
  94
  95         function is_deferred() {
  96                 return $this->deferred;
  97         }
  98
  99         function conname() {
 100                 return $this->conname;
 101         }
 102
 103         /**
 104          * @since 1.19
 105          * @return bool|mixed
 106          */
 107         function defaultValue() {
 108                 if ( $this->has_default ) {
 109                         return $this->default;
 110                 } else {
 111                         return false;
 112                 }
 113         }
 114 }