summaryrefslogtreecommitdiff
path: root/src/app/Cmpp298.Assignment3.DataAccess/InsertQueryBuilder.cs
blob: bbb3d9cbb3edf8ba402c32f3f8b37ecb6839691f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
using System.Collections.Generic;
using System.Text;

namespace Cmpp298.Assignment3.DataAccess {
	public class InsertQueryBuilder {
		private readonly string _tableName;
		private IList< CommandParameter > _parameters;

		public InsertQueryBuilder( string tableName ) {
			_tableName = tableName;
			_parameters = new List< CommandParameter >( );
		}

		public IList< CommandParameter > Parameters {
			get { return _parameters; }
		}

		public void Add( DatabaseColumn column, string value ) {
			_parameters.Add( new CommandParameter( column.ColumnName, value ) );
		}

		public override string ToString( ) {
			StringBuilder builder = new StringBuilder( );
			builder.AppendFormat( "INSERT INTO {0} ({1}) VALUES ({2});SELECT @@IDENTITY;", _tableName,
			                      GetParameterNames( string.Empty ),
			                      GetParameterNames( "@" ) );
			return builder.ToString( );
		}

		private string GetParameterNames( string prefix ) {
			StringBuilder builder = new StringBuilder( );
			foreach ( CommandParameter parameter in _parameters ) {
				builder.AppendFormat( "{0}{1},", prefix, parameter.ColumnName );
			}
			builder.Remove( builder.Length - 1, 1 );
			return builder.ToString( );
		}
	}
}