feat: sqlserver增加for update支持 (#64)

hsweb-easy-orm-rdb/src/main/java/org/hswebframework/ezorm/rdb/supports/mssql/SqlServerQuerySqlBuilder.java

@@ -0,0 +1,63 @@
+package org.hswebframework.ezorm.rdb.supports.mssql;
+
+import org.hswebframework.ezorm.rdb.executor.SqlRequest;
+import org.hswebframework.ezorm.rdb.metadata.RDBFeatureType;
+import org.hswebframework.ezorm.rdb.metadata.RDBSchemaMetadata;
+import org.hswebframework.ezorm.rdb.metadata.TableOrViewMetadata;
+import org.hswebframework.ezorm.rdb.operator.builder.DefaultQuerySqlBuilder;
+import org.hswebframework.ezorm.rdb.operator.builder.FragmentBlock;
+import org.hswebframework.ezorm.rdb.operator.builder.Paginator;
+import org.hswebframework.ezorm.rdb.operator.builder.fragments.BlockSqlFragments;
+import org.hswebframework.ezorm.rdb.operator.builder.fragments.PrepareSqlFragments;
+import org.hswebframework.ezorm.rdb.operator.builder.fragments.SqlFragments;
+import org.hswebframework.ezorm.rdb.operator.dml.query.QueryOperatorParameter;
+
+class SqlServerQuerySqlBuilder extends DefaultQuerySqlBuilder {
+    public SqlServerQuerySqlBuilder(RDBSchemaMetadata schema) {
+        super(schema);
+    }
+
+    protected SqlRequest build(TableOrViewMetadata metadata, QueryOperatorParameter parameter) {
+        BlockSqlFragments fragments = BlockSqlFragments.of();
+
+        fragments.addBlock(FragmentBlock.before, "select");
+
+        fragments.addBlock(FragmentBlock.selectColumn, select(parameter, metadata)
+                .orElseGet(() -> PrepareSqlFragments.of().addSql("*")));
+
+        fragments.addBlock(FragmentBlock.selectFrom, from(metadata, parameter));
+
+
+        if (Boolean.TRUE.equals(parameter.getForUpdate())) {
+            fragments.addBlock(FragmentBlock.selectFrom, PrepareSqlFragments.of().addSql("with(updlock)"));
+        }
+
+        join(parameter, metadata)
+                .ifPresent(join -> fragments.addBlock(FragmentBlock.join, join));
+        //where
+
+        where(parameter, metadata)
+                .ifPresent(where ->
+                                   fragments.addBlock(FragmentBlock.where, "where")
+                                            .addBlock(FragmentBlock.where, where));
+
+        //group by
+
+        //having
+
+        //order by
+        orderBy(parameter, metadata)
+                .ifPresent(order -> fragments.addBlock(FragmentBlock.orderBy, "order by")
+                                             .addBlock(FragmentBlock.orderBy, order));
+
+        //分页
+        if (!Boolean.TRUE.equals(parameter.getForUpdate()) && parameter.getPageIndex() != null && parameter.getPageSize() != null) {
+            return metadata.<Paginator>findFeature(RDBFeatureType.paginator.getId())
+                           .map(paginator -> paginator.doPaging(fragments, parameter.getPageIndex(), parameter.getPageSize()))
+                           .map(SqlFragments::toRequest)
+                           .orElseGet(fragments::toRequest);
+
+        }
+        return fragments.toRequest();
+    }
+}

hsweb-easy-orm-rdb/src/main/java/org/hswebframework/ezorm/rdb/supports/mssql/SqlServerSchemaMetadata.java

@@ -15,6 +15,7 @@ public SqlServerSchemaMetadata(String name) {
         addFeature(new SqlServer2012TableMetadataParser(this));
         addFeature(new SqlServerIndexMetadataParser(this));
         addFeature(Dialect.MSSQL);
+        addFeature(new SqlServerQuerySqlBuilder(this));
     }
 
     @Override

hsweb-easy-orm-rdb/src/test/java/org/hswebframework/ezorm/rdb/supports/BasicReactiveTests.java

@@ -193,6 +193,19 @@ public void testReactivePager() {
 
     }
 
+    @Test
+    public void testForUpdate(){
+        repository
+                .createQuery()
+                .where(BasicTestEntity::getId, "forupdate")
+                .select("id", "name")
+                .forUpdate()
+                .fetch()
+                .then()
+                .as(StepVerifier::create)
+                .verifyComplete();
+    }
+
     @Test
     public void testInsertMerge() {