LISTING 4: Code That Alters Listing 2's UPDATE Statement

ALTER PROCEDURE CustomerInsert @CustomerID nchar(10),
	@CompanyName nvarchar(80),@ContactName nvarchar(60)=NULL,
	@ContactTitle nvarchar(60)=NULL,@Address nvarchar(120)=NULL,
	@City nvarchar(30)=NULL,@Region nvarchar(30)=NULL,
	@PostalCode nvarchar(20)=NULL,@Country nvarchar(30)=NULL,
	@Phone nvarchar(48)=NULL,@Fax nvarchar(48)=NULL
AS

DECLARE @audit_id smallint
SET NOCOUNT ON
SELECT @audit_id=audit_id FROM audit WHERE audit_user=user_name()
IF @@rowcount=0
BEGIN
	INSERT INTO AUDIT (audit_user) VALUES (user_name())
	SET @audit_id=@@identity
END
UPDATE Customers SET CompanyName=@CompanyName,
ContactName=@ContactName,
	ContactTitle=@ContactTitle,Address=@Address,
City=@City,Region=@Region,
	PostalCode=@PostalCode,Country=@Country,
Phone=@Phone,Fax=@Fax,
	au_user_id=@audit_id,au_update_date=getdate()
WHERE CustomerID=@CustomerID
RETURN @@error

SET NOCOUNT OFF