feat(examples/transaction): add transaction examples (#1836)
parent
24393ee655
commit
c9f1d7db6d
@ -0,0 +1,233 @@ |
||||
// Code generated by protoc-gen-go. DO NOT EDIT.
|
||||
// versions:
|
||||
// protoc-gen-go v1.27.1
|
||||
// protoc v3.17.3
|
||||
// source: api/transaction/v1/transaction.proto
|
||||
|
||||
package v1 |
||||
|
||||
import ( |
||||
_ "google.golang.org/genproto/googleapis/api/annotations" |
||||
protoreflect "google.golang.org/protobuf/reflect/protoreflect" |
||||
protoimpl "google.golang.org/protobuf/runtime/protoimpl" |
||||
reflect "reflect" |
||||
sync "sync" |
||||
) |
||||
|
||||
const ( |
||||
// Verify that this generated code is sufficiently up-to-date.
|
||||
_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion) |
||||
// Verify that runtime/protoimpl is sufficiently up-to-date.
|
||||
_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20) |
||||
) |
||||
|
||||
type CreateUserRequest struct { |
||||
state protoimpl.MessageState |
||||
sizeCache protoimpl.SizeCache |
||||
unknownFields protoimpl.UnknownFields |
||||
|
||||
Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` |
||||
Email string `protobuf:"bytes,2,opt,name=email,proto3" json:"email,omitempty"` |
||||
} |
||||
|
||||
func (x *CreateUserRequest) Reset() { |
||||
*x = CreateUserRequest{} |
||||
if protoimpl.UnsafeEnabled { |
||||
mi := &file_api_transaction_v1_transaction_proto_msgTypes[0] |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
} |
||||
|
||||
func (x *CreateUserRequest) String() string { |
||||
return protoimpl.X.MessageStringOf(x) |
||||
} |
||||
|
||||
func (*CreateUserRequest) ProtoMessage() {} |
||||
|
||||
func (x *CreateUserRequest) ProtoReflect() protoreflect.Message { |
||||
mi := &file_api_transaction_v1_transaction_proto_msgTypes[0] |
||||
if protoimpl.UnsafeEnabled && x != nil { |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
if ms.LoadMessageInfo() == nil { |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
return ms |
||||
} |
||||
return mi.MessageOf(x) |
||||
} |
||||
|
||||
// Deprecated: Use CreateUserRequest.ProtoReflect.Descriptor instead.
|
||||
func (*CreateUserRequest) Descriptor() ([]byte, []int) { |
||||
return file_api_transaction_v1_transaction_proto_rawDescGZIP(), []int{0} |
||||
} |
||||
|
||||
func (x *CreateUserRequest) GetName() string { |
||||
if x != nil { |
||||
return x.Name |
||||
} |
||||
return "" |
||||
} |
||||
|
||||
func (x *CreateUserRequest) GetEmail() string { |
||||
if x != nil { |
||||
return x.Email |
||||
} |
||||
return "" |
||||
} |
||||
|
||||
type CreateUserReply struct { |
||||
state protoimpl.MessageState |
||||
sizeCache protoimpl.SizeCache |
||||
unknownFields protoimpl.UnknownFields |
||||
|
||||
Id string `protobuf:"bytes,1,opt,name=id,proto3" json:"id,omitempty"` |
||||
} |
||||
|
||||
func (x *CreateUserReply) Reset() { |
||||
*x = CreateUserReply{} |
||||
if protoimpl.UnsafeEnabled { |
||||
mi := &file_api_transaction_v1_transaction_proto_msgTypes[1] |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
} |
||||
|
||||
func (x *CreateUserReply) String() string { |
||||
return protoimpl.X.MessageStringOf(x) |
||||
} |
||||
|
||||
func (*CreateUserReply) ProtoMessage() {} |
||||
|
||||
func (x *CreateUserReply) ProtoReflect() protoreflect.Message { |
||||
mi := &file_api_transaction_v1_transaction_proto_msgTypes[1] |
||||
if protoimpl.UnsafeEnabled && x != nil { |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
if ms.LoadMessageInfo() == nil { |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
return ms |
||||
} |
||||
return mi.MessageOf(x) |
||||
} |
||||
|
||||
// Deprecated: Use CreateUserReply.ProtoReflect.Descriptor instead.
|
||||
func (*CreateUserReply) Descriptor() ([]byte, []int) { |
||||
return file_api_transaction_v1_transaction_proto_rawDescGZIP(), []int{1} |
||||
} |
||||
|
||||
func (x *CreateUserReply) GetId() string { |
||||
if x != nil { |
||||
return x.Id |
||||
} |
||||
return "" |
||||
} |
||||
|
||||
var File_api_transaction_v1_transaction_proto protoreflect.FileDescriptor |
||||
|
||||
var file_api_transaction_v1_transaction_proto_rawDesc = []byte{ |
||||
0x0a, 0x24, 0x61, 0x70, 0x69, 0x2f, 0x74, 0x72, 0x61, 0x6e, 0x73, 0x61, 0x63, 0x74, 0x69, 0x6f, |
||||
0x6e, 0x2f, 0x76, 0x31, 0x2f, 0x74, 0x72, 0x61, 0x6e, 0x73, 0x61, 0x63, 0x74, 0x69, 0x6f, 0x6e, |
||||
0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x0b, 0x62, 0x6c, 0x6f, 0x67, 0x2e, 0x61, 0x70, 0x69, |
||||
0x2e, 0x76, 0x31, 0x1a, 0x1c, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2f, 0x61, 0x70, 0x69, 0x2f, |
||||
0x61, 0x6e, 0x6e, 0x6f, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x2e, 0x70, 0x72, 0x6f, 0x74, |
||||
0x6f, 0x22, 0x3d, 0x0a, 0x11, 0x43, 0x72, 0x65, 0x61, 0x74, 0x65, 0x55, 0x73, 0x65, 0x72, 0x52, |
||||
0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x12, 0x12, 0x0a, 0x04, 0x6e, 0x61, 0x6d, 0x65, 0x18, 0x01, |
||||
0x20, 0x01, 0x28, 0x09, 0x52, 0x04, 0x6e, 0x61, 0x6d, 0x65, 0x12, 0x14, 0x0a, 0x05, 0x65, 0x6d, |
||||
0x61, 0x69, 0x6c, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x05, 0x65, 0x6d, 0x61, 0x69, 0x6c, |
||||
0x22, 0x21, 0x0a, 0x0f, 0x43, 0x72, 0x65, 0x61, 0x74, 0x65, 0x55, 0x73, 0x65, 0x72, 0x52, 0x65, |
||||
0x70, 0x6c, 0x79, 0x12, 0x0e, 0x0a, 0x02, 0x69, 0x64, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, |
||||
0x02, 0x69, 0x64, 0x32, 0x7c, 0x0a, 0x12, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x61, 0x63, 0x74, 0x69, |
||||
0x6f, 0x6e, 0x53, 0x65, 0x72, 0x76, 0x69, 0x63, 0x65, 0x12, 0x66, 0x0a, 0x0a, 0x43, 0x72, 0x65, |
||||
0x61, 0x74, 0x65, 0x55, 0x73, 0x65, 0x72, 0x12, 0x1e, 0x2e, 0x62, 0x6c, 0x6f, 0x67, 0x2e, 0x61, |
||||
0x70, 0x69, 0x2e, 0x76, 0x31, 0x2e, 0x43, 0x72, 0x65, 0x61, 0x74, 0x65, 0x55, 0x73, 0x65, 0x72, |
||||
0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x1c, 0x2e, 0x62, 0x6c, 0x6f, 0x67, 0x2e, 0x61, |
||||
0x70, 0x69, 0x2e, 0x76, 0x31, 0x2e, 0x43, 0x72, 0x65, 0x61, 0x74, 0x65, 0x55, 0x73, 0x65, 0x72, |
||||
0x52, 0x65, 0x70, 0x6c, 0x79, 0x22, 0x1a, 0x82, 0xd3, 0xe4, 0x93, 0x02, 0x14, 0x22, 0x0f, 0x2f, |
||||
0x76, 0x31, 0x2f, 0x74, 0x72, 0x61, 0x6e, 0x73, 0x61, 0x63, 0x74, 0x69, 0x6f, 0x6e, 0x3a, 0x01, |
||||
0x2a, 0x42, 0x5e, 0x0a, 0x12, 0x74, 0x72, 0x61, 0x6e, 0x73, 0x61, 0x63, 0x74, 0x69, 0x6f, 0x6e, |
||||
0x2e, 0x61, 0x70, 0x69, 0x2e, 0x76, 0x31, 0x50, 0x01, 0x5a, 0x46, 0x67, 0x69, 0x74, 0x68, 0x75, |
||||
0x62, 0x2e, 0x63, 0x6f, 0x6d, 0x2f, 0x67, 0x6f, 0x2d, 0x6b, 0x72, 0x61, 0x74, 0x6f, 0x73, 0x2f, |
||||
0x6b, 0x72, 0x61, 0x74, 0x6f, 0x73, 0x2f, 0x65, 0x78, 0x61, 0x6d, 0x70, 0x6c, 0x65, 0x73, 0x2f, |
||||
0x74, 0x72, 0x61, 0x6e, 0x73, 0x61, 0x63, 0x74, 0x69, 0x6f, 0x6e, 0x2f, 0x61, 0x70, 0x69, 0x2f, |
||||
0x74, 0x72, 0x61, 0x6e, 0x73, 0x61, 0x63, 0x74, 0x69, 0x6f, 0x6e, 0x2f, 0x76, 0x31, 0x3b, 0x76, |
||||
0x31, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33, |
||||
} |
||||
|
||||
var ( |
||||
file_api_transaction_v1_transaction_proto_rawDescOnce sync.Once |
||||
file_api_transaction_v1_transaction_proto_rawDescData = file_api_transaction_v1_transaction_proto_rawDesc |
||||
) |
||||
|
||||
func file_api_transaction_v1_transaction_proto_rawDescGZIP() []byte { |
||||
file_api_transaction_v1_transaction_proto_rawDescOnce.Do(func() { |
||||
file_api_transaction_v1_transaction_proto_rawDescData = protoimpl.X.CompressGZIP(file_api_transaction_v1_transaction_proto_rawDescData) |
||||
}) |
||||
return file_api_transaction_v1_transaction_proto_rawDescData |
||||
} |
||||
|
||||
var file_api_transaction_v1_transaction_proto_msgTypes = make([]protoimpl.MessageInfo, 2) |
||||
var file_api_transaction_v1_transaction_proto_goTypes = []interface{}{ |
||||
(*CreateUserRequest)(nil), // 0: blog.api.v1.CreateUserRequest
|
||||
(*CreateUserReply)(nil), // 1: blog.api.v1.CreateUserReply
|
||||
} |
||||
var file_api_transaction_v1_transaction_proto_depIdxs = []int32{ |
||||
0, // 0: blog.api.v1.TransactionService.CreateUser:input_type -> blog.api.v1.CreateUserRequest
|
||||
1, // 1: blog.api.v1.TransactionService.CreateUser:output_type -> blog.api.v1.CreateUserReply
|
||||
1, // [1:2] is the sub-list for method output_type
|
||||
0, // [0:1] is the sub-list for method input_type
|
||||
0, // [0:0] is the sub-list for extension type_name
|
||||
0, // [0:0] is the sub-list for extension extendee
|
||||
0, // [0:0] is the sub-list for field type_name
|
||||
} |
||||
|
||||
func init() { file_api_transaction_v1_transaction_proto_init() } |
||||
func file_api_transaction_v1_transaction_proto_init() { |
||||
if File_api_transaction_v1_transaction_proto != nil { |
||||
return |
||||
} |
||||
if !protoimpl.UnsafeEnabled { |
||||
file_api_transaction_v1_transaction_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} { |
||||
switch v := v.(*CreateUserRequest); i { |
||||
case 0: |
||||
return &v.state |
||||
case 1: |
||||
return &v.sizeCache |
||||
case 2: |
||||
return &v.unknownFields |
||||
default: |
||||
return nil |
||||
} |
||||
} |
||||
file_api_transaction_v1_transaction_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} { |
||||
switch v := v.(*CreateUserReply); i { |
||||
case 0: |
||||
return &v.state |
||||
case 1: |
||||
return &v.sizeCache |
||||
case 2: |
||||
return &v.unknownFields |
||||
default: |
||||
return nil |
||||
} |
||||
} |
||||
} |
||||
type x struct{} |
||||
out := protoimpl.TypeBuilder{ |
||||
File: protoimpl.DescBuilder{ |
||||
GoPackagePath: reflect.TypeOf(x{}).PkgPath(), |
||||
RawDescriptor: file_api_transaction_v1_transaction_proto_rawDesc, |
||||
NumEnums: 0, |
||||
NumMessages: 2, |
||||
NumExtensions: 0, |
||||
NumServices: 1, |
||||
}, |
||||
GoTypes: file_api_transaction_v1_transaction_proto_goTypes, |
||||
DependencyIndexes: file_api_transaction_v1_transaction_proto_depIdxs, |
||||
MessageInfos: file_api_transaction_v1_transaction_proto_msgTypes, |
||||
}.Build() |
||||
File_api_transaction_v1_transaction_proto = out.File |
||||
file_api_transaction_v1_transaction_proto_rawDesc = nil |
||||
file_api_transaction_v1_transaction_proto_goTypes = nil |
||||
file_api_transaction_v1_transaction_proto_depIdxs = nil |
||||
} |
@ -0,0 +1,27 @@ |
||||
syntax = "proto3"; |
||||
|
||||
package blog.api.v1; |
||||
|
||||
option go_package = "github.com/go-kratos/kratos/examples/transaction/api/transaction/v1;v1"; |
||||
option java_multiple_files = true; |
||||
option java_package = "transaction.api.v1"; |
||||
|
||||
import "google/api/annotations.proto"; |
||||
|
||||
service TransactionService { |
||||
rpc CreateUser(CreateUserRequest) returns (CreateUserReply) { |
||||
option (google.api.http) = { |
||||
post: "/v1/transaction" |
||||
body: "*" |
||||
}; |
||||
} |
||||
} |
||||
|
||||
message CreateUserRequest { |
||||
string name = 1; |
||||
string email = 2; |
||||
} |
||||
|
||||
message CreateUserReply { |
||||
string id = 1; |
||||
} |
@ -0,0 +1,105 @@ |
||||
// Code generated by protoc-gen-go-grpc. DO NOT EDIT.
|
||||
// versions:
|
||||
// - protoc-gen-go-grpc v1.2.0
|
||||
// - protoc v3.17.3
|
||||
// source: api/transaction/v1/transaction.proto
|
||||
|
||||
package v1 |
||||
|
||||
import ( |
||||
context "context" |
||||
grpc "google.golang.org/grpc" |
||||
codes "google.golang.org/grpc/codes" |
||||
status "google.golang.org/grpc/status" |
||||
) |
||||
|
||||
// This is a compile-time assertion to ensure that this generated file
|
||||
// is compatible with the grpc package it is being compiled against.
|
||||
// Requires gRPC-Go v1.32.0 or later.
|
||||
const _ = grpc.SupportPackageIsVersion7 |
||||
|
||||
// TransactionServiceClient is the client API for TransactionService service.
|
||||
//
|
||||
// For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream.
|
||||
type TransactionServiceClient interface { |
||||
CreateUser(ctx context.Context, in *CreateUserRequest, opts ...grpc.CallOption) (*CreateUserReply, error) |
||||
} |
||||
|
||||
type transactionServiceClient struct { |
||||
cc grpc.ClientConnInterface |
||||
} |
||||
|
||||
func NewTransactionServiceClient(cc grpc.ClientConnInterface) TransactionServiceClient { |
||||
return &transactionServiceClient{cc} |
||||
} |
||||
|
||||
func (c *transactionServiceClient) CreateUser(ctx context.Context, in *CreateUserRequest, opts ...grpc.CallOption) (*CreateUserReply, error) { |
||||
out := new(CreateUserReply) |
||||
err := c.cc.Invoke(ctx, "/blog.api.v1.TransactionService/CreateUser", in, out, opts...) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
return out, nil |
||||
} |
||||
|
||||
// TransactionServiceServer is the server API for TransactionService service.
|
||||
// All implementations must embed UnimplementedTransactionServiceServer
|
||||
// for forward compatibility
|
||||
type TransactionServiceServer interface { |
||||
CreateUser(context.Context, *CreateUserRequest) (*CreateUserReply, error) |
||||
mustEmbedUnimplementedTransactionServiceServer() |
||||
} |
||||
|
||||
// UnimplementedTransactionServiceServer must be embedded to have forward compatible implementations.
|
||||
type UnimplementedTransactionServiceServer struct { |
||||
} |
||||
|
||||
func (UnimplementedTransactionServiceServer) CreateUser(context.Context, *CreateUserRequest) (*CreateUserReply, error) { |
||||
return nil, status.Errorf(codes.Unimplemented, "method CreateUser not implemented") |
||||
} |
||||
func (UnimplementedTransactionServiceServer) mustEmbedUnimplementedTransactionServiceServer() {} |
||||
|
||||
// UnsafeTransactionServiceServer may be embedded to opt out of forward compatibility for this service.
|
||||
// Use of this interface is not recommended, as added methods to TransactionServiceServer will
|
||||
// result in compilation errors.
|
||||
type UnsafeTransactionServiceServer interface { |
||||
mustEmbedUnimplementedTransactionServiceServer() |
||||
} |
||||
|
||||
func RegisterTransactionServiceServer(s grpc.ServiceRegistrar, srv TransactionServiceServer) { |
||||
s.RegisterService(&TransactionService_ServiceDesc, srv) |
||||
} |
||||
|
||||
func _TransactionService_CreateUser_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) { |
||||
in := new(CreateUserRequest) |
||||
if err := dec(in); err != nil { |
||||
return nil, err |
||||
} |
||||
if interceptor == nil { |
||||
return srv.(TransactionServiceServer).CreateUser(ctx, in) |
||||
} |
||||
info := &grpc.UnaryServerInfo{ |
||||
Server: srv, |
||||
FullMethod: "/blog.api.v1.TransactionService/CreateUser", |
||||
} |
||||
handler := func(ctx context.Context, req interface{}) (interface{}, error) { |
||||
return srv.(TransactionServiceServer).CreateUser(ctx, req.(*CreateUserRequest)) |
||||
} |
||||
return interceptor(ctx, in, info, handler) |
||||
} |
||||
|
||||
// TransactionService_ServiceDesc is the grpc.ServiceDesc for TransactionService service.
|
||||
// It's only intended for direct use with grpc.RegisterService,
|
||||
// and not to be introspected or modified (even as a copy)
|
||||
var TransactionService_ServiceDesc = grpc.ServiceDesc{ |
||||
ServiceName: "blog.api.v1.TransactionService", |
||||
HandlerType: (*TransactionServiceServer)(nil), |
||||
Methods: []grpc.MethodDesc{ |
||||
{ |
||||
MethodName: "CreateUser", |
||||
Handler: _TransactionService_CreateUser_Handler, |
||||
}, |
||||
}, |
||||
Streams: []grpc.StreamDesc{}, |
||||
Metadata: "api/transaction/v1/transaction.proto", |
||||
} |
@ -0,0 +1,71 @@ |
||||
// Code generated by protoc-gen-go-http. DO NOT EDIT.
|
||||
// versions:
|
||||
// protoc-gen-go-http v2.1.4
|
||||
|
||||
package v1 |
||||
|
||||
import ( |
||||
context "context" |
||||
http "github.com/go-kratos/kratos/v2/transport/http" |
||||
binding "github.com/go-kratos/kratos/v2/transport/http/binding" |
||||
) |
||||
|
||||
// This is a compile-time assertion to ensure that this generated file
|
||||
// is compatible with the kratos package it is being compiled against.
|
||||
var _ = new(context.Context) |
||||
var _ = binding.EncodeURL |
||||
|
||||
const _ = http.SupportPackageIsVersion1 |
||||
|
||||
type TransactionServiceHTTPServer interface { |
||||
CreateUser(context.Context, *CreateUserRequest) (*CreateUserReply, error) |
||||
} |
||||
|
||||
func RegisterTransactionServiceHTTPServer(s *http.Server, srv TransactionServiceHTTPServer) { |
||||
r := s.Route("/") |
||||
r.POST("/v1/transaction", _TransactionService_CreateUser0_HTTP_Handler(srv)) |
||||
} |
||||
|
||||
func _TransactionService_CreateUser0_HTTP_Handler(srv TransactionServiceHTTPServer) func(ctx http.Context) error { |
||||
return func(ctx http.Context) error { |
||||
var in CreateUserRequest |
||||
if err := ctx.Bind(&in); err != nil { |
||||
return err |
||||
} |
||||
http.SetOperation(ctx, "/blog.api.v1.TransactionService/CreateUser") |
||||
h := ctx.Middleware(func(ctx context.Context, req interface{}) (interface{}, error) { |
||||
return srv.CreateUser(ctx, req.(*CreateUserRequest)) |
||||
}) |
||||
out, err := h(ctx, &in) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
reply := out.(*CreateUserReply) |
||||
return ctx.Result(200, reply) |
||||
} |
||||
} |
||||
|
||||
type TransactionServiceHTTPClient interface { |
||||
CreateUser(ctx context.Context, req *CreateUserRequest, opts ...http.CallOption) (rsp *CreateUserReply, err error) |
||||
} |
||||
|
||||
type TransactionServiceHTTPClientImpl struct { |
||||
cc *http.Client |
||||
} |
||||
|
||||
func NewTransactionServiceHTTPClient(client *http.Client) TransactionServiceHTTPClient { |
||||
return &TransactionServiceHTTPClientImpl{client} |
||||
} |
||||
|
||||
func (c *TransactionServiceHTTPClientImpl) CreateUser(ctx context.Context, in *CreateUserRequest, opts ...http.CallOption) (*CreateUserReply, error) { |
||||
var out CreateUserReply |
||||
pattern := "/v1/transaction" |
||||
path := binding.EncodeURL(pattern, in, false) |
||||
opts = append(opts, http.Operation("/blog.api.v1.TransactionService/CreateUser")) |
||||
opts = append(opts, http.PathTemplate(pattern)) |
||||
err := c.cc.Invoke(ctx, "POST", path, in, &out, opts...) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
return &out, err |
||||
} |
@ -0,0 +1,79 @@ |
||||
GOPATH:=$(shell go env GOPATH)
|
||||
VERSION=$(shell git describe --tags --always)
|
||||
INTERNAL_PROTO_FILES=$(shell find internal -name *.proto)
|
||||
API_PROTO_FILES=$(shell find api -name *.proto)
|
||||
|
||||
.PHONY: init |
||||
# init env
|
||||
init: |
||||
go install google.golang.org/protobuf/cmd/protoc-gen-go@latest
|
||||
go install google.golang.org/grpc/cmd/protoc-gen-go-grpc@latest
|
||||
go install github.com/go-kratos/kratos/cmd/kratos/v2@latest
|
||||
go install github.com/go-kratos/kratos/cmd/protoc-gen-go-http/v2@latest
|
||||
go install github.com/go-kratos/kratos/cmd/protoc-gen-go-errors/v2@latest
|
||||
go install github.com/google/gnostic/cmd/protoc-gen-openapi@v0.6.1
|
||||
|
||||
.PHONY: errors |
||||
# generate errors code
|
||||
errors: |
||||
protoc --proto_path=. \
|
||||
--proto_path=./third_party \
|
||||
--go_out=paths=source_relative:. \
|
||||
--go-errors_out=paths=source_relative:. \
|
||||
$(API_PROTO_FILES)
|
||||
|
||||
.PHONY: config |
||||
# generate internal proto
|
||||
config: |
||||
protoc --proto_path=. \
|
||||
--proto_path=./third_party \
|
||||
--go_out=paths=source_relative:. \
|
||||
$(INTERNAL_PROTO_FILES)
|
||||
|
||||
.PHONY: api |
||||
# generate api proto
|
||||
api: |
||||
protoc --proto_path=. \
|
||||
--proto_path=./third_party \
|
||||
--go_out=paths=source_relative:. \
|
||||
--go-http_out=paths=source_relative:. \
|
||||
--go-grpc_out=paths=source_relative:. \
|
||||
--openapi_out==paths=source_relative:. \
|
||||
$(API_PROTO_FILES)
|
||||
|
||||
.PHONY: build |
||||
# build
|
||||
build: |
||||
mkdir -p bin/ && go build -ldflags "-X main.Version=$(VERSION)" -o ./bin/ ./...
|
||||
|
||||
.PHONY: generate |
||||
# generate
|
||||
generate: |
||||
go generate ./...
|
||||
|
||||
.PHONY: all |
||||
# generate all
|
||||
all: |
||||
make api;
|
||||
make errors;
|
||||
make config;
|
||||
make generate;
|
||||
|
||||
# show help
|
||||
help: |
||||
@echo ''
|
||||
@echo 'Usage:'
|
||||
@echo ' make [target]'
|
||||
@echo ''
|
||||
@echo 'Targets:'
|
||||
@awk '/^[a-zA-Z\-\_0-9]+:/ { \
|
||||
helpMessage = match(lastLine, /^# (.*)/); \
|
||||
if (helpMessage) { \
|
||||
helpCommand = substr($$1, 0, index($$1, ":")-1); \
|
||||
helpMessage = substr(lastLine, RSTART + 2, RLENGTH); \
|
||||
printf "\033[36m%-22s\033[0m %s\n", helpCommand,helpMessage; \
|
||||
} \
|
||||
} \
|
||||
{ lastLine = $$0 }' $(MAKEFILE_LIST)
|
||||
|
||||
.DEFAULT_GOAL := help
|
@ -0,0 +1,19 @@ |
||||
# How to run this blog example server |
||||
1. You should ensure that your mysql server is running. |
||||
2. Ensure that the database named `testdb` has been created, |
||||
otherwise you should execute the following database script: |
||||
```mysql |
||||
create database testdb; |
||||
``` |
||||
3. Modify the `configs/config.yaml` file and add your mysql information in the data source: |
||||
```yaml |
||||
data: |
||||
database: |
||||
driver: mysql |
||||
source: root:password@tcp(127.0.0.1:3306)/testdb?parseTime=True |
||||
``` |
||||
4. Run your blog server: |
||||
```bash |
||||
$ go generate ./... |
||||
$ kratos run |
||||
``` |
@ -0,0 +1,103 @@ |
||||
package main |
||||
|
||||
import ( |
||||
"flag" |
||||
"os" |
||||
|
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/conf" |
||||
"github.com/go-kratos/kratos/v2" |
||||
"github.com/go-kratos/kratos/v2/config" |
||||
"github.com/go-kratos/kratos/v2/config/file" |
||||
"github.com/go-kratos/kratos/v2/log" |
||||
"github.com/go-kratos/kratos/v2/transport/grpc" |
||||
"github.com/go-kratos/kratos/v2/transport/http" |
||||
"go.opentelemetry.io/otel" |
||||
"go.opentelemetry.io/otel/attribute" |
||||
"go.opentelemetry.io/otel/exporters/jaeger" |
||||
"go.opentelemetry.io/otel/sdk/resource" |
||||
tracesdk "go.opentelemetry.io/otel/sdk/trace" |
||||
semconv "go.opentelemetry.io/otel/semconv/v1.4.0" |
||||
) |
||||
|
||||
// go build -ldflags "-X main.Version=x.y.z"
|
||||
var ( |
||||
// Name is the name of the compiled software.
|
||||
Name string |
||||
// Version is the version of the compiled software.
|
||||
Version string |
||||
// flagconf is the config flag.
|
||||
flagconf string |
||||
) |
||||
|
||||
func init() { |
||||
flag.StringVar(&flagconf, "conf", "../../configs", "config path, eg: -conf config.yaml") |
||||
} |
||||
|
||||
func newApp(logger log.Logger, hs *http.Server, gs *grpc.Server) *kratos.App { |
||||
return kratos.New( |
||||
kratos.Name(Name), |
||||
kratos.Version(Version), |
||||
kratos.Metadata(map[string]string{}), |
||||
kratos.Logger(logger), |
||||
kratos.Server( |
||||
hs, |
||||
gs, |
||||
), |
||||
) |
||||
} |
||||
|
||||
// Set global trace provider
|
||||
func setTracerProvider(url string) error { |
||||
// Create the Jaeger exporter
|
||||
exp, err := jaeger.New(jaeger.WithCollectorEndpoint(jaeger.WithEndpoint(url))) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
tp := tracesdk.NewTracerProvider( |
||||
// Set the sampling rate based on the parent span to 100%
|
||||
tracesdk.WithSampler(tracesdk.ParentBased(tracesdk.TraceIDRatioBased(1.0))), |
||||
// Always be sure to batch in production.
|
||||
tracesdk.WithBatcher(exp), |
||||
// Record information about this application in an Resource.
|
||||
tracesdk.WithResource(resource.NewSchemaless( |
||||
semconv.ServiceNameKey.String(Name), |
||||
attribute.String("env", "dev"), |
||||
)), |
||||
) |
||||
otel.SetTracerProvider(tp) |
||||
return nil |
||||
} |
||||
|
||||
func main() { |
||||
flag.Parse() |
||||
logger := log.NewStdLogger(os.Stdout) |
||||
|
||||
cfg := config.New( |
||||
config.WithSource( |
||||
file.NewSource(flagconf), |
||||
), |
||||
) |
||||
if err := cfg.Load(); err != nil { |
||||
panic(err) |
||||
} |
||||
|
||||
var bc conf.Bootstrap |
||||
if err := cfg.Scan(&bc); err != nil { |
||||
panic(err) |
||||
} |
||||
|
||||
if err := setTracerProvider(bc.Trace.Endpoint); err != nil { |
||||
panic(err) |
||||
} |
||||
|
||||
app, cleanup, err := initApp(bc.Server, bc.Data, logger) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
defer cleanup() |
||||
|
||||
// start and wait for stop signal
|
||||
if err := app.Run(); err != nil { |
||||
panic(err) |
||||
} |
||||
} |
@ -0,0 +1,22 @@ |
||||
//go:build wireinject
|
||||
// +build wireinject
|
||||
|
||||
// The build tag makes sure the stub is not built in the final build.
|
||||
|
||||
package main |
||||
|
||||
import ( |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/biz" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/conf" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/server" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/service" |
||||
"github.com/go-kratos/kratos/v2" |
||||
"github.com/go-kratos/kratos/v2/log" |
||||
"github.com/google/wire" |
||||
) |
||||
|
||||
// initApp init kratos application.
|
||||
func initApp(*conf.Server, *conf.Data, log.Logger) (*kratos.App, func(), error) { |
||||
panic(wire.Build(server.ProviderSet, data.ProviderSet, biz.ProviderSet, service.ProviderSet, newApp)) |
||||
} |
@ -0,0 +1,37 @@ |
||||
// Code generated by Wire. DO NOT EDIT.
|
||||
|
||||
//go:generate go run github.com/google/wire/cmd/wire
|
||||
//+build !wireinject
|
||||
|
||||
package main |
||||
|
||||
import ( |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/biz" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/conf" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/server" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/service" |
||||
"github.com/go-kratos/kratos/v2" |
||||
"github.com/go-kratos/kratos/v2/log" |
||||
) |
||||
|
||||
// Injectors from wire.go:
|
||||
|
||||
// initApp init kratos application.
|
||||
func initApp(confServer *conf.Server, confData *conf.Data, logger log.Logger) (*kratos.App, func(), error) { |
||||
dataData, cleanup, err := data.NewData(confData, logger) |
||||
if err != nil { |
||||
return nil, nil, err |
||||
} |
||||
userRepo := data.NewUserRepo(dataData, logger) |
||||
cardRepo := data.NewCardRepo(dataData, logger) |
||||
transaction := data.NewTransaction(dataData) |
||||
userUsecase := biz.NewUserUsecase(userRepo, cardRepo, transaction, logger) |
||||
transactionService := service.NewTransactionService(userUsecase, logger) |
||||
httpServer := server.NewHTTPServer(confServer, logger, transactionService) |
||||
grpcServer := server.NewGRPCServer(confServer, logger, transactionService) |
||||
app := newApp(logger, httpServer, grpcServer) |
||||
return app, func() { |
||||
cleanup() |
||||
}, nil |
||||
} |
@ -0,0 +1,18 @@ |
||||
trace: |
||||
endpoint: http://127.0.0.1:14268/api/traces |
||||
server: |
||||
http: |
||||
addr: 0.0.0.0:8000 |
||||
timeout: 1s |
||||
grpc: |
||||
addr: 0.0.0.0:9000 |
||||
timeout: 1s |
||||
data: |
||||
database: |
||||
driver: mysql |
||||
source: root:123456@tcp(127.0.0.1:3306)/testdb?parseTime=True |
||||
redis: |
||||
addr: 127.0.0.1:6379 |
||||
dial_timeout: 1s |
||||
read_timeout: 0.4s |
||||
write_timeout: 0.6s |
@ -0,0 +1,3 @@ |
||||
package generate |
||||
|
||||
//go:generate kratos proto client .
|
@ -0,0 +1 @@ |
||||
# Biz |
@ -0,0 +1,14 @@ |
||||
package biz |
||||
|
||||
import ( |
||||
"context" |
||||
|
||||
"github.com/google/wire" |
||||
) |
||||
|
||||
// ProviderSet is biz providers.
|
||||
var ProviderSet = wire.NewSet(NewUserUsecase) |
||||
|
||||
type Transaction interface { |
||||
ExecTx(context.Context, func(ctx context.Context) error) error |
||||
} |
@ -0,0 +1,50 @@ |
||||
package biz |
||||
|
||||
import ( |
||||
"context" |
||||
|
||||
"github.com/go-kratos/kratos/v2/log" |
||||
) |
||||
|
||||
type User struct { |
||||
Name string |
||||
Email string |
||||
} |
||||
|
||||
type UserRepo interface { |
||||
CreateUser(ctx context.Context, a *User) (int, error) |
||||
} |
||||
|
||||
type CardRepo interface { |
||||
CreateCard(ctx context.Context, id int) (int, error) |
||||
} |
||||
|
||||
type UserUsecase struct { |
||||
userRepo UserRepo |
||||
cardRepo CardRepo |
||||
tm Transaction |
||||
} |
||||
|
||||
func NewUserUsecase(user UserRepo, card CardRepo, tm Transaction, logger log.Logger) *UserUsecase { |
||||
return &UserUsecase{userRepo: user, cardRepo: card, tm: tm} |
||||
} |
||||
|
||||
func (u *UserUsecase) CreateUser(ctx context.Context, m *User) (int, error) { |
||||
var ( |
||||
err error |
||||
id int |
||||
) |
||||
if e := u.tm.ExecTx(ctx, func(ctx context.Context) error { |
||||
id, err = u.userRepo.CreateUser(ctx, m) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
if _, e := u.cardRepo.CreateCard(ctx, id); err != nil { |
||||
return e |
||||
} |
||||
return nil |
||||
}); e != nil { |
||||
return 0, err |
||||
} |
||||
return id, nil |
||||
} |
@ -0,0 +1,763 @@ |
||||
// Code generated by protoc-gen-go. DO NOT EDIT.
|
||||
// versions:
|
||||
// protoc-gen-go v1.26.0
|
||||
// protoc v3.17.0
|
||||
// source: conf.proto
|
||||
|
||||
package conf |
||||
|
||||
import ( |
||||
protoreflect "google.golang.org/protobuf/reflect/protoreflect" |
||||
protoimpl "google.golang.org/protobuf/runtime/protoimpl" |
||||
durationpb "google.golang.org/protobuf/types/known/durationpb" |
||||
reflect "reflect" |
||||
sync "sync" |
||||
) |
||||
|
||||
const ( |
||||
// Verify that this generated code is sufficiently up-to-date.
|
||||
_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion) |
||||
// Verify that runtime/protoimpl is sufficiently up-to-date.
|
||||
_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20) |
||||
) |
||||
|
||||
type Bootstrap struct { |
||||
state protoimpl.MessageState |
||||
sizeCache protoimpl.SizeCache |
||||
unknownFields protoimpl.UnknownFields |
||||
|
||||
Trace *Trace `protobuf:"bytes,1,opt,name=trace,proto3" json:"trace,omitempty"` |
||||
Server *Server `protobuf:"bytes,2,opt,name=server,proto3" json:"server,omitempty"` |
||||
Data *Data `protobuf:"bytes,3,opt,name=data,proto3" json:"data,omitempty"` |
||||
} |
||||
|
||||
func (x *Bootstrap) Reset() { |
||||
*x = Bootstrap{} |
||||
if protoimpl.UnsafeEnabled { |
||||
mi := &file_conf_proto_msgTypes[0] |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
} |
||||
|
||||
func (x *Bootstrap) String() string { |
||||
return protoimpl.X.MessageStringOf(x) |
||||
} |
||||
|
||||
func (*Bootstrap) ProtoMessage() {} |
||||
|
||||
func (x *Bootstrap) ProtoReflect() protoreflect.Message { |
||||
mi := &file_conf_proto_msgTypes[0] |
||||
if protoimpl.UnsafeEnabled && x != nil { |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
if ms.LoadMessageInfo() == nil { |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
return ms |
||||
} |
||||
return mi.MessageOf(x) |
||||
} |
||||
|
||||
// Deprecated: Use Bootstrap.ProtoReflect.Descriptor instead.
|
||||
func (*Bootstrap) Descriptor() ([]byte, []int) { |
||||
return file_conf_proto_rawDescGZIP(), []int{0} |
||||
} |
||||
|
||||
func (x *Bootstrap) GetTrace() *Trace { |
||||
if x != nil { |
||||
return x.Trace |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
func (x *Bootstrap) GetServer() *Server { |
||||
if x != nil { |
||||
return x.Server |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
func (x *Bootstrap) GetData() *Data { |
||||
if x != nil { |
||||
return x.Data |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
type Server struct { |
||||
state protoimpl.MessageState |
||||
sizeCache protoimpl.SizeCache |
||||
unknownFields protoimpl.UnknownFields |
||||
|
||||
Http *Server_HTTP `protobuf:"bytes,1,opt,name=http,proto3" json:"http,omitempty"` |
||||
Grpc *Server_GRPC `protobuf:"bytes,2,opt,name=grpc,proto3" json:"grpc,omitempty"` |
||||
} |
||||
|
||||
func (x *Server) Reset() { |
||||
*x = Server{} |
||||
if protoimpl.UnsafeEnabled { |
||||
mi := &file_conf_proto_msgTypes[1] |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
} |
||||
|
||||
func (x *Server) String() string { |
||||
return protoimpl.X.MessageStringOf(x) |
||||
} |
||||
|
||||
func (*Server) ProtoMessage() {} |
||||
|
||||
func (x *Server) ProtoReflect() protoreflect.Message { |
||||
mi := &file_conf_proto_msgTypes[1] |
||||
if protoimpl.UnsafeEnabled && x != nil { |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
if ms.LoadMessageInfo() == nil { |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
return ms |
||||
} |
||||
return mi.MessageOf(x) |
||||
} |
||||
|
||||
// Deprecated: Use Server.ProtoReflect.Descriptor instead.
|
||||
func (*Server) Descriptor() ([]byte, []int) { |
||||
return file_conf_proto_rawDescGZIP(), []int{1} |
||||
} |
||||
|
||||
func (x *Server) GetHttp() *Server_HTTP { |
||||
if x != nil { |
||||
return x.Http |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
func (x *Server) GetGrpc() *Server_GRPC { |
||||
if x != nil { |
||||
return x.Grpc |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
type Data struct { |
||||
state protoimpl.MessageState |
||||
sizeCache protoimpl.SizeCache |
||||
unknownFields protoimpl.UnknownFields |
||||
|
||||
Database *Data_Database `protobuf:"bytes,1,opt,name=database,proto3" json:"database,omitempty"` |
||||
Redis *Data_Redis `protobuf:"bytes,2,opt,name=redis,proto3" json:"redis,omitempty"` |
||||
} |
||||
|
||||
func (x *Data) Reset() { |
||||
*x = Data{} |
||||
if protoimpl.UnsafeEnabled { |
||||
mi := &file_conf_proto_msgTypes[2] |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
} |
||||
|
||||
func (x *Data) String() string { |
||||
return protoimpl.X.MessageStringOf(x) |
||||
} |
||||
|
||||
func (*Data) ProtoMessage() {} |
||||
|
||||
func (x *Data) ProtoReflect() protoreflect.Message { |
||||
mi := &file_conf_proto_msgTypes[2] |
||||
if protoimpl.UnsafeEnabled && x != nil { |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
if ms.LoadMessageInfo() == nil { |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
return ms |
||||
} |
||||
return mi.MessageOf(x) |
||||
} |
||||
|
||||
// Deprecated: Use Data.ProtoReflect.Descriptor instead.
|
||||
func (*Data) Descriptor() ([]byte, []int) { |
||||
return file_conf_proto_rawDescGZIP(), []int{2} |
||||
} |
||||
|
||||
func (x *Data) GetDatabase() *Data_Database { |
||||
if x != nil { |
||||
return x.Database |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
func (x *Data) GetRedis() *Data_Redis { |
||||
if x != nil { |
||||
return x.Redis |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
type Trace struct { |
||||
state protoimpl.MessageState |
||||
sizeCache protoimpl.SizeCache |
||||
unknownFields protoimpl.UnknownFields |
||||
|
||||
Endpoint string `protobuf:"bytes,1,opt,name=endpoint,proto3" json:"endpoint,omitempty"` |
||||
} |
||||
|
||||
func (x *Trace) Reset() { |
||||
*x = Trace{} |
||||
if protoimpl.UnsafeEnabled { |
||||
mi := &file_conf_proto_msgTypes[3] |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
} |
||||
|
||||
func (x *Trace) String() string { |
||||
return protoimpl.X.MessageStringOf(x) |
||||
} |
||||
|
||||
func (*Trace) ProtoMessage() {} |
||||
|
||||
func (x *Trace) ProtoReflect() protoreflect.Message { |
||||
mi := &file_conf_proto_msgTypes[3] |
||||
if protoimpl.UnsafeEnabled && x != nil { |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
if ms.LoadMessageInfo() == nil { |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
return ms |
||||
} |
||||
return mi.MessageOf(x) |
||||
} |
||||
|
||||
// Deprecated: Use Trace.ProtoReflect.Descriptor instead.
|
||||
func (*Trace) Descriptor() ([]byte, []int) { |
||||
return file_conf_proto_rawDescGZIP(), []int{3} |
||||
} |
||||
|
||||
func (x *Trace) GetEndpoint() string { |
||||
if x != nil { |
||||
return x.Endpoint |
||||
} |
||||
return "" |
||||
} |
||||
|
||||
type Server_HTTP struct { |
||||
state protoimpl.MessageState |
||||
sizeCache protoimpl.SizeCache |
||||
unknownFields protoimpl.UnknownFields |
||||
|
||||
Network string `protobuf:"bytes,1,opt,name=network,proto3" json:"network,omitempty"` |
||||
Addr string `protobuf:"bytes,2,opt,name=addr,proto3" json:"addr,omitempty"` |
||||
Timeout *durationpb.Duration `protobuf:"bytes,3,opt,name=timeout,proto3" json:"timeout,omitempty"` |
||||
} |
||||
|
||||
func (x *Server_HTTP) Reset() { |
||||
*x = Server_HTTP{} |
||||
if protoimpl.UnsafeEnabled { |
||||
mi := &file_conf_proto_msgTypes[4] |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
} |
||||
|
||||
func (x *Server_HTTP) String() string { |
||||
return protoimpl.X.MessageStringOf(x) |
||||
} |
||||
|
||||
func (*Server_HTTP) ProtoMessage() {} |
||||
|
||||
func (x *Server_HTTP) ProtoReflect() protoreflect.Message { |
||||
mi := &file_conf_proto_msgTypes[4] |
||||
if protoimpl.UnsafeEnabled && x != nil { |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
if ms.LoadMessageInfo() == nil { |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
return ms |
||||
} |
||||
return mi.MessageOf(x) |
||||
} |
||||
|
||||
// Deprecated: Use Server_HTTP.ProtoReflect.Descriptor instead.
|
||||
func (*Server_HTTP) Descriptor() ([]byte, []int) { |
||||
return file_conf_proto_rawDescGZIP(), []int{1, 0} |
||||
} |
||||
|
||||
func (x *Server_HTTP) GetNetwork() string { |
||||
if x != nil { |
||||
return x.Network |
||||
} |
||||
return "" |
||||
} |
||||
|
||||
func (x *Server_HTTP) GetAddr() string { |
||||
if x != nil { |
||||
return x.Addr |
||||
} |
||||
return "" |
||||
} |
||||
|
||||
func (x *Server_HTTP) GetTimeout() *durationpb.Duration { |
||||
if x != nil { |
||||
return x.Timeout |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
type Server_GRPC struct { |
||||
state protoimpl.MessageState |
||||
sizeCache protoimpl.SizeCache |
||||
unknownFields protoimpl.UnknownFields |
||||
|
||||
Network string `protobuf:"bytes,1,opt,name=network,proto3" json:"network,omitempty"` |
||||
Addr string `protobuf:"bytes,2,opt,name=addr,proto3" json:"addr,omitempty"` |
||||
Timeout *durationpb.Duration `protobuf:"bytes,3,opt,name=timeout,proto3" json:"timeout,omitempty"` |
||||
} |
||||
|
||||
func (x *Server_GRPC) Reset() { |
||||
*x = Server_GRPC{} |
||||
if protoimpl.UnsafeEnabled { |
||||
mi := &file_conf_proto_msgTypes[5] |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
} |
||||
|
||||
func (x *Server_GRPC) String() string { |
||||
return protoimpl.X.MessageStringOf(x) |
||||
} |
||||
|
||||
func (*Server_GRPC) ProtoMessage() {} |
||||
|
||||
func (x *Server_GRPC) ProtoReflect() protoreflect.Message { |
||||
mi := &file_conf_proto_msgTypes[5] |
||||
if protoimpl.UnsafeEnabled && x != nil { |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
if ms.LoadMessageInfo() == nil { |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
return ms |
||||
} |
||||
return mi.MessageOf(x) |
||||
} |
||||
|
||||
// Deprecated: Use Server_GRPC.ProtoReflect.Descriptor instead.
|
||||
func (*Server_GRPC) Descriptor() ([]byte, []int) { |
||||
return file_conf_proto_rawDescGZIP(), []int{1, 1} |
||||
} |
||||
|
||||
func (x *Server_GRPC) GetNetwork() string { |
||||
if x != nil { |
||||
return x.Network |
||||
} |
||||
return "" |
||||
} |
||||
|
||||
func (x *Server_GRPC) GetAddr() string { |
||||
if x != nil { |
||||
return x.Addr |
||||
} |
||||
return "" |
||||
} |
||||
|
||||
func (x *Server_GRPC) GetTimeout() *durationpb.Duration { |
||||
if x != nil { |
||||
return x.Timeout |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
type Data_Database struct { |
||||
state protoimpl.MessageState |
||||
sizeCache protoimpl.SizeCache |
||||
unknownFields protoimpl.UnknownFields |
||||
|
||||
Driver string `protobuf:"bytes,1,opt,name=driver,proto3" json:"driver,omitempty"` |
||||
Source string `protobuf:"bytes,2,opt,name=source,proto3" json:"source,omitempty"` |
||||
} |
||||
|
||||
func (x *Data_Database) Reset() { |
||||
*x = Data_Database{} |
||||
if protoimpl.UnsafeEnabled { |
||||
mi := &file_conf_proto_msgTypes[6] |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
} |
||||
|
||||
func (x *Data_Database) String() string { |
||||
return protoimpl.X.MessageStringOf(x) |
||||
} |
||||
|
||||
func (*Data_Database) ProtoMessage() {} |
||||
|
||||
func (x *Data_Database) ProtoReflect() protoreflect.Message { |
||||
mi := &file_conf_proto_msgTypes[6] |
||||
if protoimpl.UnsafeEnabled && x != nil { |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
if ms.LoadMessageInfo() == nil { |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
return ms |
||||
} |
||||
return mi.MessageOf(x) |
||||
} |
||||
|
||||
// Deprecated: Use Data_Database.ProtoReflect.Descriptor instead.
|
||||
func (*Data_Database) Descriptor() ([]byte, []int) { |
||||
return file_conf_proto_rawDescGZIP(), []int{2, 0} |
||||
} |
||||
|
||||
func (x *Data_Database) GetDriver() string { |
||||
if x != nil { |
||||
return x.Driver |
||||
} |
||||
return "" |
||||
} |
||||
|
||||
func (x *Data_Database) GetSource() string { |
||||
if x != nil { |
||||
return x.Source |
||||
} |
||||
return "" |
||||
} |
||||
|
||||
type Data_Redis struct { |
||||
state protoimpl.MessageState |
||||
sizeCache protoimpl.SizeCache |
||||
unknownFields protoimpl.UnknownFields |
||||
|
||||
Network string `protobuf:"bytes,1,opt,name=network,proto3" json:"network,omitempty"` |
||||
Addr string `protobuf:"bytes,2,opt,name=addr,proto3" json:"addr,omitempty"` |
||||
Password string `protobuf:"bytes,3,opt,name=password,proto3" json:"password,omitempty"` |
||||
Db int32 `protobuf:"varint,4,opt,name=db,proto3" json:"db,omitempty"` |
||||
DialTimeout *durationpb.Duration `protobuf:"bytes,5,opt,name=dial_timeout,json=dialTimeout,proto3" json:"dial_timeout,omitempty"` |
||||
ReadTimeout *durationpb.Duration `protobuf:"bytes,6,opt,name=read_timeout,json=readTimeout,proto3" json:"read_timeout,omitempty"` |
||||
WriteTimeout *durationpb.Duration `protobuf:"bytes,7,opt,name=write_timeout,json=writeTimeout,proto3" json:"write_timeout,omitempty"` |
||||
} |
||||
|
||||
func (x *Data_Redis) Reset() { |
||||
*x = Data_Redis{} |
||||
if protoimpl.UnsafeEnabled { |
||||
mi := &file_conf_proto_msgTypes[7] |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
} |
||||
|
||||
func (x *Data_Redis) String() string { |
||||
return protoimpl.X.MessageStringOf(x) |
||||
} |
||||
|
||||
func (*Data_Redis) ProtoMessage() {} |
||||
|
||||
func (x *Data_Redis) ProtoReflect() protoreflect.Message { |
||||
mi := &file_conf_proto_msgTypes[7] |
||||
if protoimpl.UnsafeEnabled && x != nil { |
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) |
||||
if ms.LoadMessageInfo() == nil { |
||||
ms.StoreMessageInfo(mi) |
||||
} |
||||
return ms |
||||
} |
||||
return mi.MessageOf(x) |
||||
} |
||||
|
||||
// Deprecated: Use Data_Redis.ProtoReflect.Descriptor instead.
|
||||
func (*Data_Redis) Descriptor() ([]byte, []int) { |
||||
return file_conf_proto_rawDescGZIP(), []int{2, 1} |
||||
} |
||||
|
||||
func (x *Data_Redis) GetNetwork() string { |
||||
if x != nil { |
||||
return x.Network |
||||
} |
||||
return "" |
||||
} |
||||
|
||||
func (x *Data_Redis) GetAddr() string { |
||||
if x != nil { |
||||
return x.Addr |
||||
} |
||||
return "" |
||||
} |
||||
|
||||
func (x *Data_Redis) GetPassword() string { |
||||
if x != nil { |
||||
return x.Password |
||||
} |
||||
return "" |
||||
} |
||||
|
||||
func (x *Data_Redis) GetDb() int32 { |
||||
if x != nil { |
||||
return x.Db |
||||
} |
||||
return 0 |
||||
} |
||||
|
||||
func (x *Data_Redis) GetDialTimeout() *durationpb.Duration { |
||||
if x != nil { |
||||
return x.DialTimeout |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
func (x *Data_Redis) GetReadTimeout() *durationpb.Duration { |
||||
if x != nil { |
||||
return x.ReadTimeout |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
func (x *Data_Redis) GetWriteTimeout() *durationpb.Duration { |
||||
if x != nil { |
||||
return x.WriteTimeout |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
var File_conf_proto protoreflect.FileDescriptor |
||||
|
||||
var file_conf_proto_rawDesc = []byte{ |
||||
0x0a, 0x0a, 0x63, 0x6f, 0x6e, 0x66, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x14, 0x6b, 0x72, |
||||
0x61, 0x74, 0x6f, 0x73, 0x2e, 0x69, 0x6e, 0x74, 0x65, 0x72, 0x6e, 0x61, 0x6c, 0x2e, 0x63, 0x6f, |
||||
0x6e, 0x66, 0x1a, 0x1e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2f, 0x70, 0x72, 0x6f, 0x74, 0x6f, |
||||
0x62, 0x75, 0x66, 0x2f, 0x64, 0x75, 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x2e, 0x70, 0x72, 0x6f, |
||||
0x74, 0x6f, 0x22, 0xa4, 0x01, 0x0a, 0x09, 0x42, 0x6f, 0x6f, 0x74, 0x73, 0x74, 0x72, 0x61, 0x70, |
||||
0x12, 0x31, 0x0a, 0x05, 0x74, 0x72, 0x61, 0x63, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0b, 0x32, |
||||
0x1b, 0x2e, 0x6b, 0x72, 0x61, 0x74, 0x6f, 0x73, 0x2e, 0x69, 0x6e, 0x74, 0x65, 0x72, 0x6e, 0x61, |
||||
0x6c, 0x2e, 0x63, 0x6f, 0x6e, 0x66, 0x2e, 0x54, 0x72, 0x61, 0x63, 0x65, 0x52, 0x05, 0x74, 0x72, |
||||
0x61, 0x63, 0x65, 0x12, 0x34, 0x0a, 0x06, 0x73, 0x65, 0x72, 0x76, 0x65, 0x72, 0x18, 0x02, 0x20, |
||||
0x01, 0x28, 0x0b, 0x32, 0x1c, 0x2e, 0x6b, 0x72, 0x61, 0x74, 0x6f, 0x73, 0x2e, 0x69, 0x6e, 0x74, |
||||
0x65, 0x72, 0x6e, 0x61, 0x6c, 0x2e, 0x63, 0x6f, 0x6e, 0x66, 0x2e, 0x53, 0x65, 0x72, 0x76, 0x65, |
||||
0x72, 0x52, 0x06, 0x73, 0x65, 0x72, 0x76, 0x65, 0x72, 0x12, 0x2e, 0x0a, 0x04, 0x64, 0x61, 0x74, |
||||
0x61, 0x18, 0x03, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x1a, 0x2e, 0x6b, 0x72, 0x61, 0x74, 0x6f, 0x73, |
||||
0x2e, 0x69, 0x6e, 0x74, 0x65, 0x72, 0x6e, 0x61, 0x6c, 0x2e, 0x63, 0x6f, 0x6e, 0x66, 0x2e, 0x44, |
||||
0x61, 0x74, 0x61, 0x52, 0x04, 0x64, 0x61, 0x74, 0x61, 0x22, 0xcc, 0x02, 0x0a, 0x06, 0x53, 0x65, |
||||
0x72, 0x76, 0x65, 0x72, 0x12, 0x35, 0x0a, 0x04, 0x68, 0x74, 0x74, 0x70, 0x18, 0x01, 0x20, 0x01, |
||||
0x28, 0x0b, 0x32, 0x21, 0x2e, 0x6b, 0x72, 0x61, 0x74, 0x6f, 0x73, 0x2e, 0x69, 0x6e, 0x74, 0x65, |
||||
0x72, 0x6e, 0x61, 0x6c, 0x2e, 0x63, 0x6f, 0x6e, 0x66, 0x2e, 0x53, 0x65, 0x72, 0x76, 0x65, 0x72, |
||||
0x2e, 0x48, 0x54, 0x54, 0x50, 0x52, 0x04, 0x68, 0x74, 0x74, 0x70, 0x12, 0x35, 0x0a, 0x04, 0x67, |
||||
0x72, 0x70, 0x63, 0x18, 0x02, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x21, 0x2e, 0x6b, 0x72, 0x61, 0x74, |
||||
0x6f, 0x73, 0x2e, 0x69, 0x6e, 0x74, 0x65, 0x72, 0x6e, 0x61, 0x6c, 0x2e, 0x63, 0x6f, 0x6e, 0x66, |
||||
0x2e, 0x53, 0x65, 0x72, 0x76, 0x65, 0x72, 0x2e, 0x47, 0x52, 0x50, 0x43, 0x52, 0x04, 0x67, 0x72, |
||||
0x70, 0x63, 0x1a, 0x69, 0x0a, 0x04, 0x48, 0x54, 0x54, 0x50, 0x12, 0x18, 0x0a, 0x07, 0x6e, 0x65, |
||||
0x74, 0x77, 0x6f, 0x72, 0x6b, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x07, 0x6e, 0x65, 0x74, |
||||
0x77, 0x6f, 0x72, 0x6b, 0x12, 0x12, 0x0a, 0x04, 0x61, 0x64, 0x64, 0x72, 0x18, 0x02, 0x20, 0x01, |
||||
0x28, 0x09, 0x52, 0x04, 0x61, 0x64, 0x64, 0x72, 0x12, 0x33, 0x0a, 0x07, 0x74, 0x69, 0x6d, 0x65, |
||||
0x6f, 0x75, 0x74, 0x18, 0x03, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x19, 0x2e, 0x67, 0x6f, 0x6f, 0x67, |
||||
0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x44, 0x75, 0x72, 0x61, |
||||
0x74, 0x69, 0x6f, 0x6e, 0x52, 0x07, 0x74, 0x69, 0x6d, 0x65, 0x6f, 0x75, 0x74, 0x1a, 0x69, 0x0a, |
||||
0x04, 0x47, 0x52, 0x50, 0x43, 0x12, 0x18, 0x0a, 0x07, 0x6e, 0x65, 0x74, 0x77, 0x6f, 0x72, 0x6b, |
||||
0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x07, 0x6e, 0x65, 0x74, 0x77, 0x6f, 0x72, 0x6b, 0x12, |
||||
0x12, 0x0a, 0x04, 0x61, 0x64, 0x64, 0x72, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x04, 0x61, |
||||
0x64, 0x64, 0x72, 0x12, 0x33, 0x0a, 0x07, 0x74, 0x69, 0x6d, 0x65, 0x6f, 0x75, 0x74, 0x18, 0x03, |
||||
0x20, 0x01, 0x28, 0x0b, 0x32, 0x19, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, |
||||
0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x44, 0x75, 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x52, |
||||
0x07, 0x74, 0x69, 0x6d, 0x65, 0x6f, 0x75, 0x74, 0x22, 0xdb, 0x03, 0x0a, 0x04, 0x44, 0x61, 0x74, |
||||
0x61, 0x12, 0x3f, 0x0a, 0x08, 0x64, 0x61, 0x74, 0x61, 0x62, 0x61, 0x73, 0x65, 0x18, 0x01, 0x20, |
||||
0x01, 0x28, 0x0b, 0x32, 0x23, 0x2e, 0x6b, 0x72, 0x61, 0x74, 0x6f, 0x73, 0x2e, 0x69, 0x6e, 0x74, |
||||
0x65, 0x72, 0x6e, 0x61, 0x6c, 0x2e, 0x63, 0x6f, 0x6e, 0x66, 0x2e, 0x44, 0x61, 0x74, 0x61, 0x2e, |
||||
0x44, 0x61, 0x74, 0x61, 0x62, 0x61, 0x73, 0x65, 0x52, 0x08, 0x64, 0x61, 0x74, 0x61, 0x62, 0x61, |
||||
0x73, 0x65, 0x12, 0x36, 0x0a, 0x05, 0x72, 0x65, 0x64, 0x69, 0x73, 0x18, 0x02, 0x20, 0x01, 0x28, |
||||
0x0b, 0x32, 0x20, 0x2e, 0x6b, 0x72, 0x61, 0x74, 0x6f, 0x73, 0x2e, 0x69, 0x6e, 0x74, 0x65, 0x72, |
||||
0x6e, 0x61, 0x6c, 0x2e, 0x63, 0x6f, 0x6e, 0x66, 0x2e, 0x44, 0x61, 0x74, 0x61, 0x2e, 0x52, 0x65, |
||||
0x64, 0x69, 0x73, 0x52, 0x05, 0x72, 0x65, 0x64, 0x69, 0x73, 0x1a, 0x3a, 0x0a, 0x08, 0x44, 0x61, |
||||
0x74, 0x61, 0x62, 0x61, 0x73, 0x65, 0x12, 0x16, 0x0a, 0x06, 0x64, 0x72, 0x69, 0x76, 0x65, 0x72, |
||||
0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x06, 0x64, 0x72, 0x69, 0x76, 0x65, 0x72, 0x12, 0x16, |
||||
0x0a, 0x06, 0x73, 0x6f, 0x75, 0x72, 0x63, 0x65, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x06, |
||||
0x73, 0x6f, 0x75, 0x72, 0x63, 0x65, 0x1a, 0x9d, 0x02, 0x0a, 0x05, 0x52, 0x65, 0x64, 0x69, 0x73, |
||||
0x12, 0x18, 0x0a, 0x07, 0x6e, 0x65, 0x74, 0x77, 0x6f, 0x72, 0x6b, 0x18, 0x01, 0x20, 0x01, 0x28, |
||||
0x09, 0x52, 0x07, 0x6e, 0x65, 0x74, 0x77, 0x6f, 0x72, 0x6b, 0x12, 0x12, 0x0a, 0x04, 0x61, 0x64, |
||||
0x64, 0x72, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x04, 0x61, 0x64, 0x64, 0x72, 0x12, 0x1a, |
||||
0x0a, 0x08, 0x70, 0x61, 0x73, 0x73, 0x77, 0x6f, 0x72, 0x64, 0x18, 0x03, 0x20, 0x01, 0x28, 0x09, |
||||
0x52, 0x08, 0x70, 0x61, 0x73, 0x73, 0x77, 0x6f, 0x72, 0x64, 0x12, 0x0e, 0x0a, 0x02, 0x64, 0x62, |
||||
0x18, 0x04, 0x20, 0x01, 0x28, 0x05, 0x52, 0x02, 0x64, 0x62, 0x12, 0x3c, 0x0a, 0x0c, 0x64, 0x69, |
||||
0x61, 0x6c, 0x5f, 0x74, 0x69, 0x6d, 0x65, 0x6f, 0x75, 0x74, 0x18, 0x05, 0x20, 0x01, 0x28, 0x0b, |
||||
0x32, 0x19, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, |
||||
0x75, 0x66, 0x2e, 0x44, 0x75, 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x52, 0x0b, 0x64, 0x69, 0x61, |
||||
0x6c, 0x54, 0x69, 0x6d, 0x65, 0x6f, 0x75, 0x74, 0x12, 0x3c, 0x0a, 0x0c, 0x72, 0x65, 0x61, 0x64, |
||||
0x5f, 0x74, 0x69, 0x6d, 0x65, 0x6f, 0x75, 0x74, 0x18, 0x06, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x19, |
||||
0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, |
||||
0x2e, 0x44, 0x75, 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x52, 0x0b, 0x72, 0x65, 0x61, 0x64, 0x54, |
||||
0x69, 0x6d, 0x65, 0x6f, 0x75, 0x74, 0x12, 0x3e, 0x0a, 0x0d, 0x77, 0x72, 0x69, 0x74, 0x65, 0x5f, |
||||
0x74, 0x69, 0x6d, 0x65, 0x6f, 0x75, 0x74, 0x18, 0x07, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x19, 0x2e, |
||||
0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, |
||||
0x44, 0x75, 0x72, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x52, 0x0c, 0x77, 0x72, 0x69, 0x74, 0x65, 0x54, |
||||
0x69, 0x6d, 0x65, 0x6f, 0x75, 0x74, 0x22, 0x23, 0x0a, 0x05, 0x54, 0x72, 0x61, 0x63, 0x65, 0x12, |
||||
0x1a, 0x0a, 0x08, 0x65, 0x6e, 0x64, 0x70, 0x6f, 0x69, 0x6e, 0x74, 0x18, 0x01, 0x20, 0x01, 0x28, |
||||
0x09, 0x52, 0x08, 0x65, 0x6e, 0x64, 0x70, 0x6f, 0x69, 0x6e, 0x74, 0x42, 0x3e, 0x5a, 0x3c, 0x67, |
||||
0x69, 0x74, 0x68, 0x75, 0x62, 0x2e, 0x63, 0x6f, 0x6d, 0x2f, 0x67, 0x6f, 0x2d, 0x6b, 0x72, 0x61, |
||||
0x74, 0x6f, 0x73, 0x2f, 0x6b, 0x72, 0x61, 0x74, 0x6f, 0x73, 0x2f, 0x65, 0x78, 0x61, 0x6d, 0x70, |
||||
0x6c, 0x65, 0x73, 0x2f, 0x62, 0x6c, 0x6f, 0x67, 0x2f, 0x69, 0x6e, 0x74, 0x65, 0x72, 0x6e, 0x61, |
||||
0x6c, 0x2f, 0x63, 0x6f, 0x6e, 0x66, 0x3b, 0x63, 0x6f, 0x6e, 0x66, 0x62, 0x06, 0x70, 0x72, 0x6f, |
||||
0x74, 0x6f, 0x33, |
||||
} |
||||
|
||||
var ( |
||||
file_conf_proto_rawDescOnce sync.Once |
||||
file_conf_proto_rawDescData = file_conf_proto_rawDesc |
||||
) |
||||
|
||||
func file_conf_proto_rawDescGZIP() []byte { |
||||
file_conf_proto_rawDescOnce.Do(func() { |
||||
file_conf_proto_rawDescData = protoimpl.X.CompressGZIP(file_conf_proto_rawDescData) |
||||
}) |
||||
return file_conf_proto_rawDescData |
||||
} |
||||
|
||||
var file_conf_proto_msgTypes = make([]protoimpl.MessageInfo, 8) |
||||
var file_conf_proto_goTypes = []interface{}{ |
||||
(*Bootstrap)(nil), // 0: kratos.internal.conf.Bootstrap
|
||||
(*Server)(nil), // 1: kratos.internal.conf.Server
|
||||
(*Data)(nil), // 2: kratos.internal.conf.Data
|
||||
(*Trace)(nil), // 3: kratos.internal.conf.Trace
|
||||
(*Server_HTTP)(nil), // 4: kratos.internal.conf.Server.HTTP
|
||||
(*Server_GRPC)(nil), // 5: kratos.internal.conf.Server.GRPC
|
||||
(*Data_Database)(nil), // 6: kratos.internal.conf.Data.Database
|
||||
(*Data_Redis)(nil), // 7: kratos.internal.conf.Data.Redis
|
||||
(*durationpb.Duration)(nil), // 8: google.protobuf.Duration
|
||||
} |
||||
var file_conf_proto_depIdxs = []int32{ |
||||
3, // 0: kratos.internal.conf.Bootstrap.trace:type_name -> kratos.internal.conf.Trace
|
||||
1, // 1: kratos.internal.conf.Bootstrap.server:type_name -> kratos.internal.conf.Server
|
||||
2, // 2: kratos.internal.conf.Bootstrap.data:type_name -> kratos.internal.conf.Data
|
||||
4, // 3: kratos.internal.conf.Server.http:type_name -> kratos.internal.conf.Server.HTTP
|
||||
5, // 4: kratos.internal.conf.Server.grpc:type_name -> kratos.internal.conf.Server.GRPC
|
||||
6, // 5: kratos.internal.conf.Data.database:type_name -> kratos.internal.conf.Data.Database
|
||||
7, // 6: kratos.internal.conf.Data.redis:type_name -> kratos.internal.conf.Data.Redis
|
||||
8, // 7: kratos.internal.conf.Server.HTTP.timeout:type_name -> google.protobuf.Duration
|
||||
8, // 8: kratos.internal.conf.Server.GRPC.timeout:type_name -> google.protobuf.Duration
|
||||
8, // 9: kratos.internal.conf.Data.Redis.dial_timeout:type_name -> google.protobuf.Duration
|
||||
8, // 10: kratos.internal.conf.Data.Redis.read_timeout:type_name -> google.protobuf.Duration
|
||||
8, // 11: kratos.internal.conf.Data.Redis.write_timeout:type_name -> google.protobuf.Duration
|
||||
12, // [12:12] is the sub-list for method output_type
|
||||
12, // [12:12] is the sub-list for method input_type
|
||||
12, // [12:12] is the sub-list for extension type_name
|
||||
12, // [12:12] is the sub-list for extension extendee
|
||||
0, // [0:12] is the sub-list for field type_name
|
||||
} |
||||
|
||||
func init() { file_conf_proto_init() } |
||||
func file_conf_proto_init() { |
||||
if File_conf_proto != nil { |
||||
return |
||||
} |
||||
if !protoimpl.UnsafeEnabled { |
||||
file_conf_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} { |
||||
switch v := v.(*Bootstrap); i { |
||||
case 0: |
||||
return &v.state |
||||
case 1: |
||||
return &v.sizeCache |
||||
case 2: |
||||
return &v.unknownFields |
||||
default: |
||||
return nil |
||||
} |
||||
} |
||||
file_conf_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} { |
||||
switch v := v.(*Server); i { |
||||
case 0: |
||||
return &v.state |
||||
case 1: |
||||
return &v.sizeCache |
||||
case 2: |
||||
return &v.unknownFields |
||||
default: |
||||
return nil |
||||
} |
||||
} |
||||
file_conf_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} { |
||||
switch v := v.(*Data); i { |
||||
case 0: |
||||
return &v.state |
||||
case 1: |
||||
return &v.sizeCache |
||||
case 2: |
||||
return &v.unknownFields |
||||
default: |
||||
return nil |
||||
} |
||||
} |
||||
file_conf_proto_msgTypes[3].Exporter = func(v interface{}, i int) interface{} { |
||||
switch v := v.(*Trace); i { |
||||
case 0: |
||||
return &v.state |
||||
case 1: |
||||
return &v.sizeCache |
||||
case 2: |
||||
return &v.unknownFields |
||||
default: |
||||
return nil |
||||
} |
||||
} |
||||
file_conf_proto_msgTypes[4].Exporter = func(v interface{}, i int) interface{} { |
||||
switch v := v.(*Server_HTTP); i { |
||||
case 0: |
||||
return &v.state |
||||
case 1: |
||||
return &v.sizeCache |
||||
case 2: |
||||
return &v.unknownFields |
||||
default: |
||||
return nil |
||||
} |
||||
} |
||||
file_conf_proto_msgTypes[5].Exporter = func(v interface{}, i int) interface{} { |
||||
switch v := v.(*Server_GRPC); i { |
||||
case 0: |
||||
return &v.state |
||||
case 1: |
||||
return &v.sizeCache |
||||
case 2: |
||||
return &v.unknownFields |
||||
default: |
||||
return nil |
||||
} |
||||
} |
||||
file_conf_proto_msgTypes[6].Exporter = func(v interface{}, i int) interface{} { |
||||
switch v := v.(*Data_Database); i { |
||||
case 0: |
||||
return &v.state |
||||
case 1: |
||||
return &v.sizeCache |
||||
case 2: |
||||
return &v.unknownFields |
||||
default: |
||||
return nil |
||||
} |
||||
} |
||||
file_conf_proto_msgTypes[7].Exporter = func(v interface{}, i int) interface{} { |
||||
switch v := v.(*Data_Redis); i { |
||||
case 0: |
||||
return &v.state |
||||
case 1: |
||||
return &v.sizeCache |
||||
case 2: |
||||
return &v.unknownFields |
||||
default: |
||||
return nil |
||||
} |
||||
} |
||||
} |
||||
type x struct{} |
||||
out := protoimpl.TypeBuilder{ |
||||
File: protoimpl.DescBuilder{ |
||||
GoPackagePath: reflect.TypeOf(x{}).PkgPath(), |
||||
RawDescriptor: file_conf_proto_rawDesc, |
||||
NumEnums: 0, |
||||
NumMessages: 8, |
||||
NumExtensions: 0, |
||||
NumServices: 0, |
||||
}, |
||||
GoTypes: file_conf_proto_goTypes, |
||||
DependencyIndexes: file_conf_proto_depIdxs, |
||||
MessageInfos: file_conf_proto_msgTypes, |
||||
}.Build() |
||||
File_conf_proto = out.File |
||||
file_conf_proto_rawDesc = nil |
||||
file_conf_proto_goTypes = nil |
||||
file_conf_proto_depIdxs = nil |
||||
} |
@ -0,0 +1,49 @@ |
||||
syntax = "proto3"; |
||||
package kratos.internal.conf; |
||||
|
||||
option go_package = "github.com/go-kratos/kratos/examples/blog/internal/conf;conf"; |
||||
|
||||
import "google/protobuf/duration.proto"; |
||||
|
||||
message Bootstrap { |
||||
Trace trace = 1; |
||||
Server server = 2; |
||||
Data data = 3; |
||||
} |
||||
|
||||
message Server { |
||||
message HTTP { |
||||
string network = 1; |
||||
string addr = 2; |
||||
google.protobuf.Duration timeout = 3; |
||||
} |
||||
message GRPC { |
||||
string network = 1; |
||||
string addr = 2; |
||||
google.protobuf.Duration timeout = 3; |
||||
} |
||||
HTTP http = 1; |
||||
GRPC grpc = 2; |
||||
} |
||||
|
||||
message Data { |
||||
message Database { |
||||
string driver = 1; |
||||
string source = 2; |
||||
} |
||||
message Redis { |
||||
string network = 1; |
||||
string addr = 2; |
||||
string password = 3; |
||||
int32 db = 4; |
||||
google.protobuf.Duration dial_timeout = 5; |
||||
google.protobuf.Duration read_timeout = 6; |
||||
google.protobuf.Duration write_timeout = 7; |
||||
} |
||||
Database database = 1; |
||||
Redis redis = 2; |
||||
} |
||||
|
||||
message Trace { |
||||
string endpoint = 1; |
||||
} |
@ -0,0 +1,5 @@ |
||||
# Data |
||||
|
||||
``` |
||||
go generate ./ent |
||||
``` |
@ -0,0 +1,88 @@ |
||||
package data |
||||
|
||||
import ( |
||||
"context" |
||||
|
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/biz" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/conf" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent" |
||||
|
||||
"entgo.io/ent/dialect/sql" |
||||
"github.com/go-kratos/kratos/v2/log" |
||||
"github.com/google/wire" |
||||
|
||||
// init mysql driver
|
||||
_ "github.com/go-sql-driver/mysql" |
||||
) |
||||
|
||||
// ProviderSet is data providers.
|
||||
var ProviderSet = wire.NewSet(NewData, NewTransaction, NewUserRepo, NewCardRepo) |
||||
|
||||
// Data .
|
||||
type Data struct { |
||||
db *ent.Client |
||||
} |
||||
|
||||
type contextTxKey struct{} |
||||
|
||||
func (d *Data) ExecTx(ctx context.Context, f func(ctx context.Context) error) error { |
||||
tx, err := d.db.Tx(ctx) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
ctx = context.WithValue(ctx, contextTxKey{}, tx) |
||||
if err := f(ctx); err != nil { |
||||
_ = tx.Rollback() |
||||
return err |
||||
} |
||||
return tx.Commit() |
||||
} |
||||
|
||||
func (d *Data) User(ctx context.Context) *ent.UserClient { |
||||
tx, ok := ctx.Value(contextTxKey{}).(*ent.Tx) |
||||
if ok { |
||||
return tx.User |
||||
} |
||||
return d.db.User |
||||
} |
||||
|
||||
func (d *Data) Card(ctx context.Context) *ent.CardClient { |
||||
tx, ok := ctx.Value(contextTxKey{}).(*ent.Tx) |
||||
if ok { |
||||
return tx.Card |
||||
} |
||||
return d.db.Card |
||||
} |
||||
|
||||
func NewTransaction(d *Data) biz.Transaction { |
||||
return d |
||||
} |
||||
|
||||
// NewData .
|
||||
func NewData(conf *conf.Data, logger log.Logger) (*Data, func(), error) { |
||||
log := log.NewHelper(logger) |
||||
drv, err := sql.Open( |
||||
conf.Database.Driver, |
||||
conf.Database.Source, |
||||
) |
||||
client := ent.NewClient(ent.Driver(drv)) |
||||
if err != nil { |
||||
log.Errorf("failed opening connection to sqlite: %v", err) |
||||
return nil, nil, err |
||||
} |
||||
// Run the auto migration tool.
|
||||
if err := client.Schema.Create(context.Background()); err != nil { |
||||
log.Errorf("failed creating schema resources: %v", err) |
||||
return nil, nil, err |
||||
} |
||||
|
||||
d := &Data{ |
||||
db: client, |
||||
} |
||||
return d, func() { |
||||
log.Info("message", "closing the data resources") |
||||
if err := d.db.Close(); err != nil { |
||||
log.Error(err) |
||||
} |
||||
}, nil |
||||
} |
@ -0,0 +1,109 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package ent |
||||
|
||||
import ( |
||||
"fmt" |
||||
"strings" |
||||
|
||||
"entgo.io/ent/dialect/sql" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/card" |
||||
) |
||||
|
||||
// Card is the model entity for the Card schema.
|
||||
type Card struct { |
||||
config `json:"-"` |
||||
// ID of the ent.
|
||||
ID int `json:"id,omitempty"` |
||||
// UserID holds the value of the "user_id" field.
|
||||
UserID string `json:"user_id,omitempty"` |
||||
// Money holds the value of the "money" field.
|
||||
Money string `json:"money,omitempty"` |
||||
} |
||||
|
||||
// scanValues returns the types for scanning values from sql.Rows.
|
||||
func (*Card) scanValues(columns []string) ([]interface{}, error) { |
||||
values := make([]interface{}, len(columns)) |
||||
for i := range columns { |
||||
switch columns[i] { |
||||
case card.FieldID: |
||||
values[i] = new(sql.NullInt64) |
||||
case card.FieldUserID, card.FieldMoney: |
||||
values[i] = new(sql.NullString) |
||||
default: |
||||
return nil, fmt.Errorf("unexpected column %q for type Card", columns[i]) |
||||
} |
||||
} |
||||
return values, nil |
||||
} |
||||
|
||||
// assignValues assigns the values that were returned from sql.Rows (after scanning)
|
||||
// to the Card fields.
|
||||
func (c *Card) assignValues(columns []string, values []interface{}) error { |
||||
if m, n := len(values), len(columns); m < n { |
||||
return fmt.Errorf("mismatch number of scan values: %d != %d", m, n) |
||||
} |
||||
for i := range columns { |
||||
switch columns[i] { |
||||
case card.FieldID: |
||||
value, ok := values[i].(*sql.NullInt64) |
||||
if !ok { |
||||
return fmt.Errorf("unexpected type %T for field id", value) |
||||
} |
||||
c.ID = int(value.Int64) |
||||
case card.FieldUserID: |
||||
if value, ok := values[i].(*sql.NullString); !ok { |
||||
return fmt.Errorf("unexpected type %T for field user_id", values[i]) |
||||
} else if value.Valid { |
||||
c.UserID = value.String |
||||
} |
||||
case card.FieldMoney: |
||||
if value, ok := values[i].(*sql.NullString); !ok { |
||||
return fmt.Errorf("unexpected type %T for field money", values[i]) |
||||
} else if value.Valid { |
||||
c.Money = value.String |
||||
} |
||||
} |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
// Update returns a builder for updating this Card.
|
||||
// Note that you need to call Card.Unwrap() before calling this method if this Card
|
||||
// was returned from a transaction, and the transaction was committed or rolled back.
|
||||
func (c *Card) Update() *CardUpdateOne { |
||||
return (&CardClient{config: c.config}).UpdateOne(c) |
||||
} |
||||
|
||||
// Unwrap unwraps the Card entity that was returned from a transaction after it was closed,
|
||||
// so that all future queries will be executed through the driver which created the transaction.
|
||||
func (c *Card) Unwrap() *Card { |
||||
tx, ok := c.config.driver.(*txDriver) |
||||
if !ok { |
||||
panic("ent: Card is not a transactional entity") |
||||
} |
||||
c.config.driver = tx.drv |
||||
return c |
||||
} |
||||
|
||||
// String implements the fmt.Stringer.
|
||||
func (c *Card) String() string { |
||||
var builder strings.Builder |
||||
builder.WriteString("Card(") |
||||
builder.WriteString(fmt.Sprintf("id=%v", c.ID)) |
||||
builder.WriteString(", user_id=") |
||||
builder.WriteString(c.UserID) |
||||
builder.WriteString(", money=") |
||||
builder.WriteString(c.Money) |
||||
builder.WriteByte(')') |
||||
return builder.String() |
||||
} |
||||
|
||||
// Cards is a parsable slice of Card.
|
||||
type Cards []*Card |
||||
|
||||
func (c Cards) config(cfg config) { |
||||
for _i := range c { |
||||
c[_i].config = cfg |
||||
} |
||||
} |
@ -0,0 +1,33 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package card |
||||
|
||||
const ( |
||||
// Label holds the string label denoting the card type in the database.
|
||||
Label = "card" |
||||
// FieldID holds the string denoting the id field in the database.
|
||||
FieldID = "id" |
||||
// FieldUserID holds the string denoting the user_id field in the database.
|
||||
FieldUserID = "user_id" |
||||
// FieldMoney holds the string denoting the money field in the database.
|
||||
FieldMoney = "money" |
||||
// Table holds the table name of the card in the database.
|
||||
Table = "cards" |
||||
) |
||||
|
||||
// Columns holds all SQL columns for card fields.
|
||||
var Columns = []string{ |
||||
FieldID, |
||||
FieldUserID, |
||||
FieldMoney, |
||||
} |
||||
|
||||
// ValidColumn reports if the column name is valid (part of the table columns).
|
||||
func ValidColumn(column string) bool { |
||||
for i := range Columns { |
||||
if column == Columns[i] { |
||||
return true |
||||
} |
||||
} |
||||
return false |
||||
} |
@ -0,0 +1,359 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package card |
||||
|
||||
import ( |
||||
"entgo.io/ent/dialect/sql" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/predicate" |
||||
) |
||||
|
||||
// ID filters vertices based on their ID field.
|
||||
func ID(id int) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.EQ(s.C(FieldID), id)) |
||||
}) |
||||
} |
||||
|
||||
// IDEQ applies the EQ predicate on the ID field.
|
||||
func IDEQ(id int) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.EQ(s.C(FieldID), id)) |
||||
}) |
||||
} |
||||
|
||||
// IDNEQ applies the NEQ predicate on the ID field.
|
||||
func IDNEQ(id int) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.NEQ(s.C(FieldID), id)) |
||||
}) |
||||
} |
||||
|
||||
// IDIn applies the In predicate on the ID field.
|
||||
func IDIn(ids ...int) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
// if not arguments were provided, append the FALSE constants,
|
||||
// since we can't apply "IN ()". This will make this predicate falsy.
|
||||
if len(ids) == 0 { |
||||
s.Where(sql.False()) |
||||
return |
||||
} |
||||
v := make([]interface{}, len(ids)) |
||||
for i := range v { |
||||
v[i] = ids[i] |
||||
} |
||||
s.Where(sql.In(s.C(FieldID), v...)) |
||||
}) |
||||
} |
||||
|
||||
// IDNotIn applies the NotIn predicate on the ID field.
|
||||
func IDNotIn(ids ...int) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
// if not arguments were provided, append the FALSE constants,
|
||||
// since we can't apply "IN ()". This will make this predicate falsy.
|
||||
if len(ids) == 0 { |
||||
s.Where(sql.False()) |
||||
return |
||||
} |
||||
v := make([]interface{}, len(ids)) |
||||
for i := range v { |
||||
v[i] = ids[i] |
||||
} |
||||
s.Where(sql.NotIn(s.C(FieldID), v...)) |
||||
}) |
||||
} |
||||
|
||||
// IDGT applies the GT predicate on the ID field.
|
||||
func IDGT(id int) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.GT(s.C(FieldID), id)) |
||||
}) |
||||
} |
||||
|
||||
// IDGTE applies the GTE predicate on the ID field.
|
||||
func IDGTE(id int) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.GTE(s.C(FieldID), id)) |
||||
}) |
||||
} |
||||
|
||||
// IDLT applies the LT predicate on the ID field.
|
||||
func IDLT(id int) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.LT(s.C(FieldID), id)) |
||||
}) |
||||
} |
||||
|
||||
// IDLTE applies the LTE predicate on the ID field.
|
||||
func IDLTE(id int) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.LTE(s.C(FieldID), id)) |
||||
}) |
||||
} |
||||
|
||||
// UserID applies equality check predicate on the "user_id" field. It's identical to UserIDEQ.
|
||||
func UserID(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.EQ(s.C(FieldUserID), v)) |
||||
}) |
||||
} |
||||
|
||||
// Money applies equality check predicate on the "money" field. It's identical to MoneyEQ.
|
||||
func Money(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.EQ(s.C(FieldMoney), v)) |
||||
}) |
||||
} |
||||
|
||||
// UserIDEQ applies the EQ predicate on the "user_id" field.
|
||||
func UserIDEQ(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.EQ(s.C(FieldUserID), v)) |
||||
}) |
||||
} |
||||
|
||||
// UserIDNEQ applies the NEQ predicate on the "user_id" field.
|
||||
func UserIDNEQ(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.NEQ(s.C(FieldUserID), v)) |
||||
}) |
||||
} |
||||
|
||||
// UserIDIn applies the In predicate on the "user_id" field.
|
||||
func UserIDIn(vs ...string) predicate.Card { |
||||
v := make([]interface{}, len(vs)) |
||||
for i := range v { |
||||
v[i] = vs[i] |
||||
} |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
// if not arguments were provided, append the FALSE constants,
|
||||
// since we can't apply "IN ()". This will make this predicate falsy.
|
||||
if len(v) == 0 { |
||||
s.Where(sql.False()) |
||||
return |
||||
} |
||||
s.Where(sql.In(s.C(FieldUserID), v...)) |
||||
}) |
||||
} |
||||
|
||||
// UserIDNotIn applies the NotIn predicate on the "user_id" field.
|
||||
func UserIDNotIn(vs ...string) predicate.Card { |
||||
v := make([]interface{}, len(vs)) |
||||
for i := range v { |
||||
v[i] = vs[i] |
||||
} |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
// if not arguments were provided, append the FALSE constants,
|
||||
// since we can't apply "IN ()". This will make this predicate falsy.
|
||||
if len(v) == 0 { |
||||
s.Where(sql.False()) |
||||
return |
||||
} |
||||
s.Where(sql.NotIn(s.C(FieldUserID), v...)) |
||||
}) |
||||
} |
||||
|
||||
// UserIDGT applies the GT predicate on the "user_id" field.
|
||||
func UserIDGT(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.GT(s.C(FieldUserID), v)) |
||||
}) |
||||
} |
||||
|
||||
// UserIDGTE applies the GTE predicate on the "user_id" field.
|
||||
func UserIDGTE(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.GTE(s.C(FieldUserID), v)) |
||||
}) |
||||
} |
||||
|
||||
// UserIDLT applies the LT predicate on the "user_id" field.
|
||||
func UserIDLT(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.LT(s.C(FieldUserID), v)) |
||||
}) |
||||
} |
||||
|
||||
// UserIDLTE applies the LTE predicate on the "user_id" field.
|
||||
func UserIDLTE(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.LTE(s.C(FieldUserID), v)) |
||||
}) |
||||
} |
||||
|
||||
// UserIDContains applies the Contains predicate on the "user_id" field.
|
||||
func UserIDContains(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.Contains(s.C(FieldUserID), v)) |
||||
}) |
||||
} |
||||
|
||||
// UserIDHasPrefix applies the HasPrefix predicate on the "user_id" field.
|
||||
func UserIDHasPrefix(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.HasPrefix(s.C(FieldUserID), v)) |
||||
}) |
||||
} |
||||
|
||||
// UserIDHasSuffix applies the HasSuffix predicate on the "user_id" field.
|
||||
func UserIDHasSuffix(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.HasSuffix(s.C(FieldUserID), v)) |
||||
}) |
||||
} |
||||
|
||||
// UserIDEqualFold applies the EqualFold predicate on the "user_id" field.
|
||||
func UserIDEqualFold(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.EqualFold(s.C(FieldUserID), v)) |
||||
}) |
||||
} |
||||
|
||||
// UserIDContainsFold applies the ContainsFold predicate on the "user_id" field.
|
||||
func UserIDContainsFold(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.ContainsFold(s.C(FieldUserID), v)) |
||||
}) |
||||
} |
||||
|
||||
// MoneyEQ applies the EQ predicate on the "money" field.
|
||||
func MoneyEQ(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.EQ(s.C(FieldMoney), v)) |
||||
}) |
||||
} |
||||
|
||||
// MoneyNEQ applies the NEQ predicate on the "money" field.
|
||||
func MoneyNEQ(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.NEQ(s.C(FieldMoney), v)) |
||||
}) |
||||
} |
||||
|
||||
// MoneyIn applies the In predicate on the "money" field.
|
||||
func MoneyIn(vs ...string) predicate.Card { |
||||
v := make([]interface{}, len(vs)) |
||||
for i := range v { |
||||
v[i] = vs[i] |
||||
} |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
// if not arguments were provided, append the FALSE constants,
|
||||
// since we can't apply "IN ()". This will make this predicate falsy.
|
||||
if len(v) == 0 { |
||||
s.Where(sql.False()) |
||||
return |
||||
} |
||||
s.Where(sql.In(s.C(FieldMoney), v...)) |
||||
}) |
||||
} |
||||
|
||||
// MoneyNotIn applies the NotIn predicate on the "money" field.
|
||||
func MoneyNotIn(vs ...string) predicate.Card { |
||||
v := make([]interface{}, len(vs)) |
||||
for i := range v { |
||||
v[i] = vs[i] |
||||
} |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
// if not arguments were provided, append the FALSE constants,
|
||||
// since we can't apply "IN ()". This will make this predicate falsy.
|
||||
if len(v) == 0 { |
||||
s.Where(sql.False()) |
||||
return |
||||
} |
||||
s.Where(sql.NotIn(s.C(FieldMoney), v...)) |
||||
}) |
||||
} |
||||
|
||||
// MoneyGT applies the GT predicate on the "money" field.
|
||||
func MoneyGT(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.GT(s.C(FieldMoney), v)) |
||||
}) |
||||
} |
||||
|
||||
// MoneyGTE applies the GTE predicate on the "money" field.
|
||||
func MoneyGTE(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.GTE(s.C(FieldMoney), v)) |
||||
}) |
||||
} |
||||
|
||||
// MoneyLT applies the LT predicate on the "money" field.
|
||||
func MoneyLT(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.LT(s.C(FieldMoney), v)) |
||||
}) |
||||
} |
||||
|
||||
// MoneyLTE applies the LTE predicate on the "money" field.
|
||||
func MoneyLTE(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.LTE(s.C(FieldMoney), v)) |
||||
}) |
||||
} |
||||
|
||||
// MoneyContains applies the Contains predicate on the "money" field.
|
||||
func MoneyContains(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.Contains(s.C(FieldMoney), v)) |
||||
}) |
||||
} |
||||
|
||||
// MoneyHasPrefix applies the HasPrefix predicate on the "money" field.
|
||||
func MoneyHasPrefix(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.HasPrefix(s.C(FieldMoney), v)) |
||||
}) |
||||
} |
||||
|
||||
// MoneyHasSuffix applies the HasSuffix predicate on the "money" field.
|
||||
func MoneyHasSuffix(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.HasSuffix(s.C(FieldMoney), v)) |
||||
}) |
||||
} |
||||
|
||||
// MoneyEqualFold applies the EqualFold predicate on the "money" field.
|
||||
func MoneyEqualFold(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.EqualFold(s.C(FieldMoney), v)) |
||||
}) |
||||
} |
||||
|
||||
// MoneyContainsFold applies the ContainsFold predicate on the "money" field.
|
||||
func MoneyContainsFold(v string) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s.Where(sql.ContainsFold(s.C(FieldMoney), v)) |
||||
}) |
||||
} |
||||
|
||||
// And groups predicates with the AND operator between them.
|
||||
func And(predicates ...predicate.Card) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s1 := s.Clone().SetP(nil) |
||||
for _, p := range predicates { |
||||
p(s1) |
||||
} |
||||
s.Where(s1.P()) |
||||
}) |
||||
} |
||||
|
||||
// Or groups predicates with the OR operator between them.
|
||||
func Or(predicates ...predicate.Card) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
s1 := s.Clone().SetP(nil) |
||||
for i, p := range predicates { |
||||
if i > 0 { |
||||
s1.Or() |
||||
} |
||||
p(s1) |
||||
} |
||||
s.Where(s1.P()) |
||||
}) |
||||
} |
||||
|
||||
// Not applies the not operator on the given predicate.
|
||||
func Not(p predicate.Card) predicate.Card { |
||||
return predicate.Card(func(s *sql.Selector) { |
||||
p(s.Not()) |
||||
}) |
||||
} |
@ -0,0 +1,249 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package ent |
||||
|
||||
import ( |
||||
"context" |
||||
"errors" |
||||
"fmt" |
||||
|
||||
"entgo.io/ent/dialect/sql/sqlgraph" |
||||
"entgo.io/ent/schema/field" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/card" |
||||
) |
||||
|
||||
// CardCreate is the builder for creating a Card entity.
|
||||
type CardCreate struct { |
||||
config |
||||
mutation *CardMutation |
||||
hooks []Hook |
||||
} |
||||
|
||||
// SetUserID sets the "user_id" field.
|
||||
func (cc *CardCreate) SetUserID(s string) *CardCreate { |
||||
cc.mutation.SetUserID(s) |
||||
return cc |
||||
} |
||||
|
||||
// SetMoney sets the "money" field.
|
||||
func (cc *CardCreate) SetMoney(s string) *CardCreate { |
||||
cc.mutation.SetMoney(s) |
||||
return cc |
||||
} |
||||
|
||||
// SetID sets the "id" field.
|
||||
func (cc *CardCreate) SetID(i int) *CardCreate { |
||||
cc.mutation.SetID(i) |
||||
return cc |
||||
} |
||||
|
||||
// Mutation returns the CardMutation object of the builder.
|
||||
func (cc *CardCreate) Mutation() *CardMutation { |
||||
return cc.mutation |
||||
} |
||||
|
||||
// Save creates the Card in the database.
|
||||
func (cc *CardCreate) Save(ctx context.Context) (*Card, error) { |
||||
var ( |
||||
err error |
||||
node *Card |
||||
) |
||||
if len(cc.hooks) == 0 { |
||||
if err = cc.check(); err != nil { |
||||
return nil, err |
||||
} |
||||
node, err = cc.sqlSave(ctx) |
||||
} else { |
||||
var mut Mutator = MutateFunc(func(ctx context.Context, m Mutation) (Value, error) { |
||||
mutation, ok := m.(*CardMutation) |
||||
if !ok { |
||||
return nil, fmt.Errorf("unexpected mutation type %T", m) |
||||
} |
||||
if err = cc.check(); err != nil { |
||||
return nil, err |
||||
} |
||||
cc.mutation = mutation |
||||
if node, err = cc.sqlSave(ctx); err != nil { |
||||
return nil, err |
||||
} |
||||
mutation.id = &node.ID |
||||
mutation.done = true |
||||
return node, err |
||||
}) |
||||
for i := len(cc.hooks) - 1; i >= 0; i-- { |
||||
if cc.hooks[i] == nil { |
||||
return nil, fmt.Errorf("ent: uninitialized hook (forgotten import ent/runtime?)") |
||||
} |
||||
mut = cc.hooks[i](mut) |
||||
} |
||||
if _, err := mut.Mutate(ctx, cc.mutation); err != nil { |
||||
return nil, err |
||||
} |
||||
} |
||||
return node, err |
||||
} |
||||
|
||||
// SaveX calls Save and panics if Save returns an error.
|
||||
func (cc *CardCreate) SaveX(ctx context.Context) *Card { |
||||
v, err := cc.Save(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Exec executes the query.
|
||||
func (cc *CardCreate) Exec(ctx context.Context) error { |
||||
_, err := cc.Save(ctx) |
||||
return err |
||||
} |
||||
|
||||
// ExecX is like Exec, but panics if an error occurs.
|
||||
func (cc *CardCreate) ExecX(ctx context.Context) { |
||||
if err := cc.Exec(ctx); err != nil { |
||||
panic(err) |
||||
} |
||||
} |
||||
|
||||
// check runs all checks and user-defined validators on the builder.
|
||||
func (cc *CardCreate) check() error { |
||||
if _, ok := cc.mutation.UserID(); !ok { |
||||
return &ValidationError{Name: "user_id", err: errors.New(`ent: missing required field "user_id"`)} |
||||
} |
||||
if _, ok := cc.mutation.Money(); !ok { |
||||
return &ValidationError{Name: "money", err: errors.New(`ent: missing required field "money"`)} |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
func (cc *CardCreate) sqlSave(ctx context.Context) (*Card, error) { |
||||
_node, _spec := cc.createSpec() |
||||
if err := sqlgraph.CreateNode(ctx, cc.driver, _spec); err != nil { |
||||
if sqlgraph.IsConstraintError(err) { |
||||
err = &ConstraintError{err.Error(), err} |
||||
} |
||||
return nil, err |
||||
} |
||||
if _node.ID == 0 { |
||||
id := _spec.ID.Value.(int64) |
||||
_node.ID = int(id) |
||||
} |
||||
return _node, nil |
||||
} |
||||
|
||||
func (cc *CardCreate) createSpec() (*Card, *sqlgraph.CreateSpec) { |
||||
var ( |
||||
_node = &Card{config: cc.config} |
||||
_spec = &sqlgraph.CreateSpec{ |
||||
Table: card.Table, |
||||
ID: &sqlgraph.FieldSpec{ |
||||
Type: field.TypeInt, |
||||
Column: card.FieldID, |
||||
}, |
||||
} |
||||
) |
||||
if id, ok := cc.mutation.ID(); ok { |
||||
_node.ID = id |
||||
_spec.ID.Value = id |
||||
} |
||||
if value, ok := cc.mutation.UserID(); ok { |
||||
_spec.Fields = append(_spec.Fields, &sqlgraph.FieldSpec{ |
||||
Type: field.TypeString, |
||||
Value: value, |
||||
Column: card.FieldUserID, |
||||
}) |
||||
_node.UserID = value |
||||
} |
||||
if value, ok := cc.mutation.Money(); ok { |
||||
_spec.Fields = append(_spec.Fields, &sqlgraph.FieldSpec{ |
||||
Type: field.TypeString, |
||||
Value: value, |
||||
Column: card.FieldMoney, |
||||
}) |
||||
_node.Money = value |
||||
} |
||||
return _node, _spec |
||||
} |
||||
|
||||
// CardCreateBulk is the builder for creating many Card entities in bulk.
|
||||
type CardCreateBulk struct { |
||||
config |
||||
builders []*CardCreate |
||||
} |
||||
|
||||
// Save creates the Card entities in the database.
|
||||
func (ccb *CardCreateBulk) Save(ctx context.Context) ([]*Card, error) { |
||||
specs := make([]*sqlgraph.CreateSpec, len(ccb.builders)) |
||||
nodes := make([]*Card, len(ccb.builders)) |
||||
mutators := make([]Mutator, len(ccb.builders)) |
||||
for i := range ccb.builders { |
||||
func(i int, root context.Context) { |
||||
builder := ccb.builders[i] |
||||
var mut Mutator = MutateFunc(func(ctx context.Context, m Mutation) (Value, error) { |
||||
mutation, ok := m.(*CardMutation) |
||||
if !ok { |
||||
return nil, fmt.Errorf("unexpected mutation type %T", m) |
||||
} |
||||
if err := builder.check(); err != nil { |
||||
return nil, err |
||||
} |
||||
builder.mutation = mutation |
||||
nodes[i], specs[i] = builder.createSpec() |
||||
var err error |
||||
if i < len(mutators)-1 { |
||||
_, err = mutators[i+1].Mutate(root, ccb.builders[i+1].mutation) |
||||
} else { |
||||
spec := &sqlgraph.BatchCreateSpec{Nodes: specs} |
||||
// Invoke the actual operation on the latest mutation in the chain.
|
||||
if err = sqlgraph.BatchCreate(ctx, ccb.driver, spec); err != nil { |
||||
if sqlgraph.IsConstraintError(err) { |
||||
err = &ConstraintError{err.Error(), err} |
||||
} |
||||
} |
||||
} |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
mutation.id = &nodes[i].ID |
||||
mutation.done = true |
||||
if specs[i].ID.Value != nil && nodes[i].ID == 0 { |
||||
id := specs[i].ID.Value.(int64) |
||||
nodes[i].ID = int(id) |
||||
} |
||||
return nodes[i], nil |
||||
}) |
||||
for i := len(builder.hooks) - 1; i >= 0; i-- { |
||||
mut = builder.hooks[i](mut) |
||||
} |
||||
mutators[i] = mut |
||||
}(i, ctx) |
||||
} |
||||
if len(mutators) > 0 { |
||||
if _, err := mutators[0].Mutate(ctx, ccb.builders[0].mutation); err != nil { |
||||
return nil, err |
||||
} |
||||
} |
||||
return nodes, nil |
||||
} |
||||
|
||||
// SaveX is like Save, but panics if an error occurs.
|
||||
func (ccb *CardCreateBulk) SaveX(ctx context.Context) []*Card { |
||||
v, err := ccb.Save(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Exec executes the query.
|
||||
func (ccb *CardCreateBulk) Exec(ctx context.Context) error { |
||||
_, err := ccb.Save(ctx) |
||||
return err |
||||
} |
||||
|
||||
// ExecX is like Exec, but panics if an error occurs.
|
||||
func (ccb *CardCreateBulk) ExecX(ctx context.Context) { |
||||
if err := ccb.Exec(ctx); err != nil { |
||||
panic(err) |
||||
} |
||||
} |
@ -0,0 +1,111 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package ent |
||||
|
||||
import ( |
||||
"context" |
||||
"fmt" |
||||
|
||||
"entgo.io/ent/dialect/sql" |
||||
"entgo.io/ent/dialect/sql/sqlgraph" |
||||
"entgo.io/ent/schema/field" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/card" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/predicate" |
||||
) |
||||
|
||||
// CardDelete is the builder for deleting a Card entity.
|
||||
type CardDelete struct { |
||||
config |
||||
hooks []Hook |
||||
mutation *CardMutation |
||||
} |
||||
|
||||
// Where appends a list predicates to the CardDelete builder.
|
||||
func (cd *CardDelete) Where(ps ...predicate.Card) *CardDelete { |
||||
cd.mutation.Where(ps...) |
||||
return cd |
||||
} |
||||
|
||||
// Exec executes the deletion query and returns how many vertices were deleted.
|
||||
func (cd *CardDelete) Exec(ctx context.Context) (int, error) { |
||||
var ( |
||||
err error |
||||
affected int |
||||
) |
||||
if len(cd.hooks) == 0 { |
||||
affected, err = cd.sqlExec(ctx) |
||||
} else { |
||||
var mut Mutator = MutateFunc(func(ctx context.Context, m Mutation) (Value, error) { |
||||
mutation, ok := m.(*CardMutation) |
||||
if !ok { |
||||
return nil, fmt.Errorf("unexpected mutation type %T", m) |
||||
} |
||||
cd.mutation = mutation |
||||
affected, err = cd.sqlExec(ctx) |
||||
mutation.done = true |
||||
return affected, err |
||||
}) |
||||
for i := len(cd.hooks) - 1; i >= 0; i-- { |
||||
if cd.hooks[i] == nil { |
||||
return 0, fmt.Errorf("ent: uninitialized hook (forgotten import ent/runtime?)") |
||||
} |
||||
mut = cd.hooks[i](mut) |
||||
} |
||||
if _, err := mut.Mutate(ctx, cd.mutation); err != nil { |
||||
return 0, err |
||||
} |
||||
} |
||||
return affected, err |
||||
} |
||||
|
||||
// ExecX is like Exec, but panics if an error occurs.
|
||||
func (cd *CardDelete) ExecX(ctx context.Context) int { |
||||
n, err := cd.Exec(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return n |
||||
} |
||||
|
||||
func (cd *CardDelete) sqlExec(ctx context.Context) (int, error) { |
||||
_spec := &sqlgraph.DeleteSpec{ |
||||
Node: &sqlgraph.NodeSpec{ |
||||
Table: card.Table, |
||||
ID: &sqlgraph.FieldSpec{ |
||||
Type: field.TypeInt, |
||||
Column: card.FieldID, |
||||
}, |
||||
}, |
||||
} |
||||
if ps := cd.mutation.predicates; len(ps) > 0 { |
||||
_spec.Predicate = func(selector *sql.Selector) { |
||||
for i := range ps { |
||||
ps[i](selector) |
||||
} |
||||
} |
||||
} |
||||
return sqlgraph.DeleteNodes(ctx, cd.driver, _spec) |
||||
} |
||||
|
||||
// CardDeleteOne is the builder for deleting a single Card entity.
|
||||
type CardDeleteOne struct { |
||||
cd *CardDelete |
||||
} |
||||
|
||||
// Exec executes the deletion query.
|
||||
func (cdo *CardDeleteOne) Exec(ctx context.Context) error { |
||||
n, err := cdo.cd.Exec(ctx) |
||||
switch { |
||||
case err != nil: |
||||
return err |
||||
case n == 0: |
||||
return &NotFoundError{card.Label} |
||||
default: |
||||
return nil |
||||
} |
||||
} |
||||
|
||||
// ExecX is like Exec, but panics if an error occurs.
|
||||
func (cdo *CardDeleteOne) ExecX(ctx context.Context) { |
||||
cdo.cd.ExecX(ctx) |
||||
} |
@ -0,0 +1,915 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package ent |
||||
|
||||
import ( |
||||
"context" |
||||
"errors" |
||||
"fmt" |
||||
"math" |
||||
|
||||
"entgo.io/ent/dialect/sql" |
||||
"entgo.io/ent/dialect/sql/sqlgraph" |
||||
"entgo.io/ent/schema/field" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/card" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/predicate" |
||||
) |
||||
|
||||
// CardQuery is the builder for querying Card entities.
|
||||
type CardQuery struct { |
||||
config |
||||
limit *int |
||||
offset *int |
||||
unique *bool |
||||
order []OrderFunc |
||||
fields []string |
||||
predicates []predicate.Card |
||||
// intermediate query (i.e. traversal path).
|
||||
sql *sql.Selector |
||||
path func(context.Context) (*sql.Selector, error) |
||||
} |
||||
|
||||
// Where adds a new predicate for the CardQuery builder.
|
||||
func (cq *CardQuery) Where(ps ...predicate.Card) *CardQuery { |
||||
cq.predicates = append(cq.predicates, ps...) |
||||
return cq |
||||
} |
||||
|
||||
// Limit adds a limit step to the query.
|
||||
func (cq *CardQuery) Limit(limit int) *CardQuery { |
||||
cq.limit = &limit |
||||
return cq |
||||
} |
||||
|
||||
// Offset adds an offset step to the query.
|
||||
func (cq *CardQuery) Offset(offset int) *CardQuery { |
||||
cq.offset = &offset |
||||
return cq |
||||
} |
||||
|
||||
// Unique configures the query builder to filter duplicate records on query.
|
||||
// By default, unique is set to true, and can be disabled using this method.
|
||||
func (cq *CardQuery) Unique(unique bool) *CardQuery { |
||||
cq.unique = &unique |
||||
return cq |
||||
} |
||||
|
||||
// Order adds an order step to the query.
|
||||
func (cq *CardQuery) Order(o ...OrderFunc) *CardQuery { |
||||
cq.order = append(cq.order, o...) |
||||
return cq |
||||
} |
||||
|
||||
// First returns the first Card entity from the query.
|
||||
// Returns a *NotFoundError when no Card was found.
|
||||
func (cq *CardQuery) First(ctx context.Context) (*Card, error) { |
||||
nodes, err := cq.Limit(1).All(ctx) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
if len(nodes) == 0 { |
||||
return nil, &NotFoundError{card.Label} |
||||
} |
||||
return nodes[0], nil |
||||
} |
||||
|
||||
// FirstX is like First, but panics if an error occurs.
|
||||
func (cq *CardQuery) FirstX(ctx context.Context) *Card { |
||||
node, err := cq.First(ctx) |
||||
if err != nil && !IsNotFound(err) { |
||||
panic(err) |
||||
} |
||||
return node |
||||
} |
||||
|
||||
// FirstID returns the first Card ID from the query.
|
||||
// Returns a *NotFoundError when no Card ID was found.
|
||||
func (cq *CardQuery) FirstID(ctx context.Context) (id int, err error) { |
||||
var ids []int |
||||
if ids, err = cq.Limit(1).IDs(ctx); err != nil { |
||||
return |
||||
} |
||||
if len(ids) == 0 { |
||||
err = &NotFoundError{card.Label} |
||||
return |
||||
} |
||||
return ids[0], nil |
||||
} |
||||
|
||||
// FirstIDX is like FirstID, but panics if an error occurs.
|
||||
func (cq *CardQuery) FirstIDX(ctx context.Context) int { |
||||
id, err := cq.FirstID(ctx) |
||||
if err != nil && !IsNotFound(err) { |
||||
panic(err) |
||||
} |
||||
return id |
||||
} |
||||
|
||||
// Only returns a single Card entity found by the query, ensuring it only returns one.
|
||||
// Returns a *NotSingularError when exactly one Card entity is not found.
|
||||
// Returns a *NotFoundError when no Card entities are found.
|
||||
func (cq *CardQuery) Only(ctx context.Context) (*Card, error) { |
||||
nodes, err := cq.Limit(2).All(ctx) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
switch len(nodes) { |
||||
case 1: |
||||
return nodes[0], nil |
||||
case 0: |
||||
return nil, &NotFoundError{card.Label} |
||||
default: |
||||
return nil, &NotSingularError{card.Label} |
||||
} |
||||
} |
||||
|
||||
// OnlyX is like Only, but panics if an error occurs.
|
||||
func (cq *CardQuery) OnlyX(ctx context.Context) *Card { |
||||
node, err := cq.Only(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return node |
||||
} |
||||
|
||||
// OnlyID is like Only, but returns the only Card ID in the query.
|
||||
// Returns a *NotSingularError when exactly one Card ID is not found.
|
||||
// Returns a *NotFoundError when no entities are found.
|
||||
func (cq *CardQuery) OnlyID(ctx context.Context) (id int, err error) { |
||||
var ids []int |
||||
if ids, err = cq.Limit(2).IDs(ctx); err != nil { |
||||
return |
||||
} |
||||
switch len(ids) { |
||||
case 1: |
||||
id = ids[0] |
||||
case 0: |
||||
err = &NotFoundError{card.Label} |
||||
default: |
||||
err = &NotSingularError{card.Label} |
||||
} |
||||
return |
||||
} |
||||
|
||||
// OnlyIDX is like OnlyID, but panics if an error occurs.
|
||||
func (cq *CardQuery) OnlyIDX(ctx context.Context) int { |
||||
id, err := cq.OnlyID(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return id |
||||
} |
||||
|
||||
// All executes the query and returns a list of Cards.
|
||||
func (cq *CardQuery) All(ctx context.Context) ([]*Card, error) { |
||||
if err := cq.prepareQuery(ctx); err != nil { |
||||
return nil, err |
||||
} |
||||
return cq.sqlAll(ctx) |
||||
} |
||||
|
||||
// AllX is like All, but panics if an error occurs.
|
||||
func (cq *CardQuery) AllX(ctx context.Context) []*Card { |
||||
nodes, err := cq.All(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return nodes |
||||
} |
||||
|
||||
// IDs executes the query and returns a list of Card IDs.
|
||||
func (cq *CardQuery) IDs(ctx context.Context) ([]int, error) { |
||||
var ids []int |
||||
if err := cq.Select(card.FieldID).Scan(ctx, &ids); err != nil { |
||||
return nil, err |
||||
} |
||||
return ids, nil |
||||
} |
||||
|
||||
// IDsX is like IDs, but panics if an error occurs.
|
||||
func (cq *CardQuery) IDsX(ctx context.Context) []int { |
||||
ids, err := cq.IDs(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return ids |
||||
} |
||||
|
||||
// Count returns the count of the given query.
|
||||
func (cq *CardQuery) Count(ctx context.Context) (int, error) { |
||||
if err := cq.prepareQuery(ctx); err != nil { |
||||
return 0, err |
||||
} |
||||
return cq.sqlCount(ctx) |
||||
} |
||||
|
||||
// CountX is like Count, but panics if an error occurs.
|
||||
func (cq *CardQuery) CountX(ctx context.Context) int { |
||||
count, err := cq.Count(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return count |
||||
} |
||||
|
||||
// Exist returns true if the query has elements in the graph.
|
||||
func (cq *CardQuery) Exist(ctx context.Context) (bool, error) { |
||||
if err := cq.prepareQuery(ctx); err != nil { |
||||
return false, err |
||||
} |
||||
return cq.sqlExist(ctx) |
||||
} |
||||
|
||||
// ExistX is like Exist, but panics if an error occurs.
|
||||
func (cq *CardQuery) ExistX(ctx context.Context) bool { |
||||
exist, err := cq.Exist(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return exist |
||||
} |
||||
|
||||
// Clone returns a duplicate of the CardQuery builder, including all associated steps. It can be
|
||||
// used to prepare common query builders and use them differently after the clone is made.
|
||||
func (cq *CardQuery) Clone() *CardQuery { |
||||
if cq == nil { |
||||
return nil |
||||
} |
||||
return &CardQuery{ |
||||
config: cq.config, |
||||
limit: cq.limit, |
||||
offset: cq.offset, |
||||
order: append([]OrderFunc{}, cq.order...), |
||||
predicates: append([]predicate.Card{}, cq.predicates...), |
||||
// clone intermediate query.
|
||||
sql: cq.sql.Clone(), |
||||
path: cq.path, |
||||
} |
||||
} |
||||
|
||||
// GroupBy is used to group vertices by one or more fields/columns.
|
||||
// It is often used with aggregate functions, like: count, max, mean, min, sum.
|
||||
//
|
||||
// Example:
|
||||
//
|
||||
// var v []struct {
|
||||
// UserID string `json:"user_id,omitempty"`
|
||||
// Count int `json:"count,omitempty"`
|
||||
// }
|
||||
//
|
||||
// client.Card.Query().
|
||||
// GroupBy(card.FieldUserID).
|
||||
// Aggregate(ent.Count()).
|
||||
// Scan(ctx, &v)
|
||||
//
|
||||
func (cq *CardQuery) GroupBy(field string, fields ...string) *CardGroupBy { |
||||
group := &CardGroupBy{config: cq.config} |
||||
group.fields = append([]string{field}, fields...) |
||||
group.path = func(ctx context.Context) (prev *sql.Selector, err error) { |
||||
if err := cq.prepareQuery(ctx); err != nil { |
||||
return nil, err |
||||
} |
||||
return cq.sqlQuery(ctx), nil |
||||
} |
||||
return group |
||||
} |
||||
|
||||
// Select allows the selection one or more fields/columns for the given query,
|
||||
// instead of selecting all fields in the entity.
|
||||
//
|
||||
// Example:
|
||||
//
|
||||
// var v []struct {
|
||||
// UserID string `json:"user_id,omitempty"`
|
||||
// }
|
||||
//
|
||||
// client.Card.Query().
|
||||
// Select(card.FieldUserID).
|
||||
// Scan(ctx, &v)
|
||||
//
|
||||
func (cq *CardQuery) Select(fields ...string) *CardSelect { |
||||
cq.fields = append(cq.fields, fields...) |
||||
return &CardSelect{CardQuery: cq} |
||||
} |
||||
|
||||
func (cq *CardQuery) prepareQuery(ctx context.Context) error { |
||||
for _, f := range cq.fields { |
||||
if !card.ValidColumn(f) { |
||||
return &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)} |
||||
} |
||||
} |
||||
if cq.path != nil { |
||||
prev, err := cq.path(ctx) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
cq.sql = prev |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
func (cq *CardQuery) sqlAll(ctx context.Context) ([]*Card, error) { |
||||
var ( |
||||
nodes = []*Card{} |
||||
_spec = cq.querySpec() |
||||
) |
||||
_spec.ScanValues = func(columns []string) ([]interface{}, error) { |
||||
node := &Card{config: cq.config} |
||||
nodes = append(nodes, node) |
||||
return node.scanValues(columns) |
||||
} |
||||
_spec.Assign = func(columns []string, values []interface{}) error { |
||||
if len(nodes) == 0 { |
||||
return fmt.Errorf("ent: Assign called without calling ScanValues") |
||||
} |
||||
node := nodes[len(nodes)-1] |
||||
return node.assignValues(columns, values) |
||||
} |
||||
if err := sqlgraph.QueryNodes(ctx, cq.driver, _spec); err != nil { |
||||
return nil, err |
||||
} |
||||
if len(nodes) == 0 { |
||||
return nodes, nil |
||||
} |
||||
return nodes, nil |
||||
} |
||||
|
||||
func (cq *CardQuery) sqlCount(ctx context.Context) (int, error) { |
||||
_spec := cq.querySpec() |
||||
return sqlgraph.CountNodes(ctx, cq.driver, _spec) |
||||
} |
||||
|
||||
func (cq *CardQuery) sqlExist(ctx context.Context) (bool, error) { |
||||
n, err := cq.sqlCount(ctx) |
||||
if err != nil { |
||||
return false, fmt.Errorf("ent: check existence: %w", err) |
||||
} |
||||
return n > 0, nil |
||||
} |
||||
|
||||
func (cq *CardQuery) querySpec() *sqlgraph.QuerySpec { |
||||
_spec := &sqlgraph.QuerySpec{ |
||||
Node: &sqlgraph.NodeSpec{ |
||||
Table: card.Table, |
||||
Columns: card.Columns, |
||||
ID: &sqlgraph.FieldSpec{ |
||||
Type: field.TypeInt, |
||||
Column: card.FieldID, |
||||
}, |
||||
}, |
||||
From: cq.sql, |
||||
Unique: true, |
||||
} |
||||
if unique := cq.unique; unique != nil { |
||||
_spec.Unique = *unique |
||||
} |
||||
if fields := cq.fields; len(fields) > 0 { |
||||
_spec.Node.Columns = make([]string, 0, len(fields)) |
||||
_spec.Node.Columns = append(_spec.Node.Columns, card.FieldID) |
||||
for i := range fields { |
||||
if fields[i] != card.FieldID { |
||||
_spec.Node.Columns = append(_spec.Node.Columns, fields[i]) |
||||
} |
||||
} |
||||
} |
||||
if ps := cq.predicates; len(ps) > 0 { |
||||
_spec.Predicate = func(selector *sql.Selector) { |
||||
for i := range ps { |
||||
ps[i](selector) |
||||
} |
||||
} |
||||
} |
||||
if limit := cq.limit; limit != nil { |
||||
_spec.Limit = *limit |
||||
} |
||||
if offset := cq.offset; offset != nil { |
||||
_spec.Offset = *offset |
||||
} |
||||
if ps := cq.order; len(ps) > 0 { |
||||
_spec.Order = func(selector *sql.Selector) { |
||||
for i := range ps { |
||||
ps[i](selector) |
||||
} |
||||
} |
||||
} |
||||
return _spec |
||||
} |
||||
|
||||
func (cq *CardQuery) sqlQuery(ctx context.Context) *sql.Selector { |
||||
builder := sql.Dialect(cq.driver.Dialect()) |
||||
t1 := builder.Table(card.Table) |
||||
columns := cq.fields |
||||
if len(columns) == 0 { |
||||
columns = card.Columns |
||||
} |
||||
selector := builder.Select(t1.Columns(columns...)...).From(t1) |
||||
if cq.sql != nil { |
||||
selector = cq.sql |
||||
selector.Select(selector.Columns(columns...)...) |
||||
} |
||||
for _, p := range cq.predicates { |
||||
p(selector) |
||||
} |
||||
for _, p := range cq.order { |
||||
p(selector) |
||||
} |
||||
if offset := cq.offset; offset != nil { |
||||
// limit is mandatory for offset clause. We start
|
||||
// with default value, and override it below if needed.
|
||||
selector.Offset(*offset).Limit(math.MaxInt32) |
||||
} |
||||
if limit := cq.limit; limit != nil { |
||||
selector.Limit(*limit) |
||||
} |
||||
return selector |
||||
} |
||||
|
||||
// CardGroupBy is the group-by builder for Card entities.
|
||||
type CardGroupBy struct { |
||||
config |
||||
fields []string |
||||
fns []AggregateFunc |
||||
// intermediate query (i.e. traversal path).
|
||||
sql *sql.Selector |
||||
path func(context.Context) (*sql.Selector, error) |
||||
} |
||||
|
||||
// Aggregate adds the given aggregation functions to the group-by query.
|
||||
func (cgb *CardGroupBy) Aggregate(fns ...AggregateFunc) *CardGroupBy { |
||||
cgb.fns = append(cgb.fns, fns...) |
||||
return cgb |
||||
} |
||||
|
||||
// Scan applies the group-by query and scans the result into the given value.
|
||||
func (cgb *CardGroupBy) Scan(ctx context.Context, v interface{}) error { |
||||
query, err := cgb.path(ctx) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
cgb.sql = query |
||||
return cgb.sqlScan(ctx, v) |
||||
} |
||||
|
||||
// ScanX is like Scan, but panics if an error occurs.
|
||||
func (cgb *CardGroupBy) ScanX(ctx context.Context, v interface{}) { |
||||
if err := cgb.Scan(ctx, v); err != nil { |
||||
panic(err) |
||||
} |
||||
} |
||||
|
||||
// Strings returns list of strings from group-by.
|
||||
// It is only allowed when executing a group-by query with one field.
|
||||
func (cgb *CardGroupBy) Strings(ctx context.Context) ([]string, error) { |
||||
if len(cgb.fields) > 1 { |
||||
return nil, errors.New("ent: CardGroupBy.Strings is not achievable when grouping more than 1 field") |
||||
} |
||||
var v []string |
||||
if err := cgb.Scan(ctx, &v); err != nil { |
||||
return nil, err |
||||
} |
||||
return v, nil |
||||
} |
||||
|
||||
// StringsX is like Strings, but panics if an error occurs.
|
||||
func (cgb *CardGroupBy) StringsX(ctx context.Context) []string { |
||||
v, err := cgb.Strings(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// String returns a single string from a group-by query.
|
||||
// It is only allowed when executing a group-by query with one field.
|
||||
func (cgb *CardGroupBy) String(ctx context.Context) (_ string, err error) { |
||||
var v []string |
||||
if v, err = cgb.Strings(ctx); err != nil { |
||||
return |
||||
} |
||||
switch len(v) { |
||||
case 1: |
||||
return v[0], nil |
||||
case 0: |
||||
err = &NotFoundError{card.Label} |
||||
default: |
||||
err = fmt.Errorf("ent: CardGroupBy.Strings returned %d results when one was expected", len(v)) |
||||
} |
||||
return |
||||
} |
||||
|
||||
// StringX is like String, but panics if an error occurs.
|
||||
func (cgb *CardGroupBy) StringX(ctx context.Context) string { |
||||
v, err := cgb.String(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Ints returns list of ints from group-by.
|
||||
// It is only allowed when executing a group-by query with one field.
|
||||
func (cgb *CardGroupBy) Ints(ctx context.Context) ([]int, error) { |
||||
if len(cgb.fields) > 1 { |
||||
return nil, errors.New("ent: CardGroupBy.Ints is not achievable when grouping more than 1 field") |
||||
} |
||||
var v []int |
||||
if err := cgb.Scan(ctx, &v); err != nil { |
||||
return nil, err |
||||
} |
||||
return v, nil |
||||
} |
||||
|
||||
// IntsX is like Ints, but panics if an error occurs.
|
||||
func (cgb *CardGroupBy) IntsX(ctx context.Context) []int { |
||||
v, err := cgb.Ints(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Int returns a single int from a group-by query.
|
||||
// It is only allowed when executing a group-by query with one field.
|
||||
func (cgb *CardGroupBy) Int(ctx context.Context) (_ int, err error) { |
||||
var v []int |
||||
if v, err = cgb.Ints(ctx); err != nil { |
||||
return |
||||
} |
||||
switch len(v) { |
||||
case 1: |
||||
return v[0], nil |
||||
case 0: |
||||
err = &NotFoundError{card.Label} |
||||
default: |
||||
err = fmt.Errorf("ent: CardGroupBy.Ints returned %d results when one was expected", len(v)) |
||||
} |
||||
return |
||||
} |
||||
|
||||
// IntX is like Int, but panics if an error occurs.
|
||||
func (cgb *CardGroupBy) IntX(ctx context.Context) int { |
||||
v, err := cgb.Int(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Float64s returns list of float64s from group-by.
|
||||
// It is only allowed when executing a group-by query with one field.
|
||||
func (cgb *CardGroupBy) Float64s(ctx context.Context) ([]float64, error) { |
||||
if len(cgb.fields) > 1 { |
||||
return nil, errors.New("ent: CardGroupBy.Float64s is not achievable when grouping more than 1 field") |
||||
} |
||||
var v []float64 |
||||
if err := cgb.Scan(ctx, &v); err != nil { |
||||
return nil, err |
||||
} |
||||
return v, nil |
||||
} |
||||
|
||||
// Float64sX is like Float64s, but panics if an error occurs.
|
||||
func (cgb *CardGroupBy) Float64sX(ctx context.Context) []float64 { |
||||
v, err := cgb.Float64s(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Float64 returns a single float64 from a group-by query.
|
||||
// It is only allowed when executing a group-by query with one field.
|
||||
func (cgb *CardGroupBy) Float64(ctx context.Context) (_ float64, err error) { |
||||
var v []float64 |
||||
if v, err = cgb.Float64s(ctx); err != nil { |
||||
return |
||||
} |
||||
switch len(v) { |
||||
case 1: |
||||
return v[0], nil |
||||
case 0: |
||||
err = &NotFoundError{card.Label} |
||||
default: |
||||
err = fmt.Errorf("ent: CardGroupBy.Float64s returned %d results when one was expected", len(v)) |
||||
} |
||||
return |
||||
} |
||||
|
||||
// Float64X is like Float64, but panics if an error occurs.
|
||||
func (cgb *CardGroupBy) Float64X(ctx context.Context) float64 { |
||||
v, err := cgb.Float64(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Bools returns list of bools from group-by.
|
||||
// It is only allowed when executing a group-by query with one field.
|
||||
func (cgb *CardGroupBy) Bools(ctx context.Context) ([]bool, error) { |
||||
if len(cgb.fields) > 1 { |
||||
return nil, errors.New("ent: CardGroupBy.Bools is not achievable when grouping more than 1 field") |
||||
} |
||||
var v []bool |
||||
if err := cgb.Scan(ctx, &v); err != nil { |
||||
return nil, err |
||||
} |
||||
return v, nil |
||||
} |
||||
|
||||
// BoolsX is like Bools, but panics if an error occurs.
|
||||
func (cgb *CardGroupBy) BoolsX(ctx context.Context) []bool { |
||||
v, err := cgb.Bools(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Bool returns a single bool from a group-by query.
|
||||
// It is only allowed when executing a group-by query with one field.
|
||||
func (cgb *CardGroupBy) Bool(ctx context.Context) (_ bool, err error) { |
||||
var v []bool |
||||
if v, err = cgb.Bools(ctx); err != nil { |
||||
return |
||||
} |
||||
switch len(v) { |
||||
case 1: |
||||
return v[0], nil |
||||
case 0: |
||||
err = &NotFoundError{card.Label} |
||||
default: |
||||
err = fmt.Errorf("ent: CardGroupBy.Bools returned %d results when one was expected", len(v)) |
||||
} |
||||
return |
||||
} |
||||
|
||||
// BoolX is like Bool, but panics if an error occurs.
|
||||
func (cgb *CardGroupBy) BoolX(ctx context.Context) bool { |
||||
v, err := cgb.Bool(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
func (cgb *CardGroupBy) sqlScan(ctx context.Context, v interface{}) error { |
||||
for _, f := range cgb.fields { |
||||
if !card.ValidColumn(f) { |
||||
return &ValidationError{Name: f, err: fmt.Errorf("invalid field %q for group-by", f)} |
||||
} |
||||
} |
||||
selector := cgb.sqlQuery() |
||||
if err := selector.Err(); err != nil { |
||||
return err |
||||
} |
||||
rows := &sql.Rows{} |
||||
query, args := selector.Query() |
||||
if err := cgb.driver.Query(ctx, query, args, rows); err != nil { |
||||
return err |
||||
} |
||||
defer rows.Close() |
||||
return sql.ScanSlice(rows, v) |
||||
} |
||||
|
||||
func (cgb *CardGroupBy) sqlQuery() *sql.Selector { |
||||
selector := cgb.sql.Select() |
||||
aggregation := make([]string, 0, len(cgb.fns)) |
||||
for _, fn := range cgb.fns { |
||||
aggregation = append(aggregation, fn(selector)) |
||||
} |
||||
// If no columns were selected in a custom aggregation function, the default
|
||||
// selection is the fields used for "group-by", and the aggregation functions.
|
||||
if len(selector.SelectedColumns()) == 0 { |
||||
columns := make([]string, 0, len(cgb.fields)+len(cgb.fns)) |
||||
for _, f := range cgb.fields { |
||||
columns = append(columns, selector.C(f)) |
||||
} |
||||
for _, c := range aggregation { |
||||
columns = append(columns, c) |
||||
} |
||||
selector.Select(columns...) |
||||
} |
||||
return selector.GroupBy(selector.Columns(cgb.fields...)...) |
||||
} |
||||
|
||||
// CardSelect is the builder for selecting fields of Card entities.
|
||||
type CardSelect struct { |
||||
*CardQuery |
||||
// intermediate query (i.e. traversal path).
|
||||
sql *sql.Selector |
||||
} |
||||
|
||||
// Scan applies the selector query and scans the result into the given value.
|
||||
func (cs *CardSelect) Scan(ctx context.Context, v interface{}) error { |
||||
if err := cs.prepareQuery(ctx); err != nil { |
||||
return err |
||||
} |
||||
cs.sql = cs.CardQuery.sqlQuery(ctx) |
||||
return cs.sqlScan(ctx, v) |
||||
} |
||||
|
||||
// ScanX is like Scan, but panics if an error occurs.
|
||||
func (cs *CardSelect) ScanX(ctx context.Context, v interface{}) { |
||||
if err := cs.Scan(ctx, v); err != nil { |
||||
panic(err) |
||||
} |
||||
} |
||||
|
||||
// Strings returns list of strings from a selector. It is only allowed when selecting one field.
|
||||
func (cs *CardSelect) Strings(ctx context.Context) ([]string, error) { |
||||
if len(cs.fields) > 1 { |
||||
return nil, errors.New("ent: CardSelect.Strings is not achievable when selecting more than 1 field") |
||||
} |
||||
var v []string |
||||
if err := cs.Scan(ctx, &v); err != nil { |
||||
return nil, err |
||||
} |
||||
return v, nil |
||||
} |
||||
|
||||
// StringsX is like Strings, but panics if an error occurs.
|
||||
func (cs *CardSelect) StringsX(ctx context.Context) []string { |
||||
v, err := cs.Strings(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// String returns a single string from a selector. It is only allowed when selecting one field.
|
||||
func (cs *CardSelect) String(ctx context.Context) (_ string, err error) { |
||||
var v []string |
||||
if v, err = cs.Strings(ctx); err != nil { |
||||
return |
||||
} |
||||
switch len(v) { |
||||
case 1: |
||||
return v[0], nil |
||||
case 0: |
||||
err = &NotFoundError{card.Label} |
||||
default: |
||||
err = fmt.Errorf("ent: CardSelect.Strings returned %d results when one was expected", len(v)) |
||||
} |
||||
return |
||||
} |
||||
|
||||
// StringX is like String, but panics if an error occurs.
|
||||
func (cs *CardSelect) StringX(ctx context.Context) string { |
||||
v, err := cs.String(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Ints returns list of ints from a selector. It is only allowed when selecting one field.
|
||||
func (cs *CardSelect) Ints(ctx context.Context) ([]int, error) { |
||||
if len(cs.fields) > 1 { |
||||
return nil, errors.New("ent: CardSelect.Ints is not achievable when selecting more than 1 field") |
||||
} |
||||
var v []int |
||||
if err := cs.Scan(ctx, &v); err != nil { |
||||
return nil, err |
||||
} |
||||
return v, nil |
||||
} |
||||
|
||||
// IntsX is like Ints, but panics if an error occurs.
|
||||
func (cs *CardSelect) IntsX(ctx context.Context) []int { |
||||
v, err := cs.Ints(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Int returns a single int from a selector. It is only allowed when selecting one field.
|
||||
func (cs *CardSelect) Int(ctx context.Context) (_ int, err error) { |
||||
var v []int |
||||
if v, err = cs.Ints(ctx); err != nil { |
||||
return |
||||
} |
||||
switch len(v) { |
||||
case 1: |
||||
return v[0], nil |
||||
case 0: |
||||
err = &NotFoundError{card.Label} |
||||
default: |
||||
err = fmt.Errorf("ent: CardSelect.Ints returned %d results when one was expected", len(v)) |
||||
} |
||||
return |
||||
} |
||||
|
||||
// IntX is like Int, but panics if an error occurs.
|
||||
func (cs *CardSelect) IntX(ctx context.Context) int { |
||||
v, err := cs.Int(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Float64s returns list of float64s from a selector. It is only allowed when selecting one field.
|
||||
func (cs *CardSelect) Float64s(ctx context.Context) ([]float64, error) { |
||||
if len(cs.fields) > 1 { |
||||
return nil, errors.New("ent: CardSelect.Float64s is not achievable when selecting more than 1 field") |
||||
} |
||||
var v []float64 |
||||
if err := cs.Scan(ctx, &v); err != nil { |
||||
return nil, err |
||||
} |
||||
return v, nil |
||||
} |
||||
|
||||
// Float64sX is like Float64s, but panics if an error occurs.
|
||||
func (cs *CardSelect) Float64sX(ctx context.Context) []float64 { |
||||
v, err := cs.Float64s(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Float64 returns a single float64 from a selector. It is only allowed when selecting one field.
|
||||
func (cs *CardSelect) Float64(ctx context.Context) (_ float64, err error) { |
||||
var v []float64 |
||||
if v, err = cs.Float64s(ctx); err != nil { |
||||
return |
||||
} |
||||
switch len(v) { |
||||
case 1: |
||||
return v[0], nil |
||||
case 0: |
||||
err = &NotFoundError{card.Label} |
||||
default: |
||||
err = fmt.Errorf("ent: CardSelect.Float64s returned %d results when one was expected", len(v)) |
||||
} |
||||
return |
||||
} |
||||
|
||||
// Float64X is like Float64, but panics if an error occurs.
|
||||
func (cs *CardSelect) Float64X(ctx context.Context) float64 { |
||||
v, err := cs.Float64(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Bools returns list of bools from a selector. It is only allowed when selecting one field.
|
||||
func (cs *CardSelect) Bools(ctx context.Context) ([]bool, error) { |
||||
if len(cs.fields) > 1 { |
||||
return nil, errors.New("ent: CardSelect.Bools is not achievable when selecting more than 1 field") |
||||
} |
||||
var v []bool |
||||
if err := cs.Scan(ctx, &v); err != nil { |
||||
return nil, err |
||||
} |
||||
return v, nil |
||||
} |
||||
|
||||
// BoolsX is like Bools, but panics if an error occurs.
|
||||
func (cs *CardSelect) BoolsX(ctx context.Context) []bool { |
||||
v, err := cs.Bools(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Bool returns a single bool from a selector. It is only allowed when selecting one field.
|
||||
func (cs *CardSelect) Bool(ctx context.Context) (_ bool, err error) { |
||||
var v []bool |
||||
if v, err = cs.Bools(ctx); err != nil { |
||||
return |
||||
} |
||||
switch len(v) { |
||||
case 1: |
||||
return v[0], nil |
||||
case 0: |
||||
err = &NotFoundError{card.Label} |
||||
default: |
||||
err = fmt.Errorf("ent: CardSelect.Bools returned %d results when one was expected", len(v)) |
||||
} |
||||
return |
||||
} |
||||
|
||||
// BoolX is like Bool, but panics if an error occurs.
|
||||
func (cs *CardSelect) BoolX(ctx context.Context) bool { |
||||
v, err := cs.Bool(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
func (cs *CardSelect) sqlScan(ctx context.Context, v interface{}) error { |
||||
rows := &sql.Rows{} |
||||
query, args := cs.sql.Query() |
||||
if err := cs.driver.Query(ctx, query, args, rows); err != nil { |
||||
return err |
||||
} |
||||
defer rows.Close() |
||||
return sql.ScanSlice(rows, v) |
||||
} |
@ -0,0 +1,290 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package ent |
||||
|
||||
import ( |
||||
"context" |
||||
"fmt" |
||||
|
||||
"entgo.io/ent/dialect/sql" |
||||
"entgo.io/ent/dialect/sql/sqlgraph" |
||||
"entgo.io/ent/schema/field" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/card" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/predicate" |
||||
) |
||||
|
||||
// CardUpdate is the builder for updating Card entities.
|
||||
type CardUpdate struct { |
||||
config |
||||
hooks []Hook |
||||
mutation *CardMutation |
||||
} |
||||
|
||||
// Where appends a list predicates to the CardUpdate builder.
|
||||
func (cu *CardUpdate) Where(ps ...predicate.Card) *CardUpdate { |
||||
cu.mutation.Where(ps...) |
||||
return cu |
||||
} |
||||
|
||||
// SetUserID sets the "user_id" field.
|
||||
func (cu *CardUpdate) SetUserID(s string) *CardUpdate { |
||||
cu.mutation.SetUserID(s) |
||||
return cu |
||||
} |
||||
|
||||
// SetMoney sets the "money" field.
|
||||
func (cu *CardUpdate) SetMoney(s string) *CardUpdate { |
||||
cu.mutation.SetMoney(s) |
||||
return cu |
||||
} |
||||
|
||||
// Mutation returns the CardMutation object of the builder.
|
||||
func (cu *CardUpdate) Mutation() *CardMutation { |
||||
return cu.mutation |
||||
} |
||||
|
||||
// Save executes the query and returns the number of nodes affected by the update operation.
|
||||
func (cu *CardUpdate) Save(ctx context.Context) (int, error) { |
||||
var ( |
||||
err error |
||||
affected int |
||||
) |
||||
if len(cu.hooks) == 0 { |
||||
affected, err = cu.sqlSave(ctx) |
||||
} else { |
||||
var mut Mutator = MutateFunc(func(ctx context.Context, m Mutation) (Value, error) { |
||||
mutation, ok := m.(*CardMutation) |
||||
if !ok { |
||||
return nil, fmt.Errorf("unexpected mutation type %T", m) |
||||
} |
||||
cu.mutation = mutation |
||||
affected, err = cu.sqlSave(ctx) |
||||
mutation.done = true |
||||
return affected, err |
||||
}) |
||||
for i := len(cu.hooks) - 1; i >= 0; i-- { |
||||
if cu.hooks[i] == nil { |
||||
return 0, fmt.Errorf("ent: uninitialized hook (forgotten import ent/runtime?)") |
||||
} |
||||
mut = cu.hooks[i](mut) |
||||
} |
||||
if _, err := mut.Mutate(ctx, cu.mutation); err != nil { |
||||
return 0, err |
||||
} |
||||
} |
||||
return affected, err |
||||
} |
||||
|
||||
// SaveX is like Save, but panics if an error occurs.
|
||||
func (cu *CardUpdate) SaveX(ctx context.Context) int { |
||||
affected, err := cu.Save(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return affected |
||||
} |
||||
|
||||
// Exec executes the query.
|
||||
func (cu *CardUpdate) Exec(ctx context.Context) error { |
||||
_, err := cu.Save(ctx) |
||||
return err |
||||
} |
||||
|
||||
// ExecX is like Exec, but panics if an error occurs.
|
||||
func (cu *CardUpdate) ExecX(ctx context.Context) { |
||||
if err := cu.Exec(ctx); err != nil { |
||||
panic(err) |
||||
} |
||||
} |
||||
|
||||
func (cu *CardUpdate) sqlSave(ctx context.Context) (n int, err error) { |
||||
_spec := &sqlgraph.UpdateSpec{ |
||||
Node: &sqlgraph.NodeSpec{ |
||||
Table: card.Table, |
||||
Columns: card.Columns, |
||||
ID: &sqlgraph.FieldSpec{ |
||||
Type: field.TypeInt, |
||||
Column: card.FieldID, |
||||
}, |
||||
}, |
||||
} |
||||
if ps := cu.mutation.predicates; len(ps) > 0 { |
||||
_spec.Predicate = func(selector *sql.Selector) { |
||||
for i := range ps { |
||||
ps[i](selector) |
||||
} |
||||
} |
||||
} |
||||
if value, ok := cu.mutation.UserID(); ok { |
||||
_spec.Fields.Set = append(_spec.Fields.Set, &sqlgraph.FieldSpec{ |
||||
Type: field.TypeString, |
||||
Value: value, |
||||
Column: card.FieldUserID, |
||||
}) |
||||
} |
||||
if value, ok := cu.mutation.Money(); ok { |
||||
_spec.Fields.Set = append(_spec.Fields.Set, &sqlgraph.FieldSpec{ |
||||
Type: field.TypeString, |
||||
Value: value, |
||||
Column: card.FieldMoney, |
||||
}) |
||||
} |
||||
if n, err = sqlgraph.UpdateNodes(ctx, cu.driver, _spec); err != nil { |
||||
if _, ok := err.(*sqlgraph.NotFoundError); ok { |
||||
err = &NotFoundError{card.Label} |
||||
} else if sqlgraph.IsConstraintError(err) { |
||||
err = &ConstraintError{err.Error(), err} |
||||
} |
||||
return 0, err |
||||
} |
||||
return n, nil |
||||
} |
||||
|
||||
// CardUpdateOne is the builder for updating a single Card entity.
|
||||
type CardUpdateOne struct { |
||||
config |
||||
fields []string |
||||
hooks []Hook |
||||
mutation *CardMutation |
||||
} |
||||
|
||||
// SetUserID sets the "user_id" field.
|
||||
func (cuo *CardUpdateOne) SetUserID(s string) *CardUpdateOne { |
||||
cuo.mutation.SetUserID(s) |
||||
return cuo |
||||
} |
||||
|
||||
// SetMoney sets the "money" field.
|
||||
func (cuo *CardUpdateOne) SetMoney(s string) *CardUpdateOne { |
||||
cuo.mutation.SetMoney(s) |
||||
return cuo |
||||
} |
||||
|
||||
// Mutation returns the CardMutation object of the builder.
|
||||
func (cuo *CardUpdateOne) Mutation() *CardMutation { |
||||
return cuo.mutation |
||||
} |
||||
|
||||
// Select allows selecting one or more fields (columns) of the returned entity.
|
||||
// The default is selecting all fields defined in the entity schema.
|
||||
func (cuo *CardUpdateOne) Select(field string, fields ...string) *CardUpdateOne { |
||||
cuo.fields = append([]string{field}, fields...) |
||||
return cuo |
||||
} |
||||
|
||||
// Save executes the query and returns the updated Card entity.
|
||||
func (cuo *CardUpdateOne) Save(ctx context.Context) (*Card, error) { |
||||
var ( |
||||
err error |
||||
node *Card |
||||
) |
||||
if len(cuo.hooks) == 0 { |
||||
node, err = cuo.sqlSave(ctx) |
||||
} else { |
||||
var mut Mutator = MutateFunc(func(ctx context.Context, m Mutation) (Value, error) { |
||||
mutation, ok := m.(*CardMutation) |
||||
if !ok { |
||||
return nil, fmt.Errorf("unexpected mutation type %T", m) |
||||
} |
||||
cuo.mutation = mutation |
||||
node, err = cuo.sqlSave(ctx) |
||||
mutation.done = true |
||||
return node, err |
||||
}) |
||||
for i := len(cuo.hooks) - 1; i >= 0; i-- { |
||||
if cuo.hooks[i] == nil { |
||||
return nil, fmt.Errorf("ent: uninitialized hook (forgotten import ent/runtime?)") |
||||
} |
||||
mut = cuo.hooks[i](mut) |
||||
} |
||||
if _, err := mut.Mutate(ctx, cuo.mutation); err != nil { |
||||
return nil, err |
||||
} |
||||
} |
||||
return node, err |
||||
} |
||||
|
||||
// SaveX is like Save, but panics if an error occurs.
|
||||
func (cuo *CardUpdateOne) SaveX(ctx context.Context) *Card { |
||||
node, err := cuo.Save(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return node |
||||
} |
||||
|
||||
// Exec executes the query on the entity.
|
||||
func (cuo *CardUpdateOne) Exec(ctx context.Context) error { |
||||
_, err := cuo.Save(ctx) |
||||
return err |
||||
} |
||||
|
||||
// ExecX is like Exec, but panics if an error occurs.
|
||||
func (cuo *CardUpdateOne) ExecX(ctx context.Context) { |
||||
if err := cuo.Exec(ctx); err != nil { |
||||
panic(err) |
||||
} |
||||
} |
||||
|
||||
func (cuo *CardUpdateOne) sqlSave(ctx context.Context) (_node *Card, err error) { |
||||
_spec := &sqlgraph.UpdateSpec{ |
||||
Node: &sqlgraph.NodeSpec{ |
||||
Table: card.Table, |
||||
Columns: card.Columns, |
||||
ID: &sqlgraph.FieldSpec{ |
||||
Type: field.TypeInt, |
||||
Column: card.FieldID, |
||||
}, |
||||
}, |
||||
} |
||||
id, ok := cuo.mutation.ID() |
||||
if !ok { |
||||
return nil, &ValidationError{Name: "ID", err: fmt.Errorf("missing Card.ID for update")} |
||||
} |
||||
_spec.Node.ID.Value = id |
||||
if fields := cuo.fields; len(fields) > 0 { |
||||
_spec.Node.Columns = make([]string, 0, len(fields)) |
||||
_spec.Node.Columns = append(_spec.Node.Columns, card.FieldID) |
||||
for _, f := range fields { |
||||
if !card.ValidColumn(f) { |
||||
return nil, &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)} |
||||
} |
||||
if f != card.FieldID { |
||||
_spec.Node.Columns = append(_spec.Node.Columns, f) |
||||
} |
||||
} |
||||
} |
||||
if ps := cuo.mutation.predicates; len(ps) > 0 { |
||||
_spec.Predicate = func(selector *sql.Selector) { |
||||
for i := range ps { |
||||
ps[i](selector) |
||||
} |
||||
} |
||||
} |
||||
if value, ok := cuo.mutation.UserID(); ok { |
||||
_spec.Fields.Set = append(_spec.Fields.Set, &sqlgraph.FieldSpec{ |
||||
Type: field.TypeString, |
||||
Value: value, |
||||
Column: card.FieldUserID, |
||||
}) |
||||
} |
||||
if value, ok := cuo.mutation.Money(); ok { |
||||
_spec.Fields.Set = append(_spec.Fields.Set, &sqlgraph.FieldSpec{ |
||||
Type: field.TypeString, |
||||
Value: value, |
||||
Column: card.FieldMoney, |
||||
}) |
||||
} |
||||
_node = &Card{config: cuo.config} |
||||
_spec.Assign = _node.assignValues |
||||
_spec.ScanValues = _node.scanValues |
||||
if err = sqlgraph.UpdateNode(ctx, cuo.driver, _spec); err != nil { |
||||
if _, ok := err.(*sqlgraph.NotFoundError); ok { |
||||
err = &NotFoundError{card.Label} |
||||
} else if sqlgraph.IsConstraintError(err) { |
||||
err = &ConstraintError{err.Error(), err} |
||||
} |
||||
return nil, err |
||||
} |
||||
return _node, nil |
||||
} |
@ -0,0 +1,309 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package ent |
||||
|
||||
import ( |
||||
"context" |
||||
"fmt" |
||||
"log" |
||||
|
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/migrate" |
||||
|
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/card" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/user" |
||||
|
||||
"entgo.io/ent/dialect" |
||||
"entgo.io/ent/dialect/sql" |
||||
) |
||||
|
||||
// Client is the client that holds all ent builders.
|
||||
type Client struct { |
||||
config |
||||
// Schema is the client for creating, migrating and dropping schema.
|
||||
Schema *migrate.Schema |
||||
// Card is the client for interacting with the Card builders.
|
||||
Card *CardClient |
||||
// User is the client for interacting with the User builders.
|
||||
User *UserClient |
||||
} |
||||
|
||||
// NewClient creates a new client configured with the given options.
|
||||
func NewClient(opts ...Option) *Client { |
||||
cfg := config{log: log.Println, hooks: &hooks{}} |
||||
cfg.options(opts...) |
||||
client := &Client{config: cfg} |
||||
client.init() |
||||
return client |
||||
} |
||||
|
||||
func (c *Client) init() { |
||||
c.Schema = migrate.NewSchema(c.driver) |
||||
c.Card = NewCardClient(c.config) |
||||
c.User = NewUserClient(c.config) |
||||
} |
||||
|
||||
// Open opens a database/sql.DB specified by the driver name and
|
||||
// the data source name, and returns a new client attached to it.
|
||||
// Optional parameters can be added for configuring the client.
|
||||
func Open(driverName, dataSourceName string, options ...Option) (*Client, error) { |
||||
switch driverName { |
||||
case dialect.MySQL, dialect.Postgres, dialect.SQLite: |
||||
drv, err := sql.Open(driverName, dataSourceName) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
return NewClient(append(options, Driver(drv))...), nil |
||||
default: |
||||
return nil, fmt.Errorf("unsupported driver: %q", driverName) |
||||
} |
||||
} |
||||
|
||||
// Tx returns a new transactional client. The provided context
|
||||
// is used until the transaction is committed or rolled back.
|
||||
func (c *Client) Tx(ctx context.Context) (*Tx, error) { |
||||
if _, ok := c.driver.(*txDriver); ok { |
||||
return nil, fmt.Errorf("ent: cannot start a transaction within a transaction") |
||||
} |
||||
tx, err := newTx(ctx, c.driver) |
||||
if err != nil { |
||||
return nil, fmt.Errorf("ent: starting a transaction: %w", err) |
||||
} |
||||
cfg := c.config |
||||
cfg.driver = tx |
||||
return &Tx{ |
||||
ctx: ctx, |
||||
config: cfg, |
||||
Card: NewCardClient(cfg), |
||||
User: NewUserClient(cfg), |
||||
}, nil |
||||
} |
||||
|
||||
// BeginTx returns a transactional client with specified options.
|
||||
func (c *Client) BeginTx(ctx context.Context, opts *sql.TxOptions) (*Tx, error) { |
||||
if _, ok := c.driver.(*txDriver); ok { |
||||
return nil, fmt.Errorf("ent: cannot start a transaction within a transaction") |
||||
} |
||||
tx, err := c.driver.(interface { |
||||
BeginTx(context.Context, *sql.TxOptions) (dialect.Tx, error) |
||||
}).BeginTx(ctx, opts) |
||||
if err != nil { |
||||
return nil, fmt.Errorf("ent: starting a transaction: %w", err) |
||||
} |
||||
cfg := c.config |
||||
cfg.driver = &txDriver{tx: tx, drv: c.driver} |
||||
return &Tx{ |
||||
config: cfg, |
||||
Card: NewCardClient(cfg), |
||||
User: NewUserClient(cfg), |
||||
}, nil |
||||
} |
||||
|
||||
// Debug returns a new debug-client. It's used to get verbose logging on specific operations.
|
||||
//
|
||||
// client.Debug().
|
||||
// Card.
|
||||
// Query().
|
||||
// Count(ctx)
|
||||
//
|
||||
func (c *Client) Debug() *Client { |
||||
if c.debug { |
||||
return c |
||||
} |
||||
cfg := c.config |
||||
cfg.driver = dialect.Debug(c.driver, c.log) |
||||
client := &Client{config: cfg} |
||||
client.init() |
||||
return client |
||||
} |
||||
|
||||
// Close closes the database connection and prevents new queries from starting.
|
||||
func (c *Client) Close() error { |
||||
return c.driver.Close() |
||||
} |
||||
|
||||
// Use adds the mutation hooks to all the entity clients.
|
||||
// In order to add hooks to a specific client, call: `client.Node.Use(...)`.
|
||||
func (c *Client) Use(hooks ...Hook) { |
||||
c.Card.Use(hooks...) |
||||
c.User.Use(hooks...) |
||||
} |
||||
|
||||
// CardClient is a client for the Card schema.
|
||||
type CardClient struct { |
||||
config |
||||
} |
||||
|
||||
// NewCardClient returns a client for the Card from the given config.
|
||||
func NewCardClient(c config) *CardClient { |
||||
return &CardClient{config: c} |
||||
} |
||||
|
||||
// Use adds a list of mutation hooks to the hooks stack.
|
||||
// A call to `Use(f, g, h)` equals to `card.Hooks(f(g(h())))`.
|
||||
func (c *CardClient) Use(hooks ...Hook) { |
||||
c.hooks.Card = append(c.hooks.Card, hooks...) |
||||
} |
||||
|
||||
// Create returns a create builder for Card.
|
||||
func (c *CardClient) Create() *CardCreate { |
||||
mutation := newCardMutation(c.config, OpCreate) |
||||
return &CardCreate{config: c.config, hooks: c.Hooks(), mutation: mutation} |
||||
} |
||||
|
||||
// CreateBulk returns a builder for creating a bulk of Card entities.
|
||||
func (c *CardClient) CreateBulk(builders ...*CardCreate) *CardCreateBulk { |
||||
return &CardCreateBulk{config: c.config, builders: builders} |
||||
} |
||||
|
||||
// Update returns an update builder for Card.
|
||||
func (c *CardClient) Update() *CardUpdate { |
||||
mutation := newCardMutation(c.config, OpUpdate) |
||||
return &CardUpdate{config: c.config, hooks: c.Hooks(), mutation: mutation} |
||||
} |
||||
|
||||
// UpdateOne returns an update builder for the given entity.
|
||||
func (c *CardClient) UpdateOne(ca *Card) *CardUpdateOne { |
||||
mutation := newCardMutation(c.config, OpUpdateOne, withCard(ca)) |
||||
return &CardUpdateOne{config: c.config, hooks: c.Hooks(), mutation: mutation} |
||||
} |
||||
|
||||
// UpdateOneID returns an update builder for the given id.
|
||||
func (c *CardClient) UpdateOneID(id int) *CardUpdateOne { |
||||
mutation := newCardMutation(c.config, OpUpdateOne, withCardID(id)) |
||||
return &CardUpdateOne{config: c.config, hooks: c.Hooks(), mutation: mutation} |
||||
} |
||||
|
||||
// Delete returns a delete builder for Card.
|
||||
func (c *CardClient) Delete() *CardDelete { |
||||
mutation := newCardMutation(c.config, OpDelete) |
||||
return &CardDelete{config: c.config, hooks: c.Hooks(), mutation: mutation} |
||||
} |
||||
|
||||
// DeleteOne returns a delete builder for the given entity.
|
||||
func (c *CardClient) DeleteOne(ca *Card) *CardDeleteOne { |
||||
return c.DeleteOneID(ca.ID) |
||||
} |
||||
|
||||
// DeleteOneID returns a delete builder for the given id.
|
||||
func (c *CardClient) DeleteOneID(id int) *CardDeleteOne { |
||||
builder := c.Delete().Where(card.ID(id)) |
||||
builder.mutation.id = &id |
||||
builder.mutation.op = OpDeleteOne |
||||
return &CardDeleteOne{builder} |
||||
} |
||||
|
||||
// Query returns a query builder for Card.
|
||||
func (c *CardClient) Query() *CardQuery { |
||||
return &CardQuery{ |
||||
config: c.config, |
||||
} |
||||
} |
||||
|
||||
// Get returns a Card entity by its id.
|
||||
func (c *CardClient) Get(ctx context.Context, id int) (*Card, error) { |
||||
return c.Query().Where(card.ID(id)).Only(ctx) |
||||
} |
||||
|
||||
// GetX is like Get, but panics if an error occurs.
|
||||
func (c *CardClient) GetX(ctx context.Context, id int) *Card { |
||||
obj, err := c.Get(ctx, id) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return obj |
||||
} |
||||
|
||||
// Hooks returns the client hooks.
|
||||
func (c *CardClient) Hooks() []Hook { |
||||
return c.hooks.Card |
||||
} |
||||
|
||||
// UserClient is a client for the User schema.
|
||||
type UserClient struct { |
||||
config |
||||
} |
||||
|
||||
// NewUserClient returns a client for the User from the given config.
|
||||
func NewUserClient(c config) *UserClient { |
||||
return &UserClient{config: c} |
||||
} |
||||
|
||||
// Use adds a list of mutation hooks to the hooks stack.
|
||||
// A call to `Use(f, g, h)` equals to `user.Hooks(f(g(h())))`.
|
||||
func (c *UserClient) Use(hooks ...Hook) { |
||||
c.hooks.User = append(c.hooks.User, hooks...) |
||||
} |
||||
|
||||
// Create returns a create builder for User.
|
||||
func (c *UserClient) Create() *UserCreate { |
||||
mutation := newUserMutation(c.config, OpCreate) |
||||
return &UserCreate{config: c.config, hooks: c.Hooks(), mutation: mutation} |
||||
} |
||||
|
||||
// CreateBulk returns a builder for creating a bulk of User entities.
|
||||
func (c *UserClient) CreateBulk(builders ...*UserCreate) *UserCreateBulk { |
||||
return &UserCreateBulk{config: c.config, builders: builders} |
||||
} |
||||
|
||||
// Update returns an update builder for User.
|
||||
func (c *UserClient) Update() *UserUpdate { |
||||
mutation := newUserMutation(c.config, OpUpdate) |
||||
return &UserUpdate{config: c.config, hooks: c.Hooks(), mutation: mutation} |
||||
} |
||||
|
||||
// UpdateOne returns an update builder for the given entity.
|
||||
func (c *UserClient) UpdateOne(u *User) *UserUpdateOne { |
||||
mutation := newUserMutation(c.config, OpUpdateOne, withUser(u)) |
||||
return &UserUpdateOne{config: c.config, hooks: c.Hooks(), mutation: mutation} |
||||
} |
||||
|
||||
// UpdateOneID returns an update builder for the given id.
|
||||
func (c *UserClient) UpdateOneID(id int) *UserUpdateOne { |
||||
mutation := newUserMutation(c.config, OpUpdateOne, withUserID(id)) |
||||
return &UserUpdateOne{config: c.config, hooks: c.Hooks(), mutation: mutation} |
||||
} |
||||
|
||||
// Delete returns a delete builder for User.
|
||||
func (c *UserClient) Delete() *UserDelete { |
||||
mutation := newUserMutation(c.config, OpDelete) |
||||
return &UserDelete{config: c.config, hooks: c.Hooks(), mutation: mutation} |
||||
} |
||||
|
||||
// DeleteOne returns a delete builder for the given entity.
|
||||
func (c *UserClient) DeleteOne(u *User) *UserDeleteOne { |
||||
return c.DeleteOneID(u.ID) |
||||
} |
||||
|
||||
// DeleteOneID returns a delete builder for the given id.
|
||||
func (c *UserClient) DeleteOneID(id int) *UserDeleteOne { |
||||
builder := c.Delete().Where(user.ID(id)) |
||||
builder.mutation.id = &id |
||||
builder.mutation.op = OpDeleteOne |
||||
return &UserDeleteOne{builder} |
||||
} |
||||
|
||||
// Query returns a query builder for User.
|
||||
func (c *UserClient) Query() *UserQuery { |
||||
return &UserQuery{ |
||||
config: c.config, |
||||
} |
||||
} |
||||
|
||||
// Get returns a User entity by its id.
|
||||
func (c *UserClient) Get(ctx context.Context, id int) (*User, error) { |
||||
return c.Query().Where(user.ID(id)).Only(ctx) |
||||
} |
||||
|
||||
// GetX is like Get, but panics if an error occurs.
|
||||
func (c *UserClient) GetX(ctx context.Context, id int) *User { |
||||
obj, err := c.Get(ctx, id) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return obj |
||||
} |
||||
|
||||
// Hooks returns the client hooks.
|
||||
func (c *UserClient) Hooks() []Hook { |
||||
return c.hooks.User |
||||
} |
@ -0,0 +1,60 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package ent |
||||
|
||||
import ( |
||||
"entgo.io/ent" |
||||
"entgo.io/ent/dialect" |
||||
) |
||||
|
||||
// Option function to configure the client.
|
||||
type Option func(*config) |
||||
|
||||
// Config is the configuration for the client and its builder.
|
||||
type config struct { |
||||
// driver used for executing database requests.
|
||||
driver dialect.Driver |
||||
// debug enable a debug logging.
|
||||
debug bool |
||||
// log used for logging on debug mode.
|
||||
log func(...interface{}) |
||||
// hooks to execute on mutations.
|
||||
hooks *hooks |
||||
} |
||||
|
||||
// hooks per client, for fast access.
|
||||
type hooks struct { |
||||
Card []ent.Hook |
||||
User []ent.Hook |
||||
} |
||||
|
||||
// Options applies the options on the config object.
|
||||
func (c *config) options(opts ...Option) { |
||||
for _, opt := range opts { |
||||
opt(c) |
||||
} |
||||
if c.debug { |
||||
c.driver = dialect.Debug(c.driver, c.log) |
||||
} |
||||
} |
||||
|
||||
// Debug enables debug logging on the ent.Driver.
|
||||
func Debug() Option { |
||||
return func(c *config) { |
||||
c.debug = true |
||||
} |
||||
} |
||||
|
||||
// Log sets the logging function for debug mode.
|
||||
func Log(fn func(...interface{})) Option { |
||||
return func(c *config) { |
||||
c.log = fn |
||||
} |
||||
} |
||||
|
||||
// Driver configures the client driver.
|
||||
func Driver(driver dialect.Driver) Option { |
||||
return func(c *config) { |
||||
c.driver = driver |
||||
} |
||||
} |
@ -0,0 +1,33 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package ent |
||||
|
||||
import ( |
||||
"context" |
||||
) |
||||
|
||||
type clientCtxKey struct{} |
||||
|
||||
// FromContext returns a Client stored inside a context, or nil if there isn't one.
|
||||
func FromContext(ctx context.Context) *Client { |
||||
c, _ := ctx.Value(clientCtxKey{}).(*Client) |
||||
return c |
||||
} |
||||
|
||||
// NewContext returns a new context with the given Client attached.
|
||||
func NewContext(parent context.Context, c *Client) context.Context { |
||||
return context.WithValue(parent, clientCtxKey{}, c) |
||||
} |
||||
|
||||
type txCtxKey struct{} |
||||
|
||||
// TxFromContext returns a Tx stored inside a context, or nil if there isn't one.
|
||||
func TxFromContext(ctx context.Context) *Tx { |
||||
tx, _ := ctx.Value(txCtxKey{}).(*Tx) |
||||
return tx |
||||
} |
||||
|
||||
// NewTxContext returns a new context with the given Tx attached.
|
||||
func NewTxContext(parent context.Context, tx *Tx) context.Context { |
||||
return context.WithValue(parent, txCtxKey{}, tx) |
||||
} |
@ -0,0 +1,261 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package ent |
||||
|
||||
import ( |
||||
"errors" |
||||
"fmt" |
||||
|
||||
"entgo.io/ent" |
||||
"entgo.io/ent/dialect/sql" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/card" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/user" |
||||
) |
||||
|
||||
// ent aliases to avoid import conflicts in user's code.
|
||||
type ( |
||||
Op = ent.Op |
||||
Hook = ent.Hook |
||||
Value = ent.Value |
||||
Query = ent.Query |
||||
Policy = ent.Policy |
||||
Mutator = ent.Mutator |
||||
Mutation = ent.Mutation |
||||
MutateFunc = ent.MutateFunc |
||||
) |
||||
|
||||
// OrderFunc applies an ordering on the sql selector.
|
||||
type OrderFunc func(*sql.Selector) |
||||
|
||||
// columnChecker returns a function indicates if the column exists in the given column.
|
||||
func columnChecker(table string) func(string) error { |
||||
checks := map[string]func(string) bool{ |
||||
card.Table: card.ValidColumn, |
||||
user.Table: user.ValidColumn, |
||||
} |
||||
check, ok := checks[table] |
||||
if !ok { |
||||
return func(string) error { |
||||
return fmt.Errorf("unknown table %q", table) |
||||
} |
||||
} |
||||
return func(column string) error { |
||||
if !check(column) { |
||||
return fmt.Errorf("unknown column %q for table %q", column, table) |
||||
} |
||||
return nil |
||||
} |
||||
} |
||||
|
||||
// Asc applies the given fields in ASC order.
|
||||
func Asc(fields ...string) OrderFunc { |
||||
return func(s *sql.Selector) { |
||||
check := columnChecker(s.TableName()) |
||||
for _, f := range fields { |
||||
if err := check(f); err != nil { |
||||
s.AddError(&ValidationError{Name: f, err: fmt.Errorf("ent: %w", err)}) |
||||
} |
||||
s.OrderBy(sql.Asc(s.C(f))) |
||||
} |
||||
} |
||||
} |
||||
|
||||
// Desc applies the given fields in DESC order.
|
||||
func Desc(fields ...string) OrderFunc { |
||||
return func(s *sql.Selector) { |
||||
check := columnChecker(s.TableName()) |
||||
for _, f := range fields { |
||||
if err := check(f); err != nil { |
||||
s.AddError(&ValidationError{Name: f, err: fmt.Errorf("ent: %w", err)}) |
||||
} |
||||
s.OrderBy(sql.Desc(s.C(f))) |
||||
} |
||||
} |
||||
} |
||||
|
||||
// AggregateFunc applies an aggregation step on the group-by traversal/selector.
|
||||
type AggregateFunc func(*sql.Selector) string |
||||
|
||||
// As is a pseudo aggregation function for renaming another other functions with custom names. For example:
|
||||
//
|
||||
// GroupBy(field1, field2).
|
||||
// Aggregate(ent.As(ent.Sum(field1), "sum_field1"), (ent.As(ent.Sum(field2), "sum_field2")).
|
||||
// Scan(ctx, &v)
|
||||
//
|
||||
func As(fn AggregateFunc, end string) AggregateFunc { |
||||
return func(s *sql.Selector) string { |
||||
return sql.As(fn(s), end) |
||||
} |
||||
} |
||||
|
||||
// Count applies the "count" aggregation function on each group.
|
||||
func Count() AggregateFunc { |
||||
return func(s *sql.Selector) string { |
||||
return sql.Count("*") |
||||
} |
||||
} |
||||
|
||||
// Max applies the "max" aggregation function on the given field of each group.
|
||||
func Max(field string) AggregateFunc { |
||||
return func(s *sql.Selector) string { |
||||
check := columnChecker(s.TableName()) |
||||
if err := check(field); err != nil { |
||||
s.AddError(&ValidationError{Name: field, err: fmt.Errorf("ent: %w", err)}) |
||||
return "" |
||||
} |
||||
return sql.Max(s.C(field)) |
||||
} |
||||
} |
||||
|
||||
// Mean applies the "mean" aggregation function on the given field of each group.
|
||||
func Mean(field string) AggregateFunc { |
||||
return func(s *sql.Selector) string { |
||||
check := columnChecker(s.TableName()) |
||||
if err := check(field); err != nil { |
||||
s.AddError(&ValidationError{Name: field, err: fmt.Errorf("ent: %w", err)}) |
||||
return "" |
||||
} |
||||
return sql.Avg(s.C(field)) |
||||
} |
||||
} |
||||
|
||||
// Min applies the "min" aggregation function on the given field of each group.
|
||||
func Min(field string) AggregateFunc { |
||||
return func(s *sql.Selector) string { |
||||
check := columnChecker(s.TableName()) |
||||
if err := check(field); err != nil { |
||||
s.AddError(&ValidationError{Name: field, err: fmt.Errorf("ent: %w", err)}) |
||||
return "" |
||||
} |
||||
return sql.Min(s.C(field)) |
||||
} |
||||
} |
||||
|
||||
// Sum applies the "sum" aggregation function on the given field of each group.
|
||||
func Sum(field string) AggregateFunc { |
||||
return func(s *sql.Selector) string { |
||||
check := columnChecker(s.TableName()) |
||||
if err := check(field); err != nil { |
||||
s.AddError(&ValidationError{Name: field, err: fmt.Errorf("ent: %w", err)}) |
||||
return "" |
||||
} |
||||
return sql.Sum(s.C(field)) |
||||
} |
||||
} |
||||
|
||||
// ValidationError returns when validating a field fails.
|
||||
type ValidationError struct { |
||||
Name string // Field or edge name.
|
||||
err error |
||||
} |
||||
|
||||
// Error implements the error interface.
|
||||
func (e *ValidationError) Error() string { |
||||
return e.err.Error() |
||||
} |
||||
|
||||
// Unwrap implements the errors.Wrapper interface.
|
||||
func (e *ValidationError) Unwrap() error { |
||||
return e.err |
||||
} |
||||
|
||||
// IsValidationError returns a boolean indicating whether the error is a validation error.
|
||||
func IsValidationError(err error) bool { |
||||
if err == nil { |
||||
return false |
||||
} |
||||
var e *ValidationError |
||||
return errors.As(err, &e) |
||||
} |
||||
|
||||
// NotFoundError returns when trying to fetch a specific entity and it was not found in the database.
|
||||
type NotFoundError struct { |
||||
label string |
||||
} |
||||
|
||||
// Error implements the error interface.
|
||||
func (e *NotFoundError) Error() string { |
||||
return "ent: " + e.label + " not found" |
||||
} |
||||
|
||||
// IsNotFound returns a boolean indicating whether the error is a not found error.
|
||||
func IsNotFound(err error) bool { |
||||
if err == nil { |
||||
return false |
||||
} |
||||
var e *NotFoundError |
||||
return errors.As(err, &e) |
||||
} |
||||
|
||||
// MaskNotFound masks not found error.
|
||||
func MaskNotFound(err error) error { |
||||
if IsNotFound(err) { |
||||
return nil |
||||
} |
||||
return err |
||||
} |
||||
|
||||
// NotSingularError returns when trying to fetch a singular entity and more then one was found in the database.
|
||||
type NotSingularError struct { |
||||
label string |
||||
} |
||||
|
||||
// Error implements the error interface.
|
||||
func (e *NotSingularError) Error() string { |
||||
return "ent: " + e.label + " not singular" |
||||
} |
||||
|
||||
// IsNotSingular returns a boolean indicating whether the error is a not singular error.
|
||||
func IsNotSingular(err error) bool { |
||||
if err == nil { |
||||
return false |
||||
} |
||||
var e *NotSingularError |
||||
return errors.As(err, &e) |
||||
} |
||||
|
||||
// NotLoadedError returns when trying to get a node that was not loaded by the query.
|
||||
type NotLoadedError struct { |
||||
edge string |
||||
} |
||||
|
||||
// Error implements the error interface.
|
||||
func (e *NotLoadedError) Error() string { |
||||
return "ent: " + e.edge + " edge was not loaded" |
||||
} |
||||
|
||||
// IsNotLoaded returns a boolean indicating whether the error is a not loaded error.
|
||||
func IsNotLoaded(err error) bool { |
||||
if err == nil { |
||||
return false |
||||
} |
||||
var e *NotLoadedError |
||||
return errors.As(err, &e) |
||||
} |
||||
|
||||
// ConstraintError returns when trying to create/update one or more entities and
|
||||
// one or more of their constraints failed. For example, violation of edge or
|
||||
// field uniqueness.
|
||||
type ConstraintError struct { |
||||
msg string |
||||
wrap error |
||||
} |
||||
|
||||
// Error implements the error interface.
|
||||
func (e ConstraintError) Error() string { |
||||
return "ent: constraint failed: " + e.msg |
||||
} |
||||
|
||||
// Unwrap implements the errors.Wrapper interface.
|
||||
func (e *ConstraintError) Unwrap() error { |
||||
return e.wrap |
||||
} |
||||
|
||||
// IsConstraintError returns a boolean indicating whether the error is a constraint failure.
|
||||
func IsConstraintError(err error) bool { |
||||
if err == nil { |
||||
return false |
||||
} |
||||
var e *ConstraintError |
||||
return errors.As(err, &e) |
||||
} |
@ -0,0 +1,78 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package enttest |
||||
|
||||
import ( |
||||
"context" |
||||
|
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent" |
||||
// required by schema hooks.
|
||||
_ "github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/runtime" |
||||
|
||||
"entgo.io/ent/dialect/sql/schema" |
||||
) |
||||
|
||||
type ( |
||||
// TestingT is the interface that is shared between
|
||||
// testing.T and testing.B and used by enttest.
|
||||
TestingT interface { |
||||
FailNow() |
||||
Error(...interface{}) |
||||
} |
||||
|
||||
// Option configures client creation.
|
||||
Option func(*options) |
||||
|
||||
options struct { |
||||
opts []ent.Option |
||||
migrateOpts []schema.MigrateOption |
||||
} |
||||
) |
||||
|
||||
// WithOptions forwards options to client creation.
|
||||
func WithOptions(opts ...ent.Option) Option { |
||||
return func(o *options) { |
||||
o.opts = append(o.opts, opts...) |
||||
} |
||||
} |
||||
|
||||
// WithMigrateOptions forwards options to auto migration.
|
||||
func WithMigrateOptions(opts ...schema.MigrateOption) Option { |
||||
return func(o *options) { |
||||
o.migrateOpts = append(o.migrateOpts, opts...) |
||||
} |
||||
} |
||||
|
||||
func newOptions(opts []Option) *options { |
||||
o := &options{} |
||||
for _, opt := range opts { |
||||
opt(o) |
||||
} |
||||
return o |
||||
} |
||||
|
||||
// Open calls ent.Open and auto-run migration.
|
||||
func Open(t TestingT, driverName, dataSourceName string, opts ...Option) *ent.Client { |
||||
o := newOptions(opts) |
||||
c, err := ent.Open(driverName, dataSourceName, o.opts...) |
||||
if err != nil { |
||||
t.Error(err) |
||||
t.FailNow() |
||||
} |
||||
if err := c.Schema.Create(context.Background(), o.migrateOpts...); err != nil { |
||||
t.Error(err) |
||||
t.FailNow() |
||||
} |
||||
return c |
||||
} |
||||
|
||||
// NewClient calls ent.NewClient and auto-run migration.
|
||||
func NewClient(t TestingT, opts ...Option) *ent.Client { |
||||
o := newOptions(opts) |
||||
c := ent.NewClient(o.opts...) |
||||
if err := c.Schema.Create(context.Background(), o.migrateOpts...); err != nil { |
||||
t.Error(err) |
||||
t.FailNow() |
||||
} |
||||
return c |
||||
} |
@ -0,0 +1,3 @@ |
||||
package ent |
||||
|
||||
//go:generate go run -mod=mod entgo.io/ent/cmd/ent generate ./schema
|
@ -0,0 +1,217 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package hook |
||||
|
||||
import ( |
||||
"context" |
||||
"fmt" |
||||
|
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent" |
||||
) |
||||
|
||||
// The CardFunc type is an adapter to allow the use of ordinary
|
||||
// function as Card mutator.
|
||||
type CardFunc func(context.Context, *ent.CardMutation) (ent.Value, error) |
||||
|
||||
// Mutate calls f(ctx, m).
|
||||
func (f CardFunc) Mutate(ctx context.Context, m ent.Mutation) (ent.Value, error) { |
||||
mv, ok := m.(*ent.CardMutation) |
||||
if !ok { |
||||
return nil, fmt.Errorf("unexpected mutation type %T. expect *ent.CardMutation", m) |
||||
} |
||||
return f(ctx, mv) |
||||
} |
||||
|
||||
// The UserFunc type is an adapter to allow the use of ordinary
|
||||
// function as User mutator.
|
||||
type UserFunc func(context.Context, *ent.UserMutation) (ent.Value, error) |
||||
|
||||
// Mutate calls f(ctx, m).
|
||||
func (f UserFunc) Mutate(ctx context.Context, m ent.Mutation) (ent.Value, error) { |
||||
mv, ok := m.(*ent.UserMutation) |
||||
if !ok { |
||||
return nil, fmt.Errorf("unexpected mutation type %T. expect *ent.UserMutation", m) |
||||
} |
||||
return f(ctx, mv) |
||||
} |
||||
|
||||
// Condition is a hook condition function.
|
||||
type Condition func(context.Context, ent.Mutation) bool |
||||
|
||||
// And groups conditions with the AND operator.
|
||||
func And(first, second Condition, rest ...Condition) Condition { |
||||
return func(ctx context.Context, m ent.Mutation) bool { |
||||
if !first(ctx, m) || !second(ctx, m) { |
||||
return false |
||||
} |
||||
for _, cond := range rest { |
||||
if !cond(ctx, m) { |
||||
return false |
||||
} |
||||
} |
||||
return true |
||||
} |
||||
} |
||||
|
||||
// Or groups conditions with the OR operator.
|
||||
func Or(first, second Condition, rest ...Condition) Condition { |
||||
return func(ctx context.Context, m ent.Mutation) bool { |
||||
if first(ctx, m) || second(ctx, m) { |
||||
return true |
||||
} |
||||
for _, cond := range rest { |
||||
if cond(ctx, m) { |
||||
return true |
||||
} |
||||
} |
||||
return false |
||||
} |
||||
} |
||||
|
||||
// Not negates a given condition.
|
||||
func Not(cond Condition) Condition { |
||||
return func(ctx context.Context, m ent.Mutation) bool { |
||||
return !cond(ctx, m) |
||||
} |
||||
} |
||||
|
||||
// HasOp is a condition testing mutation operation.
|
||||
func HasOp(op ent.Op) Condition { |
||||
return func(_ context.Context, m ent.Mutation) bool { |
||||
return m.Op().Is(op) |
||||
} |
||||
} |
||||
|
||||
// HasAddedFields is a condition validating `.AddedField` on fields.
|
||||
func HasAddedFields(field string, fields ...string) Condition { |
||||
return func(_ context.Context, m ent.Mutation) bool { |
||||
if _, exists := m.AddedField(field); !exists { |
||||
return false |
||||
} |
||||
for _, field := range fields { |
||||
if _, exists := m.AddedField(field); !exists { |
||||
return false |
||||
} |
||||
} |
||||
return true |
||||
} |
||||
} |
||||
|
||||
// HasClearedFields is a condition validating `.FieldCleared` on fields.
|
||||
func HasClearedFields(field string, fields ...string) Condition { |
||||
return func(_ context.Context, m ent.Mutation) bool { |
||||
if exists := m.FieldCleared(field); !exists { |
||||
return false |
||||
} |
||||
for _, field := range fields { |
||||
if exists := m.FieldCleared(field); !exists { |
||||
return false |
||||
} |
||||
} |
||||
return true |
||||
} |
||||
} |
||||
|
||||
// HasFields is a condition validating `.Field` on fields.
|
||||
func HasFields(field string, fields ...string) Condition { |
||||
return func(_ context.Context, m ent.Mutation) bool { |
||||
if _, exists := m.Field(field); !exists { |
||||
return false |
||||
} |
||||
for _, field := range fields { |
||||
if _, exists := m.Field(field); !exists { |
||||
return false |
||||
} |
||||
} |
||||
return true |
||||
} |
||||
} |
||||
|
||||
// If executes the given hook under condition.
|
||||
//
|
||||
// hook.If(ComputeAverage, And(HasFields(...), HasAddedFields(...)))
|
||||
//
|
||||
func If(hk ent.Hook, cond Condition) ent.Hook { |
||||
return func(next ent.Mutator) ent.Mutator { |
||||
return ent.MutateFunc(func(ctx context.Context, m ent.Mutation) (ent.Value, error) { |
||||
if cond(ctx, m) { |
||||
return hk(next).Mutate(ctx, m) |
||||
} |
||||
return next.Mutate(ctx, m) |
||||
}) |
||||
} |
||||
} |
||||
|
||||
// On executes the given hook only for the given operation.
|
||||
//
|
||||
// hook.On(Log, ent.Delete|ent.Create)
|
||||
//
|
||||
func On(hk ent.Hook, op ent.Op) ent.Hook { |
||||
return If(hk, HasOp(op)) |
||||
} |
||||
|
||||
// Unless skips the given hook only for the given operation.
|
||||
//
|
||||
// hook.Unless(Log, ent.Update|ent.UpdateOne)
|
||||
//
|
||||
func Unless(hk ent.Hook, op ent.Op) ent.Hook { |
||||
return If(hk, Not(HasOp(op))) |
||||
} |
||||
|
||||
// FixedError is a hook returning a fixed error.
|
||||
func FixedError(err error) ent.Hook { |
||||
return func(ent.Mutator) ent.Mutator { |
||||
return ent.MutateFunc(func(context.Context, ent.Mutation) (ent.Value, error) { |
||||
return nil, err |
||||
}) |
||||
} |
||||
} |
||||
|
||||
// Reject returns a hook that rejects all operations that match op.
|
||||
//
|
||||
// func (T) Hooks() []ent.Hook {
|
||||
// return []ent.Hook{
|
||||
// Reject(ent.Delete|ent.Update),
|
||||
// }
|
||||
// }
|
||||
//
|
||||
func Reject(op ent.Op) ent.Hook { |
||||
hk := FixedError(fmt.Errorf("%s operation is not allowed", op)) |
||||
return On(hk, op) |
||||
} |
||||
|
||||
// Chain acts as a list of hooks and is effectively immutable.
|
||||
// Once created, it will always hold the same set of hooks in the same order.
|
||||
type Chain struct { |
||||
hooks []ent.Hook |
||||
} |
||||
|
||||
// NewChain creates a new chain of hooks.
|
||||
func NewChain(hooks ...ent.Hook) Chain { |
||||
return Chain{append([]ent.Hook(nil), hooks...)} |
||||
} |
||||
|
||||
// Hook chains the list of hooks and returns the final hook.
|
||||
func (c Chain) Hook() ent.Hook { |
||||
return func(mutator ent.Mutator) ent.Mutator { |
||||
for i := len(c.hooks) - 1; i >= 0; i-- { |
||||
mutator = c.hooks[i](mutator) |
||||
} |
||||
return mutator |
||||
} |
||||
} |
||||
|
||||
// Append extends a chain, adding the specified hook
|
||||
// as the last ones in the mutation flow.
|
||||
func (c Chain) Append(hooks ...ent.Hook) Chain { |
||||
newHooks := make([]ent.Hook, 0, len(c.hooks)+len(hooks)) |
||||
newHooks = append(newHooks, c.hooks...) |
||||
newHooks = append(newHooks, hooks...) |
||||
return Chain{newHooks} |
||||
} |
||||
|
||||
// Extend extends a chain, adding the specified chain
|
||||
// as the last ones in the mutation flow.
|
||||
func (c Chain) Extend(chain Chain) Chain { |
||||
return c.Append(chain.hooks...) |
||||
} |
@ -0,0 +1,72 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package migrate |
||||
|
||||
import ( |
||||
"context" |
||||
"fmt" |
||||
"io" |
||||
|
||||
"entgo.io/ent/dialect" |
||||
"entgo.io/ent/dialect/sql/schema" |
||||
) |
||||
|
||||
var ( |
||||
// WithGlobalUniqueID sets the universal ids options to the migration.
|
||||
// If this option is enabled, ent migration will allocate a 1<<32 range
|
||||
// for the ids of each entity (table).
|
||||
// Note that this option cannot be applied on tables that already exist.
|
||||
WithGlobalUniqueID = schema.WithGlobalUniqueID |
||||
// WithDropColumn sets the drop column option to the migration.
|
||||
// If this option is enabled, ent migration will drop old columns
|
||||
// that were used for both fields and edges. This defaults to false.
|
||||
WithDropColumn = schema.WithDropColumn |
||||
// WithDropIndex sets the drop index option to the migration.
|
||||
// If this option is enabled, ent migration will drop old indexes
|
||||
// that were defined in the schema. This defaults to false.
|
||||
// Note that unique constraints are defined using `UNIQUE INDEX`,
|
||||
// and therefore, it's recommended to enable this option to get more
|
||||
// flexibility in the schema changes.
|
||||
WithDropIndex = schema.WithDropIndex |
||||
// WithFixture sets the foreign-key renaming option to the migration when upgrading
|
||||
// ent from v0.1.0 (issue-#285). Defaults to false.
|
||||
WithFixture = schema.WithFixture |
||||
// WithForeignKeys enables creating foreign-key in schema DDL. This defaults to true.
|
||||
WithForeignKeys = schema.WithForeignKeys |
||||
) |
||||
|
||||
// Schema is the API for creating, migrating and dropping a schema.
|
||||
type Schema struct { |
||||
drv dialect.Driver |
||||
universalID bool |
||||
} |
||||
|
||||
// NewSchema creates a new schema client.
|
||||
func NewSchema(drv dialect.Driver) *Schema { return &Schema{drv: drv} } |
||||
|
||||
// Create creates all schema resources.
|
||||
func (s *Schema) Create(ctx context.Context, opts ...schema.MigrateOption) error { |
||||
migrate, err := schema.NewMigrate(s.drv, opts...) |
||||
if err != nil { |
||||
return fmt.Errorf("ent/migrate: %w", err) |
||||
} |
||||
return migrate.Create(ctx, Tables...) |
||||
} |
||||
|
||||
// WriteTo writes the schema changes to w instead of running them against the database.
|
||||
//
|
||||
// if err := client.Schema.WriteTo(context.Background(), os.Stdout); err != nil {
|
||||
// log.Fatal(err)
|
||||
// }
|
||||
//
|
||||
func (s *Schema) WriteTo(ctx context.Context, w io.Writer, opts ...schema.MigrateOption) error { |
||||
drv := &schema.WriteDriver{ |
||||
Writer: w, |
||||
Driver: s.drv, |
||||
} |
||||
migrate, err := schema.NewMigrate(drv, opts...) |
||||
if err != nil { |
||||
return fmt.Errorf("ent/migrate: %w", err) |
||||
} |
||||
return migrate.Create(ctx, Tables...) |
||||
} |
@ -0,0 +1,43 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package migrate |
||||
|
||||
import ( |
||||
"entgo.io/ent/dialect/sql/schema" |
||||
"entgo.io/ent/schema/field" |
||||
) |
||||
|
||||
var ( |
||||
// CardsColumns holds the columns for the "cards" table.
|
||||
CardsColumns = []*schema.Column{ |
||||
{Name: "id", Type: field.TypeInt, Increment: true}, |
||||
{Name: "user_id", Type: field.TypeString}, |
||||
{Name: "money", Type: field.TypeString}, |
||||
} |
||||
// CardsTable holds the schema information for the "cards" table.
|
||||
CardsTable = &schema.Table{ |
||||
Name: "cards", |
||||
Columns: CardsColumns, |
||||
PrimaryKey: []*schema.Column{CardsColumns[0]}, |
||||
} |
||||
// UsersColumns holds the columns for the "users" table.
|
||||
UsersColumns = []*schema.Column{ |
||||
{Name: "id", Type: field.TypeInt, Increment: true}, |
||||
{Name: "name", Type: field.TypeString, Default: ""}, |
||||
{Name: "email", Type: field.TypeString, Default: ""}, |
||||
} |
||||
// UsersTable holds the schema information for the "users" table.
|
||||
UsersTable = &schema.Table{ |
||||
Name: "users", |
||||
Columns: UsersColumns, |
||||
PrimaryKey: []*schema.Column{UsersColumns[0]}, |
||||
} |
||||
// Tables holds all the tables in the schema.
|
||||
Tables = []*schema.Table{ |
||||
CardsTable, |
||||
UsersTable, |
||||
} |
||||
) |
||||
|
||||
func init() { |
||||
} |
@ -0,0 +1,732 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package ent |
||||
|
||||
import ( |
||||
"context" |
||||
"fmt" |
||||
"sync" |
||||
|
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/card" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/predicate" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/user" |
||||
|
||||
"entgo.io/ent" |
||||
) |
||||
|
||||
const ( |
||||
// Operation types.
|
||||
OpCreate = ent.OpCreate |
||||
OpDelete = ent.OpDelete |
||||
OpDeleteOne = ent.OpDeleteOne |
||||
OpUpdate = ent.OpUpdate |
||||
OpUpdateOne = ent.OpUpdateOne |
||||
|
||||
// Node types.
|
||||
TypeCard = "Card" |
||||
TypeUser = "User" |
||||
) |
||||
|
||||
// CardMutation represents an operation that mutates the Card nodes in the graph.
|
||||
type CardMutation struct { |
||||
config |
||||
op Op |
||||
typ string |
||||
id *int |
||||
user_id *string |
||||
money *string |
||||
clearedFields map[string]struct{} |
||||
done bool |
||||
oldValue func(context.Context) (*Card, error) |
||||
predicates []predicate.Card |
||||
} |
||||
|
||||
var _ ent.Mutation = (*CardMutation)(nil) |
||||
|
||||
// cardOption allows management of the mutation configuration using functional options.
|
||||
type cardOption func(*CardMutation) |
||||
|
||||
// newCardMutation creates new mutation for the Card entity.
|
||||
func newCardMutation(c config, op Op, opts ...cardOption) *CardMutation { |
||||
m := &CardMutation{ |
||||
config: c, |
||||
op: op, |
||||
typ: TypeCard, |
||||
clearedFields: make(map[string]struct{}), |
||||
} |
||||
for _, opt := range opts { |
||||
opt(m) |
||||
} |
||||
return m |
||||
} |
||||
|
||||
// withCardID sets the ID field of the mutation.
|
||||
func withCardID(id int) cardOption { |
||||
return func(m *CardMutation) { |
||||
var ( |
||||
err error |
||||
once sync.Once |
||||
value *Card |
||||
) |
||||
m.oldValue = func(ctx context.Context) (*Card, error) { |
||||
once.Do(func() { |
||||
if m.done { |
||||
err = fmt.Errorf("querying old values post mutation is not allowed") |
||||
} else { |
||||
value, err = m.Client().Card.Get(ctx, id) |
||||
} |
||||
}) |
||||
return value, err |
||||
} |
||||
m.id = &id |
||||
} |
||||
} |
||||
|
||||
// withCard sets the old Card of the mutation.
|
||||
func withCard(node *Card) cardOption { |
||||
return func(m *CardMutation) { |
||||
m.oldValue = func(context.Context) (*Card, error) { |
||||
return node, nil |
||||
} |
||||
m.id = &node.ID |
||||
} |
||||
} |
||||
|
||||
// Client returns a new `ent.Client` from the mutation. If the mutation was
|
||||
// executed in a transaction (ent.Tx), a transactional client is returned.
|
||||
func (m CardMutation) Client() *Client { |
||||
client := &Client{config: m.config} |
||||
client.init() |
||||
return client |
||||
} |
||||
|
||||
// Tx returns an `ent.Tx` for mutations that were executed in transactions;
|
||||
// it returns an error otherwise.
|
||||
func (m CardMutation) Tx() (*Tx, error) { |
||||
if _, ok := m.driver.(*txDriver); !ok { |
||||
return nil, fmt.Errorf("ent: mutation is not running in a transaction") |
||||
} |
||||
tx := &Tx{config: m.config} |
||||
tx.init() |
||||
return tx, nil |
||||
} |
||||
|
||||
// SetID sets the value of the id field. Note that this
|
||||
// operation is only accepted on creation of Card entities.
|
||||
func (m *CardMutation) SetID(id int) { |
||||
m.id = &id |
||||
} |
||||
|
||||
// ID returns the ID value in the mutation. Note that the ID is only available
|
||||
// if it was provided to the builder or after it was returned from the database.
|
||||
func (m *CardMutation) ID() (id int, exists bool) { |
||||
if m.id == nil { |
||||
return |
||||
} |
||||
return *m.id, true |
||||
} |
||||
|
||||
// SetUserID sets the "user_id" field.
|
||||
func (m *CardMutation) SetUserID(s string) { |
||||
m.user_id = &s |
||||
} |
||||
|
||||
// UserID returns the value of the "user_id" field in the mutation.
|
||||
func (m *CardMutation) UserID() (r string, exists bool) { |
||||
v := m.user_id |
||||
if v == nil { |
||||
return |
||||
} |
||||
return *v, true |
||||
} |
||||
|
||||
// OldUserID returns the old "user_id" field's value of the Card entity.
|
||||
// If the Card object wasn't provided to the builder, the object is fetched from the database.
|
||||
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
|
||||
func (m *CardMutation) OldUserID(ctx context.Context) (v string, err error) { |
||||
if !m.op.Is(OpUpdateOne) { |
||||
return v, fmt.Errorf("OldUserID is only allowed on UpdateOne operations") |
||||
} |
||||
if m.id == nil || m.oldValue == nil { |
||||
return v, fmt.Errorf("OldUserID requires an ID field in the mutation") |
||||
} |
||||
oldValue, err := m.oldValue(ctx) |
||||
if err != nil { |
||||
return v, fmt.Errorf("querying old value for OldUserID: %w", err) |
||||
} |
||||
return oldValue.UserID, nil |
||||
} |
||||
|
||||
// ResetUserID resets all changes to the "user_id" field.
|
||||
func (m *CardMutation) ResetUserID() { |
||||
m.user_id = nil |
||||
} |
||||
|
||||
// SetMoney sets the "money" field.
|
||||
func (m *CardMutation) SetMoney(s string) { |
||||
m.money = &s |
||||
} |
||||
|
||||
// Money returns the value of the "money" field in the mutation.
|
||||
func (m *CardMutation) Money() (r string, exists bool) { |
||||
v := m.money |
||||
if v == nil { |
||||
return |
||||
} |
||||
return *v, true |
||||
} |
||||
|
||||
// OldMoney returns the old "money" field's value of the Card entity.
|
||||
// If the Card object wasn't provided to the builder, the object is fetched from the database.
|
||||
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
|
||||
func (m *CardMutation) OldMoney(ctx context.Context) (v string, err error) { |
||||
if !m.op.Is(OpUpdateOne) { |
||||
return v, fmt.Errorf("OldMoney is only allowed on UpdateOne operations") |
||||
} |
||||
if m.id == nil || m.oldValue == nil { |
||||
return v, fmt.Errorf("OldMoney requires an ID field in the mutation") |
||||
} |
||||
oldValue, err := m.oldValue(ctx) |
||||
if err != nil { |
||||
return v, fmt.Errorf("querying old value for OldMoney: %w", err) |
||||
} |
||||
return oldValue.Money, nil |
||||
} |
||||
|
||||
// ResetMoney resets all changes to the "money" field.
|
||||
func (m *CardMutation) ResetMoney() { |
||||
m.money = nil |
||||
} |
||||
|
||||
// Where appends a list predicates to the CardMutation builder.
|
||||
func (m *CardMutation) Where(ps ...predicate.Card) { |
||||
m.predicates = append(m.predicates, ps...) |
||||
} |
||||
|
||||
// Op returns the operation name.
|
||||
func (m *CardMutation) Op() Op { |
||||
return m.op |
||||
} |
||||
|
||||
// Type returns the node type of this mutation (Card).
|
||||
func (m *CardMutation) Type() string { |
||||
return m.typ |
||||
} |
||||
|
||||
// Fields returns all fields that were changed during this mutation. Note that in
|
||||
// order to get all numeric fields that were incremented/decremented, call
|
||||
// AddedFields().
|
||||
func (m *CardMutation) Fields() []string { |
||||
fields := make([]string, 0, 2) |
||||
if m.user_id != nil { |
||||
fields = append(fields, card.FieldUserID) |
||||
} |
||||
if m.money != nil { |
||||
fields = append(fields, card.FieldMoney) |
||||
} |
||||
return fields |
||||
} |
||||
|
||||
// Field returns the value of a field with the given name. The second boolean
|
||||
// return value indicates that this field was not set, or was not defined in the
|
||||
// schema.
|
||||
func (m *CardMutation) Field(name string) (ent.Value, bool) { |
||||
switch name { |
||||
case card.FieldUserID: |
||||
return m.UserID() |
||||
case card.FieldMoney: |
||||
return m.Money() |
||||
} |
||||
return nil, false |
||||
} |
||||
|
||||
// OldField returns the old value of the field from the database. An error is
|
||||
// returned if the mutation operation is not UpdateOne, or the query to the
|
||||
// database failed.
|
||||
func (m *CardMutation) OldField(ctx context.Context, name string) (ent.Value, error) { |
||||
switch name { |
||||
case card.FieldUserID: |
||||
return m.OldUserID(ctx) |
||||
case card.FieldMoney: |
||||
return m.OldMoney(ctx) |
||||
} |
||||
return nil, fmt.Errorf("unknown Card field %s", name) |
||||
} |
||||
|
||||
// SetField sets the value of a field with the given name. It returns an error if
|
||||
// the field is not defined in the schema, or if the type mismatched the field
|
||||
// type.
|
||||
func (m *CardMutation) SetField(name string, value ent.Value) error { |
||||
switch name { |
||||
case card.FieldUserID: |
||||
v, ok := value.(string) |
||||
if !ok { |
||||
return fmt.Errorf("unexpected type %T for field %s", value, name) |
||||
} |
||||
m.SetUserID(v) |
||||
return nil |
||||
case card.FieldMoney: |
||||
v, ok := value.(string) |
||||
if !ok { |
||||
return fmt.Errorf("unexpected type %T for field %s", value, name) |
||||
} |
||||
m.SetMoney(v) |
||||
return nil |
||||
} |
||||
return fmt.Errorf("unknown Card field %s", name) |
||||
} |
||||
|
||||
// AddedFields returns all numeric fields that were incremented/decremented during
|
||||
// this mutation.
|
||||
func (m *CardMutation) AddedFields() []string { |
||||
return nil |
||||
} |
||||
|
||||
// AddedField returns the numeric value that was incremented/decremented on a field
|
||||
// with the given name. The second boolean return value indicates that this field
|
||||
// was not set, or was not defined in the schema.
|
||||
func (m *CardMutation) AddedField(name string) (ent.Value, bool) { |
||||
return nil, false |
||||
} |
||||
|
||||
// AddField adds the value to the field with the given name. It returns an error if
|
||||
// the field is not defined in the schema, or if the type mismatched the field
|
||||
// type.
|
||||
func (m *CardMutation) AddField(name string, value ent.Value) error { |
||||
switch name { |
||||
} |
||||
return fmt.Errorf("unknown Card numeric field %s", name) |
||||
} |
||||
|
||||
// ClearedFields returns all nullable fields that were cleared during this
|
||||
// mutation.
|
||||
func (m *CardMutation) ClearedFields() []string { |
||||
return nil |
||||
} |
||||
|
||||
// FieldCleared returns a boolean indicating if a field with the given name was
|
||||
// cleared in this mutation.
|
||||
func (m *CardMutation) FieldCleared(name string) bool { |
||||
_, ok := m.clearedFields[name] |
||||
return ok |
||||
} |
||||
|
||||
// ClearField clears the value of the field with the given name. It returns an
|
||||
// error if the field is not defined in the schema.
|
||||
func (m *CardMutation) ClearField(name string) error { |
||||
return fmt.Errorf("unknown Card nullable field %s", name) |
||||
} |
||||
|
||||
// ResetField resets all changes in the mutation for the field with the given name.
|
||||
// It returns an error if the field is not defined in the schema.
|
||||
func (m *CardMutation) ResetField(name string) error { |
||||
switch name { |
||||
case card.FieldUserID: |
||||
m.ResetUserID() |
||||
return nil |
||||
case card.FieldMoney: |
||||
m.ResetMoney() |
||||
return nil |
||||
} |
||||
return fmt.Errorf("unknown Card field %s", name) |
||||
} |
||||
|
||||
// AddedEdges returns all edge names that were set/added in this mutation.
|
||||
func (m *CardMutation) AddedEdges() []string { |
||||
edges := make([]string, 0, 0) |
||||
return edges |
||||
} |
||||
|
||||
// AddedIDs returns all IDs (to other nodes) that were added for the given edge
|
||||
// name in this mutation.
|
||||
func (m *CardMutation) AddedIDs(name string) []ent.Value { |
||||
return nil |
||||
} |
||||
|
||||
// RemovedEdges returns all edge names that were removed in this mutation.
|
||||
func (m *CardMutation) RemovedEdges() []string { |
||||
edges := make([]string, 0, 0) |
||||
return edges |
||||
} |
||||
|
||||
// RemovedIDs returns all IDs (to other nodes) that were removed for the edge with
|
||||
// the given name in this mutation.
|
||||
func (m *CardMutation) RemovedIDs(name string) []ent.Value { |
||||
return nil |
||||
} |
||||
|
||||
// ClearedEdges returns all edge names that were cleared in this mutation.
|
||||
func (m *CardMutation) ClearedEdges() []string { |
||||
edges := make([]string, 0, 0) |
||||
return edges |
||||
} |
||||
|
||||
// EdgeCleared returns a boolean which indicates if the edge with the given name
|
||||
// was cleared in this mutation.
|
||||
func (m *CardMutation) EdgeCleared(name string) bool { |
||||
return false |
||||
} |
||||
|
||||
// ClearEdge clears the value of the edge with the given name. It returns an error
|
||||
// if that edge is not defined in the schema.
|
||||
func (m *CardMutation) ClearEdge(name string) error { |
||||
return fmt.Errorf("unknown Card unique edge %s", name) |
||||
} |
||||
|
||||
// ResetEdge resets all changes to the edge with the given name in this mutation.
|
||||
// It returns an error if the edge is not defined in the schema.
|
||||
func (m *CardMutation) ResetEdge(name string) error { |
||||
return fmt.Errorf("unknown Card edge %s", name) |
||||
} |
||||
|
||||
// UserMutation represents an operation that mutates the User nodes in the graph.
|
||||
type UserMutation struct { |
||||
config |
||||
op Op |
||||
typ string |
||||
id *int |
||||
name *string |
||||
email *string |
||||
clearedFields map[string]struct{} |
||||
done bool |
||||
oldValue func(context.Context) (*User, error) |
||||
predicates []predicate.User |
||||
} |
||||
|
||||
var _ ent.Mutation = (*UserMutation)(nil) |
||||
|
||||
// userOption allows management of the mutation configuration using functional options.
|
||||
type userOption func(*UserMutation) |
||||
|
||||
// newUserMutation creates new mutation for the User entity.
|
||||
func newUserMutation(c config, op Op, opts ...userOption) *UserMutation { |
||||
m := &UserMutation{ |
||||
config: c, |
||||
op: op, |
||||
typ: TypeUser, |
||||
clearedFields: make(map[string]struct{}), |
||||
} |
||||
for _, opt := range opts { |
||||
opt(m) |
||||
} |
||||
return m |
||||
} |
||||
|
||||
// withUserID sets the ID field of the mutation.
|
||||
func withUserID(id int) userOption { |
||||
return func(m *UserMutation) { |
||||
var ( |
||||
err error |
||||
once sync.Once |
||||
value *User |
||||
) |
||||
m.oldValue = func(ctx context.Context) (*User, error) { |
||||
once.Do(func() { |
||||
if m.done { |
||||
err = fmt.Errorf("querying old values post mutation is not allowed") |
||||
} else { |
||||
value, err = m.Client().User.Get(ctx, id) |
||||
} |
||||
}) |
||||
return value, err |
||||
} |
||||
m.id = &id |
||||
} |
||||
} |
||||
|
||||
// withUser sets the old User of the mutation.
|
||||
func withUser(node *User) userOption { |
||||
return func(m *UserMutation) { |
||||
m.oldValue = func(context.Context) (*User, error) { |
||||
return node, nil |
||||
} |
||||
m.id = &node.ID |
||||
} |
||||
} |
||||
|
||||
// Client returns a new `ent.Client` from the mutation. If the mutation was
|
||||
// executed in a transaction (ent.Tx), a transactional client is returned.
|
||||
func (m UserMutation) Client() *Client { |
||||
client := &Client{config: m.config} |
||||
client.init() |
||||
return client |
||||
} |
||||
|
||||
// Tx returns an `ent.Tx` for mutations that were executed in transactions;
|
||||
// it returns an error otherwise.
|
||||
func (m UserMutation) Tx() (*Tx, error) { |
||||
if _, ok := m.driver.(*txDriver); !ok { |
||||
return nil, fmt.Errorf("ent: mutation is not running in a transaction") |
||||
} |
||||
tx := &Tx{config: m.config} |
||||
tx.init() |
||||
return tx, nil |
||||
} |
||||
|
||||
// SetID sets the value of the id field. Note that this
|
||||
// operation is only accepted on creation of User entities.
|
||||
func (m *UserMutation) SetID(id int) { |
||||
m.id = &id |
||||
} |
||||
|
||||
// ID returns the ID value in the mutation. Note that the ID is only available
|
||||
// if it was provided to the builder or after it was returned from the database.
|
||||
func (m *UserMutation) ID() (id int, exists bool) { |
||||
if m.id == nil { |
||||
return |
||||
} |
||||
return *m.id, true |
||||
} |
||||
|
||||
// SetName sets the "name" field.
|
||||
func (m *UserMutation) SetName(s string) { |
||||
m.name = &s |
||||
} |
||||
|
||||
// Name returns the value of the "name" field in the mutation.
|
||||
func (m *UserMutation) Name() (r string, exists bool) { |
||||
v := m.name |
||||
if v == nil { |
||||
return |
||||
} |
||||
return *v, true |
||||
} |
||||
|
||||
// OldName returns the old "name" field's value of the User entity.
|
||||
// If the User object wasn't provided to the builder, the object is fetched from the database.
|
||||
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
|
||||
func (m *UserMutation) OldName(ctx context.Context) (v string, err error) { |
||||
if !m.op.Is(OpUpdateOne) { |
||||
return v, fmt.Errorf("OldName is only allowed on UpdateOne operations") |
||||
} |
||||
if m.id == nil || m.oldValue == nil { |
||||
return v, fmt.Errorf("OldName requires an ID field in the mutation") |
||||
} |
||||
oldValue, err := m.oldValue(ctx) |
||||
if err != nil { |
||||
return v, fmt.Errorf("querying old value for OldName: %w", err) |
||||
} |
||||
return oldValue.Name, nil |
||||
} |
||||
|
||||
// ResetName resets all changes to the "name" field.
|
||||
func (m *UserMutation) ResetName() { |
||||
m.name = nil |
||||
} |
||||
|
||||
// SetEmail sets the "email" field.
|
||||
func (m *UserMutation) SetEmail(s string) { |
||||
m.email = &s |
||||
} |
||||
|
||||
// Email returns the value of the "email" field in the mutation.
|
||||
func (m *UserMutation) Email() (r string, exists bool) { |
||||
v := m.email |
||||
if v == nil { |
||||
return |
||||
} |
||||
return *v, true |
||||
} |
||||
|
||||
// OldEmail returns the old "email" field's value of the User entity.
|
||||
// If the User object wasn't provided to the builder, the object is fetched from the database.
|
||||
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
|
||||
func (m *UserMutation) OldEmail(ctx context.Context) (v string, err error) { |
||||
if !m.op.Is(OpUpdateOne) { |
||||
return v, fmt.Errorf("OldEmail is only allowed on UpdateOne operations") |
||||
} |
||||
if m.id == nil || m.oldValue == nil { |
||||
return v, fmt.Errorf("OldEmail requires an ID field in the mutation") |
||||
} |
||||
oldValue, err := m.oldValue(ctx) |
||||
if err != nil { |
||||
return v, fmt.Errorf("querying old value for OldEmail: %w", err) |
||||
} |
||||
return oldValue.Email, nil |
||||
} |
||||
|
||||
// ResetEmail resets all changes to the "email" field.
|
||||
func (m *UserMutation) ResetEmail() { |
||||
m.email = nil |
||||
} |
||||
|
||||
// Where appends a list predicates to the UserMutation builder.
|
||||
func (m *UserMutation) Where(ps ...predicate.User) { |
||||
m.predicates = append(m.predicates, ps...) |
||||
} |
||||
|
||||
// Op returns the operation name.
|
||||
func (m *UserMutation) Op() Op { |
||||
return m.op |
||||
} |
||||
|
||||
// Type returns the node type of this mutation (User).
|
||||
func (m *UserMutation) Type() string { |
||||
return m.typ |
||||
} |
||||
|
||||
// Fields returns all fields that were changed during this mutation. Note that in
|
||||
// order to get all numeric fields that were incremented/decremented, call
|
||||
// AddedFields().
|
||||
func (m *UserMutation) Fields() []string { |
||||
fields := make([]string, 0, 2) |
||||
if m.name != nil { |
||||
fields = append(fields, user.FieldName) |
||||
} |
||||
if m.email != nil { |
||||
fields = append(fields, user.FieldEmail) |
||||
} |
||||
return fields |
||||
} |
||||
|
||||
// Field returns the value of a field with the given name. The second boolean
|
||||
// return value indicates that this field was not set, or was not defined in the
|
||||
// schema.
|
||||
func (m *UserMutation) Field(name string) (ent.Value, bool) { |
||||
switch name { |
||||
case user.FieldName: |
||||
return m.Name() |
||||
case user.FieldEmail: |
||||
return m.Email() |
||||
} |
||||
return nil, false |
||||
} |
||||
|
||||
// OldField returns the old value of the field from the database. An error is
|
||||
// returned if the mutation operation is not UpdateOne, or the query to the
|
||||
// database failed.
|
||||
func (m *UserMutation) OldField(ctx context.Context, name string) (ent.Value, error) { |
||||
switch name { |
||||
case user.FieldName: |
||||
return m.OldName(ctx) |
||||
case user.FieldEmail: |
||||
return m.OldEmail(ctx) |
||||
} |
||||
return nil, fmt.Errorf("unknown User field %s", name) |
||||
} |
||||
|
||||
// SetField sets the value of a field with the given name. It returns an error if
|
||||
// the field is not defined in the schema, or if the type mismatched the field
|
||||
// type.
|
||||
func (m *UserMutation) SetField(name string, value ent.Value) error { |
||||
switch name { |
||||
case user.FieldName: |
||||
v, ok := value.(string) |
||||
if !ok { |
||||
return fmt.Errorf("unexpected type %T for field %s", value, name) |
||||
} |
||||
m.SetName(v) |
||||
return nil |
||||
case user.FieldEmail: |
||||
v, ok := value.(string) |
||||
if !ok { |
||||
return fmt.Errorf("unexpected type %T for field %s", value, name) |
||||
} |
||||
m.SetEmail(v) |
||||
return nil |
||||
} |
||||
return fmt.Errorf("unknown User field %s", name) |
||||
} |
||||
|
||||
// AddedFields returns all numeric fields that were incremented/decremented during
|
||||
// this mutation.
|
||||
func (m *UserMutation) AddedFields() []string { |
||||
return nil |
||||
} |
||||
|
||||
// AddedField returns the numeric value that was incremented/decremented on a field
|
||||
// with the given name. The second boolean return value indicates that this field
|
||||
// was not set, or was not defined in the schema.
|
||||
func (m *UserMutation) AddedField(name string) (ent.Value, bool) { |
||||
return nil, false |
||||
} |
||||
|
||||
// AddField adds the value to the field with the given name. It returns an error if
|
||||
// the field is not defined in the schema, or if the type mismatched the field
|
||||
// type.
|
||||
func (m *UserMutation) AddField(name string, value ent.Value) error { |
||||
switch name { |
||||
} |
||||
return fmt.Errorf("unknown User numeric field %s", name) |
||||
} |
||||
|
||||
// ClearedFields returns all nullable fields that were cleared during this
|
||||
// mutation.
|
||||
func (m *UserMutation) ClearedFields() []string { |
||||
return nil |
||||
} |
||||
|
||||
// FieldCleared returns a boolean indicating if a field with the given name was
|
||||
// cleared in this mutation.
|
||||
func (m *UserMutation) FieldCleared(name string) bool { |
||||
_, ok := m.clearedFields[name] |
||||
return ok |
||||
} |
||||
|
||||
// ClearField clears the value of the field with the given name. It returns an
|
||||
// error if the field is not defined in the schema.
|
||||
func (m *UserMutation) ClearField(name string) error { |
||||
return fmt.Errorf("unknown User nullable field %s", name) |
||||
} |
||||
|
||||
// ResetField resets all changes in the mutation for the field with the given name.
|
||||
// It returns an error if the field is not defined in the schema.
|
||||
func (m *UserMutation) ResetField(name string) error { |
||||
switch name { |
||||
case user.FieldName: |
||||
m.ResetName() |
||||
return nil |
||||
case user.FieldEmail: |
||||
m.ResetEmail() |
||||
return nil |
||||
} |
||||
return fmt.Errorf("unknown User field %s", name) |
||||
} |
||||
|
||||
// AddedEdges returns all edge names that were set/added in this mutation.
|
||||
func (m *UserMutation) AddedEdges() []string { |
||||
edges := make([]string, 0, 0) |
||||
return edges |
||||
} |
||||
|
||||
// AddedIDs returns all IDs (to other nodes) that were added for the given edge
|
||||
// name in this mutation.
|
||||
func (m *UserMutation) AddedIDs(name string) []ent.Value { |
||||
return nil |
||||
} |
||||
|
||||
// RemovedEdges returns all edge names that were removed in this mutation.
|
||||
func (m *UserMutation) RemovedEdges() []string { |
||||
edges := make([]string, 0, 0) |
||||
return edges |
||||
} |
||||
|
||||
// RemovedIDs returns all IDs (to other nodes) that were removed for the edge with
|
||||
// the given name in this mutation.
|
||||
func (m *UserMutation) RemovedIDs(name string) []ent.Value { |
||||
return nil |
||||
} |
||||
|
||||
// ClearedEdges returns all edge names that were cleared in this mutation.
|
||||
func (m *UserMutation) ClearedEdges() []string { |
||||
edges := make([]string, 0, 0) |
||||
return edges |
||||
} |
||||
|
||||
// EdgeCleared returns a boolean which indicates if the edge with the given name
|
||||
// was cleared in this mutation.
|
||||
func (m *UserMutation) EdgeCleared(name string) bool { |
||||
return false |
||||
} |
||||
|
||||
// ClearEdge clears the value of the edge with the given name. It returns an error
|
||||
// if that edge is not defined in the schema.
|
||||
func (m *UserMutation) ClearEdge(name string) error { |
||||
return fmt.Errorf("unknown User unique edge %s", name) |
||||
} |
||||
|
||||
// ResetEdge resets all changes to the edge with the given name in this mutation.
|
||||
// It returns an error if the edge is not defined in the schema.
|
||||
func (m *UserMutation) ResetEdge(name string) error { |
||||
return fmt.Errorf("unknown User edge %s", name) |
||||
} |
@ -0,0 +1,13 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package predicate |
||||
|
||||
import ( |
||||
"entgo.io/ent/dialect/sql" |
||||
) |
||||
|
||||
// Card is the predicate function for card builders.
|
||||
type Card func(*sql.Selector) |
||||
|
||||
// User is the predicate function for user builders.
|
||||
type User func(*sql.Selector) |
@ -0,0 +1,24 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package ent |
||||
|
||||
import ( |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/schema" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/user" |
||||
) |
||||
|
||||
// The init function reads all schema descriptors with runtime code
|
||||
// (default values, validators, hooks and policies) and stitches it
|
||||
// to their package variables.
|
||||
func init() { |
||||
userFields := schema.User{}.Fields() |
||||
_ = userFields |
||||
// userDescName is the schema descriptor for name field.
|
||||
userDescName := userFields[1].Descriptor() |
||||
// user.DefaultName holds the default value on creation for the name field.
|
||||
user.DefaultName = userDescName.Default.(string) |
||||
// userDescEmail is the schema descriptor for email field.
|
||||
userDescEmail := userFields[2].Descriptor() |
||||
// user.DefaultEmail holds the default value on creation for the email field.
|
||||
user.DefaultEmail = userDescEmail.Default.(string) |
||||
} |
@ -0,0 +1,10 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package runtime |
||||
|
||||
// The schema-stitching logic is generated in github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/runtime.go
|
||||
|
||||
const ( |
||||
Version = "v0.9.0" // Version of ent codegen.
|
||||
Sum = "h1:2S1zfpMMW6p+wctj6kcYUprNPNjLWFW06T5MdyAfmWc=" // Sum of ent codegen.
|
||||
) |
@ -0,0 +1,25 @@ |
||||
package schema |
||||
|
||||
import ( |
||||
"entgo.io/ent" |
||||
"entgo.io/ent/schema/field" |
||||
) |
||||
|
||||
// Card holds the schema definition for the Card entity.
|
||||
type Card struct { |
||||
ent.Schema |
||||
} |
||||
|
||||
// Fields of the Card.
|
||||
func (Card) Fields() []ent.Field { |
||||
return []ent.Field{ |
||||
field.Int("id"), |
||||
field.String("user_id"), |
||||
field.String("money"), |
||||
} |
||||
} |
||||
|
||||
// Edges of the Card.
|
||||
func (Card) Edges() []ent.Edge { |
||||
return nil |
||||
} |
@ -0,0 +1,25 @@ |
||||
package schema |
||||
|
||||
import ( |
||||
"entgo.io/ent" |
||||
"entgo.io/ent/schema/field" |
||||
) |
||||
|
||||
// User holds the schema definition for the User entity.
|
||||
type User struct { |
||||
ent.Schema |
||||
} |
||||
|
||||
// Fields of the User.
|
||||
func (User) Fields() []ent.Field { |
||||
return []ent.Field{ |
||||
field.Int("id"), |
||||
field.String("name").Default(""), |
||||
field.String("email").Default(""), |
||||
} |
||||
} |
||||
|
||||
// Edges of the User.
|
||||
func (User) Edges() []ent.Edge { |
||||
return nil |
||||
} |
@ -0,0 +1,213 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package ent |
||||
|
||||
import ( |
||||
"context" |
||||
"sync" |
||||
|
||||
"entgo.io/ent/dialect" |
||||
) |
||||
|
||||
// Tx is a transactional client that is created by calling Client.Tx().
|
||||
type Tx struct { |
||||
config |
||||
// Card is the client for interacting with the Card builders.
|
||||
Card *CardClient |
||||
// User is the client for interacting with the User builders.
|
||||
User *UserClient |
||||
|
||||
// lazily loaded.
|
||||
client *Client |
||||
clientOnce sync.Once |
||||
|
||||
// completion callbacks.
|
||||
mu sync.Mutex |
||||
onCommit []CommitHook |
||||
onRollback []RollbackHook |
||||
|
||||
// ctx lives for the life of the transaction. It is
|
||||
// the same context used by the underlying connection.
|
||||
ctx context.Context |
||||
} |
||||
|
||||
type ( |
||||
// Committer is the interface that wraps the Committer method.
|
||||
Committer interface { |
||||
Commit(context.Context, *Tx) error |
||||
} |
||||
|
||||
// The CommitFunc type is an adapter to allow the use of ordinary
|
||||
// function as a Committer. If f is a function with the appropriate
|
||||
// signature, CommitFunc(f) is a Committer that calls f.
|
||||
CommitFunc func(context.Context, *Tx) error |
||||
|
||||
// CommitHook defines the "commit middleware". A function that gets a Committer
|
||||
// and returns a Committer. For example:
|
||||
//
|
||||
// hook := func(next ent.Committer) ent.Committer {
|
||||
// return ent.CommitFunc(func(context.Context, tx *ent.Tx) error {
|
||||
// // Do some stuff before.
|
||||
// if err := next.Commit(ctx, tx); err != nil {
|
||||
// return err
|
||||
// }
|
||||
// // Do some stuff after.
|
||||
// return nil
|
||||
// })
|
||||
// }
|
||||
//
|
||||
CommitHook func(Committer) Committer |
||||
) |
||||
|
||||
// Commit calls f(ctx, m).
|
||||
func (f CommitFunc) Commit(ctx context.Context, tx *Tx) error { |
||||
return f(ctx, tx) |
||||
} |
||||
|
||||
// Commit commits the transaction.
|
||||
func (tx *Tx) Commit() error { |
||||
txDriver := tx.config.driver.(*txDriver) |
||||
var fn Committer = CommitFunc(func(context.Context, *Tx) error { |
||||
return txDriver.tx.Commit() |
||||
}) |
||||
tx.mu.Lock() |
||||
hooks := append([]CommitHook(nil), tx.onCommit...) |
||||
tx.mu.Unlock() |
||||
for i := len(hooks) - 1; i >= 0; i-- { |
||||
fn = hooks[i](fn) |
||||
} |
||||
return fn.Commit(tx.ctx, tx) |
||||
} |
||||
|
||||
// OnCommit adds a hook to call on commit.
|
||||
func (tx *Tx) OnCommit(f CommitHook) { |
||||
tx.mu.Lock() |
||||
defer tx.mu.Unlock() |
||||
tx.onCommit = append(tx.onCommit, f) |
||||
} |
||||
|
||||
type ( |
||||
// Rollbacker is the interface that wraps the Rollbacker method.
|
||||
Rollbacker interface { |
||||
Rollback(context.Context, *Tx) error |
||||
} |
||||
|
||||
// The RollbackFunc type is an adapter to allow the use of ordinary
|
||||
// function as a Rollbacker. If f is a function with the appropriate
|
||||
// signature, RollbackFunc(f) is a Rollbacker that calls f.
|
||||
RollbackFunc func(context.Context, *Tx) error |
||||
|
||||
// RollbackHook defines the "rollback middleware". A function that gets a Rollbacker
|
||||
// and returns a Rollbacker. For example:
|
||||
//
|
||||
// hook := func(next ent.Rollbacker) ent.Rollbacker {
|
||||
// return ent.RollbackFunc(func(context.Context, tx *ent.Tx) error {
|
||||
// // Do some stuff before.
|
||||
// if err := next.Rollback(ctx, tx); err != nil {
|
||||
// return err
|
||||
// }
|
||||
// // Do some stuff after.
|
||||
// return nil
|
||||
// })
|
||||
// }
|
||||
//
|
||||
RollbackHook func(Rollbacker) Rollbacker |
||||
) |
||||
|
||||
// Rollback calls f(ctx, m).
|
||||
func (f RollbackFunc) Rollback(ctx context.Context, tx *Tx) error { |
||||
return f(ctx, tx) |
||||
} |
||||
|
||||
// Rollback rollbacks the transaction.
|
||||
func (tx *Tx) Rollback() error { |
||||
txDriver := tx.config.driver.(*txDriver) |
||||
var fn Rollbacker = RollbackFunc(func(context.Context, *Tx) error { |
||||
return txDriver.tx.Rollback() |
||||
}) |
||||
tx.mu.Lock() |
||||
hooks := append([]RollbackHook(nil), tx.onRollback...) |
||||
tx.mu.Unlock() |
||||
for i := len(hooks) - 1; i >= 0; i-- { |
||||
fn = hooks[i](fn) |
||||
} |
||||
return fn.Rollback(tx.ctx, tx) |
||||
} |
||||
|
||||
// OnRollback adds a hook to call on rollback.
|
||||
func (tx *Tx) OnRollback(f RollbackHook) { |
||||
tx.mu.Lock() |
||||
defer tx.mu.Unlock() |
||||
tx.onRollback = append(tx.onRollback, f) |
||||
} |
||||
|
||||
// Client returns a Client that binds to current transaction.
|
||||
func (tx *Tx) Client() *Client { |
||||
tx.clientOnce.Do(func() { |
||||
tx.client = &Client{config: tx.config} |
||||
tx.client.init() |
||||
}) |
||||
return tx.client |
||||
} |
||||
|
||||
func (tx *Tx) init() { |
||||
tx.Card = NewCardClient(tx.config) |
||||
tx.User = NewUserClient(tx.config) |
||||
} |
||||
|
||||
// txDriver wraps the given dialect.Tx with a nop dialect.Driver implementation.
|
||||
// The idea is to support transactions without adding any extra code to the builders.
|
||||
// When a builder calls to driver.Tx(), it gets the same dialect.Tx instance.
|
||||
// Commit and Rollback are nop for the internal builders and the user must call one
|
||||
// of them in order to commit or rollback the transaction.
|
||||
//
|
||||
// If a closed transaction is embedded in one of the generated entities, and the entity
|
||||
// applies a query, for example: Card.QueryXXX(), the query will be executed
|
||||
// through the driver which created this transaction.
|
||||
//
|
||||
// Note that txDriver is not goroutine safe.
|
||||
type txDriver struct { |
||||
// the driver we started the transaction from.
|
||||
drv dialect.Driver |
||||
// tx is the underlying transaction.
|
||||
tx dialect.Tx |
||||
} |
||||
|
||||
// newTx creates a new transactional driver.
|
||||
func newTx(ctx context.Context, drv dialect.Driver) (*txDriver, error) { |
||||
tx, err := drv.Tx(ctx) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
return &txDriver{tx: tx, drv: drv}, nil |
||||
} |
||||
|
||||
// Tx returns the transaction wrapper (txDriver) to avoid Commit or Rollback calls
|
||||
// from the internal builders. Should be called only by the internal builders.
|
||||
func (tx *txDriver) Tx(context.Context) (dialect.Tx, error) { return tx, nil } |
||||
|
||||
// Dialect returns the dialect of the driver we started the transaction from.
|
||||
func (tx *txDriver) Dialect() string { return tx.drv.Dialect() } |
||||
|
||||
// Close is a nop close.
|
||||
func (*txDriver) Close() error { return nil } |
||||
|
||||
// Commit is a nop commit for the internal builders.
|
||||
// User must call `Tx.Commit` in order to commit the transaction.
|
||||
func (*txDriver) Commit() error { return nil } |
||||
|
||||
// Rollback is a nop rollback for the internal builders.
|
||||
// User must call `Tx.Rollback` in order to rollback the transaction.
|
||||
func (*txDriver) Rollback() error { return nil } |
||||
|
||||
// Exec calls tx.Exec.
|
||||
func (tx *txDriver) Exec(ctx context.Context, query string, args, v interface{}) error { |
||||
return tx.tx.Exec(ctx, query, args, v) |
||||
} |
||||
|
||||
// Query calls tx.Query.
|
||||
func (tx *txDriver) Query(ctx context.Context, query string, args, v interface{}) error { |
||||
return tx.tx.Query(ctx, query, args, v) |
||||
} |
||||
|
||||
var _ dialect.Driver = (*txDriver)(nil) |
@ -0,0 +1,109 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package ent |
||||
|
||||
import ( |
||||
"fmt" |
||||
"strings" |
||||
|
||||
"entgo.io/ent/dialect/sql" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/user" |
||||
) |
||||
|
||||
// User is the model entity for the User schema.
|
||||
type User struct { |
||||
config `json:"-"` |
||||
// ID of the ent.
|
||||
ID int `json:"id,omitempty"` |
||||
// Name holds the value of the "name" field.
|
||||
Name string `json:"name,omitempty"` |
||||
// Email holds the value of the "email" field.
|
||||
Email string `json:"email,omitempty"` |
||||
} |
||||
|
||||
// scanValues returns the types for scanning values from sql.Rows.
|
||||
func (*User) scanValues(columns []string) ([]interface{}, error) { |
||||
values := make([]interface{}, len(columns)) |
||||
for i := range columns { |
||||
switch columns[i] { |
||||
case user.FieldID: |
||||
values[i] = new(sql.NullInt64) |
||||
case user.FieldName, user.FieldEmail: |
||||
values[i] = new(sql.NullString) |
||||
default: |
||||
return nil, fmt.Errorf("unexpected column %q for type User", columns[i]) |
||||
} |
||||
} |
||||
return values, nil |
||||
} |
||||
|
||||
// assignValues assigns the values that were returned from sql.Rows (after scanning)
|
||||
// to the User fields.
|
||||
func (u *User) assignValues(columns []string, values []interface{}) error { |
||||
if m, n := len(values), len(columns); m < n { |
||||
return fmt.Errorf("mismatch number of scan values: %d != %d", m, n) |
||||
} |
||||
for i := range columns { |
||||
switch columns[i] { |
||||
case user.FieldID: |
||||
value, ok := values[i].(*sql.NullInt64) |
||||
if !ok { |
||||
return fmt.Errorf("unexpected type %T for field id", value) |
||||
} |
||||
u.ID = int(value.Int64) |
||||
case user.FieldName: |
||||
if value, ok := values[i].(*sql.NullString); !ok { |
||||
return fmt.Errorf("unexpected type %T for field name", values[i]) |
||||
} else if value.Valid { |
||||
u.Name = value.String |
||||
} |
||||
case user.FieldEmail: |
||||
if value, ok := values[i].(*sql.NullString); !ok { |
||||
return fmt.Errorf("unexpected type %T for field email", values[i]) |
||||
} else if value.Valid { |
||||
u.Email = value.String |
||||
} |
||||
} |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
// Update returns a builder for updating this User.
|
||||
// Note that you need to call User.Unwrap() before calling this method if this User
|
||||
// was returned from a transaction, and the transaction was committed or rolled back.
|
||||
func (u *User) Update() *UserUpdateOne { |
||||
return (&UserClient{config: u.config}).UpdateOne(u) |
||||
} |
||||
|
||||
// Unwrap unwraps the User entity that was returned from a transaction after it was closed,
|
||||
// so that all future queries will be executed through the driver which created the transaction.
|
||||
func (u *User) Unwrap() *User { |
||||
tx, ok := u.config.driver.(*txDriver) |
||||
if !ok { |
||||
panic("ent: User is not a transactional entity") |
||||
} |
||||
u.config.driver = tx.drv |
||||
return u |
||||
} |
||||
|
||||
// String implements the fmt.Stringer.
|
||||
func (u *User) String() string { |
||||
var builder strings.Builder |
||||
builder.WriteString("User(") |
||||
builder.WriteString(fmt.Sprintf("id=%v", u.ID)) |
||||
builder.WriteString(", name=") |
||||
builder.WriteString(u.Name) |
||||
builder.WriteString(", email=") |
||||
builder.WriteString(u.Email) |
||||
builder.WriteByte(')') |
||||
return builder.String() |
||||
} |
||||
|
||||
// Users is a parsable slice of User.
|
||||
type Users []*User |
||||
|
||||
func (u Users) config(cfg config) { |
||||
for _i := range u { |
||||
u[_i].config = cfg |
||||
} |
||||
} |
@ -0,0 +1,40 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package user |
||||
|
||||
const ( |
||||
// Label holds the string label denoting the user type in the database.
|
||||
Label = "user" |
||||
// FieldID holds the string denoting the id field in the database.
|
||||
FieldID = "id" |
||||
// FieldName holds the string denoting the name field in the database.
|
||||
FieldName = "name" |
||||
// FieldEmail holds the string denoting the email field in the database.
|
||||
FieldEmail = "email" |
||||
// Table holds the table name of the user in the database.
|
||||
Table = "users" |
||||
) |
||||
|
||||
// Columns holds all SQL columns for user fields.
|
||||
var Columns = []string{ |
||||
FieldID, |
||||
FieldName, |
||||
FieldEmail, |
||||
} |
||||
|
||||
// ValidColumn reports if the column name is valid (part of the table columns).
|
||||
func ValidColumn(column string) bool { |
||||
for i := range Columns { |
||||
if column == Columns[i] { |
||||
return true |
||||
} |
||||
} |
||||
return false |
||||
} |
||||
|
||||
var ( |
||||
// DefaultName holds the default value on creation for the "name" field.
|
||||
DefaultName string |
||||
// DefaultEmail holds the default value on creation for the "email" field.
|
||||
DefaultEmail string |
||||
) |
@ -0,0 +1,359 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package user |
||||
|
||||
import ( |
||||
"entgo.io/ent/dialect/sql" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/predicate" |
||||
) |
||||
|
||||
// ID filters vertices based on their ID field.
|
||||
func ID(id int) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.EQ(s.C(FieldID), id)) |
||||
}) |
||||
} |
||||
|
||||
// IDEQ applies the EQ predicate on the ID field.
|
||||
func IDEQ(id int) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.EQ(s.C(FieldID), id)) |
||||
}) |
||||
} |
||||
|
||||
// IDNEQ applies the NEQ predicate on the ID field.
|
||||
func IDNEQ(id int) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.NEQ(s.C(FieldID), id)) |
||||
}) |
||||
} |
||||
|
||||
// IDIn applies the In predicate on the ID field.
|
||||
func IDIn(ids ...int) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
// if not arguments were provided, append the FALSE constants,
|
||||
// since we can't apply "IN ()". This will make this predicate falsy.
|
||||
if len(ids) == 0 { |
||||
s.Where(sql.False()) |
||||
return |
||||
} |
||||
v := make([]interface{}, len(ids)) |
||||
for i := range v { |
||||
v[i] = ids[i] |
||||
} |
||||
s.Where(sql.In(s.C(FieldID), v...)) |
||||
}) |
||||
} |
||||
|
||||
// IDNotIn applies the NotIn predicate on the ID field.
|
||||
func IDNotIn(ids ...int) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
// if not arguments were provided, append the FALSE constants,
|
||||
// since we can't apply "IN ()". This will make this predicate falsy.
|
||||
if len(ids) == 0 { |
||||
s.Where(sql.False()) |
||||
return |
||||
} |
||||
v := make([]interface{}, len(ids)) |
||||
for i := range v { |
||||
v[i] = ids[i] |
||||
} |
||||
s.Where(sql.NotIn(s.C(FieldID), v...)) |
||||
}) |
||||
} |
||||
|
||||
// IDGT applies the GT predicate on the ID field.
|
||||
func IDGT(id int) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.GT(s.C(FieldID), id)) |
||||
}) |
||||
} |
||||
|
||||
// IDGTE applies the GTE predicate on the ID field.
|
||||
func IDGTE(id int) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.GTE(s.C(FieldID), id)) |
||||
}) |
||||
} |
||||
|
||||
// IDLT applies the LT predicate on the ID field.
|
||||
func IDLT(id int) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.LT(s.C(FieldID), id)) |
||||
}) |
||||
} |
||||
|
||||
// IDLTE applies the LTE predicate on the ID field.
|
||||
func IDLTE(id int) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.LTE(s.C(FieldID), id)) |
||||
}) |
||||
} |
||||
|
||||
// Name applies equality check predicate on the "name" field. It's identical to NameEQ.
|
||||
func Name(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.EQ(s.C(FieldName), v)) |
||||
}) |
||||
} |
||||
|
||||
// Email applies equality check predicate on the "email" field. It's identical to EmailEQ.
|
||||
func Email(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.EQ(s.C(FieldEmail), v)) |
||||
}) |
||||
} |
||||
|
||||
// NameEQ applies the EQ predicate on the "name" field.
|
||||
func NameEQ(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.EQ(s.C(FieldName), v)) |
||||
}) |
||||
} |
||||
|
||||
// NameNEQ applies the NEQ predicate on the "name" field.
|
||||
func NameNEQ(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.NEQ(s.C(FieldName), v)) |
||||
}) |
||||
} |
||||
|
||||
// NameIn applies the In predicate on the "name" field.
|
||||
func NameIn(vs ...string) predicate.User { |
||||
v := make([]interface{}, len(vs)) |
||||
for i := range v { |
||||
v[i] = vs[i] |
||||
} |
||||
return predicate.User(func(s *sql.Selector) { |
||||
// if not arguments were provided, append the FALSE constants,
|
||||
// since we can't apply "IN ()". This will make this predicate falsy.
|
||||
if len(v) == 0 { |
||||
s.Where(sql.False()) |
||||
return |
||||
} |
||||
s.Where(sql.In(s.C(FieldName), v...)) |
||||
}) |
||||
} |
||||
|
||||
// NameNotIn applies the NotIn predicate on the "name" field.
|
||||
func NameNotIn(vs ...string) predicate.User { |
||||
v := make([]interface{}, len(vs)) |
||||
for i := range v { |
||||
v[i] = vs[i] |
||||
} |
||||
return predicate.User(func(s *sql.Selector) { |
||||
// if not arguments were provided, append the FALSE constants,
|
||||
// since we can't apply "IN ()". This will make this predicate falsy.
|
||||
if len(v) == 0 { |
||||
s.Where(sql.False()) |
||||
return |
||||
} |
||||
s.Where(sql.NotIn(s.C(FieldName), v...)) |
||||
}) |
||||
} |
||||
|
||||
// NameGT applies the GT predicate on the "name" field.
|
||||
func NameGT(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.GT(s.C(FieldName), v)) |
||||
}) |
||||
} |
||||
|
||||
// NameGTE applies the GTE predicate on the "name" field.
|
||||
func NameGTE(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.GTE(s.C(FieldName), v)) |
||||
}) |
||||
} |
||||
|
||||
// NameLT applies the LT predicate on the "name" field.
|
||||
func NameLT(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.LT(s.C(FieldName), v)) |
||||
}) |
||||
} |
||||
|
||||
// NameLTE applies the LTE predicate on the "name" field.
|
||||
func NameLTE(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.LTE(s.C(FieldName), v)) |
||||
}) |
||||
} |
||||
|
||||
// NameContains applies the Contains predicate on the "name" field.
|
||||
func NameContains(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.Contains(s.C(FieldName), v)) |
||||
}) |
||||
} |
||||
|
||||
// NameHasPrefix applies the HasPrefix predicate on the "name" field.
|
||||
func NameHasPrefix(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.HasPrefix(s.C(FieldName), v)) |
||||
}) |
||||
} |
||||
|
||||
// NameHasSuffix applies the HasSuffix predicate on the "name" field.
|
||||
func NameHasSuffix(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.HasSuffix(s.C(FieldName), v)) |
||||
}) |
||||
} |
||||
|
||||
// NameEqualFold applies the EqualFold predicate on the "name" field.
|
||||
func NameEqualFold(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.EqualFold(s.C(FieldName), v)) |
||||
}) |
||||
} |
||||
|
||||
// NameContainsFold applies the ContainsFold predicate on the "name" field.
|
||||
func NameContainsFold(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.ContainsFold(s.C(FieldName), v)) |
||||
}) |
||||
} |
||||
|
||||
// EmailEQ applies the EQ predicate on the "email" field.
|
||||
func EmailEQ(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.EQ(s.C(FieldEmail), v)) |
||||
}) |
||||
} |
||||
|
||||
// EmailNEQ applies the NEQ predicate on the "email" field.
|
||||
func EmailNEQ(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.NEQ(s.C(FieldEmail), v)) |
||||
}) |
||||
} |
||||
|
||||
// EmailIn applies the In predicate on the "email" field.
|
||||
func EmailIn(vs ...string) predicate.User { |
||||
v := make([]interface{}, len(vs)) |
||||
for i := range v { |
||||
v[i] = vs[i] |
||||
} |
||||
return predicate.User(func(s *sql.Selector) { |
||||
// if not arguments were provided, append the FALSE constants,
|
||||
// since we can't apply "IN ()". This will make this predicate falsy.
|
||||
if len(v) == 0 { |
||||
s.Where(sql.False()) |
||||
return |
||||
} |
||||
s.Where(sql.In(s.C(FieldEmail), v...)) |
||||
}) |
||||
} |
||||
|
||||
// EmailNotIn applies the NotIn predicate on the "email" field.
|
||||
func EmailNotIn(vs ...string) predicate.User { |
||||
v := make([]interface{}, len(vs)) |
||||
for i := range v { |
||||
v[i] = vs[i] |
||||
} |
||||
return predicate.User(func(s *sql.Selector) { |
||||
// if not arguments were provided, append the FALSE constants,
|
||||
// since we can't apply "IN ()". This will make this predicate falsy.
|
||||
if len(v) == 0 { |
||||
s.Where(sql.False()) |
||||
return |
||||
} |
||||
s.Where(sql.NotIn(s.C(FieldEmail), v...)) |
||||
}) |
||||
} |
||||
|
||||
// EmailGT applies the GT predicate on the "email" field.
|
||||
func EmailGT(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.GT(s.C(FieldEmail), v)) |
||||
}) |
||||
} |
||||
|
||||
// EmailGTE applies the GTE predicate on the "email" field.
|
||||
func EmailGTE(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.GTE(s.C(FieldEmail), v)) |
||||
}) |
||||
} |
||||
|
||||
// EmailLT applies the LT predicate on the "email" field.
|
||||
func EmailLT(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.LT(s.C(FieldEmail), v)) |
||||
}) |
||||
} |
||||
|
||||
// EmailLTE applies the LTE predicate on the "email" field.
|
||||
func EmailLTE(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.LTE(s.C(FieldEmail), v)) |
||||
}) |
||||
} |
||||
|
||||
// EmailContains applies the Contains predicate on the "email" field.
|
||||
func EmailContains(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.Contains(s.C(FieldEmail), v)) |
||||
}) |
||||
} |
||||
|
||||
// EmailHasPrefix applies the HasPrefix predicate on the "email" field.
|
||||
func EmailHasPrefix(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.HasPrefix(s.C(FieldEmail), v)) |
||||
}) |
||||
} |
||||
|
||||
// EmailHasSuffix applies the HasSuffix predicate on the "email" field.
|
||||
func EmailHasSuffix(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.HasSuffix(s.C(FieldEmail), v)) |
||||
}) |
||||
} |
||||
|
||||
// EmailEqualFold applies the EqualFold predicate on the "email" field.
|
||||
func EmailEqualFold(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.EqualFold(s.C(FieldEmail), v)) |
||||
}) |
||||
} |
||||
|
||||
// EmailContainsFold applies the ContainsFold predicate on the "email" field.
|
||||
func EmailContainsFold(v string) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s.Where(sql.ContainsFold(s.C(FieldEmail), v)) |
||||
}) |
||||
} |
||||
|
||||
// And groups predicates with the AND operator between them.
|
||||
func And(predicates ...predicate.User) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s1 := s.Clone().SetP(nil) |
||||
for _, p := range predicates { |
||||
p(s1) |
||||
} |
||||
s.Where(s1.P()) |
||||
}) |
||||
} |
||||
|
||||
// Or groups predicates with the OR operator between them.
|
||||
func Or(predicates ...predicate.User) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
s1 := s.Clone().SetP(nil) |
||||
for i, p := range predicates { |
||||
if i > 0 { |
||||
s1.Or() |
||||
} |
||||
p(s1) |
||||
} |
||||
s.Where(s1.P()) |
||||
}) |
||||
} |
||||
|
||||
// Not applies the not operator on the given predicate.
|
||||
func Not(p predicate.User) predicate.User { |
||||
return predicate.User(func(s *sql.Selector) { |
||||
p(s.Not()) |
||||
}) |
||||
} |
@ -0,0 +1,279 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package ent |
||||
|
||||
import ( |
||||
"context" |
||||
"errors" |
||||
"fmt" |
||||
|
||||
"entgo.io/ent/dialect/sql/sqlgraph" |
||||
"entgo.io/ent/schema/field" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/user" |
||||
) |
||||
|
||||
// UserCreate is the builder for creating a User entity.
|
||||
type UserCreate struct { |
||||
config |
||||
mutation *UserMutation |
||||
hooks []Hook |
||||
} |
||||
|
||||
// SetName sets the "name" field.
|
||||
func (uc *UserCreate) SetName(s string) *UserCreate { |
||||
uc.mutation.SetName(s) |
||||
return uc |
||||
} |
||||
|
||||
// SetNillableName sets the "name" field if the given value is not nil.
|
||||
func (uc *UserCreate) SetNillableName(s *string) *UserCreate { |
||||
if s != nil { |
||||
uc.SetName(*s) |
||||
} |
||||
return uc |
||||
} |
||||
|
||||
// SetEmail sets the "email" field.
|
||||
func (uc *UserCreate) SetEmail(s string) *UserCreate { |
||||
uc.mutation.SetEmail(s) |
||||
return uc |
||||
} |
||||
|
||||
// SetNillableEmail sets the "email" field if the given value is not nil.
|
||||
func (uc *UserCreate) SetNillableEmail(s *string) *UserCreate { |
||||
if s != nil { |
||||
uc.SetEmail(*s) |
||||
} |
||||
return uc |
||||
} |
||||
|
||||
// SetID sets the "id" field.
|
||||
func (uc *UserCreate) SetID(i int) *UserCreate { |
||||
uc.mutation.SetID(i) |
||||
return uc |
||||
} |
||||
|
||||
// Mutation returns the UserMutation object of the builder.
|
||||
func (uc *UserCreate) Mutation() *UserMutation { |
||||
return uc.mutation |
||||
} |
||||
|
||||
// Save creates the User in the database.
|
||||
func (uc *UserCreate) Save(ctx context.Context) (*User, error) { |
||||
var ( |
||||
err error |
||||
node *User |
||||
) |
||||
uc.defaults() |
||||
if len(uc.hooks) == 0 { |
||||
if err = uc.check(); err != nil { |
||||
return nil, err |
||||
} |
||||
node, err = uc.sqlSave(ctx) |
||||
} else { |
||||
var mut Mutator = MutateFunc(func(ctx context.Context, m Mutation) (Value, error) { |
||||
mutation, ok := m.(*UserMutation) |
||||
if !ok { |
||||
return nil, fmt.Errorf("unexpected mutation type %T", m) |
||||
} |
||||
if err = uc.check(); err != nil { |
||||
return nil, err |
||||
} |
||||
uc.mutation = mutation |
||||
if node, err = uc.sqlSave(ctx); err != nil { |
||||
return nil, err |
||||
} |
||||
mutation.id = &node.ID |
||||
mutation.done = true |
||||
return node, err |
||||
}) |
||||
for i := len(uc.hooks) - 1; i >= 0; i-- { |
||||
if uc.hooks[i] == nil { |
||||
return nil, fmt.Errorf("ent: uninitialized hook (forgotten import ent/runtime?)") |
||||
} |
||||
mut = uc.hooks[i](mut) |
||||
} |
||||
if _, err := mut.Mutate(ctx, uc.mutation); err != nil { |
||||
return nil, err |
||||
} |
||||
} |
||||
return node, err |
||||
} |
||||
|
||||
// SaveX calls Save and panics if Save returns an error.
|
||||
func (uc *UserCreate) SaveX(ctx context.Context) *User { |
||||
v, err := uc.Save(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Exec executes the query.
|
||||
func (uc *UserCreate) Exec(ctx context.Context) error { |
||||
_, err := uc.Save(ctx) |
||||
return err |
||||
} |
||||
|
||||
// ExecX is like Exec, but panics if an error occurs.
|
||||
func (uc *UserCreate) ExecX(ctx context.Context) { |
||||
if err := uc.Exec(ctx); err != nil { |
||||
panic(err) |
||||
} |
||||
} |
||||
|
||||
// defaults sets the default values of the builder before save.
|
||||
func (uc *UserCreate) defaults() { |
||||
if _, ok := uc.mutation.Name(); !ok { |
||||
v := user.DefaultName |
||||
uc.mutation.SetName(v) |
||||
} |
||||
if _, ok := uc.mutation.Email(); !ok { |
||||
v := user.DefaultEmail |
||||
uc.mutation.SetEmail(v) |
||||
} |
||||
} |
||||
|
||||
// check runs all checks and user-defined validators on the builder.
|
||||
func (uc *UserCreate) check() error { |
||||
if _, ok := uc.mutation.Name(); !ok { |
||||
return &ValidationError{Name: "name", err: errors.New(`ent: missing required field "name"`)} |
||||
} |
||||
if _, ok := uc.mutation.Email(); !ok { |
||||
return &ValidationError{Name: "email", err: errors.New(`ent: missing required field "email"`)} |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
func (uc *UserCreate) sqlSave(ctx context.Context) (*User, error) { |
||||
_node, _spec := uc.createSpec() |
||||
if err := sqlgraph.CreateNode(ctx, uc.driver, _spec); err != nil { |
||||
if sqlgraph.IsConstraintError(err) { |
||||
err = &ConstraintError{err.Error(), err} |
||||
} |
||||
return nil, err |
||||
} |
||||
if _node.ID == 0 { |
||||
id := _spec.ID.Value.(int64) |
||||
_node.ID = int(id) |
||||
} |
||||
return _node, nil |
||||
} |
||||
|
||||
func (uc *UserCreate) createSpec() (*User, *sqlgraph.CreateSpec) { |
||||
var ( |
||||
_node = &User{config: uc.config} |
||||
_spec = &sqlgraph.CreateSpec{ |
||||
Table: user.Table, |
||||
ID: &sqlgraph.FieldSpec{ |
||||
Type: field.TypeInt, |
||||
Column: user.FieldID, |
||||
}, |
||||
} |
||||
) |
||||
if id, ok := uc.mutation.ID(); ok { |
||||
_node.ID = id |
||||
_spec.ID.Value = id |
||||
} |
||||
if value, ok := uc.mutation.Name(); ok { |
||||
_spec.Fields = append(_spec.Fields, &sqlgraph.FieldSpec{ |
||||
Type: field.TypeString, |
||||
Value: value, |
||||
Column: user.FieldName, |
||||
}) |
||||
_node.Name = value |
||||
} |
||||
if value, ok := uc.mutation.Email(); ok { |
||||
_spec.Fields = append(_spec.Fields, &sqlgraph.FieldSpec{ |
||||
Type: field.TypeString, |
||||
Value: value, |
||||
Column: user.FieldEmail, |
||||
}) |
||||
_node.Email = value |
||||
} |
||||
return _node, _spec |
||||
} |
||||
|
||||
// UserCreateBulk is the builder for creating many User entities in bulk.
|
||||
type UserCreateBulk struct { |
||||
config |
||||
builders []*UserCreate |
||||
} |
||||
|
||||
// Save creates the User entities in the database.
|
||||
func (ucb *UserCreateBulk) Save(ctx context.Context) ([]*User, error) { |
||||
specs := make([]*sqlgraph.CreateSpec, len(ucb.builders)) |
||||
nodes := make([]*User, len(ucb.builders)) |
||||
mutators := make([]Mutator, len(ucb.builders)) |
||||
for i := range ucb.builders { |
||||
func(i int, root context.Context) { |
||||
builder := ucb.builders[i] |
||||
builder.defaults() |
||||
var mut Mutator = MutateFunc(func(ctx context.Context, m Mutation) (Value, error) { |
||||
mutation, ok := m.(*UserMutation) |
||||
if !ok { |
||||
return nil, fmt.Errorf("unexpected mutation type %T", m) |
||||
} |
||||
if err := builder.check(); err != nil { |
||||
return nil, err |
||||
} |
||||
builder.mutation = mutation |
||||
nodes[i], specs[i] = builder.createSpec() |
||||
var err error |
||||
if i < len(mutators)-1 { |
||||
_, err = mutators[i+1].Mutate(root, ucb.builders[i+1].mutation) |
||||
} else { |
||||
spec := &sqlgraph.BatchCreateSpec{Nodes: specs} |
||||
// Invoke the actual operation on the latest mutation in the chain.
|
||||
if err = sqlgraph.BatchCreate(ctx, ucb.driver, spec); err != nil { |
||||
if sqlgraph.IsConstraintError(err) { |
||||
err = &ConstraintError{err.Error(), err} |
||||
} |
||||
} |
||||
} |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
mutation.id = &nodes[i].ID |
||||
mutation.done = true |
||||
if specs[i].ID.Value != nil && nodes[i].ID == 0 { |
||||
id := specs[i].ID.Value.(int64) |
||||
nodes[i].ID = int(id) |
||||
} |
||||
return nodes[i], nil |
||||
}) |
||||
for i := len(builder.hooks) - 1; i >= 0; i-- { |
||||
mut = builder.hooks[i](mut) |
||||
} |
||||
mutators[i] = mut |
||||
}(i, ctx) |
||||
} |
||||
if len(mutators) > 0 { |
||||
if _, err := mutators[0].Mutate(ctx, ucb.builders[0].mutation); err != nil { |
||||
return nil, err |
||||
} |
||||
} |
||||
return nodes, nil |
||||
} |
||||
|
||||
// SaveX is like Save, but panics if an error occurs.
|
||||
func (ucb *UserCreateBulk) SaveX(ctx context.Context) []*User { |
||||
v, err := ucb.Save(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Exec executes the query.
|
||||
func (ucb *UserCreateBulk) Exec(ctx context.Context) error { |
||||
_, err := ucb.Save(ctx) |
||||
return err |
||||
} |
||||
|
||||
// ExecX is like Exec, but panics if an error occurs.
|
||||
func (ucb *UserCreateBulk) ExecX(ctx context.Context) { |
||||
if err := ucb.Exec(ctx); err != nil { |
||||
panic(err) |
||||
} |
||||
} |
@ -0,0 +1,111 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package ent |
||||
|
||||
import ( |
||||
"context" |
||||
"fmt" |
||||
|
||||
"entgo.io/ent/dialect/sql" |
||||
"entgo.io/ent/dialect/sql/sqlgraph" |
||||
"entgo.io/ent/schema/field" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/predicate" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/user" |
||||
) |
||||
|
||||
// UserDelete is the builder for deleting a User entity.
|
||||
type UserDelete struct { |
||||
config |
||||
hooks []Hook |
||||
mutation *UserMutation |
||||
} |
||||
|
||||
// Where appends a list predicates to the UserDelete builder.
|
||||
func (ud *UserDelete) Where(ps ...predicate.User) *UserDelete { |
||||
ud.mutation.Where(ps...) |
||||
return ud |
||||
} |
||||
|
||||
// Exec executes the deletion query and returns how many vertices were deleted.
|
||||
func (ud *UserDelete) Exec(ctx context.Context) (int, error) { |
||||
var ( |
||||
err error |
||||
affected int |
||||
) |
||||
if len(ud.hooks) == 0 { |
||||
affected, err = ud.sqlExec(ctx) |
||||
} else { |
||||
var mut Mutator = MutateFunc(func(ctx context.Context, m Mutation) (Value, error) { |
||||
mutation, ok := m.(*UserMutation) |
||||
if !ok { |
||||
return nil, fmt.Errorf("unexpected mutation type %T", m) |
||||
} |
||||
ud.mutation = mutation |
||||
affected, err = ud.sqlExec(ctx) |
||||
mutation.done = true |
||||
return affected, err |
||||
}) |
||||
for i := len(ud.hooks) - 1; i >= 0; i-- { |
||||
if ud.hooks[i] == nil { |
||||
return 0, fmt.Errorf("ent: uninitialized hook (forgotten import ent/runtime?)") |
||||
} |
||||
mut = ud.hooks[i](mut) |
||||
} |
||||
if _, err := mut.Mutate(ctx, ud.mutation); err != nil { |
||||
return 0, err |
||||
} |
||||
} |
||||
return affected, err |
||||
} |
||||
|
||||
// ExecX is like Exec, but panics if an error occurs.
|
||||
func (ud *UserDelete) ExecX(ctx context.Context) int { |
||||
n, err := ud.Exec(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return n |
||||
} |
||||
|
||||
func (ud *UserDelete) sqlExec(ctx context.Context) (int, error) { |
||||
_spec := &sqlgraph.DeleteSpec{ |
||||
Node: &sqlgraph.NodeSpec{ |
||||
Table: user.Table, |
||||
ID: &sqlgraph.FieldSpec{ |
||||
Type: field.TypeInt, |
||||
Column: user.FieldID, |
||||
}, |
||||
}, |
||||
} |
||||
if ps := ud.mutation.predicates; len(ps) > 0 { |
||||
_spec.Predicate = func(selector *sql.Selector) { |
||||
for i := range ps { |
||||
ps[i](selector) |
||||
} |
||||
} |
||||
} |
||||
return sqlgraph.DeleteNodes(ctx, ud.driver, _spec) |
||||
} |
||||
|
||||
// UserDeleteOne is the builder for deleting a single User entity.
|
||||
type UserDeleteOne struct { |
||||
ud *UserDelete |
||||
} |
||||
|
||||
// Exec executes the deletion query.
|
||||
func (udo *UserDeleteOne) Exec(ctx context.Context) error { |
||||
n, err := udo.ud.Exec(ctx) |
||||
switch { |
||||
case err != nil: |
||||
return err |
||||
case n == 0: |
||||
return &NotFoundError{user.Label} |
||||
default: |
||||
return nil |
||||
} |
||||
} |
||||
|
||||
// ExecX is like Exec, but panics if an error occurs.
|
||||
func (udo *UserDeleteOne) ExecX(ctx context.Context) { |
||||
udo.ud.ExecX(ctx) |
||||
} |
@ -0,0 +1,915 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package ent |
||||
|
||||
import ( |
||||
"context" |
||||
"errors" |
||||
"fmt" |
||||
"math" |
||||
|
||||
"entgo.io/ent/dialect/sql" |
||||
"entgo.io/ent/dialect/sql/sqlgraph" |
||||
"entgo.io/ent/schema/field" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/predicate" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/user" |
||||
) |
||||
|
||||
// UserQuery is the builder for querying User entities.
|
||||
type UserQuery struct { |
||||
config |
||||
limit *int |
||||
offset *int |
||||
unique *bool |
||||
order []OrderFunc |
||||
fields []string |
||||
predicates []predicate.User |
||||
// intermediate query (i.e. traversal path).
|
||||
sql *sql.Selector |
||||
path func(context.Context) (*sql.Selector, error) |
||||
} |
||||
|
||||
// Where adds a new predicate for the UserQuery builder.
|
||||
func (uq *UserQuery) Where(ps ...predicate.User) *UserQuery { |
||||
uq.predicates = append(uq.predicates, ps...) |
||||
return uq |
||||
} |
||||
|
||||
// Limit adds a limit step to the query.
|
||||
func (uq *UserQuery) Limit(limit int) *UserQuery { |
||||
uq.limit = &limit |
||||
return uq |
||||
} |
||||
|
||||
// Offset adds an offset step to the query.
|
||||
func (uq *UserQuery) Offset(offset int) *UserQuery { |
||||
uq.offset = &offset |
||||
return uq |
||||
} |
||||
|
||||
// Unique configures the query builder to filter duplicate records on query.
|
||||
// By default, unique is set to true, and can be disabled using this method.
|
||||
func (uq *UserQuery) Unique(unique bool) *UserQuery { |
||||
uq.unique = &unique |
||||
return uq |
||||
} |
||||
|
||||
// Order adds an order step to the query.
|
||||
func (uq *UserQuery) Order(o ...OrderFunc) *UserQuery { |
||||
uq.order = append(uq.order, o...) |
||||
return uq |
||||
} |
||||
|
||||
// First returns the first User entity from the query.
|
||||
// Returns a *NotFoundError when no User was found.
|
||||
func (uq *UserQuery) First(ctx context.Context) (*User, error) { |
||||
nodes, err := uq.Limit(1).All(ctx) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
if len(nodes) == 0 { |
||||
return nil, &NotFoundError{user.Label} |
||||
} |
||||
return nodes[0], nil |
||||
} |
||||
|
||||
// FirstX is like First, but panics if an error occurs.
|
||||
func (uq *UserQuery) FirstX(ctx context.Context) *User { |
||||
node, err := uq.First(ctx) |
||||
if err != nil && !IsNotFound(err) { |
||||
panic(err) |
||||
} |
||||
return node |
||||
} |
||||
|
||||
// FirstID returns the first User ID from the query.
|
||||
// Returns a *NotFoundError when no User ID was found.
|
||||
func (uq *UserQuery) FirstID(ctx context.Context) (id int, err error) { |
||||
var ids []int |
||||
if ids, err = uq.Limit(1).IDs(ctx); err != nil { |
||||
return |
||||
} |
||||
if len(ids) == 0 { |
||||
err = &NotFoundError{user.Label} |
||||
return |
||||
} |
||||
return ids[0], nil |
||||
} |
||||
|
||||
// FirstIDX is like FirstID, but panics if an error occurs.
|
||||
func (uq *UserQuery) FirstIDX(ctx context.Context) int { |
||||
id, err := uq.FirstID(ctx) |
||||
if err != nil && !IsNotFound(err) { |
||||
panic(err) |
||||
} |
||||
return id |
||||
} |
||||
|
||||
// Only returns a single User entity found by the query, ensuring it only returns one.
|
||||
// Returns a *NotSingularError when exactly one User entity is not found.
|
||||
// Returns a *NotFoundError when no User entities are found.
|
||||
func (uq *UserQuery) Only(ctx context.Context) (*User, error) { |
||||
nodes, err := uq.Limit(2).All(ctx) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
switch len(nodes) { |
||||
case 1: |
||||
return nodes[0], nil |
||||
case 0: |
||||
return nil, &NotFoundError{user.Label} |
||||
default: |
||||
return nil, &NotSingularError{user.Label} |
||||
} |
||||
} |
||||
|
||||
// OnlyX is like Only, but panics if an error occurs.
|
||||
func (uq *UserQuery) OnlyX(ctx context.Context) *User { |
||||
node, err := uq.Only(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return node |
||||
} |
||||
|
||||
// OnlyID is like Only, but returns the only User ID in the query.
|
||||
// Returns a *NotSingularError when exactly one User ID is not found.
|
||||
// Returns a *NotFoundError when no entities are found.
|
||||
func (uq *UserQuery) OnlyID(ctx context.Context) (id int, err error) { |
||||
var ids []int |
||||
if ids, err = uq.Limit(2).IDs(ctx); err != nil { |
||||
return |
||||
} |
||||
switch len(ids) { |
||||
case 1: |
||||
id = ids[0] |
||||
case 0: |
||||
err = &NotFoundError{user.Label} |
||||
default: |
||||
err = &NotSingularError{user.Label} |
||||
} |
||||
return |
||||
} |
||||
|
||||
// OnlyIDX is like OnlyID, but panics if an error occurs.
|
||||
func (uq *UserQuery) OnlyIDX(ctx context.Context) int { |
||||
id, err := uq.OnlyID(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return id |
||||
} |
||||
|
||||
// All executes the query and returns a list of Users.
|
||||
func (uq *UserQuery) All(ctx context.Context) ([]*User, error) { |
||||
if err := uq.prepareQuery(ctx); err != nil { |
||||
return nil, err |
||||
} |
||||
return uq.sqlAll(ctx) |
||||
} |
||||
|
||||
// AllX is like All, but panics if an error occurs.
|
||||
func (uq *UserQuery) AllX(ctx context.Context) []*User { |
||||
nodes, err := uq.All(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return nodes |
||||
} |
||||
|
||||
// IDs executes the query and returns a list of User IDs.
|
||||
func (uq *UserQuery) IDs(ctx context.Context) ([]int, error) { |
||||
var ids []int |
||||
if err := uq.Select(user.FieldID).Scan(ctx, &ids); err != nil { |
||||
return nil, err |
||||
} |
||||
return ids, nil |
||||
} |
||||
|
||||
// IDsX is like IDs, but panics if an error occurs.
|
||||
func (uq *UserQuery) IDsX(ctx context.Context) []int { |
||||
ids, err := uq.IDs(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return ids |
||||
} |
||||
|
||||
// Count returns the count of the given query.
|
||||
func (uq *UserQuery) Count(ctx context.Context) (int, error) { |
||||
if err := uq.prepareQuery(ctx); err != nil { |
||||
return 0, err |
||||
} |
||||
return uq.sqlCount(ctx) |
||||
} |
||||
|
||||
// CountX is like Count, but panics if an error occurs.
|
||||
func (uq *UserQuery) CountX(ctx context.Context) int { |
||||
count, err := uq.Count(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return count |
||||
} |
||||
|
||||
// Exist returns true if the query has elements in the graph.
|
||||
func (uq *UserQuery) Exist(ctx context.Context) (bool, error) { |
||||
if err := uq.prepareQuery(ctx); err != nil { |
||||
return false, err |
||||
} |
||||
return uq.sqlExist(ctx) |
||||
} |
||||
|
||||
// ExistX is like Exist, but panics if an error occurs.
|
||||
func (uq *UserQuery) ExistX(ctx context.Context) bool { |
||||
exist, err := uq.Exist(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return exist |
||||
} |
||||
|
||||
// Clone returns a duplicate of the UserQuery builder, including all associated steps. It can be
|
||||
// used to prepare common query builders and use them differently after the clone is made.
|
||||
func (uq *UserQuery) Clone() *UserQuery { |
||||
if uq == nil { |
||||
return nil |
||||
} |
||||
return &UserQuery{ |
||||
config: uq.config, |
||||
limit: uq.limit, |
||||
offset: uq.offset, |
||||
order: append([]OrderFunc{}, uq.order...), |
||||
predicates: append([]predicate.User{}, uq.predicates...), |
||||
// clone intermediate query.
|
||||
sql: uq.sql.Clone(), |
||||
path: uq.path, |
||||
} |
||||
} |
||||
|
||||
// GroupBy is used to group vertices by one or more fields/columns.
|
||||
// It is often used with aggregate functions, like: count, max, mean, min, sum.
|
||||
//
|
||||
// Example:
|
||||
//
|
||||
// var v []struct {
|
||||
// Name string `json:"name,omitempty"`
|
||||
// Count int `json:"count,omitempty"`
|
||||
// }
|
||||
//
|
||||
// client.User.Query().
|
||||
// GroupBy(user.FieldName).
|
||||
// Aggregate(ent.Count()).
|
||||
// Scan(ctx, &v)
|
||||
//
|
||||
func (uq *UserQuery) GroupBy(field string, fields ...string) *UserGroupBy { |
||||
group := &UserGroupBy{config: uq.config} |
||||
group.fields = append([]string{field}, fields...) |
||||
group.path = func(ctx context.Context) (prev *sql.Selector, err error) { |
||||
if err := uq.prepareQuery(ctx); err != nil { |
||||
return nil, err |
||||
} |
||||
return uq.sqlQuery(ctx), nil |
||||
} |
||||
return group |
||||
} |
||||
|
||||
// Select allows the selection one or more fields/columns for the given query,
|
||||
// instead of selecting all fields in the entity.
|
||||
//
|
||||
// Example:
|
||||
//
|
||||
// var v []struct {
|
||||
// Name string `json:"name,omitempty"`
|
||||
// }
|
||||
//
|
||||
// client.User.Query().
|
||||
// Select(user.FieldName).
|
||||
// Scan(ctx, &v)
|
||||
//
|
||||
func (uq *UserQuery) Select(fields ...string) *UserSelect { |
||||
uq.fields = append(uq.fields, fields...) |
||||
return &UserSelect{UserQuery: uq} |
||||
} |
||||
|
||||
func (uq *UserQuery) prepareQuery(ctx context.Context) error { |
||||
for _, f := range uq.fields { |
||||
if !user.ValidColumn(f) { |
||||
return &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)} |
||||
} |
||||
} |
||||
if uq.path != nil { |
||||
prev, err := uq.path(ctx) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
uq.sql = prev |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
func (uq *UserQuery) sqlAll(ctx context.Context) ([]*User, error) { |
||||
var ( |
||||
nodes = []*User{} |
||||
_spec = uq.querySpec() |
||||
) |
||||
_spec.ScanValues = func(columns []string) ([]interface{}, error) { |
||||
node := &User{config: uq.config} |
||||
nodes = append(nodes, node) |
||||
return node.scanValues(columns) |
||||
} |
||||
_spec.Assign = func(columns []string, values []interface{}) error { |
||||
if len(nodes) == 0 { |
||||
return fmt.Errorf("ent: Assign called without calling ScanValues") |
||||
} |
||||
node := nodes[len(nodes)-1] |
||||
return node.assignValues(columns, values) |
||||
} |
||||
if err := sqlgraph.QueryNodes(ctx, uq.driver, _spec); err != nil { |
||||
return nil, err |
||||
} |
||||
if len(nodes) == 0 { |
||||
return nodes, nil |
||||
} |
||||
return nodes, nil |
||||
} |
||||
|
||||
func (uq *UserQuery) sqlCount(ctx context.Context) (int, error) { |
||||
_spec := uq.querySpec() |
||||
return sqlgraph.CountNodes(ctx, uq.driver, _spec) |
||||
} |
||||
|
||||
func (uq *UserQuery) sqlExist(ctx context.Context) (bool, error) { |
||||
n, err := uq.sqlCount(ctx) |
||||
if err != nil { |
||||
return false, fmt.Errorf("ent: check existence: %w", err) |
||||
} |
||||
return n > 0, nil |
||||
} |
||||
|
||||
func (uq *UserQuery) querySpec() *sqlgraph.QuerySpec { |
||||
_spec := &sqlgraph.QuerySpec{ |
||||
Node: &sqlgraph.NodeSpec{ |
||||
Table: user.Table, |
||||
Columns: user.Columns, |
||||
ID: &sqlgraph.FieldSpec{ |
||||
Type: field.TypeInt, |
||||
Column: user.FieldID, |
||||
}, |
||||
}, |
||||
From: uq.sql, |
||||
Unique: true, |
||||
} |
||||
if unique := uq.unique; unique != nil { |
||||
_spec.Unique = *unique |
||||
} |
||||
if fields := uq.fields; len(fields) > 0 { |
||||
_spec.Node.Columns = make([]string, 0, len(fields)) |
||||
_spec.Node.Columns = append(_spec.Node.Columns, user.FieldID) |
||||
for i := range fields { |
||||
if fields[i] != user.FieldID { |
||||
_spec.Node.Columns = append(_spec.Node.Columns, fields[i]) |
||||
} |
||||
} |
||||
} |
||||
if ps := uq.predicates; len(ps) > 0 { |
||||
_spec.Predicate = func(selector *sql.Selector) { |
||||
for i := range ps { |
||||
ps[i](selector) |
||||
} |
||||
} |
||||
} |
||||
if limit := uq.limit; limit != nil { |
||||
_spec.Limit = *limit |
||||
} |
||||
if offset := uq.offset; offset != nil { |
||||
_spec.Offset = *offset |
||||
} |
||||
if ps := uq.order; len(ps) > 0 { |
||||
_spec.Order = func(selector *sql.Selector) { |
||||
for i := range ps { |
||||
ps[i](selector) |
||||
} |
||||
} |
||||
} |
||||
return _spec |
||||
} |
||||
|
||||
func (uq *UserQuery) sqlQuery(ctx context.Context) *sql.Selector { |
||||
builder := sql.Dialect(uq.driver.Dialect()) |
||||
t1 := builder.Table(user.Table) |
||||
columns := uq.fields |
||||
if len(columns) == 0 { |
||||
columns = user.Columns |
||||
} |
||||
selector := builder.Select(t1.Columns(columns...)...).From(t1) |
||||
if uq.sql != nil { |
||||
selector = uq.sql |
||||
selector.Select(selector.Columns(columns...)...) |
||||
} |
||||
for _, p := range uq.predicates { |
||||
p(selector) |
||||
} |
||||
for _, p := range uq.order { |
||||
p(selector) |
||||
} |
||||
if offset := uq.offset; offset != nil { |
||||
// limit is mandatory for offset clause. We start
|
||||
// with default value, and override it below if needed.
|
||||
selector.Offset(*offset).Limit(math.MaxInt32) |
||||
} |
||||
if limit := uq.limit; limit != nil { |
||||
selector.Limit(*limit) |
||||
} |
||||
return selector |
||||
} |
||||
|
||||
// UserGroupBy is the group-by builder for User entities.
|
||||
type UserGroupBy struct { |
||||
config |
||||
fields []string |
||||
fns []AggregateFunc |
||||
// intermediate query (i.e. traversal path).
|
||||
sql *sql.Selector |
||||
path func(context.Context) (*sql.Selector, error) |
||||
} |
||||
|
||||
// Aggregate adds the given aggregation functions to the group-by query.
|
||||
func (ugb *UserGroupBy) Aggregate(fns ...AggregateFunc) *UserGroupBy { |
||||
ugb.fns = append(ugb.fns, fns...) |
||||
return ugb |
||||
} |
||||
|
||||
// Scan applies the group-by query and scans the result into the given value.
|
||||
func (ugb *UserGroupBy) Scan(ctx context.Context, v interface{}) error { |
||||
query, err := ugb.path(ctx) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
ugb.sql = query |
||||
return ugb.sqlScan(ctx, v) |
||||
} |
||||
|
||||
// ScanX is like Scan, but panics if an error occurs.
|
||||
func (ugb *UserGroupBy) ScanX(ctx context.Context, v interface{}) { |
||||
if err := ugb.Scan(ctx, v); err != nil { |
||||
panic(err) |
||||
} |
||||
} |
||||
|
||||
// Strings returns list of strings from group-by.
|
||||
// It is only allowed when executing a group-by query with one field.
|
||||
func (ugb *UserGroupBy) Strings(ctx context.Context) ([]string, error) { |
||||
if len(ugb.fields) > 1 { |
||||
return nil, errors.New("ent: UserGroupBy.Strings is not achievable when grouping more than 1 field") |
||||
} |
||||
var v []string |
||||
if err := ugb.Scan(ctx, &v); err != nil { |
||||
return nil, err |
||||
} |
||||
return v, nil |
||||
} |
||||
|
||||
// StringsX is like Strings, but panics if an error occurs.
|
||||
func (ugb *UserGroupBy) StringsX(ctx context.Context) []string { |
||||
v, err := ugb.Strings(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// String returns a single string from a group-by query.
|
||||
// It is only allowed when executing a group-by query with one field.
|
||||
func (ugb *UserGroupBy) String(ctx context.Context) (_ string, err error) { |
||||
var v []string |
||||
if v, err = ugb.Strings(ctx); err != nil { |
||||
return |
||||
} |
||||
switch len(v) { |
||||
case 1: |
||||
return v[0], nil |
||||
case 0: |
||||
err = &NotFoundError{user.Label} |
||||
default: |
||||
err = fmt.Errorf("ent: UserGroupBy.Strings returned %d results when one was expected", len(v)) |
||||
} |
||||
return |
||||
} |
||||
|
||||
// StringX is like String, but panics if an error occurs.
|
||||
func (ugb *UserGroupBy) StringX(ctx context.Context) string { |
||||
v, err := ugb.String(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Ints returns list of ints from group-by.
|
||||
// It is only allowed when executing a group-by query with one field.
|
||||
func (ugb *UserGroupBy) Ints(ctx context.Context) ([]int, error) { |
||||
if len(ugb.fields) > 1 { |
||||
return nil, errors.New("ent: UserGroupBy.Ints is not achievable when grouping more than 1 field") |
||||
} |
||||
var v []int |
||||
if err := ugb.Scan(ctx, &v); err != nil { |
||||
return nil, err |
||||
} |
||||
return v, nil |
||||
} |
||||
|
||||
// IntsX is like Ints, but panics if an error occurs.
|
||||
func (ugb *UserGroupBy) IntsX(ctx context.Context) []int { |
||||
v, err := ugb.Ints(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Int returns a single int from a group-by query.
|
||||
// It is only allowed when executing a group-by query with one field.
|
||||
func (ugb *UserGroupBy) Int(ctx context.Context) (_ int, err error) { |
||||
var v []int |
||||
if v, err = ugb.Ints(ctx); err != nil { |
||||
return |
||||
} |
||||
switch len(v) { |
||||
case 1: |
||||
return v[0], nil |
||||
case 0: |
||||
err = &NotFoundError{user.Label} |
||||
default: |
||||
err = fmt.Errorf("ent: UserGroupBy.Ints returned %d results when one was expected", len(v)) |
||||
} |
||||
return |
||||
} |
||||
|
||||
// IntX is like Int, but panics if an error occurs.
|
||||
func (ugb *UserGroupBy) IntX(ctx context.Context) int { |
||||
v, err := ugb.Int(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Float64s returns list of float64s from group-by.
|
||||
// It is only allowed when executing a group-by query with one field.
|
||||
func (ugb *UserGroupBy) Float64s(ctx context.Context) ([]float64, error) { |
||||
if len(ugb.fields) > 1 { |
||||
return nil, errors.New("ent: UserGroupBy.Float64s is not achievable when grouping more than 1 field") |
||||
} |
||||
var v []float64 |
||||
if err := ugb.Scan(ctx, &v); err != nil { |
||||
return nil, err |
||||
} |
||||
return v, nil |
||||
} |
||||
|
||||
// Float64sX is like Float64s, but panics if an error occurs.
|
||||
func (ugb *UserGroupBy) Float64sX(ctx context.Context) []float64 { |
||||
v, err := ugb.Float64s(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Float64 returns a single float64 from a group-by query.
|
||||
// It is only allowed when executing a group-by query with one field.
|
||||
func (ugb *UserGroupBy) Float64(ctx context.Context) (_ float64, err error) { |
||||
var v []float64 |
||||
if v, err = ugb.Float64s(ctx); err != nil { |
||||
return |
||||
} |
||||
switch len(v) { |
||||
case 1: |
||||
return v[0], nil |
||||
case 0: |
||||
err = &NotFoundError{user.Label} |
||||
default: |
||||
err = fmt.Errorf("ent: UserGroupBy.Float64s returned %d results when one was expected", len(v)) |
||||
} |
||||
return |
||||
} |
||||
|
||||
// Float64X is like Float64, but panics if an error occurs.
|
||||
func (ugb *UserGroupBy) Float64X(ctx context.Context) float64 { |
||||
v, err := ugb.Float64(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Bools returns list of bools from group-by.
|
||||
// It is only allowed when executing a group-by query with one field.
|
||||
func (ugb *UserGroupBy) Bools(ctx context.Context) ([]bool, error) { |
||||
if len(ugb.fields) > 1 { |
||||
return nil, errors.New("ent: UserGroupBy.Bools is not achievable when grouping more than 1 field") |
||||
} |
||||
var v []bool |
||||
if err := ugb.Scan(ctx, &v); err != nil { |
||||
return nil, err |
||||
} |
||||
return v, nil |
||||
} |
||||
|
||||
// BoolsX is like Bools, but panics if an error occurs.
|
||||
func (ugb *UserGroupBy) BoolsX(ctx context.Context) []bool { |
||||
v, err := ugb.Bools(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Bool returns a single bool from a group-by query.
|
||||
// It is only allowed when executing a group-by query with one field.
|
||||
func (ugb *UserGroupBy) Bool(ctx context.Context) (_ bool, err error) { |
||||
var v []bool |
||||
if v, err = ugb.Bools(ctx); err != nil { |
||||
return |
||||
} |
||||
switch len(v) { |
||||
case 1: |
||||
return v[0], nil |
||||
case 0: |
||||
err = &NotFoundError{user.Label} |
||||
default: |
||||
err = fmt.Errorf("ent: UserGroupBy.Bools returned %d results when one was expected", len(v)) |
||||
} |
||||
return |
||||
} |
||||
|
||||
// BoolX is like Bool, but panics if an error occurs.
|
||||
func (ugb *UserGroupBy) BoolX(ctx context.Context) bool { |
||||
v, err := ugb.Bool(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
func (ugb *UserGroupBy) sqlScan(ctx context.Context, v interface{}) error { |
||||
for _, f := range ugb.fields { |
||||
if !user.ValidColumn(f) { |
||||
return &ValidationError{Name: f, err: fmt.Errorf("invalid field %q for group-by", f)} |
||||
} |
||||
} |
||||
selector := ugb.sqlQuery() |
||||
if err := selector.Err(); err != nil { |
||||
return err |
||||
} |
||||
rows := &sql.Rows{} |
||||
query, args := selector.Query() |
||||
if err := ugb.driver.Query(ctx, query, args, rows); err != nil { |
||||
return err |
||||
} |
||||
defer rows.Close() |
||||
return sql.ScanSlice(rows, v) |
||||
} |
||||
|
||||
func (ugb *UserGroupBy) sqlQuery() *sql.Selector { |
||||
selector := ugb.sql.Select() |
||||
aggregation := make([]string, 0, len(ugb.fns)) |
||||
for _, fn := range ugb.fns { |
||||
aggregation = append(aggregation, fn(selector)) |
||||
} |
||||
// If no columns were selected in a custom aggregation function, the default
|
||||
// selection is the fields used for "group-by", and the aggregation functions.
|
||||
if len(selector.SelectedColumns()) == 0 { |
||||
columns := make([]string, 0, len(ugb.fields)+len(ugb.fns)) |
||||
for _, f := range ugb.fields { |
||||
columns = append(columns, selector.C(f)) |
||||
} |
||||
for _, c := range aggregation { |
||||
columns = append(columns, c) |
||||
} |
||||
selector.Select(columns...) |
||||
} |
||||
return selector.GroupBy(selector.Columns(ugb.fields...)...) |
||||
} |
||||
|
||||
// UserSelect is the builder for selecting fields of User entities.
|
||||
type UserSelect struct { |
||||
*UserQuery |
||||
// intermediate query (i.e. traversal path).
|
||||
sql *sql.Selector |
||||
} |
||||
|
||||
// Scan applies the selector query and scans the result into the given value.
|
||||
func (us *UserSelect) Scan(ctx context.Context, v interface{}) error { |
||||
if err := us.prepareQuery(ctx); err != nil { |
||||
return err |
||||
} |
||||
us.sql = us.UserQuery.sqlQuery(ctx) |
||||
return us.sqlScan(ctx, v) |
||||
} |
||||
|
||||
// ScanX is like Scan, but panics if an error occurs.
|
||||
func (us *UserSelect) ScanX(ctx context.Context, v interface{}) { |
||||
if err := us.Scan(ctx, v); err != nil { |
||||
panic(err) |
||||
} |
||||
} |
||||
|
||||
// Strings returns list of strings from a selector. It is only allowed when selecting one field.
|
||||
func (us *UserSelect) Strings(ctx context.Context) ([]string, error) { |
||||
if len(us.fields) > 1 { |
||||
return nil, errors.New("ent: UserSelect.Strings is not achievable when selecting more than 1 field") |
||||
} |
||||
var v []string |
||||
if err := us.Scan(ctx, &v); err != nil { |
||||
return nil, err |
||||
} |
||||
return v, nil |
||||
} |
||||
|
||||
// StringsX is like Strings, but panics if an error occurs.
|
||||
func (us *UserSelect) StringsX(ctx context.Context) []string { |
||||
v, err := us.Strings(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// String returns a single string from a selector. It is only allowed when selecting one field.
|
||||
func (us *UserSelect) String(ctx context.Context) (_ string, err error) { |
||||
var v []string |
||||
if v, err = us.Strings(ctx); err != nil { |
||||
return |
||||
} |
||||
switch len(v) { |
||||
case 1: |
||||
return v[0], nil |
||||
case 0: |
||||
err = &NotFoundError{user.Label} |
||||
default: |
||||
err = fmt.Errorf("ent: UserSelect.Strings returned %d results when one was expected", len(v)) |
||||
} |
||||
return |
||||
} |
||||
|
||||
// StringX is like String, but panics if an error occurs.
|
||||
func (us *UserSelect) StringX(ctx context.Context) string { |
||||
v, err := us.String(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Ints returns list of ints from a selector. It is only allowed when selecting one field.
|
||||
func (us *UserSelect) Ints(ctx context.Context) ([]int, error) { |
||||
if len(us.fields) > 1 { |
||||
return nil, errors.New("ent: UserSelect.Ints is not achievable when selecting more than 1 field") |
||||
} |
||||
var v []int |
||||
if err := us.Scan(ctx, &v); err != nil { |
||||
return nil, err |
||||
} |
||||
return v, nil |
||||
} |
||||
|
||||
// IntsX is like Ints, but panics if an error occurs.
|
||||
func (us *UserSelect) IntsX(ctx context.Context) []int { |
||||
v, err := us.Ints(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Int returns a single int from a selector. It is only allowed when selecting one field.
|
||||
func (us *UserSelect) Int(ctx context.Context) (_ int, err error) { |
||||
var v []int |
||||
if v, err = us.Ints(ctx); err != nil { |
||||
return |
||||
} |
||||
switch len(v) { |
||||
case 1: |
||||
return v[0], nil |
||||
case 0: |
||||
err = &NotFoundError{user.Label} |
||||
default: |
||||
err = fmt.Errorf("ent: UserSelect.Ints returned %d results when one was expected", len(v)) |
||||
} |
||||
return |
||||
} |
||||
|
||||
// IntX is like Int, but panics if an error occurs.
|
||||
func (us *UserSelect) IntX(ctx context.Context) int { |
||||
v, err := us.Int(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Float64s returns list of float64s from a selector. It is only allowed when selecting one field.
|
||||
func (us *UserSelect) Float64s(ctx context.Context) ([]float64, error) { |
||||
if len(us.fields) > 1 { |
||||
return nil, errors.New("ent: UserSelect.Float64s is not achievable when selecting more than 1 field") |
||||
} |
||||
var v []float64 |
||||
if err := us.Scan(ctx, &v); err != nil { |
||||
return nil, err |
||||
} |
||||
return v, nil |
||||
} |
||||
|
||||
// Float64sX is like Float64s, but panics if an error occurs.
|
||||
func (us *UserSelect) Float64sX(ctx context.Context) []float64 { |
||||
v, err := us.Float64s(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Float64 returns a single float64 from a selector. It is only allowed when selecting one field.
|
||||
func (us *UserSelect) Float64(ctx context.Context) (_ float64, err error) { |
||||
var v []float64 |
||||
if v, err = us.Float64s(ctx); err != nil { |
||||
return |
||||
} |
||||
switch len(v) { |
||||
case 1: |
||||
return v[0], nil |
||||
case 0: |
||||
err = &NotFoundError{user.Label} |
||||
default: |
||||
err = fmt.Errorf("ent: UserSelect.Float64s returned %d results when one was expected", len(v)) |
||||
} |
||||
return |
||||
} |
||||
|
||||
// Float64X is like Float64, but panics if an error occurs.
|
||||
func (us *UserSelect) Float64X(ctx context.Context) float64 { |
||||
v, err := us.Float64(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Bools returns list of bools from a selector. It is only allowed when selecting one field.
|
||||
func (us *UserSelect) Bools(ctx context.Context) ([]bool, error) { |
||||
if len(us.fields) > 1 { |
||||
return nil, errors.New("ent: UserSelect.Bools is not achievable when selecting more than 1 field") |
||||
} |
||||
var v []bool |
||||
if err := us.Scan(ctx, &v); err != nil { |
||||
return nil, err |
||||
} |
||||
return v, nil |
||||
} |
||||
|
||||
// BoolsX is like Bools, but panics if an error occurs.
|
||||
func (us *UserSelect) BoolsX(ctx context.Context) []bool { |
||||
v, err := us.Bools(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
// Bool returns a single bool from a selector. It is only allowed when selecting one field.
|
||||
func (us *UserSelect) Bool(ctx context.Context) (_ bool, err error) { |
||||
var v []bool |
||||
if v, err = us.Bools(ctx); err != nil { |
||||
return |
||||
} |
||||
switch len(v) { |
||||
case 1: |
||||
return v[0], nil |
||||
case 0: |
||||
err = &NotFoundError{user.Label} |
||||
default: |
||||
err = fmt.Errorf("ent: UserSelect.Bools returned %d results when one was expected", len(v)) |
||||
} |
||||
return |
||||
} |
||||
|
||||
// BoolX is like Bool, but panics if an error occurs.
|
||||
func (us *UserSelect) BoolX(ctx context.Context) bool { |
||||
v, err := us.Bool(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return v |
||||
} |
||||
|
||||
func (us *UserSelect) sqlScan(ctx context.Context, v interface{}) error { |
||||
rows := &sql.Rows{} |
||||
query, args := us.sql.Query() |
||||
if err := us.driver.Query(ctx, query, args, rows); err != nil { |
||||
return err |
||||
} |
||||
defer rows.Close() |
||||
return sql.ScanSlice(rows, v) |
||||
} |
@ -0,0 +1,322 @@ |
||||
// Code generated by entc, DO NOT EDIT.
|
||||
|
||||
package ent |
||||
|
||||
import ( |
||||
"context" |
||||
"fmt" |
||||
|
||||
"entgo.io/ent/dialect/sql" |
||||
"entgo.io/ent/dialect/sql/sqlgraph" |
||||
"entgo.io/ent/schema/field" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/predicate" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/data/ent/user" |
||||
) |
||||
|
||||
// UserUpdate is the builder for updating User entities.
|
||||
type UserUpdate struct { |
||||
config |
||||
hooks []Hook |
||||
mutation *UserMutation |
||||
} |
||||
|
||||
// Where appends a list predicates to the UserUpdate builder.
|
||||
func (uu *UserUpdate) Where(ps ...predicate.User) *UserUpdate { |
||||
uu.mutation.Where(ps...) |
||||
return uu |
||||
} |
||||
|
||||
// SetName sets the "name" field.
|
||||
func (uu *UserUpdate) SetName(s string) *UserUpdate { |
||||
uu.mutation.SetName(s) |
||||
return uu |
||||
} |
||||
|
||||
// SetNillableName sets the "name" field if the given value is not nil.
|
||||
func (uu *UserUpdate) SetNillableName(s *string) *UserUpdate { |
||||
if s != nil { |
||||
uu.SetName(*s) |
||||
} |
||||
return uu |
||||
} |
||||
|
||||
// SetEmail sets the "email" field.
|
||||
func (uu *UserUpdate) SetEmail(s string) *UserUpdate { |
||||
uu.mutation.SetEmail(s) |
||||
return uu |
||||
} |
||||
|
||||
// SetNillableEmail sets the "email" field if the given value is not nil.
|
||||
func (uu *UserUpdate) SetNillableEmail(s *string) *UserUpdate { |
||||
if s != nil { |
||||
uu.SetEmail(*s) |
||||
} |
||||
return uu |
||||
} |
||||
|
||||
// Mutation returns the UserMutation object of the builder.
|
||||
func (uu *UserUpdate) Mutation() *UserMutation { |
||||
return uu.mutation |
||||
} |
||||
|
||||
// Save executes the query and returns the number of nodes affected by the update operation.
|
||||
func (uu *UserUpdate) Save(ctx context.Context) (int, error) { |
||||
var ( |
||||
err error |
||||
affected int |
||||
) |
||||
if len(uu.hooks) == 0 { |
||||
affected, err = uu.sqlSave(ctx) |
||||
} else { |
||||
var mut Mutator = MutateFunc(func(ctx context.Context, m Mutation) (Value, error) { |
||||
mutation, ok := m.(*UserMutation) |
||||
if !ok { |
||||
return nil, fmt.Errorf("unexpected mutation type %T", m) |
||||
} |
||||
uu.mutation = mutation |
||||
affected, err = uu.sqlSave(ctx) |
||||
mutation.done = true |
||||
return affected, err |
||||
}) |
||||
for i := len(uu.hooks) - 1; i >= 0; i-- { |
||||
if uu.hooks[i] == nil { |
||||
return 0, fmt.Errorf("ent: uninitialized hook (forgotten import ent/runtime?)") |
||||
} |
||||
mut = uu.hooks[i](mut) |
||||
} |
||||
if _, err := mut.Mutate(ctx, uu.mutation); err != nil { |
||||
return 0, err |
||||
} |
||||
} |
||||
return affected, err |
||||
} |
||||
|
||||
// SaveX is like Save, but panics if an error occurs.
|
||||
func (uu *UserUpdate) SaveX(ctx context.Context) int { |
||||
affected, err := uu.Save(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return affected |
||||
} |
||||
|
||||
// Exec executes the query.
|
||||
func (uu *UserUpdate) Exec(ctx context.Context) error { |
||||
_, err := uu.Save(ctx) |
||||
return err |
||||
} |
||||
|
||||
// ExecX is like Exec, but panics if an error occurs.
|
||||
func (uu *UserUpdate) ExecX(ctx context.Context) { |
||||
if err := uu.Exec(ctx); err != nil { |
||||
panic(err) |
||||
} |
||||
} |
||||
|
||||
func (uu *UserUpdate) sqlSave(ctx context.Context) (n int, err error) { |
||||
_spec := &sqlgraph.UpdateSpec{ |
||||
Node: &sqlgraph.NodeSpec{ |
||||
Table: user.Table, |
||||
Columns: user.Columns, |
||||
ID: &sqlgraph.FieldSpec{ |
||||
Type: field.TypeInt, |
||||
Column: user.FieldID, |
||||
}, |
||||
}, |
||||
} |
||||
if ps := uu.mutation.predicates; len(ps) > 0 { |
||||
_spec.Predicate = func(selector *sql.Selector) { |
||||
for i := range ps { |
||||
ps[i](selector) |
||||
} |
||||
} |
||||
} |
||||
if value, ok := uu.mutation.Name(); ok { |
||||
_spec.Fields.Set = append(_spec.Fields.Set, &sqlgraph.FieldSpec{ |
||||
Type: field.TypeString, |
||||
Value: value, |
||||
Column: user.FieldName, |
||||
}) |
||||
} |
||||
if value, ok := uu.mutation.Email(); ok { |
||||
_spec.Fields.Set = append(_spec.Fields.Set, &sqlgraph.FieldSpec{ |
||||
Type: field.TypeString, |
||||
Value: value, |
||||
Column: user.FieldEmail, |
||||
}) |
||||
} |
||||
if n, err = sqlgraph.UpdateNodes(ctx, uu.driver, _spec); err != nil { |
||||
if _, ok := err.(*sqlgraph.NotFoundError); ok { |
||||
err = &NotFoundError{user.Label} |
||||
} else if sqlgraph.IsConstraintError(err) { |
||||
err = &ConstraintError{err.Error(), err} |
||||
} |
||||
return 0, err |
||||
} |
||||
return n, nil |
||||
} |
||||
|
||||
// UserUpdateOne is the builder for updating a single User entity.
|
||||
type UserUpdateOne struct { |
||||
config |
||||
fields []string |
||||
hooks []Hook |
||||
mutation *UserMutation |
||||
} |
||||
|
||||
// SetName sets the "name" field.
|
||||
func (uuo *UserUpdateOne) SetName(s string) *UserUpdateOne { |
||||
uuo.mutation.SetName(s) |
||||
return uuo |
||||
} |
||||
|
||||
// SetNillableName sets the "name" field if the given value is not nil.
|
||||
func (uuo *UserUpdateOne) SetNillableName(s *string) *UserUpdateOne { |
||||
if s != nil { |
||||
uuo.SetName(*s) |
||||
} |
||||
return uuo |
||||
} |
||||
|
||||
// SetEmail sets the "email" field.
|
||||
func (uuo *UserUpdateOne) SetEmail(s string) *UserUpdateOne { |
||||
uuo.mutation.SetEmail(s) |
||||
return uuo |
||||
} |
||||
|
||||
// SetNillableEmail sets the "email" field if the given value is not nil.
|
||||
func (uuo *UserUpdateOne) SetNillableEmail(s *string) *UserUpdateOne { |
||||
if s != nil { |
||||
uuo.SetEmail(*s) |
||||
} |
||||
return uuo |
||||
} |
||||
|
||||
// Mutation returns the UserMutation object of the builder.
|
||||
func (uuo *UserUpdateOne) Mutation() *UserMutation { |
||||
return uuo.mutation |
||||
} |
||||
|
||||
// Select allows selecting one or more fields (columns) of the returned entity.
|
||||
// The default is selecting all fields defined in the entity schema.
|
||||
func (uuo *UserUpdateOne) Select(field string, fields ...string) *UserUpdateOne { |
||||
uuo.fields = append([]string{field}, fields...) |
||||
return uuo |
||||
} |
||||
|
||||
// Save executes the query and returns the updated User entity.
|
||||
func (uuo *UserUpdateOne) Save(ctx context.Context) (*User, error) { |
||||
var ( |
||||
err error |
||||
node *User |
||||
) |
||||
if len(uuo.hooks) == 0 { |
||||
node, err = uuo.sqlSave(ctx) |
||||
} else { |
||||
var mut Mutator = MutateFunc(func(ctx context.Context, m Mutation) (Value, error) { |
||||
mutation, ok := m.(*UserMutation) |
||||
if !ok { |
||||
return nil, fmt.Errorf("unexpected mutation type %T", m) |
||||
} |
||||
uuo.mutation = mutation |
||||
node, err = uuo.sqlSave(ctx) |
||||
mutation.done = true |
||||
return node, err |
||||
}) |
||||
for i := len(uuo.hooks) - 1; i >= 0; i-- { |
||||
if uuo.hooks[i] == nil { |
||||
return nil, fmt.Errorf("ent: uninitialized hook (forgotten import ent/runtime?)") |
||||
} |
||||
mut = uuo.hooks[i](mut) |
||||
} |
||||
if _, err := mut.Mutate(ctx, uuo.mutation); err != nil { |
||||
return nil, err |
||||
} |
||||
} |
||||
return node, err |
||||
} |
||||
|
||||
// SaveX is like Save, but panics if an error occurs.
|
||||
func (uuo *UserUpdateOne) SaveX(ctx context.Context) *User { |
||||
node, err := uuo.Save(ctx) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return node |
||||
} |
||||
|
||||
// Exec executes the query on the entity.
|
||||
func (uuo *UserUpdateOne) Exec(ctx context.Context) error { |
||||
_, err := uuo.Save(ctx) |
||||
return err |
||||
} |
||||
|
||||
// ExecX is like Exec, but panics if an error occurs.
|
||||
func (uuo *UserUpdateOne) ExecX(ctx context.Context) { |
||||
if err := uuo.Exec(ctx); err != nil { |
||||
panic(err) |
||||
} |
||||
} |
||||
|
||||
func (uuo *UserUpdateOne) sqlSave(ctx context.Context) (_node *User, err error) { |
||||
_spec := &sqlgraph.UpdateSpec{ |
||||
Node: &sqlgraph.NodeSpec{ |
||||
Table: user.Table, |
||||
Columns: user.Columns, |
||||
ID: &sqlgraph.FieldSpec{ |
||||
Type: field.TypeInt, |
||||
Column: user.FieldID, |
||||
}, |
||||
}, |
||||
} |
||||
id, ok := uuo.mutation.ID() |
||||
if !ok { |
||||
return nil, &ValidationError{Name: "ID", err: fmt.Errorf("missing User.ID for update")} |
||||
} |
||||
_spec.Node.ID.Value = id |
||||
if fields := uuo.fields; len(fields) > 0 { |
||||
_spec.Node.Columns = make([]string, 0, len(fields)) |
||||
_spec.Node.Columns = append(_spec.Node.Columns, user.FieldID) |
||||
for _, f := range fields { |
||||
if !user.ValidColumn(f) { |
||||
return nil, &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)} |
||||
} |
||||
if f != user.FieldID { |
||||
_spec.Node.Columns = append(_spec.Node.Columns, f) |
||||
} |
||||
} |
||||
} |
||||
if ps := uuo.mutation.predicates; len(ps) > 0 { |
||||
_spec.Predicate = func(selector *sql.Selector) { |
||||
for i := range ps { |
||||
ps[i](selector) |
||||
} |
||||
} |
||||
} |
||||
if value, ok := uuo.mutation.Name(); ok { |
||||
_spec.Fields.Set = append(_spec.Fields.Set, &sqlgraph.FieldSpec{ |
||||
Type: field.TypeString, |
||||
Value: value, |
||||
Column: user.FieldName, |
||||
}) |
||||
} |
||||
if value, ok := uuo.mutation.Email(); ok { |
||||
_spec.Fields.Set = append(_spec.Fields.Set, &sqlgraph.FieldSpec{ |
||||
Type: field.TypeString, |
||||
Value: value, |
||||
Column: user.FieldEmail, |
||||
}) |
||||
} |
||||
_node = &User{config: uuo.config} |
||||
_spec.Assign = _node.assignValues |
||||
_spec.ScanValues = _node.scanValues |
||||
if err = sqlgraph.UpdateNode(ctx, uuo.driver, _spec); err != nil { |
||||
if _, ok := err.(*sqlgraph.NotFoundError); ok { |
||||
err = &NotFoundError{user.Label} |
||||
} else if sqlgraph.IsConstraintError(err) { |
||||
err = &ConstraintError{err.Error(), err} |
||||
} |
||||
return nil, err |
||||
} |
||||
return _node, nil |
||||
} |
@ -0,0 +1,58 @@ |
||||
package data |
||||
|
||||
import ( |
||||
"context" |
||||
"strconv" |
||||
|
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/biz" |
||||
"github.com/go-kratos/kratos/v2/log" |
||||
) |
||||
|
||||
type userRepo struct { |
||||
data *Data |
||||
log *log.Helper |
||||
} |
||||
|
||||
type cardRepo struct { |
||||
data *Data |
||||
log *log.Helper |
||||
} |
||||
|
||||
func (u *userRepo) CreateUser(ctx context.Context, m *biz.User) (int, error) { |
||||
user, err := u.data.User(ctx). |
||||
Create(). |
||||
SetName(m.Name). |
||||
SetEmail(m.Email). |
||||
Save(ctx) |
||||
if err != nil { |
||||
return 0, err |
||||
} |
||||
return user.ID, nil |
||||
} |
||||
|
||||
// NewUserRepo .
|
||||
func NewUserRepo(data *Data, logger log.Logger) biz.UserRepo { |
||||
return &userRepo{ |
||||
data: data, |
||||
log: log.NewHelper(logger), |
||||
} |
||||
} |
||||
|
||||
func (c *cardRepo) CreateCard(ctx context.Context, id int) (int, error) { |
||||
card, err := c.data.Card(ctx). |
||||
Create(). |
||||
SetMoney("1000"). |
||||
SetUserID(strconv.Itoa(id)). |
||||
Save(ctx) |
||||
if err != nil { |
||||
return 0, err |
||||
} |
||||
return card.ID, nil |
||||
} |
||||
|
||||
func NewCardRepo(data *Data, logger log.Logger) biz.CardRepo { |
||||
return &cardRepo{ |
||||
data: data, |
||||
log: log.NewHelper(logger), |
||||
} |
||||
} |
@ -0,0 +1,37 @@ |
||||
package server |
||||
|
||||
import ( |
||||
v1 "github.com/go-kratos/kratos/examples/transaction/api/transaction/v1" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/conf" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/service" |
||||
"github.com/go-kratos/kratos/v2/log" |
||||
"github.com/go-kratos/kratos/v2/middleware/logging" |
||||
"github.com/go-kratos/kratos/v2/middleware/recovery" |
||||
"github.com/go-kratos/kratos/v2/middleware/tracing" |
||||
"github.com/go-kratos/kratos/v2/middleware/validate" |
||||
"github.com/go-kratos/kratos/v2/transport/grpc" |
||||
) |
||||
|
||||
// NewGRPCServer new a gRPC server.
|
||||
func NewGRPCServer(c *conf.Server, logger log.Logger, transaction *service.TransactionService) *grpc.Server { |
||||
opts := []grpc.ServerOption{ |
||||
grpc.Middleware( |
||||
recovery.Recovery(), |
||||
tracing.Server(), |
||||
logging.Server(logger), |
||||
validate.Validator(), |
||||
), |
||||
} |
||||
if c.Grpc.Network != "" { |
||||
opts = append(opts, grpc.Network(c.Grpc.Network)) |
||||
} |
||||
if c.Grpc.Addr != "" { |
||||
opts = append(opts, grpc.Address(c.Grpc.Addr)) |
||||
} |
||||
if c.Grpc.Timeout != nil { |
||||
opts = append(opts, grpc.Timeout(c.Grpc.Timeout.AsDuration())) |
||||
} |
||||
srv := grpc.NewServer(opts...) |
||||
v1.RegisterTransactionServiceServer(srv, transaction) |
||||
return srv |
||||
} |
@ -0,0 +1,37 @@ |
||||
package server |
||||
|
||||
import ( |
||||
v1 "github.com/go-kratos/kratos/examples/transaction/api/transaction/v1" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/conf" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/service" |
||||
"github.com/go-kratos/kratos/v2/log" |
||||
"github.com/go-kratos/kratos/v2/middleware/logging" |
||||
"github.com/go-kratos/kratos/v2/middleware/recovery" |
||||
"github.com/go-kratos/kratos/v2/middleware/tracing" |
||||
"github.com/go-kratos/kratos/v2/middleware/validate" |
||||
"github.com/go-kratos/kratos/v2/transport/http" |
||||
) |
||||
|
||||
// NewHTTPServer new a HTTP server.
|
||||
func NewHTTPServer(c *conf.Server, logger log.Logger, transaction *service.TransactionService) *http.Server { |
||||
opts := []http.ServerOption{ |
||||
http.Middleware( |
||||
recovery.Recovery(), |
||||
tracing.Server(), |
||||
logging.Server(logger), |
||||
validate.Validator(), |
||||
), |
||||
} |
||||
if c.Http.Network != "" { |
||||
opts = append(opts, http.Network(c.Http.Network)) |
||||
} |
||||
if c.Http.Addr != "" { |
||||
opts = append(opts, http.Address(c.Http.Addr)) |
||||
} |
||||
if c.Http.Timeout != nil { |
||||
opts = append(opts, http.Timeout(c.Http.Timeout.AsDuration())) |
||||
} |
||||
srv := http.NewServer(opts...) |
||||
v1.RegisterTransactionServiceHTTPServer(srv, transaction) |
||||
return srv |
||||
} |
@ -0,0 +1,8 @@ |
||||
package server |
||||
|
||||
import ( |
||||
"github.com/google/wire" |
||||
) |
||||
|
||||
// ProviderSet is server providers.
|
||||
var ProviderSet = wire.NewSet(NewHTTPServer, NewGRPCServer) |
@ -0,0 +1 @@ |
||||
# Service |
@ -0,0 +1,20 @@ |
||||
package service |
||||
|
||||
import ( |
||||
pb "github.com/go-kratos/kratos/examples/transaction/api/transaction/v1" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/biz" |
||||
|
||||
"github.com/go-kratos/kratos/v2/log" |
||||
"github.com/google/wire" |
||||
) |
||||
|
||||
// ProviderSet is service providers.
|
||||
var ProviderSet = wire.NewSet(NewTransactionService) |
||||
|
||||
type TransactionService struct { |
||||
pb.UnimplementedTransactionServiceServer |
||||
|
||||
log *log.Helper |
||||
|
||||
user *biz.UserUsecase |
||||
} |
@ -0,0 +1,31 @@ |
||||
package service |
||||
|
||||
import ( |
||||
"context" |
||||
"strconv" |
||||
|
||||
pb "github.com/go-kratos/kratos/examples/transaction/api/transaction/v1" |
||||
"github.com/go-kratos/kratos/examples/transaction/ent/internal/biz" |
||||
|
||||
"github.com/go-kratos/kratos/v2/log" |
||||
) |
||||
|
||||
func NewTransactionService(user *biz.UserUsecase, logger log.Logger) *TransactionService { |
||||
return &TransactionService{ |
||||
user: user, |
||||
log: log.NewHelper(logger), |
||||
} |
||||
} |
||||
|
||||
func (b *TransactionService) CreateUser(ctx context.Context, in *pb.CreateUserRequest) (*pb.CreateUserReply, error) { |
||||
id, err := b.user.CreateUser(ctx, &biz.User{ |
||||
Name: in.Name, |
||||
Email: in.Email, |
||||
}) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
return &pb.CreateUserReply{ |
||||
Id: strconv.Itoa(id), |
||||
}, nil |
||||
} |
@ -0,0 +1,126 @@ |
||||
# Generated with protoc-gen-openapi |
||||
# https://github.com/google/gnostic/tree/master/apps/protoc-gen-openapi |
||||
|
||||
openapi: 3.0.3 |
||||
info: |
||||
title: BlogService |
||||
version: 0.0.1 |
||||
paths: |
||||
/v1/article/: |
||||
get: |
||||
operationId: BlogService_ListArticle |
||||
responses: |
||||
"200": |
||||
description: OK |
||||
content: |
||||
application/json: |
||||
schema: |
||||
$ref: '#/components/schemas/ListArticleReply' |
||||
post: |
||||
operationId: BlogService_CreateArticle |
||||
requestBody: |
||||
content: |
||||
application/json: |
||||
schema: |
||||
$ref: '#/components/schemas/CreateArticleRequest' |
||||
required: true |
||||
responses: |
||||
"200": |
||||
description: OK |
||||
content: |
||||
application/json: |
||||
schema: |
||||
$ref: '#/components/schemas/CreateArticleReply' |
||||
/v1/article/{id}: |
||||
get: |
||||
operationId: BlogService_GetArticle |
||||
parameters: |
||||
- name: id |
||||
in: query |
||||
schema: |
||||
type: string |
||||
responses: |
||||
"200": |
||||
description: OK |
||||
content: |
||||
application/json: |
||||
schema: |
||||
$ref: '#/components/schemas/GetArticleReply' |
||||
put: |
||||
operationId: BlogService_UpdateArticle |
||||
requestBody: |
||||
content: |
||||
application/json: |
||||
schema: |
||||
$ref: '#/components/schemas/UpdateArticleRequest' |
||||
required: true |
||||
responses: |
||||
"200": |
||||
description: OK |
||||
content: |
||||
application/json: |
||||
schema: |
||||
$ref: '#/components/schemas/UpdateArticleReply' |
||||
delete: |
||||
operationId: BlogService_DeleteArticle |
||||
parameters: |
||||
- name: id |
||||
in: query |
||||
schema: |
||||
type: string |
||||
responses: |
||||
"200": |
||||
description: OK |
||||
content: |
||||
application/json: |
||||
schema: |
||||
$ref: '#/components/schemas/DeleteArticleReply' |
||||
components: |
||||
schemas: |
||||
Article: |
||||
properties: |
||||
id: |
||||
type: integer |
||||
format: int64 |
||||
title: |
||||
type: string |
||||
content: |
||||
type: string |
||||
like: |
||||
type: integer |
||||
format: int64 |
||||
CreateArticleReply: |
||||
properties: |
||||
Article: |
||||
$ref: '#/components/schemas/Article' |
||||
CreateArticleRequest: |
||||
properties: |
||||
title: |
||||
type: string |
||||
content: |
||||
type: string |
||||
DeleteArticleReply: |
||||
properties: {} |
||||
GetArticleReply: |
||||
properties: |
||||
Article: |
||||
$ref: '#/components/schemas/Article' |
||||
ListArticleReply: |
||||
properties: |
||||
results: |
||||
type: array |
||||
items: |
||||
$ref: '#/components/schemas/Article' |
||||
UpdateArticleReply: |
||||
properties: |
||||
Article: |
||||
$ref: '#/components/schemas/Article' |
||||
UpdateArticleRequest: |
||||
properties: |
||||
id: |
||||
type: integer |
||||
format: int64 |
||||
title: |
||||
type: string |
||||
content: |
||||
type: string |
@ -0,0 +1,42 @@ |
||||
# Generated with protoc-gen-openapi |
||||
# https://github.com/google/gnostic/tree/master/apps/protoc-gen-openapi |
||||
|
||||
openapi: 3.0.3 |
||||
info: |
||||
title: TransactionService API |
||||
version: 0.0.1 |
||||
paths: |
||||
/v1/transaction: |
||||
post: |
||||
tags: |
||||
- TransactionService |
||||
operationId: TransactionService_CreateUser |
||||
requestBody: |
||||
content: |
||||
application/json: |
||||
schema: |
||||
$ref: '#/components/schemas/CreateUserRequest' |
||||
required: true |
||||
responses: |
||||
"200": |
||||
description: OK |
||||
content: |
||||
application/json: |
||||
schema: |
||||
$ref: '#/components/schemas/CreateUserReply' |
||||
components: |
||||
schemas: |
||||
CreateUserReply: |
||||
type: object |
||||
properties: |
||||
id: |
||||
type: string |
||||
CreateUserRequest: |
||||
type: object |
||||
properties: |
||||
name: |
||||
type: string |
||||
email: |
||||
type: string |
||||
tags: |
||||
- name: TransactionService |
Loading…
Reference in new issue