Create a pip package for pipe_io_server
This commit is contained in:
29
pipe_io_server/clients/c_cpp/example.c
Normal file
29
pipe_io_server/clients/c_cpp/example.c
Normal file
@ -0,0 +1,29 @@
|
||||
#include "pipe_io.h"
|
||||
|
||||
|
||||
void run2() {
|
||||
pipe_io pipeio = pipeio_new_io("in", "out");
|
||||
const char* line;
|
||||
while ((line = pipeio_recv_msg(&pipeio.reader))) {
|
||||
pipeio_send_msg(&pipeio.writer, line);
|
||||
}
|
||||
pipeio_close_io(&pipeio);
|
||||
}
|
||||
|
||||
void msg_handler(const char* msg, void* ctx) {
|
||||
pipe_writer* writer = (pipe_writer*)(ctx);
|
||||
pipeio_send_msg(writer, msg);
|
||||
}
|
||||
|
||||
void run1() {
|
||||
pipe_io pipeio = pipeio_new_io("in", "out");
|
||||
pipeio_set_msg_handler(&pipeio.reader, &msg_handler, &pipeio.writer);
|
||||
pipeio_run(&pipeio.reader);
|
||||
pipeio_close_io(&pipeio);
|
||||
}
|
||||
|
||||
int main() {
|
||||
run1(); // or run2();
|
||||
// note that run1 and run2 do the same thing
|
||||
// and you can use whichever API you prefer
|
||||
}
|
13
pipe_io_server/clients/c_cpp/example.cpp
Normal file
13
pipe_io_server/clients/c_cpp/example.cpp
Normal file
@ -0,0 +1,13 @@
|
||||
#include <iostream>
|
||||
#include <string>
|
||||
#include <fstream>
|
||||
#include "pipe_io.hpp"
|
||||
|
||||
|
||||
int main() {
|
||||
PipeIO pipe_io("in", "out");
|
||||
pipe_io.reader.set_message_handler(
|
||||
[&pipe_io](std::string msg){pipe_io.writer.send_message(msg);}
|
||||
);
|
||||
pipe_io.reader.run();
|
||||
}
|
147
pipe_io_server/clients/c_cpp/pipe_io.h
Normal file
147
pipe_io_server/clients/c_cpp/pipe_io.h
Normal file
@ -0,0 +1,147 @@
|
||||
#pragma once
|
||||
#include <stdlib.h>
|
||||
#include <stdio.h>
|
||||
#include <stdbool.h>
|
||||
#include <string.h>
|
||||
|
||||
|
||||
typedef struct pipe_reader {
|
||||
FILE* stream;
|
||||
char* line_buf;
|
||||
size_t buf_size;
|
||||
void (*msg_handler)(const char*, void*);
|
||||
void* handler_ctx;
|
||||
} pipe_reader;
|
||||
|
||||
static void null_handler(const char* msg, void* ctx) {}
|
||||
|
||||
pipe_reader pipeio_new_reader(const char* pipe_path) {
|
||||
pipe_reader preader;
|
||||
preader.stream = fopen(pipe_path, "r");
|
||||
preader.line_buf = NULL;
|
||||
preader.buf_size = 0;
|
||||
preader.msg_handler = &null_handler;
|
||||
return preader;
|
||||
}
|
||||
|
||||
bool pipeio_check_reader(const pipe_reader* preader) {
|
||||
if (!preader || preader->stream == NULL) {
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
const char* pipeio_recv_msg(pipe_reader* preader) {
|
||||
if (!pipeio_check_reader(preader)) {
|
||||
return NULL;
|
||||
}
|
||||
ssize_t read = getline(&preader->line_buf, &preader->buf_size, preader->stream);
|
||||
if (read > 0) {
|
||||
preader->line_buf[read - 1] = '\0';
|
||||
return preader->line_buf;
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
void pipeio_set_msg_handler(pipe_reader* preader, void (*msg_handler)(const char*, void*), void* handler_ctx) {
|
||||
if (msg_handler != NULL && handler_ctx != NULL) {
|
||||
preader->msg_handler = msg_handler;
|
||||
preader->handler_ctx = handler_ctx;
|
||||
}
|
||||
}
|
||||
|
||||
void pipeio_run(pipe_reader* preader) {
|
||||
const char* line;
|
||||
while ((line = pipeio_recv_msg(preader))) {
|
||||
(*preader->msg_handler)(line, preader->handler_ctx);
|
||||
}
|
||||
}
|
||||
|
||||
void pipeio_close_reader(pipe_reader* preader) {
|
||||
if (!pipeio_check_reader(preader)) {
|
||||
return;
|
||||
}
|
||||
fclose(preader->stream);
|
||||
preader->stream = NULL;
|
||||
if (preader->line_buf) {
|
||||
free(preader->line_buf);
|
||||
preader->line_buf = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
typedef struct pipe_writer {
|
||||
FILE* stream;
|
||||
} pipe_writer;
|
||||
|
||||
pipe_writer pipeio_new_writer(const char* pipe_path) {
|
||||
pipe_writer pwriter;
|
||||
pwriter.stream = fopen(pipe_path, "w");
|
||||
return pwriter;
|
||||
}
|
||||
|
||||
bool pipeio_check_writer(const pipe_writer* pwriter) {
|
||||
if (!pwriter || pwriter->stream == NULL) {
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
bool pipeio_send_msg(pipe_writer* pwriter, const char* msg) {
|
||||
if (!pipeio_check_writer(pwriter)) {
|
||||
return false;
|
||||
}
|
||||
fprintf(pwriter->stream, "%s\n", msg);
|
||||
fflush(pwriter->stream);
|
||||
return true;
|
||||
}
|
||||
|
||||
void pipeio_close_writer(pipe_writer* pwriter) {
|
||||
if (!pipeio_check_writer(pwriter)) {
|
||||
return;
|
||||
}
|
||||
fclose(pwriter->stream);
|
||||
pwriter->stream = NULL;
|
||||
}
|
||||
|
||||
|
||||
typedef struct pipe_io {
|
||||
pipe_reader reader;
|
||||
pipe_writer writer;
|
||||
} pipe_io;
|
||||
|
||||
pipe_io pipeio_new_io(const char* in_pipe_path, const char* out_pipe_path) {
|
||||
pipe_io pipeio;
|
||||
pipeio.reader = pipeio_new_reader(in_pipe_path);
|
||||
pipeio.writer = pipeio_new_writer(out_pipe_path);
|
||||
return pipeio;
|
||||
}
|
||||
|
||||
bool pipeio_check_io(const pipe_io* pipeio) {
|
||||
if (!pipeio) {
|
||||
return false;
|
||||
}
|
||||
return (pipeio_check_reader(&pipeio->reader) && pipeio_check_writer(&pipeio->writer));
|
||||
}
|
||||
|
||||
void pipeio_close_io(pipe_io* pipeio) {
|
||||
if (!pipeio_check_io(pipeio)) {
|
||||
return;
|
||||
}
|
||||
pipeio_close_reader(&pipeio->reader);
|
||||
pipeio_close_writer(&pipeio->writer);
|
||||
}
|
||||
|
||||
// const char* recv_msg_pipe_io(pipe_io* pipeio) {
|
||||
// if (!check_pipe_io(pipeio)) {
|
||||
// return NULL;
|
||||
// }
|
||||
// return recv_msg_pipe_reader(&pipeio->preader);
|
||||
// }
|
||||
|
||||
// bool send_msg_pipe_io(pipe_io* pipeio, const char* msg) {
|
||||
// if (!check_pipe_io(pipeio)) {
|
||||
// return false;
|
||||
// }
|
||||
// return send_msg_pipe_writer(&pipeio->pwriter, msg);
|
||||
// }
|
79
pipe_io_server/clients/c_cpp/pipe_io.hpp
Normal file
79
pipe_io_server/clients/c_cpp/pipe_io.hpp
Normal file
@ -0,0 +1,79 @@
|
||||
extern "C" {
|
||||
#include "pipe_io.h"
|
||||
}
|
||||
#include <string>
|
||||
#include <iostream>
|
||||
#include <functional>
|
||||
|
||||
|
||||
class PipeReader {
|
||||
public:
|
||||
PipeReader(const std::string& pipe_path)
|
||||
:preader(pipeio_new_reader(pipe_path.c_str())) {}
|
||||
|
||||
virtual ~PipeReader() {
|
||||
pipeio_close_reader(&this->preader);
|
||||
}
|
||||
|
||||
PipeReader(PipeReader&) = delete;
|
||||
PipeReader(PipeReader&&) = delete;
|
||||
PipeReader& operator=(const PipeReader&) = delete;
|
||||
PipeReader& operator=(const PipeReader&&) = delete;
|
||||
|
||||
void set_message_handler(std::function<void(std::string)> message_handler) {
|
||||
this->message_handler = message_handler;
|
||||
}
|
||||
|
||||
void run() {
|
||||
while (const char* linebuf = this->recv_message()) {
|
||||
std::string msg(linebuf);
|
||||
this->message_handler(msg);
|
||||
}
|
||||
}
|
||||
|
||||
const char* recv_message() {
|
||||
return pipeio_recv_msg(&this->preader);
|
||||
}
|
||||
|
||||
private:
|
||||
pipe_reader preader;
|
||||
std::function<void(std::string)> message_handler = [](std::string){};
|
||||
};
|
||||
|
||||
|
||||
class PipeWriter {
|
||||
public:
|
||||
PipeWriter(const std::string& pipe_path)
|
||||
:pwriter(pipeio_new_writer(pipe_path.c_str())) {}
|
||||
|
||||
virtual ~PipeWriter() {
|
||||
pipeio_close_writer(&this->pwriter);
|
||||
}
|
||||
|
||||
PipeWriter(PipeReader&) = delete;
|
||||
PipeWriter(PipeReader&&) = delete;
|
||||
PipeWriter& operator=(const PipeReader&) = delete;
|
||||
PipeWriter& operator=(const PipeReader&&) = delete;
|
||||
|
||||
void send_message(std::string msg) {
|
||||
pipeio_send_msg(&this->pwriter, msg.c_str());
|
||||
}
|
||||
|
||||
private:
|
||||
pipe_writer pwriter;
|
||||
};
|
||||
|
||||
|
||||
class PipeIO {
|
||||
public:
|
||||
PipeReader reader;
|
||||
PipeWriter writer;
|
||||
|
||||
PipeIO(const std::string& in_pipe_path, const std::string& out_pipe_path)
|
||||
:reader(in_pipe_path), writer(out_pipe_path) {}
|
||||
|
||||
PipeIO(PipeReader&) = delete;
|
||||
PipeIO(PipeReader&&) = delete;
|
||||
PipeIO& operator=(const PipeReader&) = delete;
|
||||
PipeIO& operator=(const PipeReader&&) = delete;
|
||||
};
|
14
pipe_io_server/clients/csharp/Example.cs
Normal file
14
pipe_io_server/clients/csharp/Example.cs
Normal file
@ -0,0 +1,14 @@
|
||||
using System;
|
||||
using Pipe.IO;
|
||||
|
||||
class Program
|
||||
{
|
||||
static void Main(string[] args)
|
||||
{
|
||||
using (var t = new PipeIO("in", "out"))
|
||||
{
|
||||
t.Reader.OnMessage += (String msg) => t.Writer.SendMessage(msg);
|
||||
t.Reader.Run();
|
||||
}
|
||||
}
|
||||
}
|
83
pipe_io_server/clients/csharp/PipeIO.cs
Normal file
83
pipe_io_server/clients/csharp/PipeIO.cs
Normal file
@ -0,0 +1,83 @@
|
||||
using System;
|
||||
using System.IO;
|
||||
|
||||
|
||||
namespace Pipe.IO
|
||||
{
|
||||
public class PipeReader : IDisposable
|
||||
{
|
||||
public delegate void MessageHandler(String msg);
|
||||
public event MessageHandler OnMessage;
|
||||
private StreamReader pipe;
|
||||
|
||||
public PipeReader(String pipe_path)
|
||||
{
|
||||
var fileStream = File.OpenRead(pipe_path);
|
||||
this.pipe = new StreamReader(fileStream);
|
||||
}
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
this.pipe.Close();
|
||||
}
|
||||
|
||||
public void Run()
|
||||
{
|
||||
String msg;
|
||||
while ((msg = this.RecvMessage()) != null)
|
||||
{
|
||||
if (this.OnMessage != null)
|
||||
{
|
||||
this.OnMessage(msg);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public String RecvMessage()
|
||||
{
|
||||
return this.pipe.ReadLine();
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
public class PipeWriter : IDisposable
|
||||
{
|
||||
private StreamWriter pipe;
|
||||
|
||||
public PipeWriter(String pipe)
|
||||
{
|
||||
var fileStream = File.OpenWrite(pipe);
|
||||
this.pipe = new StreamWriter(fileStream);
|
||||
}
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
this.pipe.Close();
|
||||
}
|
||||
|
||||
public void SendMessage(String msg)
|
||||
{
|
||||
this.pipe.WriteLine(msg);
|
||||
this.pipe.Flush();
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
public class PipeIO : IDisposable
|
||||
{
|
||||
public PipeReader Reader;
|
||||
public PipeWriter Writer;
|
||||
|
||||
public PipeIO(String in_pipe_path, String out_pipe_path)
|
||||
{
|
||||
this.Reader = new PipeReader(in_pipe_path);
|
||||
this.Writer = new PipeWriter(out_pipe_path);
|
||||
}
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
this.Reader.Dispose();
|
||||
this.Writer.Dispose();
|
||||
}
|
||||
}
|
||||
}
|
20
pipe_io_server/clients/go/example.go
Normal file
20
pipe_io_server/clients/go/example.go
Normal file
@ -0,0 +1,20 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"log"
|
||||
"./pipeio"
|
||||
)
|
||||
|
||||
|
||||
func main() {
|
||||
pipeio, err := pipeio.NewPipeIO("in", "out")
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
defer pipeio.Close()
|
||||
|
||||
pipeio.Reader.SetMessageHandler(func(msg []byte) {
|
||||
pipeio.Writer.SendMessage(msg)
|
||||
})
|
||||
pipeio.Reader.Run()
|
||||
}
|
116
pipe_io_server/clients/go/pipeio/pipeio.go
Normal file
116
pipe_io_server/clients/go/pipeio/pipeio.go
Normal file
@ -0,0 +1,116 @@
|
||||
package pipeio
|
||||
|
||||
import (
|
||||
"os"
|
||||
"io"
|
||||
"bufio"
|
||||
"errors"
|
||||
)
|
||||
|
||||
const defaultBufSize = 65536
|
||||
|
||||
|
||||
type pipeReader struct {
|
||||
pipe *os.File
|
||||
scanner *bufio.Scanner
|
||||
messageHandler messageHandlerFunc
|
||||
}
|
||||
|
||||
type messageHandlerFunc func([]byte)
|
||||
|
||||
func NewPipeReader(pipePath string) (*pipeReader, error) {
|
||||
pipe, err := os.Open(pipePath)
|
||||
scanner := bufio.NewScanner(pipe)
|
||||
scanner.Split(bufio.ScanLines)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
scanner.Buffer(make([]byte, defaultBufSize), defaultBufSize)
|
||||
return &pipeReader{
|
||||
pipe: pipe,
|
||||
scanner: scanner,
|
||||
messageHandler: func([]byte){},
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (preader *pipeReader) Close() {
|
||||
preader.pipe.Close()
|
||||
}
|
||||
|
||||
func (preader *pipeReader) SetMessageHandler(fun messageHandlerFunc) {
|
||||
preader.messageHandler = fun
|
||||
}
|
||||
|
||||
func (preader *pipeReader) Run() {
|
||||
for {
|
||||
msg := preader.RecvMessage()
|
||||
if msg == nil {
|
||||
break
|
||||
}
|
||||
preader.messageHandler(msg)
|
||||
}
|
||||
}
|
||||
|
||||
func (preader *pipeReader) RecvMessage() []byte {
|
||||
if preader.scanner.Scan() {
|
||||
return preader.scanner.Bytes()
|
||||
}
|
||||
if err := preader.scanner.Err(); err != nil && err != io.EOF {
|
||||
if err == bufio.ErrTooLong {
|
||||
err = errors.New("pipeReader buffer too small!")
|
||||
}
|
||||
panic(err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (preader *pipeReader) BufSize(size int) {
|
||||
preader.scanner.Buffer(make([]byte, defaultBufSize), size)
|
||||
}
|
||||
|
||||
|
||||
type pipeWriter struct {
|
||||
pipe *os.File
|
||||
}
|
||||
|
||||
func NewPipeWriter(pipePath string) (*pipeWriter, error) {
|
||||
pipe, err := os.OpenFile(pipePath, os.O_WRONLY, 0644)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &pipeWriter{pipe: pipe}, nil
|
||||
}
|
||||
|
||||
func (pwriter *pipeWriter) Close() {
|
||||
pwriter.pipe.Close()
|
||||
}
|
||||
|
||||
func (pwriter *pipeWriter) SendMessage(msg []byte) {
|
||||
pwriter.pipe.Write(append(msg, '\n'))
|
||||
}
|
||||
|
||||
|
||||
type pipeIO struct {
|
||||
Reader pipeReader
|
||||
Writer pipeWriter
|
||||
}
|
||||
|
||||
func NewPipeIO(inPipePath, outPipePath string) (*pipeIO, error) {
|
||||
reader, err := NewPipeReader(inPipePath)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
writer, err := NewPipeWriter(outPipePath)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &pipeIO{
|
||||
Reader: *reader,
|
||||
Writer: *writer,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (pipeio *pipeIO) Close() {
|
||||
pipeio.Reader.Close()
|
||||
pipeio.Writer.Close()
|
||||
}
|
13
pipe_io_server/clients/java/Example.java
Normal file
13
pipe_io_server/clients/java/Example.java
Normal file
@ -0,0 +1,13 @@
|
||||
public class Example {
|
||||
public static void main(String[] args) {
|
||||
try (PipeIO pipeio = new PipeIO("in", "out")) {
|
||||
pipeio.reader.setMessageHandler(new MessageHandler(){
|
||||
@Override
|
||||
public void call(String msg) {
|
||||
pipeio.writer.sendMessage(msg);
|
||||
}
|
||||
});
|
||||
pipeio.reader.run();
|
||||
}
|
||||
}
|
||||
}
|
3
pipe_io_server/clients/java/MessageHandler.java
Normal file
3
pipe_io_server/clients/java/MessageHandler.java
Normal file
@ -0,0 +1,3 @@
|
||||
public interface MessageHandler {
|
||||
public void call(String msg);
|
||||
}
|
17
pipe_io_server/clients/java/PipeIO.java
Normal file
17
pipe_io_server/clients/java/PipeIO.java
Normal file
@ -0,0 +1,17 @@
|
||||
import java.io.IOException;
|
||||
|
||||
|
||||
public class PipeIO implements AutoCloseable {
|
||||
public PipeReader reader;
|
||||
public PipeWriter writer;
|
||||
|
||||
public PipeIO(String in_pipe_path, String out_pipe_path) {
|
||||
this.reader = new PipeReader(in_pipe_path);
|
||||
this.writer = new PipeWriter(out_pipe_path);
|
||||
}
|
||||
|
||||
public void close() {
|
||||
this.reader.close();
|
||||
this.writer.close();
|
||||
}
|
||||
}
|
50
pipe_io_server/clients/java/PipeReader.java
Normal file
50
pipe_io_server/clients/java/PipeReader.java
Normal file
@ -0,0 +1,50 @@
|
||||
import java.io.BufferedReader;
|
||||
import java.io.FileReader;
|
||||
import java.io.IOException;
|
||||
import java.io.UncheckedIOException;
|
||||
import java.lang.AutoCloseable;
|
||||
|
||||
|
||||
public class PipeReader implements AutoCloseable {
|
||||
private BufferedReader pipe;
|
||||
private MessageHandler messageHandler;
|
||||
|
||||
public PipeReader(String pipe_path) {
|
||||
try {
|
||||
this.pipe = new BufferedReader(new FileReader(pipe_path));
|
||||
} catch (IOException e) {
|
||||
throw new UncheckedIOException(e);
|
||||
}
|
||||
this.messageHandler = new MessageHandler(){
|
||||
@Override
|
||||
public void call(String msg) {}
|
||||
};
|
||||
}
|
||||
|
||||
public void close() {
|
||||
try {
|
||||
this.pipe.close();
|
||||
} catch (IOException e) {
|
||||
throw new UncheckedIOException(e);
|
||||
}
|
||||
}
|
||||
|
||||
public void setMessageHandler(MessageHandler handler) {
|
||||
this.messageHandler = handler;
|
||||
}
|
||||
|
||||
public void run() {
|
||||
String msg;
|
||||
while ((msg = this.recvMessage()) != null) {
|
||||
this.messageHandler.call(msg);
|
||||
}
|
||||
}
|
||||
|
||||
public String recvMessage() {
|
||||
try {
|
||||
return this.pipe.readLine();
|
||||
} catch (IOException e) {
|
||||
throw new UncheckedIOException(e);
|
||||
}
|
||||
}
|
||||
}
|
35
pipe_io_server/clients/java/PipeWriter.java
Normal file
35
pipe_io_server/clients/java/PipeWriter.java
Normal file
@ -0,0 +1,35 @@
|
||||
import java.io.BufferedWriter;
|
||||
import java.io.FileWriter;
|
||||
import java.io.IOException;
|
||||
import java.io.UncheckedIOException;
|
||||
import java.lang.AutoCloseable;
|
||||
|
||||
|
||||
public class PipeWriter implements AutoCloseable {
|
||||
private BufferedWriter pipe;
|
||||
|
||||
public PipeWriter(String pipe_path) {
|
||||
try {
|
||||
this.pipe = new BufferedWriter(new FileWriter(pipe_path));
|
||||
} catch(IOException e) {
|
||||
throw new UncheckedIOException(e);
|
||||
}
|
||||
}
|
||||
|
||||
public void close() {
|
||||
try {
|
||||
this.pipe.close();
|
||||
} catch (IOException e) {
|
||||
throw new UncheckedIOException(e);
|
||||
}
|
||||
}
|
||||
|
||||
public void sendMessage(String msg) {
|
||||
try {
|
||||
this.pipe.write(msg + "\n");
|
||||
this.pipe.flush();
|
||||
} catch (IOException e) {
|
||||
throw new UncheckedIOException(e);
|
||||
}
|
||||
}
|
||||
}
|
7
pipe_io_server/clients/python/example.py
Normal file
7
pipe_io_server/clients/python/example.py
Normal file
@ -0,0 +1,7 @@
|
||||
from pipe_io import PipeIO
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
with PipeIO('in', 'out') as pipeio:
|
||||
pipeio.reader.message_handler = pipeio.writer.send_message
|
||||
pipeio.reader.run()
|
69
pipe_io_server/clients/python/pipe_io.py
Normal file
69
pipe_io_server/clients/python/pipe_io.py
Normal file
@ -0,0 +1,69 @@
|
||||
from typing import Callable
|
||||
|
||||
|
||||
class PipeReader:
|
||||
def __init__(self, pipe_path):
|
||||
self._pipe = open(pipe_path, 'rb')
|
||||
self._message_handler = lambda msg: None
|
||||
|
||||
def __enter__(self):
|
||||
return self
|
||||
|
||||
def __exit__(self, type_, value, traceback):
|
||||
self.close()
|
||||
|
||||
def close(self):
|
||||
self._pipe.close()
|
||||
|
||||
@property
|
||||
def message_handler(self):
|
||||
return self._message_handler
|
||||
|
||||
@message_handler.setter
|
||||
def message_handler(self, value):
|
||||
if not isinstance(value, Callable):
|
||||
raise ValueError("message_handler must be callable!")
|
||||
self._message_handler = value
|
||||
|
||||
def run(self):
|
||||
msg = self.recv_message()
|
||||
while msg:
|
||||
self._message_handler(msg)
|
||||
msg = self.recv_message()
|
||||
|
||||
def recv_message(self):
|
||||
return self._pipe.readline()[:-1]
|
||||
|
||||
|
||||
class PipeWriter:
|
||||
def __init__(self, pipe_path):
|
||||
self._pipe = open(pipe_path, 'wb')
|
||||
|
||||
def __enter__(self):
|
||||
return self
|
||||
|
||||
def __exit__(self, type_, value, traceback):
|
||||
self.close()
|
||||
|
||||
def close(self):
|
||||
self._pipe.close()
|
||||
|
||||
def send_message(self, message):
|
||||
self._pipe.write(message + b'\n')
|
||||
self._pipe.flush()
|
||||
|
||||
|
||||
class PipeIO:
|
||||
def __init__(self, in_pipe_path, out_pipe_path):
|
||||
self.reader = PipeReader(in_pipe_path)
|
||||
self.writer = PipeWriter(out_pipe_path)
|
||||
|
||||
def __enter__(self):
|
||||
return self
|
||||
|
||||
def __exit__(self, type_, value, traceback):
|
||||
self.close()
|
||||
|
||||
def close(self):
|
||||
self.reader.close()
|
||||
self.writer.close()
|
Reference in New Issue
Block a user