yjp 1 vuosi sitten
vanhempi
commit
201a3f1811
1 muutettua tiedostoa jossa 28 lisäystä ja 4 poistoa
  1. 28 4
      framework/core/infrastructure/database/database.go

+ 28 - 4
framework/core/infrastructure/database/database.go

@@ -516,7 +516,13 @@ func Query(executor Executor, executeParams *sql.QueryExecuteParams) ([]sql.Resu
 		results[i] = row
 	}
 
-	return results, int64(countTableRow[0]["count"].(float64)), nil
+	var totalCount int64
+	err = reflectutils.AssignInt64Value(countTableRow[0]["count"], reflect.ValueOf(&totalCount))
+	if err != nil {
+		return nil, 0, err
+	}
+
+	return results, totalCount, nil
 }
 
 // QueryOne 查询单条数据
@@ -586,7 +592,13 @@ func Count(executor Executor, executeParams *sql.CountExecuteParams) (int64, err
 		return 0, err
 	}
 
-	return int64(tableRows[0]["count"].(float64)), nil
+	var count int64
+	err = reflectutils.AssignInt64Value(tableRows[0]["count"], reflect.ValueOf(&count))
+	if err != nil {
+		return 0, err
+	}
+
+	return count, nil
 }
 
 // CheckExist 数据存在性检查
@@ -619,7 +631,13 @@ func CheckExist(executor Executor, executeParams *sql.CheckExistExecuteParams) (
 		return false, err
 	}
 
-	return int64(tableRows[0]["count"].(float64)) > 0, nil
+	var count int64
+	err = reflectutils.AssignInt64Value(tableRows[0]["count"], reflect.ValueOf(&count))
+	if err != nil {
+		return false, err
+	}
+
+	return count > 0, nil
 }
 
 // CheckHasOnlyOne 数据唯一性检查
@@ -652,7 +670,13 @@ func CheckHasOnlyOne(executor Executor, executeParams *sql.CheckHasOnlyOneExecut
 		return false, err
 	}
 
-	return int64(tableRows[0]["count"].(float64)) == 1, nil
+	var count int64
+	err = reflectutils.AssignInt64Value(tableRows[0]["count"], reflect.ValueOf(&count))
+	if err != nil {
+		return false, err
+	}
+
+	return count == 1, nil
 }
 
 // ExecuteRawSql SQL执行接口