1// Code generated by smithy-go-codegen DO NOT EDIT.
2
3package personalize
4
5import (
6	"context"
7	awsmiddleware "github.com/aws/aws-sdk-go-v2/aws/middleware"
8	"github.com/aws/aws-sdk-go-v2/aws/signer/v4"
9	"github.com/aws/smithy-go/middleware"
10	smithyhttp "github.com/aws/smithy-go/transport/http"
11)
12
13// Creates an empty dataset and adds it to the specified dataset group. Use
14// CreateDatasetImportJob to import your training data to a dataset. There are
15// three types of datasets:
16//
17// * Interactions
18//
19// * Items
20//
21// * Users
22//
23// Each dataset type
24// has an associated schema with required field types. Only the Interactions
25// dataset is required in order to train a model (also referred to as creating a
26// solution). A dataset can be in one of the following states:
27//
28// * CREATE PENDING >
29// CREATE IN_PROGRESS > ACTIVE -or- CREATE FAILED
30//
31// * DELETE PENDING > DELETE
32// IN_PROGRESS
33//
34// To get the status of the dataset, call DescribeDataset. Related
35// APIs
36//
37// * CreateDatasetGroup
38//
39// * ListDatasets
40//
41// * DescribeDataset
42//
43// * DeleteDataset
44func (c *Client) CreateDataset(ctx context.Context, params *CreateDatasetInput, optFns ...func(*Options)) (*CreateDatasetOutput, error) {
45	if params == nil {
46		params = &CreateDatasetInput{}
47	}
48
49	result, metadata, err := c.invokeOperation(ctx, "CreateDataset", params, optFns, addOperationCreateDatasetMiddlewares)
50	if err != nil {
51		return nil, err
52	}
53
54	out := result.(*CreateDatasetOutput)
55	out.ResultMetadata = metadata
56	return out, nil
57}
58
59type CreateDatasetInput struct {
60
61	// The Amazon Resource Name (ARN) of the dataset group to add the dataset to.
62	//
63	// This member is required.
64	DatasetGroupArn *string
65
66	// The type of dataset. One of the following (case insensitive) values:
67	//
68	// *
69	// Interactions
70	//
71	// * Items
72	//
73	// * Users
74	//
75	// This member is required.
76	DatasetType *string
77
78	// The name for the dataset.
79	//
80	// This member is required.
81	Name *string
82
83	// The ARN of the schema to associate with the dataset. The schema defines the
84	// dataset fields.
85	//
86	// This member is required.
87	SchemaArn *string
88}
89
90type CreateDatasetOutput struct {
91
92	// The ARN of the dataset.
93	DatasetArn *string
94
95	// Metadata pertaining to the operation's result.
96	ResultMetadata middleware.Metadata
97}
98
99func addOperationCreateDatasetMiddlewares(stack *middleware.Stack, options Options) (err error) {
100	err = stack.Serialize.Add(&awsAwsjson11_serializeOpCreateDataset{}, middleware.After)
101	if err != nil {
102		return err
103	}
104	err = stack.Deserialize.Add(&awsAwsjson11_deserializeOpCreateDataset{}, middleware.After)
105	if err != nil {
106		return err
107	}
108	if err = addSetLoggerMiddleware(stack, options); err != nil {
109		return err
110	}
111	if err = awsmiddleware.AddClientRequestIDMiddleware(stack); err != nil {
112		return err
113	}
114	if err = smithyhttp.AddComputeContentLengthMiddleware(stack); err != nil {
115		return err
116	}
117	if err = addResolveEndpointMiddleware(stack, options); err != nil {
118		return err
119	}
120	if err = v4.AddComputePayloadSHA256Middleware(stack); err != nil {
121		return err
122	}
123	if err = addRetryMiddlewares(stack, options); err != nil {
124		return err
125	}
126	if err = addHTTPSignerV4Middleware(stack, options); err != nil {
127		return err
128	}
129	if err = awsmiddleware.AddRawResponseToMetadata(stack); err != nil {
130		return err
131	}
132	if err = awsmiddleware.AddRecordResponseTiming(stack); err != nil {
133		return err
134	}
135	if err = addClientUserAgent(stack); err != nil {
136		return err
137	}
138	if err = smithyhttp.AddErrorCloseResponseBodyMiddleware(stack); err != nil {
139		return err
140	}
141	if err = smithyhttp.AddCloseResponseBodyMiddleware(stack); err != nil {
142		return err
143	}
144	if err = addOpCreateDatasetValidationMiddleware(stack); err != nil {
145		return err
146	}
147	if err = stack.Initialize.Add(newServiceMetadataMiddleware_opCreateDataset(options.Region), middleware.Before); err != nil {
148		return err
149	}
150	if err = addRequestIDRetrieverMiddleware(stack); err != nil {
151		return err
152	}
153	if err = addResponseErrorMiddleware(stack); err != nil {
154		return err
155	}
156	if err = addRequestResponseLogging(stack, options); err != nil {
157		return err
158	}
159	return nil
160}
161
162func newServiceMetadataMiddleware_opCreateDataset(region string) *awsmiddleware.RegisterServiceMetadata {
163	return &awsmiddleware.RegisterServiceMetadata{
164		Region:        region,
165		ServiceID:     ServiceID,
166		SigningName:   "personalize",
167		OperationName: "CreateDataset",
168	}
169}
170