Merge pull request #39 from wangbokun/dev

update
wangbokun · Jul 31, 2020 · 91c3a55 · 91c3a55
2 parents ad70dc5 + 4075ae0
commit 91c3a55
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/db/mysql/mysql.go b/db/mysql/mysql.go
@@ -123,17 +123,17 @@ func (my *MySQL) Update(ctx context.Context, table string, v interface{}, cond m
 }
 
 // Upset update if exist or insert, true is insert, false is update
-func (my *MySQL) Upset(ctx context.Context, table string, v interface{}, cond map[string]interface{}) (bool, error) {
+func (my *MySQL) Upset(ctx context.Context, table string, v interface{}, cond map[string]interface{}) (bool, error, sql.Result) {
 	result, err := my.Update(ctx, table, v, cond)
 	if err != nil {
-		return false, err
+		return false, err, nil
 	}
 	if n, err := result.RowsAffected(); err != nil || n != 0 {
-		return false, err
+		return false, err, result
 	}
 
-	_, err = my.Put(ctx, table, v)
-	return true, err
+	res, err := my.Put(ctx, table, v)
+	return true, err, res
 }
 
 // Get get data