Merge pull request #32 from aliyun/feature/holo_registration

feat: add datasource removal functions

datasource/hologres/hologres.go

@@ -77,3 +77,16 @@ func RegisterHologres(name, dsn string) {
 	hologresInstances[name] = m
 
 }
+
+func RemoveHologres(name string) {
+	hologres, ok := hologresInstances[name]
+	if !ok {
+		return
+	}
+
+	if hologres.DB != nil {
+		hologres.DB.Close()
+	}
+
+	delete(hologresInstances, name)
+}

datasource/tablestore/tablestore.go

@@ -35,6 +35,14 @@ func GetTableStoreClient(name string) (*TableStoreClient, error) {
 	return tablestoreInstances[name], nil
 }
 
+func DeleteTableStoreClient(name string) {
+	if _, ok := tablestoreInstances[name]; !ok {
+		return
+	}
+
+	delete(tablestoreInstances, name)
+}
+
 func (o *TableStoreClient) GetClient() *tablestore.TableStoreClient {
 	return o.client
 }