From ce96abecb04fdb06917d31ce8d2c5f43f7fc9e40 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E5=85=89=E6=98=A5?= Date: Thu, 12 May 2022 10:36:48 +0800 Subject: [PATCH] update --- app.go | 36 ++++++++++++++++++++++++++++++++++++ app_test.go | 9 +++++++++ client.go | 2 -- client_test.go | 10 ++++++++++ version.go | 3 +++ 5 files changed, 58 insertions(+), 2 deletions(-) create mode 100644 app.go create mode 100644 app_test.go create mode 100644 client_test.go create mode 100644 version.go diff --git a/app.go b/app.go new file mode 100644 index 0000000..5886c28 --- /dev/null +++ b/app.go @@ -0,0 +1,36 @@ +package golog + +import "gorm.io/gorm" + +type App struct { + Gin gin // 框架日志 + Api api // 接口日志 + Pgsql *gorm.DB // pgsql数据库 + TableName string // 日志表名 +} + +// InitClientApi 接口实例化 +func (a *App) InitClientApi() { + if a.Pgsql == nil { + panic("驱动不正常") + } + if a.TableName == "" { + panic("表名不能为空") + } + a.Api.db = a.Pgsql + a.Api.tableName = a.TableName + a.Api.AutoMigrate() +} + +// InitClientGin 框架实例化 +func (a *App) InitClientGin() { + if a.Pgsql == nil { + panic("驱动不正常") + } + if a.TableName == "" { + panic("表名不能为空") + } + a.Gin.db = a.Pgsql + a.Gin.tableName = a.TableName + a.Gin.AutoMigrate() +} diff --git a/app_test.go b/app_test.go new file mode 100644 index 0000000..cfbde7a --- /dev/null +++ b/app_test.go @@ -0,0 +1,9 @@ +package golog + +import "testing" + +var a = App{} + +func TestApp(t *testing.T) { + a.Pgsql = nil +} diff --git a/client.go b/client.go index 0cc6cb0..9700eed 100644 --- a/client.go +++ b/client.go @@ -2,8 +2,6 @@ package golog import "gorm.io/gorm" -const Version = "1.0.0" - type Client struct { Gin gin // 框架日志 Api api // 接口日志 diff --git a/client_test.go b/client_test.go new file mode 100644 index 0000000..b69cdca --- /dev/null +++ b/client_test.go @@ -0,0 +1,10 @@ +package golog + +import "testing" + +var c = &Client{} + +func TestClient(t *testing.T) { + c = NewClientGin(nil, "") + c = NewClientApi(nil, "") +} diff --git a/version.go b/version.go new file mode 100644 index 0000000..74c0a35 --- /dev/null +++ b/version.go @@ -0,0 +1,3 @@ +package golog + +const Version = "1.0.1"