View Javadoc
1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one or more
3    * contributor license agreements.  See the NOTICE file distributed with
4    * this work for additional information regarding copyright ownership.
5    * The ASF licenses this file to You under the Apache License, Version 2.0
6    * (the "License"); you may not use this file except in compliance with
7    * the License.  You may obtain a copy of the License at
8    *
9    *     http://www.apache.org/licenses/LICENSE-2.0
10   *
11   * Unless required by applicable law or agreed to in writing, software
12   * distributed under the License is distributed on an "AS IS" BASIS,
13   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14   * See the License for the specific language governing permissions and
15   * limitations under the License.
16   */
17  
18  package org.apache.shardingsphere.sql.parser.sql.common.segment.ddl.constraint;
19  
20  import lombok.Getter;
21  import lombok.RequiredArgsConstructor;
22  import lombok.Setter;
23  import org.apache.shardingsphere.sql.parser.sql.common.segment.ddl.AlterDefinitionSegment;
24  import org.apache.shardingsphere.sql.parser.sql.common.segment.ddl.CreateDefinitionSegment;
25  import org.apache.shardingsphere.sql.parser.sql.common.segment.ddl.index.IndexSegment;
26  import org.apache.shardingsphere.sql.parser.sql.common.segment.dml.column.ColumnSegment;
27  import org.apache.shardingsphere.sql.parser.sql.common.segment.generic.table.SimpleTableSegment;
28  
29  import java.util.Collection;
30  import java.util.LinkedList;
31  import java.util.Optional;
32  
33  /**
34   * Constraint definition segment.
35   */
36  @RequiredArgsConstructor
37  @Getter
38  @Setter
39  public final class ConstraintDefinitionSegment implements CreateDefinitionSegment, AlterDefinitionSegment {
40      
41      private final int startIndex;
42      
43      private final int stopIndex;
44      
45      private final Collection<ColumnSegment> primaryKeyColumns = new LinkedList<>();
46      
47      private final Collection<ColumnSegment> indexColumns = new LinkedList<>();
48      
49      private ConstraintSegment constraintName;
50      
51      private IndexSegment indexName;
52      
53      private SimpleTableSegment referencedTable;
54      
55      private boolean uniqueKey;
56      
57      private boolean primaryKey;
58      
59      /**
60       * Get constraint name.
61       *
62       * @return constraint name.
63       */
64      public Optional<ConstraintSegment> getConstraintName() {
65          return Optional.ofNullable(constraintName);
66      }
67      
68      /**
69       * Get index name.
70       *
71       * @return index name
72       */
73      public Optional<IndexSegment> getIndexName() {
74          return Optional.ofNullable(indexName);
75      }
76      
77      /**
78       * Get referenced table.
79       *
80       * @return referenced table
81       */
82      public Optional<SimpleTableSegment> getReferencedTable() {
83          return Optional.ofNullable(referencedTable);
84      }
85  }