Ejemplo n.º 1
0
func (m *JobManager) killJob(id string) {
	m.jobLock.Lock()
	job, ex := m.jobMap[id]
	if ex {
		isFail := false
		for tid, task := range job.TaskMap {
			if task.State != comp.TASK_COMPLETE && task.State != comp.TASK_FAIL {
				task.State = comp.TASK_FAIL
				task.FinishTime = time.Now()
				job.TaskMap[tid] = task
				isFail = true
			}
		}
		job.FinishTime = time.Now()
		var update map[string]interface{}
		//update job in db
		if isFail {
			update = bson.M{"$set": bson.M{models.JOB_STATUS: -1}}
		} else {
			update = bson.M{"$set": bson.M{models.JOB_STATUS: 100}}
		}
		models.UpdateJobSketchInDB(id, update)
		models.UpdateJobInDB(id, job)
		delete(m.jobMap, id)
	}
	m.jobLock.Unlock()
}
Ejemplo n.º 2
0
//Start find finished job cyclically
func (m *JobManager) updateJobInDB() {
	log.Println("Start update job state.")
	for {
		//update job status
		var finishedJobs []string = make([]string, 0)
		m.jobLock.Lock()
		for jid, job := range m.jobMap {

			all := 0
			pro := 0
			for _, ts := range job.TaskMap {
				all = all + 2
				if ts.State == comp.TASK_COMPLETE || ts.State == comp.TASK_FAIL {
					pro = pro + 2
				} else if ts.State == comp.TASK_RUN {
					pro = pro + 1
				}
			}
			rate := 100
			if all != 0 {
				rate = pro * 100 / all

			}
			//log.Println(all, " : ", pro)

			if all == pro {
				finishedJobs = append(finishedJobs, jid)
				job.FinishTime = time.Now()
			}
			update := bson.M{"$set": bson.M{models.JOB_STATUS: rate}}
			models.UpdateJobSketchInDB(jid, update)
			models.UpdateJobInDB(jid, job)
		}
		m.jobLock.Unlock()

		//delete finished job
		for _, id := range finishedJobs {
			log.Println("Delete finished job: ", id)
			m.deleteJob(id)
		}

		time.Sleep(comm.HEARTTIME)
	}
}