1212use WPGraphQL \WooCommerce \Data \Factory ;
1313use WPGraphQL \WooCommerce \Data \Loader \WC_CPT_Loader ;
1414use WPGraphQL \WooCommerce \Data \Loader \WC_Customer_Loader ;
15- use WPGraphQL \WooCommerce \Data \Loader \WC_Db_Loader ;
15+ use WPGraphQL \WooCommerce \Data \Loader \WC_Cart_Item_Loader ;
16+ use WPGraphQL \WooCommerce \Data \Loader \WC_Downloadable_Item_Loader ;
17+ use WPGraphQL \WooCommerce \Data \Loader \WC_Order_Item_Loader ;
18+ use WPGraphQL \WooCommerce \Data \Loader \WC_Shipping_Method_Loader ;
19+ use WPGraphQL \WooCommerce \Data \Loader \WC_Shipping_Zone_Loader ;
20+ use WPGraphQL \WooCommerce \Data \Loader \WC_Tax_Class_Loader ;
21+ use WPGraphQL \WooCommerce \Data \Loader \WC_Tax_Rate_Loader ;
1622use WPGraphQL \WooCommerce \WP_GraphQL_WooCommerce as WooGraphQL ;
1723
1824/**
@@ -33,7 +39,7 @@ public static function add_filters() {
3339 add_filter ( 'register_taxonomy_args ' , [ self ::class, 'register_taxonomy_args ' ], 10 , 2 );
3440
3541 // Add data-loaders to AppContext.
36- add_filter ( 'graphql_data_loaders ' , [ self ::class, 'graphql_data_loaders ' ], 10 , 2 );
42+ add_filter ( 'graphql_data_loader_classes ' , [ self ::class, 'graphql_data_loader_classes ' ], 10 , 2 );
3743
3844 // Add node resolvers.
3945 add_filter (
@@ -255,30 +261,21 @@ public static function register_taxonomy_args( $args, $taxonomy ) {
255261 *
256262 * @return array
257263 */
258- public static function graphql_data_loaders ( $ loaders, $ context ) {
264+ public static function graphql_data_loader_classes ( $ loaders ) {
259265 // WooCommerce customer loader.
260- $ customer_loader = new WC_Customer_Loader ( $ context );
261- $ loaders ['wc_customer ' ] = &$ customer_loader ;
266+ $ loaders ['wc_customer ' ] = WC_Customer_Loader::class;
262267
263268 // WooCommerce CPT loader.
264- $ cpt_loader = new WC_CPT_Loader ( $ context );
265- $ loaders ['wc_post ' ] = &$ cpt_loader ;
269+ $ loaders ['wc_post ' ] = WC_CPT_Loader::class;
266270
267271 // WooCommerce DB loaders.
268- $ cart_item_loader = new WC_Db_Loader ( $ context , 'CART_ITEM ' );
269- $ loaders ['cart_item ' ] = &$ cart_item_loader ;
270- $ downloadable_item_loader = new WC_Db_Loader ( $ context , 'DOWNLOADABLE_ITEM ' );
271- $ loaders ['downloadable_item ' ] = &$ downloadable_item_loader ;
272- $ tax_class_loader = new WC_Db_Loader ( $ context , 'TAX_CLASS ' );
273- $ loaders ['tax_class ' ] = &$ tax_class_loader ;
274- $ tax_rate_loader = new WC_Db_Loader ( $ context , 'TAX_RATE ' );
275- $ loaders ['tax_rate ' ] = &$ tax_rate_loader ;
276- $ order_item_loader = new WC_Db_Loader ( $ context , 'ORDER_ITEM ' );
277- $ loaders ['order_item ' ] = &$ order_item_loader ;
278- $ shipping_item_loader = new WC_Db_Loader ( $ context , 'SHIPPING_METHOD ' );
279- $ loaders ['shipping_method ' ] = &$ shipping_item_loader ;
280- $ shipping_zone_loader = new WC_Db_Loader ( $ context , 'SHIPPING_ZONE ' );
281- $ loaders ['shipping_zone ' ] = &$ shipping_zone_loader ;
272+ $ loaders ['cart_item ' ] = WC_Cart_Item_Loader::class;
273+ $ loaders ['downloadable_item ' ] = WC_Downloadable_Item_Loader::class;
274+ $ loaders ['tax_class ' ] = WC_Tax_Class_Loader::class;
275+ $ loaders ['tax_rate ' ] = WC_Tax_Rate_Loader::class;
276+ $ loaders ['order_item ' ] = WC_Order_Item_Loader::class;
277+ $ loaders ['shipping_method ' ] = WC_Shipping_Method_Loader::class;
278+ $ loaders ['shipping_zone ' ] = WC_Shipping_Zone_Loader::class;
282279 return $ loaders ;
283280 }
284281
0 commit comments