Bagaimana cara menghitung semua kelas dengan atribut kelas khusus?

Pertanyaan berdasarkan contoh MSDN .

Katakanlah kita memiliki beberapa kelas C # dengan HelpAttribute di aplikasi desktop mandiri. Apakah mungkin untuk menghitung semua kelas dengan atribut seperti itu? Apakah masuk akal untuk mengenali kelas dengan cara ini? Atribut khusus akan digunakan untuk membuat daftar opsi menu yang mungkin, memilih item akan membawa ke instance layar kelas tersebut. Jumlah kelas / item akan tumbuh lambat, tetapi dengan cara ini kita dapat menghindari penghitungan semuanya di tempat lain, saya pikir.

Ya, tentu saja. Menggunakan Refleksi:

static IEnumerable<Type> GetTypesWithHelpAttribute(Assembly assembly) {
    foreach(Type type in assembly.GetTypes()) {
        if (type.GetCustomAttributes(typeof(HelpAttribute), true).Length > 0) {
            yield return type;
        }
    }
}

Nah, Anda harus menghitung melalui semua kelas di semua majelis yang dimuat ke domain aplikasi saat ini. Untuk melakukan itu, Anda akan memanggil GetAssembliesmetode pada AppDomaincontoh untuk domain aplikasi saat ini.

Dari sana, Anda akan menelepon GetExportedTypes(jika Anda hanya ingin tipe publik) atau GetTypesmasing Assembly- masing untuk mendapatkan tipe yang terkandung dalam majelis.

Kemudian, Anda akan memanggil GetCustomAttributesmetode ekstensi pada setiap Typeinstance, melewati tipe atribut yang ingin Anda temukan.

Anda dapat menggunakan LINQ untuk menyederhanakan ini untuk Anda:

var typesWithMyAttribute =
    from a in AppDomain.CurrentDomain.GetAssemblies()
    from t in a.GetTypes()
    let attributes = t.GetCustomAttributes(typeof(HelpAttribute), true)
    where attributes != null && attributes.Length > 0
    select new { Type = t, Attributes = attributes.Cast<HelpAttribute>() };

Kueri di atas akan memberi Anda setiap jenis dengan atribut yang diterapkan padanya, bersama dengan instance atribut yang ditugaskan padanya.

Perhatikan bahwa jika Anda memiliki banyak rakitan yang dimuat ke domain aplikasi Anda, operasi itu bisa mahal. Anda dapat menggunakan LINQ Paralel untuk mengurangi waktu operasi, seperti:

var typesWithMyAttribute =
    // Note the AsParallel here, this will parallelize everything after.
    from a in AppDomain.CurrentDomain.GetAssemblies().AsParallel()
    from t in a.GetTypes()
    let attributes = t.GetCustomAttributes(typeof(HelpAttribute), true)
    where attributes != null && attributes.Length > 0
    select new { Type = t, Attributes = attributes.Cast<HelpAttribute>() };

Memfilternya pada spesifik Assemblyadalah sederhana:

Assembly assembly = ...;

var typesWithMyAttribute =
    from t in assembly.GetTypes()
    let attributes = t.GetCustomAttributes(typeof(HelpAttribute), true)
    where attributes != null && attributes.Length > 0
    select new { Type = t, Attributes = attributes.Cast<HelpAttribute>() };

Dan jika perakitan memiliki sejumlah besar jenis di dalamnya, maka Anda dapat menggunakan LINQ paralel:

Assembly assembly = ...;

var typesWithMyAttribute =
    // Partition on the type list initially.
    from t in assembly.GetTypes().AsParallel()
    let attributes = t.GetCustomAttributes(typeof(HelpAttribute), true)
    where attributes != null && attributes.Length > 0
    select new { Type = t, Attributes = attributes.Cast<HelpAttribute>() };

Referensi jawaban lain GetCustomAttributes . Menambahkan yang ini sebagai contoh menggunakan IsDefined

