Skip to content

Commit 56f0bae

Browse files
unit-admunit-adm
committed
[unitdb] restructure server code and more examples
1 parent a9b2814 commit 56f0bae

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

53 files changed

+135
-76
lines changed

‎examples/pubsub/main.go‎

Lines changed: 59 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,59 @@
1+
package main
2+
3+
import (
4+
"context"
5+
"fmt"
6+
"log"
7+
"os"
8+
"time"
9+
10+
unitdb "github.com/unit-io/unitdb-go"
11+
)
12+
13+
var f unitdb.MessageHandler = func(client unitdb.Client, msg unitdb.Message) {
14+
fmt.Printf("TOPIC: %s\n", msg.Topic())
15+
fmt.Printf("MSG: %s\n", msg.Payload())
16+
}
17+
18+
func main() {
19+
client, err := unitdb.NewClient(
20+
"grpc://localhost:6061",
21+
"UCBFDONCNJLaKMCAIeJBaOVfbAXUZHNPLDKKLDKLHZHKYIZLCDPQ",
22+
unitdb.WithInsecure(),
23+
unitdb.WithKeepAlive(2*time.Second),
24+
unitdb.WithPingTimeout(1*time.Second),
25+
unitdb.WithDefaultMessageHandler(f),
26+
)
27+
if err != nil {
28+
log.Fatalf("err: %s", err)
29+
}
30+
ctx := context.Background()
31+
err = client.ConnectContext(ctx)
32+
if err != nil {
33+
log.Fatalf("err: %s", err)
34+
}
35+
36+
r := client.Subscribe("teams.alpha.user1")
37+
if _, err := r.Get(ctx, 1*time.Second); err != nil {
38+
fmt.Println(err)
39+
os.Exit(1)
40+
}
41+
42+
for i := 0; i < 5; i++ {
43+
msg := fmt.Sprintf("Hi #%d time!", i)
44+
r := client.Publish("teams.alpha.user1", msg)
45+
if _, err := r.Get(ctx, 1*time.Second); err != nil {
46+
log.Fatalf("err: %s", err)
47+
}
48+
}
49+
50+
wait := time.NewTicker(1 * time.Second)
51+
<-wait.C
52+
r = client.Unsubscribe("teams.alpha.user1")
53+
if _, err := r.Get(ctx, 1*time.Second); err != nil {
54+
fmt.Println(err)
55+
os.Exit(1)
56+
}
57+
wait.Stop()
58+
client.Disconnect()
59+
}

server/cluster.go renamed to server/internal/cluster.go

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
package main
1+
package internal
22

