Function main

pkg/logwrite/logwrite.go:111–170

()

}

func main() {
	socketPath := flag.String("socket", "/var/run/memlogdq.sock", "memlogd log query socket")
	logDir := flag.String("log-dir", "/var/log", "Directory containing log files")
	maxLogFiles := flag.Int("max-log-files", 10, "Maximum number of rotated log files before deletion")
	maxLogSize := flag.Int("max-log-size", mb, "Maximum size of a log file before rotation")
	flag.Parse()

	addr := net.UnixAddr{
		Name: *socketPath,
		Net:  "unix",
	}
	conn, err := net.DialUnix("unix", nil, &addr)
	if err != nil {
		log.Fatal(err)
	}
	defer conn.Close()

	n, err := conn.Write([]byte{logDumpFollow})
	if err != nil || n < 1 {
		log.Fatalf("Failed to write request to memlogd socket: %v", err)
	}

	// map of service name to active log file
	logs := make(map[string]*LogFile)

	var msg LogMessage
	decoder := json.NewDecoder(conn)
	for {
		if err := decoder.Decode(&msg); err != nil {
			log.Println(err)
			continue
		}
		if strings.HasPrefix(msg.Source, "logwrite") {
			// don't log our own output in a loop
			continue
		}

		var logF *LogFile
		var ok bool
		if logF, ok = logs[msg.Source]; !ok {
			logF, err = NewLogFile(*logDir, msg.Source)
			if err != nil {
				log.Printf("Failed to create log file %s: %v", msg.Source, err)
				continue
			}
			logs[msg.Source] = logF
		}
		if err = logF.Write(&msg); err != nil {
			log.Printf("Failed to write to log file %s: %v", msg.Source, err)
			if err := logF.Close(); err != nil {
				log.Printf("Failed to close log file %s: %v", msg.Source, err)
			}
			delete(logs, msg.Source)
			continue
		}
		if logF.BytesWritten > *maxLogSize {
			logF.Rotate(*maxLogFiles)
		}