Commit 24fe68e5 authored by M. Rehan Abbasi's avatar M. Rehan Abbasi
Browse files

separately populate values in CreateCategoryLoad function in traffic manager

parent 8a3fd2f1

go-packages/meep-vis-traffic-mgr/traffic-mgr.go

+3 −4
Original line number Diff line number Diff line
@@ -396,19 +396,18 @@ func (tm *TrafficMgr) CreateCategoryLoad(category string, data map[string]int32) 
		FieldEighteenToTwentyOne,

		FieldTwentyOneToTwentyFour,

	}

	for i, field := range fields {

	for _, field := range fields {

		if _, found := data[field]; !found {

			return errors.New("Missing time field " + field)

		} else if loadTime[i] == 0 {

			return errors.New("Invalid " + field)

		}

		loadTime = append(loadTime, data[field])

	}



	// Create Traffic Load entry

	query := `INSERT INTO ` + CategoryTable +

		` (category, "0000-0300", "0300-0600", "0600-0900", "0900-1200", "1200-1500", "1500-1800", "1800-2100", "2100-2400")

			VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)`

	_, err = tm.db.Exec(query, category, loadTime[0:8])

	_, err = tm.db.Exec(query, category, loadTime[0], loadTime[1], loadTime[2], loadTime[3], loadTime[4], loadTime[5], loadTime[6], loadTime[7])

	if err != nil {

		log.Error(err.Error())

		return err