From 859cf9d91af0053f8ba0021887598ae2257a32a5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=88=98=E5=AD=A6=E6=B6=9B?= <liuxuetao@jwzg.com>
Date: Thu, 27 Apr 2023 18:22:28 +0800
Subject: [PATCH] omit() support called in chain

---
 chainable_api.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/chainable_api.go b/chainable_api.go
index 3dc7256e..adfc208b 100644
--- a/chainable_api.go
+++ b/chainable_api.go
@@ -178,9 +178,9 @@ func (db *DB) Omit(columns ...string) (tx *DB) {
 	tx = db.getInstance()
 
 	if len(columns) == 1 && strings.ContainsRune(columns[0], ',') {
-		tx.Statement.Omits = strings.FieldsFunc(columns[0], utils.IsValidDBNameChar)
+		tx.Statement.Omits = append(tx.Statement.Omits, strings.FieldsFunc(columns[0], utils.IsValidDBNameChar)...)
 	} else {
-		tx.Statement.Omits = columns
+		tx.Statement.Omits = append(tx.Statement.Omits, columns...)
 	}
 	return
 }