IntenseFinance/neovim
3
0
Fork 0
mirror of https://github.com/neovim/neovim.git synced 2025-02-25 18:55:25 -06:00
Code Issues Packages Projects Releases Wiki Activity

Refactor job.c module to use WStream

Browse Source 
After a job has accumulated 1mb of stdin data we assume that it's stuck and kill
it.
This commit is contained in:
Thiago de Arruda 2014-04-17 16:24:43 -03:00
parent 913e92324a
commit d59034ea93
1 changed files with 15 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
src/os/job.c
View File

@ -7,6 +7,8 @@
#include "os/job_defs.h"
#include "os/rstream.h"
#include "os/rstream_defs.h"
#include "os/wstream.h"
#include "os/wstream_defs.h"
#include "os/event.h"
#include "os/event_defs.h"
#include "os/time.h"
@ -18,6 +20,7 @@
#define EXIT_TIMEOUT 25
#define MAX_RUNNING_JOBS 100
#define JOB_BUFFER_SIZE 1024
#define JOB_WRITE_MAXMEM 1024 * 1024
struct job {
// Job id the index in the job table plus one.
@ -38,8 +41,10 @@ struct job {
// Callbacks
job_exit_cb exit_cb;
rstream_cb stdout_cb, stderr_cb;
// Readable streams
// Readable streams(std{out,err})
RStream *out, *err;
// Writable stream(stdin)
WStream *in;
// Structures for process spawning/management used by libuv
uv_process_t proc;
uv_process_options_t proc_opts;
@ -57,7 +62,6 @@ static void free_job(Job *job);
// Callbacks for libuv
static void job_prepare_cb(uv_prepare_t *handle);
static void write_cb(uv_write_t *req, int status);
static void read_cb(RStream *rstream, void *data, bool eof);
static void exit_cb(uv_process_t *proc, int64_t status, int term_signal);
static void close_cb(uv_handle_t *handle);
@ -177,6 +181,8 @@ int job_start(char **argv,
return -1;
}
job->in = wstream_new(JOB_WRITE_MAXMEM);
wstream_set_stream(job->in, (uv_stream_t *)&job->proc_stdin);
// Start the readable streams
job->out = rstream_new(read_cb, JOB_BUFFER_SIZE, job, true);
job->err = rstream_new(read_cb, JOB_BUFFER_SIZE, job, true);
@ -209,8 +215,6 @@ bool job_stop(int id)
bool job_write(int id, char *data, uint32_t len)
{
uv_buf_t uvbuf;
uv_write_t *req;
Job *job = find_job(id);
if (job == NULL || job->stopped) {
@ -218,11 +222,10 @@ bool job_write(int id, char *data, uint32_t len)
return false;
}
req = xmalloc(sizeof(uv_write_t));
req->data = data;
uvbuf.base = data;
uvbuf.len = len;
uv_write(req, (uv_stream_t *)&job->proc_stdin, &uvbuf, 1, write_cb);
if (!wstream_write(job->in, data, len, true)) {
job_stop(job->id);
return false;
}
return true;
}
@ -270,9 +273,6 @@ static void free_job(Job *job)
uv_close((uv_handle_t *)&job->proc_stdin, close_cb);
uv_close((uv_handle_t *)&job->proc_stderr, close_cb);
uv_close((uv_handle_t *)&job->proc, close_cb);
rstream_free(job->out);
rstream_free(job->err);
wstream_free(job->in);
}
/// Iterates the table, sending SIGTERM to stopped jobs and SIGKILL to those
@ -297,12 +297,6 @@ static void job_prepare_cb(uv_prepare_t *handle)
}
}
static void write_cb(uv_write_t *req, int status)
{
free(req->data);
free(req);
}
// Wraps the call to std{out,err}_cb and emits a JobExit event if necessary.
static void read_cb(RStream *rstream, void *data, bool eof)
{
@ -344,6 +338,9 @@ static void close_cb(uv_handle_t *handle)
if (--job->pending_closes == 0) {
// Only free the job memory after all the associated handles are properly
// closed by libuv
rstream_free(job->out);
rstream_free(job->err);
wstream_free(job->in);
free(job);
}
}