Merge pull request #2 from chenmin1992/release_topom

Remove my own topom

Author: hiqsociety

pkg/models/store.go

@@ -123,6 +123,19 @@ func (s *Store) Release() error {
 	return s.client.Delete(s.LockPath())
 }
 
+func (s *Store) ReleaseByToken(token string) error {
+	if b, err := s.client.Read(s.LockPath(), false); err != nil {
+		return err
+	} else if b != nil {
+		if t, err := Decode(b); err != nil {
+			return err
+		} else if t.Token == token {
+			return s.Release()
+		}
+	}
+	return nil
+}
+
 func (s *Store) LoadTopom(must bool) (*Topom, error) {
 	return LoadTopom(s.client, s.product, must)
 }

pkg/models/topom.go

@@ -18,3 +18,11 @@ type Topom struct {
 func (t *Topom) Encode() []byte {
 	return jsonEncode(t)
 }
+
+func Decode(b []byte) (*Topom, error) {
+	s := &Topom{}
+	if err := jsonDecode(s, b); err != nil {
+		return nil, err
+	}
+	return s, nil
+}

pkg/topom/topom.go

@@ -168,7 +168,7 @@ func (s *Topom) Close() error {
 	defer s.store.Close()
 
 	if s.online {
-		if err := s.store.Release(); err != nil {
+		if err := s.store.ReleaseByToken(s.model.Token); err != nil {
 			log.ErrorErrorf(err, "store: release lock of %s failed", s.config.ProductName)
 			return errors.Errorf("store: release lock of %s failed", s.config.ProductName)
 		}