introduce repo sync leave

Author: tianyuan129

repo/repo_mgmt.go

@@ -21,6 +21,11 @@ func (r *Repo) onMgmtCmd(_ enc.Name, wire enc.Wire, reply func(enc.Wire) error)
 		return
 	}
 
+	if cmd.SyncLeave != nil {
+		go r.handleSyncLeave(cmd.SyncLeave, reply)
+		return
+	}
+
 	log.Warn(r, "Unknown management command received")
 }
 
@@ -42,6 +47,25 @@ func (r *Repo) handleSyncJoin(cmd *tlv.SyncJoin, reply func(enc.Wire) error) {
 	reply(res.Encode())
 }
 
+// (AI GENERATED DESCRIPTION): Handles a SyncLeave command by stopping an SVS session for the specified group when the protocol is `SyncProtocolSvsV3`; otherwise, responds with an error status.
+func (r *Repo) handleSyncLeave(cmd *tlv.SyncLeave, reply func(enc.Wire) error) {
+	res := tlv.RepoCmdRes{Status: 200}
+
+	if cmd.Protocol != nil && cmd.Protocol.Name.Equal(tlv.SyncProtocolSvsV3) {
+		if err := r.stopSvs(cmd); err != nil {
+			res.Status = 500
+			res.Message = err.Error()
+			log.Error(r, "Failed to stop SVS", "err", err)
+		}
+		reply(res.Encode())
+		return
+	}
+
+	log.Warn(r, "Unknown sync protocol specified in command", "protocol", cmd.Protocol)
+	res.Status = 400
+	reply(res.Encode())
+}
+
 // (AI GENERATED DESCRIPTION): Starts a SyncJoin session for the specified group, initializing a new RepoSvs if one isn’t already active and storing it in the repository’s group session map.
 func (r *Repo) startSvs(cmd *tlv.SyncJoin) error {
 	if cmd.Group == nil || len(cmd.Group.Name) == 0 {
@@ -66,3 +90,29 @@ func (r *Repo) startSvs(cmd *tlv.SyncJoin) error {
 
 	return nil
 }
+
+// stopSvs stops an SVS instance for the specified group.
+func (r *Repo) stopSvs(cmd *tlv.SyncLeave) error {
+	if cmd.Group == nil || len(cmd.Group.Name) == 0 {
+		return fmt.Errorf("missing group name")
+	}
+
+	hash := cmd.Group.Name.TlvStr()
+
+	r.mutex.Lock()
+	svs, ok := r.groupsSvs[hash]
+	r.mutex.Unlock()
+	if !ok {
+		return fmt.Errorf("group not joined")
+	}
+
+	if err := svs.Stop(); err != nil {
+		return err
+	}
+
+	r.mutex.Lock()
+	delete(r.groupsSvs, hash)
+	r.mutex.Unlock()
+
+	return nil
+}

repo/tlv/definitions.go

@@ -15,6 +15,8 @@ var SyncProtocolSvsV3 = enc.Name{
 type RepoCmd struct {
 	//+field:struct:SyncJoin
 	SyncJoin *SyncJoin `tlv:"0x1DB0"`
+	//+field:struct:SyncLeave
+	SyncLeave *SyncLeave `tlv:"0x1DB1"`
 	//+field:struct:BlobFetch
 	BlobFetch *BlobFetch `tlv:"0x1DB2"`
 }
@@ -37,6 +39,13 @@ type SyncJoin struct {
 	HistorySnapshot *HistorySnapshotConfig `tlv:"0x1A4"`
 }
 
+type SyncLeave struct {
+	//+field:struct:spec.NameContainer
+	Protocol *spec.NameContainer `tlv:"0x191"`
+	//+field:struct:spec.NameContainer
+	Group *spec.NameContainer `tlv:"0x193"`
+}
+
 type HistorySnapshotConfig struct {
 	//+field:natural
 	Threshold uint64 `tlv:"0x1A5"`