@@ -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