Лучшие практики Rust Daemon - PullRequest
Новая
       33

Лучшие практики Rust Daemon

2 голосов
/ 26 апреля 2020

У меня проблемы с созданием демона для сервера сокетов unix в ржавчине. На таком языке, как node.js, я бы просто создал отдельный процесс. Но ржавчина кажется немного более сложной.

(вся работа в unix средах)

Вот краткий пример. Я написал: 

use std::os::unix::net::{UnixListener, UnixStream};

use std::path::{Path};
use std::io::{Read, Write};
use std::{fs};
use std::thread;


fn handle_stream(mut stream: UnixStream) {

    loop {
        let mut read = [0; 1028];
        match stream.read(&mut read) {
            Ok(n) => {
                if n == 0 {
                    // connection was closed
                    println!("connection closed?");
                    break;
                }
                let msg = String::from_utf8(read.to_vec()).unwrap();
                println!("SERVER: {} received from remote client.", msg);

                match stream.write_all(&read[0..n]) {
                    Ok(_) => {}
                    Err(e) => println!("Error writing to socket: {}", e),
                }
            }
            Err(err) => {
                panic!(err);
            }
        }
    }
    println!("SERVER: Ending connection with client.");
}

fn server_start() {

    // remove existing sock if exists
    let _did_remove = fs::remove_file("/Users/tom/desktop/app.sock");

    // socket location
    let socket_file = "/Users/tom/desktop/app.sock";
    let socket = Path::new(socket_file);

    // Bind to socket
    let listener = match UnixListener::bind(&socket) {
        Err(_) => panic!("failed to bind socket"),
        Ok(stream) => stream,
    };

    println!("Server started, waiting for clients");

    for conn in listener.incoming() {
        match conn {
            Ok(stream) => {
                // spawn a new thread for each incomming
                thread::spawn(|| handle_stream(stream));
            }
            Err(err) => {
                println!("Error connecting to client: {}", err);
                break;
            }
        }
    }
}

fn main() {

    server_start();
}

1 Ответ

2 голосов
/ 28 апреля 2020

Следуя сообщениям в комментариях, я решил использовать службу systemd вместо создания своего собственного демона. Кажется, это идеальный способ управления фоновыми задачами. Я отредактировал верхний код так, чтобы он имел смысл с ответом.

Systemd - linux

Вам потребуется создать файл .service и поместить его в каталог демона systemd. Например: /etc/systemd/system/test.service

Затем обновите права доступа к файлу: 

sudo chmod 644 /etc/systemd/system/test.service

Для запуска службы: 

sudo systemctl start service_name

Сервисный код: 

[Unit]
Description=Test service
After=network.target
StartLimitIntervalSec=0
[Service]
Type=simple
Restart=always
RestartSec=1
User=username
ExecStart=/usr/bin/env test

[Install]
WantedBy=multi-user.target

Launchctl - macOS

Для macOS нам нужно создать файл .plist и поместить его в каталог демонов запуска. Например: /Library/LaunchDaemons/test.plist

Следующее обновление прав доступа к файлу: 

sudo chown root:wheel /Library/LaunchDaemons/com.test.daemon.plist

Загрузка демона: 

launchctl load /Library/LaunchDaemons/com.test.daemon.plist

Запуск демона: 

launchctl start /Library/LaunchDaemons/com.test.daemon

Код Plist: 

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE plist PUBLIC "-//Apple Computer//DTD PLIST 1.0//EN"
    "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
<plist version="1.0">
<dict>
    <key>Label</key>
        <string>com.test.daemon</string>
    <key>ServiceDescription</key>
        <string>Test Server</string>
    <key>ProgramArguments</key>
        <array>             
            <string>/Users/tom/desktop/test/target/debug/test</string>
        </array>
    <key>RunAtLoad</key>
        <false/>
</dict>
</plist>
...