FilesExpand file tree

 main

/

queue.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

86 lines (75 loc) · 1.62 KB

 main

/

queue.go

Copy path

Top

File metadata and controls

• Code
• Blame

86 lines (75 loc) · 1.62 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

package goqueue

import (

"context"

"errors"

"fmt"

)

var (

ErrorQueueFull = errors.New("queue was full")

ErrorQueueEmpty = errors.New("queue was empty")

)

type Config struct {

PushBlocking bool

PopBlocking bool

MaxBuffer int64

}

type InMemoryQueue struct {

Config

receiver chan Task

}

func NewInMemoryQueue(config Config) *InMemoryQueue {

return &InMemoryQueue{

Config: config,

receiver: make(chan Task, config.MaxBuffer),

}

}

func (i *InMemoryQueue) Push(ctx context.Context, task Task) error {

if i.PushBlocking {

return i.BlockingPush(ctx, task)

} else { // non-blocking

return i.NonBlockingPush(ctx, task)

}

}

func (i *InMemoryQueue) BlockingPush(ctx context.Context, task Task) error {

select {

case <-ctx.Done():

return ctx.Err()

case i.receiver <- task:

return nil

}

}

func (i *InMemoryQueue) NonBlockingPush(ctx context.Context, task Task) error {

select {

case <-ctx.Done():

return ctx.Err()

case i.receiver <- task:

return nil

default:

return fmt.Errorf("push task failed, %w", ErrorQueueFull)

}

}

func (i *InMemoryQueue) Pop(ctx context.Context) (Task, error) {

if i.PopBlocking {

return i.BlockingPop(ctx)

} else {

return i.NonBlockingPop(ctx)

}

}

func (i *InMemoryQueue) BlockingPop(ctx context.Context) (Task, error) {

select {

case <-ctx.Done():

return nil, ctx.Err()

case task := <-i.receiver:

return task, nil

}

}

func (i *InMemoryQueue) NonBlockingPop(ctx context.Context) (Task, error) {

select {

case <-ctx.Done():

return nil, ctx.Err()

case task := <-i.receiver:

return task, nil

default:

return nil, fmt.Errorf("pop task failed, %w", ErrorQueueEmpty)

}

}