示例#1
0
        protected async Task <object> PerformInsertAsync(object ownerId, IPersistentCollection collection, IExpectation expectation,
                                                         object entry, int index, bool useBatch, bool callable, ISessionImplementor session, CancellationToken cancellationToken)
        {
            cancellationToken.ThrowIfCancellationRequested();
            object entryId = null;
            int    offset  = 0;
            var    st      = useBatch
                                ? await(session.Batcher.PrepareBatchCommandAsync(SqlInsertRowString.CommandType, SqlInsertRowString.Text, SqlInsertRowString.ParameterTypes, cancellationToken)).ConfigureAwait(false)
                                : await(session.Batcher.PrepareCommandAsync(SqlInsertRowString.CommandType, SqlInsertRowString.Text, SqlInsertRowString.ParameterTypes, cancellationToken)).ConfigureAwait(false);

            try
            {
                //offset += expectation.Prepare(st, factory.ConnectionProvider.Driver);
                offset = await(WriteKeyAsync(st, ownerId, offset, session, cancellationToken)).ConfigureAwait(false);
                if (hasIdentifier)
                {
                    entryId = collection.GetIdentifier(entry, index);
                    offset  = await(WriteIdentifierAsync(st, entryId, offset, session, cancellationToken)).ConfigureAwait(false);
                }
                if (hasIndex)
                {
                    offset = await(WriteIndexAsync(st, collection.GetIndex(entry, index, this), offset, session, cancellationToken)).ConfigureAwait(false);
                }
                await(WriteElementAsync(st, collection.GetElement(entry), offset, session, cancellationToken)).ConfigureAwait(false);
                if (useBatch)
                {
                    await(session.Batcher.AddToBatchAsync(expectation, cancellationToken)).ConfigureAwait(false);
                }
                else
                {
                    expectation.VerifyOutcomeNonBatched(await(session.Batcher.ExecuteNonQueryAsync(st, cancellationToken)).ConfigureAwait(false), st);
                }
            }
            catch (OperationCanceledException) { throw; }
            catch (Exception e)
            {
                if (useBatch)
                {
                    session.Batcher.AbortBatch(e);
                }
                throw;
            }
            finally
            {
                if (!useBatch)
                {
                    session.Batcher.CloseCommand(st, null);
                }
            }
            return(entryId);
        }
        protected override async Task <int> DoUpdateRowsAsync(object id, IPersistentCollection collection, ISessionImplementor session, CancellationToken cancellationToken)
        {
            cancellationToken.ThrowIfCancellationRequested();
            if (ArrayHelper.IsAllFalse(elementColumnIsSettable))
            {
                return(0);
            }

            try
            {
                DbCommand    st          = null;
                IExpectation expectation = Expectations.AppropriateExpectation(UpdateCheckStyle);
                //bool callable = UpdateCallable;
                bool        useBatch = expectation.CanBeBatched;
                IEnumerable entries  = collection.Entries(this);
                int         i        = 0;
                int         count    = 0;
                foreach (object entry in entries)
                {
                    if (await(collection.NeedsUpdatingAsync(entry, i, ElementType, cancellationToken)).ConfigureAwait(false))
                    {
                        int offset = 0;
                        if (useBatch)
                        {
                            if (st == null)
                            {
                                st =
                                    await(session.Batcher.PrepareBatchCommandAsync(SqlUpdateRowString.CommandType, SqlUpdateRowString.Text,
                                                                                   SqlUpdateRowString.ParameterTypes, cancellationToken)).ConfigureAwait(false);
                            }
                        }
                        else
                        {
                            st =
                                await(session.Batcher.PrepareCommandAsync(SqlUpdateRowString.CommandType, SqlUpdateRowString.Text,
                                                                          SqlUpdateRowString.ParameterTypes, cancellationToken)).ConfigureAwait(false);
                        }

                        try
                        {
                            //offset += expectation.Prepare(st, Factory.ConnectionProvider.Driver);

                            int loc = await(WriteElementAsync(st, collection.GetElement(entry), offset, session, cancellationToken)).ConfigureAwait(false);
                            if (hasIdentifier)
                            {
                                await(WriteIdentifierAsync(st, collection.GetIdentifier(entry, i), loc, session, cancellationToken)).ConfigureAwait(false);
                            }
                            else
                            {
                                loc = await(WriteKeyAsync(st, id, loc, session, cancellationToken)).ConfigureAwait(false);
                                if (HasIndex && !indexContainsFormula)
                                {
                                    await(WriteIndexToWhereAsync(st, collection.GetIndex(entry, i, this), loc, session, cancellationToken)).ConfigureAwait(false);
                                }
                                else
                                {
                                    await(WriteElementToWhereAsync(st, collection.GetSnapshotElement(entry, i), loc, session, cancellationToken)).ConfigureAwait(false);
                                }
                            }

                            if (useBatch)
                            {
                                await(session.Batcher.AddToBatchAsync(expectation, cancellationToken)).ConfigureAwait(false);
                            }
                            else
                            {
                                expectation.VerifyOutcomeNonBatched(await(session.Batcher.ExecuteNonQueryAsync(st, cancellationToken)).ConfigureAwait(false), st);
                            }
                        }

                        catch (OperationCanceledException) { throw; }
                        catch (Exception e)
                        {
                            if (useBatch)
                            {
                                session.Batcher.AbortBatch(e);
                            }
                            throw;
                        }
                        finally
                        {
                            if (!useBatch)
                            {
                                session.Batcher.CloseCommand(st, null);
                            }
                        }
                        count++;
                    }
                    i++;
                }
                return(count);
            }
            catch (DbException sqle)
            {
                throw ADOExceptionHelper.Convert(SQLExceptionConverter, sqle,
                                                 "could not update collection rows: " + MessageHelper.CollectionInfoString(this, collection, id, session),
                                                 SqlUpdateRowString.Text);
            }
        }
		protected object PerformInsert(object ownerId, IPersistentCollection collection, IExpectation expectation,
		                               object entry, int index, bool useBatch, bool callable, ISessionImplementor session)
		{
			object entryId = null;
			int offset = 0;
			IDbCommand st = useBatch
			                	? session.Batcher.PrepareBatchCommand(SqlInsertRowString.CommandType, SqlInsertRowString.Text,
			                	                                      SqlInsertRowString.ParameterTypes)
			                	: session.Batcher.PrepareCommand(SqlInsertRowString.CommandType, SqlInsertRowString.Text,
			                	                                 SqlInsertRowString.ParameterTypes);
			try
			{
				//offset += expectation.Prepare(st, factory.ConnectionProvider.Driver);
				offset = WriteKey(st, ownerId, offset, session);
				if (hasIdentifier)
				{
					entryId = collection.GetIdentifier(entry, index);
					offset = WriteIdentifier(st, entryId, offset, session);
				}
				if (hasIndex)
				{
					offset = WriteIndex(st, collection.GetIndex(entry, index, this), offset, session);
				}
				WriteElement(st, collection.GetElement(entry), offset, session);
				if (useBatch)
				{
					session.Batcher.AddToBatch(expectation);
				}
				else
				{
					expectation.VerifyOutcomeNonBatched(session.Batcher.ExecuteNonQuery(st), st);
				}
			}
			catch (Exception e)
			{
				if (useBatch)
				{
					session.Batcher.AbortBatch(e);
				}
				throw;
			}
			finally
			{
				if (!useBatch)
				{
					session.Batcher.CloseCommand(st, null);
				}
			}
			return entryId;
		}
        protected override int DoUpdateRows(object id, IPersistentCollection collection, ISessionImplementor session)
        {
            if (ArrayHelper.IsAllFalse(elementColumnIsSettable))
            {
                return(0);
            }

            try
            {
                DbCommand    st          = null;
                IExpectation expectation = Expectations.AppropriateExpectation(UpdateCheckStyle);
                //bool callable = UpdateCallable;
                bool        useBatch = expectation.CanBeBatched;
                IEnumerable entries  = collection.Entries(this);
                int         i        = 0;
                int         count    = 0;
                foreach (object entry in entries)
                {
                    if (collection.NeedsUpdating(entry, i, ElementType))
                    {
                        int offset = 0;
                        if (useBatch)
                        {
                            if (st == null)
                            {
                                st =
                                    session.Batcher.PrepareBatchCommand(SqlUpdateRowString.CommandType, SqlUpdateRowString.Text,
                                                                        SqlUpdateRowString.ParameterTypes);
                            }
                        }
                        else
                        {
                            st =
                                session.Batcher.PrepareCommand(SqlUpdateRowString.CommandType, SqlUpdateRowString.Text,
                                                               SqlUpdateRowString.ParameterTypes);
                        }

                        try
                        {
                            //offset += expectation.Prepare(st, Factory.ConnectionProvider.Driver);

                            int loc = WriteElement(st, collection.GetElement(entry), offset, session);
                            if (hasIdentifier)
                            {
                                WriteIdentifier(st, collection.GetIdentifier(entry, i), loc, session);
                            }
                            else
                            {
                                loc = WriteKey(st, id, loc, session);
                                if (HasIndex && !indexContainsFormula)
                                {
                                    WriteIndexToWhere(st, collection.GetIndex(entry, i, this), loc, session);
                                }
                                else
                                {
                                    // No nullness handled on update: updates does not occurs with sets or bags, and
                                    // indexed collections allowing formula (maps) force their element columns to
                                    // not-nullable.
                                    WriteElementToWhere(st, collection.GetSnapshotElement(entry, i), null, loc, session);
                                }
                            }

                            if (useBatch)
                            {
                                session.Batcher.AddToBatch(expectation);
                            }
                            else
                            {
                                expectation.VerifyOutcomeNonBatched(session.Batcher.ExecuteNonQuery(st), st);
                            }
                        }
                        catch (Exception e)
                        {
                            if (useBatch)
                            {
                                session.Batcher.AbortBatch(e);
                            }
                            throw;
                        }
                        finally
                        {
                            if (!useBatch)
                            {
                                session.Batcher.CloseCommand(st, null);
                            }
                        }
                        count++;
                    }
                    i++;
                }
                return(count);
            }
            catch (DbException sqle)
            {
                throw ADOExceptionHelper.Convert(SQLExceptionConverter, sqle,
                                                 "could not update collection rows: " + MessageHelper.CollectionInfoString(this, collection, id, session),
                                                 SqlUpdateRowString.Text);
            }
        }
		public void Recreate(IPersistentCollection collection, object id, ISessionImplementor session)
		{
			if (!isInverse)
			{
				if (log.IsDebugEnabled)
				{
					log.Debug("Inserting collection: " + MessageHelper.InfoString(this, id));
				}

				try
				{
					// create all the new entries
					IEnumerable entries = collection.Entries();
					IExpectation expectation = Expectations.AppropriateExpectation(insertCheckStyle);
					bool useBatch = expectation.CanBeBatched;

					int i = 0;
					int count = 0;
					collection.PreInsert(this);

					IDbCommand st = null;
					foreach (object entry in entries)
					{
						if (collection.EntryExists(entry, i))
						{
							try
							{
								int offset = 0;
								if (useBatch)
								{
									if (st == null)
									{
										st = session.Batcher.PrepareBatchCommand(
											SqlInsertRowString.CommandType,
											SqlInsertRowString.Text,
											SqlInsertRowString.ParameterTypes);
									}
								}
								else
								{
									st = session.Batcher.PrepareCommand(
										SqlInsertRowString.CommandType,
										SqlInsertRowString.Text,
										SqlInsertRowString.ParameterTypes);
								}
								//offset += expectation.Prepare(st, factory.ConnectionProvider.Driver);
								int loc = WriteKey(st, id, offset, session);
								if (hasIdentifier)
								{
									loc = WriteIdentifier(st, collection.GetIdentifier(entry, i), loc, session);
								}
								if (hasIndex /*&&! !indexIsFormula */)
								{
									loc = WriteIndex(st, collection.GetIndex(entry, i), loc, session);
								}
								loc = WriteElement(st, collection.GetElement(entry), loc, session);
								if (useBatch)
								{
									session.Batcher.AddToBatch(expectation);
								}
								else
								{
									expectation.VerifyOutcomeNonBatched(session.Batcher.ExecuteNonQuery(st), st);
								}
								collection.AfterRowInsert(this, entry, i);
								count++;
							}
							catch (Exception e)
							{
								if (useBatch)
								{
									session.Batcher.AbortBatch(e);
								}
								throw;
							}
							finally
							{
								if (!useBatch)
								{
									session.Batcher.CloseCommand(st, null);
								}
							}
						}
						i++;
					}

					if (log.IsDebugEnabled)
					{
						log.Debug(string.Format("done inserting collection: {0} rows inserted", count));
					}

					if (count == 0 && log.IsDebugEnabled)
					{
						log.Debug("collection was empty");
					}
				}
				catch (HibernateException)
				{
					// Do not call Convert on HibernateExceptions
					throw;
				}
				catch (Exception sqle)
				{
					throw Convert(sqle, "could not insert collection: " + MessageHelper.InfoString(this, id));
				}
			}
		}
		protected override int DoUpdateRows(object id, IPersistentCollection collection, ISessionImplementor session)
		{
			if (ArrayHelper.IsAllFalse(elementColumnIsSettable)) return 0;

			try
			{
				IDbCommand st = null;
				IExpectation expectation = Expectations.AppropriateExpectation(UpdateCheckStyle);
				//bool callable = UpdateCallable;
				bool useBatch = expectation.CanBeBatched;
				IEnumerable entries = collection.Entries(this);
				int i = 0;
				int count = 0;
				foreach (object entry in entries)
				{
					if (collection.NeedsUpdating(entry, i, ElementType))
					{
						int offset = 0;
						if (useBatch)
						{
							if (st == null)
							{
								st =
									session.Batcher.PrepareBatchCommand(SqlUpdateRowString.CommandType, SqlUpdateRowString.Text,
									                                    SqlUpdateRowString.ParameterTypes);
							}
						}
						else
						{
							st =
								session.Batcher.PrepareCommand(SqlUpdateRowString.CommandType, SqlUpdateRowString.Text,
								                               SqlUpdateRowString.ParameterTypes);
						}

						try
						{
							//offset += expectation.Prepare(st, Factory.ConnectionProvider.Driver);

							int loc = WriteElement(st, collection.GetElement(entry), offset, session);
							if (hasIdentifier)
							{
								WriteIdentifier(st, collection.GetIdentifier(entry, i), loc, session);
							}
							else
							{
								loc = WriteKey(st, id, loc, session);
								if (HasIndex && !indexContainsFormula)
								{
									WriteIndexToWhere(st, collection.GetIndex(entry, i, this), loc, session);
								}
								else
								{
									WriteElementToWhere(st, collection.GetSnapshotElement(entry, i), loc, session);
								}
							}

							if (useBatch)
							{
								session.Batcher.AddToBatch(expectation);
							}
							else
							{
								expectation.VerifyOutcomeNonBatched(session.Batcher.ExecuteNonQuery(st), st);
							}
						}
						catch (Exception e)
						{
							if (useBatch)
							{
								session.Batcher.AbortBatch(e);
							}
							throw;
						}
						finally
						{
							if (!useBatch)
							{
								session.Batcher.CloseCommand(st, null);
							}
						}
						count++;
					}
					i++;
				}
				return count;
			}
			catch (DbException sqle)
			{
				throw ADOExceptionHelper.Convert(SQLExceptionConverter, sqle,
				                                 "could not update collection rows: " + MessageHelper.InfoString(this, id),
				                                 SqlUpdateRowString.Text);
			}
		}
        public void Recreate(IPersistentCollection collection, object id, ISessionImplementor session)
        {
            if (!isInverse && RowInsertEnabled)
            {
                if (log.IsDebugEnabled)
                {
                    log.Debug("Inserting collection: " + MessageHelper.InfoString(this, id));
                }

                try
                {
                    // create all the new entries
                    IEnumerator entries = collection.Entries(this).GetEnumerator();
                    if (entries.MoveNext())
                    {
                        entries.Reset();
                        collection.PreInsert(this);
                        int i = 0;
                        int count = 0;

                        while (entries.MoveNext())
                        {
                            object entry = entries.Current;
                            if (collection.EntryExists(entry, i))
                            {
                                int offset = 0;
                                IDbCommand st;
                                IExpectation expectation = Expectations.AppropriateExpectation(insertCheckStyle);
                                //bool callable = InsertCallable;

                                bool useBatch = expectation.CanBeBatched;
                                if (useBatch)
                                {
                                    st =
                                        session.Batcher.PrepareBatchCommand(SqlInsertRowString.CommandType, SqlInsertRowString.Text,
                                                                            SqlInsertRowString.ParameterTypes);
                                }
                                else
                                {
                                    st =
                                        session.Batcher.PrepareCommand(SqlInsertRowString.CommandType, SqlInsertRowString.Text,
                                                                       SqlInsertRowString.ParameterTypes);
                                }
                                try
                                {
                                    //offset += expectation.Prepare(st, factory.ConnectionProvider.Driver);
                                    int loc = WriteKey(st, id, offset, session);
                                    if (hasIdentifier && !isPostInsertIdentifier)
                                        loc = WriteIdentifier(st, collection.GetIdentifier(entry, i), loc, session);

                                    if (hasIndex)
                                        loc = WriteIndex(st, collection.GetIndex(entry, i), loc, session);

                                    WriteElement(st, collection.GetElement(entry), loc, session);
                                    if (useBatch)
                                    {
                                        session.Batcher.AddToBatch(expectation);
                                    }
                                    else
                                    {
                                        expectation.VerifyOutcomeNonBatched(session.Batcher.ExecuteNonQuery(st), st);
                                    }
                                    collection.AfterRowInsert(this, entry, i);
                                    count++;
                                }
                                catch (Exception e)
                                {
                                    if (useBatch)
                                    {
                                        session.Batcher.AbortBatch(e);
                                    }
                                    throw;
                                }
                                finally
                                {
                                    if (!useBatch)
                                    {
                                        session.Batcher.CloseCommand(st, null);
                                    }
                                }
                            }
                            i++;
                        }

                        if (log.IsDebugEnabled)
                        {
                            log.Debug(string.Format("done inserting collection: {0} rows inserted", count));
                        }
                    }
                    else
                    {
                        if (log.IsDebugEnabled)
                            log.Debug("collection was empty");
                    }
                }
                catch (DbException sqle)
                {
                    throw ADOExceptionHelper.Convert(sqlExceptionConverter, sqle,
                                                     "could not insert collection: " + MessageHelper.InfoString(this, id));
                }
            }
        }
        public void InsertRows(IPersistentCollection collection, object id, ISessionImplementor session)
        {
            if (!isInverse && RowInsertEnabled)
            {
                if (log.IsDebugEnabled)
                    log.Debug("Inserting rows of collection: " + MessageHelper.InfoString(this, id, Factory));

                try
                {
                    // insert all the new entries
                    collection.PreInsert(this);
                    IExpectation expectation = Expectations.AppropriateExpectation(insertCheckStyle);
                    //bool callable = InsertCallable;
                    bool useBatch = expectation.CanBeBatched;
                    int i = 0;
                    int count = 0;

                    IEnumerable entries = collection.Entries(this);
                    foreach (object entry in entries)
                    {
                        int offset = 0;
                        if (collection.NeedsInserting(entry, i, elementType))
                        {
                            IDbCommand st = useBatch
                                                ? session.Batcher.PrepareBatchCommand(SqlInsertRowString.CommandType, SqlInsertRowString.Text,
                                                                                      SqlInsertRowString.ParameterTypes)
                                                : session.Batcher.PrepareCommand(SqlInsertRowString.CommandType, SqlInsertRowString.Text,
                                                                                 SqlInsertRowString.ParameterTypes);
                            try
                            {
                                //offset += expectation.Prepare(st, factory.ConnectionProvider.Driver);
                                offset = WriteKey(st, id, offset, session);
                                if (hasIdentifier && !isPostInsertIdentifier)
                                {
                                    offset = WriteIdentifier(st, collection.GetIdentifier(entry, i), offset, session);
                                }
                                if (hasIndex)
                                {
                                    offset = WriteIndex(st, collection.GetIndex(entry, i), offset, session);
                                }
                                WriteElement(st, collection.GetElement(entry), offset, session);
                                if (useBatch)
                                {
                                    session.Batcher.AddToBatch(expectation);
                                }
                                else
                                {
                                    expectation.VerifyOutcomeNonBatched(session.Batcher.ExecuteNonQuery(st), st);
                                }
                                collection.AfterRowInsert(this, entry, i);
                                count++;
                            }
                            catch (Exception e)
                            {
                                if (useBatch)
                                {
                                    session.Batcher.AbortBatch(e);
                                }
                                throw;
                            }
                            finally
                            {
                                if (!useBatch)
                                {
                                    session.Batcher.CloseCommand(st, null);
                                }
                            }
                        }
                        i++;
                    }

                    if (log.IsDebugEnabled)
                    {
                        log.Debug(string.Format("done inserting rows: {0} inserted", count));
                    }
                }
                catch (DbException sqle)
                {
                    throw ADOExceptionHelper.Convert(sqlExceptionConverter, sqle, "could not insert collection rows: " + MessageHelper.InfoString(this, id));
                }
            }
        }