001package ca.uhn.fhir.jpa.migrate.taskdef;
002
003/*-
004 * #%L
005 * HAPI FHIR Server - SQL Migration
006 * %%
007 * Copyright (C) 2014 - 2022 Smile CDR, Inc.
008 * %%
009 * Licensed under the Apache License, Version 2.0 (the "License");
010 * you may not use this file except in compliance with the License.
011 * You may obtain a copy of the License at
012 *
013 *      http://www.apache.org/licenses/LICENSE-2.0
014 *
015 * Unless required by applicable law or agreed to in writing, software
016 * distributed under the License is distributed on an "AS IS" BASIS,
017 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
018 * See the License for the specific language governing permissions and
019 * limitations under the License.
020 * #L%
021 */
022
023import org.apache.commons.lang3.Validate;
024import org.apache.commons.lang3.builder.EqualsBuilder;
025import org.apache.commons.lang3.builder.HashCodeBuilder;
026
027import java.util.Objects;
028
029public abstract class BaseTableTask extends BaseTask {
030        protected final ColumnTypeToDriverTypeToSqlType myColumnTypeToDriverTypeToSqlType = new ColumnTypeToDriverTypeToSqlType();
031        private String myTableName;
032
033
034        public BaseTableTask(String theProductVersion, String theSchemaVersion) {
035                super(theProductVersion, theSchemaVersion);
036        }
037
038        public String getTableName() {
039                return myTableName;
040        }
041
042        public BaseTableTask setTableName(String theTableName) {
043                Validate.notBlank(theTableName);
044                myTableName = theTableName;
045                return this;
046        }
047
048        @Override
049        public void validate() {
050                Validate.notBlank(myTableName);
051        }
052
053        @Override
054        protected void generateEquals(EqualsBuilder theBuilder, BaseTask theOtherObject) {
055                BaseTableTask otherObject = (BaseTableTask) theOtherObject;
056                theBuilder.append(myTableName, otherObject.myTableName);
057        }
058
059        protected String getSqlType(ColumnTypeEnum theColumnType, Long theColumnLength) {
060                String retVal = myColumnTypeToDriverTypeToSqlType.getColumnTypeToDriverTypeToSqlType().get(theColumnType).get(getDriverType());
061                Objects.requireNonNull(retVal);
062
063                if (theColumnType == ColumnTypeEnum.STRING) {
064                        retVal = retVal.replace("?", Long.toString(theColumnLength));
065                }
066
067                return retVal;
068        }
069
070        @Override
071        protected void generateHashCode(HashCodeBuilder theBuilder) {
072                theBuilder.append(myTableName);
073        }
074}