33
import (
44
"encoding/gob"
@@ -10,13 +10,13 @@ import (
1010
"sync"
1111
"time"
1212

13-
"github.com/unit-io/unitdb/server/message"
14-
"github.com/unit-io/unitdb/server/message/security"
15-
lp "github.com/unit-io/unitdb/server/net"
16-
"github.com/unit-io/unitdb/server/net/listener"
17-
rh "github.com/unit-io/unitdb/server/pkg/hash"
18-
"github.com/unit-io/unitdb/server/pkg/log"
19-
"github.com/unit-io/unitdb/server/pkg/uid"
13+
"github.com/unit-io/unitdb/server/internal/message"
14+
"github.com/unit-io/unitdb/server/internal/message/security"
15+
lp "github.com/unit-io/unitdb/server/internal/net"
16+
"github.com/unit-io/unitdb/server/internal/net/listener"
17+
rh "github.com/unit-io/unitdb/server/internal/pkg/hash"
18+
"github.com/unit-io/unitdb/server/internal/pkg/log"
19+
"github.com/unit-io/unitdb/server/internal/pkg/uid"
2020
)
2121

2222
const (

server/cluster_leader.go renamed to server/internal/cluster_leader.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
package main
1+
package internal
22

33
import (
44
"log"

server/config/config.go renamed to server/internal/config/config.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ package config
33
import (
44
"encoding/json"
55

6-
"github.com/unit-io/unitdb/server/pkg/log"
6+
"github.com/unit-io/unitdb/server/internal/pkg/log"
77
)
88

99
const (

server/conn.go renamed to server/internal/conn.go

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
package main
1+
package internal
22

33
import (
44
"encoding/binary"
@@ -10,14 +10,14 @@ import (
1010
"sync"
1111
"time"
1212

13-
"github.com/unit-io/unitdb/server/message"
14-
"github.com/unit-io/unitdb/server/message/security"
15-
lp "github.com/unit-io/unitdb/server/net"
16-
"github.com/unit-io/unitdb/server/net/grpc"
17-
"github.com/unit-io/unitdb/server/pkg/log"
18-
"github.com/unit-io/unitdb/server/pkg/uid"
19-
"github.com/unit-io/unitdb/server/store"
20-
"github.com/unit-io/unitdb/server/types"
13+
"github.com/unit-io/unitdb/server/internal/message"
14+
"github.com/unit-io/unitdb/server/internal/message/security"
15+
lp "github.com/unit-io/unitdb/server/internal/net"
16+
"github.com/unit-io/unitdb/server/internal/net/grpc"
17+
"github.com/unit-io/unitdb/server/internal/pkg/log"
18+
"github.com/unit-io/unitdb/server/internal/pkg/uid"
19+
"github.com/unit-io/unitdb/server/internal/store"
20+
"github.com/unit-io/unitdb/server/internal/types"
2121
)
2222

2323
type Conn struct {

server/conn_cache.go renamed to server/internal/conn_cache.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
1-
package main
1+
package internal
22

33
import (
44
"sync"
55

6-
"github.com/unit-io/unitdb/server/pkg/uid"
6+
"github.com/unit-io/unitdb/server/internal/pkg/uid"
77
)
88

99
type ConnCache struct {

server/db/adapter.go renamed to server/internal/db/adapter.go

File renamed without changes.

server/db/unitdb/adapter.go renamed to server/internal/db/unitdb/adapter.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,8 @@ import (
88

99
"github.com/unit-io/unitdb"
1010
"github.com/unit-io/unitdb/memdb"
11-
"github.com/unit-io/unitdb/server/pkg/log"
12-
"github.com/unit-io/unitdb/server/store"
11+
"github.com/unit-io/unitdb/server/internal/pkg/log"
12+
"github.com/unit-io/unitdb/server/internal/store"
1313
)
1414

1515
const (

server/globals.go renamed to server/internal/globals.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
package main
1+
package internal
22

33
var Globals struct {
44
Cluster *Cluster

server/hdl_conn.go renamed to server/internal/hdl_conn.go

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
package main
1+
package internal
22

33
import (
44
"bufio"
@@ -7,15 +7,15 @@ import (
77
"fmt"
88
"time"
99

10-
"github.com/unit-io/unitdb/server/message"
11-
"github.com/unit-io/unitdb/server/message/security"
12-
lp "github.com/unit-io/unitdb/server/net"
13-
"github.com/unit-io/unitdb/server/pkg/crypto"
14-
"github.com/unit-io/unitdb/server/pkg/log"
15-
"github.com/unit-io/unitdb/server/pkg/stats"
16-
"github.com/unit-io/unitdb/server/pkg/uid"
17-
"github.com/unit-io/unitdb/server/store"
18-
"github.com/unit-io/unitdb/server/types"
10+
"github.com/unit-io/unitdb/server/internal/message"
11+
"github.com/unit-io/unitdb/server/internal/message/security"
12+
lp "github.com/unit-io/unitdb/server/internal/net"
13+
"github.com/unit-io/unitdb/server/internal/pkg/crypto"
14+
"github.com/unit-io/unitdb/server/internal/pkg/log"
15+
"github.com/unit-io/unitdb/server/internal/pkg/stats"
16+
"github.com/unit-io/unitdb/server/internal/pkg/uid"
17+
"github.com/unit-io/unitdb/server/internal/store"
18+
"github.com/unit-io/unitdb/server/internal/types"
1919
)
2020

2121
const (

0 commit comments

Comments
 (0)