如何在自定义类中为Flutter中的不同屏幕尺寸设置应用栏的高度？

class CustomAppBar extends StatelessWidget implements PreferredSizeWidget {
  final double appBarHeight;

  CustomAppBar({this.appBarHeight});

  @override
  Size get preferredSize => Size.fromHeight(appBarHeight);

  @override
  Widget build(BuildContext context) {
    return AppBar(
      title: Text('Your App Title'),
      // 其他AppBar属性设置
    );
  }
}

class MyHomePage extends StatelessWidget {
  @override
  Widget build(BuildContext context) {
    final double screenHeight = MediaQuery.of(context).size.height;

    double appBarHeight;
    if (screenHeight < 600) {
      appBarHeight = 50.0;
    } else if (screenHeight < 900) {
      appBarHeight = 60.0;
    } else {
      appBarHeight = 70.0;
    }

    return Scaffold(
      appBar: CustomAppBar(appBarHeight: appBarHeight),
      body: Container(
        // 页面内容
      ),
    );
  }
}