UI changes

This commit is contained in:
2023-03-06 01:09:54 +00:00
parent bfe6232a41
commit a04ea65855
3 changed files with 342 additions and 225 deletions

View File

@@ -376,6 +376,10 @@ func AddBarrierIP(db *gorm.DB, SessionID string, barrierIP string, MachineName s
if exists == false {
var NewBarrierIP BarrierIP
// generate new UUID for barrier
id := uuid.New()
NewBarrierIP.ID = id.String()
NewBarrierIP.BarrierIP = barrierIP
NewBarrierIP.UserID = user.UserID
NewBarrierIP.MachineName = MachineName
@@ -416,3 +420,29 @@ func RemoveBarrierIP(db *gorm.DB, barrierIP string) error {
}
return nil
}
func GetBarrierInfo(db *gorm.DB, SessionID string) (barriers []*BarrierIP, err error) {
// Get User Information
user, err := GetUserInformation(db, SessionID)
if err != nil {
return nil, err
}
// Get barrier information for a particular
UserRows, err := db.Model(&BarrierIP{}).Where("user_id = ?", user.UserID).Rows()
if err != nil {
return nil, err
}
defer UserRows.Close()
for UserRows.Next() {
var barrier BarrierIP
err := db.ScanRows(UserRows, &barrier)
if err != nil {
return nil, err
}
barriers = append(barriers, &barrier)
}
return
}