changed signature of progress callback

1 files changed, 5 insertions, 4 deletions

diff --git a/rhimport/session.go b/rhimport/session.go
index 84d283e..ece5303 100644
--- a/rhimport/session.go
+++ b/rhimport/session.go
@@ -70,7 +70,8 @@ type SessionDoneCB struct {
 type ProgressData struct {
 	Step     int
 	StepName string
-	Progress float64
+	Current  float64
+	Total    float64
 }
 
 type sessionAddProgressHandlerResponse struct {
@@ -93,9 +94,9 @@ type sessionAddDoneHandlerRequest struct {
 	response chan<- sessionAddDoneHandlerResponse
 }
 
-func sessionProgressCallback(step int, stepName string, progress float64, userdata interface{}) bool {
+func sessionProgressCallback(step int, stepName string, current, total float64, userdata interface{}) bool {
 	out := userdata.(chan<- ProgressData)
-	out <- ProgressData{step, stepName, progress}
+	out <- ProgressData{step, stepName, current, total}
 	return true
 }
 
@@ -163,7 +164,7 @@ func (self *Session) addDoneHandler(userdata interface{}, cb DoneCB) (resp sessi
 func (self *Session) callProgressHandler(p *ProgressData) {
 	for _, cb := range self.progressCBs {
 		if cb.cb != nil {
-			if keep := cb.cb(p.Step, p.StepName, p.Progress, cb.userdata); !keep {
+			if keep := cb.cb(p.Step, p.StepName, p.Current, p.Total, cb.userdata); !keep {
 				cb.cb = nil
 			}
 		}