122 lines
2.5 KiB
Go
122 lines
2.5 KiB
Go
package p2p
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"git.sr.ht/~akilan1999/p2p-rendering-computation/config"
|
|
"io/ioutil"
|
|
"os"
|
|
"time"
|
|
)
|
|
|
|
// Get IP table Data
|
|
|
|
type IpAddresses struct {
|
|
IpAddress []IpAddress `json:"ip_address"`
|
|
}
|
|
|
|
type IpAddress struct {
|
|
Ipv4 string `json:"ipv4"`
|
|
Ipv6 string `json:"ipv6"`
|
|
Latency time.Duration `json:"latency"`
|
|
Download float64 `json:"download"`
|
|
Upload float64 `json:"upload"`
|
|
}
|
|
|
|
// ReadIpTable Read data from Ip tables from json file
|
|
func ReadIpTable()(*IpAddresses ,error){
|
|
// Get Path from config
|
|
config, err := config.ConfigInit()
|
|
if err != nil {
|
|
return nil,err
|
|
}
|
|
jsonFile, err := os.Open(config.IPTable)
|
|
// if we os.Open returns an error then handle it
|
|
if err != nil {
|
|
return nil,err
|
|
}
|
|
|
|
|
|
// defer the closing of our jsonFile so that we can parse it later on
|
|
defer jsonFile.Close()
|
|
|
|
// read our opened xmlFile as a byte array.
|
|
byteValue, _ := ioutil.ReadAll(jsonFile)
|
|
|
|
// we initialize our Users array
|
|
var ipAddresses IpAddresses
|
|
|
|
// we unmarshal our byteArray which contains our
|
|
// jsonFile's content into 'users' which we defined above
|
|
json.Unmarshal(byteValue, &ipAddresses)
|
|
|
|
return &ipAddresses, nil
|
|
}
|
|
|
|
// WriteIpTable Write to IP table json file
|
|
func (i *IpAddresses) WriteIpTable() error {
|
|
file, err := json.MarshalIndent(i, "", " ")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Get Path from config
|
|
config, err := config.ConfigInit()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = ioutil.WriteFile(config.IPTable, file, 0644)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// PrintIpTable Print Ip table data for Cli
|
|
func PrintIpTable() error {
|
|
table, err := ReadIpTable()
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for i := 0; i < len(table.IpAddress); i++ {
|
|
fmt.Printf("\nIP Address: %s\nLatency: %s\n-----------" +
|
|
"-----------------\n",table.IpAddress[i].Ipv4,
|
|
table.IpAddress[i].Latency)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// RemoveDuplicates This is a temporary fix current functions failing to remove
|
|
// Duplicate IP addresses from local IP table
|
|
func RemoveDuplicates() error {
|
|
table, err := ReadIpTable()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var NoDuplicates IpAddresses
|
|
for i, _:= range table.IpAddress {
|
|
Exists := false
|
|
for k := range NoDuplicates.IpAddress {
|
|
if NoDuplicates.IpAddress[k].Ipv4 == table.IpAddress[i].Ipv4 {
|
|
Exists = true
|
|
break
|
|
}
|
|
}
|
|
|
|
if Exists {
|
|
continue
|
|
}
|
|
NoDuplicates.IpAddress = append(NoDuplicates.IpAddress, table.IpAddress[i])
|
|
}
|
|
|
|
if err := NoDuplicates.WriteIpTable(); err != nil {
|
|
return nil
|
|
}
|
|
|
|
return nil
|
|
} |