mirror of
https://github.com/psviderski/uncloud.git
synced 2026-08-27 19:43:34 +00:00
split cluster Server into Machine and Server
This commit is contained in:
@@ -0,0 +1,54 @@
|
||||
package machine
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"google.golang.org/grpc"
|
||||
"log/slog"
|
||||
"net"
|
||||
"uncloud/internal/machine/api/pb"
|
||||
"uncloud/internal/machine/cluster"
|
||||
)
|
||||
|
||||
type Config struct {
|
||||
// DataDir is the directory where the machine stores its persistent state.
|
||||
DataDir string
|
||||
APIAddr string
|
||||
APISockPath string
|
||||
}
|
||||
|
||||
type Machine struct {
|
||||
config Config
|
||||
server *grpc.Server
|
||||
}
|
||||
|
||||
func NewMachine(config *Config) (*Machine, error) {
|
||||
m := &Machine{
|
||||
config: *config,
|
||||
server: grpc.NewServer(),
|
||||
}
|
||||
|
||||
clusterState := cluster.NewState(cluster.StatePath(config.DataDir))
|
||||
clusterServer := cluster.NewServer(clusterState)
|
||||
pb.RegisterClusterServer(m.server, clusterServer)
|
||||
|
||||
return m, nil
|
||||
}
|
||||
|
||||
func (m *Machine) Run() error {
|
||||
listener, err := net.Listen("tcp", m.config.APIAddr)
|
||||
if err != nil {
|
||||
return fmt.Errorf("listen API port: %w", err)
|
||||
}
|
||||
slog.Info("Starting API server.", "addr", m.config.APIAddr)
|
||||
if err = m.server.Serve(listener); err != nil {
|
||||
return fmt.Errorf("API server failed: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *Machine) Stop() {
|
||||
slog.Info("Stopping API server.")
|
||||
// TODO: implement timeout for graceful shutdown.
|
||||
m.server.GracefulStop()
|
||||
slog.Info("API server stopped.")
|
||||
}
|
||||
Reference in New Issue
Block a user