gnustep/libs-gdl2
0
0
Fork 0
mirror of https://github.com/gnustep/libs-gdl2.git synced 2025-02-23 03:10:55 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
libs-gdl2/EOAdaptors/Postgres95/Postgres95Channel.m
David Ayers 760c71a9b4 * EOAdaptors/Postgres95/Postgres95Channel.m 
([Postgres95Channel -describeTableNames]): Removed incorrect code
	left by copy and paste action.  Reported by Markus Hitter.


git-svn-id: svn+ssh://svn.gna.org/svn/gnustep/libs/gdl2/trunk@17003 72102866-910b-0410-8b05-ffd578937521
2003-06-22 16:36:40 +00:00

1486 lines
42 KiB
Objective-C
Raw Blame History

/**
Postgres95Channel.m <title>Postgres95Channel</title>
Copyright (C) 2000-2002 Free Software Foundation, Inc.
Author: Mirko Viviani <mirko.viviani@rccr.cremona.it>
Date: February 2000
based on the Postgres95 adaptor written by
Mircea Oancea <mircea@jupiter.elcom.pub.ro>
Author: Manuel Guesdon <mguesdon@orange-concept.com>
Date: October 2000
$Revision$
$Date$
<abstract></abstract>
This file is part of the GNUstep Database Library.
<license>
This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Library General Public
License as published by the Free Software Foundation; either
version 2 of the License, or (at your option) any later version.
This library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Library General Public License for more details.
You should have received a copy of the GNU Library General Public
License along with this library; see the file COPYING.LIB.
If not, write to the Free Software Foundation,
59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
</license>
**/
#include "config.h"
RCS_ID("$Id$")
#ifndef NeXT_Foundation_LIBRARY
#include <Foundation/NSArray.h>
#include <Foundation/NSDictionary.h>
#include <Foundation/NSString.h>
#include <Foundation/NSObjCRuntime.h>
#include <Foundation/NSUtilities.h>
#include <Foundation/NSException.h>
#include <Foundation/NSDebug.h>
#else
#include <Foundation/Foundation.h>
#endif
#include <EOControl/EONull.h>
#include <EOControl/EOQualifier.h>
#include <EOControl/EOFetchSpecification.h>
#include <EOControl/EODebug.h>
#include <EOAccess/EOAttribute.h>
#include <EOAccess/EOEntity.h>
#include <EOAccess/EOModel.h>
#include <EOAccess/EOSQLExpression.h>
#include <Postgres95EOAdaptor/Postgres95Channel.h>
#include <Postgres95EOAdaptor/Postgres95Context.h>
#include <Postgres95EOAdaptor/Postgres95Values.h>
static void __dummy_function_used_for_linking(void)
{
extern void __postgres95_values_linking_function(void);
__postgres95_values_linking_function();
__dummy_function_used_for_linking();
}
@implementation Postgres95Channel
- (id) initWithAdaptorContext: (EOAdaptorContext *)adaptorContext
{
if ((self = [super initWithAdaptorContext: adaptorContext]))
{
EOAttribute *attr = nil;
ASSIGN(_adaptorContext, adaptorContext);//TODO NO
//verify
_oidToTypeName = [[NSMutableDictionary alloc] initWithCapacity: 101];
attr = [[EOAttribute alloc] init];
[attr setName: @"nextval"];
[attr setColumnName: @"nextval"];
[attr setValueType: @"i"];
[attr setValueClassName: @"NSNumber"];
ASSIGN(_pkAttributeArray, [NSArray arrayWithObject: attr]);
RELEASE(attr);
}
return self;
}
- (void)dealloc
{
if ([self isOpen])
[self closeChannel];
DESTROY(_adaptorContext);
DESTROY(_sqlExpression);
DESTROY(_oidToTypeName);
DESTROY(_pkAttributeArray);
[super dealloc];
}
- (BOOL)isOpen
{
return (_pgConn ? YES : NO);
}
- (void)openChannel
{
//OK
NSAssert(!_pgConn, @"Channel already opened");
_pgConn = [(Postgres95Adaptor *)[[self adaptorContext] adaptor] newPGconn];
if (_pgConn)
[self _describeDatabaseTypes];
}
- (void)closeChannel
{
NSAssert(_pgConn, @"Channel not opened");
[self _cancelResults];
[(Postgres95Adaptor *)[[self adaptorContext] adaptor] releasePGconn: _pgConn
force: NO];
_pgConn = NULL;
}
- (BOOL)isFetchInProgress
{
return _isFetchInProgress;
}
- (PGconn *)pgConn
{
return _pgConn;
}
- (PGresult *)pgResult
{
return _pgResult;
}
- (void)cancelFetch
{
EOAdaptorContext *adaptorContext = nil;
EOFLOGObjectFnStart();
adaptorContext = [self adaptorContext];
[self cleanupFetch];
//NO ?? [self _cancelResults];//Done in cleanup fetch
// [_adaptorContext autoCommitTransaction];//Done in cleanup fetch
EOFLOGObjectFnStop();
}
- (void)_cancelResults
{
EOFLOGObjectFnStart();
_fetchBlobsOid = NO;
DESTROY(_attributes);
DESTROY(_origAttributes);
if (_pgResult)
{
PQclear(_pgResult);
_pgResult = NULL;
_currentResultRow = -2;
}
_isFetchInProgress = NO;
EOFLOGObjectFnStop();
}
- (BOOL)advanceRow
{
BOOL advanceRow = NO;
// fetch results where read then freed
EOFLOGObjectFnStart();
if (_pgResult)
{
// next row
_currentResultRow++;
// check if result set is finished
if (_currentResultRow >= PQntuples(_pgResult))
{
[self _cancelResults];
}
else
advanceRow = YES;
}
EOFLOGObjectFnStop();
return advanceRow;
}
- (NSArray*)lowLevelResultFieldNames: (PGresult*)res
{
NSMutableArray *names = [NSMutableArray array];
int nb = PQnfields(res);
int i;
for (i = 0; i < nb; i++)
{
char *szName = PQfname(res,i);
NSString *name = [NSString stringWithCString: szName];
[names addObject: name];
}
return names;
}
- (NSMutableDictionary *)fetchRowWithZone: (NSZone *)zone
{
//TODO
/*
//self cleanupFetch quand plus de row !!
valueClassName...externaltype on each attr
self adaptorContext
context adaptor
adaptor databaseEncoding//2
self dictionaryWithObjects:???
forAttributes:_attributes
zone:zone
//end
*/
NSMutableDictionary *dict = nil;
EOFLOGObjectFnStart();
if (_delegateRespondsTo.willFetchRow)
[_delegate adaptorChannelWillFetchRow: self];
NSDebugMLLog(@"gsdb",@"[self isFetchInProgress]: %s",
([self isFetchInProgress] ? "YES" : "NO"));
if ([self isFetchInProgress])
{
NSDebugMLLog(@"gsdb", @"ATTRIBUTES=%@", _attributes);
if (!_attributes)
[self _describeResults];
if ([self advanceRow] == NO)
{
NSDebugMLLog(@"gsdb", @"No Advance Row", "");
// Return nil to indicate that the fetch operation was finished
if (_delegateRespondsTo.didFinishFetching)
[_delegate adaptorChannelDidFinishFetching: self];
[self _cancelResults];
}
else
{
int i;
int count = [_attributes count];
id valueBuffer[100];
id *values = NULL;
EONull *nullValue = (EONull *)[EONull null];
NSDebugMLLog(@"gsdb", @"count=%d", count);
if (count > PQnfields(_pgResult))
{
NSDebugMLog(@"attempt to read %d attributes when the result set has only %d columns",
count, PQnfields(_pgResult));
NSDebugMLog(@"_attributes=%@", _attributes);
NSDebugMLog(@"result=%@", [self lowLevelResultFieldNames:
_pgResult]);
[NSException raise: Postgres95Exception
format: @"attempt to read %d attributes "
@"when the result set has only %d columns",
count, PQnfields(_pgResult)];
}
if (count > 100)
values = (id *)NSZoneMalloc(zone, count * sizeof(id));
else
values = valueBuffer;
for (i = 0; i < count; i++)
{
EOAttribute *attr = [_attributes objectAtIndex: i];
int length = 0;
const char *string = NULL;
// If the column has the NULL value insert EONull in row
if (PQgetisnull(_pgResult, _currentResultRow, i))
{
values[i] = RETAIN(nullValue); //to be compatible with others returned values
}
else
{
string = PQgetvalue(_pgResult, _currentResultRow, i);
length = PQgetlength(_pgResult, _currentResultRow, i);
// if external type for this attribute is "inversion" then this
// column represents an Oid of a large object
if ([[attr externalType] isEqual: @"inversion"])
{
if (!_fetchBlobsOid)
{
string = [self _readBinaryDataRow: (Oid)atol(string)
length:&length zone: zone];
//For efficiency reasons, the returned value is NOT autoreleased !
values[i] = [Postgres95Values newValueForBytes: string
length: length
attribute: attr];
}
else
{
//For efficiency reasons, the returned value is NOT autoreleased !
values[i] = [[NSNumber alloc]
initWithLong: atol(string)];
}
}
else
{
//For efficiency reasons, the returned value is NOT autoreleased !
values[i] = [Postgres95Values newValueForBytes: string
length: length
attribute: attr];
}
}
NSDebugMLLog(@"gsdb", @"value[%d]=%@", i, values[i]);
}
NSDebugMLLog(@"gsdb", @"values count=%d values=%p", count, values);
NSDebugMLLog(@"gsdb", @"_attributes=%@", _attributes);
dict = [self dictionaryWithObjects: values
forAttributes: _attributes
zone: zone];
/* NO: For efficiency reasons, the returned value is NOT autoreleased !
for (i = 0; i < count; i++)
[values[i] release];
*/
if (values != valueBuffer)
NSZoneFree(zone, values);
if (_delegateRespondsTo.didFetchRow)
[_delegate adaptorChannel: self didFetchRow: dict];
}
}
NSDebugMLLog(@"gsdb", @"row: %@", dict);
EOFLOGObjectFnStop();
return dict; //an EOMutableKnownKeyDictionary
}
- (BOOL)_evaluateCommandsUntilAFetch
{
BOOL ret = NO;
ExecStatusType status;
EOFLOGObjectFnStart();
// Check results
status = PQresultStatus(_pgResult);
NSDebugMLLog(@"gsdb",@"status=%d (%s)",
(int)status,
PQresStatus(status));
switch (status)
{
case PGRES_EMPTY_QUERY:
_isFetchInProgress = NO;
ret = YES;
break;
case PGRES_COMMAND_OK:
_isFetchInProgress = NO;
ret = YES;
break;
case PGRES_TUPLES_OK:
_isFetchInProgress = YES;
_currentResultRow = -1;
ret = YES;
break;
case PGRES_COPY_OUT:
_isFetchInProgress = NO;
ret = YES;
break;
case PGRES_COPY_IN:
_isFetchInProgress = NO;
ret = YES;
break;
case PGRES_BAD_RESPONSE:
case PGRES_NONFATAL_ERROR:
case PGRES_FATAL_ERROR:
{
if ([self isDebugEnabled])
NSLog(@"SQL expression '%@' caused %s",
[_sqlExpression statement], PQerrorMessage(_pgConn));
NSDebugMLLog(@"SQL expression '%@' caused %s",
[_sqlExpression statement], PQerrorMessage(_pgConn));
[NSException raise: Postgres95Exception
format: @"unexpected result returned by PQresultStatus()"];
EOFLOGObjectFnStop();
return NO;
}
default:
{
[NSException raise: Postgres95Exception
format: @"unexpected result returned by PQresultStatus():"];
break;
}
}
NSDebugMLLog(@"gsdb", @"ret=%s", (ret ? "YES" : "NO"));
NSDebugMLLog(@"gsdb", @"_isFetchInProgress=%s", (_isFetchInProgress ? "YES" : "NO"));
if (ret == YES)
{
PGnotify *notify = PQnotifies(_pgConn);
const char *insoid = NULL;
if (notify)
{
if (_postgres95DelegateRespondsTo.postgres95Notification)
[_delegate postgres95Channel: self
receivedNotification:
[NSString stringWithCString: notify->relname]];
free(notify);
}
insoid = PQoidStatus(_pgResult);
if (*insoid && _postgres95DelegateRespondsTo.postgres95InsertedRowOid)
{
Oid oid = atol(insoid);
[_delegate postgres95Channel: self insertedRowWithOid: oid];
}
}
NSDebugMLLog(@"gsdb",@"_isFetchInProgress=%s",
(_isFetchInProgress ? "YES" : "NO"));
if ([self isFetchInProgress])// Mirko: TODO remove this !
[self _describeResults];
if ([self isDebugEnabled])
{
NSString *message = [NSString stringWithCString: PQcmdStatus(_pgResult)];
if (status == PGRES_TUPLES_OK)
message = [NSString stringWithFormat:
@"Command status %@. Returned %d rows with %d columns ",
message, PQntuples(_pgResult), PQnfields(_pgResult)];
NSLog (@"Postgres95Adaptor: %@", message);
}
NSDebugMLLog(@"gsdb", @"ret=%s", (ret ? "YES" : "NO"));
EOFLOGObjectFnStop();
return ret;
}
- (BOOL)_evaluateExpression: (EOSQLExpression *)expression
withAttributes: (NSArray*)attributes
{
BOOL result = NO;
EOAdaptorContext *adaptorContext = nil;
EOFLOGObjectFnStart();
NSDebugMLLog(@"gsdb", @"expression=%@", expression);
ASSIGN(_sqlExpression, expression);
ASSIGN(_origAttributes, attributes);
// NSDebugMLLog(@"gsdb",@"EE _origAttributes=%@",_origAttributes);
// NSDebugMLLog(@"gsdb",@"EE _attributes=%@",_attributes);
NSDebugMLLog(@"gsdb", @"Postgres95Adaptor: execute command:\n%@\n",
[expression statement]);
if ([self isDebugEnabled] == YES)
NSLog(@"Postgres95Adaptor: execute command:\n%@\n",
[expression statement]);
//call PostgreSQLChannel numberOfAffectedRows
/* Send the expression to the SQL server */
_pgResult = PQexec(_pgConn, (char *)[[[expression statement] stringByAppendingString:@";"] cString]); // stephane@sente.ch: We need to add ; for INSERT, at least
NSDebugMLLog(@"gsdb", @"_pgResult=%p", (void*)_pgResult);
if (_pgResult == NULL)
{
if ([self isDebugEnabled])
{
adaptorContext = [self adaptorContext];
[(Postgres95Adaptor *)[adaptorContext adaptor]
privateReportError: _pgConn];
}
}
else
{
/* Check command results */
if ([self _evaluateCommandsUntilAFetch] != NO)
result = YES;
}
//self numberOfAffectedRows
NSDebugMLLog(@"gsdb", @"result: %s", (result ? "YES" : "NO"));
// NSDebugMLLog(@"gsdb",@"FF attributes=%@",_attributes);
EOFLOGObjectFnStop();
return result;
}
- (void)evaluateExpression: (EOSQLExpression *)expression // OK quasi
{
Postgres95Context *adaptorContext = nil;
EOFLOGObjectFnStart();
//_evaluationIsDirectCalled=1
adaptorContext = (Postgres95Context *)[self adaptorContext];
//call expression statement
//call adaptorContext adaptor
//call adaptor databaseEncoding
//call self setErrorMessage
//call expre statement
NSDebugMLLog(@"gsdb", @"expression=%@", expression);
if (_delegateRespondsTo.shouldEvaluateExpression)
{
BOOL response
= [_delegate adaptorChannel: self
shouldEvaluateExpression: expression];
if (response == NO)
return;
}
if ([self isOpen] == NO)
[NSException raise: Postgres95Exception
format: @"cannot execute SQL expression. Channel is not opened."];
[self _cancelResults];
[adaptorContext autoBeginTransaction: NO/*YES*/]; //TODO: shouldbe yes ??
if (![self _evaluateExpression: expression
withAttributes: nil])
{
NSDebugMLLog(@"gsdb", @"_evaluateExpression:withAttributes: return NO", "");
[self _cancelResults];
}
else
{
NSDebugMLLog(@"gsdb", @"expression=%@ [self isFetchInProgress]=%d",
expression,
[self isFetchInProgress]);
if (![self isFetchInProgress])//If a fetch is in progress, we don't want to commit because
//it will cancel fetch. I'm not sure it the 'good' way to do
[adaptorContext autoCommitTransaction];
if (_delegateRespondsTo.didEvaluateExpression)
[_delegate adaptorChannel: self didEvaluateExpression: expression];
}
EOFLOGObjectFnStop();
}
- (void)insertRow: (NSDictionary *)row
forEntity: (EOEntity *)entity
{
EOSQLExpression *sqlexpr = nil;
NSMutableDictionary *nrow = nil;
NSEnumerator *enumerator = nil;
NSString *attrName = nil;
Postgres95Context *adaptorContext = nil;
EOFLOGObjectFnStart();
NSDebugMLLog(@"gsdb", @"row=%@", row);
if (![self isOpen])
[NSException raise: NSInternalInconsistencyException
format: @"%@ -- %@ 0x%x: attempt to insert rows with no open channel",
NSStringFromSelector(_cmd),
NSStringFromClass([self class]),
self];
if (!row || !entity)
[NSException raise: NSInvalidArgumentException
format: @"row and entity arguments for insertRow:forEntity:"
@" must not be nil objects"];
if ([self isFetchInProgress])
[NSException raise: NSInternalInconsistencyException
format: @"%@ -- %@ 0x%x: attempt to insert rows with fetch in progress",
NSStringFromSelector(_cmd),
NSStringFromClass([self class]),
self];
/* Before creating the SQL INSERT expression we have to replace in the
row the large objects as Oids and to insert them with the large
object file-like interface */
nrow = AUTORELEASE([row mutableCopy]);
adaptorContext = (Postgres95Context *)[self adaptorContext];
[self _cancelResults]; //No done by WO
NSDebugMLLog(@"gsdb", @"autoBeginTransaction", "");
[adaptorContext autoBeginTransaction: YES];
/*:
row allKeys
allkey sortedArrayUsingSelector:compare:
each key
*/
enumerator = [row keyEnumerator];
while ((attrName = [enumerator nextObject]))
{
EOAttribute *attribute = nil;
NSString *externalType = nil;
id value = nil;
NSDebugMLLog(@"gsdb", @"attrName=%@", attrName);
attribute=[entity attributeNamed: attrName];
NSDebugMLLog(@"gsdb", @"attribute=%@", attribute);
if (!attribute)
return; //???????????
value = [row objectForKey: attrName];
NSDebugMLLog(@"gsdb", @"value=%@", value);
externalType = [attribute externalType];
NSDebugMLLog(@"gsdb", @"externalType=%@", externalType);
/* Insert the binary value into the binaryDataRow dictionary */
if ([externalType isEqual: @"inversion"])
{
id binValue = [nrow objectForKey: attrName];
Oid binOid = [self _insertBinaryData: binValue
forAttribute: attribute];
value = [NSNumber numberWithLong: binOid];
}
else if ([externalType isEqual: @"NSString"]) //??
{
//TODO: database encoding
// [[adaptorContext adaptor] databaseEncoding]
}
[nrow setObject: value
forKey: attrName];
}
NSDebugMLLog(@"gsdb", @"nrow=%@", nrow);
if ([nrow count] > 0)
{
sqlexpr = [[[_adaptorContext adaptor] expressionClass]
insertStatementForRow: nrow
entity: entity];
NSDebugMLLog(@"gsdb", @"sqlexpr=%@", sqlexpr);
if ([self _evaluateExpression: sqlexpr withAttributes: nil] == NO) //call evaluateExpression:
[NSException raise: EOGeneralAdaptorException
format: @"%@ -- %@ 0x%x: cannot insert row for entity '%@'",
NSStringFromSelector(_cmd),
NSStringFromClass([self class]),
self,
[entity name]];
}
[_adaptorContext autoCommitTransaction];
EOFLOGObjectFnStop();
}
- (unsigned)deleteRowsDescribedByQualifier: (EOQualifier *)qualifier
entity: (EOEntity *)entity
{
EOSQLExpression *sqlexpr = nil;
unsigned long rows = 0;
Postgres95Context *adaptorContext;
EOFLOGObjectFnStart();
if (![self isOpen])
[NSException raise: NSInternalInconsistencyException
format: @"%@ -- %@ 0x%x: attempt to delete rows with no open channel",
NSStringFromSelector(_cmd),
NSStringFromClass([self class]),
self];
if (!qualifier || !entity)
[NSException raise: NSInvalidArgumentException
format: @"%@ -- %@ 0x%x: qualifier and entity arguments "
@" must not be nil objects",
NSStringFromSelector(_cmd),
NSStringFromClass([self class]),
self];
if ([self isFetchInProgress])
[NSException raise: NSInternalInconsistencyException
format: @"%@ -- %@ 0x%x: attempt to delete rows with fetch in progress",
NSStringFromSelector(_cmd),
NSStringFromClass([self class]),
self];
adaptorContext = (Postgres95Context *)[self adaptorContext];
[self _cancelResults];
[_adaptorContext autoBeginTransaction: NO];
sqlexpr = [[[_adaptorContext adaptor] expressionClass]
deleteStatementWithQualifier: qualifier
entity: entity];
if ([self _evaluateExpression: sqlexpr withAttributes: nil])
rows = strtoul(PQcmdTuples(_pgResult), NULL, 10);
[adaptorContext autoCommitTransaction];
EOFLOGObjectFnStop();
return rows;
}
- (void)selectAttributes: (NSArray *)attributes
fetchSpecification: (EOFetchSpecification *)fetchSpecification
lock: (BOOL)flag
entity: (EOEntity *)entity
{
EOSQLExpression *sqlExpr = nil;
//objectForKey:EOAdaptorQuotesExternalNames ret: nil
//lastObject ret NSRegistrationDomain
//objectForKey:NSRegistrationDomain ret dict
//objectForKey:EOAdaptorQuotesExternalNames
//attr count
//PostgreSQLExpression initWithEntity:
//setUseliases:YES
//prepareSelectExpressionWithAttributes:lock:fetchSpecification:
//statement
//adaptorContext
//a con autoBeginTransaction
//end
EOFLOGObjectFnStart();
NSDebugMLog(@"TEST attributesToFetch=%@", [entity attributesToFetch]);
NSDebugMLLog(@"gsdb",@"%@ -- %@ 0x%x: isFetchInProgress=%s",
NSStringFromSelector(_cmd),
NSStringFromClass([self class]),
self,
([self isFetchInProgress] ? "YES" : "NO"));
if (![self isOpen])
[NSException raise: NSInternalInconsistencyException
format: @"%@ -- %@ 0x%x: attempt to select attributes with no open channel",
NSStringFromSelector(_cmd),
NSStringFromClass([self class]),
self];
if ([self isFetchInProgress])
[NSException raise: NSInternalInconsistencyException
format: @"%@ -- %@ 0x%x: attempt to select attributes with fetch in progress",
NSStringFromSelector(_cmd),
NSStringFromClass([self class]),
self];
if (_delegateRespondsTo.shouldSelectAttributes)
if (![_delegate adaptorChannel: self
shouldSelectAttributes: attributes
fetchSpecification: fetchSpecification
lock: flag
entity: entity])
return;
NSDebugMLLog(@"gsdb", @"%@ -- %@ 0x%x: isFetchInProgress=%s",
NSStringFromSelector(_cmd),
NSStringFromClass([self class]),
self,
([self isFetchInProgress] ? "YES" : "NO"));
[self _cancelResults];
NSDebugMLLog(@"gsdb", @"%@ -- %@ 0x%x: isFetchInProgress=%s",
NSStringFromSelector(_cmd),
NSStringFromClass([self class]),
self,
([self isFetchInProgress] ? "YES" : "NO"));
[_adaptorContext autoBeginTransaction: NO];
ASSIGN(_attributes, attributes);
// NSDebugMLLog(@"gsdb",@"00 attributes=%@",_attributes);
NSAssert([attributes count] > 0, @"No Attributes");
sqlExpr = [[[_adaptorContext adaptor] expressionClass]
selectStatementForAttributes: attributes
lock: flag
fetchSpecification: fetchSpecification
entity: entity];
NSDebugMLLog(@"gsdb", @"sqlExpr=%@", sqlExpr);
// NSDebugMLLog(@"gsdb",@"AA attributes=%@",_attributes);
[self _evaluateExpression: sqlExpr
withAttributes: attributes];
NSDebugMLLog(@"gsdb", @"After _evaluate", "");
// NSDebugMLLog(@"gsdb",@"BB attributes=%@",_attributes);
[_adaptorContext autoCommitTransaction];
NSDebugMLLog(@"gsdb", @"After autoCommitTransaction", "");
if (_delegateRespondsTo.didSelectAttributes)
[_delegate adaptorChannel: self
didSelectAttributes: attributes
fetchSpecification: fetchSpecification
lock: flag
entity: entity];
// NSDebugMLLog(@"gsdb",@"CC attributes=%@",_attributes);
EOFLOGObjectFnStop();
}
- (unsigned int)updateValues: (NSDictionary *)values
inRowsDescribedByQualifier: (EOQualifier *)qualifier
entity: (EOEntity *)entity
{
//autoBeginTransaction
//entity attributes
//externaltype on each attr
//adaptor expressionClass
//exprclass alloc initwithentity
//expr setUseAliases:NO
//exp prepareUpdateExpressionWithRow:qualifier:
//self evaluateExpression:
//autoCommitTransaction
//return number of affeted rows
//end
EOSQLExpression *sqlExpr = nil;
NSMutableDictionary *mrow = nil;
NSMutableArray *invAttributes = nil;
NSEnumerator *enumerator = nil;
NSString *attrName = nil;
NSString *externalType = nil;
EOAttribute *attr = nil;
Postgres95Context *adaptorContext = nil;
unsigned long rows = 0;
EOFLOGObjectFnStart();
if (![self isOpen])
[NSException raise: NSInternalInconsistencyException
format: @"%@ -- %@ 0x%x: attempt to update values with no open channel",
NSStringFromSelector(_cmd),
NSStringFromClass([self class]),
self];
if ([self isFetchInProgress])
[NSException raise: NSInternalInconsistencyException
format: @"%@ -- %@ 0x%x: attempt to update values with fetch in progress",
NSStringFromSelector(_cmd),
NSStringFromClass([self class]),
self];
if ([values count] > 0)
{
mrow = AUTORELEASE([values mutableCopyWithZone: [values zone]]);
// Get EOAttributes involved in update operation
// Modify "inversion" attributes to NSNumber type with the Oid
invAttributes = AUTORELEASE([[NSMutableArray alloc] initWithCapacity: [mrow count]]);
enumerator = [values keyEnumerator];
while ((attrName = [enumerator nextObject]))
{
attr = [entity attributeNamed: attrName];
externalType = [attr externalType];
if (attr == nil)
return 0; //???
/*
[mrow setObject:[attr adaptorValueByConvertingAttributeValue://Not in WO
[values objectForKey:attrName]]
forKey:attrName];
*/
[mrow setObject:[values objectForKey: attrName]
forKey: attrName];
if ([externalType isEqual: @"inversion"])
[invAttributes addObject: attr];
}
[self _cancelResults]; //Not in WO
adaptorContext = (Postgres95Context *)[self adaptorContext];
[adaptorContext autoBeginTransaction: YES];
if ([invAttributes count])
{
// Select with update qualifier to see there is only one row
// to be updated and to get the large objects (to be updatetd)
// Oid from dataserver - there is a hack here based on the fact that
// in update there in only one table and no flattened attributes
NSDictionary *dbRow = nil;
sqlExpr = [[[_adaptorContext adaptor] expressionClass]
selectStatementForAttributes: invAttributes
lock: NO
fetchSpecification:
[EOFetchSpecification
fetchSpecificationWithEntityName: [entity name]
qualifier: qualifier
sortOrderings: nil]
entity: entity];
[self _evaluateExpression: sqlExpr withAttributes: nil];
_fetchBlobsOid = YES;
dbRow = [self fetchRowWithZone: NULL];
_fetchBlobsOid = NO;
[self _cancelResults];
// Update the large objects and modify the row to update with Oid's
enumerator = [invAttributes objectEnumerator];
while ((attr = [enumerator nextObject]))
{
Oid oldOid;
Oid newOid;
NSData *data;
attrName = [attr name];
data = [mrow objectForKey: attrName];
oldOid = [[dbRow objectForKey:attrName] longValue];
newOid = [self _updateBinaryDataRow: oldOid data: data];
[mrow setObject: [NSNumber numberWithUnsignedLong: newOid]
forKey: attrName];
}
}
// Now we have all: one and only row to update and binary rows
// (large objects) where updated and their new Oid set in the row
rows = 0;
NSDebugMLLog(@"gsdb", @"[mrow count]=%d", [mrow count]);
if ([mrow count] > 0)
{
sqlExpr = [[[_adaptorContext adaptor] expressionClass]
updateStatementForRow: mrow
qualifier: qualifier
entity: entity];
//wo call evaluateExpression:
if ([self _evaluateExpression: sqlExpr withAttributes: nil])
rows = strtoul(PQcmdTuples(_pgResult), NULL, 10);
}
[adaptorContext autoCommitTransaction];
}
EOFLOGObjectFnStop();
return rows;
}
/* The binaryDataRow should contain only one binary attribute */
- (char *)_readBinaryDataRow: (Oid)oid
length: (int *)length
zone: (NSZone *)zone;
{
int fd;
int len, wrt;
char *bytes;
if (oid == 0)
{
*length = 0;
return NULL;
}
fd = lo_open(_pgConn, oid, INV_READ|INV_WRITE);
if (fd < 0)
[NSException raise: Postgres95Exception
format: @"cannot open large object Oid = %ld", oid];
lo_lseek(_pgConn, fd, 0, SEEK_END);
len = lo_tell(_pgConn, fd);
lo_lseek(_pgConn, fd, 0, SEEK_SET);
if (len < 0)
[NSException raise: Postgres95Exception
format: @"error while getting size of large object Oid = %ld", oid];
bytes = NSZoneMalloc(zone, len);
wrt = lo_read(_pgConn, fd, bytes, len);
if (len != wrt)
{
NSZoneFree(zone, bytes);
[NSException raise: Postgres95Exception
format: @"error while reading large object Oid = %ld", oid];
}
lo_close(_pgConn, fd);
*length = len;
return bytes;
}
- (Oid)_insertBinaryData: (NSData *)binaryData
forAttribute: (EOAttribute *)attr
{
int len;
const char* bytes;
Oid oid;
int fd, wrt;
if ((id)binaryData == [EONull null] || binaryData == nil)
return 0;
len = [binaryData length];
bytes = [binaryData bytes];
oid = lo_creat(_pgConn, INV_READ|INV_WRITE);
if (oid == 0)
[NSException raise: Postgres95Exception
format: @"cannot create large object"];
fd = lo_open(_pgConn, oid, INV_READ|INV_WRITE);
if (fd < 0)
[NSException raise: Postgres95Exception
format: @"cannot open large object Oid = %ld", oid];
wrt = lo_write(_pgConn, fd, (char *)bytes, len);
if (len != wrt)
[NSException raise: Postgres95Exception
format: @"error while writing large object Oid = %ld", oid];
lo_close(_pgConn, fd);
return oid;
}
- (Oid)_updateBinaryDataRow: (Oid)oid
data: (NSData *)binaryData
{
int len;
const char* bytes;
int wrt, fd;
if (oid)
lo_unlink(_pgConn, oid);
if ((id)binaryData == [EONull null] || binaryData == nil)
return 0;
len = [binaryData length];
bytes = [binaryData bytes];
oid = lo_creat(_pgConn, INV_READ|INV_WRITE);
if (oid == 0)
[NSException raise: Postgres95Exception
format: @"cannot create large object"];
fd = lo_open(_pgConn, oid, INV_READ|INV_WRITE);
if (fd < 0)
[NSException raise: Postgres95Exception
format: @"cannot open large object Oid = %ld", oid];
wrt = lo_write(_pgConn, fd, (char*)bytes, len);
if (len != wrt)
[NSException raise: Postgres95Exception
format: @"error while writing large object Oid = %ld", oid];
lo_close(_pgConn, fd);
return oid;
}
/* Read type oid and names from the database server.
Called on each openChannel to refresh info. */
- (void)_describeDatabaseTypes
{
int i, count;
_pgResult = PQexec(_pgConn,
"SELECT oid, typname FROM pg_type WHERE typrelid = 0");
if (_pgResult == NULL || PQresultStatus(_pgResult) != PGRES_TUPLES_OK)
{
_pgResult = NULL;
[NSException raise: Postgres95Exception
format: @"cannot read type name informations from database. "
@"bad response from server"];
}
if (PQnfields(_pgResult) != 2)
{
_pgResult = NULL;
[NSException raise: Postgres95Exception
format: @"cannot read type name informations from database. "
@"results should have two columns"];
}
[_oidToTypeName removeAllObjects];
count = PQntuples(_pgResult);
for (i = 0; i < count; i++)
{
char* oid = PQgetvalue(_pgResult, i, 0);
char* typ = PQgetvalue(_pgResult, i, 1);
[_oidToTypeName setObject: [NSString stringWithCString: typ]
forKey: [NSNumber numberWithLong: atol(oid)]];
}
PQclear(_pgResult);
_pgResult = NULL;
}
- (NSArray *)attributesToFetch
{
return _attributes;
}
- (void)setAttributesToFetch: (NSArray *)attributes
{
//call adaptorContext
NSDebugMLLog(@"gsdb", @"Postgres95Channel: setAttributesToFetch %p:%@",
attributes, attributes);
ASSIGN(_attributes, attributes);
}
- (NSArray *)describeResults
{
NSArray *desc;
EOFLOGObjectFnStart();
if (![self isFetchInProgress])
[NSException raise: NSInternalInconsistencyException
format: @"%@ -- %@ 0x%x: attempt to describe results with no fetch in progress",
NSStringFromSelector(_cmd),
NSStringFromClass([self class]),
self];
desc = [self attributesToFetch];
EOFLOGObjectFnStop();
return desc;
}
- (void)_describeResults
{
int colsNumber;
EOFLOGObjectFnStart();
colsNumber=_pgResult ? PQnfields(_pgResult): 0;
NSDebugMLLog(@"gsdb", @"colsNumber=%d", colsNumber);
if (colsNumber == 0)
{
[self setAttributesToFetch: [NSArray array]];
}
else if (!_attributes) //??
{
int i;
id *attributes = NULL;
attributes = alloca(colsNumber * sizeof(id));
for (i = 0; i < colsNumber; i++)
{
EOAttribute *attribute = AUTORELEASE([EOAttribute new]);
NSString *externalName;
NSString *valueClass = @"NSString";
NSString *valueType = nil;
NSDebugMLog(@"TEST attributesToFetch=%@",
[[attribute entity] attributesToFetch]);
if (_origAttributes)
{
EOAttribute *origAttr = (EOAttribute *)[_origAttributes
objectAtIndex: i];
[attribute setName: [origAttr name]];
[attribute setColumnName: [origAttr columnName]];
[attribute setExternalType: [origAttr externalType]];
[attribute setValueType: [origAttr valueType]];
[attribute setValueClassName: [origAttr valueClassName]];
}
else
{
externalName = [_oidToTypeName
objectForKey: [NSNumber
numberWithLong: PQftype(_pgResult, i)]];
if (!externalName)
[NSException raise: Postgres95Exception
format: @"cannot find type for Oid = %d",
PQftype(_pgResult, i)];
[attribute setName: [NSString stringWithFormat: @"attribute%d", i]];
[attribute setColumnName: @"unknown"];
[attribute setExternalType: externalName];
if ([externalName isEqual: @"bool"])
valueClass = @"NSNumber", valueType = @"c";
else if ([externalName isEqual: @"char"])
valueClass = @"NSNumber", valueType = @"c";
else if ([externalName isEqual: @"dt"])
valueClass = @"NSCalendarDate", valueType = nil;
else if ([externalName isEqual: @"date"])
valueClass = @"NSCalendarDate", valueType = nil;
else if ([externalName isEqual: @"time"])
valueClass = @"NSCalendarDate", valueType = nil;
else if ([externalName isEqual: @"float4"])
valueClass = @"NSNumber", valueType = @"f";
else if ([externalName isEqual: @"float8"])
valueClass = @"NSNumber", valueType = @"d";
else if ([externalName isEqual: @"int2"])
valueClass = @"NSNumber", valueType = @"i";
else if ([externalName isEqual: @"int4"])
valueClass = @"NSNumber", valueType = @"i";
else if ([externalName isEqual: @"int8"])
valueClass = @"NSNumber", valueType = @"l";
else if ([externalName isEqual: @"oid"])
valueClass = @"NSNumber", valueType = @"l";
else if ([externalName isEqual: @"varchar"])
valueClass = @"NSString", valueType = nil;
else if ([externalName isEqual: @"bpchar"])
valueClass = @"NSString", valueType = nil;
else if ([externalName isEqual: @"text"])
valueClass = @"NSString", valueType = nil;
/* else if ([externalName isEqual:@"cid"])
valueClass = @"NSNumber", valueType = @"";
else if ([externalName isEqual:@"tid"])
valueClass = @"NSNumber", valueType = @"";
else if ([externalName isEqual:@"xid"])
valueClass = @"NSNumber", valueType = @"";*/
[attribute setValueType: valueType];
[attribute setValueClassName: valueClass];
}
attributes[i] = attribute;
NSDebugMLog(@"TEST attributesToFetch=%@",
[[attribute entity] attributesToFetch]);
}
[self setAttributesToFetch: AUTORELEASE([[NSArray alloc]
initWithObjects: attributes
count: colsNumber])];
}
// NSDebugMLLog(@"gsdb",@"_attributes=%@",_attributes);
EOFLOGObjectFnStop();
}
/* The methods used to generate an model from the meta-information kept by
the database. */
- (NSArray *)describeTableNames
{
int i, count;
NSMutableArray *results = [NSMutableArray array];
NSAssert(_pgConn, @"Channel not opened");
_pgResult = PQexec(_pgConn,
"SELECT tablename FROM pg_tables WHERE pg_tables.schemaname = 'public'");
if (_pgResult == NULL
|| PQresultStatus(_pgResult) != PGRES_TUPLES_OK)
{
_pgResult = NULL;
[NSException raise: Postgres95Exception
format: @"cannot read list of tables from database. "
@"bad response from server"];
}
count = PQntuples(_pgResult);
for (i = 0; i < count; i++)
{
char *oid = PQgetvalue(_pgResult, i, 0);
[results addObject: [NSString stringWithUTF8String: oid]];
}
PQclear(_pgResult);
_pgResult = NULL;
return [NSArray arrayWithArray: results];
}
- (NSArray *)describeStoredProcedureNames
{
// TODO
[self notImplemented: _cmd];
return nil;
}
- (EOModel *)describeModelWithTableNames: (NSArray *)tableNames
{
NSEnumerator *tableEnum;
NSString *table;
EOModel *model;
EOEntity *entity;
model = AUTORELEASE([[EOModel alloc] init]);
tableEnum = [tableNames objectEnumerator];
while ((table = [tableEnum nextObject]))
{
entity = AUTORELEASE([[EOEntity alloc] init]);
[entity setExternalName: table];
[model addEntity: entity];
}
return model; //TODO
}
- (void)setDelegate:delegate
{
[super setDelegate: delegate];
_postgres95DelegateRespondsTo.postgres95InsertedRowOid =
[delegate respondsToSelector:
@selector(postgres95Channel:insertedRowWithOid:)];
_postgres95DelegateRespondsTo.postgres95Notification =
[delegate respondsToSelector:
@selector(postgres95Channel:receivedNotification:)];
}
- (NSDictionary *)primaryKeyForNewRowWithEntity:(EOEntity *)entity
{
//entity primaryKeyAttributes
//self adaptorContext
//on each attr attr: adaptorValueType
//entty externalName
//context autoBeginTransaction
//self cleanupFetch######
//attr name
//dictionary with...
NSDictionary *pk = nil;
NSString *sqlString;
NSString *key = nil;
NSNumber *pkValue = nil;
const char *string = NULL;
int length = 0;
NSString *primaryKeySequenceNameFormat;
NSString *sequenceName;
EOFLOGObjectFnStart();
primaryKeySequenceNameFormat = [(Postgres95Context*)[self adaptorContext]
primaryKeySequenceNameFormat];
NSAssert(primaryKeySequenceNameFormat, @"No primary sequence name format");
sequenceName = [NSString stringWithFormat: primaryKeySequenceNameFormat,
[entity externalName]];
sqlString = [NSString stringWithFormat: @"SELECT nextval('%@')",
sequenceName];
[self _cancelResults];
[_adaptorContext autoBeginTransaction: NO];
[self _evaluateExpression: [EOSQLExpression expressionForString:sqlString]
withAttributes: _pkAttributeArray];
if ([self isFetchInProgress] == NO
|| [self advanceRow] == NO)
{
[self _cancelResults];
[_adaptorContext autoCommitTransaction];
}
else
{
string = PQgetvalue(_pgResult, _currentResultRow, 0);
length = PQgetlength(_pgResult, _currentResultRow, 0);
pkValue = AUTORELEASE([Postgres95Values newValueForBytes: string
length: length
attribute: [_pkAttributeArray
objectAtIndex: 0]]);
NSAssert(pkValue, @"no pk value");
key = [[entity primaryKeyAttributeNames] objectAtIndex: 0];
NSAssert(key, @"pk key");
[self _cancelResults];
[_adaptorContext autoCommitTransaction];
pk = [NSDictionary dictionaryWithObject: pkValue
forKey: key];
}
EOFLOGObjectFnStop();
return pk;
}
- (void)cleanupFetch
{
Postgres95Context *adaptorContext;
EOFLOGObjectFnStart();
adaptorContext = (Postgres95Context *)[self adaptorContext];
NSDebugMLog(@"[self isFetchInProgress]=%s",
([self isFetchInProgress] ? "YES" : "NO"));
if ([self isFetchInProgress])
{
BOOL ok;
[self _cancelResults];
ok = [adaptorContext autoCommitTransaction];
//_isTransactionstarted to 0
//_evaluationIsDirectCalled=0
}
EOFLOGObjectFnStop();
}
@end /* Postgres95Channel */
Reference in a new issue View git blame Copy permalink