func main() { var g *graph.DefaultGraph g, _ = utils.ReadFileToGraph("topology.txt") g = utils.ReadFile("topology.txt") fmt.Print(g.String()) labels := utils.GetLabelList(g) fmt.Println(labels) neighbors, _ := utils.GetNeighborMap(g) fmt.Println(neighbors) }
func main() { g, labels, err1 := utils.ReadFileToGraph("topology.txt") if err1 != nil { fmt.Println("ReadFileToGraph:") fmt.Println(err1) return } labels = labels // fmt.Println(g); bytes, err2 := dot.Marshal(g, "", "", " ", false) if err2 != nil { fmt.Println("dot.Marshall:") fmt.Println(err1) return } fmt.Println(string(bytes)) }
func main() { var filename string var address string var name string var nodeLabels []string var topology graph.Graph var fileErr error var input string fmt.Print("[create|connect]> ") fmt.Scanln(&input) if input == "store" { fmt.Print("enter topology filename > ") fmt.Scanln(&filename) fmt.Print("enter topology name > ") fmt.Scanln(&name) fmt.Print("enter address and port of database > ") fmt.Scanln(&address) if address == "" { fmt.Println("- no address specified, using default database") address = "128.153.144.171:6379" } topology, fileErr = utils.ReadFileToGraph(filename) if fileErr != nil { panic(fileErr) } nodeLabels = utils.GetLabelList(topology) numberOfNodes := len(nodeLabels) labelMap := utils.GetLabelMap(topology) rdb, err := database.NewRoutingDatabase(name, "tcp", address, labelMap) fmt.Println("Connecting to data base") if err != nil { panic(err) } fmt.Println("Connected") fmt.Println("Setting paths") var src, dest string for i := 0; i < numberOfNodes; i++ { for j := 0; j < numberOfNodes; j++ { if i != j { src = nodeLabels[i] dest = nodeLabels[j] paths, distance, _ := graph.Dijkstra(topology, src, dest) path := strings.Join(paths[1:], " ") rdb.SetPath(src, dest, fmt.Sprintf("%s %s | %d", src, path, int(distance[dest]))) } } } fmt.Println("Paths set, storing paths") rdb.StorePathsInDB() fmt.Println("Paths stored in data base") rdb.Disconnect() } else if input == "grab" { fmt.Print("enter topology name > ") fmt.Scanln(&name) fmt.Print("enter address and port of database > ") fmt.Scanln(&address) if address == "" { fmt.Println("- no address specified, using default database") address = "128.153.144.171:6379" } rdb, err := database.NewRoutingDatabaseFromDB(name, "tcp", address) if err != nil { panic(err) } fmt.Println("Connected") var src, dest string for { fmt.Print("Enter a first node > ") fmt.Scanln(&input) src = input fmt.Print("Enter a second node > ") fmt.Scanln(&input) dest = input path, DBerr := rdb.GetPathFromDB(src, dest) if DBerr != nil { panic(DBerr) } fmt.Println("The shortest path is: ", path) } rdb.Disconnect() } else { fmt.Println("invalid input program terminated") } }
func main() { var ipAddress, recordName, filename, input string var db database.RoutingDatabase var DBerr error color.Set(color.FgHiGreen, color.Bold, color.BlinkSlow) fmt.Println("\nWelcome to the NRA System, written in golang\n") color.Unset() fmt.Println("Enter IP address and port of the database server you'd like to use") fmt.Println("If none is entered, the default server will be used (testing only)\n") color.Unset() color.Set(color.FgWhite) fmt.Print("IP address and port >> ") color.Unset() fmt.Scanln(&ipAddress) if ipAddress == "" { color.Set(color.FgHiBlue) fmt.Println("No ip address selected, using default database") color.Unset() ipAddress = "128.153.144.171:6379" } color.Set(color.FgWhite) fmt.Print("Enter the name of the database record you wish to use >> ") color.Unset() fmt.Scanln(&recordName) exists, DBerr := database.TopologyExists(recordName, "tcp", ipAddress) if DBerr != nil { panic(DBerr) } if exists { color.Set(color.FgHiBlue) fmt.Println("This record exists!\n") color.Unset() db, DBerr = database.ConnectToDatabase(recordName, "tcp", ipAddress) topo, TPerr := db.GetTopology() //fmt.Printf("CLI: gb.GetTopo[%v]: %v\n", TPerr, topo) if DBerr != nil { panic(DBerr) } queryPaths(db) } else { color.Set(color.FgHiBlue) fmt.Println("This record does not exist. Please check your spelling") color.Set(color.FgWhite) fmt.Print("Would you like to create a record by this name? [Y/N] >> ") color.Unset() fmt.Scanln(&input) if input == "Y" || input == "y" { color.Set(color.FgWhite) fmt.Print("Give me the name of a topology file >> ") color.Unset() fmt.Scanln(&filename) g, labels, Uerror := utils.ReadFileToGraph(filename) if Uerror != nil { panic(Uerror) } revLabels := utils.GetRevLabels(labels) paths := path.DijkstraAllPaths(g) pathMap := algorithms.ConvertAllPaths(g, paths) realPathMap := make(map[int]map[int]string) for k := range pathMap { smallMap := pathMap[k] newMap := make(map[int]string) for k2 := range smallMap { stringForPath := smallMap[k2].PathString(labels) newMap[k2] = stringForPath } realPathMap[k] = newMap } //fmt.Printf("CLI: pathMap: %v\n", pathMap) //fmt.Printf("CLI: realPathMap: %v\n", realPathMap) db, DBerr = database.NewRoutingDatabase(recordName, "tcp", ipAddress, revLabels, realPathMap, utils.GetNeighborMap(g)) topo, TPerr := db.GetTopology() //fmt.Printf("CLI: db.GetTopo[%v]: %v\n", TPerr, topo) if DBerr != nil { panic(DBerr) } color.Set(color.FgHiGreen, color.Bold) fmt.Println("Database created correctly!") color.Unset() queryPaths(db) } else { color.Set(color.FgHiGreen, color.Bold) fmt.Println("Thank you for using golang NRA!") color.Unset() } } }