001package ca.uhn.fhir.jpa.migrate.taskdef;
002
003/*-
004 * #%L
005 * HAPI FHIR Server - SQL Migration
006 * %%
007 * Copyright (C) 2014 - 2023 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        private String myTableName;
031
032
033        public BaseTableTask(String theProductVersion, String theSchemaVersion) {
034                super(theProductVersion, theSchemaVersion);
035        }
036
037        public String getTableName() {
038                return myTableName;
039        }
040
041        public BaseTableTask setTableName(String theTableName) {
042                Validate.notBlank(theTableName);
043                myTableName = theTableName;
044                return this;
045        }
046
047        @Override
048        public void validate() {
049                Validate.notBlank(myTableName);
050        }
051
052        @Override
053        protected void generateEquals(EqualsBuilder theBuilder, BaseTask theOtherObject) {
054                BaseTableTask otherObject = (BaseTableTask) theOtherObject;
055                theBuilder.append(myTableName, otherObject.myTableName);
056        }
057
058        protected String getSqlType(ColumnTypeEnum theColumnType, Long theColumnLength) {
059                String retVal = ColumnTypeToDriverTypeToSqlType.getColumnTypeToDriverTypeToSqlType().get(theColumnType).get(getDriverType());
060                Objects.requireNonNull(retVal);
061
062                if (theColumnType == ColumnTypeEnum.STRING) {
063                        retVal = retVal.replace("?", Long.toString(theColumnLength));
064                }
065
066                return retVal;
067        }
068
069        @Override
070        protected void generateHashCode(HashCodeBuilder theBuilder) {
071                theBuilder.append(myTableName);
072        }
073}