Commit aafa7ab8 authored by Sushant Mahajan's avatar Sushant Mahajan

allowed provision to disable debugging

parent 933059b0
...@@ -15,9 +15,6 @@ import ( ...@@ -15,9 +15,6 @@ import (
// Logger // Logger
var Info *log.Logger var Info *log.Logger
// Flag for enabling/disabling logging functionality
var DEBUG = true
var rft *raft.Raft var rft *raft.Raft
type AppendEntries struct{} type AppendEntries struct{}
...@@ -60,12 +57,11 @@ func initInterServerCommunication(server *raft.ServerConfig, rft *raft.Raft, ch ...@@ -60,12 +57,11 @@ func initInterServerCommunication(server *raft.ServerConfig, rft *raft.Raft, ch
} }
// Initialize Logger // Initialize Logger
func initLogger(serverId int) { func initLogger(serverId int, toDebug bool) {
// Logger Initializaion // Logger Initializaion
if !DEBUG { if !toDebug {
Info = log.New(ioutil.Discard, "INFO: ", log.Ldate|log.Ltime|log.Lshortfile) Info = log.New(ioutil.Discard, "INFO: ", log.Ldate|log.Ltime|log.Lshortfile)
} else { } else {
Info = log.New(os.Stdout, "INFO: ", log.Ldate|log.Ltime|log.Lshortfile) Info = log.New(os.Stdout, "INFO: ", log.Ldate|log.Ltime|log.Lshortfile)
} }
...@@ -96,7 +92,11 @@ func main() { ...@@ -96,7 +92,11 @@ func main() {
Info.Println("argument ", os.Args[1], "is not string") Info.Println("argument ", os.Args[1], "is not string")
} }
initLogger(sid) if len(os.Args) > 3 {
initLogger(sid, true)
} else {
initLogger(sid, false)
}
Info.Println("Starting") Info.Println("Starting")
serverCount, err2 := strconv.Atoi((os.Args[2])) serverCount, err2 := strconv.Atoi((os.Args[2]))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment