Gradle android插件，如何根据不同的产品风格复制和加载不同的动态库？

buildscript {
    repositories {
        google()
        jcenter()
    }
    dependencies {
        classpath 'com.android.tools.build:gradle:4.2.0'
    }
}

android {
    ...
    flavorDimensions "style"
    productFlavors {
        style1 {
            dimension "style"
            // 配置style1的相关属性
        }
        style2 {
            dimension "style"
            // 配置style2的相关属性
        }
    }
}

android {
    ...
    sourceSets {
        style1 {
            jniLibs.srcDirs = ['src/style1/libs']
        }
        style2 {
            jniLibs.srcDirs = ['src/style2/libs']
        }
    }
}

if (BuildConfig.FLAVOR.equals("style1")) {
    System.loadLibrary("lib1");
} else if (BuildConfig.FLAVOR.equals("style2")) {
    System.loadLibrary("lib2");
}