package PgTest

import (
	"context"

	"demo/internal/svc"
	"demo/internal/types"
	Pgmodel "demo/model/pgTest" // 引入model

	"github.com/google/uuid"
	"github.com/zeromicro/go-zero/core/logx"
)

type PgInsertTestLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewPgInsertTestLogic(ctx context.Context, svcCtx *svc.ServiceContext) *PgInsertTestLogic {
	return &PgInsertTestLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *PgInsertTestLogic) PgInsertTest() (resp *types.TestResponse, err error) {
	// 组装数据
	uid := uuid.New()
	server := l.svcCtx.Config.Name
	// insertData := new (Pgmodel.Test)
	// insertData.Id = uid.String()
	// insertData.Server = server
	insertData := &Pgmodel.Test{
		Id:     uid.String(),
		Server: server,
	}

	// 插入数据库
	_, err = l.svcCtx.PgModel.Insert(l.ctx, insertData)
	if err != nil {
		return nil, err
	}
	resp = &types.TestResponse{
		Success: true,
		Message: "插入成功",
	}

	return resp, nil
}