camper/pkg/database/types.go

82 lines
1.9 KiB
Go

/*
* SPDX-FileCopyrightText: 2023 jordi fita mas <jfita@peritasoft.com>
* SPDX-License-Identifier: AGPL-3.0-only
*/
package database
import (
"context"
"github.com/jackc/pgtype"
"github.com/jackc/pgx/v4"
)
const (
RedsysRequestTypeName = "redsys_request"
RedsysSignedRequestTypeName = "redsys_signed_request"
)
var (
oidCache = make(map[string]uint32)
)
func registerTypes(ctx context.Context, conn *pgx.Conn) error {
uriOID, err := registerType(ctx, conn, &pgtype.Text{}, "uri")
if err != nil {
return err
}
redsysRequestType, err := pgtype.NewCompositeType(
RedsysRequestTypeName,
[]pgtype.CompositeTypeField{
{"transaction_type", pgtype.Int4OID},
{"amount", pgtype.TextOID},
{"order_number", pgtype.TextOID},
{"product", pgtype.TextOID},
{"card_holder", pgtype.TextOID},
{"success_uri", uriOID},
{"failure_uri", uriOID},
{"notification_uri", uriOID},
{"lang_tag", pgtype.TextOID},
},
conn.ConnInfo(),
)
if err != nil {
return err
}
if _, err = registerType(ctx, conn, redsysRequestType, redsysRequestType.TypeName()); err != nil {
return err
}
redsysSignedRequestType, err := pgtype.NewCompositeType(
RedsysSignedRequestTypeName,
[]pgtype.CompositeTypeField{
{"merchant_parameters", pgtype.TextOID},
{"signature", pgtype.TextOID},
{"signature_version", pgtype.TextOID},
},
conn.ConnInfo(),
)
if err != nil {
return err
}
if _, err = registerType(ctx, conn, redsysSignedRequestType, redsysSignedRequestType.TypeName()); err != nil {
return err
}
return nil
}
func registerType(ctx context.Context, conn *pgx.Conn, value pgtype.Value, name string) (oid uint32, err error) {
var found bool
if oid, found = oidCache[name]; !found {
if err = conn.QueryRow(ctx, "select $1::regtype::oid", name).Scan(&oid); err != nil {
return
}
oidCache[name] = oid
}
conn.ConnInfo().RegisterDataType(pgtype.DataType{Value: value, Name: name, OID: oid})
return
}