Exemple #1
0
func TestIdGeneratorNoneEmpty(t *testing.T) {
	// prepare data first.
	id := struct {
		ID uint64 `json:"id"`
	}{4}
	// save the id into local disk
	path := filepath.Join(orderDir, "test1_sky."+idExt)
	err := util.SaveJSON(path, id, 0600)
	assert.Nil(t, err)

	// remove file
	defer os.RemoveAll(filepath.Join(orderDir, "test1_sky."+idExt))

	idg := newIDGenerator("test1/sky")
	closing := make(chan bool)
	go idg.Run(closing)
	for i := 1; i < 10; i++ {
		nid := idg.GetID()
		expected := uint64(4 + i)
		if nid != expected {
			t.Fatal("id error")
		}
	}
	close(closing)
	// check the file value.
	time.Sleep(500 * time.Millisecond)
	err = util.LoadJSON(path, &id)
	assert.Nil(t, err)
	if id.ID != 13 {
		t.Fatal("sync file failed")
	}
}
Exemple #2
0
// Saves known peers to disk as a newline delimited list of addresses to
// <dir><PeerDatabaseFilename>
func (self Peerlist) Save(dir string) error {

	logger.Debug("PEX: SavingPeerList")

	filename := PeerDatabaseFilename
	fn := filepath.Join(dir, filename)
	err := util.SaveJSON(fn, self, 0600)
	if err != nil {
		logger.Notice("SavePeerList Failed: %s", err)
	}
	return err
}
Exemple #3
0
func (ig IDGenerator) Run(closing chan bool) {
	id := struct {
		ID uint64 `json:"id"`
	}{}
	if _, err := os.Stat(ig.Path); !os.IsNotExist(err) {
		if err := util.LoadJSON(ig.Path, &id); err != nil {
			panic(err)
		}
	}
	for {
		select {
		case <-closing:
			return
		default:
			id.ID += 1
			ig.IDC <- id.ID

			if err := util.SaveJSON(ig.Path, id, 0600); err != nil {
				panic(err)
			}
		}
	}
}
Exemple #4
0
// Run start the manager, tm is the match tick time, closing is used for stopping the manager from running.
func (m *Manager) Start(tm time.Duration, closing chan bool) {
	// start the id generators
	for _, g := range m.idg {
		go g.Run(closing)
	}

	// start the match timer.
	wg := sync.WaitGroup{}
	for p, bk := range m.books {
		wg.Add(1)
		go func(cp string, b *Book, orderChan chan Order, c chan bool, w *sync.WaitGroup) {
			orders := []Order{}
			for {
				select {
				case <-c:
					w.Done()
					return
				case <-time.After(tm):
					orders = b.Match()
					for _, o := range orders {
						orderChan <- o
					}
					// update order book in local disk.
					pairs := strings.Split(cp, "/")
					if len(pairs) != 2 {
						panic("error coin pair name")
					}
					filename := strings.Join(pairs, "_")
					if err := util.SaveJSON(filepath.Join(orderDir, filename+"."+orderExt), b.Copy().ToMarshalable(), 0600); err != nil {
						panic(err)
					}
				}
			}
		}(p, bk, m.chans[p], closing, &wg)
	}
	wg.Wait()
}
func TestLoadManager(t *testing.T) {
	// prepare data
	coinPair := []string{"test", "sky"}
	bk := Book{}
	var BidOrderList = []Order{
		Order{ID: 1, Type: Bid, Price: 100, CreatedAt: 132424, Amount: 1, RestAmt: 1},
		Order{ID: 2, Type: Bid, Price: 102, CreatedAt: 132425, Amount: 1, RestAmt: 1},
		Order{ID: 3, Type: Bid, Price: 103, CreatedAt: 132428, Amount: 1, RestAmt: 1},
		Order{ID: 4, Type: Bid, Price: 101, CreatedAt: 132429, Amount: 1, RestAmt: 1},
		Order{ID: 5, Type: Bid, Price: 103, CreatedAt: 132430, Amount: 1, RestAmt: 1},
	}

	var AskOrderList = []Order{
		Order{ID: 6, Type: Ask, Price: 100, CreatedAt: 132424, Amount: 1, RestAmt: 1},
		Order{ID: 7, Type: Ask, Price: 102, CreatedAt: 132425, Amount: 1, RestAmt: 1},
		Order{ID: 8, Type: Ask, Price: 101, CreatedAt: 132429, Amount: 1, RestAmt: 1},
		Order{ID: 9, Type: Ask, Price: 103, CreatedAt: 132428, Amount: 1, RestAmt: 1},
		Order{ID: 10, Type: Ask, Price: 103, CreatedAt: 132438, Amount: 1, RestAmt: 1},
	}

	for _, od := range BidOrderList {
		bk.AddBid(od)
	}

	for _, od := range AskOrderList {
		bk.AddAsk(od)
	}

	// write book to files.
	err := util.SaveJSON(filepath.Join(orderDir, strings.Join(coinPair, "_")+"."+orderExt), bk.ToMarshalable(), 0600)
	assert.Nil(t, err)
	m, err := LoadManager()
	assert.Nil(t, err)
	bk1 := m.GetBook(strings.Join(coinPair, "/"))
	assert.Equal(t, bk, bk1)
}
Exemple #6
0
// Saves to filename
func (self *ReadableWallet) Save(filename string) error {
	logger.Info("Saving readable wallet to %s with filename %s", filename,
		self.Filename)
	return util.SaveJSON(filename, self, 0600)
}
Exemple #7
0
// persistance to disc. Save as JSON
func (self *ExchangeAccountManager) save() error {
	logger.Debug("save accounts")
	a := self.ToMarshalable()
	// for self.Accounts
	return util.SaveJSON(filepath.Join(acntDir, acntName), a, 0600)
}
Exemple #8
0
// Saves to filename
func (self *ReadableWallet) Save(filename string) error {
	return util.SaveJSON(filename, self, 0600)
}