140 lines
3.8 KiB
C++
140 lines
3.8 KiB
C++
#include "borderserver.h"
|
|
#include <iostream>
|
|
|
|
using namespace std;
|
|
using namespace nxs;
|
|
|
|
bool BorderServer::Create(const string &file) {
|
|
ram_used = 0;
|
|
return IndexFile<Border>::Create(file, 2 * sizeof(Link));
|
|
}
|
|
|
|
bool BorderServer::Load(const string &file, bool rdonly) {
|
|
ram_used = 0;
|
|
bool success = IndexFile<Border>::Load(file, rdonly);
|
|
if(!success) return false;
|
|
for(unsigned int i = 0; i < size(); i++)
|
|
operator[](i).links = NULL;
|
|
return true;
|
|
}
|
|
|
|
void BorderServer::Close() {
|
|
if(!Opened()) return;
|
|
Flush();
|
|
IndexFile<Border>::Close();
|
|
}
|
|
|
|
void BorderServer::Flush() {
|
|
std::map<unsigned int, list<unsigned int>::iterator>::iterator i;
|
|
for(i = index.begin(); i != index.end(); i++) {
|
|
unsigned int patch = (*i).first;
|
|
FlushBorder(patch);
|
|
}
|
|
pqueue.clear();
|
|
index.clear();
|
|
}
|
|
|
|
void BorderServer::AddBorder(unsigned short _size, unsigned int used) {
|
|
Border entry;
|
|
assert((Length() % sizeof(Link)) == 0);
|
|
|
|
entry.start = Length()/ sizeof(Link);
|
|
entry.size = _size;
|
|
entry.used = used;
|
|
entry.links = NULL;
|
|
push_back(entry);
|
|
Redim((int64)entry.start * (int64)sizeof(Link) + (int64)_size * (int64)sizeof(Link));
|
|
}
|
|
|
|
Border &BorderServer::GetBorder(unsigned int border, bool flush) {
|
|
Border &entry = operator[](border);
|
|
//assert(entry.size != 0);
|
|
if(index.count(border)) {
|
|
//assert(entry.links);
|
|
list<unsigned int>::iterator i = index[border];
|
|
pqueue.erase(i);
|
|
pqueue.push_front(border);
|
|
index[border] = pqueue.begin();
|
|
} else {
|
|
while(flush && ram_used > ram_max) {
|
|
assert(pqueue.size());
|
|
unsigned int to_flush = pqueue.back();
|
|
pqueue.pop_back();
|
|
index.erase(to_flush);
|
|
FlushBorder(to_flush);
|
|
}
|
|
entry.links = GetRegion(entry.start, entry.size);
|
|
pqueue.push_front(border);
|
|
index[border] = pqueue.begin();
|
|
ram_used += entry.size;
|
|
}
|
|
return entry;
|
|
}
|
|
//TODO Change when remving borderentry class.
|
|
void BorderServer::ResizeBorder(unsigned int border, unsigned int used) {
|
|
assert(border < size());
|
|
Border &entry = GetBorder(border);
|
|
if(used <= entry.size) {
|
|
entry.used = used;
|
|
return;
|
|
}
|
|
|
|
int capacity = used;
|
|
if(capacity < entry.size * 2)
|
|
capacity = entry.size * 2;
|
|
|
|
unsigned int newstart = Length()/sizeof(Link);
|
|
Redim((int64)(newstart + capacity) * (int64)sizeof(Link));
|
|
Link *newlinks = new Link[capacity];
|
|
if(entry.used > 0) {
|
|
assert(entry.links);
|
|
memcpy(newlinks, entry.links, entry.used * sizeof(Link));
|
|
delete []entry.links;
|
|
entry.links = NULL;
|
|
}
|
|
assert(entry.links == NULL);
|
|
entry.links = newlinks;
|
|
entry.start = newstart;
|
|
entry.size = capacity;
|
|
entry.used = used;
|
|
}
|
|
|
|
void BorderServer::FlushBorder(unsigned int border) {
|
|
Border &entry = operator[](border);
|
|
//assert(entry.links);
|
|
if(entry.size && !MFile::IsReadOnly()) { //write back patch
|
|
MFile::SetPosition((int64)entry.start * (int64)sizeof(Link));
|
|
MFile::WriteBuffer(entry.links, entry.used * (int64)sizeof(Link));
|
|
}
|
|
if(entry.links)
|
|
delete [](entry.links);
|
|
entry.links = NULL;
|
|
ram_used -= entry.size;
|
|
}
|
|
|
|
Link *BorderServer::GetRegion(unsigned int start, unsigned int size) {
|
|
if(size == 0) return NULL;
|
|
SetPosition((int64)start * (int64)sizeof(Link));
|
|
Link *buf = new Link[size];
|
|
assert(buf);
|
|
ReadBuffer(buf, (int64)size * (int64)sizeof(Link));
|
|
return buf;
|
|
}
|
|
|
|
bool BorderServer::LoadHeader() {
|
|
unsigned int magic;
|
|
ReadBuffer(&magic, sizeof(unsigned int));
|
|
if(magic != 0x3042584e) { //NXB0
|
|
cerr << "Invalid magic. Not a nxs file\n";
|
|
return false;
|
|
}
|
|
ReadBuffer(&offset, sizeof(int64));
|
|
return true;
|
|
}
|
|
|
|
void BorderServer::SaveHeader() {
|
|
unsigned int magic = 0x3042584e; // NXB0
|
|
WriteBuffer(&magic, sizeof(unsigned int));
|
|
WriteBuffer(&offset, sizeof(int64));
|
|
}
|