@@ -108,9 +108,11 @@ func NewStores(cfg *Config, clock clock.Clock) (*stores, error) {
108108
109109 acctStore := accounts .NewSQLStore (sqlStore .BaseDB , clock )
110110 sessStore := session .NewSQLStore (sqlStore .BaseDB , clock )
111+ rulesStore := firewalldb .NewSQLDB (sqlStore .BaseDB )
111112
112113 stores .accounts = acctStore
113114 stores .sessions = sessStore
115+ stores .rules = rulesStore
114116 stores .closeFns ["sqlite" ] = sqlStore .BaseDB .Close
115117
116118 case DatabaseBackendPostgres :
@@ -121,9 +123,11 @@ func NewStores(cfg *Config, clock clock.Clock) (*stores, error) {
121123
122124 acctStore := accounts .NewSQLStore (sqlStore .BaseDB , clock )
123125 sessStore := session .NewSQLStore (sqlStore .BaseDB , clock )
126+ rulesStore := firewalldb .NewSQLDB (sqlStore .BaseDB )
124127
125128 stores .accounts = acctStore
126129 stores .sessions = sessStore
130+ stores .rules = rulesStore
127131 stores .closeFns ["postgres" ] = sqlStore .BaseDB .Close
128132
129133 default :
@@ -161,5 +165,9 @@ func NewStores(cfg *Config, clock clock.Clock) (*stores, error) {
161165 stores .firewallBolt = firewallBoltDB
162166 stores .closeFns ["bbolt-firewalldb" ] = firewallBoltDB .Close
163167
168+ if stores .rules == nil {
169+ stores .rules = firewallBoltDB
170+ }
171+
164172 return stores , nil
165173}
0 commit comments