APIPark/module/upstream/iml.go

package upstream

import (
	"context"
	"errors"
	"fmt"

	"github.com/APIParkLab/APIPark/service/universally/commit"
	"github.com/eolinker/go-common/utils"

	"github.com/APIParkLab/APIPark/service/cluster"
	"github.com/APIParkLab/APIPark/service/service"

	"gorm.io/gorm"

	"github.com/APIParkLab/APIPark/service/upstream"

	"github.com/eolinker/go-common/store"

	upstream_dto "github.com/APIParkLab/APIPark/module/upstream/dto"
)

var (
	_        IUpstreamModule       = (*imlUpstreamModule)(nil)
	_        IExportUpstreamModule = (*imlUpstreamModule)(nil)
	asServer                       = map[string]bool{
		"as_server": true,
	}
)

type imlUpstreamModule struct {
	serviceService  service.IServiceService   `autowired:""`
	upstreamService upstream.IUpstreamService `autowired:""`
	transaction     store.ITransaction        `autowired:""`
}

func (i *imlUpstreamModule) ExportAll(ctx context.Context) ([]*upstream_dto.ExportUpstream, error) {
	latestCommits, err := i.upstreamService.ListLatestCommit(ctx, cluster.DefaultClusterID)
	if err != nil {
		return nil, err
	}
	commitMap := utils.SliceToMap(latestCommits, func(c *commit.Commit[upstream.Config]) string {
		return c.Target
	})
	list, err := i.upstreamService.List(ctx)
	if err != nil {
		return nil, err
	}
	items := make([]*upstream_dto.ExportUpstream, 0, len(list))
	for _, u := range list {
		c, ok := commitMap[u.UUID]
		if !ok {
			continue
		}

		items = append(items, &upstream_dto.ExportUpstream{
			ID:       u.UUID,
			Name:     u.Name,
			Service:  u.Service,
			Upstream: upstream_dto.FromClusterConfig(c.Data),
		})
	}
	return items, nil
}

func (i *imlUpstreamModule) Get(ctx context.Context, pid string) (upstream_dto.UpstreamConfig, error) {
	_, err := i.serviceService.Check(ctx, pid, asServer)
	if err != nil {
		return nil, err
	}
	_, err = i.upstreamService.Get(ctx, pid)
	if err != nil {
		if !errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, err
		}
		return nil, nil
	}
	commit, err := i.upstreamService.LatestCommit(ctx, pid, cluster.DefaultClusterID)
	if err != nil {
		if !errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, err
		}
		return nil, nil
	}

	return upstream_dto.FromClusterConfig(commit.Data), nil
}

func (i *imlUpstreamModule) Save(ctx context.Context, pid string, upstreamConfig upstream_dto.UpstreamConfig) (upstream_dto.UpstreamConfig, error) {
	pInfo, err := i.serviceService.Check(ctx, pid, asServer)
	if err != nil {
		return nil, err
	}

	err = i.transaction.Transaction(ctx, func(ctx context.Context) error {
		err = i.upstreamService.SaveCommit(ctx, pid, cluster.DefaultClusterID, upstream_dto.ConvertUpstream(upstreamConfig))
		if err != nil {
			return err
		}

		return i.upstreamService.Save(ctx, &upstream.SaveUpstream{
			UUID:    pid,
			Name:    fmt.Sprintf("upstream-%s", pid),
			Service: pid,
			Team:    pInfo.Team,
		})

	})
	if err != nil {
		return nil, err
	}
	return i.Get(ctx, pid)
}