1. 程式人生 > >Postgresql快速寫入/讀取大量資料(.net)

Postgresql快速寫入/讀取大量資料(.net)

環境及測試

使用.net驅動npgsql連線post資料庫。配置:win10 x64, i5-4590, 16G DDR3, SSD 850EVO.

postgresql 9.6.3,資料庫與資料都安裝在SSD上,預設配置,無擴充套件。

CREATE TABLE public.mesh
(
  x integer NOT NULL,
  y integer NOT NULL,
  z integer,
  CONSTRAINT prim PRIMARY KEY (x, y)
)

1. 匯入

使用資料備份,csv格式匯入,檔案位於機械硬碟上,480MB,資料量2500w+。

  • 使用COPY

    copy
    mesh from 'd:/user.csv' csv

    執行時間107s

  • 使用insert

    單連線,c# release any cpu 非除錯模式。

class Program
{
    static void Main(string[] args)
    {
        var list = GetData("D:\\user.csv");
        TimeCalc.LogStartTime();
        using (var sm = new SqlManipulation(@"Strings", SqlType.PostgresQL))
        {
            sm.Init();
            foreach
(var n in list) { sm.ExcuteNonQuery($"insert into mesh(x,y,z) values({n.x},{n.y},{n.z})"); } } TimeCalc.ShowTotalDuration(); Console.ReadKey(); } static List<(int x, int y, int z)> GetData(string filepath) { List<ValueTuple<int
, int, int>> list = new List<(int, int, int)>(); foreach (var n in File.ReadLines(filepath)) { string[] x = n.Split(','); list.Add((Convert.ToInt32(x[0]), Convert.ToInt32(x[1]), Convert.ToInt32(x[2]))); } return list; } }

Postgresql CPU佔用率很低,但是跑了一年,程式依然不能結束,沒有耐性了...,這麼插入不行。

  • multiline insert

使用multiline插入,一條語句插入約100條資料。

var bag = GetData("D:\\user.csv");
//使用時,直接執行stringbuilder的tostring方法。
List<StringBuilder> listbuilder = new List<StringBuilder>();
StringBuilder sb = new StringBuilder();
for (int i = 0; i < bag.Count; i++)
{
    if (i % 100 == 0)
    {
        sb = new StringBuilder();
        listbuilder.Add(sb);
        sb.Append("insert into mesh(x,y,z) values");
        sb.Append($"({bag[i].x}, {bag[i].y}, {bag[i].z})");
    }
    else
        sb.Append($",({bag[i].x}, {bag[i].y}, {bag[i].z})");
}

Postgresql CPU佔用率差不多27%,磁碟寫入大約45MB/S,感覺就是在幹活,最後時間217.36s。
改為1000一行的話,CPU佔用率提高,但是磁碟寫入平均來看有所降低,最後時間160.58s.

  • prepare語法

prepare語法可以讓postgresql提前規劃sql,優化效能。

使用單行插入 CPU佔用率不到25%,磁碟寫入63MB/S左右,但是,使用單行插入的方式,效率沒有改觀,時間太長還是等不來結果。

使用多行插入 CPU佔用率30%,磁碟寫入50MB/S,最後結果163.02,最後的時候出了個異常,就是最後一組資料長度不滿足條件,無傷大雅。

static void Main(string[] args)
{
    var bag = GetData("D:\\user.csv");
    List<StringBuilder> listbuilder = new List<StringBuilder>();
    StringBuilder sb = new StringBuilder();
    for (int i = 0; i < bag.Count; i++)
    {
        if (i % 1000 == 0)
        {
            sb = new StringBuilder();
            listbuilder.Add(sb);
            //sb.Append("insert into mesh(x,y,z) values");
            sb.Append($"{bag[i].x}, {bag[i].y}, {bag[i].z}");
        }
        else
            sb.Append($",{bag[i].x}, {bag[i].y}, {bag[i].z}");
    }
    StringBuilder sbp = new StringBuilder();
    sbp.Append("PREPARE insertplan (");
    for (int i = 0; i < 1000; i++)
    {
        sbp.Append("int,int,int,");
    }
    sbp.Remove(sbp.Length - 1, 1);
    sbp.Append(") AS INSERT INTO mesh(x, y, z) values");
    for (int i = 0; i < 1000; i++)
    {
        sbp.Append($"(${i*3 + 1},${i* 3 + 2},${i*3+ 3}),");
    }
    sbp.Remove(sbp.Length - 1, 1);
    TimeCalc.LogStartTime();

    using (var sm = new SqlManipulation(@"string", SqlType.PostgresQL))
    {
        sm.Init();
        sm.ExcuteNonQuery(sbp.ToString());
        foreach (var n in listbuilder)
        {
            sm.ExcuteNonQuery($"EXECUTE insertplan({n.ToString()})");
        }
    }
    TimeCalc.ShowTotalDuration();

    Console.ReadKey();
}
  • 使用Transaction

    在前面的基礎上,使用事務改造。每條語句插入1000條資料,每1000條作為一個事務,CPU 30%,磁碟34MB/S,耗時170.16s。
    改成100條一個事務,耗時167.78s。

  • 使用多執行緒

    還在前面的基礎上,使用多執行緒,每個執行緒建立一個連線,一個連線處理100條sql語句,每條sql語句插入1000條資料,以此種方式進行匯入。注意,連線字串可以將maxpoolsize設定大一些,我機器上實測,不設定會報連線超時錯誤。

