如何为匿名类型编写扩展方法?

时间:2012-01-30 17:16:14

标签: c# linq linq-to-sql extension-methods

我正在尝试为我的可枚举列表创建一个CSV扩展方法,我很难过。这是我创建简单枚举列表的方法:

var CAquery = from temp in CAtemp
 join casect in CAdb.sectors
 on temp.sector_code equals casect.sector_code
 select new
 {       
     CUSIP = temp.equity_cusip,
     CompName = temp.company_name,
     Exchange = temp.primary_exchange       
 };

CAquery.WriteToCSVFile();

这是我到目前为止创建扩展方法(我认为是错误的):

public static class CSVExtensions
{        
    public static void WriteToCSVFile(this IEnumerable<T> myList)
    {

你看到我做错了吗?

2 个答案:

答案 0 :(得分:12)

您必须在方法签名中指定泛型类型参数:

public static class CSVExtensions
{        
    public static void WriteToCSVFile<T>(this IEnumerable<T> myList)
    {
       //your code here
    }
}

您是否真的尝试编写一个适用于任何IEnumerable<T>的扩展方法,或者您的类型更具体?如果是更晚的情况,则应将T替换为您要支持的类型(或添加足够的约束)。

修改

根据评论 - 您应该在查询中投射到一个类而不是一个匿名类型 - 然后您可以对此特定类型使用扩展方法,即:

class CompanyTicker
{
  public string CUSIP {get;set;}
  public string CompName  {get;set;}
  public string Exchange {get;set;}
}

现在您的查询可以是:

var CAquery = from temp in CAtemp
 join casect in CAdb.sectors
 on temp.sector_code equals casect.sector_code
 select new CompanyTicker
 {       
     CUSIP = temp.equity_cusip,
     CompName = temp.company_name,
     Exchange = temp.primary_exchange       
 };

您的扩展方法(现在不需要是通用的)变为:

public static class CSVExtensions
{        
    public static void WriteToCSVFile(this IEnumerable<CompanyTicker> myList)
    {
       //your code here
    }
}

答案 1 :(得分:2)

可以使用反射做你想做的事情。但是,如果编写非泛型代码,性能会稍差一些。

以下是完整的代码示例:

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Reflection;

class Program
{
    static void Main(string[] args)
    {
        var seq =
        Enumerable.Range(0, 100)
            .Select(i => new { Name = "Item" + i, Value = i })
            ;

        seq.WriteCsv(Console.Out);
        Console.ReadLine();
    }
}

public static class CsvExtension
{

    public static void WriteCsv<T>(this IEnumerable<T> seq, TextWriter writer)
    {
        var type = typeof(T);

        MethodInfo[] getters = type.GetProperties().Select(pi => pi.GetGetMethod()).ToArray();


        // only supporting simple properties
        // indexer properties will probably fail
        var args = new object[0];

        foreach (var item in seq)
        {
            for (int i = 0; i < getters.Length; i++)
            {
                if (i != 0)
                    writer.Write(",");

                Object value = getters[i].Invoke(item, args);
                var str = value.ToString();

                if (str.Contains(",") || str.Contains("\""))
                {
                    var escaped = str.Replace("\"", "\\\"");
                    writer.Write("\"");
                    writer.Write(escaped);
                    writer.Write("\"");
                }
                else
                {
                    writer.Write(str);
                }
            }


            writer.WriteLine();
        }
    }
}