@@ -108,9 +108,11 @@ func NewStores(cfg *Config, clock clock.Clock) (*stores, error) {
108
108
109
109
acctStore := accounts .NewSQLStore (sqlStore .BaseDB , clock )
110
110
sessStore := session .NewSQLStore (sqlStore .BaseDB , clock )
111
+ firewallStore := firewalldb .NewSQLDB (sqlStore .BaseDB )
111
112
112
113
stores .accounts = acctStore
113
114
stores .sessions = sessStore
115
+ stores .firewall = firewalldb .NewDB (firewallStore )
114
116
stores .closeFns ["sqlite" ] = sqlStore .BaseDB .Close
115
117
116
118
case DatabaseBackendPostgres :
@@ -121,9 +123,11 @@ func NewStores(cfg *Config, clock clock.Clock) (*stores, error) {
121
123
122
124
acctStore := accounts .NewSQLStore (sqlStore .BaseDB , clock )
123
125
sessStore := session .NewSQLStore (sqlStore .BaseDB , clock )
126
+ firewallStore := firewalldb .NewSQLDB (sqlStore .BaseDB )
124
127
125
128
stores .accounts = acctStore
126
129
stores .sessions = sessStore
130
+ stores .firewall = firewalldb .NewDB (firewallStore )
127
131
stores .closeFns ["postgres" ] = sqlStore .BaseDB .Close
128
132
129
133
default :
@@ -157,7 +161,10 @@ func NewStores(cfg *Config, clock clock.Clock) (*stores, error) {
157
161
err )
158
162
}
159
163
160
- stores .firewall = firewalldb .NewDB (firewallBoltDB )
164
+ if stores .firewall == nil {
165
+ stores .firewall = firewalldb .NewDB (firewallBoltDB )
166
+ }
167
+
161
168
stores .firewallBolt = firewallBoltDB
162
169
stores .closeFns ["bbolt-firewalldb" ] = firewallBoltDB .Close
163
170
0 commit comments