ceph/branches/sage/pgs/crush/test/overload.cc
sageweil 9213a23f14 eek
git-svn-id: https://ceph.svn.sf.net/svnroot/ceph@1138 29311d96-e01e-0410-9327-a35deaab8ce9
2007-02-28 18:42:55 +00:00

336 lines
7.5 KiB
C++

#include "../crush.h"
using namespace crush;
#include "../../common/Clock.h"
#include <math.h>
#include <iostream>
#include <vector>
using namespace std;
Clock g_clock;
Bucket *make_bucket(Crush& c, vector<int>& wid, int h, int& ndisks)
{
if (h == 0) {
// uniform
Hash hash(123);
vector<int> disks;
for (int i=0; i<wid[h]; i++)
disks.push_back(ndisks++);
float w = ((ndisks-1)/100+1)*10;
UniformBucket *b = new UniformBucket(1, 0, w, disks);
//b->make_primes(hash);
c.add_bucket(b);
//cout << h << " uniformbucket with " << wid[h] << " disks weight " << w << endl;
return b;
} else {
// mixed
Bucket *b = new TreeBucket(h+1);
for (int i=0; i<wid[h]; i++) {
Bucket *n = make_bucket(c, wid, h-1, ndisks);
b->add_item(n->get_id(), n->get_weight());
}
c.add_bucket(b);
//cout << h << " mixedbucket with " << wid[h] << endl;
return b;
}
}
int make_hierarchy(Crush& c, vector<int>& wid, int& ndisks)
{
Bucket *b = make_bucket(c, wid, wid.size()-1, ndisks);
return b->get_id();
}
float go(int dep, int utilization )
{
Hash h(73232313);
int overloadcutoff = (int)((float)10000.0 / (float)utilization);
//cout << "util " << utilization << " cutoff " << overloadcutoff << endl;
// crush
Crush c;
// buckets
int root = -1;
int ndisks = 0;
vector<int> wid;
for (int d=0; d<dep; d++)
wid.push_back(10);
if (1) {
root = make_hierarchy(c, wid, ndisks);
}
//cout << ndisks << " disks" << endl;
// rule
int numrep = 1;
Rule rule;
rule.steps.push_back(RuleStep(CRUSH_RULE_TAKE, root));
rule.steps.push_back(RuleStep(CRUSH_RULE_CHOOSE, numrep, 0));
rule.steps.push_back(RuleStep(CRUSH_RULE_EMIT));
//c.overload[10] = .1;
int pg_per_base = 20;
int pg_med = 10*pg_per_base;
int pg_per = pg_per_base*5.5;//100;
int numpg = pg_per*ndisks/numrep;
vector<int> ocount(ndisks);
//cout << ndisks << " disks, " << endl;
//cout << pg_per << " pgs per disk" << endl;
// cout << numpg << " logical pgs" << endl;
//cout << "numrep is " << numrep << endl;
int place = 100000;
int times = place / numpg;
if (!times) times = 1;
//cout << "looping " << times << " times" << endl;
float tavg[10];
float tvar[10];
for (int j=0;j<10;j++) {
tvar[j] = 0;
tavg[j] = 0;
}
int tvarnum = 0;
float overloadsum = 0.0;
float adjustsum = 0.0;
float afteroverloadsum = 0.0;
float aslowdown = 0.0;
int chooses = 0;
int xs = 1;
for (int t=0; t<times; t++) {
vector<int> v(numrep);
c.overload.clear();
for (int z=0; z<ndisks; z++) ocount[z] = 0;
utime_t t1a = g_clock.now();
for (int x=xs; x<numpg+xs; x++) {
//cout << H(x) << "\t" << h(x) << endl;
c.do_rule(rule, x, v);
chooses += numrep;
//cout << "v = " << v << endl;// " " << v[0] << " " << v[1] << " " << v[2] << endl;
for (int i=0; i<numrep; i++) {
//int d = b.choose_r(x, i, h);
//v[i] = d;
ocount[v[i]]++;
}
//cout << v << "\t" << ocount << endl;
}
utime_t t1b = g_clock.now();
// overloaded?
int overloaded = 0;
int adjusted = 0;
for (int i=0; i<ocount.size(); i++) {
int target = (i/100+1)*pg_per_base;
int cutoff = target * overloadcutoff / 100;
int adjoff = target + (cutoff - target)*3/4;
if (ocount[i] > cutoff)
overloaded++;
if (ocount[i] > adjoff) {
adjusted++;
c.overload[i] = (float)target / (float)ocount[i];
//cout << "setting overload " << i << " to " << c.overload[i] << endl;
//cout << "disk " << i << " has " << ocount[i] << endl;
}
ocount[i] = 0;
}
//cout << overloaded << " overloaded" << endl;
overloadsum += (float)overloaded / (float)ndisks;
adjustsum += (float)adjusted / (float)ndisks;
// keep adjusting!
for (int bla=0; bla<5; bla++) {
utime_t t2a = g_clock.now();
// second pass
for (int x=xs; x<numpg+xs; x++) {
//cout << H(x) << "\t" << h(x) << endl;
c.do_rule(rule, x, v);
//cout << "v = " << v << endl;// " " << v[0] << " " << v[1] << " " << v[2] << endl;
for (int i=0; i<numrep; i++) {
//int d = b.choose_r(x, i, h);
//v[i] = d;
ocount[v[i]]++;
}
//cout << v << "\t" << ocount << endl;
}
utime_t t2b = g_clock.now();
int numover = 0;
for (int i=0; i<ocount.size(); i++) {
int target = (i/100+1)*pg_per_base;
int cutoff = target * overloadcutoff / 100;
int adjoff = cutoff;//target + (cutoff - target)*3/4;
if (ocount[i] >= adjoff) {
numover++;
if (c.overload.count(i) == 0) {
c.overload[i] = 1.0;
adjusted++;
}
//else cout << "(re)adjusting " << i << endl;
c.overload[i] *= (float)target / (float)ocount[i];
//cout << "setting overload " << i << " to " << c.overload[i] << endl;
//cout << "disk " << i << " has " << ocount[i] << endl;
}
ocount[i] = 0;
}
if (!numover) break;
cout << "readjusting" << endl;
}
utime_t t3a = g_clock.now();
for (int x=xs; x<numpg+xs; x++) {
//cout << H(x) << "\t" << h(x) << endl;
c.do_rule(rule, x, v);
//cout << "v = " << v << endl;// " " << v[0] << " " << v[1] << " " << v[2] << endl;
for (int i=0; i<numrep; i++) {
//int d = b.choose_r(x, i, h);
//v[i] = d;
ocount[v[i]]++;
}
//cout << v << "\t" << ocount << endl;
}
xs += numpg;
utime_t t3b = g_clock.now();
t1b -= t1a;
double t1 = (double)t1b;
t3b -= t3a;
double t3 = (double)t3b;
double slowdown = t3/t1;
//cout << "slowdown " << slowdown << endl;
aslowdown += slowdown;
int still = 0;
for (int i=0; i<ocount.size(); i++) {
int target = (i/100+1)*pg_per_base;
int cutoff = target * overloadcutoff / 100;
//int adjoff = target + (cutoff - target)/3;
if (ocount[i] > cutoff) {
still++;
//c.overload[ocount[i]] = 100.0 / (float)ocount[i];
if (c.overload.count(i)) cout << "[adjusted] ";
cout << "disk " << i << " has " << ocount[i] << endl;
}
}
//if (still) cout << "overload was " << overloaded << " now " << still << endl;
afteroverloadsum += (float)still / (float)ndisks;
//cout << "collisions: " << c.collisions << endl;
//cout << "r bumps: " << c.bumps << endl;
int n = ndisks/10;
float avg[10];
float var[10];
for (int i=0;i<10;i++) {
int s = n*i;
avg[i] = 0.0;
for (int j=0; j<n; j++)
avg[i] += ocount[j+s];
avg[i] /= n;//ocount.size();
var[i] = 0.0;
for (int j=0; j<n; j++)
var[i] += (ocount[j+s] - avg[i]) * (ocount[j+s] - avg[i]);
var[i] /= n;//ocount.size();
tvar[i] += var[i];
tavg[i] += avg[i];
}
//cout << "avg " << avg << " var " << var << " sd " << sqrt(var) << endl;
tvarnum++;
}
overloadsum /= tvarnum;
adjustsum /= tvarnum;
float avar = 0.0;
for (int j=0;j<10;j++) {
tvar[j] /= tvarnum;
tavg[j] /= tvarnum;
avar += tvar[j];
}
avar /= 10;
avar = sqrt(avar);
avar /= 5.5 * (float)pg_per_base;
afteroverloadsum /= tvarnum;
aslowdown /= tvarnum;
//int collisions = c.collisions[0] + c.collisions[1] + c.collisions[2] + c.collisions[3];
//float crate = (float) collisions / (float)chooses;
//cout << "collisions: " << c.collisions << endl;
//cout << "total variance " << tvar << endl;
//cout << " overlaod " << overloadsum << endl;
cout << overloadcutoff << "\t" << utilization
<< "\t" << overloadsum << "\t" << adjustsum << "\t" << afteroverloadsum
<< "\t" << aslowdown << "\t" << avar << "\t-";
for (int i=0;i<10;i++)
cout << "\t" << tavg[i] << "\t" << tvar[i];// << "\t" << tvar[i]/tavg[i];
cout << endl;
return tvar[0];
}
int main()
{
float var = go(3,50);
/* for (int d=70; d<100; d += 5) {
float var = go(3,d);
//cout << "## depth = " << d << endl;
//cout << d << "\t" << var << endl;
}*/
go(3,96);
go(3,97);
go(3,98);
go(3,99);
}