CPU佔用率上到80%, 磁碟這裡需要注意,由於生成了非常多個Postgresql server程序,不好統計,累積算上應該有小100MB/S,最終時間,98.18s。

使用TPL,由於Parallel.ForEach返回的結果沒有檢查,可能導致時間不是很準確(偏小)。

var lists = new List<List<string>>();
var listt = new List<string>();
for (int i = 0; i < listbuilder.Count; i++)
{
    if (i % 1000 == 0)
    {
        listt = new List<string>();
        lists.Add(listt);
    }
    listt.Add(listbuilder[i].ToString());
}
TimeCalc.LogStartTime();
Parallel.ForEach(lists, (x) =>
{
    using (var sm = new SqlManipulation(@";string;MaxPoolSize=1000;", SqlType.PostgresQL))
    {
        sm.Init();
        foreach (var n in x)
        {
            sm.ExcuteNonQuery(n);
        }
    }
});
TimeCalc.ShowTotalDuration();
寫入方式 耗時(1000條/行)
COPY 107s
insert N/A
多行insert 160.58s
prepare多行insert 163.02s
事務多行insert 170.16s
多連線多行insert 98.18s

2. 寫入更新

資料實時更新,數量可能繼續增長,使用簡單的insert或者update是不行的,操作使用postgresql 9.5以後支援的新語法。

insert into mesh on conflict (x,y) do update set z = excluded.z

吐槽postgresql這麼晚才支援on conflict,mysql早有了...

在表中既有資料2500w+的前提下,重複往資料庫裡面寫這些資料。這裡只做多行插入更新測試,其他的結果應該差不多。

普通多行插入,耗時272.15s。
多執行緒插入的情況,耗時362.26s,CPU佔用率一度到了100%。猜測多連線的情況下,更新互鎖導致效能下降。

3. 讀取

  • Select方法

標準讀取還是用select方法,ADO.NET直接讀取。

使用adapter方式,耗時135.39s;使用dbreader方式,耗時71.62s。

  • Copy方法

postgresql的copy方法提供stdout binary方式,可以指定一條查詢進行輸出,耗時53.20s。

public List<(int x, int y, int z)> BulkIQueryNpg()
{
    List<(int, int, int)> dict = new List<(int, int, int)>();
    using (var reader = ((NpgsqlConnection)_conn).BeginBinaryExport("COPY (select x,y,z from mesh) TO STDOUT (FORMAT BINARY)"))
    {
        while (reader.StartRow() != -1)
        {
            var x = reader.Read<int>(NpgsqlDbType.Integer);
            var y = reader.Read<int>(NpgsqlDbType.Integer);
            var z = reader.Read<int>(NpgsqlDbType.Integer);
            dict.Add((x, y, z));
        }
    }
    return dict;
}

結論

總結測試結果,對於較多資料的情況下,可以得出以下結論:

  • 向空資料表匯入或者沒有重複資料表的匯入,優先使用COPY語句(為什麼有這個前提詳見P.S.);

  • 使用一條語句插入多條資料的方式能夠大幅度改善插入效能,可以實驗確定最優條數;

  • 使用transaction或者prepare插入,在本場景中優化效果不明顯;

  • 使用多連線/多執行緒操作,速度上有優勢,但是把握不好容易造成資源佔用率過高,連線數太大也容易影響其他應用;

  • 寫入更新是postgresql新特性,使用會造成一定的效能消耗(相對直接插入);

  • 讀取資料時,使用COPY語句能夠獲得較好的效能;

  • ado.net dbreader物件由於不需要fill的過程,讀取速度也較快(雖然趕不上COPY),也可優先考慮。

P.S.

  • 為什麼不用mysql

沒有最好的,只有最合適的,講道理我也是挺喜歡用mysql的。使用postgresql的原因主要在於:

postgresql匯入匯出的sql指令“copy”直接支援Binary模式到stdin和stdout,如果程式想直接整合,那麼用這個是比較方便的;相比較,mysql的sql語法(load data infile)並不支援到stdin或者stdout,匯出可以通過mysqldump.exe實現,匯入暫時沒什麼特別好的辦法(mysqlimport或許可以)。

  • 相較於mysql缺點

postgresql使用copy匯入的時候,如果目標表已經有資料,那麼在有主鍵約束的表遇到錯誤時,COPY自動終止,而且可能導致不完全插入的情況,換言之,是不支援匯入的過程進行update操作;mysql的load語法可以顯式指定出錯之後的動作(IGNORE/REPLACE),不會打斷匯入過程。

  • 其他

如果需要使用mysql從程式匯入資料,可以考慮先通過程式匯出到檔案,然後藉助檔案進行匯入,據說效率也要比insert高出不少。