resubmit漸進式防重復提交框架示例

Posted on by

resubmit

resubmit 是一款為 java 設計的漸進式防止重復提交框架。

推薦閱讀：

面試官：你們的項目中是怎麼做防止重復提交的？

創作目的

有時候手動加防止重復提交很麻煩，每次手動編寫不利於復用。

所以希望從從簡到繁實現一個工具，便於平時使用。

特性

  • 漸進式實現，可獨立 spring 使用
  • 基於註解+字節碼，配置靈活
  • 支持編程式的調用
  • 支持註解式，完美整合 spring
  • 支持整合 spring-boot

變更日志

maven 引入

<dependency>
    <group>com.github.houbb</group>
    <artifact>resubmit-core</artifact>
    <version>1.0.0</version>
</dependency>

編碼

  • UserService.java

@Resubmit 對應的屬性如下：

屬性 說明 默認值
value() 多久內禁止重復提交，單位為毫秒。 60000 
@Resubmit(5000)
public void queryInfo(final String id) {
    System.out.println("query info: " + id);
}
  • 測試代碼

如果在指定時間差內，重復請求，則會拋出異常 ResubmitException

@Test(expected = ResubmitException.class)
public void errorTest() {
    UserService service = ResubmitProxy.getProxy(new UserService());
    service.queryInfo("1");
    service.queryInfo("1");
}

相同的參數直接提交2次，就會報錯。

  • 測試場景2

如果等待超過指定的 5s，就不會報錯。

@Test
public void untilTtlTest() {
    UserService service = ResubmitProxy.getProxy(new UserService());
    service.queryInfo("1");
    DateUtil.sleep(TimeUnit.SECONDS, 6);
    service.queryInfo("1");
}

自定義

ResubmitProxy.getProxy(new UserService());

可以獲取 UserService 對應的代理。

等價於：

ResubmitBs resubmitBs = ResubmitBs.newInstance()
                .cache(new CommonCacheServiceMap())
                .keyGenerator(new KeyGenerator())
                .tokenGenerator(new HttpServletRequestTokenGenerator());
UserService service = ResubmitProxy.getProxy(new UserService(), resubmitBs);

其中 ResubmitBs 作為引導類，對應的策略都支持自定義。

屬性 說明 默認值
cache() 緩存實現策略 默認為基於 ConcurrentHashMap 實現的基於內存的緩存實現
keyGenerator() key 實現策略，用於唯一標識一個方法+參數，判斷是否為相同的提交 md5 策略
tokenGenerator() token 實現策略，用於唯一標識一個用戶。 從 HttpServletRequest 中的 header 屬性 resubmit_token 中獲取

spring 整合使用

maven 引入

<dependency>
    <group>com.github.houbb</group>
    <artifact>resubmit-spring</artifact>
    <version>1.0.0</version>
</dependency>

代碼編寫

  • UserService.java
@Service
public class UserService {
    @Resubmit(5000)
    public void queryInfo(final String id) {
        System.out.println("query info: " + id);
    }
}
  • SpringConfig.java
@ComponentScan("com.github.houbb.resubmit.test.service")
@EnableResubmit
@Configuration
public class SpringConfig {
}

@EnableResubmit 註解說明

@EnableResubmit 中用戶可以指定對應的實現策略，便於更加靈活的適應業務場景。

和 ResubmitBs 中支持自定義的屬性一一對應。

屬性 說明 默認值
cache() 緩存實現策略 默認為基於 ConcurrentHashMap 實現的基於內存的緩存實現
keyGenerator() key 實現策略，用於唯一標識一個方法+參數，判斷是否為相同的提交 md5 策略
tokenGenerator() token 實現策略，用於唯一標識一個用戶。 從 HttpServletRequest 中的 header 屬性 resubmit_token 中獲取

測試代碼

@ContextConfiguration(classes = SpringConfig.class)
@RunWith(SpringJUnit4ClassRunner.class)
public class ResubmitSpringTest {
    @Autowired
    private UserService service;
    @Test(expected = ResubmitException.class)
    public void queryTest() {
        service.queryInfo("1");
        service.queryInfo("1");
    }
}

整合 spring-boot

maven 引入

<dependency>
    <groupId>com.github.houbb</groupId>
    <artifactId>resubmit-springboot-starter</artifactId>
    <version>1.0.0</version>
</dependency>

代碼實現

  • UserService.java

這個方法實現和前面的一樣。

@Service
public class UserService {
    @Resubmit(5000)
    public void queryInfo(final String id) {
        System.out.println("query info: " + id);
    }
}
  • Application.java

啟動入口

@SpringBootApplication
public class ResubmitApplication {
    public static void main(String[] args) {
        SpringApplication.run(ResubmitApplication.class, args);
    }
}

測試代碼

@ContextConfiguration(classes = ResubmitApplication.class)
@RunWith(SpringJUnit4ClassRunner.class)
public class ResubmitSpringBootStarterTest {
    @Autowired
    private UserService service;
    @Test(expected = ResubmitException.class)
    public void queryTest() {
        service.queryInfo("1");
        service.queryInfo("1");
    }
}

自定義策略

上面提到 @EnableResubmit 中的策略支持自定義。

此處僅以 cache 為例，為瞭簡單，默認是基於本地內存的緩存實現。

如果你不是單點應用，那麼基於 redis 的緩存更加合適

自定義緩存 cache

實現緩存

隻需要實現 ICommonCacheService 接口即可。

public class MyDefineCache extends CommonCacheServiceMap {
    // 這裡隻是作為演示，實際生產建議使用 redis 作為統一緩存
    @Override
    public synchronized void set(String key, String value, long expireMills) {
        System.out.println("------------- 自定義的設置實現");
        super.set(key, value, expireMills);
    }
}

core 中指定使用

在非 spring 項目中，可以在引導類中指定我們定義的緩存。

ResubmitBs resubmitBs = ResubmitBs.newInstance()
                .cache(new MyDefineCache());
UserService service = ResubmitProxy.getProxy(new UserService(), resubmitBs);

其他使用方式保持不變。

spring 中指定使用

在 spring 項目中，我們需要調整一下配置，其他不變。

@ComponentScan("com.github.houbb.resubmit.test.service")
@Configuration
@EnableResubmit(cache = "myDefineCache")
public class SpringDefineConfig {
    @Bean("myDefineCache")
    public ICommonCacheService myDefineCache() {
        return new MyDefineCache();
    }
}

@EnableResubmit(cache = "myDefineCache") 指定我們自定義的緩存策略名稱。

Redis 的內置緩存策略

為瞭便於復用，基於 redis 的緩存策略已實現，後續有時間進行講解。

開源地址

為瞭便於大傢學習使用，目前防重復提交框架已開源。

歡迎大傢 fork+star，鼓勵一下老馬~

github.com/houbb/resub…

以上就是resubmit漸進式防重復提交框架示例的詳細內容，更多關於resubmit防重復提交的資料請關註WalkonNet其它相關文章！

推薦閱讀：