@@ -138,12 +138,12 @@ def set_app_db_entry(self, table_name, *args):
138138 else :
139139 pb_string = value
140140
141- table = ProducerStateTable (self .dvs .get_app_db ().db_connection , table_name )
141+ table = ProducerStateTable (self .dvs .get_dpu_app_db ().db_connection , table_name )
142142 table [key ] = {'pb' : pb_string }
143143
144144 def remove_app_db_entry (self , table_name , * key_parts ):
145145 key = ":" .join (key_parts )
146- table = ProducerStateTable (self .dvs .get_app_db ().db_connection , table_name )
146+ table = ProducerStateTable (self .dvs .get_dpu_app_db ().db_connection , table_name )
147147 del table [key ]
148148
149149 def get_asic_db_entry (self , table_name , key ):
@@ -204,7 +204,7 @@ def get_attr_to_sai_object_map(self, table_name, attribute):
204204 return attr_to_sai_object_map
205205
206206 def get_app_db_keys (self , table_name ):
207- table = Table (self .dvs .get_app_db ().db_connection , table_name )
207+ table = Table (self .dvs .get_dpu_app_db ().db_connection , table_name )
208208 return table .get_keys ()
209209
210210 def get_asic_db_keys (self , table_name ):
@@ -214,27 +214,27 @@ def get_asic_db_keys(self, table_name):
214214 def __init__ (self , dvs ):
215215 self .dvs = dvs
216216 self .app_dash_routing_type_table = ProducerStateTable (
217- self .dvs .get_app_db ().db_connection , "DASH_ROUTING_TYPE_TABLE" )
217+ self .dvs .get_dpu_app_db ().db_connection , "DASH_ROUTING_TYPE_TABLE" )
218218 self .app_dash_appliance_table = ProducerStateTable (
219- self .dvs .get_app_db ().db_connection , "DASH_APPLIANCE_TABLE" )
219+ self .dvs .get_dpu_app_db ().db_connection , "DASH_APPLIANCE_TABLE" )
220220 self .app_dash_vnet_table = ProducerStateTable (
221- self .dvs .get_app_db ().db_connection , "DASH_VNET_TABLE" )
221+ self .dvs .get_dpu_app_db ().db_connection , "DASH_VNET_TABLE" )
222222 self .app_dash_eni_table = ProducerStateTable (
223- self .dvs .get_app_db ().db_connection , "DASH_ENI_TABLE" )
223+ self .dvs .get_dpu_app_db ().db_connection , "DASH_ENI_TABLE" )
224224 self .app_dash_vnet_map_table = ProducerStateTable (
225- self .dvs .get_app_db ().db_connection , "DASH_VNET_MAPPING_TABLE" )
225+ self .dvs .get_dpu_app_db ().db_connection , "DASH_VNET_MAPPING_TABLE" )
226226 self .app_dash_route_table = ProducerStateTable (
227- self .dvs .get_app_db ().db_connection , "DASH_ROUTE_TABLE" )
227+ self .dvs .get_dpu_app_db ().db_connection , "DASH_ROUTE_TABLE" )
228228 self .app_dash_route_rule_table = ProducerStateTable (
229- self .dvs .get_app_db ().db_connection , "DASH_ROUTE_RULE_TABLE" )
229+ self .dvs .get_dpu_app_db ().db_connection , "DASH_ROUTE_RULE_TABLE" )
230230 self .app_dash_eni_route_table = ProducerStateTable (
231- self .dvs .get_app_db ().db_connection , "DASH_ENI_ROUTE_TABLE" )
231+ self .dvs .get_dpu_app_db ().db_connection , "DASH_ENI_ROUTE_TABLE" )
232232 self .app_dash_route_group_table = ProducerStateTable (
233- self .dvs .get_app_db ().db_connection , "DASH_ROUTE_GROUP_TABLE" )
233+ self .dvs .get_dpu_app_db ().db_connection , "DASH_ROUTE_GROUP_TABLE" )
234234 self .app_dash_meter_policy_table = ProducerStateTable (
235- self .dvs .get_app_db ().db_connection , "DASH_METER_POLICY_TABLE" )
235+ self .dvs .get_dpu_app_db ().db_connection , "DASH_METER_POLICY_TABLE" )
236236 self .app_dash_meter_rule_table = ProducerStateTable (
237- self .dvs .get_app_db ().db_connection , "DASH_METER_RULE_TABLE" )
237+ self .dvs .get_dpu_app_db ().db_connection , "DASH_METER_RULE_TABLE" )
238238
239239 self .asic_dash_appliance_table = Table (
240240 self .dvs .get_asic_db ().db_connection , "ASIC_STATE:SAI_OBJECT_TYPE_DASH_APPLIANCE" )
0 commit comments