refactor(controller): unify backend initialization in log, stats, and user handlers

Author: ImMohammad20000

controller/rpc/log.go

@@ -8,7 +8,12 @@ import (
 )
 
 func (s *Service) GetLogs(_ *common.Empty, stream common.NodeService_GetLogsServer) error {
-	logChan := s.Backend().Logs()
+	backend, err := s.backend()
+	if err != nil {
+		return err
+	}
+
+	logChan := backend.Logs()
 
 	for {
 		select {

controller/rpc/stats.go

@@ -3,11 +3,27 @@ package rpc
 import (
 	"context"
 
+	"github.com/pasarguard/node/backend"
 	"github.com/pasarguard/node/common"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
 )
 
+func (s *Service) backend() (backend.Backend, error) {
+	backend := s.Backend()
+	if backend == nil {
+		return nil, status.Errorf(codes.Unavailable, "backend not initialized")
+	}
+	return backend, nil
+}
+
 func (s *Service) GetStats(ctx context.Context, request *common.StatRequest) (*common.StatResponse, error) {
-	stats, err := s.Backend().GetStats(ctx, request)
+	backend, err := s.backend()
+	if err != nil {
+		return nil, err
+	}
+
+	stats, err := backend.GetStats(ctx, request)
 	if err != nil {
 		err = common.InterceptNotFound(err)
 		return nil, err
@@ -16,7 +32,12 @@ func (s *Service) GetStats(ctx context.Context, request *common.StatRequest) (*c
 }
 
 func (s *Service) GetUserOnlineStats(ctx context.Context, request *common.StatRequest) (*common.OnlineStatResponse, error) {
-	stats, err := s.Backend().GetUserOnlineStats(ctx, request.GetName())
+	backend, err := s.backend()
+	if err != nil {
+		return nil, err
+	}
+
+	stats, err := backend.GetUserOnlineStats(ctx, request.GetName())
 	if err != nil {
 		err = common.InterceptNotFound(err)
 		return nil, err
@@ -25,7 +46,12 @@ func (s *Service) GetUserOnlineStats(ctx context.Context, request *common.StatRe
 }
 
 func (s *Service) GetUserOnlineIpListStats(ctx context.Context, request *common.StatRequest) (*common.StatsOnlineIpListResponse, error) {
-	stats, err := s.Backend().GetUserOnlineIpListStats(ctx, request.GetName())
+	backend, err := s.backend()
+	if err != nil {
+		return nil, err
+	}
+
+	stats, err := backend.GetUserOnlineIpListStats(ctx, request.GetName())
 	if err != nil {
 		err = common.InterceptNotFound(err)
 		return nil, err
@@ -34,7 +60,12 @@ func (s *Service) GetUserOnlineIpListStats(ctx context.Context, request *common.
 }
 
 func (s *Service) GetBackendStats(ctx context.Context, _ *common.Empty) (*common.BackendStatsResponse, error) {
-	return s.Backend().GetSysStats(ctx)
+	backend, err := s.backend()
+	if err != nil {
+		return nil, err
+	}
+
+	return backend.GetSysStats(ctx)
 }
 
 func (s *Service) GetOutboundsLatency(ctx context.Context, request *common.LatencyRequest) (*common.LatencyResponse, error) {

controller/rpc/user.go

@@ -15,6 +15,11 @@ import (
 )
 
 func (s *Service) SyncUser(stream grpc.ClientStreamingServer[common.User, common.Empty]) error {
+	backend, err := s.backend()
+	if err != nil {
+		return err
+	}
+
 	for {
 		user, err := stream.Recv()
 		if err != nil {
@@ -27,15 +32,20 @@ func (s *Service) SyncUser(stream grpc.ClientStreamingServer[common.User, common
 
 		log.Printf("Got user: %v", user.GetEmail())
 
-		if err = s.Backend().SyncUser(stream.Context(), user); err != nil {
+		if err = backend.SyncUser(stream.Context(), user); err != nil {
 			log.Printf("Error syncing user: %v", err)
 			return status.Errorf(codes.Internal, "failed to update user: %v", err)
 		}
 	}
 }
 
 func (s *Service) SyncUsers(ctx context.Context, users *common.Users) (*common.Empty, error) {
-	if err := s.Backend().SyncUsers(ctx, users.GetUsers()); err != nil {
+	backend, err := s.backend()
+	if err != nil {
+		return nil, err
+	}
+
+	if err := backend.SyncUsers(ctx, users.GetUsers()); err != nil {
 		return nil, err
 	}
 
@@ -72,14 +82,19 @@ func (s *Service) SyncUsersChunked(stream grpc.ClientStreamingServer[common.User
 		return status.Error(codes.InvalidArgument, err.Error())
 	}
 
+	backend, err := s.backend()
+	if err != nil {
+		return err
+	}
+
 	// Large chunk: update in-memory then restart (no API calls).
 	if len(users) > 100 {
-		if err := s.Backend().UpdateUsersAndRestart(stream.Context(), users); err != nil {
+		if err := backend.UpdateUsersAndRestart(stream.Context(), users); err != nil {
 			return status.Errorf(codes.Internal, "failed to update users: %v", err)
 		}
 	} else {
 		// Small chunk: update via API without restart.
-		if err := s.Backend().UpdateUsers(stream.Context(), users); err != nil {
+		if err := backend.UpdateUsers(stream.Context(), users); err != nil {
 			return status.Errorf(codes.Internal, "failed to update users: %v", err)
 		}
 	}