Make TransactionBuilder type-aware
[ovsdb.git] / library / impl / src / main / java / org / opendaylight / ovsdb / lib / operations / TransactionBuilder.java
1 /*
2  * Copyright © 2014, 2017 EBay Software Foundation and others. All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8 package org.opendaylight.ovsdb.lib.operations;
9
10 import static java.util.Objects.requireNonNull;
11
12 import com.google.common.util.concurrent.ListenableFuture;
13 import java.util.ArrayList;
14 import java.util.List;
15 import org.eclipse.jdt.annotation.NonNull;
16 import org.opendaylight.ovsdb.lib.OvsdbClient;
17 import org.opendaylight.ovsdb.lib.schema.DatabaseSchema;
18 import org.opendaylight.ovsdb.lib.schema.typed.TyperUtils;
19
20 public class TransactionBuilder {
21     private final List<Operation> operations = new ArrayList<>();
22     private final @NonNull DatabaseSchema databaseSchema;
23     private final OvsdbClient ovs;
24
25     public TransactionBuilder(final OvsdbClient ovs, final DatabaseSchema schema) {
26         this.ovs = ovs;
27         this.databaseSchema = requireNonNull(schema);
28     }
29
30     public List<Operation> getOperations() {
31         return operations;
32     }
33
34     public TransactionBuilder add(final Operation operation) {
35         operations.add(operation);
36         return this;
37     }
38
39     public List<Operation> build() {
40         return operations;
41     }
42
43     public ListenableFuture<List<OperationResult>> execute() {
44         return ovs.transact(databaseSchema, operations);
45     }
46
47     public @NonNull DatabaseSchema getDatabaseSchema() {
48         return databaseSchema;
49     }
50
51     public <T> T getTypedRowWrapper(final Class<T> klazz) {
52         return TyperUtils.getTypedRowWrapper(databaseSchema, klazz);
53     }
54
55     public <T> T getTypedRowSchema(final Class<T> klazz) {
56         return TyperUtils.getTypedRowWrapper(databaseSchema, klazz, null);
57     }
58 }