Go語言 如何配制 高性能sql.DB 的
有很多教程是關(guān)于Go的sql.DB
類型和如何使用它來執(zhí)行SQL數(shù)據(jù)庫查詢的。但大多數(shù)內(nèi)容都沒有講述SetMaxOpenConns()
,?SetMaxIdleConns()
?和?SetConnMaxLifetime()
方法, 您可以使用它們來配置sql.DB
的行為并改變其性能。
開放和空閑連接
一個sql.DB對象就是一個數(shù)據(jù)庫連接池,它包含“正在用”和“空閑的”連接。一個正在用的連接指的是,你正用它來執(zhí)行數(shù)據(jù)庫任務(wù),例如執(zhí)行SQL語句或行查詢。當(dāng)任務(wù)完成連接就是空閑的。
當(dāng)您創(chuàng)建sql.DB執(zhí)行數(shù)據(jù)庫任務(wù)時,它將首先檢查連接池中是否有可用的空閑連接。如果有可用的連接,那么Go將重用現(xiàn)有連接,并在執(zhí)行任務(wù)期間將其標記為正在使用。如果池中沒有空閑連接,而您需要一個空閑連接,那么Go將創(chuàng)建一個新的連接。
SetMaxOpenConns方法
默認情況下,在同一時間打開連接的數(shù)量是沒有限制(包含使用中+空閑)。但你可以通過SetMaxOpenConns()
方法實現(xiàn)自定義限制,如下所示:

在這個示例代碼中,連接池現(xiàn)在有5個并發(fā)打開的連接數(shù)。如果所有5個連接都已經(jīng)被標記為正在使用,并且需要另一個新的連接,那么應(yīng)用程序?qū)⒈黄鹊却钡?個連接中的一個被釋放并變?yōu)榭臻e。
為了說明更改MaxOpenConns
的影響,我運行了一個基準測試,將最大打開連接數(shù)設(shè)置為1、2、5、10和無限?;鶞蕼y試在PostgreSQL
數(shù)據(jù)庫上執(zhí)行并行的INSERT
語句,您可以在這里找到代碼。測試結(jié)果:
BenchmarkMaxOpenConns1-8 ? ? ? ? ? ? ? ? 500 ? ? ? 3129633 ns/op ? ? ? ? 478 B/op ? ? ? ? 10 allocs/op
BenchmarkMaxOpenConns2-8 ? ? ? ? ? ? ? ?1000 ? ? ? 2181641 ns/op ? ? ? ? 470 B/op ? ? ? ? 10 allocs/op
BenchmarkMaxOpenConns5-8 ? ? ? ? ? ? ? ?2000 ? ? ? ?859654 ns/op ? ? ? ? 493 B/op ? ? ? ? 10 allocs/op
BenchmarkMaxOpenConns10-8 ? ? ? ? ? ? ? 2000 ? ? ? ?545394 ns/op ? ? ? ? 510 B/op ? ? ? ? 10 allocs/op
BenchmarkMaxOpenConnsUnlimited-8 ? ? ? ?2000 ? ? ? ?531030 ns/op ? ? ? ? 479 B/op ? ? ? ? ?9 allocs/op
PASS
對于這個基準測試,我們可以看到,允許打開的連接越多,在數(shù)據(jù)庫上執(zhí)行INSERT
操作所花費的時間就越少(打開的連接數(shù)為1時,執(zhí)行速度3129633ns/op,而無限連接:531030ns/op——大約快了6倍)。這是因為允許打開的連接越多,可以并發(fā)執(zhí)行的數(shù)據(jù)庫查詢就越多。
SetMaxIdleConns方法
默認情況下,sql.DB允許連接池中最多保留2個空閑連接。你可以通過SetMaxIdleConns()
方法改變它,如下所示:

