seata-go/pkg/rm/common/handler/rm_handler_facade.go

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package handler

import (
	"context"
	"sync"

	"github.com/seata/seata-go/pkg/protocol/branch"
	"github.com/seata/seata-go/pkg/protocol/message"
)

var (
	onceRMHandlerFacade = &sync.Once{}
	rmHandler           *RMHandlerFacade
)

type RMHandlerFacade struct {
	rmHandlerMap sync.Map
}

func GetRMHandlerFacadeInstance() *RMHandlerFacade {
	if rmHandler == nil {
		onceRMHandlerFacade.Do(func() {
			rmHandler = &RMHandlerFacade{}
		})
	}
	return rmHandler
}

// Handle branch commit response.
func (h *RMHandlerFacade) HandleBranchCommitRequest(ctx context.Context, request message.BranchCommitRequest) (*message.BranchCommitResponse, error) {
	return h.getRMHandler(request.BranchType).HandleBranchCommitRequest(ctx, request)
}

// Handle branch rollback response.
// TODO
func (h *RMHandlerFacade) HandleBranchRollbackRequest(ctx context.Context, request message.BranchRollbackRequest) (*message.BranchRollbackResponse, error) {
	return h.getRMHandler(request.BranchType).HandleBranchRollbackRequest(ctx, request)
}

// Handle delete undo log .
// TODO
func (h *RMHandlerFacade) HandleUndoLogDeleteRequest(ctx context.Context, request message.UndoLogDeleteRequest) error {
	return h.getRMHandler(request.BranchType).HandleUndoLogDeleteRequest(ctx, request)
}

func (h *RMHandlerFacade) RegisteRMHandler(handler *CommonRMHandler) {
	if handler == nil {
		return
	}
	h.rmHandlerMap.Store(handler.GetBranchType(), handler)
}

func (h *RMHandlerFacade) getRMHandler(branchType branch.BranchType) *CommonRMHandler {
	if handler, ok := h.rmHandlerMap.Load(branchType); ok {
		return handler.(*CommonRMHandler)
	}
	return nil
}