From 78f90975758be58efd1338cae9987ef9ae135b45 Mon Sep 17 00:00:00 2001 From: Ryan Vacek Date: Tue, 4 Jun 2019 16:16:36 -0600 Subject: [PATCH] server: include case with no fp provided --- server/viper.go | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/server/viper.go b/server/viper.go index 89d26fde..f4af7d99 100644 --- a/server/viper.go +++ b/server/viper.go @@ -87,6 +87,10 @@ func ConfigureFlagSet(applicationName string, f *pflag.FlagSet) { // // the CPU profile is created on the server's start func CreateCPUProfileFile(v *viper.Viper, fp *pflag.FlagSet, l log.Logger) { + if fp == nil { + return + } + flag := fp.Lookup("cpuprofile") if flag == nil { return @@ -110,6 +114,10 @@ func CreateCPUProfileFile(v *viper.Viper, fp *pflag.FlagSet, l log.Logger) { // the memory profile is created on the server's exit. // this function should be used within the application. func CreateMemoryProfileFile(v *viper.Viper, fp *pflag.FlagSet, l log.Logger) { + if fp == nil { + return + } + flag := fp.Lookup("memprofile") if flag == nil { return @@ -280,6 +288,7 @@ func Initialize(applicationName string, arguments []string, f *pflag.FlagSet, v return } - // CreateCPUProfileFile(v, f, logger) + CreateCPUProfileFile(v, f, logger) + return }