#include "dat.h"
#include <stdint.h>
#include <stdlib.h>
#include <sys/socket.h>
struct Server srv = {
.port = Portdef,
.wal = {
.filesize = Filesizedef,
.wantsync = 1,
.syncrate = DEFAULT_FSYNC_MS * 1000000,
},
};
// srv_acquire_wal tries to lock the wal dir specified by s->wal and
// replay entries from it to initialize the s state with jobs.
// On errors it exits from the program.
void srv_acquire_wal(Server *s) {
if (s->wal.use) {
// We want to make sure that only one beanstalkd tries
// to use the wal directory at a time. So acquire a lock
// now and never release it.
if (!waldirlock(&s->wal)) {
twarnx("failed to lock wal dir %s", s->wal.dir);
exit(10);
}
Job list = {.prev=NULL, .next=NULL};
list.prev = list.next = &list;
walinit(&s->wal, &list);
int ok = prot_replay(s, &list);
if (!ok) {
twarnx("failed to replay log");
exit(1);
}
}
}
void
srvserve(Server *s)
{
int r;
Socket *sock;
if (sockinit() == -1) {
twarnx("sockinit");
exit(1);
}
s->sock.x = s;
s->sock.f = (Handle)srvaccept;
s->conns.less = conn_less;
s->conns.setpos = conn_setpos;
r = listen(s->sock.fd, 1024);
if (r == -1) {
twarn("listen");
return;
}
r = sockwant(&s->sock, 'r');
if (r == -1) {
twarn("sockwant");
exit(2);
}
for (;;) {
int64 period = prottick(s);
int rw = socknext(&sock, period);
if (rw == -1) {
twarnx("socknext");
exit(1);
}
if (rw) {
sock->f(sock->x, rw);
}
}
}
void
srvaccept(Server *s, int ev)
{
h_accept(s->sock.fd, ev, s);
}