Assembly assembly = ...
var typesWithHelpAttribute = 
        from type in assembly.GetTypes()
        where type.IsDefined(typeof(HelpAttribute), false)
        select type;

Seperti yang telah dinyatakan, refleksi adalah jalan yang harus ditempuh. Jika Anda akan sering menyebutnya, saya sangat menyarankan untuk menyimpan hasilnya, karena refleksi, terutama menghitung melalui setiap kelas, bisa sangat lambat.

Ini adalah potongan kode saya yang menjalankan semua jenis di semua rakitan yang dimuat:

// this is making the assumption that all assemblies we need are already loaded.
foreach (Assembly assembly in AppDomain.CurrentDomain.GetAssemblies()) 
{
    foreach (Type type in assembly.GetTypes())
    {
        var attribs = type.GetCustomAttributes(typeof(MyCustomAttribute), false);
        if (attribs != null && attribs.Length > 0)
        {
            // add to a cache.
        }
    }
}

Ini adalah peningkatan kinerja di atas solusi yang diterima. Iterasi meskipun semua kelas bisa lambat karena ada begitu banyak. Kadang-kadang Anda dapat menyaring seluruh unit tanpa melihat jenisnya.

Misalnya, jika Anda mencari atribut yang Anda nyatakan sendiri, Anda tidak mengharapkan DLL sistem mengandung tipe apa pun dengan atribut itu. Properti Assembly.GlobalAssemblyCache adalah cara cepat untuk memeriksa DLL sistem. Ketika saya mencoba ini pada program nyata saya menemukan saya bisa melewati 30.101 jenis dan saya hanya perlu memeriksa 1.983 jenis.

Cara lain untuk memfilter adalah dengan menggunakan Assembly.ReferencedAssemblies. Agaknya jika Anda ingin kelas dengan atribut tertentu, dan atribut itu didefinisikan dalam rakitan tertentu, maka Anda hanya peduli tentang rakitan itu dan rakitan lain yang merujuknya. Dalam pengujian saya ini membantu sedikit lebih banyak daripada memeriksa properti GlobalAssemblyCache.

Saya menggabungkan keduanya dan membuatnya lebih cepat. Kode di bawah ini mencakup kedua filter.

        string definedIn = typeof(XmlDecoderAttribute).Assembly.GetName().Name;
        foreach (Assembly assembly in AppDomain.CurrentDomain.GetAssemblies())
            // Note that we have to call GetName().Name.  Just GetName() will not work.  The following
            // if statement never ran when I tried to compare the results of GetName().
            if ((!assembly.GlobalAssemblyCache) && ((assembly.GetName().Name == definedIn) || assembly.GetReferencedAssemblies().Any(a => a.Name == definedIn)))
                foreach (Type type in assembly.GetTypes())
                    if (type.GetCustomAttributes(typeof(XmlDecoderAttribute), true).Length > 0)

Dalam hal batasan .NET Portable , kode berikut harus berfungsi:

    public static IEnumerable<TypeInfo> GetAtributedTypes( Assembly[] assemblies, 
                                                           Type attributeType )
    {
        var typesAttributed =
            from assembly in assemblies
            from type in assembly.DefinedTypes
            where type.IsDefined(attributeType, false)
            select type;
        return typesAttributed;
    }

atau untuk sejumlah besar majelis yang menggunakan loop-state based yield return:

    public static IEnumerable<TypeInfo> GetAtributedTypes( Assembly[] assemblies, 
                                                           Type attributeType )
    {
        foreach (var assembly in assemblies)
        {
            foreach (var typeInfo in assembly.DefinedTypes)
            {
                if (typeInfo.IsDefined(attributeType, false))
                {
                    yield return typeInfo;
                }
            }
        }
    }

Kami dapat memperbaiki jawaban Andrew dan mengonversi semuanya menjadi satu permintaan LINQ.

    public static IEnumerable<Type> GetTypesWithHelpAttribute(Assembly assembly)
    {
        return assembly.GetTypes().Where(type => type.GetCustomAttributes(typeof(HelpAttribute), true).Length > 0);
    }