從理論上講,允許池中有更多的空閑連接將提高性能,因為這樣就不太可能從頭開始建立新連接——因此有助于提升數(shù)據(jù)庫性能。
讓我們來看看相同的基準測試,最大空閑連接設(shè)置為none, 1,2,5和10:
BenchmarkMaxIdleConnsNone-8 ? ? ? ? ?300 ? ? ? 4567245 ns/op ? ? ? 58174 B/op ? ? ? ?625 allocs/op
BenchmarkMaxIdleConns1-8 ? ? ? ? ? ?2000 ? ? ? ?568765 ns/op ? ? ? ?2596 B/op ? ? ? ? 32 allocs/op
BenchmarkMaxIdleConns2-8 ? ? ? ? ? ?2000 ? ? ? ?529359 ns/op ? ? ? ? 596 B/op ? ? ? ? 11 allocs/op
BenchmarkMaxIdleConns5-8 ? ? ? ? ? ?2000 ? ? ? ?506207 ns/op ? ? ? ? 451 B/op ? ? ? ? ?9 allocs/op
BenchmarkMaxIdleConns10-8 ? ? ? ? ? 2000 ? ? ? ?501639 ns/op ? ? ? ? 450 B/op ? ? ? ? ?9 allocs/op
PASS
當(dāng)MaxIdleConns
設(shè)置為none
時,必須為每個INSERT
從頭創(chuàng)建一個新的連接,我們可以從基準測試中看到,平均運行時和內(nèi)存使用量相對較高。
只允許保留和重用一個空閑連接對基準測試影響特別明顯——它將平均運行時間減少了大約8倍,內(nèi)存使用量減少了大約20倍。繼續(xù)增加空閑連接池的大小會使性能變得更好,盡管改進并不明顯。
那么,您應(yīng)該維護一個大的空閑連接池嗎?答案取決于應(yīng)用程序。重要的是要意識到保持空閑連接是有代價的—它占用了可以用于應(yīng)用程序和數(shù)據(jù)庫的內(nèi)存。
還有一種可能是,如果一個連接空閑時間太長,那么它可能會變得不可用。例如,MySQL的wait_timeout
設(shè)置將自動關(guān)閉任何8小時(默認)內(nèi)未使用的連接。
當(dāng)發(fā)生這種情況時,sql.DB
會優(yōu)雅地處理它。壞連接將自動重試兩次,然后放棄,此時Go將該連接從連接池中刪除,并創(chuàng)建一個新的連接。因此,將MaxIdleConns
設(shè)置得太大可能會導(dǎo)致連接變得不可用,與空閑連接池更小(使用更頻繁的連接更少)相比,會占有更多的資源。所以,如果你很可能很快就會再次使用,你只需保持一個空閑的連接。
最后要指出的是,MaxIdleConns
應(yīng)該總是小于或等于MaxOpenConns
.
。Go強制執(zhí)行此操作,并在必要時自動減少MaxIdleConns
。
SetConnMaxLifetime方法
現(xiàn)在讓我們看看SetConnMaxLifetime()
方法,它設(shè)置連接可重用的最大時間長度。如果您的SQL數(shù)據(jù)庫也實現(xiàn)了最大連接生命周期,或者—例如—您希望方便地在負載均衡器后交換數(shù)據(jù)庫,那么這將非常有用。
你可以這樣使用它:

