1Imports System.ComponentModel
  2
  3imports System.Configuration.Install
  4
  5imports System.IO
  6
  7imports System.Reflection
  8
  9 
 10
 11<runinstaller(true)> Public Class DBCustomActionClass DBCustomAction
 12
 13inherits System.Configuration.Install.Installer
 14
 15 
 16
 17组件设计器生成的代码#region "组件设计器生成的代码 "
 18
 19public Sub New()Sub New()
 20
 21mybase.new()
 22
 23'该调用是组件设计器所必需的
 24
 25initializecomponent()
 26
 27'在 InitializeComponent() 调用之后添加任何初始化
 28
 29end Sub
 30
 31' Installer 重写 dispose 以清理组件列表。
 32
 33protected Overloads Overrides Sub Dispose()Sub Dispose(ByVal disposing As Boolean)
 34
 35if disposing Then
 36
 37if Not (components Is Nothing) Then
 38
 39components.dispose()
 40
 41end If
 42
 43end If
 44
 45mybase.dispose(disposing)
 46
 47end Sub
 48
 49private components As System.ComponentModel.IContainer
 50
 51<system.diagnostics.debuggerstepthrough()> Private Sub InitializeComponent()Sub InitializeComponent()
 52
 53end Sub
 54
 55#end Region
 56
 57'执行sql 语句
 58
 59private Sub ExecuteSql()Sub ExecuteSql(ByVal conn As String, ByVal DatabaseName As String, ByVal Sql As String)
 60
 61dim mySqlConnection As New SqlClient.SqlConnection(conn)
 62
 63dim Command As New SqlClient.SqlCommand(Sql, mySqlConnection)
 64
 65command.connection.open()
 66
 67command.connection.changedatabase(databasename)
 68
 69try
 70
 71command.executenonquery()
 72
 73finally
 74
 75'close Connection
 76
 77command.connection.close()
 78
 79end Try
 80
 81end Sub
 82
 83public Overrides Sub Install()Sub Install(ByVal stateSaver As System.Collections.IDictionary)
 84MyBase.Install(stateSaver)
 85
 86' ------------------------建立数据库-------------------------------------------------
 87
 88try
 89
 90dim connStr As String = String.Format("data source={0};user id={1};password={2};persist security info=false;packet size=4096", Me.Context.Parameters.Item("server"), Me.Context.Parameters.Item("user"), Me.Context.Parameters.Item("pwd"))
 91
 92'根据输入的数据库名称建立数据库
 93
 94executesql(connstr, "master", "CREATE DATABASE " + Me.Context.Parameters.Item("dbname"))
 95
 96'调用osql执行脚本
 97
 98dim sqlProcess As New System.Diagnostics.Process
 99
100sqlprocess.startinfo.filename = "osql.exe "
101
102sqlprocess.startinfo.arguments = String.Format(" -U {0} -P {1} -d {2} -i {3}db.sql", Me.Context.Parameters.Item("user"), Me.Context.Parameters.Item("pwd"), Me.Context.Parameters.Item("dbname"), Me.Context.Parameters.Item("targetdir"))
103
104sqlprocess.startinfo.windowstyle = ProcessWindowStyle.Hidden
105
106sqlprocess.start()
107
108sqlprocess.waitforexit() '等待执行
109
110sqlprocess.close()
111
112'删除脚本文件
113
114dim sqlFileInfo As New System.IO.FileInfo(String.Format("{0}db.sql", Me.Context.Parameters.Item("targetdir")))
115
116if sqlFileInfo.Exists Then
117
118sqlfileinfo.delete()
119
120end If
121
122catch ex As Exception
123
124throw ex
125
126end Try
127
128 
129
130' ---------------------将连接字符串写入Web.config-----------------------------------
131
132try
133
134dim FileInfo As System.IO.FileInfo = New System.IO.FileInfo(Me.Context.Parameters.Item("targetdir") & "\web.config")
135
136if Not FileInfo.Exists Then
137
138throw New InstallException("没有找到配置文件")
139
140end If
141
142'实例化xml文档
143
144dim XmlDocument As New System.Xml.XmlDocument
145
146xmldocument.load(fileinfo.fullname)
147
148 
149
150'查找到appsettings中的节点
151
152dim Node As System.Xml.XmlNode
153
154dim FoundIt As Boolean = False
155
156for Each Node In XmlDocument.Item("configuration").Item("appSettings")
157
158if Node.Name = "add" Then
159
160if Node.Attributes.GetNamedItem("key").Value = "connString" Then
161
162'写入连接字符串
163
164node.attributes.getnameditem("value").value = String.Format("Persist Security Info=False;Data Source={0};Initial Catalog={1};User ID={2};Password={3};Packet Size=4096;Pooling=true;Max Pool Size=100;Min Pool Size=1", _
165
166me.context.parameters.item("server"), Me.Context.Parameters.Item("dbname"), Me.Context.Parameters.Item("user"), Me.Context.Parameters.Item("pwd"))
167
168foundit = True
169
170end If
171
172end If
173
174next Node
175
176if Not FoundIt Then
177
178throw New InstallException("web.Config 文件没有包含connString连接字符串设置")
179
180end If
181
182xmldocument.save(fileinfo.fullname)
183
184catch ex As Exception
185
186throw ex
187
188end Try
189
190end Sub
191
192end Class
193
194