1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
open Core
open Async
let json_arg = Command.Arg_type.create Yojson.Safe.from_string
let call ?endpoint_url ?profile ?region f m result_to_json error_to_json =
let region =
match region with
| Some region -> Some (Awso.Region.of_string region)
| None -> None in
(Awso_async.Cfg.get_exn ?profile ?region ()) >>=
(fun cfg ->
(f ?endpoint_url ?cfg:(Some cfg) m) >>=
(fun result ->
match result with
| Error err ->
(match error_to_json with
| None ->
failwithf
"endpoint error, but no error values defined in boto"
()
| Some to_json ->
let s = (err |> to_json) |> Yojson.Safe.to_string in
failwithf "AWS error: %s" s ())
| Ok result ->
((match result_to_json with
| None -> print_endline "ok response from endpoint"
| Some to_json ->
((result |> to_json) |> Yojson.Safe.to_string) |>
print_endline);
return ())))
let batch_execute_statement =
Command.async ~summary:""
([%map_open.Command
let cli_profile =
flag "-cli-profile" (optional string) ~doc:"NAME aws profile to use"
and cli_region =
flag "-cli-region" (optional string) ~doc:"REGION override region"
and endpoint_url =
flag "-endpoint-url" (optional string)
~doc:"URL override endpoint url"
and database = flag "database" (optional string) ~doc:"STRING DbName"
and schema = flag "schema" (optional string) ~doc:"STRING DbName"
and parameterSets =
flag "parameter-sets" (optional json_arg)
~doc:"JSON SqlParameterSets"
and transactionId =
flag "transaction-id" (optional string) ~doc:"STRING Id"
and resourceArn =
flag "resource-arn" (required string) ~doc:"STRING Arn"
and secretArn = flag "secret-arn" (required string) ~doc:"STRING Arn"
and sql = flag "sql" (required string) ~doc:"STRING SqlStatement" in
fun () ->
call ?endpoint_url ?profile:cli_profile ?region:cli_region
Io.batch_execute_statement
(Values.BatchExecuteStatementRequest.make ?database ?schema
?parameterSets:(Option.map ~f:Values.SqlParameterSets.of_json
parameterSets) ?transactionId ~resourceArn
~secretArn ~sql ())
(Some Values.BatchExecuteStatementResponse.to_json)
(Some Values.BatchExecuteStatementResponse.error_to_json)])
let begin_transaction =
Command.async ~summary:""
([%map_open.Command
let cli_profile =
flag "-cli-profile" (optional string) ~doc:"NAME aws profile to use"
and cli_region =
flag "-cli-region" (optional string) ~doc:"REGION override region"
and endpoint_url =
flag "-endpoint-url" (optional string)
~doc:"URL override endpoint url"
and database = flag "database" (optional string) ~doc:"STRING DbName"
and schema = flag "schema" (optional string) ~doc:"STRING DbName"
and resourceArn =
flag "resource-arn" (required string) ~doc:"STRING Arn"
and secretArn = flag "secret-arn" (required string) ~doc:"STRING Arn" in
fun () ->
call ?endpoint_url ?profile:cli_profile ?region:cli_region
Io.begin_transaction
(Values.BeginTransactionRequest.make ?database ?schema
~resourceArn ~secretArn ())
(Some Values.BeginTransactionResponse.to_json)
(Some Values.BeginTransactionResponse.error_to_json)])
let commit_transaction =
Command.async ~summary:""
([%map_open.Command
let cli_profile =
flag "-cli-profile" (optional string) ~doc:"NAME aws profile to use"
and cli_region =
flag "-cli-region" (optional string) ~doc:"REGION override region"
and endpoint_url =
flag "-endpoint-url" (optional string)
~doc:"URL override endpoint url"
and resourceArn =
flag "resource-arn" (required string) ~doc:"STRING Arn"
and secretArn = flag "secret-arn" (required string) ~doc:"STRING Arn"
and transactionId =
flag "transaction-id" (required string) ~doc:"STRING Id" in
fun () ->
call ?endpoint_url ?profile:cli_profile ?region:cli_region
Io.commit_transaction
(Values.CommitTransactionRequest.make ~resourceArn ~secretArn
~transactionId ())
(Some Values.CommitTransactionResponse.to_json)
(Some Values.CommitTransactionResponse.error_to_json)])
let execute_sql =
Command.async ~summary:""
([%map_open.Command
let cli_profile =
flag "-cli-profile" (optional string) ~doc:"NAME aws profile to use"
and cli_region =
flag "-cli-region" (optional string) ~doc:"REGION override region"
and endpoint_url =
flag "-endpoint-url" (optional string)
~doc:"URL override endpoint url"
and database = flag "database" (optional string) ~doc:"STRING DbName"
and schema = flag "schema" (optional string) ~doc:"STRING DbName"
and dbClusterOrInstanceArn =
flag "db-cluster-or-instance-arn" (required string)
~doc:"STRING Arn"
and awsSecretStoreArn =
flag "aws-secret-store-arn" (required string) ~doc:"STRING Arn"
and sqlStatements =
flag "sql-statements" (required string) ~doc:"STRING SqlStatement" in
fun () ->
call ?endpoint_url ?profile:cli_profile ?region:cli_region
Io.execute_sql
(Values.ExecuteSqlRequest.make ?database ?schema
~dbClusterOrInstanceArn ~awsSecretStoreArn ~sqlStatements ())
(Some Values.ExecuteSqlResponse.to_json)
(Some Values.ExecuteSqlResponse.error_to_json)])
let execute_statement =
Command.async ~summary:""
([%map_open.Command
let cli_profile =
flag "-cli-profile" (optional string) ~doc:"NAME aws profile to use"
and cli_region =
flag "-cli-region" (optional string) ~doc:"REGION override region"
and endpoint_url =
flag "-endpoint-url" (optional string)
~doc:"URL override endpoint url"
and database = flag "database" (optional string) ~doc:"STRING DbName"
and schema = flag "schema" (optional string) ~doc:"STRING DbName"
and parameters =
flag "parameters" (optional json_arg) ~doc:"JSON SqlParametersList"
and transactionId =
flag "transaction-id" (optional string) ~doc:"STRING Id"
and includeResultMetadata =
flag "include-result-metadata" (optional bool) ~doc:"BOOL Boolean"
and continueAfterTimeout =
flag "continue-after-timeout" (optional bool) ~doc:"BOOL Boolean"
and resultSetOptions =
flag "result-set-options" (optional json_arg)
~doc:"JSON ResultSetOptions"
and formatRecordsAs =
flag "format-records-as" (optional json_arg)
~doc:"JSON RecordsFormatType"
and resourceArn =
flag "resource-arn" (required string) ~doc:"STRING Arn"
and secretArn = flag "secret-arn" (required string) ~doc:"STRING Arn"
and sql = flag "sql" (required string) ~doc:"STRING SqlStatement" in
fun () ->
call ?endpoint_url ?profile:cli_profile ?region:cli_region
Io.execute_statement
(Values.ExecuteStatementRequest.make ?database ?schema
?parameters:(Option.map ~f:Values.SqlParametersList.of_json
parameters) ?transactionId
?includeResultMetadata ?continueAfterTimeout
?resultSetOptions:(Option.map
~f:Values.ResultSetOptions.of_json
resultSetOptions)
?formatRecordsAs:(Option.map
~f:Values.RecordsFormatType.of_json
formatRecordsAs) ~resourceArn ~secretArn
~sql ()) (Some Values.ExecuteStatementResponse.to_json)
(Some Values.ExecuteStatementResponse.error_to_json)])
let rollback_transaction =
Command.async ~summary:""
([%map_open.Command
let cli_profile =
flag "-cli-profile" (optional string) ~doc:"NAME aws profile to use"
and cli_region =
flag "-cli-region" (optional string) ~doc:"REGION override region"
and endpoint_url =
flag "-endpoint-url" (optional string)
~doc:"URL override endpoint url"
and resourceArn =
flag "resource-arn" (required string) ~doc:"STRING Arn"
and secretArn = flag "secret-arn" (required string) ~doc:"STRING Arn"
and transactionId =
flag "transaction-id" (required string) ~doc:"STRING Id" in
fun () ->
call ?endpoint_url ?profile:cli_profile ?region:cli_region
Io.rollback_transaction
(Values.RollbackTransactionRequest.make ~resourceArn ~secretArn
~transactionId ())
(Some Values.RollbackTransactionResponse.to_json)
(Some Values.RollbackTransactionResponse.error_to_json)])
let main =
Command.group
~summary:((Awso.Service.to_string Values.service) ^ " commands")
[("batch-execute-statement", batch_execute_statement);
("begin-transaction", begin_transaction);
("commit-transaction", commit_transaction);
("execute-sql", execute_sql);
("execute-statement", execute_statement);
("rollback-transaction", rollback_transaction)]