示例#1
0
        /// <summary>
        /// Add all the parameters needed to the command just before it executes
        /// </summary>
        /// <param name="command">The raw command prior to execution</param>
        /// <param name="identity">Information about the query</param>
        public void AddParameters(IDbCommand command, Identity identity)
        {
            var literals = SqlMapper.GetLiteralTokens(identity.sql);

            if (templates != null)
            {
                foreach (var template in templates)
                {
                    var newIdent = identity.ForDynamicParameters(template.GetType());
                    Action <IDbCommand, object> appender;

                    lock (paramReaderCache)
                    {
                        if (!paramReaderCache.TryGetValue(newIdent, out appender))
                        {
                            appender = SqlMapper.CreateParamInfoGenerator(newIdent, true, RemoveUnused, literals);
                            paramReaderCache[newIdent] = appender;
                        }
                    }

                    appender(command, template);
                }

                // The parameters were added to the command, but not the
                // DynamicParameters until now.
                foreach (IDbDataParameter param in command.Parameters)
                {
                    // If someone makes a DynamicParameters with a template,
                    // then explicitly adds a parameter of a matching name,
                    // it will already exist in 'parameters'.
                    if (!parameters.ContainsKey(param.ParameterName))
                    {
                        parameters.Add(param.ParameterName, new ParamInfo
                        {
                            AttachedParam      = param,
                            CameFromTemplate   = true,
                            DbType             = param.DbType,
                            Name               = param.ParameterName,
                            ParameterDirection = param.Direction,
                            Size               = param.Size,
                            Value              = param.Value
                        });
                    }
                }

                // Now that the parameters are added to the command, let's place our output callbacks
                var tmp = outputCallbacks;
                if (tmp != null)
                {
                    foreach (var generator in tmp)
                    {
                        generator();
                    }
                }
            }

            foreach (var param in parameters.Values)
            {
                if (param.CameFromTemplate)
                {
                    continue;
                }

                var    dbType = param.DbType;
                var    val    = param.Value;
                string name   = Clean(param.Name);
                var    isCustomQueryParameter = val is ICustomQueryParameter;

                ITypeHandler handler = null;
                if (dbType == null && val != null && !isCustomQueryParameter)
                {
                    dbType = SqlMapper.LookupDbType(val.GetType(), name, true, out handler);
                }
                if (dbType == DynamicParameters.EnumerableMultiParameter)
                {
#pragma warning disable 612, 618
                    SqlMapper.PackListParameters(command, name, val);
#pragma warning restore 612, 618
                }
                else if (isCustomQueryParameter)
                {
                    ((ICustomQueryParameter)val).AddParameter(command, name);
                }
                else
                {
                    bool             add = !command.Parameters.Contains(name);
                    IDbDataParameter p;
                    if (add)
                    {
                        p = command.CreateParameter();
                        p.ParameterName = name;
                    }
                    else
                    {
                        p = (IDbDataParameter)command.Parameters[name];
                    }

                    p.Direction = param.ParameterDirection;
                    if (handler == null)
                    {
                        p.Value = val ?? DBNull.Value;
                        if (dbType != null && p.DbType != dbType)
                        {
                            p.DbType = dbType.Value;
                        }
                        var s = val as string;
                        if (s != null)
                        {
                            if (s.Length <= DbString.DefaultLength)
                            {
                                p.Size = DbString.DefaultLength;
                            }
                        }
                        if (param.Size != null)
                        {
                            p.Size = param.Size.Value;
                        }
                    }
                    else
                    {
                        if (dbType != null)
                        {
                            p.DbType = dbType.Value;
                        }
                        if (param.Size != null)
                        {
                            p.Size = param.Size.Value;
                        }
                        handler.SetValue(p, val ?? DBNull.Value);
                    }

                    if (add)
                    {
                        command.Parameters.Add(p);
                    }
                    param.AttachedParam = p;
                }
            }

            // note: most non-priveleged implementations would use: this.ReplaceLiterals(command);
            if (literals.Count != 0)
            {
                SqlMapper.ReplaceLiterals(this, command, literals);
            }
        }
示例#2
0
        private void AddParameter(ParameterInfo parameter, IDbCommand command)
        {
            var dbType        = parameter.DbType;
            var parameterName = parameter.Name.Clean();

            if (parameter.Value is ICustomQueryParameter)
            {
                ((ICustomQueryParameter)parameter.Value).AddParameter(command, parameterName);

                return;
            }

            ITypeHandler handler = null;

            if (dbType == null && parameter.Value != null)
            {
                dbType = parameter.Value.GetType().LookupDbType(parameterName, true, out handler);
            }

            if (handler == null && ((parameter.Value?.GetType().ImplementsIEnumerable() ?? false) || (parameter.Value?.GetType().IsArray ?? false)))
            {
                AddEnumerationParameter(parameter, command, dbType);

                return;
            }

            var addParameter  = !command.Parameters.Contains(parameterName);
            var dataParameter = GetOrCreateDataParameter(command, parameterName, addParameter);

            if (handler == null)
            {
                dataParameter.Value = parameter.Value.SanitizeParameterValue();
                if (dbType != null && dataParameter.DbType != dbType)
                {
                    dataParameter.DbType = dbType.Value;
                }

                var s = parameter.Value as string;
                if (s?.Length <= DefaultValues.DBSTRINGDEFAULTLENGTH)
                {
                    dataParameter.Size = DefaultValues.DBSTRINGDEFAULTLENGTH;
                }

                SetBasicParameterInfos(parameter, dataParameter);
            }
            else
            {
                if (dbType != null)
                {
                    dataParameter.DbType = dbType.Value;
                }

                SetBasicParameterInfos(parameter, dataParameter);

                handler.SetValue(dataParameter, parameter.Value ?? DBNull.Value);
            }

            if (addParameter)
            {
                command.Parameters.Add(dataParameter);
            }
        }