为一个通道使用多个接收器

我正在尝试使用go-json-rest在 golang 中编写 REST 服务


该服务的目的只是将接收到的数据转换为 CSV 并记录下来。由于负载可能很重,我想使用 goroutines 进行日志记录。目前我已经创建了四个 LogWorkers(goroutine) 每个 goroutine 将把 CSV 记录到单独的文件中。


当我执行代码时,日志总是从最后一个 goroutine 触发。我看到在我的日志文件夹中创建了一个来自第四个例程的文件。


这是我的服务器代码


package main


import (

    "github.com/ant0ine/go-json-rest/rest"

    "log"

    "net/http"

    "strconv"

    "time"

)


const workerCount = 4

var evChannel = make(chan Event)

var workers = make([]*LogWorker, workerCount)

const maxLogFileSize = 100 // In MB

const maxLogFileBackups = 30

const maxLogFileAge = 5

const logFileName = "/home/sam/tmp/go_logs/event_"


func main() {

    // Initialize workers

    // Four workers is being created

    for i := 0; i < workerCount; i++ {

        var fileName = logFileName + strconv.Itoa(i)

        workers[i] = NewLogWorker(fileName, maxLogFileSize, maxLogFileBackups, maxLogFileAge)

        go workers[i].Work(evChannel)

    }    


    // Initialize REST API

    api := rest.NewApi()

    //api.Use(rest.DefaultDevStack...)

    api.Use(rest.DefaultCommonStack...)

    router, err := rest.MakeRouter(

        rest.Post("/events", StoreEvents),

    )

    if err != nil {

        log.Fatal(err)

    }

    api.SetApp(router)

    log.Fatal(http.ListenAndServe(":4545", api.MakeHandler()))

}


func StoreEvents(w rest.ResponseWriter, r *rest.Request) {

    event := Event{}

    err := r.DecodeJsonPayload(&event)

    if err != nil {

        rest.Error(w, err.Error(), http.StatusInternalServerError)

        return

    }

    // TODO : Add validation if needed

    // Add code to parse the request and add further information to event 

    // log.Println()

    select {

        case evChannel <- event:

        case <- time.After(5 * time.Second):

      // throw away the message, so sad

    }    

    // evChannel <- event

    //log.Println(Csv(event))

    w.WriteHeader(http.StatusOK)

}


请注意 event 是一个包含一些字符串字段的结构。SO 中已经有一个类似的问题。当我尝试在playground执行 goroutine 时,它仍然打印上次goroutine的值。提供的答案有一些等待。完成。由于我的工人需要连续运行,我认为我不能使用它。


请帮我找出为什么我的所有 goroutines (LogWorkers) 都没有被使用?


桃花长相依
浏览 163回答 1
1回答

潇湘沐

您正在每个 goroutine 中设置日志包的默认全局记录器输出。您可能想要做更多类似的事情:func (lw *LogWorker) Work(evChannel chan Event) {&nbsp; &nbsp; fmt.Println(lw.FileName)&nbsp; &nbsp; lg := log.New(&lumberjack.Logger {&nbsp; &nbsp; &nbsp; &nbsp; Filename:&nbsp; &nbsp;lw.FileName,&nbsp; &nbsp; &nbsp; &nbsp; MaxSize:&nbsp; &nbsp; lw.MaxSize,&nbsp; &nbsp; &nbsp; &nbsp; MaxBackups: lw.MaxBackups,&nbsp; &nbsp; &nbsp; &nbsp; MaxAge:&nbsp; &nbsp; &nbsp;lw.MaxAge,&nbsp; &nbsp; }, "", 0)&nbsp; &nbsp; for {&nbsp; &nbsp; &nbsp; &nbsp; event := <- evChannel&nbsp; &nbsp; &nbsp; &nbsp; lg.Println(Csv(event))&nbsp; &nbsp; }}这将为每个 goroutine 提供一个记录器。在您的版本中,您可能只有最后一个要执行(可能是最后一个 goroutine 生成,但不能保证)为了进一步改进,您可能还希望将 for 循环写为:for event := range evChannel {&nbsp; &nbsp; lg.Println(Csv(event))}这样,它会在通道关闭时终止 goroutine,而不是在关闭通道中的空值上旋转。 请参阅此处以供参考
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Go