|
|
@@ -291,30 +291,11 @@ func GetOwnerDatasetByID(id int64, user *User) (*Dataset, error) { |
|
|
|
return nil, errors.New("dataset not fount") |
|
|
|
} |
|
|
|
|
|
|
|
func IncreaseDownloadCount(datasetID int64, uid int64) error { |
|
|
|
var err error |
|
|
|
sess := x.NewSession() |
|
|
|
defer sess.Close() |
|
|
|
if err = sess.Begin(); err != nil { |
|
|
|
return err |
|
|
|
} |
|
|
|
|
|
|
|
dataset := &Dataset{ID: datasetID, UserID: uid} |
|
|
|
has, err := sess.Get(dataset) |
|
|
|
if err != nil { |
|
|
|
return err |
|
|
|
} else if !has { |
|
|
|
return errors.New("not found") |
|
|
|
} |
|
|
|
|
|
|
|
func IncreaseDownloadCount(datasetID int64) error { |
|
|
|
// Update download count. |
|
|
|
if _, err := sess.Exec("UPDATE `dataset` SET download_times=download_times+1 WHERE id=?", datasetID); err != nil { |
|
|
|
if _, err := x.Exec("UPDATE `dataset` SET download_times=download_times+1 WHERE id=?", datasetID); err != nil { |
|
|
|
return fmt.Errorf("increase dataset count: %v", err) |
|
|
|
} |
|
|
|
|
|
|
|
if err = sess.Commit(); err != nil { |
|
|
|
sess.Close() |
|
|
|
return fmt.Errorf("Commit: %v", err) |
|
|
|
} |
|
|
|
return nil |
|
|
|
} |