Skip to content

Commit 2aafcc6

Browse files
committed
Minor fixes :P
1 parent eef9713 commit 2aafcc6

File tree

2 files changed

+38
-48
lines changed

2 files changed

+38
-48
lines changed

backend/cmd/main.go

+2-7
Original file line numberDiff line numberDiff line change
@@ -256,16 +256,11 @@ func main() {
256256
}
257257
}
258258

259-
boardIps := make([]net.IP, 0)
260-
for boardName, _ := range adj.Info.BoardIds {
259+
boardIps := make([]net.IP, 0, len(adj.Info.BoardIds))
260+
for boardName := range adj.Info.BoardIds {
261261
boardIps = append(boardIps, net.ParseIP(adj.Info.Addresses[boardName]))
262262
}
263263

264-
boardIpsStr := make([]string, 0)
265-
for boardName, _ := range adj.Info.BoardIds {
266-
boardIpsStr = append(boardIpsStr, adj.Info.Addresses[boardName])
267-
}
268-
269264
filter := getFilter(boardIps, net.ParseIP(adj.Info.Addresses[BACKEND]), adj.Info.Ports[UDP])
270265
trace.Warn().Str("filter", filter).Msg("filter")
271266
err = source.SetBPFFilter(filter)

backend/internal/adj/adj.go

+36-41
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,11 @@ package adj
22

33
import (
44
"encoding/json"
5+
"os"
6+
"path"
7+
58
"github.com/HyperloopUPV-H8/h9-backend/internal/utils"
69
"github.com/go-git/go-git/v5"
7-
"os"
810
)
911

1012
const (
@@ -24,27 +26,12 @@ func NewADJ() (*ADJ, error) {
2426
return nil, err
2527
}
2628

27-
// TESTING
28-
for key, value := range infoJSON.Ports {
29-
println("Ports: ", key, value)
30-
}
31-
for key, value := range infoJSON.Addresses {
32-
println("Addresses: ", key, value)
29+
var info Info = Info{
30+
Ports: infoJSON.Ports,
31+
MessageIds: infoJSON.MessageIds,
32+
Units: make(map[string]utils.Operations),
3333
}
3434
for key, value := range infoJSON.Units {
35-
println("Units: ", key, value)
36-
}
37-
for key, value := range infoJSON.MessageIds {
38-
println("MessageIds: ", key, value)
39-
}
40-
for key, value := range infoJSON.BoardIds {
41-
println("BoardIds: ", key, value)
42-
}
43-
44-
var info Info
45-
for key, value := range infoJSON.Units {
46-
info.Units = make(map[string]utils.Operations)
47-
4835
info.Units[key], err = utils.NewOperations(value)
4936
if err != nil {
5037
return nil, err
@@ -66,10 +53,22 @@ func NewADJ() (*ADJ, error) {
6653
}
6754

6855
boards, err := getBoards(boardsList.Boards)
56+
if err != nil {
57+
return nil, err
58+
}
6959

7060
info.BoardIds, err = getBoardIds(boardsList.Boards)
61+
if err != nil {
62+
return nil, err
63+
}
7164

7265
info.Addresses, err = getAddresses(boards)
66+
if err != nil {
67+
return nil, err
68+
}
69+
for target, address := range infoJSON.Addresses {
70+
info.Addresses[target] = address
71+
}
7372

7473
adj := &ADJ{
7574
Info: info,
@@ -112,13 +111,10 @@ func checkRepo() bool {
112111
}
113112

114113
func getBoards(boardsList map[string]string) (map[string]Board, error) {
115-
var boards map[string]Board
114+
boards := make(map[string]Board, len(boardsList))
116115
for boardName, boardPath := range boardsList {
117-
if _, err := os.Stat(boardPath); os.IsNotExist(err) {
118-
return nil, err
119-
}
120-
121-
boardRaw, err := os.ReadFile(boardPath)
116+
fullPath := path.Join(RepoPath, boardPath)
117+
boardRaw, err := os.ReadFile(fullPath)
122118
if err != nil {
123119
return nil, err
124120
}
@@ -128,9 +124,10 @@ func getBoards(boardsList map[string]string) (map[string]Board, error) {
128124
return nil, err
129125
}
130126

131-
var board Board
132-
board.Name = boardName
133-
board.IP = boardJSON.IP
127+
board := Board{
128+
Name: boardName,
129+
IP: boardJSON.IP,
130+
}
134131

135132
board.Packets, err = getBoardPackets(boardJSON.PacketsPaths)
136133
if err != nil {
@@ -197,13 +194,10 @@ func getBoardMeasurements(measurementsPaths []string) ([]Measurement, error) {
197194
}
198195

199196
func getBoardIds(boards map[string]string) (map[string]uint16, error) {
200-
var boardIds map[string]uint16
197+
boardIds := make(map[string]uint16, len(boards))
201198
for boardName, boardPath := range boards {
202-
if _, err := os.Stat(boardPath); os.IsNotExist(err) {
203-
return nil, err
204-
}
205-
206-
boardRaw, err := os.ReadFile(boardPath)
199+
fullPath := path.Join(RepoPath, boardPath)
200+
boardRaw, err := os.ReadFile(fullPath)
207201
if err != nil {
208202
return nil, err
209203
}
@@ -220,18 +214,19 @@ func getBoardIds(boards map[string]string) (map[string]uint16, error) {
220214
}
221215

222216
func getBoardStructures(board Board) []Structure {
223-
var structures []Structure
224-
for _, packet := range board.Packets {
225-
var structure Structure
226-
structure.Packet = packet
227-
structure.Measurements = board.Measurements
217+
structures := make([]Structure, len(board.Packets))
218+
for i, packet := range board.Packets {
219+
structures[i] = Structure{
220+
Packet: packet,
221+
Measurements: board.Measurements,
222+
}
228223
}
229224

230225
return structures
231226
}
232227

233228
func getAddresses(boards map[string]Board) (map[string]string, error) {
234-
var addresses map[string]string
229+
addresses := make(map[string]string, len(boards))
235230
for boardName, board := range boards {
236231
addresses[boardName] = board.IP
237232
}

0 commit comments

Comments
 (0)