source: trunk/prototype/services/PostgreSQL.php @ 5641

Revision 5641, 9.9 KB checked in by acoutinho, 12 years ago (diff)

Ticket #2434 - Correcao de bugs e melhorias no modulo expressoCalendar

Line 
1<?php
2
3class PostgreSQL implements Service
4{
5    private $con; //Conexão com o banco de dados
6    private $config; //Configuração
7    public  $error = false; //Armazena um erro caso ocorra
8   
9    public function find ( $uri, $justthese = false, $criteria = false ){
10                   
11        $map =  Config::get($uri['concept'], 'PostgreSQL.mapping');
12       
13        $criteria = ($criteria !== false) ? $this->parseCriteria ( $criteria , $map) : '';
14
15        $justthese = self::parseJustthese($justthese, $map);
16
17        return $this->execResultSql( 'SELECT '.$justthese['select'].' FROM '. (Config::get($uri['concept'],'PostgreSQL.concept')) .' '.$criteria );
18    }
19
20   public function read ( $uri, $justthese = false , $criteria = false){
21   
22      $map =  Config::get($uri['concept'], 'PostgreSQL.mapping');   
23      $justthese = self::parseJustthese($justthese, $map);
24      $criteria = ($criteria !== false) ? $this->parseCriteria ( $criteria , $map , ' WHERE id = \''.addslashes( $uri['id'] ).'\'') : ' WHERE id = \''.addslashes( $uri['id'] ).'\'';
25   
26      return $this->execResultSql( 'SELECT '.$justthese['select'].' FROM '. (Config::get($uri['concept'],'PostgreSQL.concept')) .$criteria , true );
27    }
28       
29    public function deleteAll ( $uri,   $justthese = false, $criteria = false ){
30            $map = Config::get($uri['concept'], 'PostgreSQL.mapping');
31            if(!self::parseCriteria ( $criteria , $map)) return false; //Validador para não apagar tabela inteira
32            return $this->execSql( 'DELETE FROM '.(Config::get($uri['concept'],'PostgreSQL.concept')).' '.self::parseCriteria ( $criteria ,$map) );
33    }
34
35    public function delete ( $uri, $justthese = false ){
36            return $this->execSql('DELETE FROM '.(Config::get($uri['concept'],'PostgreSQL.concept')).' WHERE id = '.addslashes($uri['id']));
37    }
38
39    public function replace ( $uri,  $data, $criteria = false ){
40            $map = Config::get($uri['concept'], 'PostgreSQL.mapping');
41            return $this->execSql('UPDATE '.(Config::get($uri['concept'],'PostgreSQL.concept')).' '. self::parseUpdateData( $data ,$map).' '.self::parseCriteria($criteria , $map));
42    }
43               
44    public function update ( $uri,  $data ){
45            $map = Config::get($uri['concept'], 'PostgreSQL.mapping');
46            return $this->execSql('UPDATE '.(Config::get($uri['concept'],'PostgreSQL.concept')).' '. self::parseUpdateData( $data ,$map).' WHERE id = \''. addslashes( $uri['id']) .'\'');
47    }
48
49    public function create ( $uri,  $data ){   
50            return $this->execResultSql( 'INSERT INTO '.(Config::get($uri['concept'],'PostgreSQL.concept')).' '.self::parseInsertData( $data , $uri['concept'] ), true );
51    }
52
53    private function execSql( $sql ){
54            if(!$this->con)     $this->open( $this->config );
55
56            $rs = pg_query( $this->con, $sql );
57
58            if ( !$rs ){
59                    $this->error = pg_last_error ( $this->con );
60                    return false;
61            }
62
63            return $rs;
64    }
65       
66    public function execResultSql( $sql, $unique = false ){
67            $rs = $this->execSql( $sql );
68
69            if( $rs && $rs > 0 )
70                return self::parseSelectResult( $rs , $unique );
71
72            return $rs;
73    }
74
75    public function begin( $uri ) {
76   
77        if(!$this->con)
78            $this->open( $this->config );
79       
80        $this->error = false;
81        pg_query($this->con, "BEGIN WORK");
82    }
83
84    public function commit($uri ) {
85   
86        if( $this->error !== false )
87        {
88            $error = $this->error;
89            $this->error = false;
90
91            throw new Exception( $error );
92        }
93
94        pg_query($this->con, "COMMIT");
95
96        return( true );
97    }
98
99    public function rollback( $uri ){
100   
101        pg_query($this->con, "ROLLBACK");
102    }
103
104    public function open  ( $config ){
105                       
106        $this->config = $config;
107       
108        $rs = '';
109        $rs .= ( isset($this->config['host']) && $this->config['host'] )  ? ' host='.$this->config['host'] : '' ;
110        $rs .= ( isset($this->config['user']) && $this->config['user'] )  ? ' user='.$this->config['user'] : '' ;
111        $rs .= ( isset($this->config['password']) && $this->config['password'] )  ? ' password='.$this->config['password'] : '' ;
112        $rs .= ( isset($this->config['dbname']) && $this->config['dbname'] )  ? ' dbname='.$this->config['dbname'] : '' ;
113        $rs .= ( isset($this->config['port']) && $this->config['port'] )  ? ' port='.$this->config['port'] : '' ;
114               
115            return( $this->con = pg_connect( $rs ) );
116            //$this->con = pg_connect('host='.$config['host'].' user='.$config['user'].' password='.$config['password'].' dbname='.$config['dbname'].'  options=\'--client_encoding=UTF8\'');
117    }
118
119    public function close(){
120
121            pg_close($this->con);
122           
123            $this->con = false;
124
125    }
126
127    public function setup(){}
128
129    public function teardown(){}
130
131    private static function parseInsertData( $data , $concept){
132         
133            $map = Config::get($concept, 'PostgreSQL.mapping');
134       
135            $ind = array();
136            $val = array();
137           
138            foreach ($data as $i => $v){
139                    if(!isset($map[$i])) continue;
140               
141                    $ind[] = $map[$i];
142                    $val[] = '\''.addslashes($v).'\'';
143            }
144           
145            return '('.implode(',', $ind).') VALUES ('.implode(',', $val).') RETURNING id';
146    }
147       
148    private static function parseUpdateData( $data , &$map){
149                                           
150            $d = array();
151            foreach ($data as $i => $v)
152            {
153                if(!isset($map[$i])) continue;
154               
155                $d[] = $map[$i].' = \''.addslashes ($v).'\'';
156            }
157           
158            return 'SET '.implode(',', $d);
159    }
160       
161    private static function parseSelectResult( $result , $unique = false){
162           
163            $return = array();
164           
165            if  (!$result) return false;
166            if (pg_num_rows($result) === 0) return $return;
167            else
168              while( $row = pg_fetch_assoc( $result ))
169                      $return[] = $row;
170           
171            if($unique === true) return $return[0];
172           
173            return $return;
174    }
175       
176    private static function parseCriteria( $criteria  , &$map , $query = '' ){               
177       
178            if( isset($criteria["filter"]) && $criteria["filter"] !== NULL )
179            {
180                    /*
181                  * ex: array   (
182                  *               [0] 'OR',
183                  *               [1] array( 'OR', array( array( '=', 'campo', 'valor' ) ),
184                  *               [2] array( '=', 'campo' , 'valor' ),
185                  *               [3] array( 'IN', 'campo', array( '1' , '2' , '3' ) )
186                  *             )
187                  * OR
188                  *         array( '=' , 'campo' , 'valor' )
189                */
190                $query .= ($query === '') ?  'WHERE ('.self::parseFilter( $criteria['filter'] , $map).')' : ' AND ('.self::parseFilter( $criteria['filter'] , $map).')';
191            }
192            /*
193              * ex: array( 'table1' => 'table2' ,  'table1' => 'table2')
194              *         
195              */
196            if( isset($criteria["join"]) )
197            {
198                foreach ($criteria["join"] as $i => $v)
199                    $query .= ' AND '.$i.' = '.$v.' ';
200            }
201           
202            if( isset($criteria["group"]) )
203            {
204                    $query .= ' GROUP BY '.( is_array($criteria["group"]) ? implode(', ', $criteria["group"]) : $criteria["group"] ).' ';
205            }
206           
207            if( isset($criteria["order"]) )
208            {
209                    $query .= ' ORDER BY '.self::parseOrder( $criteria["order"], $map ).' ';
210            }
211            if( isset($criteria["limit"]) )
212            {
213                    $query .= ' LIMIT '. $criteria["limit"] .' ';
214            }
215            if( isset($criteria["offset"]) )
216            {
217                    $query .= ' OFFSET '. $criteria["offset"] .' ';
218            }
219           
220            return $query;
221    }
222   
223    private static function parseFilter( $filter ,&$map){
224   
225        if( !is_array( $filter ) || count($filter) <= 0) return null;
226               
227        $op = self::parseOperator( array_shift( $filter ) );
228       
229        if( is_array($filter[0]) )
230        {
231            $nested = array();
232
233            foreach( $filter as $i => $f )
234                if( $n = self::parseFilter( $f , $map))
235                    $nested[] = $n;
236
237               
238            return (count($nested) > 0 ) ? '('.implode( ' '.$op.' ', $nested ).')' : '';
239        }
240
241        if(!isset($map[$filter[0]])) return '';
242                 
243        $filter[0] = $map[$filter[0]];
244       
245        $igSuffix = $igPrefix = '';
246               
247        if( strpos( $op[0], 'i' ) === 0 )
248        {
249            $op[0] = substr( $op[0], 1 );
250            $filter[0] = 'upper("'.$filter[0].'")';
251            $igPrefix = 'upper(';
252            $igSuffix = ')';
253        }
254
255        if( is_array($filter[1]) )
256            return( $filter[0].' '.$op[0]." ($igPrefix'".implode( "'$igSuffix,$igPrefix'", array_map("addslashes" , $filter[1]) )."'$igSuffix)" );
257
258        return( $filter[0].' '.$op[0]." $igPrefix'".$op[1].addslashes( $filter[1] ).$op[2]."'$igSuffix" );
259    }
260
261    private static function parseOperator( $op ){
262   
263        switch(strtolower($op))
264        {
265            case 'and':
266            case 'or': return( $op );
267            case 'in': return array( $op );
268            case '^': return array( 'like', '%',  '' );
269            case '$': return array( 'like',  '', '%' );
270            case '*': return array( 'like', '%', '%' );
271            case 'i^': return array( 'ilike', '%',  '' );
272            case 'i$': return array( 'ilike',  '', '%' );
273            case 'i*': return array( 'ilike', '%', '%' );
274            default : return array( $op,  '',  '' );
275        }
276    }
277   
278    private static function parseJustthese($justthese , &$map)
279    {
280                 
281        if(!is_array($justthese)) //Caso seja um full select pegar todas as keys
282            $justthese = array_keys($map);
283
284        $return = array();
285
286        foreach ($justthese as &$value)
287        {
288            if(!isset($map[$value])) continue; //Escapa itens não existentes no mapa
289
290            if(is_array($map[$value]))
291                $return['deepness'][$value] = $map[$value];
292            else
293                $return['select'][] = $map[$value] .' as "'. $value. '"';
294        }
295       
296        $return['select'] = implode(', ', $return['select']);
297        return $return; 
298    }
299
300   private static function parseOrder($order , &$map)
301    {
302                 
303        if($notArray = !is_array($order)) //Caso seja um full select pegar todas as keys
304            $order = array( $order );
305
306        $return = array();
307
308        foreach ($order as &$value)
309        {
310            if(!isset($map[$value])) continue; //Escapa itens não existentes no mapa
311
312            $value = $map[$value];
313        }
314
315        return ( $notArray ?  $order[0] : implode(', ', $order) );
316    }
317}
318
319?>
Note: See TracBrowser for help on using the repository browser.