package ctrl import ( "fmt" "log" "os" "path/filepath" "time" ) // ----- // Handler for receiving hello messages. func methodREQHello(proc process, message Message, node string) ([]byte, error) { data := fmt.Sprintf("%v, Received hello from %#v\n", time.Now().Format("Mon Jan _2 15:04:05 2006"), message.FromNode) fileName := message.FileName folderTree := filepath.Join(proc.configuration.SubscribersDataFolder, message.Directory, string(message.FromNode)) // Check if folder structure exist, if not create it. if _, err := os.Stat(folderTree); os.IsNotExist(err) { err := os.MkdirAll(folderTree, 0770) if err != nil { return nil, fmt.Errorf("error: failed to create errorLog directory tree %v: %v", folderTree, err) } er := fmt.Errorf("info: Creating subscribers data folder at %v", folderTree) proc.errorKernel.logDebug(er, proc.configuration) } // Open file and write data. file := filepath.Join(folderTree, fileName) //f, err := os.OpenFile(file, os.O_APPEND|os.O_RDWR|os.O_CREATE|os.O_SYNC, 0660) f, err := os.OpenFile(file, os.O_TRUNC|os.O_RDWR|os.O_CREATE|os.O_SYNC, 0660) if err != nil { er := fmt.Errorf("error: methodREQHello.handler: failed to open file: %v", err) return nil, er } defer f.Close() _, err = f.Write([]byte(data)) f.Sync() if err != nil { er := fmt.Errorf("error: methodEventTextLogging.handler: failed to write to file: %v", err) proc.errorKernel.errSend(proc, message, er, logWarning) } // -------------------------- // send the message to the procFuncCh which is running alongside the process // and can hold registries and handle special things for an individual process. proc.procFuncCh <- message ackMsg := []byte("confirmed from: " + node + ": " + fmt.Sprint(message.ID)) return ackMsg, nil } // --- // Handle the writing of error logs. func methodREQErrorLog(proc process, message Message, node string) ([]byte, error) { proc.metrics.promErrorMessagesReceivedTotal.Inc() // If it was a request type message we want to check what the initial messages // method, so we can use that in creating the file name to store the data. fileName, folderTree := selectFileNaming(message, proc) // Check if folder structure exist, if not create it. if _, err := os.Stat(folderTree); os.IsNotExist(err) { err := os.MkdirAll(folderTree, 0770) if err != nil { return nil, fmt.Errorf("error: failed to create errorLog directory tree %v: %v", folderTree, err) } er := fmt.Errorf("info: Creating subscribers data folder at %v", folderTree) proc.errorKernel.logDebug(er, proc.configuration) } // Open file and write data. file := filepath.Join(folderTree, fileName) f, err := os.OpenFile(file, os.O_APPEND|os.O_RDWR|os.O_CREATE|os.O_SYNC, 0660) if err != nil { er := fmt.Errorf("error: methodREQErrorLog.handler: failed to open file: %v", err) return nil, er } defer f.Close() _, err = f.Write(message.Data) f.Sync() if err != nil { er := fmt.Errorf("error: methodEventTextLogging.handler: failed to write to file: %v", err) proc.errorKernel.errSend(proc, message, er, logWarning) } ackMsg := []byte("confirmed from: " + node + ": " + fmt.Sprint(message.ID)) return ackMsg, nil } // --- // Handler to write directly to console. // This handler handles the writing to console. func methodREQToConsole(proc process, message Message, node string) ([]byte, error) { switch { case len(message.MethodArgs) > 0 && message.MethodArgs[0] == "stderr": log.Printf("* DEBUG: MethodArgs: got stderr \n") fmt.Fprintf(os.Stderr, "%v", string(message.Data)) fmt.Println() default: fmt.Fprintf(os.Stdout, "%v", string(message.Data)) fmt.Println() } ackMsg := []byte("confirmed from: " + node + ": " + fmt.Sprint(message.ID)) return ackMsg, nil } // --- // handler to be used as a reply method when testing requests. // We can then within the test listen on the testCh for received // data and validate it. // If no test is listening the data will be dropped. func methodREQTest(proc process, message Message, node string) ([]byte, error) { go func() { // Try to send the received message data on the test channel. If we // have a test started the data will be read from the testCh. // If no test is reading from the testCh the data will be dropped. select { case proc.errorKernel.testCh <- message.Data: default: // drop. } }() ackMsg := []byte("confirmed from: " + node + ": " + fmt.Sprint(message.ID)) return ackMsg, nil }