Refactoring the internal struct
This commit is contained in:
@@ -141,18 +141,18 @@ func RemoveTask(name, nodeid string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func RetrieveTasks() []utilities.InternalQueueTaskData {
|
||||
func RetrieveTasks() []utilities.RequestTaskData {
|
||||
rows, err := db.Query(declStat.ListAllTasks)
|
||||
if err != nil {
|
||||
log.Println("Query error:", err)
|
||||
return []utilities.InternalQueueTaskData{}
|
||||
return []utilities.RequestTaskData{}
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
var tasks []utilities.InternalQueueTaskData
|
||||
var tasks []utilities.RequestTaskData
|
||||
|
||||
for rows.Next() {
|
||||
var task utilities.InternalQueueTaskData
|
||||
var task utilities.RequestTaskData
|
||||
|
||||
err := rows.Scan(&task.Name, &task.UUID, &task.Command, &task.Nodeid, &task.Creation, &task.Expire)
|
||||
if err != nil {
|
||||
|
||||
@@ -32,7 +32,7 @@ func taskRoutine(venvName string, pyListArgs []string) {
|
||||
|
||||
wg.Add(1)
|
||||
|
||||
go func(task utilities.InternalQueueTaskData) {
|
||||
go func(task utilities.RequestTaskData) {
|
||||
defer wg.Done()
|
||||
|
||||
log.Println(infoTag, fmt.Sprintf("Node online. NodeID: %s", task.Nodeid), "Executing!")
|
||||
|
||||
@@ -35,15 +35,6 @@ type TokenListBody struct {
|
||||
}
|
||||
|
||||
type RequestTaskData struct {
|
||||
Name string `json:"name"`
|
||||
Command string `json:"command"`
|
||||
Nodeids []string `json:"nodeids"`
|
||||
Creation string `json:"creation"`
|
||||
Status string `json:"status"`
|
||||
Expire string `json:"expire"`
|
||||
}
|
||||
|
||||
type InternalQueueTaskData struct {
|
||||
Name string `json:"name"`
|
||||
UUID string `json:"uuid"`
|
||||
Command string `json:"command"`
|
||||
|
||||
@@ -25,24 +25,24 @@ func (t TaskCreateBody) GetName() string { return t.Details.Name }
|
||||
func CheckOldRemnants(configDir, certDir, databaseDir, fullDatabasePath string) {
|
||||
remnantConfigDir := StatPath(configDir)
|
||||
if !remnantConfigDir {
|
||||
log.Println(InfoTag, "Creating Config (conf) folder...")
|
||||
log.Println(InfoTag+"[STAT]", "Creating Config (conf) folder...")
|
||||
os.Mkdir(configDir, os.FileMode(0755))
|
||||
}
|
||||
|
||||
remnantCertDir := StatPath(certDir)
|
||||
if !remnantCertDir {
|
||||
log.Println(InfoTag, "Creating Certificate (cert) folder...")
|
||||
log.Println(InfoTag+"[STAT]", "Creating Certificate (cert) folder...")
|
||||
os.Mkdir(certDir, os.FileMode(0755))
|
||||
}
|
||||
|
||||
remnantDatabaseDir := StatPath(databaseDir)
|
||||
if !remnantDatabaseDir {
|
||||
log.Println(InfoTag, "Creating Database (data) folder...")
|
||||
log.Println(InfoTag+"[STAT]", "Creating Database (data) folder...")
|
||||
os.Mkdir(databaseDir, os.FileMode(0755))
|
||||
}
|
||||
remnantDatabaseFile := StatPath(fullDatabasePath)
|
||||
if !remnantDatabaseFile {
|
||||
log.Println(InfoTag, "Creating Database (ghostserver.db) file...")
|
||||
log.Println(InfoTag+"[STAT]", "Creating Database (ghostserver.db) file...")
|
||||
os.Create(fullDatabasePath)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user