Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
FLIPPED-AURORA
gin-vue-admin
提交
ba98288a
G
gin-vue-admin
项目概览
FLIPPED-AURORA
/
gin-vue-admin
大约 1 年 前同步成功
通知
333
Star
18155
Fork
5506
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
G
gin-vue-admin
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
ba98288a
编写于
5月 30, 2022
作者:
Sliver_Horn
提交者:
GitHub
5月 30, 2022
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
feature: viper 读取gin自带的环境变量GIN_MODE,用于区分开发,测试,生产三个环境 #1098 (#1105)
上级
70f2a94c
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
38 addition
and
22 deletion
+38
-22
server/core/internal/constant.go
server/core/internal/constant.go
+9
-0
server/core/viper.go
server/core/viper.go
+29
-16
server/utils/constant.go
server/utils/constant.go
+0
-6
未找到文件。
server/core/internal/constant.go
0 → 100644
浏览文件 @
ba98288a
package
internal
const
(
ConfigEnv
=
"GVA_CONFIG"
ConfigDefaultFile
=
"config.yaml"
ConfigTestFile
=
"config.test.yaml"
ConfigDebugFile
=
"config.debug.yaml"
ConfigReleaseFile
=
"config.release.yaml"
)
server/core/viper.go
浏览文件 @
ba98288a
...
...
@@ -3,6 +3,8 @@ package core
import
(
"flag"
"fmt"
"github.com/flipped-aurora/gin-vue-admin/server/core/internal"
"github.com/gin-gonic/gin"
"os"
"path/filepath"
"time"
...
...
@@ -11,31 +13,42 @@ import (
"github.com/flipped-aurora/gin-vue-admin/server/global"
_
"github.com/flipped-aurora/gin-vue-admin/server/packfile"
"github.com/flipped-aurora/gin-vue-admin/server/utils"
"github.com/fsnotify/fsnotify"
"github.com/spf13/viper"
)
// Viper //
// 优先级: 命令行 > 环境变量 > 默认值
// Author [SliverHorn](https://github.com/SliverHorn)
func
Viper
(
path
...
string
)
*
viper
.
Viper
{
var
config
string
if
len
(
path
)
==
0
{
flag
.
StringVar
(
&
config
,
"c"
,
""
,
"choose config file."
)
flag
.
Parse
()
if
config
==
""
{
// 优先级: 命令行 > 环境变量 > 默认值
if
configEnv
:=
os
.
Getenv
(
utils
.
ConfigEnv
);
configEnv
==
""
{
config
=
utils
.
ConfigFile
fmt
.
Printf
(
"您正在使用config的默认值,config的路径为%v
\n
"
,
utils
.
ConfigFile
)
}
else
{
if
config
==
""
{
// 判断命令行参数是否为空
if
configEnv
:=
os
.
Getenv
(
internal
.
ConfigEnv
);
configEnv
==
""
{
// 判断 internal.ConfigEnv 常量存储的环境变量是否为空
switch
gin
.
Mode
()
{
case
gin
.
DebugMode
:
config
=
internal
.
ConfigDefaultFile
fmt
.
Printf
(
"您正在使用gin模式的%s环境名称,config的路径为%s
\n
"
,
gin
.
EnvGinMode
,
internal
.
ConfigDefaultFile
)
case
gin
.
ReleaseMode
:
config
=
internal
.
ConfigReleaseFile
fmt
.
Printf
(
"您正在使用gin模式的%s环境名称,config的路径为%s
\n
"
,
gin
.
EnvGinMode
,
internal
.
ConfigReleaseFile
)
case
gin
.
TestMode
:
config
=
internal
.
ConfigTestFile
fmt
.
Printf
(
"您正在使用gin模式的%s环境名称,config的路径为%s
\n
"
,
gin
.
EnvGinMode
,
internal
.
ConfigTestFile
)
}
}
else
{
// internal.ConfigEnv 常量存储的环境变量不为空 将值赋值于config
config
=
configEnv
fmt
.
Printf
(
"您正在使用
GVA_CONFIG环境变量,config的路径为%v
\n
"
,
config
)
fmt
.
Printf
(
"您正在使用
%s环境变量,config的路径为%s
\n
"
,
internal
.
ConfigEnv
,
config
)
}
}
else
{
fmt
.
Printf
(
"您正在使用命令行的-c参数传递的值,config的路径为%
v
\n
"
,
config
)
}
else
{
// 命令行参数不为空 将值赋值于config
fmt
.
Printf
(
"您正在使用命令行的-c参数传递的值,config的路径为%
s
\n
"
,
config
)
}
}
else
{
}
else
{
// 函数传递的可变参数的第一个值赋值于config
config
=
path
[
0
]
fmt
.
Printf
(
"您正在使用func Viper()传递的值,config的路径为%
v
\n
"
,
config
)
fmt
.
Printf
(
"您正在使用func Viper()传递的值,config的路径为%
s
\n
"
,
config
)
}
v
:=
viper
.
New
()
...
...
@@ -49,15 +62,15 @@ func Viper(path ...string) *viper.Viper {
v
.
OnConfigChange
(
func
(
e
fsnotify
.
Event
)
{
fmt
.
Println
(
"config file changed:"
,
e
.
Name
)
if
err
:
=
v
.
Unmarshal
(
&
global
.
GVA_CONFIG
);
err
!=
nil
{
if
err
=
v
.
Unmarshal
(
&
global
.
GVA_CONFIG
);
err
!=
nil
{
fmt
.
Println
(
err
)
}
})
if
err
:
=
v
.
Unmarshal
(
&
global
.
GVA_CONFIG
);
err
!=
nil
{
if
err
=
v
.
Unmarshal
(
&
global
.
GVA_CONFIG
);
err
!=
nil
{
fmt
.
Println
(
err
)
}
// root 适配性
// 根据root位置去找到对应迁移位置,保证root路径有效
//
root 适配性
根据root位置去找到对应迁移位置,保证root路径有效
global
.
GVA_CONFIG
.
AutoCode
.
Root
,
_
=
filepath
.
Abs
(
".."
)
global
.
BlackCache
=
local_cache
.
NewCache
(
local_cache
.
SetDefaultExpire
(
time
.
Second
*
time
.
Duration
(
global
.
GVA_CONFIG
.
JWT
.
ExpiresTime
)),
...
...
server/utils/constant.go
已删除
100644 → 0
浏览文件 @
70f2a94c
package
utils
const
(
ConfigEnv
=
"GVA_CONFIG"
ConfigFile
=
"config.yaml"
)
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录