在這個例子中,所有的連接都將在創(chuàng)建后1小時“過期”,并且在過期后無法重用。但注意:
- 這并不能保證連接將在池中存在整整一個小時;很有可能,由于某些原因,連接變得不可用,并在此之前自動關(guān)閉。
- 一個連接在創(chuàng)建后一個多小時仍然可以被使用——它只是在這個時間之后不能被重用。
- 這不是空閑超時。連接將在第一次創(chuàng)建后1小時過期——而不是在最后一次空閑后1小時。
- 每隔一秒自動運行一次清理操作,從連接池中刪除“過期”的連接。
從理論上講,ConnMaxLifetime
越短,連接過期的頻率就越高——因此,需要從頭創(chuàng)建連接的頻率就越高。為了說明這一點,我運行了將ConnMaxLifetime
設(shè)置為100ms、200ms、500ms、1000ms和無限(永遠重用)的基準測試,默認設(shè)置為無限打開連接和2個空閑連接。
這些時間段顯然比您在大多數(shù)應(yīng)用程序中使用的時間要短得多,但它們有助于很好地說明行為。
BenchmarkConnMaxLifetime100-8 ? ? ? ? ? ? ? 2000 ? ? ? ?637902 ns/op ? ? ? ?2770 B/op ? ? ? ? 34 allocs/op
BenchmarkConnMaxLifetime200-8 ? ? ? ? ? ? ? 2000 ? ? ? ?576053 ns/op ? ? ? ?1612 B/op ? ? ? ? 21 allocs/op
BenchmarkConnMaxLifetime500-8 ? ? ? ? ? ? ? 2000 ? ? ? ?558297 ns/op ? ? ? ? 913 B/op ? ? ? ? 14 allocs/op
BenchmarkConnMaxLifetime1000-8 ? ? ? ? ? ? ?2000 ? ? ? ?543601 ns/op ? ? ? ? 740 B/op ? ? ? ? 12 allocs/op
BenchmarkConnMaxLifetimeUnlimited-8 ? ? ? ? 3000 ? ? ? ?532789 ns/op ? ? ? ? 412 B/op ? ? ? ? ?9 allocs/op
PASS
在這些特定的基準測試中,我們可以看到,與無限生存期相比,在100ms生存期時內(nèi)存使用量增加了3倍以上,而且每個INSERT
的平均運行時也稍微長一些。
如果您在代碼中設(shè)置了ConnMaxLifetime
,那么一定要記住連接將過期(隨后重新創(chuàng)建)的頻率。例如,如果您總共有100個連接,而ConnMaxLifetime
為1分鐘,那么您的應(yīng)用程序可能每秒鐘殺死和重新創(chuàng)建1.67個連接(平均值)。您不希望這個頻率太大,最終會阻礙性能,而不是提高性能。
連接數(shù)量超出
最后,如果不說明超過數(shù)據(jù)庫連接數(shù)量的硬限制將會發(fā)生什么,那么本文就不完整了。為了說明這一點,我將修改postgresql.conf
文件,這樣總共只允許5個連接(默認是100個)…

然后在無限連接的情況下重新運行基準測試……
BenchmarkMaxOpenConnsUnlimited-8 ? ?--- FAIL: BenchmarkMaxOpenConnsUnlimited-8
main_test.go:14: pq: sorry, too many clients already
main_test.go:14: pq: sorry, too many clients already
main_test.go:14: pq: sorry, too many clients already
FAIL
一旦達到5個連接的硬限制,數(shù)據(jù)庫驅(qū)動程序(pq)立即返回一個太多客戶端連接的錯誤消息,而無法完成INSERT
。為了防止這個錯誤,我們需要將sql.DB中打開連接的最大總數(shù)(正在使用的+空閑的)設(shè)置為低于5。像這樣:

現(xiàn)在,sql.DB在任何時候最多只能創(chuàng)建3個連接,基準測試運行時應(yīng)該不會出現(xiàn)任何錯誤。
但是這樣做需要注意:當(dāng)達到開放連接數(shù)限制,并且所有連接都在使用時,應(yīng)用程序需要執(zhí)行的任何新的數(shù)據(jù)庫任務(wù)都將被迫等待,直到連接標記為空閑。
例如,在web應(yīng)用程序的上下文中,用戶的HTTP請求看起來會“掛起”,甚至在等待數(shù)據(jù)庫任務(wù)運行時可能會超時。
為了減輕這種情況,你應(yīng)該始終在一個上下文中傳遞。在調(diào)用數(shù)據(jù)庫時,啟用上下文的方法(如ExecContext()
),使用固定的、快速的超時上下文對象。
總結(jié)
1、根據(jù)經(jīng)驗,應(yīng)該顯式設(shè)置MaxOpenConns
值。這應(yīng)該小于數(shù)據(jù)庫和基礎(chǔ)設(shè)施對連接數(shù)量的硬性限制。
2、一般來說,更高的MaxOpenConns
和MaxIdleConns
值將帶來更好的性能。但你應(yīng)該注意到效果是遞減的,連接池空閑連接太多(連接沒有被重用,最終會變壞)實際上會導(dǎo)致性能下降。
3、為了降低上面第2點帶來的風(fēng)險,您可能需要設(shè)置一個相對較短的ConnMaxLifetime
。但你也不希望它太短,導(dǎo)致連接被殺死或不必要地頻繁重建。
4、MaxIdleConns
應(yīng)該總是小于或等于MaxOpenConns
。
對于中小型web應(yīng)用程序,我通常使用以下設(shè)置作為起點,然后根據(jù)實際吞吐量水平的負載測試結(jié)果進行優(yōu)化。

文章鏈接:https://www.jb51.net/article/231582.htm