tablet/diamond-orm
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ensure that cfg.LogTo is properly closed on disconnect if its underlying type is a file

Browse Source
This commit is contained in:
☙◦ The Tablet ❀ GamerGirlandCo ◦❧ 2025-07-16 23:52:41 -04:00
parent bb5d6e6db3
commit 74f32a5352
Signed by: tablet
GPG Key ID: 924A5F6AF051E87C
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
diamond.go
View File

@ -20,7 +20,7 @@ const defaultKey = "default"
type Config struct { type Config struct {
DryRun bool // when true, queries will not run on the underlying database DryRun bool // when true, queries will not run on the underlying database
LogLevel slog.Level // controls the level of information logged; defaults to slog.LevelInfo if not set LogLevel slog.Level // controls the level of information logged; defaults to slog.LevelInfo if not set
LogTo io.Writer // where to write log output to; defaults to os.Stdout LogTo io.Writer // where to write log output to; defaults to os.Stdout
} }
type Engine struct { type Engine struct {
@ -121,6 +121,9 @@ func (e *Engine) logSql(msg, sql string) {
// Disconnect - closes and disposes of this Engine's connection pool. // Disconnect - closes and disposes of this Engine's connection pool.
func (e *Engine) Disconnect() { func (e *Engine) Disconnect() {
e.conn.Close() e.conn.Close()
if asFile, ok := e.cfg.LogTo.(*os.File); ok {
_ = asFile.Close()
}
} }
// Open - creates a new connection according to `connString` // Open - creates a new connection according to `connString`