From 819e8041e3301ee5af408212c679f7258e0a0ab5 Mon Sep 17 00:00:00 2001 From: s7v7nislands Date: Wed, 3 Nov 2021 14:43:16 +0800 Subject: [PATCH] refactor code --- models/connection.go | 17 ++++++++--------- 1 file changed, 8 insertions(+), 9 deletions(-) diff --git a/models/connection.go b/models/connection.go index 89a8fd5ac..e753c558b 100644 --- a/models/connection.go +++ b/models/connection.go @@ -32,15 +32,14 @@ type SortField struct { func TruncateAll(conn *storage.Connection) error { return conn.Transaction(func(tx *storage.Connection) error { - if err := tx.RawQuery("TRUNCATE " + (&pop.Model{Value: User{}}).TableName()).Exec(); err != nil { - return err - } - if err := tx.RawQuery("TRUNCATE " + (&pop.Model{Value: RefreshToken{}}).TableName()).Exec(); err != nil { - return err - } - if err := tx.RawQuery("TRUNCATE " + (&pop.Model{Value: AuditLogEntry{}}).TableName()).Exec(); err != nil { - return err + + tables := []pop.Value{User{}, RefreshToken{}, AuditLogEntry{}, Instance{}} + for _, v := range tables { + if err := tx.RawQuery("TRUNCATE " + (&pop.Model{Value: v}).TableName()).Exec(); err != nil { + return err + } } - return tx.RawQuery("TRUNCATE " + (&pop.Model{Value: Instance{}}).TableName()).Exec() + + return nil }) }