戈朗。帮助优化一段代码

Question

我有代码if config.Process.TraceLog != "" {    f, err := os.OpenFile(config.Process.TraceLog, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)    if err != nil {        fmt.Println("error opening log file: %v", err)        os.Exit(1)    }    Trace.SetOutput(f)}if config.Process.InfoLog != "" {    f, err := os.OpenFile(config.Process.InfoLog, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)    if err != nil {        fmt.Println("error opening log file: %v", err)        os.Exit(1)    }    Info.SetOutput(f)}if config.Process.WarningLog != "" {    f, err := os.OpenFile(config.Process.WarningLog, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)    if err != nil {        fmt.Println("error opening log file: %v", err)        os.Exit(1)    }    Warning.SetOutput(f)}if config.Process.ErrorLog != "" {    f, err := os.OpenFile(config.Process.ErrorLog, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)    if err != nil {        fmt.Println("error opening log file: %v", err)        os.Exit(1)    }    Error.SetOutput(f)}缩短它的好方法是什么？我不想重复相同的非常相似的块 4 次。如何使它成为一个循环？我尝试使用文件名和记录器对象创建地图，但不知道如何正确发布参考

Answer 1

这就是我最终做到的

m := map[string]*log.Logger{

    config.Process.TraceLog:   Trace,

    config.Process.InfoLog:    Info,

    config.Process.WarningLog: Warning,

    config.Process.ErrorLog:   Error,

}

for filepath, l := range m {

    if filepath != "" {

        f, err := os.OpenFile(filepath, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)

        if err != nil {

            fmt.Println("error opening log file: %v", err)

            os.Exit(1)

        }

        l.SetOutput(f)

    }

}

Answer 2

这样的事情可能会起作用（您可能需要调整界面中的函数类型或完全删除界面）：

setLogIfNotEmpty(config.Process.TraceLog, Trace)

setLogIfNotEmpty(config.Process.InfoLog, Info)

setLogIfNotEmpty(config.Process.WarningLog, Warning)

setLogIfNotEmpty(config.Process.ErrorLog, Error)

interface SetOutputer {

    SetOutput(*os.File)

}

func setLogIfNotEmpty(file string, log SetOutputer) {

    if file != "" {

        f, err := os.OpenFile(file, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)

        if err != nil {

            fmt.Println("error opening log file: %v", err)

            os.Exit(1)

        }

        log.SetOutput(f)

    }

}