frostfs-node/cmd/frostfs-node/accounting.go

package main

import (
	"context"
	"net"
	"strings"

	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/morph/client/balance"
	accountingTransportGRPC "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/network/transport/accounting/grpc"
	accountingService "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/accounting"
	accounting "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/accounting/morph"
	accountingGRPC "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/accounting/grpc"
	"google.golang.org/grpc"
)

func initAccountingService(ctx context.Context, c *cfg) {
	c.initMorphComponents(ctx)

	balanceMorphWrapper, err := balance.NewFromMorph(c.cfgMorph.client, c.cfgAccounting.scriptHash, 0)
	fatalOnErr(err)

	server := accountingTransportGRPC.New(
		accountingService.NewSignService(
			&c.key.PrivateKey,
			accountingService.NewExecutionService(
				accounting.NewExecutor(balanceMorphWrapper),
				c.respSvc,
			),
		),
	)

	c.cfgGRPC.performAndSave(func(_ string, _ net.Listener, s *grpc.Server) {
		accountingGRPC.RegisterAccountingServiceServer(s, server)

		// TODO(@aarifullin): #1487 remove the dual service support.
		s.RegisterService(frostFSServiceDesc(accountingGRPC.AccountingService_ServiceDesc), server)
	})
}

// frostFSServiceDesc creates a service descriptor with the new namespace for dual service support.
func frostFSServiceDesc(sd grpc.ServiceDesc) *grpc.ServiceDesc {
	sdLegacy := new(grpc.ServiceDesc)
	*sdLegacy = sd

	const (
		legacyNamespace           = "neo.fs.v2"
		apemanagerLegacyNamespace = "frostfs.v2"
		newNamespace              = "frost.fs"
	)

	if strings.HasPrefix(sd.ServiceName, legacyNamespace) {
		sdLegacy.ServiceName = strings.ReplaceAll(sd.ServiceName, legacyNamespace, newNamespace)
	} else if strings.HasPrefix(sd.ServiceName, apemanagerLegacyNamespace) {
		sdLegacy.ServiceName = strings.ReplaceAll(sd.ServiceName, apemanagerLegacyNamespace, newNamespace)
	}
	return sdLegacy
}