study/chsrc
1
0
Fork 0
mirror of https://github.com/RubyMetric/chsrc synced 2025-06-09 04:16:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

Measure fastly

Browse Source
This commit is contained in:
Aoran Zeng 2024-09-01 23:27:49 +08:00
parent d7a2c7d2c3
commit 47058e2fb0
No known key found for this signature in database
GPG Key ID: 8F8BA8488E10ED98
1 changed files with 45 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
include/chsrc.h
View File

@ -7,13 +7,14 @@
* Contributors : Peng Gao <gn3po4g@outlook.com> * Contributors : Peng Gao <gn3po4g@outlook.com>
* | * |
* Created on : <2023-08-29> * Created on : <2023-08-29>
* Last modified : <2024-08-30> * Last modified : <2024-09-01>
* *
* chsrc 头文件 * chsrc 头文件
* ------------------------------------------------------------*/ * ------------------------------------------------------------*/
#include "xy.h" #include "xy.h"
#include "source.h" #include "source.h"
#include <pthread.h>
#define App_Name "chsrc" #define App_Name "chsrc"
@ -359,9 +360,11 @@ to_human_readable_speed (double speed)
* 功劳和版权属于原作者，由 @ccmywish 修改为C语言，并做了额外调整 * 功劳和版权属于原作者，由 @ccmywish 修改为C语言，并做了额外调整
* *
* @return 返回测得的速度，若出错，返回-1 * @return 返回测得的速度，若出错，返回-1
*
* 该函数实际原型为 double * (*)(const char*)
*/ */
double void *
measure_speed (const char *url) measure_speed (void *url)
{ {
char *time_sec = "6"; char *time_sec = "6";
@ -404,14 +407,18 @@ measure_speed (const char *url)
if (200!=http_code) if (200!=http_code)
{ {
char* httpcodestr = yellow (xy_2strjoin ("HTTP码 ", buf)); char *httpcodestr = yellow (xy_2strjoin ("HTTP码 ", buf));
puts (xy_strjoin (3, speedstr, " | ", httpcodestr)); puts (xy_strjoin (3, speedstr, " | ", httpcodestr));
} }
else else
{ {
puts (speedstr); puts (speedstr);
} }
return speed;
double *speed_exported = xy_malloc0 (sizeof(double));
*speed_exported = speed;
return speed_exported;
} }
@ -467,15 +474,20 @@ auto_select_ (SourceInfo *sources, size_t size, const char *target_name)
} }
double speeds[size]; double speeds[size];
bool measured[size]; // 是否测速了
double speed = 0.0; double speed = 0.0;
pthread_t *threads = xy_malloc0 (sizeof(pthread_t) * size);
for (int i=0; i<size; i++) for (int i=0; i<size; i++)
{ {
SourceInfo src = sources[i]; SourceInfo src = sources[i];
const char* url = src.mirror->__bigfile_url; const char *url = src.mirror->__bigfile_url;
if (NULL==url) if (NULL==url)
{ {
if (xy_streql ("upstream", src.mirror->code)) if (xy_streql ("upstream", src.mirror->code))
{ {
speed = 0;
continue; /* 上游默认源不测速 */ continue; /* 上游默认源不测速 */
} }
else else
@ -485,6 +497,8 @@ auto_select_ (SourceInfo *sources, size_t size, const char *target_name)
chsrc_warn (xy_strjoin (3, msg1, src.mirror->code, msg2)); chsrc_warn (xy_strjoin (3, msg1, src.mirror->code, msg2));
speed = 0; speed = 0;
} }
speeds[i] = speed;
measured[i] = false;
} }
else else
{ {
@ -492,9 +506,32 @@ auto_select_ (SourceInfo *sources, size_t size, const char *target_name)
printf ("%s", xy_strjoin (3, msg, src.mirror->site , " ... ")); printf ("%s", xy_strjoin (3, msg, src.mirror->site , " ... "));
fflush (stdout); fflush (stdout);
speed = measure_speed (url);
char *url_ = xy_strdup (url);
int ret = pthread_create (&threads[i], NULL, measure_speed, url_);
if (ret!=0)
{
chsrc_error ("Unable to measure speed\n");
exit (Exit_UserCause);
}
else
{
measured[i] = true;
}
// speed = measure_speed (url);
}
}
/* 汇总 */
for (int i=0; i<size; i++)
{
double *spd;
if (measured[i]==true)
{
pthread_join (threads[i], (void *)&spd);
speeds[i] = *spd;
} }
speeds[i] = speed;
} }
int fast_idx = get_max_ele_idx_in_dbl_ary (speeds, size); int fast_idx = get_max_ele_idx_in_dbl_ary (speeds, size);