[Feature] Alembic support

contrib/starrocks-python-client/setup.py

@@ -57,6 +57,7 @@
         "sqlalchemy>=2.0",
         "sqlalchemy-utils>=0.41.2",
         "pymysql>=1.1.0",
+        "alembic>=1.4.0"
     ],
     setup_requires=["pytest-runner"],
     tests_require=["pytest", "mock"],

contrib/starrocks-python-client/starrocks/__init__.py

@@ -14,4 +14,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = "1.1.1"
+__version__ = "1.2.0"

contrib/starrocks-python-client/starrocks/alembic.py

@@ -0,0 +1,28 @@
+from alembic.ddl.mysql import MySQLImpl
+from sqlalchemy import Column
+from sqlalchemy import BIGINT
+from sqlalchemy import MetaData
+from sqlalchemy import Table
+from sqlalchemy import VARCHAR
+from typing import Optional
+
+
+class StarrocksImpl(MySQLImpl):
+    __dialect__ = "starrocks"
+
+    def version_table_impl(
+        self,
+        *,
+        version_table: str,
+        version_table_schema: Optional[str],
+        version_table_pk: bool,
+        **kw,
+    ) -> Table:
+        return Table(
+            "alembic_version",
+            MetaData(),
+            Column("id", BIGINT, autoincrement=True, primary_key=True),
+            Column("version_num", VARCHAR(32), primary_key=False),
+            starrocks_primary_key="id",
+        )
+